@praeviso/code-env-switch 0.1.2 → 0.1.4

package/bin/statusline/index.js

@@ -1,390 +1,25 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.buildStatuslineResult = buildStatuslineResult;
-/**
- * Statusline builder
- */
-const fs = require("fs");
-const path = require("path");
-const child_process_1 = require("child_process");
-const constants_1 = require("../constants");
 const type_1 = require("../profile/type");
 const usage_1 = require("../usage");
-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, colorCode) {
-    if (!COLOR_ENABLED)
-        return text;
-    return `\x1b[${colorCode}m${text}${ANSI_RESET}`;
-}
-function dim(text) {
-    return colorize(text, "2");
-}
-function getCwdSegment(cwd) {
-    if (!cwd)
-        return null;
-    const base = path.basename(cwd) || cwd;
-    const segment = `${ICON_CWD} ${base}`;
-    return dim(segment);
-}
-function isRecord(value) {
-    return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-function readStdinJson() {
-    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;
-    }
-    catch {
-        return null;
-    }
-}
-function firstNonEmpty(...values) {
-    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) {
-    if (value === null || value === undefined || value === "")
-        return null;
-    const num = Number(value);
-    if (!Number.isFinite(num))
-        return null;
-    return num;
-}
-function firstNumber(...values) {
-    for (const value of values) {
-        const num = coerceNumber(value);
-        if (num !== null)
-            return num;
-    }
-    return null;
-}
-function normalizeTypeValue(value) {
-    if (!value)
-        return null;
-    const normalized = (0, type_1.normalizeType)(value);
-    if (normalized)
-        return normalized;
-    const trimmed = String(value).trim();
-    return trimmed ? trimmed : null;
-}
-function detectTypeFromEnv() {
-    const matches = constants_1.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) {
-    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) {
-    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) {
-    if (!input || !input.model_provider)
-        return null;
-    const provider = String(input.model_provider).trim();
-    return provider ? provider : null;
-}
-function getInputProfile(input) {
-    if (!input || !isRecord(input.profile))
-        return null;
-    return input.profile;
-}
-function getInputUsage(input) {
-    var _a, _b, _c, _d;
-    if (!input)
-        return null;
-    if (isRecord(input.usage)) {
-        return input.usage;
-    }
-    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;
-        return {
-            todayTokens: (_a = firstNumber(record.todayTokens, record.today, record.today_tokens, record.daily, record.daily_tokens)) !== null && _a !== void 0 ? _a : null,
-            totalTokens: (_b = firstNumber(record.totalTokens, record.total, record.total_tokens)) !== null && _b !== void 0 ? _b : null,
-            inputTokens: (_c = firstNumber(record.inputTokens, record.input, record.input_tokens)) !== null && _c !== void 0 ? _c : null,
-            outputTokens: (_d = firstNumber(record.outputTokens, record.output, record.output_tokens)) !== null && _d !== void 0 ? _d : null,
-        };
-    }
-    return null;
-}
-function getContextUsedTokens(input) {
-    if (!input)
-        return null;
-    return coerceNumber(input.context_window_used_tokens);
-}
-function getContextLeftPercent(input, type) {
-    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) {
-    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) {
-    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) {
-    if (!cwd)
-        return null;
-    const result = (0, child_process_1.spawnSync)("git", ["-C", cwd, "status", "--porcelain=v2", "-b"], {
-        encoding: "utf8",
-        stdio: ["ignore", "pipe", "ignore"],
-    });
-    if (result.status !== 0 || !result.stdout)
-        return null;
-    const status = {
-        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) {
-    if (!status || !status.branch)
-        return null;
-    const meta = [];
-    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, configPath, type, profileKey, profileName, syncUsage) {
-    try {
-        const normalized = (0, type_1.normalizeType)(type || "");
-        if (!normalized || (!profileKey && !profileName))
-            return null;
-        const totals = (0, usage_1.readUsageTotalsIndex)(config, configPath, syncUsage);
-        if (!totals)
-            return null;
-        const usage = (0, usage_1.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) {
-    var _a;
-    if (!usage)
-        return null;
-    const today = (_a = usage.todayTokens) !== null && _a !== void 0 ? _a : (usage.inputTokens !== null || usage.outputTokens !== null
-        ? (usage.inputTokens || 0) + (usage.outputTokens || 0)
-        : usage.totalTokens);
-    if (today === null)
-        return null;
-    const text = `Today ${(0, usage_1.formatTokenCount)(today)}`;
-    return colorize(`${ICON_USAGE} ${text}`, "33");
-}
-function formatModelSegment(model, provider) {
-    if (!model)
-        return null;
-    const providerLabel = provider ? `${provider}:${model}` : model;
-    return colorize(`${ICON_MODEL} ${providerLabel}`, "35");
-}
-function formatProfileSegment(type, profileKey, profileName) {
-    const name = profileName || profileKey;
-    if (!name)
-        return null;
-    const label = type ? `${type}:${name}` : name;
-    return colorize(`${ICON_PROFILE} ${label}`, "37");
-}
-function formatContextSegment(contextLeft) {
-    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) {
-    if (usedTokens === null)
-        return null;
-    return colorize(`${ICON_CONTEXT} ${(0, usage_1.formatTokenCount)(usedTokens)} used`, "36");
-}
-function formatModeSegment(reviewMode) {
-    if (!reviewMode)
-        return null;
-    return colorize(`${ICON_REVIEW} review`, "34");
-}
+const debug_1 = require("./debug");
+const format_1 = require("./format");
+const git_1 = require("./git");
+const input_1 = require("./input");
+const usage_2 = require("./usage");
+const utils_1 = require("./utils");
 function buildStatuslineResult(args, config, configPath) {
-    const stdinInput = readStdinJson();
-    const inputProfile = getInputProfile(stdinInput);
-    let typeCandidate = firstNonEmpty(args.type, process.env.CODE_ENV_TYPE, inputProfile ? inputProfile.type : null, stdinInput ? stdinInput.type : null);
+    const stdinInput = (0, input_1.readStdinJson)();
+    const inputProfile = (0, input_1.getInputProfile)(stdinInput);
+    let typeCandidate = (0, utils_1.firstNonEmpty)(args.type, process.env.CODE_ENV_TYPE, inputProfile ? inputProfile.type : null, stdinInput ? stdinInput.type : null);
     if (!typeCandidate) {
-        typeCandidate = detectTypeFromEnv();
+        typeCandidate = (0, input_1.detectTypeFromEnv)();
     }
-    let type = normalizeTypeValue(typeCandidate);
-    const envProfile = resolveEnvProfile(type);
-    let profileKey = firstNonEmpty(args.profileKey, envProfile.key, inputProfile ? inputProfile.key : null);
-    let profileName = firstNonEmpty(args.profileName, envProfile.name, inputProfile ? inputProfile.name : null);
+    let type = (0, input_1.normalizeTypeValue)(typeCandidate);
+    const envProfile = (0, input_1.resolveEnvProfile)(type);
+    const profileKey = (0, utils_1.firstNonEmpty)(args.profileKey, envProfile.key, inputProfile ? inputProfile.key : null);
+    let profileName = (0, utils_1.firstNonEmpty)(args.profileName, envProfile.name, inputProfile ? inputProfile.name : null);
     if (profileKey && !profileName && config.profiles && config.profiles[profileKey]) {
         const profile = config.profiles[profileKey];
         profileName = (0, type_1.getProfileDisplayName)(profileKey, profile, type || undefined);
@@ -400,43 +35,81 @@ function buildStatuslineResult(args, config, configPath) {
         if (inferred)
             type = inferred;
     }
-    const cwd = firstNonEmpty(args.cwd, process.env.CODE_ENV_CWD, getWorkspaceDir(stdinInput), stdinInput ? stdinInput.cwd : null, process.cwd());
-    const model = firstNonEmpty(args.model, process.env.CODE_ENV_MODEL, getModelFromInput(stdinInput));
-    const modelProvider = firstNonEmpty(process.env.CODE_ENV_MODEL_PROVIDER, getModelProviderFromInput(stdinInput));
+    const cwd = (0, utils_1.firstNonEmpty)(args.cwd, process.env.CODE_ENV_CWD, (0, input_1.getWorkspaceDir)(stdinInput), stdinInput ? stdinInput.cwd : null, process.cwd());
+    const sessionId = (0, input_1.getSessionId)(stdinInput);
+    const stdinUsageTotals = (0, usage_2.getUsageTotalsFromInput)(stdinInput);
+    const usageType = (0, type_1.normalizeType)(type || "");
+    (0, debug_1.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) {
+        (0, usage_1.syncUsageFromStatuslineInput)(config, configPath, usageType, profileKey, profileName, sessionId, stdinUsageTotals, cwd);
+    }
+    const model = (0, utils_1.firstNonEmpty)(args.model, process.env.CODE_ENV_MODEL, (0, input_1.getModelFromInput)(stdinInput));
+    const modelProvider = (0, utils_1.firstNonEmpty)(process.env.CODE_ENV_MODEL_PROVIDER, (0, input_1.getModelProviderFromInput)(stdinInput));
     const usage = {
-        todayTokens: firstNumber(args.usageToday, process.env.CODE_ENV_USAGE_TODAY),
-        totalTokens: firstNumber(args.usageTotal, process.env.CODE_ENV_USAGE_TOTAL),
-        inputTokens: firstNumber(args.usageInput, process.env.CODE_ENV_USAGE_INPUT),
-        outputTokens: firstNumber(args.usageOutput, process.env.CODE_ENV_USAGE_OUTPUT),
+        todayTokens: (0, utils_1.firstNumber)(args.usageToday, process.env.CODE_ENV_USAGE_TODAY),
+        totalTokens: (0, utils_1.firstNumber)(args.usageTotal, process.env.CODE_ENV_USAGE_TOTAL),
+        inputTokens: (0, utils_1.firstNumber)(args.usageInput, process.env.CODE_ENV_USAGE_INPUT),
+        outputTokens: (0, utils_1.firstNumber)(args.usageOutput, process.env.CODE_ENV_USAGE_OUTPUT),
     };
     const hasExplicitUsage = usage.todayTokens !== null ||
         usage.totalTokens !== null ||
         usage.inputTokens !== null ||
         usage.outputTokens !== null;
+    const stdinUsage = (0, usage_2.normalizeInputUsage)((0, input_1.getInputUsage)(stdinInput));
     let finalUsage = hasExplicitUsage ? usage : null;
     if (!finalUsage) {
-        finalUsage = resolveUsageFromRecords(config, configPath, type, profileKey, profileName, args.syncUsage);
+        finalUsage = stdinUsage;
+    }
+    if (!finalUsage) {
+        finalUsage = (0, usage_2.resolveUsageFromRecords)(config, configPath, type, profileKey, profileName, args.syncUsage);
     }
-    let gitStatus = getGitStatus(cwd);
+    let gitStatus = (0, git_1.getGitStatus)(cwd);
     if (!gitStatus) {
-        gitStatus = getGitStatusFromInput(stdinInput);
+        gitStatus = (0, input_1.getGitStatusFromInput)(stdinInput);
     }
     else {
-        const inputGit = getGitStatusFromInput(stdinInput);
+        const inputGit = (0, input_1.getGitStatusFromInput)(stdinInput);
         if (inputGit && (!gitStatus.branch || gitStatus.branch === "HEAD")) {
             gitStatus.branch = inputGit.branch;
         }
     }
-    const gitSegment = formatGitSegment(gitStatus);
-    const profileSegment = formatProfileSegment(type, profileKey, profileName);
-    const modelSegment = formatModelSegment(model, modelProvider);
-    const usageSegment = formatUsageSegment(finalUsage);
-    const contextLeft = getContextLeftPercent(stdinInput, type);
-    const contextSegment = formatContextSegment(contextLeft);
-    const contextUsedTokens = getContextUsedTokens(stdinInput);
-    const contextUsedSegment = contextSegment === null ? formatContextUsedSegment(contextUsedTokens) : null;
-    const modeSegment = formatModeSegment((stdinInput === null || stdinInput === void 0 ? void 0 : stdinInput.review_mode) === true);
-    const cwdSegment = getCwdSegment(cwd);
+    const gitSegment = (0, git_1.formatGitSegment)(gitStatus);
+    const profileSegment = (0, format_1.formatProfileSegment)(type, profileKey, profileName);
+    const modelSegment = (0, format_1.formatModelSegment)(model, modelProvider);
+    const usageSegment = (0, format_1.formatUsageSegment)(finalUsage);
+    const contextLeft = (0, input_1.getContextLeftPercent)(stdinInput, type);
+    const contextSegment = (0, format_1.formatContextSegment)(contextLeft);
+    const contextUsedTokens = (0, input_1.getContextUsedTokens)(stdinInput);
+    const contextUsedSegment = contextSegment === null ? (0, format_1.formatContextUsedSegment)(contextUsedTokens) : null;
+    const modeSegment = (0, format_1.formatModeSegment)((stdinInput === null || stdinInput === void 0 ? void 0 : stdinInput.review_mode) === true);
+    const cwdSegment = (0, format_1.getCwdSegment)(cwd);
     const segments = [];
     if (gitSegment)
         segments.push(gitSegment);