prituscode 2.0.0 → 2.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prituscode",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "PRITUS CODE - Production-ready terminal AI coding agent CLI",
   "main": "prituscode.js",
   "bin": {

package/prituscode.js

@@ -6,6 +6,8 @@ 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() {
@@ -36,6 +38,8 @@ loadEnv()
 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')
 
 const MODELS = {
   1: { id: 'flash', label: 'Flash', desc: 'Fast & lightweight' },
@@ -53,6 +57,53 @@ CREATE_FILE: <filename>
 \`\`\`
 `
 
+// --- 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
+  }
+}
+
+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
+}
+
+function clearAuthSession() {
+  try {
+    if (fs.existsSync(AUTH_FILE)) {
+      fs.unlinkSync(AUTH_FILE)
+    }
+    return true
+  } catch (err) {
+    return false
+  }
+}
+
+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 + '"')
+  }
+}
+
 // --- ANSI & FORMATTING HELPERS ---
 function stripAnsi(str) {
   return str.replace(/\x1B\[[0-9;]{0,}[A-Za-z]/g, '')
@@ -206,24 +257,36 @@ function renderHeader() {
   const banner = getBigBanner()
 
   console.log('')
-  for (let i = 0; i < 7; i++) {
-    const dia = diamond[i]
-    const ban = i < 6 ? cyan(banner[i]) : ''
-    console.log(dia + '   ' + ban)
+  // 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) {
+function renderWelcomeBox(cwd, authSession) {
   const displayCwd = formatPath(cwd)
-  const line1 = 'Welcome to Pritus Code!'
-  const line2 = '/help for help, /status for your current setup'
-  const line3 = 'cwd: ' + displayCwd
+  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 = line1.length
-  if (line2.length > innerWidth) innerWidth = line2.length
-  if (line3.length > innerWidth) innerWidth = line3.length
+  let innerWidth = 0
+  for (let i = 0; i < lines.length; i++) {
+    if (lines[i].length > innerWidth) {
+      innerWidth = lines[i].length
+    }
+  }
   innerWidth += 4
 
   const topBorder = '┌' + '─'.repeat(innerWidth) + '┐'
@@ -231,14 +294,18 @@ function renderWelcomeBox(cwd) {
 
   console.log(cyan(topBorder))
   
-  const pad1 = ' '.repeat(innerWidth - line1.length - 2)
-  console.log(cyan('│ ') + bold(line1) + pad1 + cyan(' │'))
+  const pad1 = ' '.repeat(innerWidth - lines[0].length - 2)
+  console.log(cyan('│ ') + bold(lines[0]) + pad1 + cyan(' │'))
 
-  const pad2 = ' '.repeat(innerWidth - line2.length - 2)
-  console.log(cyan('│ ') + dim(line2) + pad2 + 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 - line3.length - 2)
-  console.log(cyan('│ ') + dim(line3) + pad3 + 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('')
@@ -253,10 +320,11 @@ function renderTips() {
   console.log('')
 }
 
-function renderStatusBar(currentModelId, cwd) {
+function renderStatusBar(currentModelId, cwd, authSession) {
   const displayCwd = formatPath(cwd)
   const selectedObj = getModelInfo(currentModelId)
-  const barText = 'workspace (' + displayCwd + ')  /model ' + selectedObj.id + '  ? for shortcuts'
+  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('')
 }
@@ -271,6 +339,76 @@ function getModelInfo(keyOrId) {
   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')
+    }
+  })
+
+  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)
+  })
+}
+
 // --- MODEL SELECTOR ---
 function promptModelSelection(currentModelId, callback) {
   console.log('Select a model:')
@@ -392,7 +530,7 @@ function expandFileAttachments(text) {
 }
 
 // --- HTTP API CLIENT ---
-function sendChatRequest(apiUrl, modelId, userPrompt, messagesHistory, callback) {
+function sendChatRequest(apiUrl, modelId, userPrompt, messagesHistory, authSession, callback) {
   const expandedPrompt = expandFileAttachments(userPrompt)
   
   let targetUrl = apiUrl
@@ -420,15 +558,21 @@ function sendChatRequest(apiUrl, modelId, userPrompt, messagesHistory, callback)
     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: {
-      'Content-Type': 'application/json',
-      'Content-Length': Buffer.byteLength(payload)
-    }
+    headers: reqHeaders
   }
 
   const req = httpModule.request(reqOpts, (res) => {
@@ -476,6 +620,9 @@ function handleSlashCommand(input, state, rl) {
     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')
@@ -487,11 +634,44 @@ function handleSlashCommand(input, state, rl) {
       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('')
@@ -539,7 +719,7 @@ Describe your project and architectural guidelines here for Pritus Code.
     case '/clear':
       console.clear()
       renderHeader()
-      renderStatusBar(state.currentModel, process.cwd())
+      renderStatusBar(state.currentModel, process.cwd(), state.authSession)
       rl.prompt()
       return true
 
@@ -585,7 +765,7 @@ function startRepl(state) {
 
     process.stdout.write(dim('Thinking... \r'))
 
-    sendChatRequest(state.apiUrl, state.currentModel, input, messagesHistory, (err, responseText) => {
+    sendChatRequest(state.apiUrl, state.currentModel, input, messagesHistory, state.authSession, (err, responseText) => {
       // Clear line
       process.stdout.write('             \r')
 
@@ -623,7 +803,8 @@ function main() {
 
   const state = {
     apiUrl: DEFAULT_API_URL,
-    currentModel: DEFAULT_MODEL
+    currentModel: DEFAULT_MODEL,
+    authSession: loadAuthSession()
   }
 
   // Parse direct flags
@@ -636,6 +817,7 @@ function main() {
     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')
@@ -643,6 +825,17 @@ function main() {
     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 = []
 
@@ -666,7 +859,7 @@ function main() {
   if (queryArgs.length > 0) {
     const singlePrompt = queryArgs.join(' ')
     process.stdout.write(dim('Processing request...\n'))
-    sendChatRequest(state.apiUrl, state.currentModel, singlePrompt, [], (err, responseText) => {
+    sendChatRequest(state.apiUrl, state.currentModel, singlePrompt, [], state.authSession, (err, responseText) => {
       if (err) {
         console.error('\x1b[31mError: ' + err.message + '\x1b[39m')
         process.exit(1)
@@ -680,12 +873,12 @@ function main() {
 
   // Interactive mode startup
   renderHeader()
-  renderWelcomeBox(process.cwd())
+  renderWelcomeBox(process.cwd(), state.authSession)
   renderTips()
 
   promptModelSelection(state.currentModel, (selectedModel) => {
     state.currentModel = selectedModel
-    renderStatusBar(state.currentModel, process.cwd())
+    renderStatusBar(state.currentModel, process.cwd(), state.authSession)
     startRepl(state)
   })
 }