cloud-pc-templates 1.0.2 → 1.1.0

package/handlers/ollamacloud.js

@@ -0,0 +1,147 @@
+const http = require('http');
+const https = require('https');
+const fs = require('fs');
+const path = require('path');
+const { spawn } = require('child_process');
+const os = require('os');
+
+// Function to get masked API key input
+function promptForApiKey() {
+  return new Promise((resolve) => {
+    process.stdout.write('Enter API Key: ');
+    
+    const stdin = process.stdin;
+    
+    // Handle both TTY and non-TTY environments
+    if (stdin.isTTY) {
+      stdin.setRawMode(true);
+    }
+    stdin.resume();
+    
+    let apiKey = '';
+    let isFirstInput = true;
+    
+    stdin.on('data', (byte) => {
+      const char = byte.toString();
+      
+      if (char === '\n' || char === '\r' || char === '\u0004') {
+        // Enter or EOF
+        if (stdin.isTTY) {
+          stdin.setRawMode(false);
+        }
+        stdin.pause();
+        stdin.removeAllListeners('data');
+        console.log('');
+        resolve(apiKey);
+      } else if (char === '\u0003') {
+        // Ctrl+C
+        if (stdin.isTTY) {
+          stdin.setRawMode(false);
+        }
+        process.exit();
+      } else if (char === '\x7f' || char === '\b') {
+        // Backspace
+        if (apiKey.length > 0) {
+          apiKey = apiKey.slice(0, -1);
+          process.stdout.write('\x1b[D\x1b[K');
+        }
+      } else if (char >= '\x20' && char <= '\x7e') {
+        // Printable character
+        apiKey += char;
+        process.stdout.write('*');
+      }
+    });
+  });
+}
+
+// Function to check health endpoint
+function checkHealthEndpoint(endpoint) {
+  return new Promise((resolve) => {
+    const url = new URL(endpoint);
+    const protocol = url.protocol === 'https:' ? https : http;
+    
+    const request = protocol.request(url, { method: 'GET' }, (res) => {
+      resolve(res.statusCode === 200);
+    });
+    
+    request.on('error', () => {
+      resolve(false);
+    });
+    
+    request.end();
+  });
+}
+
+// Function to download and run the proxy
+async function downloadAndRunProxy(endpoint) {
+  const url = 'https://raw.githubusercontent.com/devashish234073/cloud-pc-templates-marketplace/refs/heads/main/JS-PROXIES/ollama-proxy.js';
+  const tempFile = path.join(os.tmpdir(), 'ollama-proxy.js');
+  
+  // Download the file
+  await new Promise((resolve, reject) => {
+    const file = fs.createWriteStream(tempFile);
+    https.get(url, (res) => {
+      res.pipe(file);
+      file.on('finish', () => {
+        file.close();
+        resolve();
+      });
+    }).on('error', reject);
+  });
+  
+  // Get API key from user
+  const apiKey = await promptForApiKey();
+  
+  // Run the proxy with API key passed as environment variable
+  return new Promise((resolve, reject) => {
+    const env = Object.assign({}, process.env, { API_KEY: apiKey });
+    const child = spawn('node', [tempFile], { env });
+    
+    child.on('close', (code) => {
+      if (code === 0) {
+        // Wait a bit for the server to start, then validate
+        setTimeout(async () => {
+          const isHealthy = await checkHealthEndpoint(endpoint);
+          if (isHealthy) {
+            console.log('✓ Logged in');
+            console.log(`  - Endpoint checked: ${endpoint}`);
+          } else {
+            console.log('✓ Proxy started');
+            console.log(`  - Endpoint: ${endpoint}`);
+          }
+          resolve();
+        }, 1000);
+      } else {
+        reject(new Error('Proxy process failed'));
+      }
+    });
+    
+    child.on('error', reject);
+  });
+}
+
+// Function to check and login to Ollama Cloud
+async function checkAndLoginOllamaCloud() {
+  const endpoint = 'http://localhost:3004/health';
+  
+  try {
+    const isHealthy = await checkHealthEndpoint(endpoint);
+    if (isHealthy) {
+      console.log('✓ Already logged in');
+      console.log(`  - Endpoint checked: ${endpoint}`);
+      return;
+    }
+    
+    // Not healthy, download and run proxy
+    await downloadAndRunProxy(endpoint);
+  } catch (error) {
+    console.error('Error during login:', error.message);
+  }
+}
+
+module.exports = {
+  checkAndLoginOllamaCloud,
+  checkHealthEndpoint,
+  downloadAndRunProxy,
+  promptForApiKey
+};

package/index.js

@@ -1,5 +1,7 @@
 #!/usr/bin/env node
 
+const { checkAndLoginOllamaCloud } = require('./handlers/ollamacloud');
+
 // Command tree structure
 const commandTree = {
   help: {
@@ -53,12 +55,11 @@ function aiDefault() {
 }
 
 // AI Login function
-function aiLogin(mode) {
-  console.log(`✓ AI Login initialized with mode: ${mode}`);
+async function aiLogin(mode) {
   if (mode === 'ollamacloud') {
-    console.log('  - Connecting to Ollama Cloud...');
-    console.log('  - Initializing cloud connection...');
+    await checkAndLoginOllamaCloud();
   } else if (mode === 'ollamalocal') {
+    console.log(`✓ AI Login initialized with mode: ${mode}`);
     console.log('  - Connecting to Ollama Local...');
     console.log('  - Initializing local connection...');
   }
@@ -107,8 +108,11 @@ function traverseCommandTree(args, startNode, startPath = []) {
   if (currentNode.subcommands) {
     showAvailableOptions(currentNode, path);
   } else if (currentNode.handler) {
-    // Execute the handler if available
-    currentNode.handler();
+    // Execute the handler if available (could be async)
+    const result = currentNode.handler();
+    if (result instanceof Promise) {
+      result.catch(err => console.error('Error:', err.message));
+    }
   } else {
     console.log('Command complete but no action defined');
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cloud-pc-templates",
-  "version": "1.0.2",
+  "version": "1.1.0",
   "description": "",
   "main": "index.js",
   "bin": {