zobro-cli 1.0.9 → 1.1.0

package/.env

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

package/index.js

@@ -1,42 +1,38 @@
 #!/usr/bin/env node
 
-const fs = require('fs'); // Node.js File System
-const path = require('path'); // Node.js Path module
+const fs = require('fs');
+const path = require('path');
 const yargs = require('yargs/yargs');
 const { hideBin } = require('yargs/helpers');
 
-// ⚠️ Paste your official Google Gemini API key here.
-const API_KEY = "AIzaSyAGMfEj6xZbqr3d48P-oDdjpe-85p7fws8";
+// 🛡️ SECURE: No hardcoded key. It pulls from your computer's system settings.
+const API_KEY = process.env.GEMINI_API_KEY;
 
-/**
- * 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) {
+    // Check if the key exists before running
+    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 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 two keys:
-1. "filename": (string) The full relative path for the file (e.g., "index.html", "src/styles.css", "js/app.js").
+1. "filename": (string) The full relative path for the file.
 2. "code": (string) The complete code for that file.
 
-Only return the raw JSON array, with no other text, explanations, or markdown fences.
+Only return the raw JSON array, with no other text or markdown fences.
+User Prompt: "${prompt}"`;
 
-User Prompt: "${prompt}"
-`;
-
-    const body = {
-        contents: [{ parts: [{ text: newPrompt }] }]
-    };
+    const body = { contents: [{ parts: [{ text: newPrompt }] }] };
 
     try {
-        console.log(`🤖 Generating project structure for "${prompt}"...`);
+        console.log(`🤖 Building project in "${directoryName}"...`);
 
         const response = await fetch(url, {
             method: 'POST',
@@ -53,75 +49,39 @@ User Prompt: "${prompt}"
         const data = await response.json();
         let responseText = data.candidates[0].content.parts[0].text;
 
-        // --- NEW JSON PARSING & FILE CREATION LOGIC ---
-        console.log("🤖 AI response received. 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
+        const files = JSON.parse(responseText);
         fs.mkdirSync(directoryName, { recursive: true });
-        console.log(`📁 Created base directory: ${directoryName}`);
 
-        // Loop through the files array and create each file
         for (const file of files) {
             const filePath = path.join(directoryName, file.filename);
             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
+            if (!fs.existsSync(fileDir)) fs.mkdirSync(fileDir, { recursive: true });
             fs.writeFileSync(filePath, file.code);
-            console.log(`✅ Created file: ${filePath}`);
+            console.log(`✅ Created: ${file.filename}`);
         }
 
-        console.log(`\n🎉 Project "${directoryName}" created successfully!`);
+        console.log(`\n🎉 Done! Project is ready.`);
 
     } catch (error) {
         console.error('❌ An error occurred:', error.message);
     }
 }
 
-// --- NEW YARGS SETUP ---
 yargs(hideBin(process.argv))
     .command(
-        '$0 <prompt>', // The default command
-        'Generates a full project structure from a text prompt.',
+        '$0 <prompt>',
+        'Generates a full project structure.',
         (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
-                });
+                .positional('prompt', { type: 'string' })
+                .option('directory', { alias: 'd', type: 'string', demandOption: true });
         },
         (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.0.9",
+  "version": "1.1.0",
   "main": "index.js",
 "bin": {
   "zobro-cli": "./index.js"