icoa-cli 2.16.17 → 2.17.1

package/dist/commands/ai4ctf.js

@@ -1,9 +1,9 @@
 import chalk from 'chalk';
 import { createChatSession } from '../lib/gemini.js';
-import { isTokenCapReached, addTokenUsage, getTokenUsage } from '../lib/budget.js';
+import { addTokenUsage } from '../lib/budget.js';
 import { getConfig } from '../lib/config.js';
 import { logCommand } from '../lib/logger.js';
-import { printMarkdown, printError, printInfo } from '../lib/ui.js';
+import { printMarkdown, printError } from '../lib/ui.js';
 function getChallengeContext() {
     const config = getConfig();
     if (config.currentChallengeName && config.currentChallengeCategory) {
@@ -14,24 +14,60 @@ function getChallengeContext() {
 // Chat state — shared with REPL
 let chatActive = false;
 let chatSession = null;
+let chatTokensUsed = 0;
+const DEMO_TOKEN_CAP = 5000;
 export function isChatActive() {
     return chatActive;
 }
+function drawTokenBar() {
+    const cap = DEMO_TOKEN_CAP;
+    const used = chatTokensUsed;
+    const pct = Math.min(Math.round((used / cap) * 100), 100);
+    const width = 20;
+    const filled = Math.round((pct / 100) * width);
+    const empty = width - filled;
+    const color = pct > 80 ? chalk.red : pct > 50 ? chalk.yellow : chalk.green;
+    console.log(chalk.gray('  Tokens: ') + color('█'.repeat(filled)) + chalk.gray('░'.repeat(empty)) + chalk.gray(` ${used}/${cap} (${pct}%)`));
+}
 export async function handleChatMessage(input) {
     if (!chatSession)
         return 'exit';
     if (input === 'exit' || input === 'back' || input === 'quit') {
         chatActive = false;
         chatSession = null;
+        // Anonymous stats
+        fetch('https://practice.icoa2026.au:9090/api/icoa/demo-stats', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ type: 'ai4ctf', tokensUsed: chatTokensUsed, timestamp: new Date().toISOString() }),
+            signal: AbortSignal.timeout(5000),
+        }).catch(() => { });
+        console.log();
+        console.log(chalk.gray('  ─────────────────────────────────────────'));
+        console.log(chalk.white('  AI4CTF Session Report'));
+        console.log(chalk.gray(`  Tokens used: ${chatTokensUsed}/${DEMO_TOKEN_CAP}`));
+        console.log(chalk.gray(`  AI Model: Google Gemma 4 (gemma-4-31b-it)`));
+        console.log(chalk.gray('  ─────────────────────────────────────────'));
         console.log();
-        printInfo('Returning to ICOA terminal.');
+        console.log(chalk.white('  Next: try ') + chalk.bold.red('ctf4ai') + chalk.white(' — trick the AI into saying "koala"'));
         console.log();
         return 'exit';
     }
-    if (isTokenCapReached()) {
+    if (chatTokensUsed >= DEMO_TOKEN_CAP) {
         chatActive = false;
         chatSession = null;
-        printError('Token budget exhausted. Exiting chat mode.');
+        console.log();
+        console.log(chalk.yellow('  Token limit reached!'));
+        drawTokenBar();
+        console.log();
+        console.log(chalk.gray('  ─────────────────────────────────────────'));
+        console.log(chalk.white('  AI4CTF Session Report'));
+        console.log(chalk.gray(`  Tokens used: ${chatTokensUsed}/${DEMO_TOKEN_CAP}`));
+        console.log(chalk.gray(`  AI Model: Google Gemma 4 (gemma-4-31b-it)`));
+        console.log(chalk.gray('  ─────────────────────────────────────────'));
+        console.log();
+        console.log(chalk.white('  Next: try ') + chalk.bold.red('ctf4ai') + chalk.white(' — trick the AI into saying "koala"'));
+        console.log();
         return 'exit';
     }
     logCommand(`ai4ctf: ${input}`);
@@ -39,11 +75,11 @@ export async function handleChatMessage(input) {
     try {
         const response = await chatSession.sendMessage(input);
         process.stdout.write('\x1b[1A\x1b[2K');
+        chatTokensUsed += response.tokensUsed;
         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]`));
+        drawTokenBar();
         console.log();
     }
     catch (err) {
@@ -59,32 +95,55 @@ export function registerAi4ctfCommand(program) {
         .description('Chat with your AI teammate')
         .action(async () => {
         logCommand('ai4ctf');
-        if (isTokenCapReached()) {
-            printError('Token budget exhausted. No more AI interactions available.');
-            return;
-        }
-        const context = getChallengeContext();
-        const tokenState = getTokenUsage();
+        const config = getConfig();
+        const modelName = config.geminiModel || 'gemma-4-31b-it';
+        // Demo challenge context
+        const demoContext = {
+            name: 'Hidden Message',
+            category: 'Cryptography',
+        };
         try {
-            chatSession = await createChatSession(context);
+            chatSession = await createChatSession(demoContext);
         }
         catch (err) {
             printError(err.message);
             return;
         }
         chatActive = true;
-        // Welcome banner
+        chatTokensUsed = 0;
+        // Guided welcome
         console.log();
-        console.log(chalk.magenta('  ┌─────────────────────────────────────────┐'));
-        console.log(chalk.magenta('  │') + chalk.bold.white('  AI Teammate — Chat Mode') + chalk.magenta('                │'));
-        if (context) {
-            const ctxStr = `  Challenge: ${context.name} (${context.category})`;
-            console.log(chalk.magenta('  │') + chalk.gray(ctxStr.padEnd(41)) + chalk.magenta('│'));
-        }
-        const tokenStr = `  Tokens: ${tokenState.used.toLocaleString()}/${tokenState.cap.toLocaleString()}`;
-        console.log(chalk.magenta('  │') + chalk.gray(tokenStr.padEnd(41)) + chalk.magenta('│'));
-        console.log(chalk.magenta('  │') + chalk.gray("  Type 'exit' to return".padEnd(41)) + chalk.magenta('│'));
-        console.log(chalk.magenta('  └─────────────────────────────────────────┘'));
+        console.log(chalk.green.bold('  ═══ AI4CTF Demo — AI as Your Teammate ═══'));
+        console.log();
+        console.log(chalk.white('  Here\'s a sample CTF challenge:'));
+        console.log();
+        console.log(chalk.cyan('  ┌─────────────────────────────────────────────────┐'));
+        console.log(chalk.cyan('  │') + chalk.bold.white('  Challenge: Hidden Message [Cryptography]       ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  │') + chalk.white('                                                  ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  │') + chalk.white('  You intercepted this encoded text:               ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  │') + chalk.green('  aWNvYXt3M2xjMG1lXzJfYWk0Y3RmfQ==              ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  │') + chalk.white('                                                  ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  │') + chalk.white('  Can you decode it to find the flag?              ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  │') + chalk.gray('  Flag format: icoa{...}                           ') + chalk.cyan('│'));
+        console.log(chalk.cyan('  └─────────────────────────────────────────────────┘'));
+        console.log();
+        console.log(chalk.white('  In competition, you get AI help at 3 levels:'));
+        console.log();
+        console.log(chalk.yellow('    hint a') + chalk.gray('    General guidance — "What type of encoding is this?"'));
+        console.log(chalk.gray('              50 uses per competition. Safe to use freely.'));
+        console.log(chalk.yellow('    hint b') + chalk.gray('    Deep analysis — "How do I decode Base64?"'));
+        console.log(chalk.gray('              10 uses. Use when you\'re stuck.'));
+        console.log(chalk.yellow('    hint c') + chalk.gray('    Critical assist — Nearly gives you the answer'));
+        console.log(chalk.gray('              2 uses only. Last resort!'));
+        console.log();
+        console.log(chalk.gray('  ─────────────────────────────────────────'));
+        console.log(chalk.white('  Try now: ask the AI anything about the challenge above.'));
+        console.log(chalk.gray('  Example: "What encoding is aWNvYXt3M2xjMG1lXzJfYWk0Y3RmfQ==?"'));
+        console.log(chalk.gray('  Or just chat freely! You can also try hint a, hint b, hint c.'));
+        console.log();
+        drawTokenBar();
+        console.log(chalk.gray(`  AI Model: Google Gemma 4 (${modelName})`));
+        console.log(chalk.gray('  Type "exit" when done.'));
         console.log();
     });
 }

package/dist/commands/ctf4ai-demo.js

@@ -27,6 +27,12 @@ export async function handleCtf4aiMessage(input) {
     if (input === 'exit' || input === 'back' || input === 'quit') {
         ctf4aiActive = false;
         ctf4aiSession = null;
+        fetch('https://practice.icoa2026.au:9090/api/icoa/demo-stats', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ type: 'ctf4ai', solved: false, tokensUsed: ctf4aiTokens, timestamp: new Date().toISOString() }),
+            signal: AbortSignal.timeout(5000),
+        }).catch(() => { });
         console.log();
         console.log(chalk.gray('  CTF4AI challenge ended.'));
         console.log(chalk.gray(`  Tokens used: ${ctf4aiTokens}/${CTF4AI_TOKEN_LIMIT}`));
@@ -67,6 +73,12 @@ export async function handleCtf4aiMessage(input) {
             console.log();
             ctf4aiActive = false;
             ctf4aiSession = null;
+            fetch('https://practice.icoa2026.au:9090/api/icoa/demo-stats', {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({ type: 'ctf4ai', solved: true, tokensUsed: ctf4aiTokens, timestamp: new Date().toISOString() }),
+                signal: AbortSignal.timeout(5000),
+            }).catch(() => { });
             return 'solved';
         }
         return 'continue';

package/dist/commands/exam.js

@@ -581,6 +581,22 @@ export function registerExamCommand(program) {
                     }
                     drawProgress(100, 'Complete!');
                     console.log();
+                    // Anonymous stats (fire-and-forget)
+                    const helpInfo = getHelpState(state);
+                    const statsLang = getConfig().language || 'en';
+                    fetch('https://practice.icoa2026.au:9090/api/icoa/demo-stats', {
+                        method: 'POST',
+                        headers: { 'Content-Type': 'application/json' },
+                        body: JSON.stringify({
+                            score,
+                            total: state.questions.length,
+                            lang: statsLang,
+                            helpUsed: helpInfo.used,
+                            helpMax: helpInfo.max,
+                            timestamp: new Date().toISOString(),
+                        }),
+                        signal: AbortSignal.timeout(5000),
+                    }).catch(() => { });
                     clearExamState();
                     const pct = Math.round(score / state.questions.length * 100);
                     console.log();

package/package.json

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