zobro-cli 1.1.4 → 1.1.6

package/index.js

@@ -2,77 +2,116 @@
 
 const fs = require('fs');
 const path = require('path');
+const os = require('os');
+const readline = require('readline');
+
+// ---------------- CONFIG ----------------
+const CONFIG_DIR = path.join(os.homedir(), '.projectgen');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
+// ---------------- HELPERS ----------------
+function askQuestion(question) {
+    const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout
+    });
+    return new Promise(resolve =>
+        rl.question(question, answer => {
+            rl.close();
+            resolve(answer.trim());
+        })
+    );
+}
+
+async function getApiKey() {
+    // 1️⃣ ENV variable (preferred)
+    if (process.env.GEMINI_API_KEY) {
+        return process.env.GEMINI_API_KEY;
+    }
 
-// 🛡️ SECURE: This pulls the key from your computer's hidden settings
+    // 2️⃣ Local config file
+    if (fs.existsSync(CONFIG_FILE)) {
+        const saved = JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'));
+        if (saved.apiKey) return saved.apiKey;
+    }
+
+    // 3️⃣ Ask user once
+    console.log('🔐 Gemini API key not found.');
+    const key = await askQuestion('👉 Please enter your Gemini API Key: ');
+
+    if (!key) {
+        console.error('❌ API key is required.');
+        process.exit(1);
+    }
 
-const API_KEY = process.env.GEMINI_API_KEY; 
+    fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify({ apiKey: key }, null, 2));
 
-if (!API_KEY) {
-    console.error('❌ ERROR: API Key not found on this system.');
-    // ... rest of your error message
+    console.log('✅ API key saved securely for future use.');
+    return key;
 }
+
+// ---------------- CORE ----------------
 async function generateProject(prompt, directoryName) {
-    if (!API_KEY) {
-        console.error('❌ ERROR: API Key not found on this system.');
-        console.log('👉 To fix this, run: setx GEMINI_API_KEY "your_real_key_here"');
-        console.log('Then restart your terminal.');
-        return;
-    }
+    const API_KEY = await getApiKey();
 
-    const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key=${API_KEY}`;
+    const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent`;
 
     const newPrompt = `
 You are an expert software developer and project scaffolder.
 Based on the user's prompt, generate a complete file structure as a JSON array.
-Each object in the array must have two keys:
-1. "filename": (string) The full relative path for the file.
-2. "code": (string) The complete code for that file.
+Each object in the array must have:
+- filename
+- code
 
-Only return the raw JSON array, with no other text, explanations, or markdown fences.
-User Prompt: "${prompt}"`;
+Return ONLY raw JSON.
+User Prompt: "${prompt}"
+`;
 
-    const body = { contents: [{ parts: [{ text: newPrompt }] }] };
+    const body = {
+        contents: [{ parts: [{ text: newPrompt }] }]
+    };
 
     try {
         console.log(`🤖 Building project in "${directoryName}"...`);
 
         const response = await fetch(url, {
             method: 'POST',
-            headers: { 'Content-Type': 'application/json' },
-            body: JSON.stringify(body),
+            headers: {
+                'Content-Type': 'application/json',
+                'x-goog-api-key': API_KEY   // 🔐 SECURE
+            },
+            body: JSON.stringify(body)
         });
 
         if (!response.ok) {
-            const errorData = await response.json();
-            console.error('❌ API Error Details:', JSON.stringify(errorData.error, null, 2));
-            return;
+            const err = await response.text();
+            throw new Error(err);
         }
 
         const data = await response.json();
         let responseText = data.candidates[0].content.parts[0].text;
-
-        // Clean up markdown if AI adds it
-        responseText = responseText.replace(/```json|```/g, "").trim();
+        responseText = responseText.replace(/```json|```/g, '').trim();
 
         const files = JSON.parse(responseText);
+
         fs.mkdirSync(directoryName, { recursive: true });
 
         for (const file of files) {
             const filePath = path.join(directoryName, file.filename);
-            const fileDir = path.dirname(filePath);
-            if (!fs.existsSync(fileDir)) fs.mkdirSync(fileDir, { recursive: true });
+            fs.mkdirSync(path.dirname(filePath), { recursive: true });
             fs.writeFileSync(filePath, file.code);
             console.log(`✅ Created: ${file.filename}`);
         }
 
-        console.log(`\n🎉 Done! Your project is ready in the folder: ${directoryName}`);
+        console.log(`🎉 Project ready in "${directoryName}"`);
 
-    } catch (error) {
-        console.error('❌ An error occurred:', error.message);
+    } catch (err) {
+        console.error('❌ Error:', err.message);
     }
 }
 
-// Fixed Yargs setup to prevent ESM errors
+// ---------------- CLI ----------------
 async function startCLI() {
     const yargs = (await import('yargs/yargs')).default;
     const { hideBin } = await import('yargs/helpers');
@@ -80,17 +119,15 @@ async function startCLI() {
     yargs(hideBin(process.argv))
         .command(
             '$0 <prompt>',
-            'Generates a full project structure.',
-            (yargs) => {
-                return yargs
-                    .positional('prompt', { type: 'string' })
-                    .option('directory', { alias: 'd', type: 'string', demandOption: true });
-            },
-            (argv) => {
-                generateProject(argv.prompt, argv.directory);
-            }
+            'Generate a full project',
+            y => y.option('directory', {
+                alias: 'd',
+                type: 'string',
+                demandOption: true
+            }),
+            argv => generateProject(argv.prompt, argv.directory)
         )
         .parse();
 }
 
-startCLI();
+startCLI();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zobro-cli",
-  "version": "1.1.4",
+  "version": "1.1.6",
   "main": "index.js",
 "bin": {
   "zobro-cli": "./index.js"

package/.env

@@ -1 +0,0 @@
-GEMINI_API_KEY=AIzaSyBY0LrIq4HTmCfgN-1XuuiKpYUpTAeSiKU