icoa-cli 2.3.1 → 2.3.3

package/dist/commands/ai4ctf.d.ts

@@ -1,2 +1,4 @@
 import { Command } from 'commander';
+export declare function isChatActive(): boolean;
+export declare function handleChatMessage(input: string): Promise<'continue' | 'exit'>;
 export declare function registerAi4ctfCommand(program: Command): void;

package/dist/commands/ai4ctf.js

@@ -1,4 +1,3 @@
-import { createInterface } from 'node:readline';
 import chalk from 'chalk';
 import { createChatSession } from '../lib/gemini.js';
 import { isTokenCapReached, addTokenUsage, getTokenUsage } from '../lib/budget.js';
@@ -12,28 +11,68 @@ function getChallengeContext() {
     }
     return undefined;
 }
+// Chat state — shared with REPL
+let chatActive = false;
+let chatSession = null;
+export function isChatActive() {
+    return chatActive;
+}
+export async function handleChatMessage(input) {
+    if (!chatSession)
+        return 'exit';
+    if (input === 'exit' || input === 'back' || input === 'quit') {
+        chatActive = false;
+        chatSession = null;
+        console.log();
+        printInfo('Returning to ICOA terminal.');
+        console.log();
+        return 'exit';
+    }
+    if (isTokenCapReached()) {
+        chatActive = false;
+        chatSession = null;
+        printError('Token budget exhausted. Exiting chat mode.');
+        return 'exit';
+    }
+    logCommand(`ai4ctf: ${input}`);
+    console.log(chalk.gray('  Thinking...'));
+    try {
+        const response = await chatSession.sendMessage(input);
+        process.stdout.write('\x1b[1A\x1b[2K');
+        addTokenUsage(response.tokensUsed);
+        const updated = getTokenUsage();
+        console.log();
+        printMarkdown(response.text);
+        console.log(chalk.gray(`  [${response.tokensUsed.toLocaleString()} tokens | ${updated.used.toLocaleString()}/${updated.cap.toLocaleString()} total]`));
+        console.log();
+    }
+    catch (err) {
+        process.stdout.write('\x1b[1A\x1b[2K');
+        printError(`AI error: ${err.message}`);
+        console.log();
+    }
+    return 'continue';
+}
 export function registerAi4ctfCommand(program) {
     program
         .command('ai4ctf')
         .description('Chat with your AI teammate')
         .action(async () => {
         logCommand('ai4ctf');
-        // Check token cap
         if (isTokenCapReached()) {
             printError('Token budget exhausted. No more AI interactions available.');
             return;
         }
         const context = getChallengeContext();
         const tokenState = getTokenUsage();
-        // Create chat session
-        let chat;
         try {
-            chat = await createChatSession(context);
+            chatSession = await createChatSession(context);
         }
         catch (err) {
             printError(err.message);
             return;
         }
+        chatActive = true;
         // Welcome banner
         console.log();
         console.log(chalk.magenta('  ┌─────────────────────────────────────────┐'));
@@ -47,64 +86,5 @@ export function registerAi4ctfCommand(program) {
         console.log(chalk.magenta('  │') + chalk.gray("  Type 'exit' to return".padEnd(41)) + chalk.magenta('│'));
         console.log(chalk.magenta('  └─────────────────────────────────────────┘'));
         console.log();
-        // Inner chat REPL
-        return new Promise((resolve) => {
-            const chatRl = createInterface({
-                input: process.stdin,
-                output: process.stdout,
-                prompt: chalk.magenta('ai> '),
-                terminal: true,
-            });
-            chatRl.prompt();
-            chatRl.on('line', async (line) => {
-                const input = line.trim();
-                if (!input) {
-                    chatRl.prompt();
-                    return;
-                }
-                // Exit commands
-                if (input === 'exit' || input === 'back' || input === 'quit') {
-                    chatRl.close();
-                    return;
-                }
-                // Token cap check
-                if (isTokenCapReached()) {
-                    console.log();
-                    printError('Token budget exhausted. Exiting chat mode.');
-                    chatRl.close();
-                    return;
-                }
-                // Log the chat message
-                logCommand(`ai4ctf: ${input}`);
-                // Send message
-                console.log(chalk.gray('  Thinking...'));
-                try {
-                    const response = await chat.sendMessage(input);
-                    // Clear "Thinking..." line
-                    process.stdout.write('\x1b[1A\x1b[2K');
-                    // Track tokens
-                    addTokenUsage(response.tokensUsed);
-                    const updated = getTokenUsage();
-                    // Display response
-                    console.log();
-                    printMarkdown(response.text);
-                    console.log(chalk.gray(`  [${response.tokensUsed.toLocaleString()} tokens | ${updated.used.toLocaleString()}/${updated.cap.toLocaleString()} total]`));
-                    console.log();
-                }
-                catch (err) {
-                    // Clear "Thinking..." line
-                    process.stdout.write('\x1b[1A\x1b[2K');
-                    printError(`AI error: ${err.message}`);
-                    console.log();
-                }
-                chatRl.prompt();
-            });
-            chatRl.on('close', () => {
-                console.log();
-                printInfo('Returning to ICOA terminal.');
-                console.log();
-                resolve();
-            });
-        });
     });
 }

package/dist/index.js

@@ -37,7 +37,7 @@ ${LINE}
    ${chalk.white('Sydney, Australia')}  ${chalk.gray('Jun 27 - Jul 2, 2026')}
    ${chalk.cyan.underline('https://icoa2026.au')}
 
-   ${chalk.gray('CLI-Native Competition Terminal  v2.3.1')}
+   ${chalk.gray('CLI-Native Competition Terminal  v2.3.3')}
 
 ${LINE}
 `;

package/dist/repl.js

@@ -4,6 +4,7 @@ import chalk from 'chalk';
 import { isConnected, getConfig } from './lib/config.js';
 import { isActivated, activateToken, isFreeCommand, isDeviceMatch, recordExit, recordResume, isFirstRunOrUpgrade, markVersionSeen } from './lib/access.js';
 import { setReplMode } from './lib/ui.js';
+import { isChatActive, handleChatMessage } from './commands/ai4ctf.js';
 import { resetTerminalTheme } from './lib/theme.js';
 import { ensureSandbox, runInSandbox, isDockerAvailable } from './lib/sandbox.js';
 import { logCommand } from './lib/logger.js';
@@ -27,7 +28,7 @@ const BLOCKED_COMMANDS = new Set([
     'iptables', 'ufw', // firewall
 ]);
 const INTERCEPT = '__REPL_NO_EXIT__';
-const VERSION = '2.3.1';
+const VERSION = '2.3.3';
 export async function startRepl(program, resumeMode) {
     const config = getConfig();
     const connected = isConnected();
@@ -147,6 +148,18 @@ export async function startRepl(program, resumeMode) {
             return;
         const input = line.trim();
         if (!input) {
+            rl.setPrompt(isChatActive() ? chalk.magenta('ai4ctf> ') : chalk.green('icoa> '));
+            rl.prompt();
+            return;
+        }
+        // If in AI chat mode, route to chat handler
+        if (isChatActive()) {
+            processing = true;
+            const result = await handleChatMessage(input);
+            processing = false;
+            if (result === 'exit') {
+                rl.setPrompt(chalk.green('icoa> '));
+            }
             rl.prompt();
             return;
         }
@@ -287,8 +300,6 @@ export async function startRepl(program, resumeMode) {
         process.exit = (() => {
             throw new Error(INTERCEPT);
         });
-        // Pause main readline so sub-REPLs (ai4ctf) can use stdin
-        rl.pause();
         try {
             await program.parseAsync(['node', 'icoa', ...args]);
         }
@@ -310,7 +321,10 @@ export async function startRepl(program, resumeMode) {
         finally {
             process.exit = realExit;
             processing = false;
-            rl.resume();
+        }
+        // Switch prompt if entering chat mode
+        if (isChatActive()) {
+            rl.setPrompt(chalk.magenta('ai4ctf> '));
         }
         console.log();
         rl.prompt();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "icoa-cli",
-  "version": "2.3.1",
+  "version": "2.3.3",
   "description": "ICOA CLI — The world's first CLI-native CTF competition terminal",
   "type": "module",
   "bin": {