@wopr-network/platform-core 1.74.0 → 1.74.1

package/src/monetization/credits/runtime-cron.ts

@@ -4,9 +4,25 @@ import { logger } from "../../config/logger.js";
 import type { IBotInstanceRepository } from "../../fleet/bot-instance-repository.js";
 import { RESOURCE_TIERS } from "../../fleet/resource-tiers.js";
 
+/** Monthly bot cost in dollars. */
+export const MONTHLY_BOT_COST_DOLLARS = 5;
+
+/**
+ * Compute the daily bot cost for a given date, prorated by the actual
+ * number of days in that month. Uses nano-dollar precision so totals
+ * sum to exactly $5.00/month (no over/under-billing).
+ */
+export function dailyBotCost(date: string): Credit {
+  const d = new Date(date);
+  const year = d.getFullYear();
+  const month = d.getMonth();
+  const daysInMonth = new Date(year, month + 1, 0).getDate();
+  return Credit.fromDollars(MONTHLY_BOT_COST_DOLLARS / daysInMonth);
+}
+
 /**
- * Bot runtime cost: $5/bot/month prorated daily.
- * $5.00 / 30 ≈ $0.1667/day, rounded to $0.17.
+ * @deprecated Use dailyBotCost(date) for accurate per-month proration.
+ * Kept for backwards compat in tests.
  */
 export const DAILY_BOT_COST = Credit.fromCents(17);
 
@@ -118,7 +134,8 @@ export async function runRuntimeDeductions(cfg: RuntimeCronConfig): Promise<Runt
         continue;
       }
 
-      const totalCost = DAILY_BOT_COST.multiply(botCount);
+      const dailyCost = dailyBotCost(cfg.date);
+      const totalCost = dailyCost.multiply(botCount);
       let didBillAnything = false;
 
       // Bill runtime debit (skipped if already billed on a previous run)
@@ -126,7 +143,7 @@ export async function runRuntimeDeductions(cfg: RuntimeCronConfig): Promise<Runt
         if (!balance.lessThan(totalCost)) {
           // Full deduction
           await cfg.ledger.debit(tenantId, totalCost, "bot_runtime", {
-            description: `Daily runtime: ${botCount} bot(s) x $${DAILY_BOT_COST.toDollars().toFixed(2)}`,
+            description: `Daily runtime: ${botCount} bot(s) x $${dailyCost.toDollars().toFixed(4)}`,
             referenceId: runtimeRef,
           });
         } else {

package/src/monetization/credits/storage-tier-billing.test.ts

@@ -8,6 +8,7 @@ import { DrizzleBotBilling } from "./bot-billing.js";
 describe("bot-billing storage tier", () => {
   let pool: PGlite;
   let db: DrizzleDb;
+  let repo: DrizzleBotInstanceRepository;
   let billing: DrizzleBotBilling;
 
   beforeAll(async () => {
@@ -22,16 +23,19 @@ describe("bot-billing storage tier", () => {
 
   beforeEach(async () => {
     await rollbackTestTransaction(pool);
-    billing = new DrizzleBotBilling(new DrizzleBotInstanceRepository(db));
+    repo = new DrizzleBotInstanceRepository(db);
+    billing = new DrizzleBotBilling(repo);
   });
 
   it("new bot defaults to standard storage tier", async () => {
     await billing.registerBot("bot-1", "tenant-1", "TestBot");
+    await repo.startBilling("bot-1");
     expect(await billing.getStorageTier("bot-1")).toBe("standard");
   });
 
   it("setStorageTier updates tier", async () => {
     await billing.registerBot("bot-1", "tenant-1", "TestBot");
+    await repo.startBilling("bot-1");
     await billing.setStorageTier("bot-1", "pro");
     expect(await billing.getStorageTier("bot-1")).toBe("pro");
   });
@@ -42,8 +46,11 @@ describe("bot-billing storage tier", () => {
 
   it("getStorageTierCostsForTenant sums active bot storage costs", async () => {
     await billing.registerBot("bot-1", "tenant-1", "Bot1");
+    await repo.startBilling("bot-1");
     await billing.registerBot("bot-2", "tenant-1", "Bot2");
+    await repo.startBilling("bot-2");
     await billing.registerBot("bot-3", "tenant-1", "Bot3");
+    await repo.startBilling("bot-3");
     await billing.setStorageTier("bot-1", "plus"); // 3 credits/day
     await billing.setStorageTier("bot-2", "max"); // 15 credits/day
     // bot-3 stays standard                        // 0 credits/day
@@ -52,6 +59,7 @@ describe("bot-billing storage tier", () => {
 
   it("getStorageTierCostsForTenant excludes suspended bots", async () => {
     await billing.registerBot("bot-1", "tenant-1", "Bot1");
+    await repo.startBilling("bot-1");
     await billing.setStorageTier("bot-1", "pro"); // 8 credits/day
     await billing.suspendBot("bot-1");
     expect((await billing.getStorageTierCostsForTenant("tenant-1")).toCents()).toBe(0);

package/src/monetization/credits/storage-tier-cron.test.ts

@@ -3,10 +3,11 @@ import { Credit, DrizzleLedger } from "@wopr-network/platform-core/credits";
 import { afterAll, beforeAll, beforeEach, describe, expect, it } from "vitest";
 import type { DrizzleDb } from "../../db/index.js";
 import { createTestDb, truncateAllTables } from "../../test/db.js";
-import { runRuntimeDeductions } from "./runtime-cron.js";
+import { dailyBotCost, runRuntimeDeductions } from "./runtime-cron.js";
 
 describe("runtime cron with storage tiers", () => {
   const TODAY = "2025-01-01";
+  const BASE_COST_CREDIT = dailyBotCost(TODAY);
   let pool: PGlite;
   let db: DrizzleDb;
   let ledger: DrizzleLedger;
@@ -36,8 +37,9 @@ describe("runtime cron with storage tiers", () => {
     });
     expect(result.processed).toBe(1);
     const balance = await ledger.balance("t1");
-    // 1000 - 17 (base) - 8 (pro storage surcharge) = 975
-    expect(balance.toCents()).toBe(975);
+    // 1000 - dailyBotCost (base) - 8 (pro storage surcharge)
+    const expected = Credit.fromCents(1000).subtract(BASE_COST_CREDIT).subtract(Credit.fromCents(8));
+    expect(balance.toCents()).toBe(expected.toCents());
   });
 
   it("debits only base cost for standard storage tier (zero surcharge)", async () => {
@@ -49,7 +51,8 @@ describe("runtime cron with storage tiers", () => {
       getStorageTierCosts: async () => Credit.ZERO,
     });
     expect(result.processed).toBe(1);
-    expect((await ledger.balance("t1")).toCents()).toBe(983); // 1000 - 17
+    const expectedStd = Credit.fromCents(1000).subtract(BASE_COST_CREDIT);
+    expect((await ledger.balance("t1")).toCents()).toBe(expectedStd.toCents());
   });
 
   it("skips storage surcharge when callback not provided (backward compat)", async () => {
@@ -60,11 +63,14 @@ describe("runtime cron with storage tiers", () => {
       getActiveBotCount: async () => 1,
     });
     expect(result.processed).toBe(1);
-    expect((await ledger.balance("t1")).toCents()).toBe(983); // 1000 - 17
+    const expectedBackcompat = Credit.fromCents(1000).subtract(BASE_COST_CREDIT);
+    expect((await ledger.balance("t1")).toCents()).toBe(expectedBackcompat.toCents());
   });
 
   it("suspends tenant when storage surcharge exhausts remaining balance", async () => {
-    await ledger.credit("t1", Credit.fromCents(20), "purchase"); // Only 20 cents
+    // Seed just enough for base cost + 3 cents, so storage surcharge (8) exceeds remainder
+    const seed = BASE_COST_CREDIT.add(Credit.fromCents(3));
+    await ledger.credit("t1", seed, "purchase");
     const suspended: string[] = [];
     const result = await runRuntimeDeductions({
       ledger,
@@ -75,7 +81,7 @@ describe("runtime cron with storage tiers", () => {
         suspended.push(tenantId);
       },
     });
-    // 20 - 17 = 3 remaining, then 8 surcharge > 3, so partial debit + suspend
+    // seed - BASE_COST = 3 remaining, then 8 surcharge > 3, so partial debit + suspend
     expect(result.processed).toBe(1);
     expect(result.suspended).toContain("t1");
     expect((await ledger.balance("t1")).toCents()).toBe(0);