1 { Marker } = require 'marker'
3 { interfaces: Ci } = Components
5 HTMLDocument = Ci.nsIDOMHTMLDocument
6 XULDocument = Ci.nsIDOMXULDocument
8 CONTAINER_ID = 'VimFxHintMarkerContainer'
10 createHintsContainer = (document) ->
11 container = document.createElement('div')
12 container.id = CONTAINER_ID
13 container.className = 'VimFxReset'
16 # Creates and injects hint markers into the DOM
17 injectHints = (document) ->
19 inner = (document, startIndex = 1) ->
20 # First remove previous hints container
23 # For now we aren't able to handle hint markers in XUL Documents :(
24 if document instanceof HTMLDocument# or document instanceof XULDocument
25 if document.documentElement
26 # Find and create markers
27 markers = Marker.createMarkers(document, startIndex)
29 container = createHintsContainer(document)
31 # For performance use Document Fragment
32 fragment = document.createDocumentFragment()
34 fragment.appendChild(marker.markerElement)
36 container.appendChild(fragment)
37 document.documentElement.appendChild(container)
39 for frame in document.defaultView.frames
40 markers = markers.concat(inner(frame.document, markers.length+1))
44 return inner(document)
46 # Remove previously injected hints from the DOM
47 removeHints = (document) ->
48 if container = document.getElementById(CONTAINER_ID)
49 document.documentElement.removeChild(container)
51 for frame in document.defaultView.frames
52 removeHints(frame.document)
55 exports.injectHints = injectHints
56 exports.removeHints = removeHints