Basics working

hay · Dec 6, 2017 · 46985cd · 46985cd
1 parent 37c343c
commit 46985cd
Show file tree

Hide file tree

Showing 9 changed files with 124 additions and 21 deletions.
diff --git a/background.html b/background.html
@@ -0,0 +1,8 @@
+<!doctype html>
+<html>
+<head>
+</head>
+<body>
+    <script type="module" src="background.js"></script>
+</body>
+</html>
diff --git a/background.js b/background.js
@@ -1,14 +1,14 @@
-var sites = null;
-var log = console.log.bind(console);
+const log = console.log.bind(console);
+const hosts = {};
+const queries = {};
 
-function getFilterPath(url) {
-    var url = new URL(url);
-    const hostname = url.hostname.replace('www.', '');
-    return sites[hostname] ? `scripts/${sites[hostname]}.js` : false;
+function getHostname(url) {
+    url = new URL(url);
+    return url.hostname.replace('www.', '');
 }
 
 // Why doesn't Promise have this natively?
-Promise.series = function(series) {
+function runPromises(series) {
     return series.reduce((p, fn) => p.then(fn), Promise.resolve());
 };
 
@@ -20,7 +20,7 @@ function executeScripts(tabId, scripts) {
         });
     });
 
-    Promise.series(scripts).then(() => log('Loaded all scripts'));
+    runPromises(scripts).then(() => log('Loaded all scripts'));
 }
 
 function handleUpdate(tabId, changeInfo, tab) {
@@ -43,13 +43,81 @@ function handleUpdate(tabId, changeInfo, tab) {
     ]);
 }
 
+function getScriptsInDirectory(directory, callback) {
+    chrome.runtime.getPackageDirectoryEntry((entry) => {
+        entry.getDirectory(directory, { create : false }, (dir) => {
+            const reader = dir.createReader();
+            reader.readEntries((results) => {
+                results = results.map(entry => `./${directory}/${entry.name}`);
+                callback(results);
+            });
+        });
+    })
+}
+
+function executeModuleInTab(tabId, module) {
+    log(`Loading module ${module.src}`);
+
+    if (module.css) {
+        chrome.tabs.insertCSS(tabId, {
+            code : module.css
+        });
+    }
+}
+
+function parseModule(module, src) {
+    // For easy reference later
+    module.src = src;
+
+    if (module.host) {
+        hosts[module.host] = module;
+    } else if (module.query) {
+        queries[module.query] = module;
+    }
+}
+
+function setupInjector() {
+    chrome.tabs.onUpdated.addListener((tabId, change, tab) => {
+        if (change.status === 'complete') {
+            const hostname = getHostname(tab.url);
+
+            if (hostname in hosts) {
+                executeModuleInTab(tabId, hosts[hostname]);
+                return;
+            }
+
+            for (let query in queries) {
+                chrome.tabs.executeScript(tabId, {
+                    code : `!!document.querySelector('${query}');`
+                }, (result) => {
+                    if (result[0]) {
+                        executeModuleInTab(tabId, queries[query])
+                    }
+                });
+            }
+        }
+    });
+}
+
 function main() {
+    getScriptsInDirectory("scripts2", (scripts) => {
+        scripts = scripts.map((script) => {
+            return import(script).then(m => parseModule(m.default, script));
+        });
+
+        Promise.all(scripts).then(() => {
+            setupInjector();
+        });
+    });
+
+    /*
     var scriptsUrl = chrome.extension.getURL('scripts.json');
 
     fetch(scriptsUrl).then((res) => res.json()).then((scripts) => {
         sites = scripts;
         chrome.tabs.onUpdated.addListener(handleUpdate);
     });
+    */
 }
 
 main();
diff --git a/bower.json b/bower.json
diff --git a/lib/.gitignore b/lib/.gitignore
diff --git a/manifest.json b/manifest.json
@@ -11,7 +11,6 @@
         "scripts/*.js"
     ],
     "background" : {
-        "scripts" : ["background.js"],
-        "persistent" : false
+        "page" : "background.html"
     }
 }
diff --git a/scripts/revue.js b/scripts/revue.js
@@ -7,4 +7,8 @@ $("body").append(`<style>
         display: flex;
         justify-content: center;
     }
-</style>`);
+</style>`);
+
+export default {
+    'name' : 'revue'
+}
diff --git a/scripts2/flickr.js b/scripts2/flickr.js
@@ -0,0 +1,6 @@
+// Remove Spaceball to allow right-click to download photos
+export default {
+    host : 'flickr.com',
+
+    css : `.flickr-logo-container { display: none; }`
+};
diff --git a/scripts2/medium.js b/scripts2/medium.js
@@ -0,0 +1,16 @@
+// Remove nonsense
+export default {
+    query : 'meta[name="twitter:app:name:iphone"][content="Medium"]',
+
+    css : `
+        .js-stickyFooter,
+        .metabar--spacer,
+        .js-postShareWidget {
+            display: none;
+        }
+
+        .metabar {
+            position: relative !important;
+        }
+    `
+};
diff --git a/test.js b/test.js
@@ -0,0 +1,12 @@
+function load(filename) {
+    import(filename).then((test) => {
+        test = test.default;
+        console.log(test.name);
+        test.run();
+    });
+}
+
+[
+    './scripts/test.js',
+    './scripts/test2.js'
+].forEach(load);