oh-my-githubcopilot 1.4.0

package/dist/hooks/token-tracker.mjs

@@ -0,0 +1,181 @@
+// src/hooks/token-tracker.mts
+import { readFileSync as readFileSync2, writeFileSync as writeFileSync2, mkdirSync as mkdirSync2 } from "fs";
+import { homedir as homedir2 } from "os";
+import { join as join2 } from "path";
+
+// src/spending/tracker.mts
+import { readFileSync, writeFileSync, mkdirSync } from "fs";
+import { homedir } from "os";
+import { join, dirname } from "path";
+var SPENDING_PATH = join(homedir(), ".omp", "state", "spending-monthly.json");
+function currentMonth() {
+  const now = /* @__PURE__ */ new Date();
+  return `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, "0")}`;
+}
+function loadSpending(sessionId) {
+  let raw;
+  try {
+    raw = JSON.parse(readFileSync(SPENDING_PATH, "utf-8"));
+  } catch {
+    return {
+      version: 1,
+      sessionId,
+      sessionPremiumRequests: 0,
+      month: currentMonth(),
+      monthlyPremiumRequests: 0
+    };
+  }
+  const month = currentMonth();
+  if (raw.month !== month) {
+    return {
+      version: 1,
+      sessionId,
+      sessionPremiumRequests: 0,
+      month,
+      monthlyPremiumRequests: 0
+    };
+  }
+  if (raw.sessionId !== sessionId) {
+    return {
+      version: 1,
+      sessionId,
+      sessionPremiumRequests: 0,
+      month,
+      monthlyPremiumRequests: raw.monthlyPremiumRequests
+    };
+  }
+  return { ...raw, version: 1 };
+}
+function saveSpending(state) {
+  try {
+    mkdirSync(dirname(SPENDING_PATH), { recursive: true });
+    writeFileSync(SPENDING_PATH, JSON.stringify(state, null, 2), "utf-8");
+  } catch (e) {
+    console.warn(`[OMP] spending: failed to save state: ${e}`);
+  }
+}
+function incrementSpending(sessionId) {
+  const state = loadSpending(sessionId);
+  state.sessionPremiumRequests += 1;
+  state.monthlyPremiumRequests += 1;
+  saveSpending(state);
+  return state;
+}
+
+// src/hooks/token-tracker.mts
+import { fileURLToPath } from "url";
+var MODEL_CONTEXTS = {
+  "claude-sonnet-4.5": 2e5,
+  "claude-sonnet-4": 2e5,
+  "claude-sonnet-4.6": 2e5,
+  "claude-opus-4.6": 2e5,
+  "gpt-5": 128e3,
+  "gpt-5.4-mini": 128e3,
+  "gemini-3-pro": 128e3,
+  default: 2e5
+};
+var WARNING_THRESHOLDS = [60, 80, 90];
+function estimateTokens(input) {
+  if (!input) return 0;
+  try {
+    const str = typeof input === "string" ? input : JSON.stringify(input);
+    return Math.ceil(str.length / 4);
+  } catch {
+    return 0;
+  }
+}
+function getStatePath(sessionId) {
+  const base = join2(homedir2(), ".omp", "state");
+  if (sessionId) {
+    return join2(base, "sessions", sessionId, "session.json");
+  }
+  return join2(base, "session.json");
+}
+function ensureDir(path) {
+  mkdirSync2(path.substring(0, path.lastIndexOf("/")), { recursive: true });
+}
+function processHook(input) {
+  const start = Date.now();
+  const log = [];
+  try {
+    if (input.hook_type !== "PostToolUse") {
+      return {
+        status: "skip",
+        latencyMs: Date.now() - start,
+        mutations: [],
+        log: []
+      };
+    }
+    const statePath = getStatePath(input.session_id);
+    let state;
+    try {
+      state = JSON.parse(readFileSync2(statePath, "utf-8"));
+    } catch {
+      state = {
+        tokens_estimated: 0,
+        token_budget: 2e5,
+        context_pct: 0,
+        warnings_issued: /* @__PURE__ */ new Set()
+      };
+    }
+    const inputTokens = estimateTokens(input.tool_input);
+    const outputTokens = estimateTokens(input.tool_output);
+    const delta = inputTokens + outputTokens;
+    state.tokens_estimated += delta;
+    state.context_pct = Math.min(100, Math.round(state.tokens_estimated / state.token_budget * 100));
+    const mutations = [
+      { type: "set_token_budget", budget: state.token_budget }
+    ];
+    for (const threshold of WARNING_THRESHOLDS) {
+      const key = `warn_${threshold}`;
+      if (state.context_pct >= threshold && !state.warnings_issued.has(key)) {
+        state.warnings_issued.add(key);
+        const message = threshold >= 90 ? `CRITICAL: Context at ${state.context_pct}%. Tokens near budget limit.` : threshold >= 80 ? `WARNING: Context at ${state.context_pct}%. Consider enabling ecomode.` : `INFO: Context at ${state.context_pct}%.`;
+        mutations.push({ type: "log", level: threshold >= 80 ? "warn" : "info", message });
+        log.push(message);
+      }
+    }
+    try {
+      ensureDir(statePath);
+      writeFileSync2(statePath, JSON.stringify(state), "utf-8");
+    } catch (e) {
+      log.push(`Failed to write state: ${e}`);
+    }
+    const sessionId = input.session_id ?? `omp-${Date.now()}`;
+    try {
+      incrementSpending(sessionId);
+    } catch {
+    }
+    return {
+      status: "ok",
+      latencyMs: Date.now() - start,
+      mutations,
+      log
+    };
+  } catch (err) {
+    return {
+      status: "error",
+      latencyMs: Date.now() - start,
+      mutations: [],
+      log: [`Error: ${err}`]
+    };
+  }
+}
+if (process.argv[1] === fileURLToPath(import.meta.url)) {
+  const input = JSON.parse(await readStdin());
+  const output = processHook(input);
+  console.log(JSON.stringify(output));
+}
+async function readStdin() {
+  const chunks = [];
+  for await (const chunk of process.stdin) {
+    chunks.push(chunk);
+  }
+  return chunks.join("");
+}
+export {
+  MODEL_CONTEXTS,
+  estimateTokens,
+  processHook
+};
+//# sourceMappingURL=token-tracker.mjs.map

package/dist/hooks/token-tracker.mjs.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../src/hooks/token-tracker.mts", "../../src/spending/tracker.mts"],
+  "sourcesContent": ["/**\n * token-tracker hook\n * Trigger: post-message (PostToolUse equivalent)\n * Priority: 70\n *\n * Estimates token usage from character counts (1 token \u2248 4 chars).\n * Accumulates in session state. Warns at 60%, 80%, 90% thresholds.\n */\n\nimport { readFileSync, writeFileSync, mkdirSync } from \"fs\";\nimport { homedir } from \"os\";\nimport { join } from \"path\";\nimport { incrementSpending } from \"../spending/tracker.mjs\";\n\nexport interface HookInput {\n  hook_type: \"PostToolUse\";\n  tool_name?: string;\n  tool_input?: unknown;\n  tool_output?: unknown;\n  session_id?: string;\n}\n\nexport interface HookOutput {\n  modifiedResult?: unknown;\n  status: \"ok\" | \"skip\" | \"error\";\n  latencyMs: number;\n  mutations: Array<{ type: \"set_token_budget\"; budget: number } | { type: \"emit_hud\"; hudEmit: unknown } | { type: \"log\"; level: \"info\" | \"warn\" | \"error\"; message: string }>;\n  log: string[];\n}\n\ninterface SessionState {\n  tokens_estimated: number;\n  token_budget: number;\n  context_pct: number;\n  warnings_issued: Set<string>;\n}\n\n// Model context windows in tokens (for future model-specific budget lookup)\n// Exported for potential external use\nexport const MODEL_CONTEXTS = {\n  \"claude-sonnet-4.5\": 200_000,\n  \"claude-sonnet-4\": 200_000,\n  \"claude-sonnet-4.6\": 200_000,\n  \"claude-opus-4.6\": 200_000,\n  \"gpt-5\": 128_000,\n  \"gpt-5.4-mini\": 128_000,\n  \"gemini-3-pro\": 128_000,\n  default: 200_000,\n};\n\nconst WARNING_THRESHOLDS = [60, 80, 90];\n\nexport function estimateTokens(input: unknown): number {\n  if (!input) return 0;\n  try {\n    const str = typeof input === \"string\" ? input : JSON.stringify(input);\n    return Math.ceil(str.length / 4);\n  } catch {\n    return 0;\n  }\n}\n\nfunction getStatePath(sessionId?: string): string {\n  const base = join(homedir(), \".omp\", \"state\");\n  if (sessionId) {\n    return join(base, \"sessions\", sessionId, \"session.json\");\n  }\n  return join(base, \"session.json\");\n}\n\nfunction ensureDir(path: string): void {\n  mkdirSync(path.substring(0, path.lastIndexOf(\"/\")), { recursive: true });\n}\n\nexport function processHook(input: HookInput): HookOutput {\n  const start = Date.now();\n  const log: string[] = [];\n\n  try {\n    if (input.hook_type !== \"PostToolUse\") {\n      return {\n        status: \"skip\",\n        latencyMs: Date.now() - start,\n        mutations: [],\n        log: [],\n      };\n    }\n\n    const statePath = getStatePath(input.session_id);\n    let state: SessionState;\n\n    try {\n      state = JSON.parse(readFileSync(statePath, \"utf-8\"));\n    } catch {\n      // Initialize state if not found\n      state = {\n        tokens_estimated: 0,\n        token_budget: 200_000,\n        context_pct: 0,\n        warnings_issued: new Set(),\n      };\n    }\n\n    const inputTokens = estimateTokens(input.tool_input);\n    const outputTokens = estimateTokens(input.tool_output);\n    const delta = inputTokens + outputTokens;\n\n    state.tokens_estimated += delta;\n    state.context_pct = Math.min(100, Math.round((state.tokens_estimated / state.token_budget) * 100));\n\n    const mutations: HookOutput[\"mutations\"] = [\n      { type: \"set_token_budget\", budget: state.token_budget },\n    ];\n\n    // Check warning thresholds\n    for (const threshold of WARNING_THRESHOLDS) {\n      const key = `warn_${threshold}`;\n      if (state.context_pct >= threshold && !state.warnings_issued.has(key)) {\n        state.warnings_issued.add(key);\n        const message =\n          threshold >= 90\n            ? `CRITICAL: Context at ${state.context_pct}%. Tokens near budget limit.`\n            : threshold >= 80\n            ? `WARNING: Context at ${state.context_pct}%. Consider enabling ecomode.`\n            : `INFO: Context at ${state.context_pct}%.`;\n        mutations.push({ type: \"log\", level: threshold >= 80 ? \"warn\" : \"info\", message });\n        log.push(message);\n      }\n    }\n\n    // Write state back\n    try {\n      ensureDir(statePath);\n      writeFileSync(statePath, JSON.stringify(state), \"utf-8\");\n    } catch (e) {\n      log.push(`Failed to write state: ${e}`);\n    }\n\n    // Track premium request spending\n    const sessionId = input.session_id ?? `omp-${Date.now()}`;\n    try { incrementSpending(sessionId); } catch { /* non-blocking */ }\n\n    return {\n      status: \"ok\",\n      latencyMs: Date.now() - start,\n      mutations,\n      log,\n    };\n  } catch (err) {\n    return {\n      status: \"error\",\n      latencyMs: Date.now() - start,\n      mutations: [],\n      log: [`Error: ${err}`],\n    };\n  }\n}\n\n// Main entry point \u2014 only runs when executed directly (not imported)\nimport { fileURLToPath } from \"url\";\n\nif (process.argv[1] === fileURLToPath(import.meta.url)) {\n  const input: HookInput = JSON.parse(await readStdin());\n  const output = processHook(input);\n  console.log(JSON.stringify(output));\n}\n\nasync function readStdin(): Promise<string> {\n  const chunks: string[] = [];\n  for await (const chunk of process.stdin) {\n    chunks.push(chunk);\n  }\n  return chunks.join(\"\");\n}\n", "/**\n * Spending tracker for OMP.\n * Tracks premium API requests per session and per calendar month.\n * Persists to ~/.omp/state/spending-monthly.json\n *\n * // v1.1 known limitation: no /omp:spending reset command. To reset monthly counter manually: rm ~/.omp/state/spending-monthly.json\n */\n\nimport { readFileSync, writeFileSync, mkdirSync } from \"fs\";\nimport { homedir } from \"os\";\nimport { join, dirname } from \"path\";\nimport type { SpendingState } from \"./types.mjs\";\n\nconst SPENDING_PATH = join(homedir(), \".omp\", \"state\", \"spending-monthly.json\");\n\nfunction currentMonth(): string {\n  const now = new Date();\n  return `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, \"0\")}`;\n}\n\nexport function loadSpending(sessionId: string): SpendingState {\n  let raw: SpendingState;\n  try {\n    raw = JSON.parse(readFileSync(SPENDING_PATH, \"utf-8\")) as SpendingState;\n  } catch {\n    // Missing or malformed file \u2014 start fresh\n    return {\n      version: 1,\n      sessionId,\n      sessionPremiumRequests: 0,\n      month: currentMonth(),\n      monthlyPremiumRequests: 0,\n    };\n  }\n\n  const month = currentMonth();\n\n  // Reset monthly counter when month rolls over\n  if (raw.month !== month) {\n    return {\n      version: 1,\n      sessionId,\n      sessionPremiumRequests: 0,\n      month,\n      monthlyPremiumRequests: 0,\n    };\n  }\n\n  // Reset session counter when session changes\n  if (raw.sessionId !== sessionId) {\n    return {\n      version: 1,\n      sessionId,\n      sessionPremiumRequests: 0,\n      month,\n      monthlyPremiumRequests: raw.monthlyPremiumRequests,\n    };\n  }\n\n  return { ...raw, version: 1 };\n}\n\nexport function saveSpending(state: SpendingState): void {\n  try {\n    mkdirSync(dirname(SPENDING_PATH), { recursive: true });\n    writeFileSync(SPENDING_PATH, JSON.stringify(state, null, 2), \"utf-8\");\n  } catch (e) {\n    console.warn(`[OMP] spending: failed to save state: ${e}`);\n  }\n}\n\nexport function incrementSpending(sessionId: string): SpendingState {\n  const state = loadSpending(sessionId);\n  state.sessionPremiumRequests += 1;\n  state.monthlyPremiumRequests += 1;\n  saveSpending(state);\n  return state;\n}\n"],
+  "mappings": ";AASA,SAAS,gBAAAA,eAAc,iBAAAC,gBAAe,aAAAC,kBAAiB;AACvD,SAAS,WAAAC,gBAAe;AACxB,SAAS,QAAAC,aAAY;;;ACHrB,SAAS,cAAc,eAAe,iBAAiB;AACvD,SAAS,eAAe;AACxB,SAAS,MAAM,eAAe;AAG9B,IAAM,gBAAgB,KAAK,QAAQ,GAAG,QAAQ,SAAS,uBAAuB;AAE9E,SAAS,eAAuB;AAC9B,QAAM,MAAM,oBAAI,KAAK;AACrB,SAAO,GAAG,IAAI,YAAY,CAAC,IAAI,OAAO,IAAI,SAAS,IAAI,CAAC,EAAE,SAAS,GAAG,GAAG,CAAC;AAC5E;AAEO,SAAS,aAAa,WAAkC;AAC7D,MAAI;AACJ,MAAI;AACF,UAAM,KAAK,MAAM,aAAa,eAAe,OAAO,CAAC;AAAA,EACvD,QAAQ;AAEN,WAAO;AAAA,MACL,SAAS;AAAA,MACT;AAAA,MACA,wBAAwB;AAAA,MACxB,OAAO,aAAa;AAAA,MACpB,wBAAwB;AAAA,IAC1B;AAAA,EACF;AAEA,QAAM,QAAQ,aAAa;AAG3B,MAAI,IAAI,UAAU,OAAO;AACvB,WAAO;AAAA,MACL,SAAS;AAAA,MACT;AAAA,MACA,wBAAwB;AAAA,MACxB;AAAA,MACA,wBAAwB;AAAA,IAC1B;AAAA,EACF;AAGA,MAAI,IAAI,cAAc,WAAW;AAC/B,WAAO;AAAA,MACL,SAAS;AAAA,MACT;AAAA,MACA,wBAAwB;AAAA,MACxB;AAAA,MACA,wBAAwB,IAAI;AAAA,IAC9B;AAAA,EACF;AAEA,SAAO,EAAE,GAAG,KAAK,SAAS,EAAE;AAC9B;AAEO,SAAS,aAAa,OAA4B;AACvD,MAAI;AACF,cAAU,QAAQ,aAAa,GAAG,EAAE,WAAW,KAAK,CAAC;AACrD,kBAAc,eAAe,KAAK,UAAU,OAAO,MAAM,CAAC,GAAG,OAAO;AAAA,EACtE,SAAS,GAAG;AACV,YAAQ,KAAK,yCAAyC,CAAC,EAAE;AAAA,EAC3D;AACF;AAEO,SAAS,kBAAkB,WAAkC;AAClE,QAAM,QAAQ,aAAa,SAAS;AACpC,QAAM,0BAA0B;AAChC,QAAM,0BAA0B;AAChC,eAAa,KAAK;AAClB,SAAO;AACT;;;ADkFA,SAAS,qBAAqB;AAxHvB,IAAM,iBAAiB;AAAA,EAC5B,qBAAqB;AAAA,EACrB,mBAAmB;AAAA,EACnB,qBAAqB;AAAA,EACrB,mBAAmB;AAAA,EACnB,SAAS;AAAA,EACT,gBAAgB;AAAA,EAChB,gBAAgB;AAAA,EAChB,SAAS;AACX;AAEA,IAAM,qBAAqB,CAAC,IAAI,IAAI,EAAE;AAE/B,SAAS,eAAe,OAAwB;AACrD,MAAI,CAAC,MAAO,QAAO;AACnB,MAAI;AACF,UAAM,MAAM,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,KAAK;AACpE,WAAO,KAAK,KAAK,IAAI,SAAS,CAAC;AAAA,EACjC,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,aAAa,WAA4B;AAChD,QAAM,OAAOC,MAAKC,SAAQ,GAAG,QAAQ,OAAO;AAC5C,MAAI,WAAW;AACb,WAAOD,MAAK,MAAM,YAAY,WAAW,cAAc;AAAA,EACzD;AACA,SAAOA,MAAK,MAAM,cAAc;AAClC;AAEA,SAAS,UAAU,MAAoB;AACrC,EAAAE,WAAU,KAAK,UAAU,GAAG,KAAK,YAAY,GAAG,CAAC,GAAG,EAAE,WAAW,KAAK,CAAC;AACzE;AAEO,SAAS,YAAY,OAA8B;AACxD,QAAM,QAAQ,KAAK,IAAI;AACvB,QAAM,MAAgB,CAAC;AAEvB,MAAI;AACF,QAAI,MAAM,cAAc,eAAe;AACrC,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,WAAW,KAAK,IAAI,IAAI;AAAA,QACxB,WAAW,CAAC;AAAA,QACZ,KAAK,CAAC;AAAA,MACR;AAAA,IACF;AAEA,UAAM,YAAY,aAAa,MAAM,UAAU;AAC/C,QAAI;AAEJ,QAAI;AACF,cAAQ,KAAK,MAAMC,cAAa,WAAW,OAAO,CAAC;AAAA,IACrD,QAAQ;AAEN,cAAQ;AAAA,QACN,kBAAkB;AAAA,QAClB,cAAc;AAAA,QACd,aAAa;AAAA,QACb,iBAAiB,oBAAI,IAAI;AAAA,MAC3B;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,MAAM,UAAU;AACnD,UAAM,eAAe,eAAe,MAAM,WAAW;AACrD,UAAM,QAAQ,cAAc;AAE5B,UAAM,oBAAoB;AAC1B,UAAM,cAAc,KAAK,IAAI,KAAK,KAAK,MAAO,MAAM,mBAAmB,MAAM,eAAgB,GAAG,CAAC;AAEjG,UAAM,YAAqC;AAAA,MACzC,EAAE,MAAM,oBAAoB,QAAQ,MAAM,aAAa;AAAA,IACzD;AAGA,eAAW,aAAa,oBAAoB;AAC1C,YAAM,MAAM,QAAQ,SAAS;AAC7B,UAAI,MAAM,eAAe,aAAa,CAAC,MAAM,gBAAgB,IAAI,GAAG,GAAG;AACrE,cAAM,gBAAgB,IAAI,GAAG;AAC7B,cAAM,UACJ,aAAa,KACT,wBAAwB,MAAM,WAAW,iCACzC,aAAa,KACb,uBAAuB,MAAM,WAAW,kCACxC,oBAAoB,MAAM,WAAW;AAC3C,kBAAU,KAAK,EAAE,MAAM,OAAO,OAAO,aAAa,KAAK,SAAS,QAAQ,QAAQ,CAAC;AACjF,YAAI,KAAK,OAAO;AAAA,MAClB;AAAA,IACF;AAGA,QAAI;AACF,gBAAU,SAAS;AACnB,MAAAC,eAAc,WAAW,KAAK,UAAU,KAAK,GAAG,OAAO;AAAA,IACzD,SAAS,GAAG;AACV,UAAI,KAAK,0BAA0B,CAAC,EAAE;AAAA,IACxC;AAGA,UAAM,YAAY,MAAM,cAAc,OAAO,KAAK,IAAI,CAAC;AACvD,QAAI;AAAE,wBAAkB,SAAS;AAAA,IAAG,QAAQ;AAAA,IAAqB;AAEjE,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,WAAW,KAAK,IAAI,IAAI;AAAA,MACxB;AAAA,MACA;AAAA,IACF;AAAA,EACF,SAAS,KAAK;AACZ,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,WAAW,KAAK,IAAI,IAAI;AAAA,MACxB,WAAW,CAAC;AAAA,MACZ,KAAK,CAAC,UAAU,GAAG,EAAE;AAAA,IACvB;AAAA,EACF;AACF;AAKA,IAAI,QAAQ,KAAK,CAAC,MAAM,cAAc,YAAY,GAAG,GAAG;AACtD,QAAM,QAAmB,KAAK,MAAM,MAAM,UAAU,CAAC;AACrD,QAAM,SAAS,YAAY,KAAK;AAChC,UAAQ,IAAI,KAAK,UAAU,MAAM,CAAC;AACpC;AAEA,eAAe,YAA6B;AAC1C,QAAM,SAAmB,CAAC;AAC1B,mBAAiB,SAAS,QAAQ,OAAO;AACvC,WAAO,KAAK,KAAK;AAAA,EACnB;AACA,SAAO,OAAO,KAAK,EAAE;AACvB;",
+  "names": ["readFileSync", "writeFileSync", "mkdirSync", "homedir", "join", "join", "homedir", "mkdirSync", "readFileSync", "writeFileSync"]
+}