@askexenow/exe-os 0.9.7 → 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");
@@ -98,8 +108,8 @@ var init_config = __esm({
 import { readdir } from "fs/promises";
 import { createReadStream } from "fs";
 import { createInterface } from "readline";
-import path3 from "path";
-import os3 from "os";
+import path4 from "path";
+import os4 from "os";
 
 // src/lib/database.ts
 import { createClient } from "@libsql/client";
@@ -107,19 +117,39 @@ 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";
 var EMPLOYEES_PATH = path2.join(EXE_AI_DIR, "exe-employees.json");
+var IDENTITY_DIR = path2.join(EXE_AI_DIR, "identity");
+
+// src/lib/database-adapter.ts
+import os3 from "os";
+import path3 from "path";
+import { createRequire } from "module";
+import { pathToFileURL } from "url";
+var BOOLEAN_COLUMNS_BY_TABLE = {
+  memories: /* @__PURE__ */ new Set(["has_error", "draft"]),
+  behaviors: /* @__PURE__ */ new Set(["active"]),
+  notifications: /* @__PURE__ */ new Set(["read"]),
+  users: /* @__PURE__ */ new Set(["has_personal_memory"])
+};
+var BOOLEAN_COLUMN_NAMES = new Set(
+  Object.values(BOOLEAN_COLUMNS_BY_TABLE).flatMap((cols) => [...cols])
+);
 
 // src/lib/database.ts
 var _resilientClient = null;
 var _daemonClient = null;
+var _adapterClient = null;
 function getClient() {
-  if (!_resilientClient) {
+  if (!_adapterClient) {
     throw new Error("Database client not initialized. Call initDatabase() first.");
   }
+  if (process.env.DATABASE_URL) {
+    return _adapterClient;
+  }
   if (process.env.EXE_IS_DAEMON === "1") {
     return _resilientClient;
   }
@@ -155,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}$/, "");
@@ -166,29 +198,33 @@ 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 = path3.join(os3.homedir(), ".claude", "projects");
+  const claudeDir = path4.join(os4.homedir(), ".claude", "projects");
   let projectDirs = [];
   try {
     const entries = await readdir(claudeDir);
-    projectDirs = entries.map((e) => path3.join(claudeDir, e));
+    projectDirs = entries.map((e) => path4.join(claudeDir, e));
   } catch {
     return [];
   }
   const agentTotals = /* @__PURE__ */ new Map();
   for (const [sessionUuid, agentId] of sessionAgent) {
     for (const dir of projectDirs) {
-      const jsonlPath = path3.join(dir, `${sessionUuid}.jsonl`);
+      const jsonlPath = path4.join(dir, `${sessionUuid}.jsonl`);
       try {
         const usage = await extractSessionUsage(jsonlPath);
         if (usage.input === 0 && usage.output === 0) continue;
@@ -212,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,
@@ -222,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;