pinglet-cli 0.6.2 → 0.8.0

package/dist/hud/speed.js

@@ -0,0 +1,109 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getOutputSpeed = getOutputSpeed;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const crypto_1 = require("crypto");
+const config_1 = require("../config");
+const storage_1 = require("../storage");
+/**
+ * 출력 속도(tok/s) — claude-hud의 speed-tracker와 같은 방식.
+ * 이전 tick의 output_tokens와 시각을 transcript 경로별 캐시에 남겨 두고 증가분으로 계산한다.
+ *   - 2초를 넘긴 창은 버린다 (유휴 뒤 첫 tick이 거대한 속도로 보이지 않게)
+ *   - 0.5초 미만 창은 보고하지 않는다 (렌더가 초당 여러 번 돌면 노이즈가 커진다)
+ *   - output_tokens가 없으면 transcript 파일 크기 증가(≈4바이트/토큰)로 추정한다
+ */
+const SPEED_WINDOW_MS = 2000;
+const MIN_DELTA_MS = 500;
+const BYTES_PER_TOKEN = 4;
+const CACHE_DIR = path.join(config_1.PINGLET_DIR, "hud-cache", "speed");
+function cachePath(transcriptPath, suffix) {
+    const hash = (0, crypto_1.createHash)("sha256").update(path.resolve(transcriptPath)).digest("hex").slice(0, 32);
+    return path.join(CACHE_DIR, `${hash}${suffix}.json`);
+}
+function readSample(file) {
+    try {
+        const parsed = JSON.parse(fs.readFileSync(file, "utf8"));
+        return typeof parsed?.value === "number" && typeof parsed.timestamp === "number" ? parsed : null;
+    }
+    catch {
+        return null;
+    }
+}
+function writeSample(file, sample) {
+    try {
+        (0, storage_1.atomicWrite)(file, JSON.stringify(sample));
+    }
+    catch {
+        // 캐시 실패는 표시에 영향을 주면 안 된다.
+    }
+}
+function rate(file, value, now) {
+    const previous = readSample(file);
+    if (!previous || value < previous.value) {
+        writeSample(file, { value, timestamp: now });
+        return null;
+    }
+    const deltaValue = value - previous.value;
+    const deltaMs = now - previous.timestamp;
+    if (deltaMs > SPEED_WINDOW_MS || deltaValue <= 0) {
+        writeSample(file, { value, timestamp: now });
+        return null;
+    }
+    if (deltaMs < MIN_DELTA_MS)
+        return null;
+    writeSample(file, { value, timestamp: now });
+    return deltaValue / (deltaMs / 1000);
+}
+function getOutputSpeed(payload, now = Date.now()) {
+    const transcriptPath = typeof payload.transcript_path === "string" ? payload.transcript_path.trim() : "";
+    if (!transcriptPath)
+        return null; // 세션 키가 없으면 다른 세션과 캐시가 섞인다.
+    const outputTokens = payload.context_window?.current_usage?.output_tokens;
+    if (typeof outputTokens === "number" && Number.isFinite(outputTokens)) {
+        return rate(cachePath(transcriptPath, ""), outputTokens, now);
+    }
+    try {
+        const stat = fs.statSync(transcriptPath);
+        if (!stat.isFile())
+            return null;
+        const tokens = rate(cachePath(transcriptPath, ".fs"), stat.size, now);
+        return tokens === null ? null : tokens / BYTES_PER_TOKEN;
+    }
+    catch {
+        return null;
+    }
+}

package/dist/hud/stdin.js

@@ -0,0 +1,134 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTotalTokens = getTotalTokens;
+exports.getContextPercent = getContextPercent;
+exports.getModelName = getModelName;
+exports.formatModelName = formatModelName;
+exports.resolveEffortLevel = resolveEffortLevel;
+exports.parseScopedWindows = parseScopedWindows;
+exports.getUsage = getUsage;
+exports.formatTokens = formatTokens;
+exports.formatUsd = formatUsd;
+const num = (v) => typeof v === "number" && Number.isFinite(v) ? v : null;
+function getTotalTokens(payload) {
+    const usage = payload.context_window?.current_usage;
+    return ((num(usage?.input_tokens) ?? 0) +
+        (num(usage?.cache_creation_input_tokens) ?? 0) +
+        (num(usage?.cache_read_input_tokens) ?? 0));
+}
+/**
+ * 컨텍스트 사용률. Claude Code가 주는 used_percentage를 우선하고(/context와 일치),
+ * 0이거나 없으면 토큰 합/창 크기로 계산한다 — 첫 응답 전에는 used_percentage가
+ * 0으로 오지만 current_usage에는 초기 컨텍스트(시스템 프롬프트 등)가 이미 있다.
+ */
+function getContextPercent(payload) {
+    const native = num(payload.context_window?.used_percentage);
+    if (native !== null && native > 0)
+        return Math.min(100, Math.max(0, Math.round(native)));
+    const size = num(payload.context_window?.context_window_size);
+    if (!size || size <= 0)
+        return 0;
+    return Math.min(100, Math.round((getTotalTokens(payload) / size) * 100));
+}
+function getModelName(payload) {
+    const displayName = payload.model?.display_name?.trim();
+    if (displayName)
+        return displayName;
+    return payload.model?.id?.trim() || "Unknown";
+}
+/**
+ *   full:    표시 이름 그대로 (예: "Opus 4.6 (1M context)")
+ *   compact: 컨텍스트 창 접미사 제거 (예: "Opus 4.6")
+ *   short:   접미사와 "Claude " 접두사 모두 제거
+ */
+function formatModelName(name, format) {
+    if (format === "full")
+        return name;
+    let result = name.replace(/\s*\([^)]*\bcontext\b[^)]*\)/i, "").trim();
+    if (format === "short")
+        result = result.replace(/^Claude\s+/i, "");
+    return result;
+}
+const EFFORT_SYMBOLS = {
+    low: "○",
+    medium: "◔",
+    high: "◑",
+    xhigh: "◕",
+    max: "●",
+};
+/**
+ * stdin의 effort(문자열 또는 `{ level }`)를 표시용으로 정리한다.
+ * ultracode는 stdin에 일반 레벨로 오므로 transcript 신호가 있으면 `ultracode(<level>)`로 감싼다.
+ */
+function resolveEffortLevel(effort, ultracodeActive) {
+    const raw = typeof effort === "string"
+        ? effort
+        : effort && typeof effort === "object" && typeof effort.level === "string" ? effort.level : "";
+    const level = raw.toLowerCase().trim().replace(/[\p{Cc}\p{Cf}]/gu, "").slice(0, 16);
+    if (!level)
+        return null;
+    const symbol = EFFORT_SYMBOLS[level] ?? "";
+    return ultracodeActive === true ? { level: `ultracode(${level})`, symbol } : { level, symbol };
+}
+function percent(v) {
+    const n = num(v);
+    return n === null ? null : Math.round(Math.min(100, Math.max(0, n)));
+}
+function resetAt(v) {
+    const n = num(v);
+    return n === null || n <= 0 ? null : new Date(n * 1000);
+}
+const SCOPED_MAX_WINDOWS = 8;
+/** `rate_limits.model_scoped` — 잘못된 항목은 버리고 개수·라벨 길이를 제한한다 (stdin은 신뢰하지 않는다). */
+function parseScopedWindows(modelScoped) {
+    if (!Array.isArray(modelScoped))
+        return [];
+    const windows = [];
+    for (const raw of modelScoped) {
+        if (windows.length >= SCOPED_MAX_WINDOWS)
+            break;
+        const entry = raw;
+        const label = typeof entry?.display_name === "string"
+            ? entry.display_name.replace(/[\p{Cc}\p{Cf}]/gu, "").replace(/\s+/g, " ").trim().slice(0, 64)
+            : "";
+        if (!label)
+            continue;
+        const utilization = entry?.utilization;
+        const value = utilization === null ? null : percent(utilization);
+        if (utilization !== null && value === null)
+            continue;
+        const resetRaw = entry?.resets_at;
+        const reset = typeof resetRaw === "string" && resetRaw.length <= 64 && !Number.isNaN(Date.parse(resetRaw))
+            ? new Date(resetRaw)
+            : null;
+        windows.push({ label, percent: value, resetAt: reset });
+    }
+    return windows;
+}
+function getUsage(payload) {
+    const limits = payload.rate_limits;
+    if (!limits)
+        return null;
+    const fiveHour = percent(limits.five_hour?.used_percentage);
+    const sevenDay = percent(limits.seven_day?.used_percentage);
+    const scopedWindows = parseScopedWindows(limits.model_scoped);
+    if (fiveHour === null && sevenDay === null && scopedWindows.length === 0)
+        return null;
+    return {
+        fiveHour,
+        sevenDay,
+        fiveHourResetAt: resetAt(limits.five_hour?.resets_at),
+        sevenDayResetAt: resetAt(limits.seven_day?.resets_at),
+        scopedWindows,
+    };
+}
+function formatTokens(n) {
+    if (n >= 1_000_000)
+        return `${(n / 1_000_000).toFixed(1)}M`;
+    if (n >= 1000)
+        return `${(n / 1000).toFixed(0)}k`;
+    return String(Math.max(0, Math.trunc(n)));
+}
+function formatUsd(v) {
+    return v >= 100 ? `$${v.toFixed(0)}` : v >= 10 ? `$${v.toFixed(1)}` : `$${v.toFixed(2)}`;
+}