@askexenow/exe-os 0.9.8 → 0.9.9

package/dist/lib/token-spend.js

@@ -3,9 +3,18 @@ var __esm = (fn, res) => function __init() {
   return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
 };
 
+// src/lib/secure-files.ts
+import { chmodSync, existsSync, mkdirSync } from "fs";
+import { chmod, mkdir } from "fs/promises";
+var init_secure_files = __esm({
+  "src/lib/secure-files.ts"() {
+    "use strict";
+  }
+});
+
 // src/lib/config.ts
-import { readFile, writeFile, mkdir, chmod } from "fs/promises";
-import { readFileSync, existsSync, renameSync } from "fs";
+import { readFile, writeFile } from "fs/promises";
+import { readFileSync, existsSync as existsSync2, renameSync } from "fs";
 import path from "path";
 import os from "os";
 function resolveDataDir() {
@@ -13,7 +22,7 @@ function resolveDataDir() {
   if (process.env.EXE_MEM_DIR) return process.env.EXE_MEM_DIR;
   const newDir = path.join(os.homedir(), ".exe-os");
   const legacyDir = path.join(os.homedir(), ".exe-mem");
-  if (!existsSync(newDir) && existsSync(legacyDir)) {
+  if (!existsSync2(newDir) && existsSync2(legacyDir)) {
     try {
       renameSync(legacyDir, newDir);
       process.stderr.write(`[exe-os] Migrated data directory: ~/.exe-mem \u2192 ~/.exe-os
@@ -28,6 +37,7 @@ var EXE_AI_DIR, DB_PATH, MODELS_DIR, CONFIG_PATH, LEGACY_LANCE_PATH, CURRENT_CON
 var init_config = __esm({
   "src/lib/config.ts"() {
     "use strict";
+    init_secure_files();
     EXE_AI_DIR = resolveDataDir();
     DB_PATH = path.join(EXE_AI_DIR, "memories.db");
     MODELS_DIR = path.join(EXE_AI_DIR, "models");
@@ -107,7 +117,7 @@ import { createClient } from "@libsql/client";
 // src/lib/employees.ts
 init_config();
 import { readFile as readFile2, writeFile as writeFile2, mkdir as mkdir2 } from "fs/promises";
-import { existsSync as existsSync2, symlinkSync, readlinkSync, readFileSync as readFileSync2, renameSync as renameSync2, unlinkSync, writeFileSync } from "fs";
+import { existsSync as existsSync3, symlinkSync, readlinkSync, readFileSync as readFileSync2, renameSync as renameSync2, unlinkSync, writeFileSync } from "fs";
 import { execSync } from "child_process";
 import path2 from "path";
 import os2 from "os";
@@ -175,6 +185,8 @@ var MODEL_PRICING = {
   "claude-3-haiku": { input: 0.25 / 1e6, output: 1.25 / 1e6, cacheRead: 0.03 / 1e6, cacheWrite: 0.3 / 1e6 }
 };
 var DEFAULT_PRICING = MODEL_PRICING["claude-sonnet-4"];
+var CACHE_TTL_MS = 5 * 60 * 1e3;
+var _spendCache = /* @__PURE__ */ new Map();
 function getPricing(model) {
   if (MODEL_PRICING[model]) return MODEL_PRICING[model];
   const stripped = model.replace(/-\d{8}$/, "");
@@ -186,15 +198,19 @@ function getPricing(model) {
   return DEFAULT_PRICING;
 }
 async function getAgentSpend(period = "7d") {
+  const cached = _spendCache.get(period);
+  if (cached && Date.now() < cached.expires) {
+    return cached.result;
+  }
   const cutoff = periodToCutoff(period);
   const client = getClient();
-  const result = await client.execute({
+  const dbResult = await client.execute({
     sql: `SELECT session_uuid, agent_id FROM session_agent_map WHERE started_at >= ?`,
     args: [cutoff]
   });
-  if (result.rows.length === 0) return [];
+  if (dbResult.rows.length === 0) return [];
   const sessionAgent = /* @__PURE__ */ new Map();
-  for (const row of result.rows) {
+  for (const row of dbResult.rows) {
     sessionAgent.set(row.session_uuid, row.agent_id);
   }
   const claudeDir = path4.join(os4.homedir(), ".claude", "projects");
@@ -232,7 +248,7 @@ async function getAgentSpend(period = "7d") {
       }
     }
   }
-  return Array.from(agentTotals.entries()).map(([agentId, t]) => ({
+  const result = Array.from(agentTotals.entries()).map(([agentId, t]) => ({
     agentId,
     inputTokens: t.input,
     outputTokens: t.output,
@@ -242,6 +258,8 @@ async function getAgentSpend(period = "7d") {
     sessions: t.sessions.size,
     period
   })).sort((a, b) => b.costUSD - a.costUSD);
+  _spendCache.set(period, { result, expires: Date.now() + CACHE_TTL_MS });
+  return result;
 }
 async function extractSessionUsage(jsonlPath) {
   let input = 0;