From a688fde76944c8fed9dcb7ad02ad5c40f0529e39 Mon Sep 17 00:00:00 2001
From: Samuel Crypto <work@samuelallan.info>
Date: Sun, 7 Oct 2018 14:02:31 -0400
Subject: [PATCH] Added new JS file to seamlessly store and retrieve list of
 proxified containers from browser local storage

---
 src/js/proxified-containers.js | 88 ++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 src/js/proxified-containers.js

diff --git a/src/js/proxified-containers.js b/src/js/proxified-containers.js
new file mode 100644
index 0000000..1fb77f8
--- /dev/null
+++ b/src/js/proxified-containers.js
@@ -0,0 +1,88 @@
+//This object allows other scripts to access the list mapping containers to their proxies
+window.proxifiedContainers = {
+  //Resolves to a proxy object which can be used in the return of the listener required for browser.proxy.onRequest.addListener
+  retrieve: function(cookieStoreId = null) {
+    return new Promise((resolve, reject) => {
+      browser.storage.local.get("proxifiedContainersKey").then((results) => {
+          //Steps to test:
+          //1. Is result empty? If so we must inform the caller to intialize proxifiedContainersStore with some initial info.
+          //2. Is cookieStoreId null? This means the caller probably wants everything currently in the proxifiedContainersStore object store
+          //3. If there doesn't exist an entry for the associated cookieStoreId, inform the caller of this
+          //4. Normal operation - if the cookieStoreId exists in the map, we can simply resolve with the correct proxy value
+
+          var results_array = results["proxifiedContainersKey"];
+
+          if (Object.getOwnPropertyNames(results).length == 0) {
+            reject({error: "uninitialized", message: ""});
+          }
+          
+          else if(cookieStoreId == null) {
+            resolve(results_array);
+          }          
+
+
+          else {
+            var val = results_array.find(o => o.cookieStoreId === cookieStoreId);
+            if(val == null) {
+              reject({error: "doesnotexist", message: ""});
+            }
+            else {
+              resolve(val);
+            }
+          }
+
+        }, (error) => {
+          reject({error: "internal", message: error});
+        }
+      );
+    });
+  },
+  set: function(cookieStoreId, proxy, initialize = false) {
+    return new Promise((resolve, reject) => {
+      if(initialize === true) {
+        var proxifiedContainersStore = [];
+        browser.storage.local.set({proxifiedContainersKey: proxifiedContainersStore});
+      }
+
+      //Assumes proxy is a properly formatted object
+      window.proxifiedContainers.retrieve().then((proxifiedContainersStore) => {
+
+        var index = proxifiedContainersStore.findIndex(i => i.cookieStoreId === cookieStoreId);
+        if(index === -1)
+        {
+          proxifiedContainersStore.push({cookieStoreId: cookieStoreId, proxy: proxy});
+          index = proxifiedContainersStore.length - 1;
+        }
+        else
+        {
+          proxifiedContainersStore[index] = {cookieStoreId: cookieStoreId, proxy: proxy};
+        }
+        
+        browser.storage.local.set({proxifiedContainersKey: proxifiedContainersStore});
+        resolve(proxifiedContainersStore[index]);
+      }, (errorObj) => {
+        reject(errorObj);
+      });
+    });
+  },
+  parseProxy: function(proxy_str) {
+    var regexp = /(\b(\w+):(\w+)@)?(((?:\d{1,3}\.){3}\d{1,3}\b)|(\b(\w+)(\.(\w+))+))(:(\d+))?/;
+    if(regexp.test(proxy_str) !== true)
+      return false;
+
+    else
+    {
+      var matches = regexp.exec(proxy_str);
+
+      var result = {
+        type: "http",
+        host: matches[4],
+        port: parseInt(matches[11]) || 8080,
+        username: matches[2] || "",
+        password: matches[3] || ""
+      };
+
+      return result;
+    }
+  }
+};