From 008dc27f52838a1d4991e9880a71c147623b7ecd Mon Sep 17 00:00:00 2001
From: Louis Lam <louislam@users.noreply.github.com>
Date: Thu, 7 Apr 2022 23:03:45 +0800
Subject: [PATCH] Reload proxy settings for monitors in the monitorList

---
 server/proxy.js                                | 17 +++++++++++++++++
 server/socket-handlers/proxy-socket-handler.js |  6 ++++--
 2 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/server/proxy.js b/server/proxy.js
index 392a0af7..af72402d 100644
--- a/server/proxy.js
+++ b/server/proxy.js
@@ -3,6 +3,7 @@ const HttpProxyAgent = require("http-proxy-agent");
 const HttpsProxyAgent = require("https-proxy-agent");
 const SocksProxyAgent = require("socks-proxy-agent");
 const { debug } = require("../src/util");
+const server = require("./server");
 
 class Proxy {
 
@@ -144,6 +145,22 @@ class Proxy {
             httpsAgent
         };
     }
+
+    /**
+     * Reload proxy settings for current monitors
+     * @returns {Promise<void>}
+     */
+    static async reloadProxy() {
+        let updatedList = await R.getAssoc("SELECT id, proxy_id FROM monitor");
+
+        for (let monitorID in server.monitorList) {
+            let monitor = server.monitorList[monitorID];
+
+            if (updatedList[monitorID]) {
+                monitor.proxy_id = updatedList[monitorID].proxy_id;
+            }
+        }
+    }
 }
 
 /**
diff --git a/server/socket-handlers/proxy-socket-handler.js b/server/socket-handlers/proxy-socket-handler.js
index a05a00b9..817bdd49 100644
--- a/server/socket-handlers/proxy-socket-handler.js
+++ b/server/socket-handlers/proxy-socket-handler.js
@@ -1,6 +1,7 @@
 const { checkLogin } = require("../util-server");
 const { Proxy } = require("../proxy");
 const { sendProxyList } = require("../client");
+const server = require("../server");
 
 module.exports.proxySocketHandler = (socket) => {
     socket.on("addProxy", async (proxy, proxyID, callback) => {
@@ -11,7 +12,8 @@ module.exports.proxySocketHandler = (socket) => {
             await sendProxyList(socket);
 
             if (proxy.applyExisting) {
-                // TODO: await restartMonitors(socket.userID);
+                await Proxy.reloadProxy();
+                await server.sendMonitorList(socket);
             }
 
             callback({
@@ -34,7 +36,7 @@ module.exports.proxySocketHandler = (socket) => {
 
             await Proxy.delete(proxyID, socket.userID);
             await sendProxyList(socket);
-            // TODO: await restartMonitors(socket.userID);
+            await Proxy.reloadProxy();
 
             callback({
                 ok: true,