subrouter-cli 0.0.0-stage → 0.1.0

package/src/index.ts

@@ -0,0 +1,248 @@
+#!/usr/bin/env node
+// sub — a personal agentic coding CLI for the subrouter AI router.
+
+import fs from 'node:fs';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { RouterClient, errorMessage, dialectFor } from './client.ts';
+import { resolveConfig, saveConfigFile, type ResolvedConfig } from './config.ts';
+import { startRepl } from './repl.ts';
+import { renderMarkdown } from './markdown.ts';
+import { EFFORT_LEVELS, parseEffort, type EffortLevel } from './effort.ts';
+import { loadImage, addImages, type ImageAttachment } from './images.ts';
+import type { ChatMessage, ChatRequest } from './types.ts';
+
+const here = path.dirname(fileURLToPath(import.meta.url));
+
+function version(): string {
+  try {
+    const pkg = JSON.parse(fs.readFileSync(path.join(here, '..', 'package.json'), 'utf8')) as {
+      version?: string;
+    };
+    return pkg.version ?? '0.0.0';
+  } catch {
+    return '0.1.0';
+  }
+}
+
+function printHelp(): void {
+  console.log(`sub — agentic coding CLI for the subrouter AI router
+
+Usage:
+  sub                     start the REPL
+  sub -p "prompt"         one-shot, non-interactive
+  sub --check             connectivity + model catalog diagnostic
+  sub --model <id>        set the session model
+  sub --effort <level>    auto, low, medium, high, xhigh, max
+  sub --image <path>      attach an image (repeat for more; requires a vision model)
+  sub --absolute          entire drive + arbitrary commands; warning/typed acknowledgement
+  sub --auto              guarded auto: files + recognized tasks; risky/unknown commands ask
+  sub --auto-approve      allow project file changes only; all terminal commands still ask
+  sub --debug             print raw SSE frames (credentials never shown)
+  sub --no-color          plain output
+  sub --version           print the version
+
+API key: set SUBROUTER_API_KEY or create ~/.config/subrouter-cli/config.json
+         (see config.example.json in this repo).`);
+}
+
+interface Flags {
+  check: boolean;
+  showVersion: boolean;
+  showHelp: boolean;
+  model?: string;
+  effort?: EffortLevel;
+  prompt?: string;
+  images: string[];
+  debug: boolean;
+  autoApprove?: boolean;
+  autoMode?: boolean;
+  absoluteMode?: boolean;
+}
+
+function parseArgs(argv: string[]): Flags | null {
+  const flags: Flags = { check: false, showVersion: false, showHelp: false, images: [], debug: false };
+  for (let i = 0; i < argv.length; i++) {
+    const arg = argv[i];
+    if (arg === '--check') flags.check = true;
+    else if (arg === '--version' || arg === '-v') flags.showVersion = true;
+    else if (arg === '--help' || arg === '-h') flags.showHelp = true;
+    else if (arg === '--debug') flags.debug = true;
+    else if (arg === '--absolute') flags.absoluteMode = true;
+    else if (arg === '--auto') flags.autoMode = true;
+    else if (arg === '--auto-approve') flags.autoApprove = true;
+    else if (arg === '--no-color') process.env.NO_COLOR = '1';
+    else if ((arg === '--model' || arg === '-m') && i + 1 < argv.length) flags.model = argv[++i];
+    else if (arg === '--effort') {
+      const effort = parseEffort(argv[++i]);
+      if (!effort) {
+        console.error(`--effort requires one of: ${EFFORT_LEVELS.join(', ')}`);
+        return null;
+      }
+      flags.effort = effort;
+    } else if (arg === '--image') {
+      const file = argv[++i];
+      if (!file || file.startsWith('--')) {
+        console.error('--image requires a local image path');
+        return null;
+      }
+      flags.images.push(file);
+    } else if ((arg === '--prompt' || arg === '-p') && i + 1 < argv.length) flags.prompt = argv[++i];
+    else {
+      console.error(`Unknown argument: ${arg}`);
+      console.error('Run `sub --help` for usage.');
+      return null;
+    }
+  }
+  if ([flags.absoluteMode, flags.autoMode, flags.autoApprove].filter(Boolean).length > 1) {
+    console.error('Choose --absolute, --auto or --auto-approve, not both or multiple.');
+    return null;
+  }
+  return flags;
+}
+
+function say(line: string): void {
+  process.stdout.write(line + '\n');
+}
+
+async function runCheck(client: RouterClient, config: ResolvedConfig): Promise<number> {
+  say(`base:    ${config.baseUrl}`);
+  const ok = await client.healthz();
+  say(`healthz: ${ok ? 'ok' : 'FAILED'}`);
+  try {
+    const models = await client.listModels();
+    say(`models:  ${models.length} in catalog`);
+    for (const m of models.slice(0, 5)) {
+      const ctx = m.contextLength ? ` (${m.contextLength.toLocaleString()} ctx)` : '';
+      say(`  - ${m.id}${ctx}`);
+    }
+    if (models.length > 5) say('  …');
+  } catch (err) {
+    const e = err as { status?: number };
+    say(`models:  FAILED (${e.status ?? '?'}) ${errorMessage(err)}`);
+    return 1;
+  }
+  return ok ? 0 : 1;
+}
+
+/**
+ * Session model: --model flag > config default > first-run bootstrap.
+ * The bootstrap prefers the model verified to work on this key, then the first catalog
+ * entry, and persists the choice — never a hardcoded guess.
+ */
+async function resolveSessionModel(
+  flagsModel: string | undefined,
+  config: ResolvedConfig,
+  client: RouterClient,
+): Promise<string> {
+  if (flagsModel) return flagsModel;
+  if (config.defaultModel) return config.defaultModel;
+  let catalog;
+  try {
+    catalog = await client.listModels();
+  } catch (err) {
+    throw new Error(
+      'No default model configured and the catalog fetch failed — pass --model <id>. ' +
+        `(${errorMessage(err)})`,
+    );
+  }
+  const preferred = ['opencode/deepseek-v4-pro', 'deepseek/deepseek-v4-pro'];
+  const pick = preferred.find((id) => catalog.some((m) => m.id === id)) ?? catalog[0]?.id;
+  if (!pick) throw new Error('The catalog is empty — pass --model <id>.');
+  saveConfigFile({ default_model: pick });
+  return pick;
+}
+
+function systemPrompt(): string {
+  return `You are sub, a coding agent running in the user's terminal. Be concise and direct. Today: ${new Date().toISOString().slice(0, 10)}.`;
+}
+
+async function runOneShot(
+  client: RouterClient,
+  config: ResolvedConfig,
+  model: string,
+  prompt: string,
+  images: ImageAttachment[],
+): Promise<number> {
+  const messages: ChatMessage[] = [{ role: 'user', content: prompt, ...(images.length ? { images } : {}) }];
+  const req: ChatRequest = {
+    model,
+    dialect: dialectFor(model),
+    messages,
+    system: systemPrompt(),
+    stream: false,
+    maxTokens: config.maxOutputTokens,
+    effort: config.reasoningEffort,
+  };
+  try {
+    const turn = await client.complete(req);
+    const content = turn.message.content || '(no response)';
+    say(renderMarkdown(content).replace(/\n+$/, ''));
+    if (turn.stopReason === 'length') say('! response truncated — consider a shorter scope');
+    return 0;
+  } catch (err) {
+    say('× ' + errorMessage(err));
+    return 1;
+  }
+}
+
+async function main(): Promise<number> {
+  const flags = parseArgs(process.argv.slice(2));
+  if (!flags) return 2;
+  if (flags.showVersion) {
+    console.log(version());
+    return 0;
+  }
+  if (flags.showHelp) {
+    printHelp();
+    return 0;
+  }
+
+  if (flags.absoluteMode && (flags.prompt !== undefined || flags.check || !process.stdin.isTTY || !process.stdout.isTTY)) {
+    console.error('--absolute requires an interactive REPL and typed warning acknowledgement; not -p or --check.');
+    return 2;
+  }
+  if (flags.check && flags.images.length) {
+    console.error('--image is for prompts, not --check');
+    return 2;
+  }
+  let images: ImageAttachment[] = [];
+  try {
+    for (const file of flags.images) images = addImages(images, [await loadImage(file, process.cwd())]);
+  } catch (err) {
+    console.error('× ' + errorMessage(err));
+    return 2;
+  }
+
+  const { config, warnings } = resolveConfig();
+  if (flags.effort) config.reasoningEffort = flags.effort;
+  for (const w of warnings) console.error('! ' + w);
+  if (!config.apiKey) {
+    console.error('No API key found.');
+    console.error('Set the SUBROUTER_API_KEY environment variable, or create:');
+    console.error('  ~/.config/subrouter-cli/config.json  (see config.example.json)');
+    return 1;
+  }
+
+  const client = new RouterClient({
+    baseUrl: config.baseUrl,
+    apiKey: config.apiKey,
+    debug: flags.debug ? (line) => console.error('[debug] ' + line) : undefined,
+  });
+
+  if (flags.check) return await runCheck(client, config);
+
+  const model = await resolveSessionModel(flags.model, config, client);
+  if (flags.prompt !== undefined) {
+    return await runOneShot(client, config, model, flags.prompt, images);
+  }
+  startRepl({ client, config, model, images, autoApprove: flags.autoApprove, autoMode: flags.autoMode, absoluteMode: flags.absoluteMode });
+  return 0;
+}
+
+try {
+  process.exitCode = await main();
+} catch (err) {
+  console.error('× ' + errorMessage(err));
+  process.exitCode = 1;
+}