@devness/useai 0.6.11 → 0.6.12

package/dist/index.js

@@ -684,7 +684,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.6.11";
+    VERSION = "0.6.12";
   }
 });
 
@@ -34589,6 +34589,12 @@ var init_session_state = __esm({
       modelId;
       /** Token estimates for the useai_start tool call. */
       startCallTokensEst;
+      /** Whether a UseAI session is actively in-progress (between useai_start and useai_end). */
+      inProgress;
+      /** Timestamp when the current session was marked in-progress. */
+      inProgressSince;
+      /** Session ID that was auto-sealed by seal-active hook (for useai_end fallback). */
+      autoSealedSessionId;
       constructor() {
         this.sessionId = generateSessionId();
         this.conversationId = generateSessionId();
@@ -34596,6 +34602,9 @@ var init_session_state = __esm({
         this.mcpSessionId = null;
         this.modelId = null;
         this.startCallTokensEst = null;
+        this.inProgress = false;
+        this.inProgressSince = null;
+        this.autoSealedSessionId = null;
         this.sessionStartTime = Date.now();
         this.heartbeatCount = 0;
         this.sessionRecordCount = 0;
@@ -34622,6 +34631,8 @@ var init_session_state = __esm({
         this.sessionPromptWordCount = null;
         this.modelId = null;
         this.startCallTokensEst = null;
+        this.inProgress = false;
+        this.inProgressSince = null;
         this.detectProject();
       }
       detectProject() {
@@ -34715,7 +34726,7 @@ __export(register_tools_exports, {
   registerTools: () => registerTools
 });
 import { createHash as createHash3, randomUUID as randomUUID3 } from "crypto";
-import { existsSync as existsSync8, renameSync as renameSync2 } from "fs";
+import { existsSync as existsSync8, readFileSync as readFileSync5, renameSync as renameSync2 } from "fs";
 import { join as join7 } from "path";
 function coerceJsonString(schema) {
   return external_exports.preprocess((val) => {
@@ -34750,6 +34761,130 @@ function resolveClient2(server2, session2) {
   }
   session2.setClient(detectClient());
 }
+function enrichAutoSealedSession(sealedSessionId, session2, args) {
+  const sealedPath = join7(SEALED_DIR, `${sealedSessionId}.jsonl`);
+  const activePath = join7(ACTIVE_DIR, `${sealedSessionId}.jsonl`);
+  const chainPath = existsSync8(sealedPath) ? sealedPath : existsSync8(activePath) ? activePath : null;
+  if (!chainPath) {
+    return "No active session to end (already sealed or never started).";
+  }
+  let startData = {};
+  let duration3 = 0;
+  let endedAt = (/* @__PURE__ */ new Date()).toISOString();
+  let startedAt2 = endedAt;
+  try {
+    const content = readFileSync5(chainPath, "utf-8").trim();
+    const lines = content.split("\n").filter(Boolean);
+    if (lines.length > 0) {
+      const firstRecord = JSON.parse(lines[0]);
+      startData = firstRecord.data;
+      startedAt2 = firstRecord.timestamp;
+      const lastRecord = JSON.parse(lines[lines.length - 1]);
+      if (lastRecord.type === "session_seal" && lastRecord.data["seal"]) {
+        try {
+          const sealObj = JSON.parse(lastRecord.data["seal"]);
+          duration3 = sealObj.duration_seconds ?? Math.round((new Date(lastRecord.timestamp).getTime() - new Date(startedAt2).getTime()) / 1e3);
+          endedAt = sealObj.ended_at ?? lastRecord.timestamp;
+        } catch {
+          duration3 = Math.round((new Date(lastRecord.timestamp).getTime() - new Date(startedAt2).getTime()) / 1e3);
+          endedAt = lastRecord.timestamp;
+        }
+      } else {
+        duration3 = Math.round((new Date(lastRecord.timestamp).getTime() - new Date(startedAt2).getTime()) / 1e3);
+        endedAt = lastRecord.timestamp;
+      }
+    }
+  } catch {
+    return "No active session to end (chain file unreadable).";
+  }
+  const taskType = args.task_type ?? startData.task_type ?? "coding";
+  const languages = args.languages ?? [];
+  const filesTouched = args.files_touched_count ?? 0;
+  let milestoneCount = 0;
+  if (args.milestones && args.milestones.length > 0) {
+    const config2 = getConfig();
+    if (config2.milestone_tracking) {
+      const durationMinutes = Math.round(duration3 / 60);
+      const allMilestones = getMilestones();
+      for (const m of args.milestones) {
+        allMilestones.push({
+          id: `m_${randomUUID3().slice(0, 8)}`,
+          session_id: sealedSessionId,
+          title: m.title,
+          private_title: m.private_title,
+          project: startData.project ?? session2.project ?? void 0,
+          category: m.category,
+          complexity: m.complexity ?? "medium",
+          duration_minutes: durationMinutes,
+          languages,
+          client: startData.client ?? session2.clientName,
+          created_at: (/* @__PURE__ */ new Date()).toISOString(),
+          published: false,
+          published_at: null,
+          chain_hash: ""
+        });
+        milestoneCount++;
+      }
+      writeJson(MILESTONES_FILE, allMilestones);
+    }
+  }
+  let sessionScore;
+  let frameworkId;
+  if (args.evaluation) {
+    const config2 = getConfig();
+    const framework = getFramework(config2.evaluation_framework);
+    sessionScore = Math.round(framework.computeSessionScore(args.evaluation));
+    frameworkId = framework.id;
+  }
+  const richSeal = {
+    session_id: sealedSessionId,
+    conversation_id: startData.conversation_id,
+    conversation_index: startData.conversation_index,
+    client: startData.client ?? session2.clientName,
+    task_type: taskType,
+    languages,
+    files_touched: filesTouched,
+    project: startData.project ?? session2.project ?? void 0,
+    title: startData.title ?? void 0,
+    private_title: startData.private_title ?? void 0,
+    model: startData.model ?? session2.modelId ?? void 0,
+    evaluation: args.evaluation ?? void 0,
+    session_score: sessionScore,
+    evaluation_framework: frameworkId,
+    started_at: startedAt2,
+    ended_at: endedAt,
+    duration_seconds: duration3,
+    heartbeat_count: 0,
+    record_count: 0,
+    chain_start_hash: "",
+    chain_end_hash: "",
+    seal_signature: ""
+  };
+  const allSessions = getSessions();
+  const existingIdx = allSessions.findIndex((s) => s.session_id === sealedSessionId);
+  if (existingIdx >= 0) {
+    const existing = allSessions[existingIdx];
+    allSessions[existingIdx] = {
+      ...existing,
+      // Enrich with data from useai_end call
+      task_type: taskType,
+      languages,
+      files_touched: filesTouched,
+      evaluation: args.evaluation ?? existing.evaluation,
+      session_score: sessionScore ?? existing.session_score,
+      evaluation_framework: frameworkId ?? existing.evaluation_framework
+    };
+  } else {
+    allSessions.push(richSeal);
+  }
+  writeJson(SESSIONS_FILE, allSessions);
+  const durationStr = formatDuration(duration3);
+  const langStr = languages.length > 0 ? ` using ${languages.join(", ")}` : "";
+  const milestoneStr = milestoneCount > 0 ? ` \xB7 ${milestoneCount} milestone${milestoneCount > 1 ? "s" : ""} recorded` : "";
+  const evalStr = args.evaluation ? ` \xB7 eval: ${args.evaluation.task_outcome} (prompt: ${args.evaluation.prompt_quality}/5)` : "";
+  const scoreStr = sessionScore !== void 0 ? ` \xB7 score: ${sessionScore}/100 (${frameworkId})` : "";
+  return `Session ended (enriched auto-seal): ${durationStr} ${taskType}${langStr}${milestoneStr}${evalStr}${scoreStr}`;
+}
 function registerTools(server2, session2, opts) {
   server2.tool(
     "useai_start",
@@ -34768,6 +34903,7 @@ function registerTools(server2, session2, opts) {
         opts.sealBeforeReset();
       }
       session2.reset();
+      session2.autoSealedSessionId = null;
       resolveClient2(server2, session2);
       if (conversation_id) {
         if (conversation_id !== prevConvId) {
@@ -34795,6 +34931,8 @@ function registerTools(server2, session2, opts) {
       if (private_title) chainData.private_title = private_title;
       if (model) chainData.model = model;
       const record2 = session2.appendToChain("session_start", chainData);
+      session2.inProgress = true;
+      session2.inProgressSince = Date.now();
       writeMcpMapping(session2.mcpSessionId, session2.sessionId);
       const responseText = `useai session started \u2014 ${session2.sessionTaskType} on ${session2.clientName} \xB7 ${session2.sessionId.slice(0, 8)} \xB7 conv ${session2.conversationId.slice(0, 8)}#${session2.conversationIndex} \xB7 ${session2.signingAvailable ? "signed" : "unsigned"}`;
       const paramsJson = JSON.stringify({ task_type, title, private_title, project, model });
@@ -34862,6 +35000,17 @@ function registerTools(server2, session2, opts) {
     },
     async ({ task_type, languages, files_touched_count, milestones: milestonesInput, evaluation }) => {
       if (session2.sessionRecordCount === 0) {
+        if (session2.autoSealedSessionId) {
+          const enrichResult = enrichAutoSealedSession(
+            session2.autoSealedSessionId,
+            session2,
+            { task_type, languages, files_touched_count, milestones: milestonesInput, evaluation }
+          );
+          session2.autoSealedSessionId = null;
+          session2.inProgress = false;
+          session2.inProgressSince = null;
+          return { content: [{ type: "text", text: enrichResult }] };
+        }
         return {
           content: [{ type: "text", text: "No active session to end (already sealed or never started)." }]
         };
@@ -35009,6 +35158,8 @@ function registerTools(server2, session2, opts) {
       const sessions2 = getSessions().filter((s) => s.session_id !== seal.session_id);
       sessions2.push(seal);
       writeJson(SESSIONS_FILE, sessions2);
+      session2.inProgress = false;
+      session2.inProgressSince = null;
       return {
         content: [
           {
@@ -35521,7 +35672,7 @@ __export(daemon_exports, {
 });
 import { createServer } from "http";
 import { createHash as createHash4, randomUUID as randomUUID4 } from "crypto";
-import { existsSync as existsSync10, readdirSync, readFileSync as readFileSync5, appendFileSync as appendFileSync2, renameSync as renameSync3, writeFileSync as writeFileSync5, unlinkSync as unlinkSync5, statSync } from "fs";
+import { existsSync as existsSync10, readdirSync, readFileSync as readFileSync6, appendFileSync as appendFileSync2, renameSync as renameSync3, writeFileSync as writeFileSync5, unlinkSync as unlinkSync5, statSync } from "fs";
 import { join as join9 } from "path";
 function getActiveUseaiSessionIds() {
   const ids = /* @__PURE__ */ new Set();
@@ -35534,7 +35685,7 @@ function sealOrphanFile(sessionId) {
   const filePath = join9(ACTIVE_DIR, `${sessionId}.jsonl`);
   if (!existsSync10(filePath)) return;
   try {
-    const content = readFileSync5(filePath, "utf-8").trim();
+    const content = readFileSync6(filePath, "utf-8").trim();
     if (!content) return;
     const lines = content.split("\n").filter(Boolean);
     if (lines.length === 0) return;
@@ -35772,8 +35923,10 @@ function autoSealSession(active) {
   upsertSessionSeal(seal);
 }
 function sealSessionData(active) {
+  const sealedId = active.session.sessionId;
   autoSealSession(active);
   active.session.reset();
+  active.session.autoSealedSessionId = sealedId;
 }
 function resetIdleTimer(sessionId) {
   const active = sessions.get(sessionId);
@@ -35877,7 +36030,7 @@ function readChainMetadata(useaiSessionId) {
   const chainPath = existsSync10(activePath) ? activePath : existsSync10(sealedPath) ? sealedPath : null;
   if (!chainPath) return null;
   try {
-    const firstLine = readFileSync5(chainPath, "utf-8").split("\n")[0];
+    const firstLine = readFileSync6(chainPath, "utf-8").split("\n")[0];
     if (!firstLine) return null;
     const record2 = JSON.parse(firstLine);
     const d = record2.data;
@@ -35946,7 +36099,7 @@ function recoverHeartbeat(staleMcpSessionId, rpcId, res) {
     return true;
   }
   try {
-    const content = readFileSync5(chainPath, "utf-8").trim();
+    const content = readFileSync6(chainPath, "utf-8").trim();
     const lines = content.split("\n").filter(Boolean);
     if (lines.length === 0) return false;
     const firstRecord = JSON.parse(lines[0]);
@@ -35981,7 +36134,7 @@ function recoverEndSession(staleMcpSessionId, args, rpcId, res) {
   const chainPath = existsSync10(activePath) ? activePath : existsSync10(sealedPath) ? sealedPath : null;
   if (!chainPath) return false;
   const isAlreadySealed = chainPath === sealedPath;
-  const content = readFileSync5(chainPath, "utf-8").trim();
+  const content = readFileSync6(chainPath, "utf-8").trim();
   if (!content) return false;
   const lines = content.split("\n").filter(Boolean);
   if (lines.length === 0) return false;
@@ -36387,8 +36540,14 @@ async function startDaemon(port) {
     }
     if (url.pathname === "/api/seal-active" && req.method === "POST") {
       let sealed = 0;
+      let skipped = 0;
       for (const [, active] of sessions) {
         if (active.session.sessionRecordCount > 0 && !isSessionAlreadySealed(active.session)) {
+          const { inProgress, inProgressSince } = active.session;
+          if (inProgress && inProgressSince && Date.now() - inProgressSince < SEAL_GRACE_MS) {
+            skipped++;
+            continue;
+          }
           sealSessionData(active);
           sealed++;
         }
@@ -36397,7 +36556,7 @@ async function startDaemon(port) {
         "Content-Type": "application/json",
         "Access-Control-Allow-Origin": "*"
       });
-      res.end(JSON.stringify({ sealed }));
+      res.end(JSON.stringify({ sealed, skipped }));
       return;
     }
     if ((url.pathname.startsWith("/api/local/") || url.pathname === "/api/seal-active") && req.method === "OPTIONS") {
@@ -36554,7 +36713,7 @@ async function startDaemon(port) {
   console.log(`UseAI daemon listening on http://127.0.0.1:${listenPort}`);
   console.log(`PID: ${process.pid}`);
 }
-var IDLE_TIMEOUT_MS, sessions, daemonSigningKey, ORPHAN_SWEEP_INTERVAL_MS, startedAt, updateCheckCache, UPDATE_CHECK_TTL_MS;
+var IDLE_TIMEOUT_MS, SEAL_GRACE_MS, sessions, daemonSigningKey, ORPHAN_SWEEP_INTERVAL_MS, startedAt, updateCheckCache, UPDATE_CHECK_TTL_MS;
 var init_daemon2 = __esm({
   "src/daemon.ts"() {
     "use strict";
@@ -36568,6 +36727,7 @@ var init_daemon2 = __esm({
     init_html();
     init_local_api();
     IDLE_TIMEOUT_MS = 30 * 60 * 1e3;
+    SEAL_GRACE_MS = 30 * 60 * 1e3;
     sessions = /* @__PURE__ */ new Map();
     daemonSigningKey = null;
     ORPHAN_SWEEP_INTERVAL_MS = 15 * 60 * 1e3;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devness/useai",
-  "version": "0.6.11",
+  "version": "0.6.12",
   "description": "Track your AI-assisted development workflow. MCP server that records usage metrics across all your AI tools.",
   "keywords": [
     "mcp",