javascript - hightlight选择不同div中的文本

Question

我有2个div，我想突出显示我选择的文本，无论div是什么。

我尝试使用div代替span但未向我提供我想要的结果。

代码：

function setColor() {
  var range = window.getSelection().getRangeAt(0),
    span = document.createElement('span');
  span.className = 'highlight';
  span.appendChild(range.extractContents());
  range.insertNode(span);
}

.highlight {
  background-color: rgba(250, 166, 26, .2);
  padding: 2px;
  border-radius: 2px;
}

<div id="test">
  Div 1 text
</div>
<div>
  Div 2 text
</div>
<button onClick="setColor()">Apply color</button>