zobro-cli 1.1.6 → 1.1.7

package/index.js

@@ -1,70 +1,33 @@
 #!/usr/bin/env node
 
-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;
-    }
-
-    // 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);
-    }
-
-    fs.mkdirSync(CONFIG_DIR, { recursive: true });
-    fs.writeFileSync(CONFIG_FILE, JSON.stringify({ apiKey: key }, null, 2));
-
-    console.log('✅ API key saved securely for future use.');
-    return key;
-}
-
-// ---------------- CORE ----------------
+const fs = require('fs'); // Node.js File System
+const path = require('path'); // Node.js Path module
+const yargs = require('yargs/yargs');
+const { hideBin } = require('yargs/helpers');
+
+// ⚠️ Paste your official Google Gemini API key here.
+const API_KEY = "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";
+
+/**
+ * Generates a full project structure based on a prompt.
+ * @param {string} prompt - The user's project request.
+ * @param {string} directoryName - The name of the folder to create the project in.
+ */
 async function generateProject(prompt, directoryName) {
-    const API_KEY = await getApiKey();
-
-    const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent`;
+    const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key=${API_KEY}`;
 
+    // --- This is the new, "smarter" prompt ---
+    // We are asking the AI to act as a file structure generator
+    // and return a specific JSON format.
     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:
-- filename
-- code
+Each object in the array must have two keys:
+1. "filename": (string) The full relative path for the file (e.g., "index.html", "src/styles.css", "js/app.js").
+2. "code": (string) The complete code for that file.
+
+Only return the raw JSON array, with no other text, explanations, or markdown fences.
 
-Return ONLY raw JSON.
 User Prompt: "${prompt}"
 `;
 
@@ -73,61 +36,92 @@ User Prompt: "${prompt}"
     };
 
     try {
-        console.log(`🤖 Building project in "${directoryName}"...`);
+        console.log(`Code Running in Expresss`);
 
         const response = await fetch(url, {
             method: 'POST',
-            headers: {
-                'Content-Type': 'application/json',
-                'x-goog-api-key': API_KEY   // 🔐 SECURE
-            },
-            body: JSON.stringify(body)
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(body),
         });
 
         if (!response.ok) {
-            const err = await response.text();
-            throw new Error(err);
+            const errorData = await response.json();
+            console.error('❌ API Error Details:', JSON.stringify(errorData.error, null, 2));
+            throw new Error(`API request failed with status ${response.status}`);
         }
 
         const data = await response.json();
         let responseText = data.candidates[0].content.parts[0].text;
-        responseText = responseText.replace(/```json|```/g, '').trim();
 
-        const files = JSON.parse(responseText);
+        // --- NEW JSON PARSING & FILE CREATION LOGIC ---
+       // console.log("Building project...");
+
+        // Clean up potential markdown fences from the AI's response
+        if (responseText.startsWith("```json")) {
+            responseText = responseText.substring(7, responseText.length - 3).trim();
+        }
+
+        let files;
+        try {
+            files = JSON.parse(responseText);
+            if (!Array.isArray(files)) throw new Error("AI did not return a JSON array.");
+        } catch (parseError) {
+            console.error("❌ ERROR: Failed to parse the AI's response. The response was not valid JSON.");
+            console.error("Raw AI Response:", responseText);
+            return;
+        }
 
+        // Create the main project directory
         fs.mkdirSync(directoryName, { recursive: true });
+        console.log(``);
 
+        // Loop through the files array and create each file
         for (const file of files) {
             const filePath = path.join(directoryName, file.filename);
-            fs.mkdirSync(path.dirname(filePath), { recursive: true });
+            const fileDir = path.dirname(filePath);
+
+            // Create subdirectories if they don't exist
+            if (!fs.existsSync(fileDir)) {
+                fs.mkdirSync(fileDir, { recursive: true });
+            }
+
+            // Write the code to the file
             fs.writeFileSync(filePath, file.code);
-            console.log(`✅ Created: ${file.filename}`);
+            //console.log(`Created file: ${filePath}`);
         }
 
-        console.log(`🎉 Project ready in "${directoryName}"`);
+       // console.log(`\n🎉 Project "${directoryName}" created successfully!`);
 
-    } catch (err) {
-        console.error('❌ Error:', err.message);
+    } catch (error) {
+        console.error('❌ An error occurred:', error.message);
     }
 }
 
-// ---------------- CLI ----------------
-async function startCLI() {
-    const yargs = (await import('yargs/yargs')).default;
-    const { hideBin } = await import('yargs/helpers');
-
-    yargs(hideBin(process.argv))
-        .command(
-            '$0 <prompt>',
-            'Generate a full project',
-            y => y.option('directory', {
-                alias: 'd',
-                type: 'string',
-                demandOption: true
-            }),
-            argv => generateProject(argv.prompt, argv.directory)
-        )
-        .parse();
-}
-
-startCLI();
+// --- NEW YARGS SETUP ---
+yargs(hideBin(process.argv))
+    .command(
+        '$0 <prompt>', // The default command
+        'Generates a full project structure from a text prompt.',
+        (yargs) => {
+            return yargs
+                .positional('prompt', {
+                    describe: 'The project you want to generate',
+                    type: 'string',
+                })
+                .option('directory', { // Replaces the old 'output' flag
+                    alias: 'd',
+                    describe: 'The name of the new directory to create the project in',
+                    type: 'string',
+                    demandOption: true, // This flag is now required
+                });
+        },
+        (argv) => {
+            if (!API_KEY || API_KEY === "YOUR_GEMINI_API_KEY_HERE") {
+                console.error('❌ ERROR: Please add your API key to the index.js file.');
+                return;
+            }
+            generateProject(argv.prompt, argv.directory);
+        }
+    )
+    .demandCommand(1, 'Please provide a prompt.')
+    .parse();

package/package.json

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