Spaces:

omnitool-ai
/

omnitool_on_hf

Running

App Files Files Community

manu-sapiens commited on Nov 29, 2023

Commit

6cb66da

1 Parent(s): d44e3b5

double button management server - working on local

Browse files

Files changed (1) hide show

myNodeServer.js +121 -36

myNodeServer.js CHANGED Viewed

@@ -10,32 +10,7 @@ const VERSION = '0.0.4a';
 console.log(`************ Omnitool Proxy Server v${VERSION} ************`);
 let omnitoolLogs = [];
-// Function to start the request forwarding server
-async function startRequestForwardingServer() {
-  const server = http.createServer((req, res) => {
-    const options = {
-      hostname: TARGET_HOST,
-      port: TARGET_PORT,
-      path: req.url,
-      method: req.method,
-      headers: req.headers,
-    };
-    const proxy = http.request(options, (proxyRes) => {
-      res.writeHead(proxyRes.statusCode, proxyRes.headers);
-      proxyRes.pipe(res, { end: true });
-    });
-    req.pipe(proxy, { end: true });
-  });
-  server.listen(PROXY_PORT, '0.0.0.0');
-  console.log(`Request forwarding server listening on port ${PROXY_PORT}`);
-}
-// Function to start the background process
-function startYarnStartProcess()
 {
     if (ALREADY_STARTING) return;
     ALREADY_STARTING = true;
@@ -43,7 +18,8 @@ function startYarnStartProcess()
     console.log('Starting Omnitool Server...');
     return new Promise((resolve, reject) =>
     {
-        const omnitoolStartProcess = spawn('./omnitool_start.sh',  {detached: true});
         omnitoolStartProcess.stdout.on('data', (data) =>
         {
             omnitoolLogs.push(data.toString());
@@ -71,22 +47,131 @@ function startYarnStartProcess()
             }
         });
     });
-    omnitoolStartProcess.unref();
 }
 // Main function to start everything
-async function startServers() {
-  try {
-    startYarnStartProcess();
-    await startRequestForwardingServer();
-  } catch (error) {
-    console.error('Failed to start servers:', error);
-  }
 }
 // Start the servers
-startServers();
 /*

 console.log(`************ Omnitool Proxy Server v${VERSION} ************`);
 let omnitoolLogs = [];
+async function startOmnitoolServer()
 {
     if (ALREADY_STARTING) return;
     ALREADY_STARTING = true;
     console.log('Starting Omnitool Server...');
     return new Promise((resolve, reject) =>
     {
+        const omnitoolStartProcess = spawn('./omnitool_start.sh');
         omnitoolStartProcess.stdout.on('data', (data) =>
         {
             omnitoolLogs.push(data.toString());
             }
         });
     });
+}
+async function startRequestForwardingServer() {
+    const server = http.createServer((req, res) => handleRoutes(req, res));
+    server.listen(PROXY_PORT, '0.0.0.0');
+    console.log(`Request forwarding server listening on port ${PROXY_PORT}`);
+  }
+async function handleRoutes(req, res)
+{
+    if (!OMNITOOL_READY && (req.method === 'GET'))
+    {
+        console.log(`Handling request: ${req.url}`);
+        switch (req.url)
+        {
+            case '/start-omnitool-server':
+                {
+                    console.log(`Omnitool Server:ALREADY_STARTING =  ${ALREADY_STARTING}`)
+                    if (ALREADY_STARTING)
+                    {
+                        res.writeHead(200, { 'Content-Type': 'text/html'});
+                        res.end("Omnitool server already starting");
+                        return;
+                    }
+                    try
+                    {
+                        await startOmnitoolServer();
+                        res.writeHead(200, { 'Content-Type': 'text/html'});
+                        res.end("Omnitool server started successfully");
+                    }
+                    catch (error)
+                    {
+                        console.error(error);
+                        ALREADY_STARTING = false;
+                        res.writeHead(500, { 'Content-Type': 'text/html'});
+                        res.end(`Error starting Omnitool server: ${error}`);
+                    }
+                    return;
+                }
+            case '/omnitool-logs':
+                {
+                    res.writeHead(200, { 'Content-Type': 'application/json'});
+                    const reply = { logs: omnitoolLogs, ready: OMNITOOL_READY };
+                    res.end(JSON.stringify(reply));
+                    return;
+                }
+            case '/':
+                {
+                    const localUrl = req.headers['host'];
+                    let htmlContent = `
+                        <html>
+                            <head><title>Proxy Server</title></head>
+                            <body>
+                                <button id="startServerButton" onclick="startServer()">Start Omnitool Server</button>
+                                <button id="exitIframeButton" onclick="window.open('http://${localUrl}', '_blank')" >GOTO ${localUrl}</button>
+                                <div id="logs" style="white-space: pre-wrap;"></div>
+                                <script>
+                                    function startServer() {
+                                        document.getElementById('startServerButton').disabled = true;
+                                        fetch('/start-omnitool-server')
+                                            .then(response => response.json())
+                                            .then(data => {
+                                                startLogPolling();
+                                            });
+                                    }
+                                    function startLogPolling() {
+                                        const interval = setInterval(() => {
+                                            fetch('/omnitool-logs')
+                                                .then(response => response.json())
+                                                .then(data => {
+                                                    document.getElementById('logs').innerText = data.logs.join("\\n");
+                                                    if (data.ready) {
+                                                        document.getElementById('logs').innerText = "Omnitool is ready! Please open the following link in a new tab: https://${localUrl}";
+                                                        clearInterval(interval);
+                                                        //window.location.reload(); // Refresh the page when Omnitool is ready
+                                                    }
+                                                });
+                                        }, 1000);
+                                    }
+                                    startLogPolling();
+                                </script>
+                            </body>
+                        </html>`;
+                    res.writeHead(200, { 'Content-Type': 'text/html' });
+                    res.end(htmlContent);
+                    return;
+                }
+            default:
+                console.log(`Doing nothing with this request: ${req.url}`);
+                res.writeHead(200, { 'Content-Type': 'text/html' });
+                res.end(`Doing nothing with this request: ${req.url}`);
+                return;
+        }
+    }
+    const options = { hostname: TARGET_HOST, port: TARGET_PORT, path: req.url, method: req.method, headers: req.headers, };
+    const proxy = http.request(options, (proxyRes) =>
+    {
+        res.writeHead(proxyRes.statusCode, proxyRes.headers);
+        proxyRes.pipe(res, { end: true });
+    });
+    req.pipe(proxy, { end: true });
+    return;
 }
 // Main function to start everything
+async function startManagementServer()
+{
+    try
+    {
+        await startRequestForwardingServer();
+    } catch (error)
+    {
+        console.error('Failed to start servers:', error);
+    }
 }
 // Start the servers
+startManagementServer();
 /*