diff options
| -rw-r--r-- | content_scripts/vimium_frontend.coffee | 2 | ||||
| -rw-r--r-- | lib/dom_utils.coffee | 7 |
2 files changed, 8 insertions, 1 deletions
diff --git a/content_scripts/vimium_frontend.coffee b/content_scripts/vimium_frontend.coffee index 351a2690..0f42c65b 100644 --- a/content_scripts/vimium_frontend.coffee +++ b/content_scripts/vimium_frontend.coffee @@ -1142,7 +1142,7 @@ CursorHider = initializePreDomReady() window.addEventListener("DOMContentLoaded", registerFrame) window.addEventListener("unload", unregisterFrame) -window.addEventListener("DOMContentLoaded", initializeOnDomReady) +DomUtils.runWhenDOMLoaded initializeOnDomReady window.onbeforeunload = -> chrome.runtime.sendMessage( diff --git a/lib/dom_utils.coffee b/lib/dom_utils.coffee index ba5e279f..762c6418 100644 --- a/lib/dom_utils.coffee +++ b/lib/dom_utils.coffee @@ -178,5 +178,12 @@ DomUtils = event.preventDefault() @suppressPropagation(event) + # Calls func either now (if the DOM has already loaded), or when the DOM is loaded. + runWhenDOMLoaded: (func) -> + if document.readyState == "loading" + window.addEventListener "DOMContentLoaded", func + else + func() + root = exports ? window root.DomUtils = DomUtils |