added more status information and a button to try to get the user's name from hf whoami-v2 endpoint

myNodeServer.js

@@ -3,7 +3,7 @@
  * All rights reserved.
  */
 //@ts-check
-const VERSION = '0.0.8d';
+const VERSION = '0.0.8e';
 
 const express = require('express');
 const http = require('http');
@@ -18,6 +18,13 @@ const CONTAINER_HOST = '127.0.0.1';
 const OMNI_URL = 'http://127.0.0.1:1688'; // URL of the OMNITOOL service
 const PROXY_PORT_OMNITOOL = 4444;
 const CONTAINER_PORT_OMNITOOL = 1688;
+const OMNITOOL_HEALTH_PING = '/api/v1/mercenaries/ping';
+
+
+const OMNITOOL_SPACE_NAME = 'omnitool_test_3';
+const OMNITOOL_SPACE_OWNER = 'manu-sapiens';
+const HF_SPACE_URL = `https://huggingface.co/spaces/${OMNITOOL_SPACE_OWNER}/${OMNITOOL_SPACE_NAME}`;
+const HF_SPACE_DUPLICATE_URL = `${HF_SPACE_URL}?duplicate=true`;
 
 const DELAY_OMNITOOL_SET_TO_RUNNING = 2000; // 2 seconds
 const CHECK_OMNI_INTERVAL = 60000; // 1 minute
@@ -32,6 +39,8 @@ global.PROTOCOL = "";
 global.PROXY_STARTED = false;
 global.logs = [];
 global.OMNITOOL_PROXY = null;
+global.CONNECTED_TO_MASTER = false;
+global.SPACE_OWNER = "";
 
 
 console.log(`************ Omnitool Proxy Server v${VERSION} ************`);
@@ -55,7 +64,22 @@ const COMMON_STYLES = `
             background-color: yellow;
             color: black;
             font-weight: bold;
-        }                    
+        }    
+        .button-like-link {
+            display: inline-block;
+            background-color: #f0f0f0; /* Button color */
+            color: #000; /* Text color */
+            padding: 10px 15px;
+            text-align: center;
+            text-decoration: none;
+            border: 1px solid #ddd;
+            border-radius: 5px;
+            font-size: 16px;
+            cursor: pointer;
+        }
+        .button-like-link:hover {
+            background-color: #e9e9e9;
+        }                     
         @keyframes pulseAnimation {
             0% { transform: scale(1); }
             50% { transform: scale(1.1); }
@@ -80,7 +104,7 @@ async function startOmnitoolServer()
             if (data.toString().includes(`Server has started and is ready to accept connections`))
             {
                 console.log('Omnitool server started successfully');
-                setOmnitoolRunning(true);   
+                setOmnitoolRunning(true);
             }
 
         });
@@ -124,7 +148,7 @@ function setOmnitoolRunning(set_running)
                 global.OMNITOOL_RUNNING = true;
             }, DELAY_OMNITOOL_SET_TO_RUNNING); // Delay by 2 second
         }
-        
+
     }
     else
     {
@@ -183,7 +207,7 @@ async function handleGetStartOmnitoolServer(req, res)
 async function handleGetOmnitoolLogs(req, res)
 {
     res.writeHead(200, { 'Content-Type': 'application/json' });
-    const reply = { logs: global.logs, ready:global.OMNITOOL_READY };
+    const reply = { logs: global.logs, ready: global.OMNITOOL_READY };
     res.end(JSON.stringify(reply));
     return;
 }
@@ -218,6 +242,19 @@ async function proxyRequest(req, res)
 
 function getButtonsString(buttonsHTML)
 {
+    // Add a button for the Hugging Face API call
+    const huggingFaceButton = '<button onclick="getHuggingFaceUserName()">Get Hugging Face User Name</button>';
+
+
+    // Message to display based on CONNECTED_TO_MASTER
+    let connectionStatusMessage;
+    if (global.CONNECTED_TO_MASTER)
+    {
+        connectionStatusMessage = '<div style="color: red;">You are now connected DIRECTLY to the REFERENCE Omnitool Space on Huggingface. It is HIGHLY recommended that you duplicate this space and make it private to secure your keys, recipes and outputs.</div>';
+    } else
+    {
+        connectionStatusMessage = `<div style="color: green;">You are now connected to the space of ${global.SPACE_OWNER}. If this is NOT YOU, it is HIGHLY recommended that you duplicate this space and make it private to secure your keys, recipes and outputs.</div><div>In addition, consider subscribing to the 20GB local storage option from Huggingface to preserve your keys and work between reboot of this server.</div>`;
+    }
 
     return `
         <html>
@@ -226,9 +263,22 @@ function getButtonsString(buttonsHTML)
                 ${COMMON_STYLES}
             </head>
             <body>
+                <h1>Proxy Server</h1>
+                <p>Version: ${VERSION}</p>
+                <p>Local URL: ${global.LOCAL_URL}</p>
+                <p>Space Owner: ${global.SPACE_OWNER}</p>
+                <p>Connected to master: ${global.CONNECTED_TO_MASTER}</p>
+                <p>Omni URL: ${OMNI_URL}</p>
+                <p>Omni Running: ${global.OMNITOOL_RUNNING}</p>
+                <p>Omni Ready: ${global.OMNITOOL_READY}</p>
+               
+                ${connectionStatusMessage} <!-- Display the connection status message -->
                 ${buttonsHTML}
+                ${huggingFaceButton}
                 ${LOG_CONTAINER_HTML}
                 <script>
+                    // Script to check Hugging Face login status and control 'Clone Repo' button
+                
                     function startServer() {
                         document.getElementById('startServerButton').classList.remove
                         ('highlight-button');
@@ -273,15 +323,34 @@ function getButtonsString(buttonsHTML)
                         element.scrollTop = element.scrollHeight;
                     }
                     
+                    // Function to handle the Hugging Face API call
+                    function getHuggingFaceUserName() {
+                        fetch('https://huggingface.co/api/whoami-v2')
+                            .then(response => response.json())
+                            .then(data => {
+                                if(data && data.name) {
+                                    alert('Hugging Face User Name: ' + data.name);
+                                } else {
+                                    alert('No name field in the response.');
+                                }
+                            })
+                            .catch(error => {
+                                console.error('Error fetching Hugging Face user name:', error);
+                                alert('Error fetching data.');
+                            });
+                    }
+
                     startLogPolling();
                 </script>
             </body>
         </html>`;
 }
 
+
 async function handleGetRoot(req, res)
 {
-    if (global.LOCAL_URL === "") setGlobals(req);
+    setGlobals(req);
+    
     console.log(`req.session.isVisited = ${req.session.isVisited}`);
     console.log(`global.OMNITOOL_RUNNING = ${global.OMNITOOL_RUNNING}`);
 
@@ -292,13 +361,16 @@ async function handleGetRoot(req, res)
 
     if (!global.OMNITOOL_RUNNING)
     {
-        console.log('Omnitool server is not running')
+        console.log('Omnitool server is not running');
         let startButtonClass = '';
+        let cloneButtonClass = '';
         if (!global.ALREADY_STARTING) startButtonClass = 'highlight-button';
         const gotoButtonClass = '';
-        
-        let buttonsHTML = `<button id="startServerButton" class="${startButtonClass}" onclick="startServer()">Start Omnitool Server</button>`;
-        buttonsHTML += `<button id="exitIframeButton" class="${gotoButtonClass}" onclick="exitIframe()">GOTO OMNITOOL</button>`;
+
+        let buttonsHTML = `
+        <button id="startServerButton" class="${startButtonClass}" onclick="startServer()">Start Omnitool Server</button>
+        <button id="exitIframeButton" class="${gotoButtonClass}" onclick="exitIframe()">GOTO OMNITOOL</button>
+        <a href="${HF_SPACE_DUPLICATE_URL}" target="_blank" class="button-like-link" id="duplicateRepoButton">DUPLICATE Repo</a>`;
 
         const html = getButtonsString(buttonsHTML);
         res.writeHead(200, { 'Content-Type': 'text/html' });
@@ -309,7 +381,7 @@ async function handleGetRoot(req, res)
 
     if (!req.session.isVisited)
     {
-        console.log('First time visitor')
+        console.log('First time visitor');
         const gotoButtonClass = 'highlight-button';
         const buttonsHTML = `<button id="exitIframeButton" class="${gotoButtonClass}" onclick="exitIframe()">GOTO OMNITOOL</button>`;
         const html = getButtonsString(buttonsHTML);
@@ -321,17 +393,36 @@ async function handleGetRoot(req, res)
 
     proxyRequest(req, res);
 }
+
+function extractSpaceOwner(url, spaceName)
+{
+    const regex = new RegExp(`^(?:https?:\\/\\/)?([^\\/]*?)-${spaceName}\\.hf\\.space\\/?$`);
+    const match = url.match(regex);
+    return match ? match[1] : null;
+}
+
 function setGlobals(req)
 {
-    
+    console.log(`Setting global using req.headers['host'] = ${req.headers['host']}`);
     global.LOCAL_URL = req.headers['host'];
     if (req.protocol === "https") global.PROTOCOL = 'https'; else global.PROTOCOL = 'http';
+    console.log(`global.LOCAL_URL = ${global.LOCAL_URL}\nglobal.PROTOCOL = ${global.PROTOCOL}`);
+
+    if (!global.LOCAL_URL) throw new Error('No host header found in request' + JSON.stringify(req.headers, null, 2));
 
-    if (!global.LOCAL_URL) throw new Error('No host header found in request' + JSON.stringify(req.headers, null, 2));    
-    
     const hostname = global.LOCAL_URL.split(':')[0];
     const newUrl = `${global.PROTOCOL}://${hostname}`;//:${PROXY_PORT_OMNITOOL}`;
-    global.PING_URL = `${newUrl}/api/v1/mercenaries/ping`;
+    global.PING_URL = `${newUrl}${OMNITOOL_HEALTH_PING}`;
+    console.log(`global.PING_URL = ${global.PING_URL}`);
+
+    // New logic to set CONNECTED_TO_MASTER
+    global.CONNECTED_TO_MASTER = false;
+    global.SPACE_OWNER = extractSpaceOwner(global.LOCAL_URL, OMNITOOL_SPACE_NAME);
+ console.log(`global.SPACE_OWNER = ${global.SPACE_OWNER}`);
+
+    if (global.SPACE_OWNER === OMNITOOL_SPACE_OWNER) { global.CONNECTED_TO_MASTER = true; }
+    console.log(`global.CONNECTED_TO_MASTER = ${global.CONNECTED_TO_MASTER}`);
+
 }
 
 async function startServer()
@@ -341,18 +432,21 @@ async function startServer()
     {
         console.log(`Server running on port ${PROXY_PORT_OMNITOOL}`);
     });
-    
+
 }
 
-function omnitoolProxyMiddleware(req, res, next) {
-    if (global.OMNITOOL_RUNNING && req.session.isVisited) {
+function omnitoolProxyMiddleware(req, res, next)
+{
+    if (global.OMNITOOL_RUNNING && req.session.isVisited)
+    {
         // Proxy all requests to OMNITOOL when conditions are met
         return createProxyMiddleware({
             target: OMNI_URL,
             changeOrigin: true,
             ws: true // if you need WebSocket support
         })(req, res, next);
-    } else {
+    } else
+    {
         // Continue with normal processing for other cases
         next();
     }
@@ -361,9 +455,7 @@ function omnitoolProxyMiddleware(req, res, next) {
 
 async function main(app)
 {
-    
-    
-    checkOmnitoolStatus();
+    await checkOmnitoolStatus();
 
     // Configure session middleware
     app.use(session({
@@ -380,21 +472,21 @@ async function main(app)
         console.error(`There was an error starting the server: ${error}`);
         return;
     }
-    
-    
+
+
     app.use(omnitoolProxyMiddleware);
 
     app.get('/favicon.ico', (req, res) => res.status(204));
     app.get('/start-omnitool-server', (req, res) => handleGetStartOmnitoolServer(req, res));
     app.get('/omnitool-logs', (req, res) => handleGetOmnitoolLogs(req, res));
     app.get('/burst-iframe', (req, res) => handleGetBurstIframe(req, res));
-    app.get('/', (req, res) =>  handleGetRoot(req, res)); 
+    app.get('/', (req, res) => handleGetRoot(req, res));
 
     setInterval(async () =>
     {
         await checkOmnitoolStatus();
     }, CHECK_OMNI_INTERVAL);
-    
+
 }
 
 // Define the proxy middleware outside the function
@@ -404,11 +496,14 @@ global.OMNITOOL_PROXY = createProxyMiddleware({
     ws: true // if you need WebSocket support
 });
 
-function omnitoolProxyMiddleware(req, res, next) {
-    if (global.OMNITOOL_RUNNING && req.session.isVisited) {
+function omnitoolProxyMiddleware(req, res, next)
+{
+    if (global.OMNITOOL_RUNNING && req.session.isVisited)
+    {
         // Use the predefined proxy middleware
         return global.OMNITOOL_PROXY(req, res, next);
-    } else {
+    } else
+    {
         // Continue with normal processing for other cases
         next();
     }
@@ -416,3 +511,4 @@ function omnitoolProxyMiddleware(req, res, next) {
 main(app);
 
 
+