@kapishdima/ai-ledger 0.0.4 → 0.0.6

package/dist/plugin.js

@@ -3739,19 +3739,37 @@ async function deductBalance(provider, cost) {
 }
 
 // plugin.ts
+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);
@@ -3769,14 +3787,9 @@ var plugin_default = async (input) => {
         created_at: new Date().toISOString(),
         synced: 0
       });
-      input.client.tui.showToast({
-        body: {
-          title: `ai-ledger \xB7 ${msg.providerID}`,
-          message: `-$${msg.cost.toFixed(4)}  |  balance: $${newBalance.toFixed(2)}`,
-          variant: newBalance < 0 ? "warning" : "info",
-          duration: 4000
-        }
-      });
+      const existing = sessions.get(msg.sessionID);
+      const totalCost = (existing?.totalCost ?? 0) + msg.cost;
+      sessions.set(msg.sessionID, { totalCost, lastBalance: newBalance, providerID: msg.providerID });
     }
   };
 };

package/package.json

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