1 files changed, 38 insertions, 0 deletions

diff --git a/src/helpers/webview-ime-focus-helpers.js b/src/helpers/webview-ime-focus-helpers.js
new file mode 100644
index 000000000..2593a5f26
--- /dev/null
+++ b/src/helpers/webview-ime-focus-helpers.js
@@ -0,0 +1,38 @@
+module.exports.releaseDocumentFocus = () => {
+  const element = document.createElement('span');
+  document.body.appendChild(element);
+
+  const range = document.createRange();
+  range.setStart(element, 0);
+
+  const selection = window.getSelection();
+  selection.removeAllRanges();
+  selection.addRange(range);
+  selection.removeAllRanges();
+
+  document.body.removeChild(element);
+};
+
+module.exports.claimDocumentFocus = () => {
+  const { activeElement } = document;
+  const selection = window.getSelection();
+
+  let selectionStart;
+  let selectionEnd;
+  let range;
+
+  if (activeElement) ({ selectionStart, selectionEnd } = activeElement);
+  if (selection.rangeCount) range = selection.getRangeAt(0);
+
+  const restoreOriginalSelection = () => {
+    if (selectionStart >= 0 && selectionEnd >= 0) {
+      activeElement.selectionStart = selectionStart;
+      activeElement.selectionEnd = selectionEnd;
+    } else if (range) {
+      selection.addRange(range);
+    }
+  };
+
+  exports.releaseDocumentFocus();
+  window.requestAnimationFrame(restoreOriginalSelection);
+};