@praeviso/code-env-switch 0.1.2 → 0.1.4

package/src/statusline/index.ts

@@ -1,478 +1,38 @@
 /**
  * Statusline builder
  */
-import * as fs from "fs";
-import * as path from "path";
-import { spawnSync } from "child_process";
 import type { Config, StatuslineArgs } from "../types";
-import { DEFAULT_PROFILE_TYPES } from "../constants";
 import { normalizeType, inferProfileType, getProfileDisplayName } from "../profile/type";
+import { syncUsageFromStatuslineInput } from "../usage";
+import { appendStatuslineDebug } from "./debug";
 import {
-    formatTokenCount,
-    readUsageTotalsIndex,
-    resolveUsageTotalsForProfile,
-} from "../usage";
-
-interface StatuslineInputProfile {
-    key?: string;
-    name?: string;
-    type?: string;
-}
-
-interface StatuslineInputUsage {
-    todayTokens?: number;
-    totalTokens?: number;
-    inputTokens?: number;
-    outputTokens?: number;
-}
-
-interface StatuslineInputModel {
-    id?: string;
-    displayName?: string;
-    display_name?: string;
-}
-
-interface StatuslineInput {
-    cwd?: string;
-    type?: string;
-    profile?: StatuslineInputProfile;
-    model?: string | StatuslineInputModel;
-    model_provider?: string;
-    usage?: StatuslineInputUsage;
-    token_usage?: StatuslineInputUsage | number | Record<string, unknown>;
-    git_branch?: string;
-    task_running?: boolean;
-    review_mode?: boolean;
-    context_window_percent?: number;
-    context_window_used_tokens?: number;
-    workspace?: {
-        current_dir?: string;
-        project_dir?: string;
-    };
-    cost?: Record<string, unknown>;
-    version?: string;
-    output_style?: { name?: string };
-    session_id?: string;
-    transcript_path?: string;
-    hook_event_name?: string;
-}
-
-interface StatuslineUsage {
-    todayTokens: number | null;
-    totalTokens: number | null;
-    inputTokens: number | null;
-    outputTokens: number | null;
-}
-
-interface GitStatus {
-    branch: string | null;
-    ahead: number;
-    behind: number;
-    staged: number;
-    unstaged: number;
-    untracked: number;
-    conflicted: number;
-}
-
-const COLOR_ENABLED = !process.env.NO_COLOR && process.env.TERM !== "dumb";
-const ANSI_RESET = "\x1b[0m";
-const ICON_GIT = "⎇";
-const ICON_PROFILE = "👤";
-const ICON_MODEL = "⚙";
-const ICON_USAGE = "⚡";
-const ICON_CONTEXT = "🧠";
-const ICON_REVIEW = "📝";
-const ICON_CWD = "📁";
-
-function colorize(text: string, colorCode: string): string {
-    if (!COLOR_ENABLED) return text;
-    return `\x1b[${colorCode}m${text}${ANSI_RESET}`;
-}
-
-function dim(text: string): string {
-    return colorize(text, "2");
-}
-
-function getCwdSegment(cwd: string): string | null {
-    if (!cwd) return null;
-    const base = path.basename(cwd) || cwd;
-    const segment = `${ICON_CWD} ${base}`;
-    return dim(segment);
-}
-
-export interface StatuslineJson {
-    cwd: string;
-    type: string | null;
-    profile: { key: string | null; name: string | null };
-    model: string | null;
-    usage: StatuslineUsage | null;
-    git: GitStatus | null;
-}
-
-export interface StatuslineResult {
-    text: string;
-    json: StatuslineJson;
-}
-
-function isRecord(value: unknown): value is Record<string, unknown> {
-    return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-
-function readStdinJson(): StatuslineInput | null {
-    if (process.stdin.isTTY) return null;
-    try {
-        const raw = fs.readFileSync(0, "utf8");
-        const trimmed = raw.trim();
-        if (!trimmed) return null;
-        const parsed = JSON.parse(trimmed);
-        if (!isRecord(parsed)) return null;
-        return parsed as StatuslineInput;
-    } catch {
-        return null;
-    }
-}
-
-function firstNonEmpty(...values: Array<string | null | undefined>): string | null {
-    for (const value of values) {
-        if (value === null || value === undefined) continue;
-        const text = String(value).trim();
-        if (text) return text;
-    }
-    return null;
-}
-
-function coerceNumber(value: unknown): number | null {
-    if (value === null || value === undefined || value === "") return null;
-    const num = Number(value);
-    if (!Number.isFinite(num)) return null;
-    return num;
-}
-
-function firstNumber(...values: Array<unknown>): number | null {
-    for (const value of values) {
-        const num = coerceNumber(value);
-        if (num !== null) return num;
-    }
-    return null;
-}
-
-function normalizeTypeValue(value: string | null): string | null {
-    if (!value) return null;
-    const normalized = normalizeType(value);
-    if (normalized) return normalized;
-    const trimmed = String(value).trim();
-    return trimmed ? trimmed : null;
-}
-
-function detectTypeFromEnv(): string | null {
-    const matches = DEFAULT_PROFILE_TYPES.filter((type) => {
-        const suffix = type.toUpperCase();
-        return (
-            process.env[`CODE_ENV_PROFILE_KEY_${suffix}`] ||
-            process.env[`CODE_ENV_PROFILE_NAME_${suffix}`]
-        );
-    });
-    if (matches.length === 1) return matches[0];
-    return null;
-}
-
-function resolveEnvProfile(type: string | null): { key: string | null; name: string | null } {
-    const genericKey = process.env.CODE_ENV_PROFILE_KEY || null;
-    const genericName = process.env.CODE_ENV_PROFILE_NAME || null;
-    if (!type) {
-        return { key: genericKey, name: genericName };
-    }
-    const suffix = type.toUpperCase();
-    const key = process.env[`CODE_ENV_PROFILE_KEY_${suffix}`] || genericKey;
-    const name = process.env[`CODE_ENV_PROFILE_NAME_${suffix}`] || genericName;
-    return { key: key || null, name: name || null };
-}
-
-function getModelFromInput(input: StatuslineInput | null): string | null {
-    if (!input) return null;
-    const raw = input.model;
-    if (!raw) return null;
-    if (typeof raw === "string") return raw;
-    if (isRecord(raw)) {
-        const displayName = raw.displayName || raw.display_name;
-        if (displayName) return String(displayName);
-        if (raw.id) return String(raw.id);
-    }
-    return null;
-}
-
-function getModelProviderFromInput(input: StatuslineInput | null): string | null {
-    if (!input || !input.model_provider) return null;
-    const provider = String(input.model_provider).trim();
-    return provider ? provider : null;
-}
-
-function getInputProfile(input: StatuslineInput | null): StatuslineInputProfile | null {
-    if (!input || !isRecord(input.profile)) return null;
-    return input.profile as StatuslineInputProfile;
-}
-
-function getInputUsage(input: StatuslineInput | null): StatuslineInputUsage | null {
-    if (!input) return null;
-    if (isRecord(input.usage)) {
-        return input.usage as StatuslineInputUsage;
-    }
-    const tokenUsage = input.token_usage;
-    if (tokenUsage === null || tokenUsage === undefined) return null;
-    if (typeof tokenUsage === "number") {
-        return {
-            todayTokens: null,
-            totalTokens: coerceNumber(tokenUsage),
-            inputTokens: null,
-            outputTokens: null,
-        };
-    }
-    if (isRecord(tokenUsage)) {
-        const record = tokenUsage as Record<string, unknown>;
-        return {
-            todayTokens:
-                firstNumber(
-                    record.todayTokens,
-                    record.today,
-                    record.today_tokens,
-                    record.daily,
-                    record.daily_tokens
-                ) ?? null,
-            totalTokens:
-                firstNumber(
-                    record.totalTokens,
-                    record.total,
-                    record.total_tokens
-                ) ?? null,
-            inputTokens:
-                firstNumber(
-                    record.inputTokens,
-                    record.input,
-                    record.input_tokens
-                ) ?? null,
-            outputTokens:
-                firstNumber(
-                    record.outputTokens,
-                    record.output,
-                    record.output_tokens
-                ) ?? null,
-        };
-    }
-    return null;
-}
-
-function getContextUsedTokens(input: StatuslineInput | null): number | null {
-    if (!input) return null;
-    return coerceNumber(input.context_window_used_tokens);
-}
-
-function getContextLeftPercent(
-    input: StatuslineInput | null,
-    type: string | null
-): number | null {
-    if (!input) return null;
-    const raw = coerceNumber(input.context_window_percent);
-    if (raw === null || raw < 0) return null;
-    const percent = raw <= 1 ? raw * 100 : raw;
-    if (percent > 100) return null;
-    const usedTokens = getContextUsedTokens(input);
-    const normalizedType = normalizeTypeValue(type);
-    // Prefer treating the percent as "remaining" for codex/claude and when usage is absent.
-    const preferRemaining =
-        normalizedType === "codex" ||
-        normalizedType === "claude" ||
-        usedTokens === null ||
-        (usedTokens <= 0 && percent >= 99);
-    const left = preferRemaining ? percent : 100 - percent;
-    return Math.max(0, Math.min(100, left));
-}
-
-function getWorkspaceDir(input: StatuslineInput | null): string | null {
-    if (!input || !isRecord(input.workspace)) return null;
-    const currentDir = input.workspace.current_dir;
-    if (currentDir) {
-        const trimmed = String(currentDir).trim();
-        if (trimmed) return trimmed;
-    }
-    const projectDir = input.workspace.project_dir;
-    if (!projectDir) return null;
-    const trimmed = String(projectDir).trim();
-    return trimmed ? trimmed : null;
-}
-
-function getGitStatusFromInput(
-    input: StatuslineInput | null
-): GitStatus | null {
-    if (!input || !input.git_branch) return null;
-    const branch = String(input.git_branch).trim();
-    if (!branch) return null;
-    return {
-        branch,
-        ahead: 0,
-        behind: 0,
-        staged: 0,
-        unstaged: 0,
-        untracked: 0,
-        conflicted: 0,
-    };
-}
-
-function getGitStatus(cwd: string): GitStatus | null {
-    if (!cwd) return null;
-    const result = spawnSync("git", ["-C", cwd, "status", "--porcelain=v2", "-b"], {
-        encoding: "utf8",
-        stdio: ["ignore", "pipe", "ignore"],
-    });
-    if (result.status !== 0 || !result.stdout) return null;
-    const status: GitStatus = {
-        branch: null,
-        ahead: 0,
-        behind: 0,
-        staged: 0,
-        unstaged: 0,
-        untracked: 0,
-        conflicted: 0,
-    };
-
-    const lines = result.stdout.split(/\r?\n/);
-    for (const line of lines) {
-        if (!line) continue;
-        if (line.startsWith("# branch.head ")) {
-            status.branch = line.slice("# branch.head ".length).trim();
-            continue;
-        }
-        if (line.startsWith("# branch.ab ")) {
-            const parts = line
-                .slice("# branch.ab ".length)
-                .trim()
-                .split(/\s+/);
-            for (const part of parts) {
-                if (part.startsWith("+")) status.ahead = Number(part.slice(1)) || 0;
-                if (part.startsWith("-")) status.behind = Number(part.slice(1)) || 0;
-            }
-            continue;
-        }
-        if (line.startsWith("? ")) {
-            status.untracked += 1;
-            continue;
-        }
-        if (line.startsWith("u ")) {
-            status.conflicted += 1;
-            continue;
-        }
-        if (line.startsWith("1 ") || line.startsWith("2 ")) {
-            const parts = line.split(/\s+/);
-            const xy = parts[1] || "";
-            const staged = xy[0];
-            const unstaged = xy[1];
-            if (staged && staged !== ".") status.staged += 1;
-            if (unstaged && unstaged !== ".") status.unstaged += 1;
-            continue;
-        }
-    }
-
-    if (!status.branch) {
-        status.branch = "HEAD";
-    }
-    return status;
-}
-
-function formatGitSegment(status: GitStatus | null): string | null {
-    if (!status || !status.branch) return null;
-    const meta: string[] = [];
-    const dirtyCount = status.staged + status.unstaged + status.untracked;
-    if (status.ahead > 0) meta.push(`↑${status.ahead}`);
-    if (status.behind > 0) meta.push(`↓${status.behind}`);
-    if (status.conflicted > 0) meta.push(`✖${status.conflicted}`);
-    if (dirtyCount > 0) meta.push(`+${dirtyCount}`);
-    const suffix = meta.length > 0 ? ` [${meta.join("")}]` : "";
-    const text = `${ICON_GIT} ${status.branch}${suffix}`;
-    const hasConflicts = status.conflicted > 0;
-    const isDirty = dirtyCount > 0;
-    if (hasConflicts) return colorize(text, "31");
-    if (isDirty) return colorize(text, "33");
-    if (status.ahead > 0 || status.behind > 0) return colorize(text, "36");
-    return colorize(text, "32");
-}
-
-function resolveUsageFromRecords(
-    config: Config,
-    configPath: string | null,
-    type: string | null,
-    profileKey: string | null,
-    profileName: string | null,
-    syncUsage: boolean
-): StatuslineUsage | null {
-    try {
-        const normalized = normalizeType(type || "");
-        if (!normalized || (!profileKey && !profileName)) return null;
-        const totals = readUsageTotalsIndex(config, configPath, syncUsage);
-        if (!totals) return null;
-        const usage = resolveUsageTotalsForProfile(
-            totals,
-            normalized,
-            profileKey,
-            profileName
-        );
-        if (!usage) return null;
-        return {
-            todayTokens: usage.today,
-            totalTokens: usage.total,
-            inputTokens: null,
-            outputTokens: null,
-        };
-    } catch {
-        return null;
-    }
-}
-
-function formatUsageSegment(usage: StatuslineUsage | null): string | null {
-    if (!usage) return null;
-    const today =
-        usage.todayTokens ??
-        (usage.inputTokens !== null || usage.outputTokens !== null
-            ? (usage.inputTokens || 0) + (usage.outputTokens || 0)
-            : usage.totalTokens);
-    if (today === null) return null;
-    const text = `Today ${formatTokenCount(today)}`;
-    return colorize(`${ICON_USAGE} ${text}`, "33");
-}
-
-function formatModelSegment(
-    model: string | null,
-    provider: string | null
-): string | null {
-    if (!model) return null;
-    const providerLabel = provider ? `${provider}:${model}` : model;
-    return colorize(`${ICON_MODEL} ${providerLabel}`, "35");
-}
-
-function formatProfileSegment(
-    type: string | null,
-    profileKey: string | null,
-    profileName: string | null
-): string | null {
-    const name = profileName || profileKey;
-    if (!name) return null;
-    const label = type ? `${type}:${name}` : name;
-    return colorize(`${ICON_PROFILE} ${label}`, "37");
-}
-
-function formatContextSegment(contextLeft: number | null): string | null {
-    if (contextLeft === null) return null;
-    const left = Math.max(0, Math.min(100, Math.round(contextLeft)));
-    return colorize(`${ICON_CONTEXT} ${left}% left`, "36");
-}
-
-function formatContextUsedSegment(usedTokens: number | null): string | null {
-    if (usedTokens === null) return null;
-    return colorize(`${ICON_CONTEXT} ${formatTokenCount(usedTokens)} used`, "36");
-}
-
-function formatModeSegment(reviewMode: boolean): string | null {
-    if (!reviewMode) return null;
-    return colorize(`${ICON_REVIEW} review`, "34");
-}
+    formatContextSegment,
+    formatContextUsedSegment,
+    formatModeSegment,
+    formatModelSegment,
+    formatProfileSegment,
+    formatUsageSegment,
+    getCwdSegment,
+} from "./format";
+import { formatGitSegment, getGitStatus } from "./git";
+import {
+    detectTypeFromEnv,
+    getContextLeftPercent,
+    getContextUsedTokens,
+    getGitStatusFromInput,
+    getInputProfile,
+    getInputUsage,
+    getModelFromInput,
+    getModelProviderFromInput,
+    getSessionId,
+    getWorkspaceDir,
+    normalizeTypeValue,
+    readStdinJson,
+    resolveEnvProfile,
+} from "./input";
+import { getUsageTotalsFromInput, normalizeInputUsage, resolveUsageFromRecords } from "./usage";
+import { firstNonEmpty, firstNumber } from "./utils";
+import type { StatuslineResult, StatuslineUsage } from "./types";
 
 export function buildStatuslineResult(
     args: StatuslineArgs,
@@ -496,7 +56,7 @@ export function buildStatuslineResult(
     let type = normalizeTypeValue(typeCandidate);
     const envProfile = resolveEnvProfile(type);
 
-    let profileKey = firstNonEmpty(
+    const profileKey = firstNonEmpty(
         args.profileKey,
         envProfile.key,
         inputProfile ? inputProfile.key : null
@@ -530,6 +90,50 @@ export function buildStatuslineResult(
         process.cwd()
     )!;
 
+    const sessionId = getSessionId(stdinInput);
+    const stdinUsageTotals = getUsageTotalsFromInput(stdinInput);
+    const usageType = normalizeType(type || "");
+    appendStatuslineDebug(configPath, {
+        timestamp: new Date().toISOString(),
+        typeCandidate,
+        resolvedType: type,
+        usageType,
+        profile: { key: profileKey, name: profileName },
+        sessionId,
+        stdinUsageTotals,
+        cwd,
+        args: {
+            type: args.type,
+            profileKey: args.profileKey,
+            profileName: args.profileName,
+            usageToday: args.usageToday,
+            usageTotal: args.usageTotal,
+            usageInput: args.usageInput,
+            usageOutput: args.usageOutput,
+            syncUsage: args.syncUsage,
+        },
+        env: {
+            CODE_ENV_TYPE: process.env.CODE_ENV_TYPE,
+            CODE_ENV_PROFILE_KEY: process.env.CODE_ENV_PROFILE_KEY,
+            CODE_ENV_PROFILE_NAME: process.env.CODE_ENV_PROFILE_NAME,
+            CODE_ENV_CWD: process.env.CODE_ENV_CWD,
+            CODE_ENV_STATUSLINE: process.env.CODE_ENV_STATUSLINE,
+        },
+        input: stdinInput,
+    });
+    if (args.syncUsage && sessionId && stdinUsageTotals) {
+        syncUsageFromStatuslineInput(
+            config,
+            configPath,
+            usageType,
+            profileKey,
+            profileName,
+            sessionId,
+            stdinUsageTotals,
+            cwd
+        );
+    }
+
     const model = firstNonEmpty(
         args.model,
         process.env.CODE_ENV_MODEL,
@@ -565,7 +169,12 @@ export function buildStatuslineResult(
         usage.inputTokens !== null ||
         usage.outputTokens !== null;
 
+    const stdinUsage = normalizeInputUsage(getInputUsage(stdinInput));
+
     let finalUsage: StatuslineUsage | null = hasExplicitUsage ? usage : null;
+    if (!finalUsage) {
+        finalUsage = stdinUsage;
+    }
     if (!finalUsage) {
         finalUsage = resolveUsageFromRecords(
             config,
@@ -624,3 +233,17 @@ export function buildStatuslineResult(
         },
     };
 }
+
+export type {
+    GitStatus,
+    StatuslineInput,
+    StatuslineInputContextWindow,
+    StatuslineInputContextWindowUsage,
+    StatuslineInputModel,
+    StatuslineInputProfile,
+    StatuslineInputUsage,
+    StatuslineJson,
+    StatuslineResult,
+    StatuslineUsage,
+    StatuslineUsageTotals,
+} from "./types";