@stage-labs/metro 0.1.0-beta.26 → 0.1.0-beta.261

package/runtime/node_modules/@metro-labs/daemon/src/claude/files.ts

@@ -0,0 +1,359 @@
+import { createReadStream, existsSync, readdirEntries, readdirNames, readRange, statSync } from '../agent-user/agent-fs.js';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { createInterface } from 'node:readline';
+import { ApiError } from '@metro-labs/http/api-error';
+import { isRecord } from '@metro-labs/core/is-record';
+import { removeHome, writeHomeText } from '../agent-user/home-fs.js';
+import { agentUser } from '../agent-user/user.js';
+
+export const PROJECT_RE = /^[A-Za-z0-9._-]{1,200}$/;
+export const SESSION_RE = /^[A-Za-z0-9][A-Za-z0-9-]{7,63}$/;
+const MEMORY_RE = /^[A-Za-z0-9][A-Za-z0-9._-]{0,119}\.md$/;
+const FOLDER_RE = /^[A-Za-z0-9][A-Za-z0-9._-]{0,119}$/;
+const MEMORY_DEPTH = 6;
+const MEMORY_FILES_MAX = 5000;
+const EDGE_BYTES = 64 * 1024;
+const TEXT_CAP = 20_000;
+const RESULT_CAP = 8_000;
+const INPUT_CAP = 4_000;
+const FILE_CAP = 1_000_000;
+const MEMORY_MAX = 256 * 1024;
+const MEMORY_MODE = 0o644;
+const NOISE = ['<task-notification>', '<system-reminder>', '<local-command', '<command-name>'];
+
+export function claudeDir(): string {
+  const explicit = process.env.METRO_CLAUDE_DIR?.trim();
+  if (explicit !== undefined && explicit !== '') return explicit;
+  const cfg = process.env.CLAUDE_CONFIG_DIR?.trim();
+  if (cfg !== undefined && cfg !== '') return cfg;
+  const user = agentUser();
+  return join(user === null ? homedir() : user.home, '.claude');
+}
+
+const str = (v: unknown): string | null => (typeof v === 'string' && v !== '' ? v : null);
+
+export function safeName(value: string, re: RegExp, what: string): string {
+  if (!re.test(value) || value === '.' || value === '..')
+    throw new ApiError(`that is not a ${what}`, 400);
+  return value;
+}
+
+function projectDir(project: string, dir: string): string {
+  const path = join(dir, 'projects', safeName(project, PROJECT_RE, 'Claude project id'));
+  if (!existsSync(path)) throw new ApiError('no such Claude project', 404);
+  return path;
+}
+
+function parseLines(text: string): Record<string, unknown>[] {
+  const out: Record<string, unknown>[] = [];
+  for (const line of text.split('\n')) {
+    if (line.trim() === '') continue;
+    try {
+      const parsed: unknown = JSON.parse(line);
+      if (isRecord(parsed)) out.push(parsed);
+    } catch {
+      continue;
+    }
+  }
+  return out;
+}
+
+function readEdge(path: string, fromEnd: boolean): Record<string, unknown>[] {
+  const size = statSync(path).size;
+  const length = Math.min(EDGE_BYTES, size);
+  const text = readRange(path, fromEnd ? size - length : 0, length).toString('utf8');
+  const cut = fromEnd && length < size ? text.indexOf('\n') + 1 : 0;
+  return parseLines(text.slice(cut));
+}
+
+export interface ClaudeProject {
+  id: string;
+  cwd: string | null;
+  sessions: number;
+  lastActiveAt: string | null;
+  hasMemory: boolean;
+}
+
+function sessionFiles(path: string): { name: string; mtime: number; size: number }[] {
+  return readdirNames(path)
+    .filter((name) => name.endsWith('.jsonl'))
+    .map((name) => {
+      const s = statSync(join(path, name));
+      return { name, mtime: s.mtimeMs, size: s.size };
+    })
+    .sort((a, b) => b.mtime - a.mtime);
+}
+
+function cwdOf(path: string): string | null {
+  for (const line of readEdge(path, false)) {
+    const cwd = str(line.cwd);
+    if (cwd !== null) return cwd;
+  }
+  return null;
+}
+
+export function listClaudeProjects(dir = claudeDir()): ClaudeProject[] {
+  const root = join(dir, 'projects');
+  if (!existsSync(root)) return [];
+  return readdirEntries(root)
+    .filter((entry) => entry.isDirectory() && PROJECT_RE.test(entry.name))
+    .map((entry) => {
+      const path = join(root, entry.name);
+      const files = sessionFiles(path);
+      const newest = files[0];
+      return {
+        id: entry.name,
+        cwd: newest === undefined ? null : cwdOf(join(path, newest.name)),
+        sessions: files.length,
+        lastActiveAt: newest === undefined ? null : new Date(newest.mtime).toISOString(),
+        hasMemory: existsSync(join(path, 'memory')),
+      };
+    })
+    .sort((a, b) => (b.lastActiveAt ?? '').localeCompare(a.lastActiveAt ?? ''));
+}
+
+export interface ClaudeSession {
+  id: string;
+  title: string;
+  startedAt: string | null;
+  lastAt: string;
+  bytes: number;
+  gitBranch: string | null;
+  version: string | null;
+}
+
+function firstPrompt(lines: Record<string, unknown>[]): string | null {
+  for (const line of lines) {
+    if (line.type !== 'user' || line.isSidechain === true) continue;
+    const message = isRecord(line.message) ? line.message : {};
+    const text = promptText(message.content);
+    if (text !== null && !NOISE.some((n) => text.startsWith(n))) return text.slice(0, 120);
+  }
+  return null;
+}
+
+function promptText(content: unknown): string | null {
+  if (typeof content === 'string') return content.trim() === '' ? null : content.trim();
+  if (!Array.isArray(content)) return null;
+  for (const block of content)
+    if (isRecord(block) && block.type === 'text') {
+      const text = str(block.text);
+      if (text !== null) return text.trim();
+    }
+  return null;
+}
+
+function sessionOf(path: string, id: string, mtime: number, size: number): ClaudeSession {
+  const head = readEdge(path, false);
+  const tail = readEdge(path, true);
+  const titled = [...tail].reverse().find((l) => l.type === 'ai-title' && str(l.aiTitle) !== null);
+  const meta = head.find((l) => l.type === 'user' || l.type === 'assistant') ?? {};
+  return {
+    id,
+    title: (titled === undefined ? null : str(titled.aiTitle)) ?? firstPrompt(head) ?? 'Untitled session',
+    startedAt: str(head.find((l) => str(l.timestamp) !== null)?.timestamp) ?? null,
+    lastAt: new Date(mtime).toISOString(),
+    bytes: size,
+    gitBranch: str(meta.gitBranch),
+    version: str(meta.version),
+  };
+}
+
+export function listClaudeSessions(project: string, dir = claudeDir()): ClaudeSession[] {
+  const path = projectDir(project, dir);
+  return sessionFiles(path).map((f) =>
+    sessionOf(join(path, f.name), f.name.slice(0, -'.jsonl'.length), f.mtime, f.size),
+  );
+}
+
+type Block =
+  | { kind: 'text'; text: string }
+  | { kind: 'tool_use'; name: string; input: string }
+  | { kind: 'tool_result'; text: string; isError: boolean }
+  | { kind: 'thinking' }
+  | { kind: 'image' };
+
+interface TranscriptEntry {
+  uuid: string;
+  at: string | null;
+  role: 'user' | 'assistant';
+  blocks: Block[];
+}
+
+export interface TranscriptPage {
+  entries: TranscriptEntry[];
+  total: number;
+  next: number | null;
+}
+
+const cap = (text: string, max: number): string =>
+  text.length > max ? `${text.slice(0, max)}\n… (${String(text.length - max)} more characters)` : text;
+
+function resultText(content: unknown): string {
+  if (typeof content === 'string') return content;
+  if (!Array.isArray(content)) return '';
+  return content
+    .map((b) => (isRecord(b) && b.type === 'text' ? (str(b.text) ?? '') : ''))
+    .filter((t) => t !== '')
+    .join('\n');
+}
+
+function textBlock(raw: Record<string, unknown>): Block | null {
+  const text = str(raw.text);
+  if (text === null || NOISE.some((n) => text.startsWith(n))) return null;
+  return { kind: 'text', text: cap(text, TEXT_CAP) };
+}
+
+const BLOCKS: Record<string, (raw: Record<string, unknown>) => Block | null> = {
+  text: textBlock,
+  tool_use: (raw) => ({
+    kind: 'tool_use',
+    name: str(raw.name) ?? 'tool',
+    input: cap(JSON.stringify(raw.input ?? {}, null, 1), INPUT_CAP),
+  }),
+  tool_result: (raw) => ({
+    kind: 'tool_result',
+    text: cap(resultText(raw.content), RESULT_CAP),
+    isError: raw.is_error === true,
+  }),
+  thinking: () => ({ kind: 'thinking' }),
+  image: () => ({ kind: 'image' }),
+};
+
+function blockOf(raw: unknown): Block | null {
+  if (!isRecord(raw) || typeof raw.type !== 'string') return null;
+  const make = BLOCKS[raw.type];
+  return make === undefined ? null : make(raw);
+}
+
+function entryOf(line: Record<string, unknown>): TranscriptEntry | null {
+  if ((line.type !== 'user' && line.type !== 'assistant') || line.isSidechain === true) return null;
+  const message = isRecord(line.message) ? line.message : {};
+  const content = message.content;
+  const blocks =
+    typeof content === 'string'
+      ? [blockOf({ type: 'text', text: content })]
+      : Array.isArray(content)
+        ? content.map(blockOf)
+        : [];
+  const kept = blocks.filter((b): b is Block => b !== null);
+  if (kept.length === 0) return null;
+  return { uuid: str(line.uuid) ?? '', at: str(line.timestamp), role: line.type, blocks: kept };
+}
+
+export async function readTranscript(
+  project: string,
+  session: string,
+  offset: number,
+  limit: number,
+  dir = claudeDir(),
+): Promise<TranscriptPage> {
+  const path = join(projectDir(project, dir), `${safeName(session, SESSION_RE, 'session id')}.jsonl`);
+  if (!existsSync(path)) throw new ApiError('no such session', 404);
+  const entries: TranscriptEntry[] = [];
+  let total = 0;
+  const lines = createInterface({ input: createReadStream(path), crlfDelay: Infinity });
+  for await (const line of lines) {
+    const parsed = parseLines(line)[0];
+    const entry = parsed === undefined ? null : entryOf(parsed);
+    if (entry === null) continue;
+    if (total >= offset && entries.length < limit) entries.push(entry);
+    total += 1;
+  }
+  return { entries, total, next: offset + limit < total ? offset + limit : null };
+}
+
+export function deleteClaudeSession(project: string, session: string, dir = claudeDir()): void {
+  const root = projectDir(project, dir);
+  const id = safeName(session, SESSION_RE, 'session id');
+  const file = join(root, `${id}.jsonl`);
+  if (!existsSync(file)) throw new ApiError('no such session', 404);
+  removeHome(file);
+  removeHome(join(root, id), true);
+}
+
+export interface MemoryFile {
+  name: string;
+  bytes: number;
+  modifiedAt: string;
+}
+
+export interface MemoryListing {
+  files: MemoryFile[];
+  index: string | null;
+}
+
+function memoryDir(project: string, dir: string): string {
+  return join(projectDir(project, dir), 'memory');
+}
+
+function readCapped(path: string): string {
+  const size = statSync(path).size;
+  const text = readRange(path, 0, Math.min(size, FILE_CAP)).toString('utf8');
+  return size > FILE_CAP ? `${text}\n\n… truncated` : text;
+}
+
+function memoryName(name: string): string[] {
+  const segments = name.split('/');
+  const file = segments.at(-1) ?? '';
+  const folders = segments.slice(0, -1);
+  if (folders.length >= MEMORY_DEPTH || !MEMORY_RE.test(file) || folders.some((f) => !FOLDER_RE.test(f)))
+    throw new ApiError('that is not a memory file name', 400);
+  return segments;
+}
+
+function memoryPath(project: string, name: string, dir: string): string {
+  return join(memoryDir(project, dir), ...memoryName(name));
+}
+
+function walkMemory(root: string, folders: string[], out: MemoryFile[]): void {
+  for (const entry of readdirEntries(join(root, ...folders))) {
+    if (out.length >= MEMORY_FILES_MAX) return;
+    const inner = [...folders, entry.name];
+    if (entry.isDirectory() && FOLDER_RE.test(entry.name) && inner.length < MEMORY_DEPTH) walkMemory(root, inner, out);
+    else if (entry.isFile() && MEMORY_RE.test(entry.name) && !(folders.length === 0 && entry.name === 'MEMORY.md')) {
+      const s = statSync(join(root, ...inner));
+      out.push({ name: inner.join('/'), bytes: s.size, modifiedAt: s.mtime.toISOString() });
+    }
+  }
+}
+
+export function listMemory(project: string, dir = claudeDir()): MemoryListing {
+  const path = memoryDir(project, dir);
+  if (!existsSync(path)) return { files: [], index: null };
+  const files: MemoryFile[] = [];
+  walkMemory(path, [], files);
+  files.sort((a, b) => b.modifiedAt.localeCompare(a.modifiedAt) || a.name.localeCompare(b.name));
+  const index = join(path, 'MEMORY.md');
+  return { files, index: existsSync(index) ? readCapped(index) : null };
+}
+
+const stampOf = (raw: string | undefined): Date | null => {
+  if (raw === undefined) return null;
+  const at = new Date(raw);
+  return Number.isNaN(at.getTime()) ? null : at;
+};
+
+export function writeMemoryFile(project: string, name: string, text: string, dir = claudeDir(), modifiedAt?: string): MemoryFile {
+  if (text.trim() === '') throw new ApiError('a memory file cannot be empty', 400);
+  if (Buffer.byteLength(text, 'utf8') > MEMORY_MAX)
+    throw new ApiError(`a memory file is at most ${String(MEMORY_MAX)} bytes`, 400);
+  const path = memoryPath(project, name, dir);
+  writeHomeText(path, text, MEMORY_MODE, stampOf(modifiedAt) ?? undefined);
+  const stat = statSync(path);
+  return { name, bytes: stat.size, modifiedAt: stat.mtime.toISOString() };
+}
+
+export function deleteMemoryFile(project: string, name: string, dir = claudeDir()): string {
+  const path = memoryPath(project, name, dir);
+  if (!existsSync(path)) throw new ApiError('no such memory file', 404);
+  removeHome(path);
+  return name;
+}
+
+export function readMemoryFile(project: string, name: string, dir = claudeDir()): string {
+  const path = memoryPath(project, name, dir);
+  if (!existsSync(path)) throw new ApiError('no such memory file', 404);
+  return readCapped(path);
+}

package/runtime/node_modules/@metro-labs/daemon/src/claude/login.ts

@@ -0,0 +1,242 @@
+import { randomBytes } from 'node:crypto';
+import { spawnSync } from 'node:child_process';
+import { markOnboardingDone } from './onboarding.js';
+import { ApiError } from '@metro-labs/http/api-error';
+import { isRecord } from '@metro-labs/core/is-record';
+import { errMsg, log } from '@metro-labs/core/log';
+import { agentCommand, asAgent, claudeBin } from '../agent-user/user.js';
+
+const LOGIN_ARGS = ['auth', 'login', '--claudeai'];
+const STATUS_COMMAND = ['auth', 'status', '--json'];
+const TTL_MS = 10 * 60_000;
+const OUTPUT_MAX = 8_000;
+const COLS = 100;
+const ROWS = 40;
+const ESC = String.fromCharCode(27);
+const CSI = /^\[[0-9;?]*[a-zA-Z]/;
+const URL_RE = /https:\/\/[^\s"'`]+/g;
+
+type LoginState = 'pending' | 'done' | 'failed';
+
+export interface LoginView {
+  id: string;
+  state: LoginState;
+  url: string | null;
+  output: string;
+  error: string | null;
+}
+
+export interface LoginTarget {
+  key: string;
+  env: Record<string, string>;
+  settle: (ok: boolean) => Promise<string | null>;
+}
+
+interface Live {
+  id: string;
+  startedAt: number;
+  state: LoginState;
+  output: string;
+  error: string | null;
+  target: LoginTarget | null;
+  write: (text: string) => void;
+  kill: () => void;
+}
+
+export interface LoginDeps {
+  command?: string[];
+}
+
+let live: Live | null = null;
+
+export function plainText(raw: string): string {
+  return raw
+    .split(ESC)
+    .map((part, index) => (index === 0 ? part : part.replace(CSI, '')))
+    .join('');
+}
+
+export function loginUrlIn(output: string): string | null {
+  const found = output.match(URL_RE) ?? [];
+  const trimmed = found.map((url) => url.replace(/[.,)]+$/, ''));
+  return trimmed.find((url) => url.includes('claude.ai') || url.includes('anthropic.com')) ?? trimmed[0] ?? null;
+}
+
+const view = (session: Live): LoginView => ({
+  id: session.id,
+  state: session.state,
+  url: loginUrlIn(session.output),
+  output: session.output,
+  error: session.error,
+});
+
+function sweep(now: number): void {
+  if (live !== null && live.state !== 'pending') return;
+  if (live !== null && now - live.startedAt > TTL_MS) {
+    live.kill();
+    live = null;
+  }
+}
+
+const CHECK_MS = 30_000;
+
+interface Checked<T> {
+  at: number;
+  value: T;
+}
+
+const checks: { installed?: Checked<boolean>; account?: Checked<ClaudeAccount> } = {};
+
+function remembered<T>(slot: Checked<T> | undefined, read: () => T, keep: (c: Checked<T>) => void): T {
+  const now = Date.now();
+  if (slot !== undefined && now - slot.at < CHECK_MS) return slot.value;
+  const value = read();
+  keep({ at: now, value });
+  return value;
+}
+
+function forgetClaudeChecks(): void {
+  delete checks.installed;
+  delete checks.account;
+}
+
+export function claudeInstalled(): boolean {
+  return remembered(checks.installed, () => {
+    const run = spawnSync(...asAgent(claudeBin(), ['--version']), { stdio: 'ignore' });
+    return run.error === undefined && run.status === 0;
+  }, (c) => {
+    checks.installed = c;
+  });
+}
+
+export interface ClaudeAccount {
+  signedIn: boolean;
+  account: string | null;
+  plan: string | null;
+}
+
+const SIGNED_OUT: ClaudeAccount = { signedIn: false, account: null, plan: null };
+
+export function claudeAccount(): ClaudeAccount {
+  return remembered(checks.account, () => claudeAccountIn({}), (c) => {
+    checks.account = c;
+  });
+}
+
+const textOf = (value: unknown): string | null => (typeof value === 'string' && value !== '' ? value : null);
+
+export function claudeAccountIn(env: Record<string, string>): ClaudeAccount {
+  const run = spawnSync(...asAgent(claudeBin(), STATUS_COMMAND, env), { encoding: 'utf8', stdio: ['ignore', 'pipe', 'ignore'], env: { ...process.env, ...env } });
+  if (run.error !== undefined || typeof run.stdout !== 'string') return SIGNED_OUT;
+  try {
+    const parsed: unknown = JSON.parse(run.stdout);
+    if (!isRecord(parsed)) return SIGNED_OUT;
+    const account = textOf(parsed.email) ?? textOf(parsed.organization) ?? textOf(parsed.orgName);
+    return { signedIn: parsed.authenticated === true || parsed.loggedIn === true || account !== null, account, plan: textOf(parsed.subscriptionType) };
+  } catch {
+    return SIGNED_OUT;
+  }
+}
+
+function note(session: Live, chunk: string): void {
+  const next = session.output + plainText(chunk);
+  session.output = next.length > OUTPUT_MAX ? next.slice(next.length - OUTPUT_MAX) : next;
+}
+
+async function verdict(session: Live, code: number): Promise<string | null> {
+  const problem = code === 0 ? null : `the login ended with status ${String(code)}`;
+  if (session.target === null) return problem;
+  try {
+    return (await session.target.settle(problem === null)) ?? problem;
+  } catch (err) {
+    return errMsg(err);
+  }
+}
+
+function runningFor(target: LoginTarget | null): Live | null {
+  if (live === null || live.state !== 'pending') return null;
+  if (live.target?.key === target?.key) return live;
+  live.kill();
+  live = null;
+  return null;
+}
+
+export function startClaudeLogin(deps: LoginDeps = {}, now = Date.now(), target: LoginTarget | null = null): LoginView {
+  sweep(now);
+  const running = runningFor(target);
+  if (running !== null) return view(running);
+  const env = { TERM: 'xterm-256color', ...target?.env };
+  const command = deps.command ?? agentCommand([claudeBin(), ...LOGIN_ARGS], env);
+  const session: Live = {
+    id: randomBytes(9).toString('base64url'),
+    startedAt: now,
+    state: 'pending',
+    output: '',
+    error: null,
+    target,
+    write: () => undefined,
+    kill: () => undefined,
+  };
+  const terminal = new Bun.Terminal({
+    cols: COLS,
+    rows: ROWS,
+    data: (_term, chunk: Uint8Array) => {
+      note(session, Buffer.from(chunk).toString('utf8'));
+    },
+  });
+  const proc = Bun.spawn(command, { terminal, env: { ...process.env, ...env } });
+  session.write = (text: string) => {
+    terminal.write(Buffer.from(`${text}\r`, 'utf8'));
+  };
+  session.kill = () => {
+    proc.kill();
+    terminal.close();
+  };
+  proc.exited
+    .then(async (code) => {
+      session.write = () => undefined;
+      terminal.close();
+      forgetClaudeChecks();
+      session.error = await verdict(session, code);
+      session.state = session.error === null ? 'done' : 'failed';
+      const onboarding = session.error === null ? markOnboardingDone() : 'skipped';
+      log.info({ state: session.state, onboarding, own: session.target !== null }, 'claude-login: the official login finished');
+    })
+    .catch((err: unknown) => {
+      session.state = 'failed';
+      session.error = errMsg(err);
+    });
+  live = session;
+  log.info({ command: command[0] }, 'claude-login: started');
+  return view(session);
+}
+
+function current(id: string): Live {
+  const session = live;
+  if (session?.id !== id) throw new ApiError('that sign-in is over; start another one', 404);
+  return session;
+}
+
+export function claudeLoginView(id: string, now = Date.now()): LoginView {
+  sweep(now);
+  return view(current(id));
+}
+
+export function answerClaudeLogin(id: string, text: string): LoginView {
+  const session = current(id);
+  if (session.state !== 'pending') throw new ApiError('that sign-in has already finished', 409);
+  session.write(text);
+  return view(session);
+}
+
+export function endClaudeLogin(id: string): { ended: true } {
+  const session = current(id);
+  session.kill();
+  live = null;
+  return { ended: true };
+}
+
+export const forgetClaudeLogin = (): void => {
+  live?.kill();
+  live = null;
+};