prituscode 1.0.0

package/README.md

@@ -0,0 +1,79 @@
+# PRITUS CODE CLI
+
+**Terminal-based AI coding assistant.** Local Vision Intelligence in your terminal.
+
+## Install
+
+```bash
+# Install globally
+npm install -g prituscode
+
+# Or run directly with npx (no install needed)
+npx prituscode
+```
+
+## Usage
+
+### Interactive mode
+```bash
+prituscode
+```
+
+### One-shot mode
+```bash
+prituscode "build a responsive navbar with Tailwind"
+```
+
+### Custom endpoint
+```bash
+prituscode --endpoint https://your-pritus-deployment.com
+```
+
+### Select model tier
+```bash
+prituscode --model thinking
+```
+
+## Commands (interactive mode)
+
+| Command | Description |
+|---------|-------------|
+| `/help` | Show available commands |
+| `/clear` | Clear conversation history |
+| `/model` | Switch model tier (`flash` / `pro` / `thinking`) |
+| `/status` | Show connection status |
+| `/exit` | Exit the CLI |
+
+## Environment Variables
+
+| Variable | Default | Description |
+|----------|---------|-------------|
+| `PRITUS_API_URL` | `https://pritusai.netlify.app` | API endpoint |
+| `PRITUS_MODEL` | `pro` | Default model tier |
+
+## Run without installing
+
+```bash
+# Clone and run directly
+git clone https://github.com/sarthaksahoo/pritus-ai
+cd pritus-ai/cli
+node prituscode.js
+```
+
+## Publish to npm
+
+```bash
+npm login          # log in to your npm account
+cd cli
+npm publish
+```
+
+After publishing, anyone can run `npx prituscode` globally.
+
+## Requirements
+
+- Node.js 18+ (for built-in `fetch`)
+
+## License
+
+MIT © Sarthak Sahoo

package/package.json

@@ -0,0 +1,17 @@
+{
+  "name": "prituscode",
+  "version": "1.0.0",
+  "description": "Terminal-based AI coding agent CLI",
+  "main": "prituscode.js",
+  "bin": {
+    "prituscode": "./prituscode.js"
+  },
+  "scripts": {
+    "start": "node ./prituscode.js"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "author": "",
+  "license": "ISC"
+}

package/prituscode.js

@@ -0,0 +1,359 @@
+#!/usr/bin/env node
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const readline = require('readline');
+const { execSync } = require('child_process');
+
+// --- ANSI COLORS & STYLES ---
+const c = {
+    rst: '\x1b[0m', bold: '\x1b[1m', dim: '\x1b[2m',
+    black: '\x1b[30m', red: '\x1b[31m', green: '\x1b[32m',
+    yellow: '\x1b[33m', blue: '\x1b[34m', magenta: '\x1b[35m',
+    cyan: '\x1b[36m', white: '\x1b[37m',
+    bgMagenta: '\x1b[45m', bgBlue: '\x1b[44m'
+};
+
+// --- CONFIG & STATE ---
+const args = process.argv.slice(2);
+let config = {
+    tier: 'pro',
+    endpoint: 'https://pritusai.vercel.app/api/pritus/chat',
+    oneShotPrompt: ''
+};
+const messages = [];
+
+// Parse flags
+for (let i = 0; i < args.length; i++) {
+    if (args[i] === '--endpoint' && args[i + 1]) {
+        config.endpoint = args[++i];
+    } else if (args[i] === '--model' && args[i + 1]) {
+        config.tier = args[++i];
+    } else if (!args[i].startsWith('--')) {
+        config.oneShotPrompt = args.slice(i).join(' ');
+        break;
+    }
+}
+
+// --- SYSTEM PROMPT ---
+const AGENT_SYSTEM = `
+You are PRITUS CODE, an elite terminal-based AI coding agent.
+When generating or modifying code, you must ACT as an agent. Use the exact format below:
+
+THINKING: <one sentence reasoning about what you will do>
+CREATE_FILE: <filepath>
+<entire file content here>
+DONE: <short summary of what was done>
+
+If the user is just asking a question and no files need to be written, simply reply conversationally without those tags. Do not reveal your underlying model name under any circumstance.
+`;
+messages.push({ role: 'system', content: AGENT_SYSTEM });
+
+// --- UTILS ---
+const sleep = ms => new Promise(r => setTimeout(r, ms));
+
+const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    prompt: `${c.magenta}${c.bold}> ${c.rst}`
+});
+
+const ask = query => new Promise(resolve => {
+    rl.question(query, resolve);
+});
+
+// --- UI COMPONENTS ---
+function clearConsole() {
+    process.stdout.write('\x1B[2J\x1B[0f');
+}
+
+function printLogo() {
+    // Blue to Magenta ANSI 256 gradient approximation
+    const d1 = '\x1b[38;5;33m'; 
+    const d2 = '\x1b[38;5;39m';
+    const d3 = '\x1b[38;5;135m';
+    const d4 = '\x1b[38;5;165m';
+    const d5 = '\x1b[38;5;199m';
+    
+    console.log(`
+${d1}      /\\      ${c.bold}██████╗ ██████╗ ██╗████████╗██╗   ██╗███████╗${c.rst}
+${d2}     /  \\     ${c.bold}██╔══██╗██╔══██╗██║╚══██╔══╝██║   ██║██╔════╝${c.rst}
+${d3}    /____\\    ${c.bold}██████╔╝██████╔╝██║   ██║   ██║   ██║███████╗${c.rst}
+${d4}    \\    /    ${c.bold}██╔═══╝ ██╔══██╗██║   ██║   ██║   ██║╚════██║${c.rst}
+${d5}     \\  /     ${c.bold}██║     ██║  ██║██║   ██║   ╚██████╔╝███████║${c.rst}
+${c.magenta}      \\/      ${c.bold}╚═╝     ╚═╝  ╚═╝╚═╝   ╚═╝    ╚═════╝ ╚══════╝${c.rst}
+                                             ${c.dim}v1.0.0${c.rst}
+`);
+}
+
+function printWelcomeBox() {
+    const cwd = process.cwd();
+    const boxWidth = Math.max(60, cwd.length + 20);
+    const line = '─'.repeat(boxWidth - 2);
+    console.log(`${c.magenta}┌${line}┐${c.rst}`);
+    console.log(`${c.magenta}│${c.rst} ${c.bold}Welcome to PRITUS CODE${c.rst}`.padEnd(boxWidth + c.magenta.length + c.rst.length + c.bold.length + c.rst.length - 1) + `${c.magenta}│${c.rst}`);
+    console.log(`${c.magenta}│${c.rst} ${c.dim}Type /help for commands, @filename to attach files${c.rst}`.padEnd(boxWidth + c.magenta.length + c.rst.length + c.dim.length + c.rst.length - 1) + `${c.magenta}│${c.rst}`);
+    console.log(`${c.magenta}│${c.rst} ${c.cyan}Workspace: ${cwd}${c.rst}`.padEnd(boxWidth + c.magenta.length + c.rst.length + c.cyan.length + c.rst.length - 1) + `${c.magenta}│${c.rst}`);
+    console.log(`${c.magenta}└${line}┘${c.rst}\n`);
+    
+    console.log(`${c.bold}Tips:${c.rst}`);
+    console.log(`  1. Use @file to include file context`);
+    console.log(`  2. Type /model to switch intelligence tiers`);
+    console.log(`  3. Type /init to set up project instructions`);
+    console.log(`  4. Let the agent build files for you ${c.green}✓${c.rst}\n`);
+}
+
+function printStatusBar() {
+    const cwd = path.basename(process.cwd());
+    const bar = `${c.bgMagenta}${c.white} ${cwd} ${c.rst} ${c.dim}model:${c.rst} ${config.tier} ${c.dim}| /help for menu${c.rst}`;
+    console.log(`\n${bar}`);
+}
+
+async function startLoading(text = 'Thinking') {
+    const frames = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+    let i = 0;
+    const interval = setInterval(() => {
+        process.stdout.write(`\r${c.magenta}${frames[i]}${c.rst} ${c.dim}${text}...${c.rst}`);
+        i = (i + 1) % frames.length;
+    }, 80);
+    return () => {
+        clearInterval(interval);
+        process.stdout.write('\r\x1b[K'); // clear line
+    };
+}
+
+async function streamText(text) {
+    const words = text.split(' ');
+    for (const word of words) {
+        process.stdout.write(word + ' ');
+        await sleep(20); // typing effect
+    }
+    console.log();
+}
+
+// --- COMMANDS & FEATURES ---
+async function selectModel() {
+    console.log(`\n${c.bold}Select Model Tier:${c.rst}`);
+    console.log(`  1. Flash - Fast & lightweight`);
+    console.log(`  2. Pro - Balanced general use`);
+    console.log(`  3. Thinking - Deep reasoning`);
+    console.log(`  4. Ultra Code - Elite code generation`);
+    
+    const choice = await ask(`\n${c.dim}Press 1-4 or Enter for Pro: ${c.rst}`);
+    switch(choice.trim()) {
+        case '1': config.tier = 'flash'; break;
+        case '3': config.tier = 'thinking'; break;
+        case '4': config.tier = 'code'; break;
+        case '2': 
+        default: config.tier = 'pro'; break;
+    }
+    console.log(`${c.green}✓ Switched to ${config.tier} tier${c.rst}\n`);
+}
+
+function processAtFiles(input) {
+    const fileRegex = /@([\w.\-\/]+)/g;
+    let modifiedInput = input;
+    let match;
+    let addedFiles = [];
+
+    while ((match = fileRegex.exec(input)) !== null) {
+        const filepath = match[1];
+        try {
+            const content = fs.readFileSync(filepath, 'utf-8');
+            modifiedInput += `\n\n[Context from ${filepath}]:\n\`\`\`\n${content}\n\`\`\``;
+            addedFiles.push(filepath);
+        } catch (e) {
+            console.log(`${c.red}⚠ Could not read @file: ${filepath}${c.rst}`);
+        }
+    }
+    if (addedFiles.length > 0) {
+        console.log(`${c.dim}Attached: ${addedFiles.join(', ')}${c.rst}`);
+    }
+    return modifiedInput;
+}
+
+// --- AGENT PARSER ---
+async function parseAndExecuteAgent(response) {
+    // Regex to extract blocks
+    const thinkingMatch = response.match(/THINKING:\s*(.*?)\n/i);
+    const doneMatch = response.match(/DONE:\s*(.*)/i);
+    const fileRegex = /CREATE_FILE:\s*(.*?)\n([\s\S]*?)(?=DONE:|CREATE_FILE:|$)/gi;
+
+    let isAgentic = false;
+
+    if (thinkingMatch) {
+        isAgentic = true;
+        console.log(`\n🤔 ${c.dim}${thinkingMatch[1].trim()}${c.rst}\n`);
+    }
+
+    let match;
+    let fileCount = 0;
+    while ((match = fileRegex.exec(response)) !== null) {
+        isAgentic = true;
+        const filepath = match[1].trim();
+        const content = match[2].trim();
+        
+        console.log(`${c.cyan}┌─ CREATE: ${filepath}${c.rst}`);
+        
+        try {
+            const fullPath = path.resolve(process.cwd(), filepath);
+            fs.mkdirSync(path.dirname(fullPath), { recursive: true });
+            fs.writeFileSync(fullPath, content);
+            console.log(`${c.cyan}└─ ${c.green}✓ Created successfully${c.rst}`);
+            fileCount++;
+        } catch (err) {
+            console.log(`${c.cyan}└─ ${c.red}✗ Failed: ${err.message}${c.rst}`);
+        }
+    }
+
+    if (isAgentic && doneMatch) {
+        console.log(`\n${c.green}✓ Agent Finished:${c.rst} ${doneMatch[1].trim()} (${fileCount} files created)`);
+    }
+
+    // If it didn't use the agent format, just stream it conversationally
+    if (!isAgentic) {
+        await streamText(response.trim());
+    }
+}
+
+// --- API LAYER ---
+async function handleChat(prompt) {
+    const finalPrompt = processAtFiles(prompt);
+    messages.push({ role: 'user', content: finalPrompt });
+
+    const stopLoading = await startLoading();
+    
+    try {
+        const res = await fetch(config.endpoint, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ 
+                messages, 
+                tier: config.tier, 
+                plan: 'pro' 
+            })
+        });
+
+        stopLoading();
+
+        if (!res.ok) {
+            throw new Error(`Status ${res.status}`);
+        }
+
+        const data = await res.json();
+        // Assuming API returns { response: "..." } or similar
+        const reply = data.response || data.reply || data.content || data.message || (data.choices && data.choices[0].message.content);
+        
+        if (!reply) {
+            console.log(`${c.red}⚠ Unexpected API response format.${c.rst}`);
+            return;
+        }
+
+        messages.push({ role: 'assistant', content: reply });
+        await parseAndExecuteAgent(reply);
+
+    } catch (error) {
+        stopLoading();
+        console.log(`${c.red}Pritus AI is suffering from heavy traffic right now. (Error: ${error.message})${c.rst}`);
+        // Remove the failed message so they can retry
+        messages.pop();
+    }
+}
+
+// --- SIMULATED GOOGLE LOGIN ---
+async function handleLogin() {
+    console.log(`\n${c.bold}Google Authentication${c.rst}`);
+    const code = Math.random().toString(36).substring(2, 10).toUpperCase();
+    console.log(`${c.dim}Please go to: ${c.white}https://google.com/device${c.rst}`);
+    console.log(`${c.dim}And enter this code: ${c.magenta}${c.bold}${code}${c.rst}\n`);
+    
+    const stop = await startLoading('Waiting for authorization');
+    await sleep(4000); // Simulate waiting for external auth
+    stop();
+    
+    const tokenPath = path.join(os.homedir(), '.prituscode_auth');
+    fs.writeFileSync(tokenPath, `mock_token_${Date.now()}`);
+    console.log(`${c.green}✓ Successfully logged in with Google!${c.rst}\n`);
+}
+
+
+// --- MAIN REPL ---
+async function main() {
+    if (config.oneShotPrompt) {
+        console.log(`${c.magenta}PRITUS CODE${c.rst} ${c.dim}> One-shot mode${c.rst}\n`);
+        await handleChat(config.oneShotPrompt);
+        process.exit(0);
+    }
+
+    clearConsole();
+    printLogo();
+    printWelcomeBox();
+    
+    // Auto-prompt for model on first run if in interactive mode
+    await selectModel();
+    printStatusBar();
+
+    rl.prompt();
+
+    rl.on('line', async (line) => {
+        const input = line.trim();
+        if (!input) {
+            rl.prompt();
+            return;
+        }
+
+        switch (input.toLowerCase()) {
+            case '/exit':
+            case '/quit':
+                console.log(`${c.dim}Goodbye!${c.rst}`);
+                process.exit(0);
+                break;
+            case '/clear':
+                clearConsole();
+                messages.length = 1; // Keep system prompt
+                console.log(`${c.green}✓ Context cleared.${c.rst}`);
+                break;
+            case '/model':
+                await selectModel();
+                break;
+            case '/status':
+                console.log(`\n${c.bold}PRITUS STATUS${c.rst}`);
+                console.log(`Tier: ${c.magenta}${config.tier}${c.rst}`);
+                console.log(`Messages in context: ${messages.length - 1}`);
+                console.log(`Workspace: ${process.cwd()}\n`);
+                break;
+            case '/init':
+                fs.writeFileSync('PRITUS.md', '# PRITUS CODE Instructions\n\nAdd your custom project rules here. The agent will read this automatically if referenced.');
+                console.log(`${c.green}✓ Created PRITUS.md${c.rst}`);
+                break;
+            case '/login':
+                await handleLogin();
+                break;
+            case '/help':
+                console.log(`\n${c.bold}Commands:${c.rst}`);
+                console.log(`  @filename    Include file contents in your prompt`);
+                console.log(`  /model       Change AI tier (flash, pro, thinking, code)`);
+                console.log(`  /init        Create a PRITUS.md instruction file`);
+                console.log(`  /clear       Clear conversation history`);
+                console.log(`  /status      Show current agent status`);
+                console.log(`  /login       Authenticate with Google`);
+                console.log(`  /exit        Close PRITUS CODE\n`);
+                break;
+            default:
+                await handleChat(input);
+                break;
+        }
+        
+        printStatusBar();
+        rl.prompt();
+    });
+}
+
+// Handle interrupts gracefully
+rl.on('SIGINT', () => {
+    console.log(`\n${c.dim}Operation cancelled. Type /exit to quit.${c.rst}`);
+    rl.prompt();
+});
+
+main().catch(console.error);