@kapishdima/ai-ledger 0.0.5 → 0.0.6

package/dist/plugin.js

@@ -3739,21 +3739,37 @@ 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();
+  async function flushTurn(sessionID, state) {
+    sessions.delete(sessionID);
+    const balanceSign = state.lastBalance < 0 ? "\u26A0\uFE0F " : "";
+    const text2 = `\`ai-ledger\` \xB7 ${state.providerID}  \u2014  **-$${state.totalCost.toFixed(4)}** spent  |  **$${state.lastBalance.toFixed(2)}** remaining`;
+    await input.client.session.prompt({
+      path: { id: sessionID },
+      body: {
+        noReply: true,
+        parts: [{ type: "text", text: balanceSign + text2, synthetic: true }]
+      }
+    });
+  }
   return {
     event: async ({ event }) => {
+      if (event.type === "session.idle") {
+        const sessionID = event.properties?.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;
       if (!msg.cost || msg.cost === 0)
         return;
       const wallet = await getWallet(msg.providerID);
@@ -3771,22 +3787,9 @@ 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 });
     }
   };
 };

package/package.json

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