@selesai/code 0.10.2 → 0.12.0

package/dist/extensions/question/selection-mode.ts

@@ -1,47 +1,47 @@
-// SelectionMode strategy — isolates single-vs-multi selection behavior.
-
-import type { QuestionOption } from "./types.ts";
-
-export interface QuestionListState {
-	selectedIndex: number;
-	checked: ReadonlySet<number>;
-	options: QuestionOption[];
-}
-
-export interface SelectionMode {
-	readonly multi: boolean;
-	toggle(checked: Set<number>, index: number): void;
-	buildResult(state: QuestionListState): string[];
-	/** Returns whether to toggle (multi) or just select (single), and the new selectedIndex. */
-	handleNumberKey(idx: number, count: number): { toggle: boolean; selectIndex: number };
-}
-
-export const SingleSelect: SelectionMode = {
-	multi: false,
-	toggle() {},
-	buildResult(state) {
-		const value = state.options[state.selectedIndex]?.value;
-		return value ? [value] : [];
-	},
-	handleNumberKey(idx, count) {
-		return { toggle: false, selectIndex: Math.min(idx, count - 1) };
-	},
-};
-
-export const MultiSelect: SelectionMode = {
-	multi: true,
-	toggle(checked, index) {
-		if (index < 0) return;
-		if (checked.has(index)) checked.delete(index);
-		else checked.add(index);
-	},
-	buildResult(state) {
-		return Array.from(state.checked)
-			.sort((a, b) => a - b)
-			.map((i) => state.options[i]?.value)
-			.filter((value): value is string => !!value);
-	},
-	handleNumberKey(idx, count) {
-		return { toggle: true, selectIndex: Math.min(idx, count - 1) };
-	},
+// SelectionMode strategy — isolates single-vs-multi selection behavior.
+
+import type { QuestionOption } from "./types.ts";
+
+export interface QuestionListState {
+	selectedIndex: number;
+	checked: ReadonlySet<number>;
+	options: QuestionOption[];
+}
+
+export interface SelectionMode {
+	readonly multi: boolean;
+	toggle(checked: Set<number>, index: number): void;
+	buildResult(state: QuestionListState): string[];
+	/** Returns whether to toggle (multi) or just select (single), and the new selectedIndex. */
+	handleNumberKey(idx: number, count: number): { toggle: boolean; selectIndex: number };
+}
+
+export const SingleSelect: SelectionMode = {
+	multi: false,
+	toggle() {},
+	buildResult(state) {
+		const value = state.options[state.selectedIndex]?.value;
+		return value ? [value] : [];
+	},
+	handleNumberKey(idx, count) {
+		return { toggle: false, selectIndex: Math.min(idx, count - 1) };
+	},
+};
+
+export const MultiSelect: SelectionMode = {
+	multi: true,
+	toggle(checked, index) {
+		if (index < 0) return;
+		if (checked.has(index)) checked.delete(index);
+		else checked.add(index);
+	},
+	buildResult(state) {
+		return Array.from(state.checked)
+			.sort((a, b) => a - b)
+			.map((i) => state.options[i]?.value)
+			.filter((value): value is string => !!value);
+	},
+	handleNumberKey(idx, count) {
+		return { toggle: true, selectIndex: Math.min(idx, count - 1) };
+	},
 };

package/dist/extensions/question/shortcuts.ts

@@ -1,44 +1,44 @@
-// Shortcut resolution for the optional-comment toggle key.
-
-import { matchesKey } from "@earendil-works/pi-tui";
-
-import { DEFAULT_COMMENT_TOGGLE_KEY, DISABLED_SHORTCUT, SHORTCUT_DISABLE_VALUES } from "./constants.ts";
-import type { ResolvedShortcut, ResolvedShortcuts } from "./types.ts";
-
-export function normalizeShortcutSpec(value: string | null | undefined): string | null | undefined {
-	if (value === undefined) return undefined;
-	if (value === null) return null;
-	const trimmed = value.trim().toLowerCase();
-	if (SHORTCUT_DISABLE_VALUES.has(trimmed)) return null;
-	return trimmed;
-}
-
-export function isValidShortcutSpec(spec: string): boolean {
-	if (!spec) return false;
-	if (!/^[a-z0-9+_\-!@#$%^&*()|~`'":;,./<>?[\]{}=\\]+$/i.test(spec)) return false;
-	if (spec.startsWith("+") || spec.endsWith("+")) return false;
-	if (spec.includes("++")) return false;
-	return true;
-}
-
-export function buildShortcut(spec: string): ResolvedShortcut {
-	return { disabled: false, spec, matches: (data: string) => matchesKey(data, spec as any) };
-}
-
-export function resolveShortcut(
-	paramValue: string | null | undefined,
-	envValue: string | undefined,
-	defaultSpec: string,
-): ResolvedShortcut {
-	for (const raw of [paramValue, envValue, defaultSpec]) {
-		const normalized = normalizeShortcutSpec(raw);
-		if (normalized === undefined) continue;
-		if (normalized === null) return DISABLED_SHORTCUT;
-		if (isValidShortcutSpec(normalized)) return buildShortcut(normalized);
-	}
-	return DISABLED_SHORTCUT;
-}
-
-export function resolveShortcuts(commentToggleKey: string | null | undefined, commentEnv: string | undefined): ResolvedShortcuts {
-	return { commentToggle: resolveShortcut(commentToggleKey, commentEnv, DEFAULT_COMMENT_TOGGLE_KEY) };
-}
+// Shortcut resolution for the optional-comment toggle key.
+
+import { matchesKey } from "@earendil-works/pi-tui";
+
+import { DEFAULT_COMMENT_TOGGLE_KEY, DISABLED_SHORTCUT, SHORTCUT_DISABLE_VALUES } from "./constants.ts";
+import type { ResolvedShortcut, ResolvedShortcuts } from "./types.ts";
+
+export function normalizeShortcutSpec(value: string | null | undefined): string | null | undefined {
+	if (value === undefined) return undefined;
+	if (value === null) return null;
+	const trimmed = value.trim().toLowerCase();
+	if (SHORTCUT_DISABLE_VALUES.has(trimmed)) return null;
+	return trimmed;
+}
+
+export function isValidShortcutSpec(spec: string): boolean {
+	if (!spec) return false;
+	if (!/^[a-z0-9+_\-!@#$%^&*()|~`'":;,./<>?[\]{}=\\]+$/i.test(spec)) return false;
+	if (spec.startsWith("+") || spec.endsWith("+")) return false;
+	if (spec.includes("++")) return false;
+	return true;
+}
+
+export function buildShortcut(spec: string): ResolvedShortcut {
+	return { disabled: false, spec, matches: (data: string) => matchesKey(data, spec as any) };
+}
+
+export function resolveShortcut(
+	paramValue: string | null | undefined,
+	envValue: string | undefined,
+	defaultSpec: string,
+): ResolvedShortcut {
+	for (const raw of [paramValue, envValue, defaultSpec]) {
+		const normalized = normalizeShortcutSpec(raw);
+		if (normalized === undefined) continue;
+		if (normalized === null) return DISABLED_SHORTCUT;
+		if (isValidShortcutSpec(normalized)) return buildShortcut(normalized);
+	}
+	return DISABLED_SHORTCUT;
+}
+
+export function resolveShortcuts(commentToggleKey: string | null | undefined, commentEnv: string | undefined): ResolvedShortcuts {
+	return { commentToggle: resolveShortcut(commentToggleKey, commentEnv, DEFAULT_COMMENT_TOGGLE_KEY) };
+}

package/dist/extensions/question/types.ts

@@ -1,132 +1,132 @@
-// Shared type definitions for the question extension.
-
-import type { KeybindingsManager } from "@selesai/code";
-
-export interface RawQuestionOption {
-	value: string;
-	label: string;
-	description?: string;
-}
-
-export interface QuestionOption {
-	value: string;
-	label: string;
-	description?: string;
-}
-
-interface RawQuestionBase {
-	id?: string;
-	label?: string;
-	question: string;
-	context?: string;
-}
-
-export interface RawSelectQuestion extends RawQuestionBase {
-	type: "select";
-	options: RawQuestionOption[];
-	allowOther?: boolean;
-}
-
-export interface RawMultiSelectQuestion extends RawQuestionBase {
-	type: "multiselect";
-	options: RawQuestionOption[];
-	allowOther?: boolean;
-}
-
-export interface RawTextQuestion extends RawQuestionBase {
-	type: "text";
-}
-
-export type RawQuestion = RawSelectQuestion | RawMultiSelectQuestion | RawTextQuestion;
-
-export interface PreparedQuestionBase {
-	id: string;
-	label: string;
-	question: string;
-	context?: string;
-}
-
-export interface PreparedSelectQuestion extends PreparedQuestionBase {
-	type: "select";
-	options: QuestionOption[];
-	allowOther: boolean;
-}
-
-export interface PreparedMultiSelectQuestion extends PreparedQuestionBase {
-	type: "multiselect";
-	options: QuestionOption[];
-	allowOther: boolean;
-}
-
-export interface PreparedTextQuestion extends PreparedQuestionBase {
-	type: "text";
-	options: [];
-	allowOther: false;
-}
-
-export type PreparedQuestion = PreparedSelectQuestion | PreparedMultiSelectQuestion | PreparedTextQuestion;
-
-export type QuestionResponse =
-	| { kind: "selection"; values: string[]; otherText?: string }
-	| { kind: "text"; text: string };
-
-export type QuestionAnswer =
-	| { id: string; status: "answered"; response: QuestionResponse }
-	| { id: string; status: "skipped" | "unanswered" };
-
-export type QuestionToolResult =
-	| { status: "submitted"; answers: QuestionAnswer[] }
-	| { status: "cancelled"; reason: "user" };
-
-export interface DraftQuestionState {
-	status: "answered" | "skipped";
-	response?: QuestionResponse;
-}
-
-// ---------------------------------------------------------------------------
-// Legacy list internals.
-// ---------------------------------------------------------------------------
-
-export type ResolvedShortcut =
-	| { disabled: false; spec: string; matches: (data: string) => boolean }
-	| { disabled: true; spec: null; matches: (data: string) => false };
-
-export interface ResolvedShortcuts {
-	commentToggle: ResolvedShortcut;
-}
-
-export interface AnnotatedRow {
-	line: string;
-	selected: boolean;
-}
-
-export interface ItemBlock {
-	itemIndex: number;
-	lines: string[];
-}
-
-export type ListItem =
-	| { type: "option"; option: QuestionOption }
-	| { type: "comment-toggle"; option: QuestionOption }
-	| { type: "freeform"; option: QuestionOption };
-
-export interface RenderRowsParams {
-	options: QuestionOption[];
-	selectedIndex: number;
-	width: number;
-	allowFreeform: boolean;
-	allowComment?: boolean;
-	commentEnabled?: boolean;
-	maxRows?: number;
-	hideDescriptions?: boolean;
-}
-
-export type QuestionMode = "select" | "other" | "text";
-
-export interface CustomFactoryResult {
-	handleInput(data: string): void;
-	render(width: number): string[];
-	focused: boolean;
-}
-
-export type QuestionKeybindings = KeybindingsManager;
+// Shared type definitions for the question extension.
+
+import type { KeybindingsManager } from "@selesai/code";
+
+export interface RawQuestionOption {
+	value: string;
+	label: string;
+	description?: string;
+}
+
+export interface QuestionOption {
+	value: string;
+	label: string;
+	description?: string;
+}
+
+interface RawQuestionBase {
+	id?: string;
+	label?: string;
+	question: string;
+	context?: string;
+}
+
+export interface RawSelectQuestion extends RawQuestionBase {
+	type: "select";
+	options: RawQuestionOption[];
+	allowOther?: boolean;
+}
+
+export interface RawMultiSelectQuestion extends RawQuestionBase {
+	type: "multiselect";
+	options: RawQuestionOption[];
+	allowOther?: boolean;
+}
+
+export interface RawTextQuestion extends RawQuestionBase {
+	type: "text";
+}
+
+export type RawQuestion = RawSelectQuestion | RawMultiSelectQuestion | RawTextQuestion;
+
+export interface PreparedQuestionBase {
+	id: string;
+	label: string;
+	question: string;
+	context?: string;
+}
+
+export interface PreparedSelectQuestion extends PreparedQuestionBase {
+	type: "select";
+	options: QuestionOption[];
+	allowOther: boolean;
+}
+
+export interface PreparedMultiSelectQuestion extends PreparedQuestionBase {
+	type: "multiselect";
+	options: QuestionOption[];
+	allowOther: boolean;
+}
+
+export interface PreparedTextQuestion extends PreparedQuestionBase {
+	type: "text";
+	options: [];
+	allowOther: false;
+}
+
+export type PreparedQuestion = PreparedSelectQuestion | PreparedMultiSelectQuestion | PreparedTextQuestion;
+
+export type QuestionResponse =
+	| { kind: "selection"; values: string[]; otherText?: string }
+	| { kind: "text"; text: string };
+
+export type QuestionAnswer =
+	| { id: string; status: "answered"; response: QuestionResponse }
+	| { id: string; status: "skipped" | "unanswered" };
+
+export type QuestionToolResult =
+	| { status: "submitted"; answers: QuestionAnswer[] }
+	| { status: "cancelled"; reason: "user" };
+
+export interface DraftQuestionState {
+	status: "answered" | "skipped";
+	response?: QuestionResponse;
+}
+
+// ---------------------------------------------------------------------------
+// Legacy list internals.
+// ---------------------------------------------------------------------------
+
+export type ResolvedShortcut =
+	| { disabled: false; spec: string; matches: (data: string) => boolean }
+	| { disabled: true; spec: null; matches: (data: string) => false };
+
+export interface ResolvedShortcuts {
+	commentToggle: ResolvedShortcut;
+}
+
+export interface AnnotatedRow {
+	line: string;
+	selected: boolean;
+}
+
+export interface ItemBlock {
+	itemIndex: number;
+	lines: string[];
+}
+
+export type ListItem =
+	| { type: "option"; option: QuestionOption }
+	| { type: "comment-toggle"; option: QuestionOption }
+	| { type: "freeform"; option: QuestionOption };
+
+export interface RenderRowsParams {
+	options: QuestionOption[];
+	selectedIndex: number;
+	width: number;
+	allowFreeform: boolean;
+	allowComment?: boolean;
+	commentEnabled?: boolean;
+	maxRows?: number;
+	hideDescriptions?: boolean;
+}
+
+export type QuestionMode = "select" | "other" | "text";
+
+export interface CustomFactoryResult {
+	handleInput(data: string): void;
+	render(width: number): string[];
+	focused: boolean;
+}
+
+export type QuestionKeybindings = KeybindingsManager;

package/dist/extensions/test-resolve-hook-impl.mjs

@@ -1,7 +1,7 @@
-export async function resolve(specifier, context, next) {
-	if (specifier === "@selesai/code" || specifier.startsWith("@selesai/code/")) {
-		const url = new URL("../../src/index.ts", import.meta.url);
-		return next(url.href, context);
-	}
-	return next(specifier, context);
+export async function resolve(specifier, context, next) {
+	if (specifier === "@selesai/code" || specifier.startsWith("@selesai/code/")) {
+		const url = new URL("../../src/index.ts", import.meta.url);
+		return next(url.href, context);
+	}
+	return next(specifier, context);
 }

package/dist/extensions/test-resolve-hook.mjs

@@ -1,7 +1,7 @@
-// Register ESM resolve hook to rewrite "@selesai/code" -> this repo's src.
-// Node 24 loads .ts natively (no jiti), so we need a real resolve hook.
-// Usage: node --import ./src/extensions/test-resolve-hook.mjs --test <file>
-import { register } from "node:module";
-import { pathToFileURL } from "node:url";
-
+// Register ESM resolve hook to rewrite "@selesai/code" -> this repo's src.
+// Node 24 loads .ts natively (no jiti), so we need a real resolve hook.
+// Usage: node --import ./src/extensions/test-resolve-hook.mjs --test <file>
+import { register } from "node:module";
+import { pathToFileURL } from "node:url";
+
 register(pathToFileURL("./src/extensions/test-resolve-hook-impl.mjs").href);

package/dist/extensions/tokenin-onboarding.ts

@@ -16,6 +16,7 @@ import type { AuthStorage, ExtensionAPI, ExtensionCommandContext, ExtensionConte
 
 export const TOKEN_IN_PROVIDER = "tokenin";
 export const TOKEN_IN_DASHBOARD_URL = "https://token.selesai.in/dashboard/tokens";
+export const TOKEN_IN_DEFAULT_BASE_URL = "https://lite.andlet.me/v1";
 export const ONBOARDING_MARKER_NAME = ".tokenInOnboardingComplete";
 export const PLACEHOLDER_API_KEY = "sk-xxx";
 
@@ -239,6 +240,93 @@ async function promptForToken(ctx: ExtensionContext): Promise<string | undefined
 	return ctx.ui.input("Paste your Token-In token", "sk-...");
 }
 
+// ---------------------------------------------------------------------------
+// /tokenin usage — LiteLLM /key/info quota lookup
+// ---------------------------------------------------------------------------
+
+export interface TokenInUsage {
+	spend: number;
+	maxBudget?: number;
+	budgetResetAt?: string;
+}
+
+/** Parse a LiteLLM /key/info payload into spend/budget/reset. Exported for tests. */
+export function parseTokenInUsage(payload: unknown): TokenInUsage | undefined {
+	if (!payload || typeof payload !== "object" || Array.isArray(payload)) return undefined;
+	const obj = payload as Record<string, unknown>;
+	const info =
+		obj.info && typeof obj.info === "object" && !Array.isArray(obj.info)
+			? (obj.info as Record<string, unknown>)
+			: obj;
+	const spend = Number(info.spend);
+	if (!Number.isFinite(spend) || spend < 0) return undefined;
+	const usage: TokenInUsage = { spend };
+	const maxBudget = Number(info.max_budget);
+	if (Number.isFinite(maxBudget) && maxBudget > 0) usage.maxBudget = maxBudget;
+	if (typeof info.budget_reset_at === "string") usage.budgetResetAt = info.budget_reset_at;
+	return usage;
+}
+
+/** Format usage as a boxed multi-line display with a progress bar. Exported for tests. */
+export function formatTokenInUsage(usage: TokenInUsage): string {
+	const money = (v: number) => `$${v.toFixed(2)}`;
+	const pct = usage.maxBudget !== undefined ? (usage.spend / usage.maxBudget) * 100 : 0;
+	const barWidth = 40;
+	const filled = usage.maxBudget !== undefined ? Math.min(barWidth, Math.round((pct / 100) * barWidth)) : 0;
+	const bar = "█".repeat(filled) + "░".repeat(barWidth - filled);
+	const pctText = usage.maxBudget !== undefined ? `${pct.toFixed(1)}%` : "—";
+
+	let summary = `Spent ${money(usage.spend)}`;
+	if (usage.maxBudget !== undefined) {
+		summary += ` of ${money(usage.maxBudget)} budget (${pct.toFixed(1)}%)`;
+	} else {
+		summary += " (no budget limit set)";
+	}
+
+	let footer = usage.maxBudget !== undefined ? `${money(Math.max(0, usage.maxBudget - usage.spend))} remaining` : "";
+	if (usage.budgetResetAt) {
+		const d = new Date(usage.budgetResetAt);
+		if (!Number.isNaN(d.getTime())) {
+			const ms = d.getTime() - Date.now();
+			if (ms > 0) {
+				const days = Math.floor(ms / 86_400_000);
+				const hours = Math.floor((ms % 86_400_000) / 3_600_000);
+				const reset = days > 0 ? `${days}d ${hours}h` : `${Math.max(1, Math.ceil(ms / 3_600_000))}h`;
+				footer += footer ? ` · resets in ${reset}` : `resets in ${reset}`;
+			}
+		}
+	}
+
+	const W = 50; // total box width
+	const inner = W - 4; // content width between │ padding
+	const pad = (s: string) => s + " ".repeat(Math.max(0, inner - s.length));
+	const top = `╭─ Token-In usage ${"─".repeat(W - 19)}╮`;
+	const bottom = `╰${"─".repeat(W - 2)}╯`;
+	return [
+		top,
+		`│ ${pad(summary)} │`,
+		`│ ${bar} ${pctText.padStart(5)} │`,
+		`│ ${pad(footer)} │`,
+		bottom,
+	].join("\n");
+}
+
+/** Query LiteLLM /key/info at the proxy root with the account key. Exported for tests. */
+export async function fetchTokenInUsage(apiKey: string, baseUrl: string): Promise<TokenInUsage> {
+	const root = baseUrl.replace(/\/v1\/?$/, "");
+	const response = await fetch(`${root}/key/info`, {
+		headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+		signal: AbortSignal.timeout(5_000),
+	});
+	if (!response.ok) {
+		throw new Error(`Token-In usage endpoint returned HTTP ${response.status}`);
+	}
+	const payload: unknown = await response.json();
+	const usage = parseTokenInUsage(payload);
+	if (!usage) throw new Error("Token-In usage endpoint returned an unexpected payload");
+	return usage;
+}
+
 // ---------------------------------------------------------------------------
 // /tokenin subcommand handlers
 // ---------------------------------------------------------------------------
@@ -327,6 +415,25 @@ async function handleTokenInSwitch(ctx: ExtensionCommandContext): Promise<void>
 	ctx.ui.notify(`Switched to account ${account.label}.`, "info");
 }
 
+async function handleTokenInUsage(ctx: ExtensionCommandContext): Promise<void> {
+	const apiKey = getStoredTokenInApiKey(ctx.modelRegistry.authStorage);
+	if (!apiKey) {
+		ctx.ui.notify("No active Token-In token. Use /tokenin add first.", "warning");
+		return;
+	}
+	const auth = readTokenInAuth();
+	const activeId = getActiveTokenInAccountId(ctx.modelRegistry.authStorage);
+	const account = auth.accounts.find((a) => a.id === activeId);
+	const baseUrl = account?.baseUrl ?? getTokenInBaseUrl(readModelsJson()) ?? TOKEN_IN_DEFAULT_BASE_URL;
+	try {
+		const usage = await fetchTokenInUsage(apiKey, baseUrl);
+		ctx.ui.notify(formatTokenInUsage(usage), "info");
+	} catch (err) {
+		const message = err instanceof Error ? err.message : String(err);
+		ctx.ui.notify(`Could not fetch Token-In usage: ${message}`, "warning");
+	}
+}
+
 async function handleTokenInRemove(ctx: ExtensionCommandContext): Promise<void> {
 	if (!ctx.hasUI) {
 		ctx.ui.notify("This command requires interactive mode.", "warning");
@@ -441,12 +548,13 @@ export default function tokenInOnboardingExtension(pi: ExtensionAPI): void {
 	});
 
 	pi.registerCommand("tokenin", {
-		description: "Manage TokenIN accounts: /tokenin add|switch|remove",
+		description: "Manage TokenIN accounts: /tokenin add|switch|remove|usage",
 		getArgumentCompletions: (prefix: string) => {
 			const subcommands = [
 				{ name: "add", description: "Add a TokenIN account" },
 				{ name: "switch", description: "Switch active account" },
 				{ name: "remove", description: "Remove a saved account" },
+				{ name: "usage", description: "Show Token-IN quota usage" },
 			];
 			const lowerPrefix = prefix.toLowerCase();
 			const items = subcommands.filter((s) => s.name.toLowerCase().startsWith(lowerPrefix));
@@ -463,8 +571,10 @@ export default function tokenInOnboardingExtension(pi: ExtensionAPI): void {
 				await handleTokenInSwitch(ctx);
 			} else if (sub === "remove") {
 				await handleTokenInRemove(ctx);
+			} else if (sub === "usage") {
+				await handleTokenInUsage(ctx);
 			} else {
-				ctx.ui.notify("Usage: /tokenin add|switch|remove", "info");
+				ctx.ui.notify("Usage: /tokenin add|switch|remove|usage", "info");
 			}
 		},
 	});