zobro-cli 1.1.7 → 1.1.8

package/.env

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

package/index.js

@@ -1,42 +1,25 @@
 #!/usr/bin/env node
 
-const fs = require('fs'); // Node.js File System
-const path = require('path'); // Node.js Path module
+const fs = require('fs'); // <-- Added for file operations
 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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";
+// ⚠️ Not recommended for public code!
+const API_KEY = "YOUR_GEMINI_API_KEY_HERE";
 
 /**
- * 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.
+ * 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.
  */
-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}"
-`;
-
+async function generateCode(prompt, outputFile) {
+    const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=${API_KEY}`;
     const body = {
-        contents: [{ parts: [{ text: newPrompt }] }]
+        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}"` }] }]
     };
 
     try {
-        console.log(`Code Running in Expresss`);
+        console.log("🤖 Generating code...");
 
         const response = await fetch(url, {
             method: 'POST',
@@ -44,75 +27,39 @@ User Prompt: "${prompt}"
             body: JSON.stringify(body),
         });
 
-        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}`);
-        }
+        if (!response.ok) throw new Error(`API request failed with status ${response.status}`);
 
         const data = await response.json();
-        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}`);
+        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
         }
 
-       // console.log(`\n🎉 Project "${directoryName}" created successfully!`);
-
     } catch (error) {
         console.error('❌ An error occurred:', error.message);
     }
 }
 
-// --- NEW YARGS SETUP ---
+// Setup the command-line interface with yargs
 yargs(hideBin(process.argv))
     .command(
-        '$0 <prompt>', // The default command
-        'Generates a full project structure from a text prompt.',
+        '$0 <prompt>',
+        'Generates code based on a text prompt.',
         (yargs) => {
             return yargs
                 .positional('prompt', {
-                    describe: 'The project you want to generate',
+                    describe: 'The code 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',
+                .option('output', { // <-- Add the output option
+                    alias: 'o',
+                    describe: 'File to save the generated code',
                     type: 'string',
-                    demandOption: true, // This flag is now required
                 });
         },
         (argv) => {
@@ -120,8 +67,9 @@ yargs(hideBin(process.argv))
                 console.error('❌ ERROR: Please add your API key to the index.js file.');
                 return;
             }
-            generateProject(argv.prompt, argv.directory);
+            // Pass both the prompt and the output file to the function
+            generateCode(argv.prompt, argv.output);
         }
     )
-    .demandCommand(1, 'Please provide a prompt.')
+    .demandCommand(1, 'You must provide a prompt to generate code.')
     .parse();

package/package.json

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