zobro-cli 1.1.6 → 1.1.8

package/.env

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

package/index.js

@@ -1,133 +1,75 @@
 #!/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 ----------------
-async function generateProject(prompt, directoryName) {
-    const API_KEY = await getApiKey();
-
-    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:
-- filename
-- code
-
-Return ONLY raw JSON.
-User Prompt: "${prompt}"
-`;
-
+const fs = require('fs'); // <-- Added for file operations
+const yargs = require('yargs/yargs');
+const { hideBin } = require('yargs/helpers');
+
+// ⚠️ Not recommended for public code!
+const API_KEY = "YOUR_GEMINI_API_KEY_HERE";
+
+/**
+ * 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 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(`🤖 Building project in "${directoryName}"...`);
+        console.log("🤖 Generating code...");
 
         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);
-        }
+        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;
-        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);
-            fs.mkdirSync(path.dirname(filePath), { recursive: true });
-            fs.writeFileSync(filePath, file.code);
-            console.log(`✅ Created: ${file.filename}`);
+        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(`🎉 Project ready in "${directoryName}"`);
-
-    } 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();
+// Setup the command-line interface with yargs
+yargs(hideBin(process.argv))
+    .command(
+        '$0 <prompt>',
+        'Generates code based on a text prompt.',
+        (yargs) => {
+            return yargs
+                .positional('prompt', {
+                    describe: 'The code you want to generate',
+                    type: 'string',
+                })
+                .option('output', { // <-- Add the output option
+                    alias: 'o',
+                    describe: 'File to save the generated code',
+                    type: 'string',
+                });
+        },
+        (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;
+            }
+            // Pass both the prompt and the output file to the function
+            generateCode(argv.prompt, argv.output);
+        }
+    )
+    .demandCommand(1, 'You must provide a prompt to generate code.')
+    .parse();

package/package.json

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