Merge pull request #500 from AndrewBastin/refactor/strategy-proxy-han…

…dling

Refactoring proxy handling to be done in strategies

functions/network.js

@@ -1,5 +1,4 @@
 import AxiosStrategy from "./strategies/AxiosStrategy";
-import ProxyStrategy from "./strategies/ProxyStrategy";
 import FirefoxStrategy from "./strategies/FirefoxStrategy";
 
 
@@ -10,10 +9,6 @@ const runAppropriateStrategy = (req, store) => {
     return FirefoxStrategy(req, store); 
   }
 
-  if (store.state.postwoman.settings.PROXY_ENABLED) {
-    return ProxyStrategy(req, store);
-  }
-
   return AxiosStrategy(req, store);
 }
 

functions/strategies/AxiosStrategy.js

@@ -1,8 +1,24 @@
 import axios from "axios";
 
-const axiosStrategy = async (req, _store) => {
+const axiosWithProxy = async (req, store) => {
+  const { data } = await axios.post(
+    store.state.postwoman.settings.PROXY_URL ||
+    "https://postwoman.apollotv.xyz/",
+    req
+  );
+  return data;
+}
+
+const axiosWithoutProxy = async (req, _store) => {
   const res = await axios(req);
   return res;
 };
 
+const axiosStrategy = (req, store) => {
+  if (store.state.postwoman.settings.PROXY_ENABLED) {
+    return axiosWithProxy(req, store);
+  }
+  return axiosWithoutProxy(req, store);
+}
+
 export default axiosStrategy;

functions/strategies/FirefoxStrategy.js

@@ -1,15 +1,45 @@
-const firefoxStrategy = (req, _store) => new Promise((resolve, reject) => {
 
+const firefoxWithProxy = (req, store) => new Promise((resolve, reject) => {
+  const eventListener = (event) => {
+    window.removeEventListener("firefoxExtSendRequestComplete", event);
+
+    if (event.detail.error) {
+      reject(JSON.parse(event.detail.error));
+    } else {
+      resolve(JSON.parse(event.detail.response));
+    }
+  };
+
+  window.addEventListener("firefoxExtSendRequestComplete", eventListener);
+
+  window.firefoxExtSendRequest({
+    method: "post",
+    url: store.state.postwoman.settings.PROXY_URL || "https://postwoman.apollotv.xyz/",
+    data: req
+  });
+});
+
+const firefoxWithoutProxy = (req, _store) => new Promise((resolve, reject) => {
   const eventListener = (event) => {
     window.removeEventListener("firefoxExtSendRequestComplete", eventListener);
 
-    if (event.detail.error) reject(JSON.parse(event.detail.error));
-    else resolve(JSON.parse(event.detail.response));
+    if (event.detail.error) {
+      reject(JSON.parse(event.detail.error));
+    } else {
+      resolve(JSON.parse(event.detail.response));
+    }
   };
 
   window.addEventListener("firefoxExtSendRequestComplete", eventListener);
 
   window.firefoxExtSendRequest(req);
 });
 
+const firefoxStrategy = (req, store) => {
+  if (store.state.postwoman.settings.PROXY_ENABLED) {
+    return firefoxWithProxy(req, store);
+  }
+  return firefoxWithoutProxy(req, store);
+}
+
 export default firefoxStrategy;