@kapishdima/ai-ledger 0.0.5 → 0.0.7

package/dist/plugin.js

@@ -3739,24 +3739,53 @@ async function deductBalance(provider, cost) {
 }
 
 // plugin.ts
-var pending = new Map;
-var DEBOUNCE_MS = 1500;
+var sessions = new Map;
 var plugin_default = async (input) => {
   await initializeDatabase();
+  function log(msg) {
+    input.client.app.log({ body: { service: "ai-ledger", level: "info", message: msg } });
+  }
+  async function flushTurn(sessionID, state) {
+    sessions.delete(sessionID);
+    const text2 = `ai-ledger \xB7 ${state.providerID} \xB7 -$${state.totalCost.toFixed(4)} spent | $${state.lastBalance.toFixed(2)} remaining`;
+    log(`flushing turn for ${sessionID}: ${text2}`);
+    try {
+      const res = await input.client.session.prompt({
+        path: { id: sessionID },
+        query: { directory: input.directory },
+        body: {
+          noReply: true,
+          parts: [{ type: "text", text: text2 }]
+        }
+      });
+      log(`session.prompt result: ${JSON.stringify(res)}`);
+    } catch (e) {
+      log(`session.prompt error: ${e?.message}`);
+    }
+  }
   return {
     event: async ({ event }) => {
+      log(`event: ${event.type}`);
+      if (event.type === "session.idle") {
+        const sessionID = event.properties?.sessionID;
+        log(`session.idle for ${sessionID}, accumulated: ${sessions.has(sessionID)}`);
+        const state = sessions.get(sessionID);
+        if (state)
+          await flushTurn(sessionID, state);
+        return;
+      }
       if (event.type !== "message.updated")
         return;
       const msg = event.properties?.info;
-      if (!msg)
+      if (!msg?.finish)
         return;
       if (msg.role !== "assistant")
         return;
-      if (!msg.finish)
-        return;
+      log(`assistant message: cost=${msg.cost} provider=${msg.providerID}`);
       if (!msg.cost || msg.cost === 0)
         return;
       const wallet = await getWallet(msg.providerID);
+      log(`wallet for ${msg.providerID}: ${wallet ? "found" : "NOT FOUND"}`);
       if (!wallet)
         return;
       const newBalance = await deductBalance(msg.providerID, msg.cost);
@@ -3771,22 +3800,10 @@ var plugin_default = async (input) => {
         created_at: new Date().toISOString(),
         synced: 0
       });
-      const existing = pending.get(msg.providerID);
-      if (existing)
-        clearTimeout(existing.timer);
+      const existing = sessions.get(msg.sessionID);
       const totalCost = (existing?.totalCost ?? 0) + msg.cost;
-      const timer = setTimeout(() => {
-        pending.delete(msg.providerID);
-        input.client.tui.showToast({
-          body: {
-            title: `ai-ledger \xB7 ${msg.providerID}`,
-            message: `-$${totalCost.toFixed(4)}  |  balance: $${newBalance.toFixed(2)}`,
-            variant: newBalance < 0 ? "warning" : "info",
-            duration: 4000
-          }
-        });
-      }, DEBOUNCE_MS);
-      pending.set(msg.providerID, { totalCost, lastBalance: newBalance, timer });
+      sessions.set(msg.sessionID, { totalCost, lastBalance: newBalance, providerID: msg.providerID });
+      log(`accumulated for session ${msg.sessionID}: totalCost=${totalCost}`);
     }
   };
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kapishdima/ai-ledger",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "description": "AI provider balance and spending tracker for OpenCode",
   "type": "module",
   "main": "dist/plugin.js",