blob: 4210a60e398988684c2172d65578cc0907e97c9c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
# Fetch the Vimium secret, register the port received from the parent window, and stop listening for messages
# on the window object. vimiumSecret is accessible only within the current instance of Vimium. So a
# malicious host page trying to register its own port can do no better than guessing.
registerPort = (event) ->
chrome.storage.local.get "vimiumSecret", ({vimiumSecret: secret}) ->
return unless event.source == window.parent and event.data == secret
UIComponentServer.portOpen event.ports[0]
window.removeEventListener "message", registerPort
window.addEventListener "message", registerPort
UIComponentServer =
ownerPagePort: null
handleMessage: null
portOpen: (@ownerPagePort) ->
@ownerPagePort.onmessage = (event) => @handleMessage? event
@registerIsReady()
registerHandler: (@handleMessage) ->
postMessage: (message) ->
@ownerPagePort?.postMessage message
# We require both that the DOM is ready and that the port has been opened before the UI component is ready.
# These events can happen in either order. We count them, and notify the content script when we've seen
# both.
registerIsReady: do ->
uiComponentIsReadyCount =
if document.readyState == "loading"
window.addEventListener "DOMContentLoaded", -> UIComponentServer.registerIsReady()
0
else
1
-> @postMessage "uiComponentIsReady" if ++uiComponentIsReadyCount == 2
root = exports ? window
root.UIComponentServer = UIComponentServer
|
