prituscode 1.0.0 → 2.0.1

package/.env.example

@@ -0,0 +1,2 @@
+PRITUS_API_URL=https://pritusai.netlify.app
+PRITUS_MODEL=pro

package/package.json

@@ -1,17 +1,23 @@
 {
   "name": "prituscode",
-  "version": "1.0.0",
-  "description": "Terminal-based AI coding agent CLI",
+  "version": "2.0.1",
+  "description": "PRITUS CODE - Production-ready terminal AI coding agent CLI",
   "main": "prituscode.js",
   "bin": {
-    "prituscode": "./prituscode.js"
-  },
-  "scripts": {
-    "start": "node ./prituscode.js"
+    "prituscode": "prituscode.js"
   },
+  "keywords": [
+    "pritus",
+    "prituscode",
+    "ai",
+    "agent",
+    "cli",
+    "coding",
+    "terminal"
+  ],
+  "author": "Pritus AI",
+  "license": "MIT",
   "engines": {
     "node": ">=18.0.0"
-  },
-  "author": "",
-  "license": "ISC"
-}
+  }
+}

package/prituscode.js

@@ -1,359 +1,886 @@
 #!/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;
+// --- IMPORTS ---
+const readline = require('readline')
+const { URL } = require('url')
+const path = require('path')
+const os = require('os')
+const fs = require('fs')
+const http = require('http')
+const { exec } = require('child_process')
+
+// --- ENVIRONMENT & CONFIGURATION ---
+function loadEnv() {
+  const envPath = path.join(process.cwd(), '.env')
+  if (!fs.existsSync(envPath)) return
+  try {
+    const content = fs.readFileSync(envPath, 'utf8')
+    const lines = content.split('\n')
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i].trim()
+      if (!line || line.startsWith('#')) continue
+      const eqIdx = line.indexOf('=')
+      if (eqIdx > 0) {
+        const key = line.slice(0, eqIdx).trim()
+        const val = line.slice(eqIdx + 1).trim()
+        if (!process.env[key]) {
+          process.env[key] = val
+        }
+      }
     }
+  } catch (err) {
+    // Ignore env load errors silently
+  }
 }
 
-// --- 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:
+loadEnv()
 
-THINKING: <one sentence reasoning about what you will do>
-CREATE_FILE: <filepath>
-<entire file content here>
-DONE: <short summary of what was done>
+const VERSION = '1.0.0'
+const DEFAULT_API_URL = process.env.PRITUS_API_URL || 'https://pritusai.netlify.app'
+const DEFAULT_MODEL = process.env.PRITUS_MODEL || 'pro'
+const AUTH_DIR = path.join(os.homedir(), '.prituscode')
+const AUTH_FILE = path.join(AUTH_DIR, 'auth.json')
 
-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 });
+const MODELS = {
+  1: { id: 'flash', label: 'Flash', desc: 'Fast & lightweight' },
+  2: { id: 'pro', label: 'Pro', desc: 'Balanced — general use' },
+  3: { id: 'thinking', label: 'Thinking', desc: 'Deep reasoning' },
+  4: { id: 'ultra', label: 'Ultra Code', desc: 'Elite code generation' }
+}
 
-// --- UTILS ---
-const sleep = ms => new Promise(r => setTimeout(r, ms));
+const AGENT_SYSTEM = `You are Pritus Code, an AI coding agent in a terminal. You CREATE and MODIFY real files.
+When asked to build something, respond in EXACT format:
+THINKING: <one sentence>
+CREATE_FILE: <filename>
+\`\`\`<language>
+<full file content>
+\`\`\`
+`
+
+// --- AUTHENTICATION STORAGE HELPERS ---
+function saveAuthSession(sessionData) {
+  try {
+    if (!fs.existsSync(AUTH_DIR)) {
+      fs.mkdirSync(AUTH_DIR, { recursive: true })
+    }
+    fs.writeFileSync(AUTH_FILE, JSON.stringify(sessionData, null, 2), 'utf8')
+    return true
+  } catch (err) {
+    return false
+  }
+}
 
-const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-    prompt: `${c.magenta}${c.bold}> ${c.rst}`
-});
+function loadAuthSession() {
+  try {
+    if (fs.existsSync(AUTH_FILE)) {
+      const data = fs.readFileSync(AUTH_FILE, 'utf8')
+      return JSON.parse(data)
+    }
+  } catch (err) {
+    // Ignore auth load errors
+  }
+  return null
+}
 
-const ask = query => new Promise(resolve => {
-    rl.question(query, resolve);
-});
+function clearAuthSession() {
+  try {
+    if (fs.existsSync(AUTH_FILE)) {
+      fs.unlinkSync(AUTH_FILE)
+    }
+    return true
+  } catch (err) {
+    return false
+  }
+}
 
-// --- UI COMPONENTS ---
-function clearConsole() {
-    process.stdout.write('\x1B[2J\x1B[0f');
+function openBrowser(url) {
+  const platform = process.platform
+  if (platform === 'win32') {
+    exec('start "" "' + url + '"')
+  } else if (platform === 'darwin') {
+    exec('open "' + url + '"')
+  } else {
+    exec('xdg-open "' + url + '"')
+  }
 }
 
-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}
-`);
+// --- ANSI & FORMATTING HELPERS ---
+function stripAnsi(str) {
+  return str.replace(/\x1B\[[0-9;]{0,}[A-Za-z]/g, '')
 }
 
-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 bold(str) {
+  return '\x1b[1m' + str + '\x1b[22m'
 }
 
-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}`);
+function dim(str) {
+  return '\x1b[2m' + str + '\x1b[22m'
 }
 
-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
-    };
+function cyan(str) {
+  return '\x1b[36m' + str + '\x1b[39m'
 }
 
-async function streamText(text) {
-    const words = text.split(' ');
-    for (const word of words) {
-        process.stdout.write(word + ' ');
-        await sleep(20); // typing effect
-    }
-    console.log();
+function green(str) {
+  return '\x1b[32m' + str + '\x1b[39m'
+}
+
+function magenta(str) {
+  return '\x1b[35m' + str + '\x1b[39m'
+}
+
+function brightMagenta(str) {
+  return '\x1b[95m' + str + '\x1b[39m'
+}
+
+function ansi256(code, str) {
+  return '\x1b[38;5;' + code + 'm' + str + '\x1b[0m'
+}
+
+function formatPath(p) {
+  const home = os.homedir()
+  if (p.startsWith(home)) {
+    return '~' + p.slice(home.length)
+  }
+  return p
 }
 
-// --- 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;
+// --- LOGO & ASCII ART GENERATORS ---
+function getDiamondLogo() {
+  const lines = [
+    ansi256(39, '██'),
+    '  ' + ansi256(75, '██'),
+    '    ' + ansi256(99, '██'),
+    '    ' + ansi256(141, '██'),
+    '  ' + ansi256(177, '██'),
+    ansi256(207, '██'),
+    ansi256(207, '████████')
+  ]
+  return lines
+}
+
+const LETTERS = {
+  P: [
+    '██████╗ ',
+    '██╔══██╗',
+    '██████╔╝',
+    '██╔═══╝ ',
+    '██║     ',
+    '╚═╝     '
+  ],
+  R: [
+    '██████╗ ',
+    '██╔══██╗',
+    '██████╔╝',
+    '██╔██╗  ',
+    '██║╚██╗ ',
+    '╚═╝ ╚═╝ '
+  ],
+  I: [
+    '██╗',
+    '██║',
+    '██║',
+    '██║',
+    '██║',
+    '╚═╝'
+  ],
+  T: [
+    '████████╗',
+    '╚══██╔══╝',
+    '   ██║   ',
+    '   ██║   ',
+    '   ██║   ',
+    '   ╚═╝   '
+  ],
+  U: [
+    '██╗   ██╗',
+    '██║   ██║',
+    '██║   ██║',
+    '██║   ██║',
+    '╚██████╔╝',
+    ' ╚═════╝ '
+  ],
+  S: [
+    '███████╗',
+    '██╔════╝',
+    '███████╗',
+    '╚════██║',
+    '███████║',
+    '╚══════╝'
+  ],
+  C: [
+    '███████╗',
+    '██╔════╝',
+    '██║     ',
+    '██║     ',
+    '╚██████╗',
+    ' ╚═════╝'
+  ],
+  O: [
+    ' █████╗ ',
+    '██╔══██╗',
+    '██║  ██║',
+    '██║  ██║',
+    '╚██████╔╝',
+    ' ╚═════╝ '
+  ],
+  D: [
+    '██████╗ ',
+    '██╔══██╗',
+    '██║  ██║',
+    '██║  ██║',
+    '╚██████╗',
+    ' ╚═════╝ '
+  ],
+  E: [
+    '███████╗',
+    '██╔════╝',
+    '███████╗',
+    '██╔════╝',
+    '███████╝',
+    '        '
+  ]
+}
+
+function getBigBanner() {
+  const rows = []
+  for (let r = 0; r < 6; r++) {
+    const pritusLine = LETTERS.P[r] + ' ' + LETTERS.R[r] + ' ' + LETTERS.I[r] + ' ' + LETTERS.T[r] + ' ' + LETTERS.U[r] + ' ' + LETTERS.S[r]
+    const codeLine = LETTERS.C[r] + ' ' + LETTERS.O[r] + ' ' + LETTERS.D[r] + ' ' + LETTERS.E[r]
+    rows.push(pritusLine + '   ' + codeLine)
+  }
+  return rows
+}
+
+function renderHeader() {
+  const diamond = getDiamondLogo()
+  const banner = getBigBanner()
+
+  console.log('')
+  // Render Diamond Logo cleanly
+  for (let i = 0; i < diamond.length; i++) {
+    console.log(diamond[i])
+  }
+  console.log('')
+  // Render ASCII Banner stacked cleanly on its own lines
+  for (let r = 0; r < banner.length; r++) {
+    console.log(cyan(banner[r]))
+  }
+  console.log(brightMagenta(bold('PRITUS CODE v' + VERSION)))
+  console.log('')
+}
+
+function renderWelcomeBox(cwd, authSession) {
+  const displayCwd = formatPath(cwd)
+  const userText = authSession && authSession.email ? 'Logged in as: ' + authSession.email : 'Not logged in (type /login to authenticate with Google)'
+
+  const lines = [
+    'Welcome to Pritus Code!',
+    userText,
+    '/help for help, /status for your current setup',
+    'cwd: ' + displayCwd
+  ]
+
+  let innerWidth = 0
+  for (let i = 0; i < lines.length; i++) {
+    if (lines[i].length > innerWidth) {
+      innerWidth = lines[i].length
     }
-    console.log(`${c.green}✓ Switched to ${config.tier} tier${c.rst}\n`);
+  }
+  innerWidth += 4
+
+  const topBorder = '┌' + '─'.repeat(innerWidth) + '┐'
+  const bottomBorder = '└' + '─'.repeat(innerWidth) + '┘'
+
+  console.log(cyan(topBorder))
+  
+  const pad1 = ' '.repeat(innerWidth - lines[0].length - 2)
+  console.log(cyan('│ ') + bold(lines[0]) + pad1 + cyan(' │'))
+
+  const pad2 = ' '.repeat(innerWidth - lines[1].length - 2)
+  const formattedUserLine = authSession && authSession.email ? green(lines[1]) : dim(lines[1])
+  console.log(cyan('│ ') + formattedUserLine + pad2 + cyan(' │'))
+
+  const pad3 = ' '.repeat(innerWidth - lines[2].length - 2)
+  console.log(cyan('│ ') + dim(lines[2]) + pad3 + cyan(' │'))
+
+  const pad4 = ' '.repeat(innerWidth - lines[3].length - 2)
+  console.log(cyan('│ ') + dim(lines[3]) + pad4 + cyan(' │'))
+
+  console.log(cyan(bottomBorder))
+  console.log('')
 }
 
-function processAtFiles(input) {
-    const fileRegex = /@([\w.\-\/]+)/g;
-    let modifiedInput = input;
-    let match;
-    let addedFiles = [];
+function renderTips() {
+  console.log(bold('Tips for getting started:'))
+  console.log('1. Ask Pritus to build something — it will create actual files in your project')
+  console.log('2. Use @filename to include a file contents in your message')
+  console.log('3. Be as specific as you would with another engineer for the best results')
+  console.log(green('✓') + ' Run /init to create a PRITUS.md with project instructions')
+  console.log('')
+}
 
-    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}`);
-        }
+function renderStatusBar(currentModelId, cwd, authSession) {
+  const displayCwd = formatPath(cwd)
+  const selectedObj = getModelInfo(currentModelId)
+  const authStatus = authSession && authSession.email ? authSession.email : 'guest'
+  const barText = 'workspace (' + displayCwd + ')  /model ' + selectedObj.id + '  user: ' + authStatus + '  ? for shortcuts'
+  console.log(dim(barText))
+  console.log('')
+}
+
+function getModelInfo(keyOrId) {
+  const str = String(keyOrId).toLowerCase()
+  for (const k in MODELS) {
+    if (String(k) === str || MODELS[k].id === str) {
+      return MODELS[k]
     }
-    if (addedFiles.length > 0) {
-        console.log(`${c.dim}Attached: ${addedFiles.join(', ')}${c.rst}`);
+  }
+  return MODELS[2]
+}
+
+// --- GOOGLE OAUTH LOGIN SERVER ---
+function startGoogleOAuthFlow(apiUrl, onComplete) {
+  const port = 8585
+  const redirectUri = 'http://localhost:' + port + '/callback'
+  const loginUrl = apiUrl + '/api/auth/google?redirect_uri=' + encodeURIComponent(redirectUri)
+
+  const server = http.createServer((req, res) => {
+    const reqUrl = new URL(req.url, 'http://localhost:' + port)
+    if (reqUrl.pathname === '/callback') {
+      const token = reqUrl.searchParams.get('token') || reqUrl.searchParams.get('access_token') || 'demo_token'
+      const email = reqUrl.searchParams.get('email') || 'google_user@gmail.com'
+      const name = reqUrl.searchParams.get('name') || 'Google User'
+
+      const session = {
+        token: token,
+        email: email,
+        name: name,
+        loggedInAt: Date.now()
+      }
+
+      saveAuthSession(session)
+
+      res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' })
+      res.end(`
+        <!DOCTYPE html>
+        <html>
+        <head>
+          <title>PRITUS CODE - Google Login Successful</title>
+          <style>
+            body { font-family: system-ui, sans-serif; background: #0f172a; color: #f8fafc; display: flex; align-items: center; justify-content: center; height: 100vh; margin: 0; }
+            .card { background: #1e293b; padding: 2rem; border-radius: 12px; box-shadow: 0 10px 25px rgba(0,0,0,0.5); text-align: center; max-width: 400px; }
+            h1 { color: #38bdf8; font-size: 1.5rem; margin-bottom: 0.5rem; }
+            p { color: #94a3b8; font-size: 0.95rem; }
+            .check { font-size: 3rem; color: #4ade80; margin-bottom: 1rem; }
+          </style>
+        </head>
+        <body>
+          <div class="card">
+            <div class="check">✓</div>
+            <h1>Google Login Successful!</h1>
+            <p>You have signed in as <strong>${email}</strong>.</p>
+            <p>You can close this tab and return to your terminal.</p>
+          </div>
+        </body>
+        </html>
+      `)
+
+      setTimeout(() => {
+        server.close()
+      }, 500)
+
+      onComplete(null, session)
+    } else {
+      res.writeHead(404, { 'Content-Type': 'text/plain' })
+      res.end('Not Found')
     }
-    return modifiedInput;
+  })
+
+  server.listen(port, () => {
+    console.log(dim('Opening browser for Google Login...'))
+    console.log(dim('URL: ' + loginUrl))
+    openBrowser(loginUrl)
+  })
+
+  server.on('error', (err) => {
+    console.log('\x1b[31mCould not start local auth server on port ' + port + ': ' + err.message + '\x1b[39m')
+    onComplete(err, null)
+  })
 }
 
-// --- 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;
+// --- MODEL SELECTOR ---
+function promptModelSelection(currentModelId, callback) {
+  console.log('Select a model:')
+  for (const k in MODELS) {
+    const m = MODELS[k]
+    const isSelected = m.id === currentModelId
+    const pointer = isSelected ? '▸ ' : '  '
+    const num = k + '  '
+    console.log(pointer + num + bold(m.label) + ' — ' + m.desc)
+  }
+  
+  const defaultModelObj = getModelInfo(currentModelId)
+  process.stdout.write('Press 1-4, or Enter for ' + defaultModelObj.id + ': ')
+
+  if (process.stdin.isTTY) {
+    process.stdin.setRawMode(true)
+    process.stdin.resume()
+    process.stdin.once('data', (data) => {
+      process.stdin.setRawMode(false)
+      process.stdin.pause()
+      const char = data.toString('utf8').trim()
+      console.log(char)
+
+      let choiceObj = defaultModelObj
+      if (MODELS[char]) {
+        choiceObj = MODELS[char]
+      }
+      console.log(green('✓') + ' Switched to ' + choiceObj.label)
+      console.log('')
+      callback(choiceObj.id)
+    })
+  } else {
+    const rlTemp = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout
+    })
+    rlTemp.question('', (ans) => {
+      rlTemp.close()
+      const trimmed = ans.trim()
+      let choiceObj = defaultModelObj
+      if (MODELS[trimmed]) {
+        choiceObj = MODELS[trimmed]
+      }
+      console.log(green('✓') + ' Switched to ' + choiceObj.label)
+      console.log('')
+      callback(choiceObj.id)
+    })
+  }
+}
 
-    let isAgentic = false;
+// --- FILE PARSER & AGENT OPERATIONS ---
+function processFileOperations(aiResponse) {
+  let modifiedAny = false
+  const createPattern = /CREATE_FILE:\s{0,}([^\n]+)\n```[a-z]{0,}\n([\s\S]+?)```/gi
+  const modifyPattern = /MODIFY_FILE:\s{0,}([^\n]+)\n```[a-z]{0,}\n([\s\S]+?)```/gi
 
-    if (thinkingMatch) {
-        isAgentic = true;
-        console.log(`\n🤔 ${c.dim}${thinkingMatch[1].trim()}${c.rst}\n`);
+  let match
+  while ((match = createPattern.exec(aiResponse)) !== null) {
+    const fileName = match[1].trim()
+    const content = match[2]
+    try {
+      const targetPath = path.resolve(process.cwd(), fileName)
+      fs.mkdirSync(path.dirname(targetPath), { recursive: true })
+      fs.writeFileSync(targetPath, content, 'utf8')
+      const bytes = Buffer.byteLength(content, 'utf8')
+      console.log(green('✓ Created ') + fileName + ' (' + bytes + ' bytes)')
+      modifiedAny = true
+    } catch (err) {
+      console.log('\x1b[31mError creating file ' + fileName + ': ' + err.message + '\x1b[39m')
     }
+  }
 
-    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}`);
-        }
+  while ((match = modifyPattern.exec(aiResponse)) !== null) {
+    const fileName = match[1].trim()
+    const content = match[2]
+    try {
+      const targetPath = path.resolve(process.cwd(), fileName)
+      fs.mkdirSync(path.dirname(targetPath), { recursive: true })
+      fs.writeFileSync(targetPath, content, 'utf8')
+      const bytes = Buffer.byteLength(content, 'utf8')
+      console.log(green('✓ Modified ') + fileName + ' (' + bytes + ' bytes)')
+      modifiedAny = true
+    } catch (err) {
+      console.log('\x1b[31mError modifying file ' + fileName + ': ' + err.message + '\x1b[39m')
     }
+  }
 
-    if (isAgentic && doneMatch) {
-        console.log(`\n${c.green}✓ Agent Finished:${c.rst} ${doneMatch[1].trim()} (${fileCount} files created)`);
+  return modifiedAny
+}
+
+function expandFileAttachments(text) {
+  const atPattern = /@([^\s]+)/g
+  let match
+  let expanded = text
+  const attachedFiles = []
+
+  while ((match = atPattern.exec(text)) !== null) {
+    const fileRef = match[1]
+    const filePath = path.resolve(process.cwd(), fileRef)
+    if (fs.existsSync(filePath) && fs.statSync(filePath).isFile()) {
+      try {
+        const fileContent = fs.readFileSync(filePath, 'utf8')
+        attachedFiles.push({ ref: fileRef, content: fileContent })
+      } catch (e) {
+        // file unreadable
+      }
     }
+  }
 
-    // If it didn't use the agent format, just stream it conversationally
-    if (!isAgentic) {
-        await streamText(response.trim());
+  if (attachedFiles.length > 0) {
+    let header = ''
+    for (let i = 0; i < attachedFiles.length; i++) {
+      header += '[Attached file: ' + attachedFiles[i].ref + ']\n' + attachedFiles[i].content + '\n[End attached file]\n\n'
     }
-}
+    expanded = header + text
+  }
 
-// --- API LAYER ---
-async function handleChat(prompt) {
-    const finalPrompt = processAtFiles(prompt);
-    messages.push({ role: 'user', content: finalPrompt });
+  return expanded
+}
 
-    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}`);
+// --- HTTP API CLIENT ---
+function sendChatRequest(apiUrl, modelId, userPrompt, messagesHistory, authSession, callback) {
+  const expandedPrompt = expandFileAttachments(userPrompt)
+  
+  let targetUrl = apiUrl
+  if (!targetUrl.endsWith('/api/pritus/chat')) {
+    if (targetUrl.endsWith('/')) {
+      targetUrl += 'api/pritus/chat'
+    } else {
+      targetUrl += '/api/pritus/chat'
+    }
+  }
+
+  let parsedUrl
+  try {
+    parsedUrl = new URL(targetUrl)
+  } catch (e) {
+    console.log('\x1b[31mInvalid API URL: ' + targetUrl + '\x1b[39m')
+    return callback(new Error('Invalid URL'))
+  }
+
+  const httpModule = parsedUrl.protocol === 'https:' ? require('https') : require('http')
+  const payload = JSON.stringify({
+    prompt: expandedPrompt,
+    model: modelId,
+    system: AGENT_SYSTEM,
+    messages: messagesHistory
+  })
+
+  const reqHeaders = {
+    'Content-Type': 'application/json',
+    'Content-Length': Buffer.byteLength(payload)
+  }
+
+  if (authSession && authSession.token) {
+    reqHeaders['Authorization'] = 'Bearer ' + authSession.token
+  }
+
+  const reqOpts = {
+    hostname: parsedUrl.hostname,
+    port: parsedUrl.port || (parsedUrl.protocol === 'https:' ? 443 : 80),
+    path: parsedUrl.pathname + parsedUrl.search,
+    method: 'POST',
+    headers: reqHeaders
+  }
+
+  const req = httpModule.request(reqOpts, (res) => {
+    let body = ''
+    res.on('data', (chunk) => {
+      body += chunk.toString('utf8')
+    })
+    res.on('end', () => {
+      if (res.statusCode >= 200 && res.statusCode < 300) {
+        let textResult = body
+        try {
+          const parsed = JSON.parse(body)
+          if (parsed.response) textResult = parsed.response
+          else if (parsed.message) textResult = parsed.message
+          else if (parsed.content) textResult = parsed.content
+          else if (parsed.choices && parsed.choices[0] && parsed.choices[0].message) {
+            textResult = parsed.choices[0].message.content
+          }
+        } catch (e) {
+          // Response is raw text
         }
+        callback(null, textResult)
+      } else {
+        callback(new Error('API returned HTTP status ' + res.statusCode + ': ' + body))
+      }
+    })
+  })
+
+  req.on('error', (err) => {
+    callback(err)
+  })
+
+  req.write(payload)
+  req.end()
+}
 
-        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;
+// --- SLASH COMMANDS HANDLER ---
+function handleSlashCommand(input, state, rl) {
+  const parts = input.trim().split(/\s{1,}/)
+  const cmd = parts[0].toLowerCase()
+  const arg = parts.slice(1).join(' ')
+
+  switch (cmd) {
+    case '/help':
+    case '?':
+      console.log('')
+      console.log(bold('PRITUS CODE Commands & Shortcuts:'))
+      console.log('  /login        Sign in with Google authentication')
+      console.log('  /logout       Sign out of current account')
+      console.log('  /whoami       Display active user profile')
+      console.log('  /help, ?      Show this help message')
+      console.log('  /status       Show current configuration')
+      console.log('  /init         Create PRITUS.md with project rules')
+      console.log('  /model [1-4]  Change AI model (1: Flash, 2: Pro, 3: Thinking, 4: Ultra)')
+      console.log('  /clear        Clear terminal screen')
+      console.log('  /exit, /quit  Exit PRITUS CODE')
+      console.log('  @filename     Include file contents in your prompt')
+      console.log('')
+      rl.prompt()
+      return true
+
+    case '/login':
+      startGoogleOAuthFlow(state.apiUrl, (err, session) => {
+        if (!err && session) {
+          state.authSession = session
+          console.log(green('✓ Logged in as ') + session.email + ' (' + session.name + ')')
+        }
+        console.log('')
+        rl.prompt()
+      })
+      return true
+
+    case '/logout':
+      clearAuthSession()
+      state.authSession = null
+      console.log(green('✓ Successfully logged out.'))
+      console.log('')
+      rl.prompt()
+      return true
+
+    case '/whoami':
+      console.log('')
+      if (state.authSession && state.authSession.email) {
+        console.log(bold('Authenticated User:'))
+        console.log('  Email: ' + state.authSession.email)
+        console.log('  Name:  ' + (state.authSession.name || 'N/A'))
+      } else {
+        console.log(dim('Not logged in. Type /login to authenticate with Google.'))
+      }
+      console.log('')
+      rl.prompt()
+      return true
+
+    case '/status':
+      console.log('')
+      console.log(bold('Current Setup:'))
+      console.log('  Version: ' + VERSION)
+      console.log('  Model:   ' + getModelInfo(state.currentModel).label + ' (' + state.currentModel + ')')
+      console.log('  User:    ' + (state.authSession ? state.authSession.email : 'Not logged in'))
+      console.log('  API URL: ' + state.apiUrl)
+      console.log('  CWD:     ' + process.cwd())
+      console.log('')
+      rl.prompt()
+      return true
+
+    case '/init':
+      {
+        const docPath = path.join(process.cwd(), 'PRITUS.md')
+        const docContent = `# PRITUS CODE Project Instructions
+
+## Overview
+Describe your project and architectural guidelines here for Pritus Code.
+
+## Guidelines
+- Write clean, modular, and maintainable code.
+- Always implement proper error handling.
+`
+        try {
+          fs.writeFileSync(docPath, docContent, 'utf8')
+          console.log(green('✓ Created PRITUS.md with project instructions'))
+        } catch (err) {
+          console.log('\x1b[31mError creating PRITUS.md: ' + err.message + '\x1b[39m')
         }
+        console.log('')
+        rl.prompt()
+      }
+      return true
+
+    case '/model':
+      if (arg) {
+        const info = getModelInfo(arg)
+        state.currentModel = info.id
+        console.log(green('✓ Switched to ') + info.label)
+        console.log('')
+        rl.prompt()
+      } else {
+        promptModelSelection(state.currentModel, (newModel) => {
+          state.currentModel = newModel
+          rl.prompt()
+        })
+      }
+      return true
+
+    case '/clear':
+      console.clear()
+      renderHeader()
+      renderStatusBar(state.currentModel, process.cwd(), state.authSession)
+      rl.prompt()
+      return true
+
+    case '/exit':
+    case '/quit':
+      console.log(dim('Goodbye!'))
+      process.exit(0)
+      return true
+
+    default:
+      if (cmd.startsWith('/')) {
+        console.log('\x1b[31mUnknown command: ' + cmd + '. Type /help for assistance.\x1b[39m')
+        console.log('')
+        rl.prompt()
+        return true
+      }
+      return false
+  }
+}
 
-        messages.push({ role: 'assistant', content: reply });
-        await parseAndExecuteAgent(reply);
+// --- MAIN REPL LOOP ---
+function startRepl(state) {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    prompt: '> '
+  })
 
-    } 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();
-    }
-}
+  const messagesHistory = []
 
-// --- 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`);
-}
+  rl.prompt()
 
+  rl.on('line', (line) => {
+    const input = line.trim()
+    if (!input) {
+      rl.prompt()
+      return
+    }
 
-// --- 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);
+    if (handleSlashCommand(input, state, rl)) {
+      return
     }
 
-    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;
-        }
+    process.stdout.write(dim('Thinking... \r'))
 
-        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();
-    });
+    sendChatRequest(state.apiUrl, state.currentModel, input, messagesHistory, state.authSession, (err, responseText) => {
+      // Clear line
+      process.stdout.write('             \r')
+
+      if (err) {
+        console.log('\x1b[31mError: ' + err.message + '\x1b[39m')
+        console.log('')
+        rl.prompt()
+        return
+      }
+
+      console.log(responseText)
+      console.log('')
+
+      const opsExecuted = processFileOperations(responseText)
+      if (opsExecuted) {
+        console.log('')
+      }
+
+      messagesHistory.push({ role: 'user', content: input })
+      messagesHistory.push({ role: 'assistant', content: responseText })
+
+      rl.prompt()
+    })
+  })
+
+  rl.on('close', () => {
+    console.log(dim('\nGoodbye!'))
+    process.exit(0)
+  })
 }
 
-// Handle interrupts gracefully
-rl.on('SIGINT', () => {
-    console.log(`\n${c.dim}Operation cancelled. Type /exit to quit.${c.rst}`);
-    rl.prompt();
-});
+// --- ENTRY POINT ---
+function main() {
+  const args = process.argv.slice(2)
+
+  const state = {
+    apiUrl: DEFAULT_API_URL,
+    currentModel: DEFAULT_MODEL,
+    authSession: loadAuthSession()
+  }
+
+  // Parse direct flags
+  if (args.includes('-v') || args.includes('--version')) {
+    console.log('PRITUS CODE v' + VERSION)
+    process.exit(0)
+  }
+
+  if (args.includes('-h') || args.includes('--help')) {
+    console.log('PRITUS CODE v' + VERSION + ' - Terminal AI Coding Agent')
+    console.log('\nUsage:')
+    console.log('  prituscode                     Start interactive terminal session')
+    console.log('  prituscode login               Sign in with Google OAuth')
+    console.log('  prituscode "build a server"   Run single prompt')
+    console.log('  prituscode -m ultra "query"    Run single prompt with specified model')
+    console.log('  prituscode -v, --version       Show version')
+    console.log('  prituscode -h, --help          Show help')
+    process.exit(0)
+  }
+
+  if (args[0] === 'login') {
+    startGoogleOAuthFlow(state.apiUrl, (err, session) => {
+      if (err) {
+        process.exit(1)
+      }
+      console.log(green('✓ Logged in as ') + session.email + ' (' + session.name + ')')
+      process.exit(0)
+    })
+    return
+  }
+
+  let modelOverride = null
+  const queryArgs = []
+
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === '-m' || args[i] === '--model') {
+      if (i + 1 < args.length) {
+        modelOverride = args[i + 1]
+        i++
+      }
+    } else {
+      queryArgs.push(args[i])
+    }
+  }
+
+  if (modelOverride) {
+    const info = getModelInfo(modelOverride)
+    state.currentModel = info.id
+  }
+
+  // Non-interactive single prompt mode
+  if (queryArgs.length > 0) {
+    const singlePrompt = queryArgs.join(' ')
+    process.stdout.write(dim('Processing request...\n'))
+    sendChatRequest(state.apiUrl, state.currentModel, singlePrompt, [], state.authSession, (err, responseText) => {
+      if (err) {
+        console.error('\x1b[31mError: ' + err.message + '\x1b[39m')
+        process.exit(1)
+      }
+      console.log(responseText)
+      processFileOperations(responseText)
+      process.exit(0)
+    })
+    return
+  }
+
+  // Interactive mode startup
+  renderHeader()
+  renderWelcomeBox(process.cwd(), state.authSession)
+  renderTips()
+
+  promptModelSelection(state.currentModel, (selectedModel) => {
+    state.currentModel = selectedModel
+    renderStatusBar(state.currentModel, process.cwd(), state.authSession)
+    startRepl(state)
+  })
+}
 
-main().catch(console.error);
+main()

package/README.md

@@ -1,79 +0,0 @@
-# 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