zobro-cli 1.1.8 → 1.2.0

package/index.js

@@ -1,25 +1,42 @@
 #!/usr/bin/env node
 
-const fs = require('fs'); // <-- Added for file operations
+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');
 
-// ⚠️ Not recommended for public code!
-const API_KEY = "YOUR_GEMINI_API_KEY_HERE";
+// ⚠️ Paste your official Google Gemini API key here.
+const API_KEY = "AIzaSyCuqR5sFFQMzso-mwknatXRyH9_-9lav5s";
 
 /**
- * Generates code and saves it to a file or prints it to the console.
- * @param {string} prompt - The user's question.
- * @param {string} outputFile - The name of the file to save the code in.
+ * 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 generateCode(prompt, outputFile) {
-    const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=${API_KEY}`;
+async function generateProject(prompt, directoryName) {
+    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").
+2. "code": (string) The complete code for that file.
+
+Only return the raw JSON array, with no other text, explanations, or markdown fences.
+
+User Prompt: "${prompt}"
+`;
+
     const body = {
-        contents: [{ parts: [{ text: `You are an expert code generation assistant. Provide only the code for the following prompt, without any extra explanation or markdown fences. Prompt: "${prompt}"` }] }]
+        contents: [{ parts: [{ text: newPrompt }] }]
     };
 
     try {
-        console.log("🤖 Generating code...");
+        console.log(`Code Running in Expresss`);
 
         const response = await fetch(url, {
             method: 'POST',
@@ -27,39 +44,75 @@ async function generateCode(prompt, outputFile) {
             body: JSON.stringify(body),
         });
 
-        if (!response.ok) throw new Error(`API request failed with status ${response.status}`);
+        if (!response.ok) {
+            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();
-        const code = data.candidates[0].content.parts[0].text;
-
-        // --- NEW FILE SAVING LOGIC ---
-        if (outputFile) {
-            fs.writeFileSync(outputFile, code);
-            console.log(`✅ Code successfully saved to ${outputFile}`);
-        } else {
-            console.log(code); // If no file is specified, print to console
+        let responseText = data.candidates[0].content.parts[0].text;
+
+        // --- 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);
+            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: ${filePath}`);
         }
 
+       // console.log(`\n🎉 Project "${directoryName}" created successfully!`);
+
     } catch (error) {
         console.error('❌ An error occurred:', error.message);
     }
 }
 
-// Setup the command-line interface with yargs
+// --- NEW YARGS SETUP ---
 yargs(hideBin(process.argv))
     .command(
-        '$0 <prompt>',
-        'Generates code based on a text prompt.',
+        '$0 <prompt>', // The default command
+        'Generates a full project structure from a text prompt.',
         (yargs) => {
             return yargs
                 .positional('prompt', {
-                    describe: 'The code you want to generate',
+                    describe: 'The project you want to generate',
                     type: 'string',
                 })
-                .option('output', { // <-- Add the output option
-                    alias: 'o',
-                    describe: 'File to save the generated code',
+                .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) => {
@@ -67,9 +120,8 @@ yargs(hideBin(process.argv))
                 console.error('❌ ERROR: Please add your API key to the index.js file.');
                 return;
             }
-            // Pass both the prompt and the output file to the function
-            generateCode(argv.prompt, argv.output);
+            generateProject(argv.prompt, argv.directory);
         }
     )
-    .demandCommand(1, 'You must provide a prompt to generate code.')
+    .demandCommand(1, 'Please provide a prompt.')
     .parse();

package/package.json

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

package/.env

@@ -1 +0,0 @@
-API_KEY=AIzaSyCzb_DQm2IL5v4reb2YcVxI_IoQ8z9iUEA