1 files changed, 38 insertions, 6 deletions
diff --git a/content_scripts/hud.coffee b/content_scripts/hud.coffee
index b2780491..42a960da 100644
--- a/content_scripts/hud.coffee
+++ b/content_scripts/hud.coffee
@@ -9,6 +9,8 @@ HUD =
   findMode: null
   abandon: -> @hudUI?.hide false
 
+  pasteListener: null # Set by @pasteFromClipboard to handle the value returned by pasteResponse
+
   # This HUD is styled to precisely mimick the chrome HUD on Mac. Use the "has_popup_and_link_hud.html"
   # test harness to tweak these styles to match Chrome's. One limitation of our HUD display is that
   # it doesn't sit on top of horizontal scrollbars like Chrome's HUD does.
@@ -35,7 +37,9 @@ HUD =
       @tween.fade 1.0, 150
 
   search: (data) ->
-    @findMode.findInPlace data.query
+    # NOTE(mrmr1993): On Firefox, window.find moves the window focus away from the HUD. We use postFindFocus
+    # to put it back, so the user can continue typing.
+    @findMode.findInPlace data.query, {"postFindFocus": @hudUI.iframeElement.contentWindow}
 
     # Show the number of matches in the HUD UI.
     matchCount = if FindMode.query.parsedQuery.length > 0 then FindMode.query.matchCount else 0
@@ -66,20 +70,47 @@ HUD =
 
     focusNode = DomUtils.getSelectionFocusElement()
     document.activeElement?.blur()
-    focusNode?.focus()
+    focusNode?.focus?()
 
     if exitEventIsEnter
-      handleEnterForFindMode()
+      FindMode.handleEnter()
       if FindMode.query.hasResults
         postExit = -> new PostFindMode
     else if exitEventIsEscape
-      # We don't want FindMode to handle the click events that handleEscapeForFindMode can generate, so we
+      # We don't want FindMode to handle the click events that FindMode.handleEscape can generate, so we
       # wait until the mode is closed before running it.
-      postExit = handleEscapeForFindMode
+      postExit = FindMode.handleEscape
 
     @findMode.exit()
     postExit?()
 
+  # These commands manage copying and pasting from the clipboard in the HUD frame.
+  # NOTE(mrmr1993): We need this to copy and paste on Firefox:
+  # * an element can't be focused in the background page, so copying/pasting doesn't work
+  # * we don't want to disrupt the focus in the page, in case the page is listening for focus/blur events.
+  # * the HUD shouldn't be active for this frame while any of the copy/paste commands are running.
+  copyToClipboard: (text) ->
+    DomUtils.documentComplete =>
+      @init()
+      @hudUI?.postMessage {name: "copyToClipboard", data: text}
+
+  pasteFromClipboard: (@pasteListener) ->
+    DomUtils.documentComplete =>
+      @init()
+      # Show the HUD frame, so Firefox will actually perform the paste.
+      @hudUI.toggleIframeElementClasses "vimiumUIComponentHidden", "vimiumUIComponentVisible"
+      @tween.fade 0, 0
+      @hudUI.postMessage {name: "pasteFromClipboard"}
+
+  pasteResponse: ({data}) ->
+    # Hide the HUD frame again.
+    @hudUI.toggleIframeElementClasses "vimiumUIComponentVisible", "vimiumUIComponentHidden"
+    @unfocusIfFocused()
+    @pasteListener data
+
+  unfocusIfFocused: ->
+    document.activeElement.blur() if document.activeElement == @hudUI?.iframeElement
+
 class Tween
   opacity: 0
   intervalId: -1
@@ -125,5 +156,6 @@ class Tween
       }
     """
 
-root = exports ? window
+root = exports ? (window.root ?= {})
 root.HUD = HUD
+extend window, root unless exports?