From 3c40b52f96692017f008801dcfc73c985e391a1e Mon Sep 17 00:00:00 2001
From: eight <eight04@gmail.com>
Date: Fri, 1 Sep 2017 18:21:45 +0800
Subject: [PATCH] Add 'injectResource' message to inject js/css

---
 background/background.js | 51 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/background/background.js b/background/background.js
index 70b26e0a..61da3214 100644
--- a/background/background.js
+++ b/background/background.js
@@ -341,5 +341,56 @@ function onRuntimeMessage(request, sender, sendResponse) {
         .then(sendResponse)
         .catch(() => sendResponse(null));
       return KEEP_CHANNEL_OPEN;
+
+    case 'injectResource':
+      injectResource(request, sender.tab.id).then(sendResponse);
+      return KEEP_CHANNEL_OPEN;
+  }
+}
+
+function injectResource({resources}, tabId) {
+  return Promise.all(doInject())
+    .then(() => ({status: 'success'}))
+    .catch(err => ({status: 'error', error: err.message}));
+
+  function *doInject() {
+    for (const resource of resources) {
+      const type = resource.match(/\.\w+$/i)[0];
+      if (type === '.js') {
+        yield injectScript(resource, tabId);
+      } else if (type === '.css') {
+        yield injectStyle(resource, tabId);
+      }
+    }
+  }
+
+  function injectScript(url, tabId) {
+    return new Promise((resolve, reject) => {
+      chrome.tabs.executeScript(tabId, {
+        file: url,
+        runAt: 'document_start'
+      }, () => {
+        if (chrome.runtime.lastError) {
+          reject(chrome.runtime.lastError);
+        } else {
+          resolve();
+        }
+      });
+    });
+  }
+
+  function injectStyle(url, tabId) {
+    return new Promise((resolve, reject) => {
+      chrome.tabs.insertCSS(tabId, {
+        file: url,
+        runAt: 'document_start'
+      }, () => {
+        if (chrome.runtime.lastError) {
+          reject(chrome.runtime.lastError);
+        } else {
+          resolve();
+        }
+      });
+    });
   }
 }