zobro-cli 1.0.4 → 1.0.6

package/index.js

@@ -1,26 +1,42 @@
 #!/usr/bin/env node
 
-const fs = require('fs');
+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 = "AIzaSyAGMfEj6xZbqr3d48P-oDdjpe-85p7fws8";
 
-async function generateCode(prompt, outputFile) {
-    // --- THIS IS THE FINAL, CORRECTED URL ---
+/**
+ * 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 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("WAIT Server started on port 3000....");
+        console.log(`🤖 Generating project structure for "${prompt}"...`);
 
         const response = await fetch(url, {
             method: 'POST',
@@ -35,44 +51,77 @@ async function generateCode(prompt, outputFile) {
         }
 
         const data = await response.json();
-        const code = data.candidates[0].content.parts[0].text;
+        let responseText = data.candidates[0].content.parts[0].text;
+
+        // --- NEW JSON PARSING & FILE CREATION LOGIC ---
+        console.log("🤖 AI response received. Building project...");
 
-        if (outputFile) {
-            fs.writeFileSync(outputFile, code);
-            console.log(`✅ Code successfully saved to ${outputFile}`);
-        } else {
-            console.log(code);
+        // 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(`📁 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
+            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);
     }
 }
 
-// Yargs setup (no changes needed here)
+// --- NEW YARGS SETUP ---
 yargs(hideBin(process.argv))
     .command(
-        '$0 <prompt>',
-        'Generates code based on a text prompt using Google Gemini.',
+        '$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', {
-                    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) => {
             if (!API_KEY || API_KEY === "YOUR_GEMINI_API_KEY_HERE") {
-                console.error('❌ ERROR: Please replace "YOUR_GEMINI_API_KEY_HERE" with your actual key in the index.js file.');
+                console.error('❌ ERROR: Please add your API key to the index.js file.');
                 return;
             }
-            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.0.4",
+  "version": "1.0.6",
   "main": "index.js",
 "bin": {
   "zobro-cli": "./index.js"
@@ -12,7 +12,7 @@
   "author": "Secondary chor",
   "license": "ISC",
   "type": "commonjs",
-  "description": "No cheat in exam for world",
+  "description": "No cheat in exam for world ",
   "dependencies": {
     "yargs": "^17.7.2"
   }

package/shashi.txt

@@ -1,386 +0,0 @@
-```javascript
-// server.js
-
-// --- 1. Import Dependencies ---
-const express = require('express');
-const session = require('express-session');
-const path = require('path');
-const multer = require('multer'); // For file uploads, though not strictly part of 'login' itself, included as requested.
-
-// --- 2. Initialize Express App ---
-const app = express();
-const PORT = process.env.PORT || 3000;
-
-// --- 3. Mock User Data (In-memory for simplicity) ---
-// In a real application, this would come from a database.
-const users = [
-    { id: 1, username: 'user1', password: 'password1' },
-    { id: 2, username: 'user2', password: 'password2' }
-];
-
-// --- 4. Configure Express Middleware ---
-
-// Set EJS as the view engine
-app.set('view engine', 'ejs');
-app.set('views', path.join(__dirname, 'views'));
-
-// Serve static files (e.g., CSS, JS, images)
-app.use(express.static(path.join(__dirname, 'public')));
-
-// Parse URL-encoded bodies (form data)
-app.use(express.urlencoded({ extended: true }));
-
-// Parse JSON bodies (for API requests if any)
-app.use(express.json());
-
-// Configure express-session
-app.use(session({
-    secret: 'aVerySecretKeyForThisDemoApp!123', // Replace with a strong, random string in production
-    resave: false, // Don't save session if unmodified
-    saveUninitialized: false, // Don't create session until something is stored
-    cookie: {
-        maxAge: 1000 * 60 * 60 * 24, // Session lasts 1 day
-        httpOnly: true, // Prevent client-side JS from accessing the cookie
-        secure: false // Set to true if using HTTPS (for production)
-    }
-}));
-
-// Configure Multer for file uploads
-// Note: This is a basic setup. For production, consider file size limits, types, and error handling.
-const storage = multer.diskStorage({
-    destination: function (req, file, cb) {
-        cb(null, 'uploads/'); // Files will be saved in the 'uploads/' directory
-    },
-    filename: function (req, file, cb) {
-        cb(null, Date.now() + '-' + file.originalname);
-    }
-});
-
-const upload = multer({ storage: storage });
-
-// Create 'uploads' directory if it doesn't exist
-const fs = require('fs');
-const uploadsDir = path.join(__dirname, 'uploads');
-if (!fs.existsSync(uploadsDir)) {
-    fs.mkdirSync(uploadsDir);
-}
-
-
-// --- 5. Custom Middleware for Authentication Check ---
-function isAuthenticated(req, res, next) {
-    if (req.session.userId) {
-        return next(); // User is authenticated, proceed to the next middleware/route handler
-    }
-    res.redirect('/login'); // User is not authenticated, redirect to login page
-}
-
-// --- 6. Routes ---
-
-// Default route - Redirect to login if not authenticated, otherwise to dashboard
-app.get('/', (req, res) => {
-    if (req.session.userId) {
-        res.redirect('/dashboard');
-    } else {
-        res.redirect('/login');
-    }
-});
-
-// Login Page (GET)
-app.get('/login', (req, res) => {
-    // If user is already logged in, redirect to dashboard
-    if (req.session.userId) {
-        return res.redirect('/dashboard');
-    }
-    res.render('login', { error: req.query.error });
-});
-
-// Login Submission (POST)
-app.post('/login', (req, res) => {
-    const { username, password } = req.body;
-
-    const user = users.find(u => u.username === username && u.password === password);
-
-    if (user) {
-        req.session.userId = user.id; // Store user ID in session
-        req.session.username = user.username; // Store username in session
-        res.redirect('/dashboard');
-    } else {
-        res.redirect('/login?error=Invalid username or password');
-    }
-});
-
-// Register Page (GET) - Optional, for demo purposes
-app.get('/register', (req, res) => {
-    res.render('register', { error: req.query.error });
-});
-
-// Register Submission (POST) - Optional, for demo purposes
-app.post('/register', (req, res) => {
-    const { username, password } = req.body;
-
-    if (!username || !password) {
-        return res.redirect('/register?error=Username and password are required');
-    }
-
-    if (users.some(u => u.username === username)) {
-        return res.redirect('/register?error=Username already exists');
-    }
-
-    const newId = users.length > 0 ? Math.max(...users.map(u => u.id)) + 1 : 1;
-    users.push({ id: newId, username, password });
-    console.log('New user registered:', { id: newId, username, password });
-    res.redirect('/login?message=Registration successful, please login');
-});
-
-
-// Dashboard Page (GET) - Protected route
-app.get('/dashboard', isAuthenticated, (req, res) => {
-    const currentUser = users.find(u => u.id === req.session.userId);
-    if (currentUser) {
-        res.render('dashboard', { username: currentUser.username });
-    } else {
-        // This case should ideally not happen if isAuthenticated works, but for safety
-        req.session.destroy(() => {
-            res.redirect('/login?error=Session corrupted, please login again.');
-        });
-    }
-});
-
-// Logout (POST)
-app.post('/logout', (req, res) => {
-    req.session.destroy(err => {
-        if (err) {
-            console.error('Error destroying session:', err);
-            return res.redirect('/dashboard'); // Or handle error
-        }
-        res.redirect('/login?message=Logged out successfully');
-    });
-});
-
-// Example route for file upload using multer (not directly tied to login, but demonstrates multer)
-app.post('/upload-profile-picture', isAuthenticated, upload.single('profilePicture'), (req, res) => {
-    if (!req.file) {
-        return res.status(400).send('No file uploaded.');
-    }
-    // Here you would save the file path to the user's profile in your database
-    console.log(`File uploaded by user ${req.session.username}:`, req.file.path);
-    res.send(`File uploaded successfully! Path: ${req.file.path}`);
-});
-
-
-// --- 7. Start the Server ---
-app.listen(PORT, () => {
-    console.log(`Server is running on http://localhost:${PORT}`);
-    console.log('Try logging in with user: user1, pass: password1');
-});
-```
-```html
-<!-- views/login.ejs -->
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Login</title>
-    <link rel="stylesheet" href="/style.css">
-</head>
-<body>
-    <div class="container">
-        <h2>Login</h2>
-        <% if (error) { %>
-            <p class="error-message"><%= error %></p>
-        <% } %>
-        <% if (typeof message !== 'undefined' && message) { %>
-            <p class="success-message"><%= message %></p>
-        <% } %>
-        <form action="/login" method="POST">
-            <div class="form-group">
-                <label for="username">Username:</label>
-                <input type="text" id="username" name="username" required>
-            </div>
-            <div class="form-group">
-                <label for="password">Password:</label>
-                <input type="password" id="password" name="password" required>
-            </div>
-            <button type="submit">Login</button>
-        </form>
-        <p>Don't have an account? <a href="/register">Register here</a>.</p>
-    </div>
-</body>
-</html>
-```
-```html
-<!-- views/register.ejs -->
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Register</title>
-    <link rel="stylesheet" href="/style.css">
-</head>
-<body>
-    <div class="container">
-        <h2>Register</h2>
-        <% if (error) { %>
-            <p class="error-message"><%= error %></p>
-        <% } %>
-        <form action="/register" method="POST">
-            <div class="form-group">
-                <label for="username">Username:</label>
-                <input type="text" id="username" name="username" required>
-            </div>
-            <div class="form-group">
-                <label for="password">Password:</label>
-                <input type="password" id="password" name="password" required>
-            </div>
-            <button type="submit">Register</button>
-        </form>
-        <p>Already have an account? <a href="/login">Login here</a>.</p>
-    </div>
-</body>
-</html>
-```
-```html
-<!-- views/dashboard.ejs -->
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Dashboard</title>
-    <link rel="stylesheet" href="/style.css">
-</head>
-<body>
-    <div class="container">
-        <h2>Welcome to the Dashboard, <%= username %>!</h2>
-        <p>This is a protected page, accessible only after successful login.</p>
-
-        <h3>Upload Profile Picture</h3>
-        <form action="/upload-profile-picture" method="POST" enctype="multipart/form-data">
-            <div class="form-group">
-                <label for="profilePicture">Select an image:</label>
-                <input type="file" id="profilePicture" name="profilePicture" accept="image/*" required>
-            </div>
-            <button type="submit">Upload</button>
-        </form>
-
-        <form action="/logout" method="POST" class="logout-form">
-            <button type="submit">Logout</button>
-        </form>
-    </div>
-</body>
-</html>
-```
-```css
-/* public/style.css */
-body {
-    font-family: Arial, sans-serif;
-    background-color: #f4f4f4;
-    display: flex;
-    justify-content: center;
-    align-items: center;
-    min-height: 100vh;
-    margin: 0;
-}
-
-.container {
-    background-color: #fff;
-    padding: 30px;
-    border-radius: 8px;
-    box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
-    width: 100%;
-    max-width: 400px;
-    text-align: center;
-}
-
-h2 {
-    color: #333;
-    margin-bottom: 20px;
-}
-
-.form-group {
-    margin-bottom: 15px;
-    text-align: left;
-}
-
-label {
-    display: block;
-    margin-bottom: 5px;
-    color: #555;
-    font-weight: bold;
-}
-
-input[type="text"],
-input[type="password"],
-input[type="file"] {
-    width: calc(100% - 20px);
-    padding: 10px;
-    border: 1px solid #ddd;
-    border-radius: 4px;
-    font-size: 16px;
-}
-
-button[type="submit"] {
-    background-color: #007bff;
-    color: white;
-    padding: 10px 15px;
-    border: none;
-    border-radius: 4px;
-    font-size: 16px;
-    cursor: pointer;
-    transition: background-color 0.3s ease;
-    width: 100%;
-    margin-top: 10px;
-}
-
-button[type="submit"]:hover {
-    background-color: #0056b3;
-}
-
-.error-message {
-    color: #dc3545;
-    background-color: #f8d7da;
-    border: 1px solid #f5c6cb;
-    padding: 10px;
-    border-radius: 4px;
-    margin-bottom: 15px;
-    font-size: 0.9em;
-}
-
-.success-message {
-    color: #28a745;
-    background-color: #d4edda;
-    border: 1px solid #c3e6cb;
-    padding: 10px;
-    border-radius: 4px;
-    margin-bottom: 15px;
-    font-size: 0.9em;
-}
-
-p {
-    margin-top: 20px;
-    color: #666;
-}
-
-a {
-    color: #007bff;
-    text-decoration: none;
-}
-
-a:hover {
-    text-decoration: underline;
-}
-
-.logout-form {
-    margin-top: 30px;
-}
-
-.logout-form button {
-    background-color: #6c757d;
-}
-
-.logout-form button:hover {
-    background-color: #5a6268;
-}
-```