From fc90f78e26dcf9531ec3702f99a3a45fd8cd4a18 Mon Sep 17 00:00:00 2001
From: Shin'ya Ueoka <ueokande@i-beam.org>
Date: Sat, 14 Oct 2017 13:18:46 +0900
Subject: [PATCH] load content script all_frames and click link

---
 manifest.json                    |  1 +
 src/content/components/follow.js | 44 +++++++++++++++-----------------
 2 files changed, 21 insertions(+), 24 deletions(-)

diff --git a/manifest.json b/manifest.json
index 369c5a2..19f4974 100644
--- a/manifest.json
+++ b/manifest.json
@@ -14,6 +14,7 @@
   },
   "content_scripts": [
     {
+      "all_frames": true,
       "matches": [ "http://*/*", "https://*/*" ],
       "js": [ "build/content.js" ]
     }
diff --git a/src/content/components/follow.js b/src/content/components/follow.js
index 3307893..3f28cc2 100644
--- a/src/content/components/follow.js
+++ b/src/content/components/follow.js
@@ -90,33 +90,29 @@ export default class FollowComponent {
     });
   }
 
+  openLink(element) {
+    if (!this.state.newTab) {
+      element.click();
+      return;
+    }
+
+    let href = element.getAttribute('href');
+
+    // eslint-disable-next-line no-script-url
+    if (!href || href === '#' || href.toLowerCase().startsWith('javascript:')) {
+      return;
+    }
+    return browser.runtime.sendMessage({
+      type: messages.OPEN_URL,
+      url: element.href,
+      newTab: this.state.newTab,
+    });
+  }
+
   activate(element) {
     switch (element.tagName.toLowerCase()) {
     case 'a':
-      if (this.state.newTab) {
-        // getAttribute() to avoid to resolve absolute path
-        let href = element.getAttribute('href');
-
-        // eslint-disable-next-line no-script-url
-        if (!href || href === '#' || href.startsWith('javascript:')) {
-          return;
-        }
-        return browser.runtime.sendMessage({
-          type: messages.OPEN_URL,
-          url: element.href,
-          newTab: this.state.newTab,
-        });
-      }
-      if (element.href.startsWith('http://') ||
-        element.href.startsWith('https://') ||
-        element.href.startsWith('ftp://')) {
-        return browser.runtime.sendMessage({
-          type: messages.OPEN_URL,
-          url: element.href,
-          newTab: this.state.newTab,
-        });
-      }
-      return element.click();
+      return this.openLink(element, this.state.newTab);
     case 'input':
       switch (element.type) {
       case 'file':