codeshark-cli 0.1.1

package/dist/repl.js

@@ -0,0 +1,202 @@
+import { createInterface } from "node:readline/promises";
+import { stdin as input, stdout as output } from "node:process";
+import { bold, dim, hex } from "./ansi.js";
+import { runAgent } from "./agent.js";
+import { ProviderError, errorMessage } from "./provider/types.js";
+import { runSetup } from "./setup.js";
+import { loadConfig, modelLabel, saveConfig } from "./config.js";
+import { DEFAULT_MODEL_ID, MODELS, findModel } from "./models.js";
+import { launchKeysPage } from "./keysPage.js";
+const PROMPT = "🦈 ";
+function replClosed(rl) {
+    return Boolean(rl.closed);
+}
+function briefArgs(args) {
+    const s = JSON.stringify(args);
+    return s.length > 90 ? s.slice(0, 90) + "…" : s;
+}
+function printHelp() {
+    console.log([
+        "",
+        bold("  CodeShark commands"),
+        dim("    /help            show this help"),
+        dim("    /model           show the active model and the full catalog"),
+        dim("    /model <name>    switch model, e.g. /model glm or /model kimi-k3"),
+        dim("    /keys            open the password-protected local API key page"),
+        dim("    /key-status      show masked key status in the terminal"),
+        dim("    /setup           guided setup wizard (providers / keys)"),
+        dim("    /clear           clear the screen"),
+        dim("    /quit            exit (Ctrl+C also works)"),
+        "",
+    ].join("\n"));
+}
+export function printModelInfo() {
+    const cfg = loadConfig();
+    console.log("");
+    console.log(`  Active: ${bold(hex("#4ade80", modelLabel(cfg)))}`);
+    console.log("");
+    console.log(bold("  Model catalog"));
+    for (const m of MODELS) {
+        const active = m.id === (findModel(cfg.model ?? "")?.id ?? DEFAULT_MODEL_ID);
+        const marker = active ? hex("#4ade80", "●") : dim("○");
+        console.log(`    ${marker} ${m.label.padEnd(28)} ${dim(m.context.padEnd(5))} ${m.notes}`);
+    }
+    console.log("");
+    console.log(dim("  Switch with: /model <name>   e.g. /model glm, /model kimi, /model gpt"));
+    console.log("");
+}
+export function switchModel(query) {
+    const q = query.trim();
+    if (!q) {
+        printModelInfo();
+        return;
+    }
+    const cfg = loadConfig();
+    const entry = findModel(q) ??
+        MODELS.find((m) => m.id.toLowerCase().includes(q.toLowerCase()) || m.label.toLowerCase().includes(q.toLowerCase()));
+    if (!entry) {
+        console.log(hex("#f87171", `  ✗ No model matches "${q}".`));
+        console.log(dim("  Available: " + MODELS.map((m) => m.label).join(", ")));
+        return;
+    }
+    cfg.model = entry.id;
+    saveConfig(cfg);
+    console.log(`  ${hex("#4ade80", "✓")} Switched to ${bold(entry.label)} ${dim(`(${entry.context} context)`)}`);
+    console.log(dim("  Applies immediately — just start chatting."));
+    console.log("");
+}
+function printKeys() {
+    const cfg = loadConfig();
+    console.log("");
+    console.log(bold("  API keys"));
+    const ur = Boolean(cfg.unorouterApiKey ?? process.env.UNOROUTER_API_KEY);
+    const or = Boolean(cfg.openrouterApiKey ?? process.env.OPENROUTER_API_KEY);
+    const nv = Boolean(cfg.nvidiaApiKey ?? process.env.NVIDIA_API_KEY);
+    const gm = Boolean(cfg.geminiApiKey ?? process.env.GEMINI_API_KEY);
+    console.log(`    ${ur ? hex("#4ade80", "✓") : dim("○")} UnoRouter    ${ur ? dim("(configured)") : dim("(not set)")}  → https://unorouter.com/en/tokens ${dim("key: shown once")}`);
+    console.log(`    ${or ? hex("#4ade80", "✓") : dim("○")} OpenRouter   ${or ? dim("(configured)") : dim("(not set)")}  → https://openrouter.ai/keys  ${dim("key: sk-or-v1-…")}`);
+    console.log(`    ${nv ? hex("#4ade80", "✓") : dim("○")} NVIDIA NIM   ${nv ? dim("(configured)") : dim("(not set)")}  → https://build.nvidia.com   ${dim("key: nvapi-…")}`);
+    console.log(`    ${gm ? hex("#4ade80", "✓") : dim("○")} Google AI Studio ${gm ? dim("(configured)") : dim("(not set)")}  → https://aistudio.google.com/apikey ${dim("key: AIza…")}`);
+    console.log("");
+    console.log(dim("  Add one: run /setup, or paste it into ~/.codeshark.json like:"));
+    console.log(dim('    { "unorouterApiKey": "ur-…", "openrouterApiKey": "sk-or-v1-…" }'));
+    console.log(dim("  Or set env vars: UNOROUTER_API_KEY, OPENROUTER_API_KEY, NVIDIA_API_KEY, GEMINI_API_KEY"));
+    console.log("");
+}
+export async function startRepl(opts) {
+    const rl = createInterface({ input, output });
+    rl.setPrompt(PROMPT);
+    let history = [];
+    console.log(dim("  Type a message and press Enter. /help for commands · Ctrl+C to quit."));
+    rl.prompt();
+    rl.on("SIGINT", () => {
+        console.log("");
+        console.log(dim("Bye! 🦈"));
+        process.exit(0);
+    });
+    rl.on("line", async (raw) => {
+        const line = raw.trim();
+        if (!line) {
+            if (!replClosed(rl))
+                rl.prompt();
+            return;
+        }
+        // Commands
+        if (line.startsWith("/")) {
+            const [cmd, ...rest] = line.slice(1).split(/\s+/);
+            switch (cmd) {
+                case "help":
+                case "h":
+                    printHelp();
+                    break;
+                case "model":
+                case "m":
+                    if (rest.length)
+                        switchModel(rest.join(" "));
+                    else
+                        printModelInfo();
+                    break;
+                case "keys":
+                case "key":
+                    console.log(dim(`  Opening local key vault… ${await launchKeysPage()}`));
+                    console.log(dim("  The browser page asks for a password before showing keys."));
+                    break;
+                case "key-status":
+                case "keystatus":
+                    printKeys();
+                    break;
+                case "clear":
+                    history = [];
+                    process.stdout.write("\u001b[2J\u001b[H");
+                    break;
+                case "setup":
+                    rl.pause();
+                    await runSetup();
+                    rl.resume();
+                    console.log(dim("  Provider chain reloaded."));
+                    break;
+                case "quit":
+                case "exit":
+                case "q":
+                    console.log(dim("Bye! 🦈"));
+                    process.exit(0);
+                    break;
+                default:
+                    console.log(dim(`  Unknown command "${cmd}". Try /help.`));
+            }
+            if (!replClosed(rl))
+                rl.prompt();
+            return;
+        }
+        // Agent run
+        rl.pause();
+        console.log("");
+        const events = {
+            onText: (delta) => process.stdout.write(delta),
+            onToolCall: (call) => {
+                console.log(dim(`  ⚙ ${call.name}(${briefArgs(call.args)})`));
+            },
+            onDebug: (msg) => console.log(dim(msg)),
+        };
+        try {
+            const result = await runAgent(line, {
+                clients: opts.getClients(),
+                registry: opts.registry,
+                cwd: opts.cwd,
+                debug: opts.debug,
+                initialMessages: history,
+            }, events);
+            history = result.history;
+            if (result.streamedText) {
+                if (!result.streamedText.endsWith("\n"))
+                    process.stdout.write("\n");
+            }
+            else if (result.text) {
+                process.stdout.write(result.text.endsWith("\n") ? result.text : result.text + "\n");
+            }
+        }
+        catch (e) {
+            console.log("");
+            if (e instanceof ProviderError) {
+                console.log(hex("#f87171", `  ✗ ${e.message}`));
+                if (e.kind === "network") {
+                    console.log(dim("  The provider is unreachable. Run /keys to configure your own key."));
+                }
+                else if (e.kind === "auth" || e.kind === "rate_limit") {
+                    console.log(dim("  Run /setup to fix this, or /model to pick a different model."));
+                }
+            }
+            else {
+                console.log(hex("#f87171", `  ✗ ${errorMessage(e)}`));
+            }
+        }
+        console.log("");
+        if (!replClosed(rl)) {
+            rl.resume();
+            rl.prompt();
+        }
+    });
+    await new Promise(() => {
+        // Keep the interface alive; SIGINT handles exit.
+    });
+}

package/dist/setup.js

@@ -0,0 +1,178 @@
+import { createInterface } from "node:readline/promises";
+import { stdin as input, stdout as output } from "node:process";
+import { exec } from "node:child_process";
+import { bold, dim, hex } from "./ansi.js";
+import { configPath, hasApiKey, loadConfig, saveConfig } from "./config.js";
+import { createUnoRouterClient } from "./provider/unorouter.js";
+import { createGeminiClient } from "./provider/gemini.js";
+import { ollamaAvailable } from "./provider/ollama.js";
+import { MODELS } from "./models.js";
+import { errorMessage } from "./provider/types.js";
+function openBrowser(url) {
+    try {
+        const cmd = process.platform === "win32"
+            ? `cmd /c start "" "${url}"`
+            : process.platform === "darwin"
+                ? `open "${url}"`
+                : `xdg-open "${url}"`;
+        exec(cmd, () => { });
+    }
+    catch {
+        // ignore — we print the URL anyway
+    }
+}
+/**
+ * Ask for a secret WITHOUT echoing it to the terminal, so pasting an API
+ * key never leaves it visible on screen (or in terminal scrollback).
+ */
+function secretQuestion(rl) {
+    return async (prompt) => {
+        const anyRl = rl;
+        const original = anyRl._writeToOutput;
+        anyRl._writeToOutput = () => { }; // swallow the echo of typed characters
+        process.stdout.write(prompt);
+        const answer = await rl.question("");
+        anyRl._writeToOutput = original;
+        process.stdout.write("\n");
+        return answer.trim();
+    };
+}
+async function testClient(label, client) {
+    process.stdout.write(`  Testing ${label}… `);
+    try {
+        await client.chat([{ role: "user", content: "Reply with exactly: OK" }], []);
+        console.log(hex("#4ade80", "✓ works"));
+        return true;
+    }
+    catch (e) {
+        console.log(hex("#f87171", `✗ ${errorMessage(e)}`));
+        return false;
+    }
+}
+/** After keys are saved, offer a numbered model picker and save the choice. */
+async function pickModel(rl, cfg) {
+    const available = MODELS.filter((m) => hasApiKey(cfg, m.provider));
+    if (!available.length)
+        return;
+    console.log("");
+    console.log(bold("  Pick your default model:"));
+    available.forEach((m, i) => {
+        console.log(`    [${i + 1}] ${m.label.padEnd(30)} ${dim(m.context.padEnd(5))}`);
+        console.log(`        ${dim(m.notes)}`);
+    });
+    console.log("");
+    const answer = await rl.question(`  Choose [1-${available.length}] (default 1): `);
+    const idx = (parseInt(answer.trim(), 10) || 1) - 1;
+    const chosen = available[Math.max(0, Math.min(idx, available.length - 1))] ?? available[0];
+    if (!chosen)
+        return;
+    cfg.model = chosen.id;
+    console.log(`  ${hex("#4ade80", "✓")} Default model: ${bold(chosen.label)}`);
+}
+function saveDone(cfg) {
+    saveConfig(cfg);
+    console.log("");
+    console.log(hex("#4ade80", `  ✓ Saved to ${configPath()}`));
+    console.log("");
+}
+/** The `codeshark setup` command: guided provider / key configuration. */
+export async function runSetup() {
+    const rl = createInterface({ input, output });
+    try {
+        await runSetupFlow(loadConfig(), rl, { title: "🦈  CodeShark setup" });
+    }
+    finally {
+        rl.close();
+    }
+}
+/**
+ * The provider/key configuration flow, shared by `codeshark setup` and the
+ * first-run prompt at CLI startup. Returns true when a choice was saved.
+ */
+export async function runSetupFlow(cfg, rl, opts = {}) {
+    const askSecret = secretQuestion(rl);
+    console.log("");
+    console.log(bold(hex("#d96b43", opts.title ?? "🦈  CodeShark setup")));
+    console.log("");
+    console.log(bold("  How do you want to talk to the models?"));
+    console.log(dim("    [1] Shared CodeShark gateway — zero setup. No key on this computer;"));
+    console.log(dim("        the gateway's keys stay private on the server."));
+    console.log(dim("    [2] My own API key (recommended) — your own rate limits and privacy;"));
+    console.log(dim("        typed hidden, stored only on this computer, never shown on screen."));
+    console.log(dim("    [3] Local Ollama — fully offline, runs on your machine."));
+    console.log(dim("    [4] Cancel — change nothing"));
+    console.log("");
+    const choice = (await rl.question("  Choose [1-4]: ")).trim();
+    const is = (v, ...words) => choice === v || words.some((w) => choice.toLowerCase() === w);
+    if (is("4", "cancel", "c", "skip", "s")) {
+        console.log(dim("  Skipped — no changes saved."));
+        return false;
+    }
+    if (is("3", "ollama", "o")) {
+        cfg.provider = "ollama";
+        if (await ollamaAvailable(cfg)) {
+            console.log(hex("#4ade80", "  ✓ Ollama reachable"));
+        }
+        else {
+            console.log(hex("#f87171", "  ✗ Ollama not reachable — install from https://ollama.com and run `ollama serve`."));
+        }
+        saveDone(cfg);
+        return true;
+    }
+    if (is("1", "gateway", "g")) {
+        cfg.provider = "gateway";
+        saveDone(cfg);
+        console.log(dim("  Zero setup — you're ready. Switch anytime with /setup."));
+        console.log("");
+        return true;
+    }
+    // Choice 2 — the user's own key (typed into the terminal, never echoed).
+    console.log(bold("  Step 1 — UnoRouter key (unlocks all 5 models)"));
+    console.log(dim("    1. Open https://unorouter.com/en/tokens  (sign up with Discord/GitHub, no card)"));
+    console.log(dim("    2. Create an API key on the Tokens page — it is shown exactly once — and copy it"));
+    console.log(dim("    Your key is typed hidden: it will not appear on screen."));
+    openBrowser("https://unorouter.com/en/tokens");
+    const urKey = await askSecret("  Paste your UnoRouter key (Enter to skip): ");
+    if (urKey) {
+        cfg.unorouterApiKey = urKey;
+        cfg.provider = "unorouter";
+        await testClient("UnoRouter (GLM 5.3 Flash Thinking)", createUnoRouterClient(cfg, urKey));
+        await pickModel(rl, cfg);
+        saveDone(cfg);
+        return true;
+    }
+    console.log(bold("  Step 2 — Google Gemini key (optional)"));
+    console.log(dim("    Get a key at https://aistudio.google.com/apikey (AIza…)."));
+    const gk = await askSecret("  Paste your Gemini key (Enter to skip): ");
+    if (gk) {
+        cfg.geminiApiKey = gk;
+        cfg.provider = "gemini";
+        await testClient("Gemini", createGeminiClient(cfg, gk));
+        saveDone(cfg);
+        return true;
+    }
+    // No key pasted: loop back to the provider choice.
+    console.log(dim("  No keys pasted."));
+    console.log("");
+    console.log(dim("    [1] Use shared CodeShark gateway (zero setup)   [2] Use local Ollama   [3] Cancel"));
+    const fallback = await rl.question("  Choose: ");
+    if (fallback.trim() === "2") {
+        cfg.provider = "ollama";
+        if (await ollamaAvailable(cfg)) {
+            console.log(hex("#4ade80", "  ✓ Ollama reachable"));
+        }
+        else {
+            console.log(hex("#f87171", "  ✗ Ollama not reachable — install from https://ollama.com and run `ollama serve`."));
+        }
+    }
+    else if (fallback.trim() === "3") {
+        console.log(dim("  Cancelled — nothing changed."));
+        return false;
+    }
+    else {
+        cfg.provider = "gateway";
+        console.log(dim("  Using the shared gateway — zero setup. You can add your own key later with /setup."));
+    }
+    saveDone(cfg);
+    return true;
+}

package/dist/system.js

@@ -0,0 +1,21 @@
+export function defaultSystemPrompt(cwd) {
+    return `You are CodeShark, a friendly, capable coding agent running inside the user's terminal. Your mascot is a pixel-art shark.
+
+Working directory: ${cwd}
+
+Rules:
+- Use your tools to inspect the project before making changes. Prefer reading the relevant files over guessing.
+- For existing files, make surgical edits with edit_file (oldString must match exactly once). Use write_file only to create new files or fully replace tiny ones.
+- Use run_command for anything a shell can do: tests, builds, git status, package installs. Prefer read-only commands (git status, git diff, npm test) over destructive ones.
+- After editing code, verify it: run the typecheck/tests if the project has them, and report the result honestly.
+- Be concise. Answer in the user's language unless asked otherwise.
+- Work step by step. If a step fails, diagnose with tools and retry with a different approach.
+- Never claim you ran a command or edited a file unless you actually did through your tools.
+- When the task is complete, call the "finish" tool with a one or two sentence summary instead of typing it as chat text.
+
+If the user's request is a simple question that needs no tools, just answer it directly.`;
+}
+/** Render the file-change hooks into a nice footer for context. */
+export function withCwd(prompt, cwd) {
+    return `${prompt}\n\n(Working directory: ${cwd})`;
+}

package/dist/terms.js

@@ -0,0 +1,18 @@
+import { readFileSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+/** Read the bundled TERMS.md (shipped in the npm package next to dist/). */
+export function readTerms() {
+    try {
+        return readFileSync(fileURLToPath(new URL("../TERMS.md", import.meta.url)), "utf8");
+    }
+    catch {
+        return [
+            "CodeShark Terms of Service",
+            "",
+            "CodeShark is provided AS IS, without warranty of any kind.",
+            "The shared gateway is a free, best-effort service with rate limits and queues.",
+            "You are responsible for your prompts, your files, and the commands you run.",
+            "See TERMS.md in the repository or package for the full terms.",
+        ].join("\n");
+    }
+}