chels-code 0.0.0-stage → 0.5.0

package/dist/test/ui.test.js

@@ -0,0 +1,370 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import assert from "node:assert/strict";
+import { mkdtemp, readFile, rm } from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import test from "node:test";
+import { render } from "ink-testing-library";
+import { ActionConfirmation } from "../src/ui/ConfirmPanel.js";
+import { App } from "../src/ui/App.js";
+import { TextArea } from "../src/ui/TextArea.js";
+import { sanitize } from "../src/ui/theme.js";
+const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+function output(app) {
+    return app.frames.map((frame) => sanitize(frame)).join("\n");
+}
+async function waitFor(app, predicate, timeout = 4_000) {
+    const startedAt = Date.now();
+    while (Date.now() - startedAt < timeout) {
+        if (predicate(output(app)))
+            return;
+        await delay(20);
+    }
+    throw new Error(`Timed out waiting for UI output.\n${output(app).slice(-4_000)}`);
+}
+async function waitForFrame(app, predicate, timeout = 4_000) {
+    const startedAt = Date.now();
+    while (Date.now() - startedAt < timeout) {
+        if (predicate(sanitize(app.lastFrame() ?? "")))
+            return;
+        await delay(20);
+    }
+    throw new Error(`Timed out waiting for the current frame.\n${sanitize(app.lastFrame() ?? "")}`);
+}
+async function waitForReady(app, timeout = 4_000) {
+    const startedAt = Date.now();
+    while (Date.now() - startedAt < timeout) {
+        const frame = sanitize(app.lastFrame() ?? "");
+        if (frame.includes("Setup failed"))
+            return;
+        if (frame.includes("Ask anything") && !frame.includes("Thinking"))
+            return;
+        await delay(20);
+    }
+    throw new Error(`Timed out waiting for the input to become ready.\n${output(app).slice(-4_000)}`);
+}
+test("text area supports multiline input, cursor movement, and enter to send", async () => {
+    let submitted = null;
+    const app = render(_jsx(TextArea, { onSubmit: (text) => { submitted = text; }, columns: 60, rows: 24 }));
+    try {
+        await delay(20);
+        app.stdin.write("hello");
+        await delay(20);
+        app.stdin.write("\u001b[13;2u");
+        await delay(20);
+        app.stdin.write("world");
+        await delay(20);
+        app.stdin.write("\u001b[A");
+        await delay(20);
+        app.stdin.write("!");
+        await delay(20);
+        app.stdin.write("\r");
+        await delay(30);
+        assert.equal(submitted, "hello!\nworld");
+        assert.match(sanitize(app.lastFrame() ?? ""), /Ask anything/);
+    }
+    finally {
+        app.unmount();
+    }
+});
+test("command confirmation resolves approval with y", async () => {
+    let resolved = null;
+    const app = render(_jsx(ActionConfirmation, { action: { kind: "command", command: "npm test" }, onResolve: (approved) => { resolved = approved; } }));
+    try {
+        await delay(20);
+        assert.match(sanitize(app.lastFrame() ?? ""), /Run this PowerShell command/);
+        app.stdin.write("y");
+        await delay(20);
+        assert.equal(resolved, true);
+    }
+    finally {
+        app.unmount();
+    }
+});
+test("runs a mocked minimal session with plan approval and write approval", async (t) => {
+    const originalFetch = globalThis.fetch;
+    const workspace = await mkdtemp(path.join(os.tmpdir(), "chels-ui-"));
+    let writeIssued = false;
+    const jsonResponse = (payload) => new Response(JSON.stringify(payload), {
+        status: 200,
+        headers: { "Content-Type": "application/json" },
+    });
+    globalThis.fetch = (async (input, init) => {
+        const url = new URL(String(input));
+        const host = url.hostname;
+        if (url.pathname.endsWith("/models")) {
+            const model = host === "openrouter.ai"
+                ? {
+                    id: "qwen/qwen3-coder:free",
+                    pricing: { prompt: 0, completion: 0 },
+                    supported_parameters: ["tools"],
+                    context_length: 128_000,
+                }
+                : host === "integrate.api.nvidia.com"
+                    ? {
+                        id: "meta/llama-3.3-70b-instruct",
+                        supported_parameters: ["tools"],
+                        context_length: 128_000,
+                    }
+                    : {
+                        id: "llama-3.3-70b-versatile",
+                        supported_parameters: ["tools"],
+                        context_length: 128_000,
+                    };
+            return jsonResponse({ data: [model] });
+        }
+        if (url.pathname.endsWith("/key")) {
+            return jsonResponse({ data: { limit: 5, limit_remaining: 5, usage: 0 } });
+        }
+        const body = JSON.parse(String(init?.body));
+        const system = body.messages.find((message) => message.role === "system")?.content ?? "";
+        await delay(60);
+        if (system.includes("Combine three AI work breakdowns")) {
+            return jsonResponse({
+                choices: [{
+                        message: {
+                            role: "assistant",
+                            content: JSON.stringify({
+                                tasks: [{
+                                        title: "Write greeting",
+                                        instructions: "Create hello.txt with a short greeting.",
+                                        provider: "openrouter",
+                                    }],
+                            }),
+                        },
+                    }],
+            });
+        }
+        if (system.includes("final reporting agent")) {
+            return jsonResponse({
+                choices: [{
+                        message: {
+                            role: "assistant",
+                            content: "Final: created hello.txt.",
+                        },
+                    }],
+            });
+        }
+        if (system.includes("working on an approved subtask")) {
+            if (!writeIssued) {
+                writeIssued = true;
+                return jsonResponse({
+                    choices: [{
+                            message: {
+                                role: "assistant",
+                                content: null,
+                                tool_calls: [{
+                                        id: "call_write",
+                                        type: "function",
+                                        function: {
+                                            name: "write_file",
+                                            arguments: JSON.stringify({
+                                                path: "hello.txt",
+                                                content: "hello from ChelsCode\n",
+                                            }),
+                                        },
+                                    }],
+                            },
+                        }],
+                });
+            }
+            return jsonResponse({
+                choices: [{
+                        message: {
+                            role: "assistant",
+                            content: "All done. hello.txt was written.",
+                        },
+                    }],
+            });
+        }
+        return jsonResponse({
+            choices: [{
+                    message: {
+                        role: "assistant",
+                        content: `Proposal from ${host}.`,
+                    },
+                }],
+        });
+    });
+    const app = render(_jsx(App, { apiKeys: { openrouter: "test", nvidia: "test", groq: "test" }, workspace: workspace, version: "0.5.0" }));
+    t.after(async () => {
+        globalThis.fetch = originalFetch;
+        app.unmount();
+        await rm(workspace, { recursive: true, force: true });
+    });
+    await waitForReady(app);
+    const initial = output(app);
+    assert.ok(initial.includes("I like cats"), "the remaining tagline should be visible");
+    assert.ok(initial.includes("██████╗"), "the start screen should show the big ChelsCode wordmark");
+    assert.ok(initial.includes("Build · MystBloom V1.0  Free ChelsCode · max"), "the chat should show the build, model, plan, and effort status line");
+    assert.ok(initial.includes("MystBloom V1.0"), "the assistant name should be visible");
+    assert.match(initial, /0 \(0%\)  shift\+tab mode  ctrl\+p commands/, "the footer should start at zero usage");
+    assert.ok(!initial.includes("enter send"), "the old hint row should be gone");
+    assert.ok(!initial.includes("Three AIs. One coding team."), "the longer tagline should be gone");
+    assert.ok(!initial.includes("Connected · 3 model providers ready."), "the connected notice should be gone");
+    assert.doesNotMatch(initial, /●\s*OpenRouter\s+●\s*NVIDIA/, "the provider chip row should be gone");
+    assert.match(initial, /Ask anything/);
+    app.stdin.write("Create a greeting file");
+    await delay(30);
+    app.stdin.write("\r");
+    await waitFor(app, (text) => text.includes("Proceed with this plan?"));
+    assert.match(output(app), /MystBloom V1\.0/, "the plan should show the assistant name");
+    await delay(30);
+    app.stdin.write("y");
+    await waitFor(app, (text) => text.includes("Write this file?"));
+    await delay(30);
+    app.stdin.write("y");
+    await waitFor(app, (text) => text.includes("◆ Done")
+        && text.includes("Final: created hello.txt.")
+        && text.includes("Made"));
+    const finalOutput = output(app);
+    assert.match(finalOutput, /> Create a greeting file/, "the user message should use the > prefix");
+    assert.match(finalOutput, /[\u2800-\u28FF] Thinking/, "the live status should animate while working");
+    assert.match(finalOutput, /◆ Done/, "the response should finish with Done");
+    assert.match(finalOutput, /\+\s*hello\.txt/, "Done should list the file that was made");
+    assert.ok(finalOutput.includes("Final: created hello.txt."), "one AI should write the final answer");
+    assert.ok(!finalOutput.includes("All done. hello.txt was written."), "intermediate subtask chatter should stay hidden");
+    assert.ok(!finalOutput.includes("Subtask"), "subtask headers should stay hidden");
+    assert.ok(!finalOutput.includes("◆ You"), "the old You heading should be gone");
+    assert.doesNotMatch(finalOutput, /●\s*OpenRouter\s+·/, "provider/model answer headers should be gone");
+    assert.doesNotMatch(finalOutput, /●\s*NVIDIA/, "provider attribution should stay hidden");
+    assert.equal(await readFile(path.join(workspace, "hello.txt"), "utf8"), "hello from ChelsCode\n");
+});
+test("answers a simple question with one AI instead of the three-AI plan", async (t) => {
+    const originalFetch = globalThis.fetch;
+    const workspace = await mkdtemp(path.join(os.tmpdir(), "chels-question-"));
+    let completionCalls = 0;
+    let visibleTools = [];
+    const jsonResponse = (payload) => new Response(JSON.stringify(payload), {
+        status: 200,
+        headers: { "Content-Type": "application/json" },
+    });
+    globalThis.fetch = (async (input, init) => {
+        const url = new URL(String(input));
+        if (url.pathname.endsWith("/models")) {
+            const model = url.hostname === "openrouter.ai"
+                ? {
+                    id: "qwen/qwen3-coder:free",
+                    pricing: { prompt: 0, completion: 0 },
+                    supported_parameters: ["tools"],
+                    context_length: 128_000,
+                }
+                : url.hostname === "integrate.api.nvidia.com"
+                    ? {
+                        id: "meta/llama-3.3-70b-instruct",
+                        supported_parameters: ["tools"],
+                        context_length: 128_000,
+                    }
+                    : {
+                        id: "llama-3.3-70b-versatile",
+                        supported_parameters: ["tools"],
+                        context_length: 128_000,
+                    };
+            return jsonResponse({ data: [model] });
+        }
+        if (url.pathname.endsWith("/key")) {
+            return jsonResponse({ data: { limit: 5, limit_remaining: 5, usage: 0 } });
+        }
+        completionCalls += 1;
+        const body = JSON.parse(String(init?.body));
+        visibleTools = body.tools?.map((tool) => tool.function.name) ?? [];
+        await delay(60);
+        return jsonResponse({
+            choices: [{
+                    message: {
+                        role: "assistant",
+                        content: "Hi! Ask me anything about the workspace.",
+                    },
+                }],
+            usage: { prompt_tokens: 1_000, completion_tokens: 200, total_tokens: 1_200, cost: 0.5 },
+        });
+    });
+    const app = render(_jsx(App, { apiKeys: { openrouter: "test", nvidia: "test", groq: "test" }, workspace: workspace, version: "0.5.0" }));
+    t.after(async () => {
+        globalThis.fetch = originalFetch;
+        app.unmount();
+        await rm(workspace, { recursive: true, force: true });
+    });
+    await waitForReady(app);
+    const initial = output(app);
+    assert.ok(initial.includes("Build · MystBloom V1.0  Free ChelsCode · max"), "the chat should show the build, model, plan, and effort status line");
+    assert.ok(initial.includes("MystBloom V1.0"), "the assistant name should be visible");
+    assert.match(initial, /0 \(0%\)  shift\+tab mode  ctrl\+p commands/, "the footer should start at zero usage");
+    assert.ok(!initial.includes("enter send"), "the old hint row should be gone");
+    app.stdin.write("hi");
+    await delay(30);
+    app.stdin.write("\r");
+    await waitFor(app, (text) => text.includes("Hi! Ask me anything about the workspace."));
+    await waitFor(app, (text) => text.includes("1.2K (10%)"));
+    const text = output(app);
+    assert.match(text, /1\.2K \(10%\)/, "the footer should show session tokens and credit usage");
+    assert.match(text, /1\.2K \(10%\)  shift\+tab mode  ctrl\+p commands/, "the token footer should show both shortcuts and not include the model name");
+    assert.match(text, /> hi/, "the user message should use the > prefix");
+    assert.match(text, /[\u2800-\u28FF] Thinking/, "the live status should animate while working");
+    assert.match(text, /◆ Done/, "the answer should finish with Done");
+    assert.ok(text.includes("Hi! Ask me anything about the workspace."), "the one AI answer should still be shown");
+    assert.ok(!text.includes("Made"), "a question should not claim anything was made");
+    assert.ok(!text.includes("● OpenRouter"), "provider/model answer headers should be gone");
+    assert.ok(!text.includes("Analysis"), "a question should not trigger the three-AI analysis");
+    assert.ok(!text.includes("Proceed with this plan?"), "a question should not ask for plan approval");
+    assert.ok(!text.includes("perspectives received"), "a question should not show three-way progress");
+    assert.equal(completionCalls, 1, "only one AI should answer a question");
+    assert.deepEqual(visibleTools, ["list_files", "read_file", "get_current_datetime", "get_weather", "web_search", "get_news"], "a question should get read-only tools plus current-info lookups");
+    app.stdin.write("\u0010");
+    await waitForFrame(app, (text) => text.includes("Commands"));
+    const menu = sanitize(app.lastFrame() ?? "");
+    assert.match(menu, /Mode/);
+    assert.match(menu, /Effort/);
+    assert.match(menu, /Exit/);
+    assert.match(menu, /Help/);
+    await delay(30);
+    app.stdin.write("\u001b[B");
+    await delay(30);
+    app.stdin.write("\u001b[B");
+    await delay(30);
+    app.stdin.write("\r");
+    await waitFor(app, (text) => text.includes("enter send · shift+enter newline"));
+    app.stdin.write("\u0010");
+    await waitForFrame(app, (text) => text.includes("Commands"));
+    await delay(30);
+    app.stdin.write("\u001b[B");
+    await delay(30);
+    app.stdin.write("\r");
+    await waitForFrame(app, (text) => text.includes("enter apply"));
+    const effortMenu = sanitize(app.lastFrame() ?? "");
+    assert.match(effortMenu, /low/);
+    assert.match(effortMenu, /medium/);
+    assert.match(effortMenu, /high/);
+    assert.match(effortMenu, /max/);
+    assert.match(effortMenu, /ultra/);
+    assert.match(effortMenu, /max  current/, "max should be the default effort");
+    await delay(30);
+    app.stdin.write("\u001b[B");
+    await delay(30);
+    app.stdin.write("\r");
+    await waitFor(app, (text) => text.includes("Build · MystBloom V1.0  Free ChelsCode · ultra"));
+    app.stdin.write("\u0010");
+    await waitForFrame(app, (text) => text.includes("Commands"));
+    await delay(30);
+    app.stdin.write("\r");
+    await waitForFrame(app, (text) => text.includes("enter switch"));
+    const modeMenu = sanitize(app.lastFrame() ?? "");
+    assert.match(modeMenu, /Build/);
+    assert.match(modeMenu, /Plan/);
+    assert.match(modeMenu, /Build  current/, "Build should be the default mode");
+    await delay(30);
+    app.stdin.write("\u001b[B");
+    await delay(30);
+    app.stdin.write("\r");
+    await waitFor(app, (text) => text.includes("Plan · MystBloom V1.0  Free ChelsCode · ultra"));
+    assert.ok(sanitize(app.lastFrame() ?? "").includes("Describe what you want to plan…"), "the mode menu should update the input prompt");
+    app.stdin.write("\u001b[Z");
+    await waitForFrame(app, (text) => text.includes("Build · MystBloom V1.0  Free ChelsCode · ultra"));
+    assert.ok(sanitize(app.lastFrame() ?? "").includes("Describe what you want to build…"), "Shift+Tab should switch back to Build without inserting spaces");
+    app.stdin.write("\u0014");
+    await waitForFrame(app, (text) => text.includes("Plan · MystBloom V1.0  Free ChelsCode · ultra"));
+    assert.ok(sanitize(app.lastFrame() ?? "").includes("Describe what you want to plan…"), "Ctrl+T should switch modes when the terminal intercepts Shift+Tab");
+    app.stdin.write("\t");
+    await waitForFrame(app, (text) => text.includes("Build · MystBloom V1.0  Free ChelsCode · ultra"));
+    assert.ok(sanitize(app.lastFrame() ?? "").includes("Describe what you want to build…"), "a plain Tab should toggle the mode instead of inserting whitespace");
+});

package/package.json

@@ -1,6 +1,32 @@
 {
   "name": "chels-code",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "0.5.0",
+  "description": "A TypeScript + Ink terminal coding assistant for OpenRouter, NVIDIA, and Groq",
+  "type": "module",
+  "bin": {
+    "chelscode": "dist/src/cli.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "dev": "tsx src/cli.tsx",
+    "start": "npm run build && node dist/src/cli.js",
+    "test": "node --import tsx --test \"test/*.test.ts\" \"test/*.test.tsx\""
+  },
+  "engines": {
+    "node": ">=22"
+  },
+  "dependencies": {
+    "ink": "^8.0.0",
+    "react": "^19.3.1"
+  },
+  "devDependencies": {
+    "@types/node": "^26.6.6",
+    "@types/react": "^19.3.0",
+    "ink-testing-library": "^4.0.0",
+    "tsx": "^4.23.15",
+    "typescript": "^7.0.2"
+  }
+}