@vgai/p2p-colyseus 0.5.4 → 0.5.6

package/package.json

@@ -2,7 +2,7 @@
   "name": "@vgai/p2p-colyseus",
   "author": "Volter AI, Inc.",
   "license": "Apache-2.0",
-  "version": "0.5.4",
+  "version": "0.5.6",
   "type": "module",
   "description": "Colyseus-compatible transport for vgai — real Colyseus, loopback, WebRTC, or relay, chosen at runtime.",
   "homepage": "https://github.com/volter-ai/vgai-engine#readme",

package/src/cloudflare/durable-room.ts

@@ -18,6 +18,8 @@ interface DurableObjectStateLike {
 }
 
 interface DurableRoomEnv {
+  VGAI_BILLING?: { fetch(request: Request): Promise<Response> };
+  VGAI_BILLING_SERVICE_TOKEN?: string;
   P2P_COLYSEUS_RELAY_DISABLED?: string | undefined;
   P2P_COLYSEUS_RELAY_DAILY_BUDGET_USD?: string | undefined;
   P2P_COLYSEUS_RELAY_ESTIMATED_USD_PER_GIB?: string | undefined;
@@ -428,14 +430,75 @@ export class P2PColyseusDurableRoom {
     bytes: number,
   ): Promise<ErrorSignalEnvelope | null> {
     if (!principal) return null;
-    return this.consumeDailyCounter(
+    const cap = await this.consumeDailyCounter(
       `relay-account-bytes:${principal}`,
       bytes,
       getFreeRelayBytesPerAccountPerDay(this.env),
     );
+    if (cap || !getRequireAccount(this.env)) return cap;
+    if (!this.env.VGAI_BILLING) {
+      return { kind: 'error', message: String(P2P_CLOSE_CODES.relayQuotaExceeded) };
+    }
+    const serviceToken = this.env.VGAI_BILLING_SERVICE_TOKEN?.trim();
+    if (!serviceToken) {
+      return { kind: 'error', message: String(P2P_CLOSE_CODES.relayQuotaExceeded) };
+    }
+    const day = dayKey(Date.now());
+    const offset =
+      (await this.state?.storage?.get<number>(`relay-account-bytes:${principal}:${day}`)) ?? bytes;
+    const authorization = await this.env.VGAI_BILLING.fetch(
+      new Request('https://billing.internal/usage/authorize', {
+        method: 'POST',
+        headers: billingHeaders(serviceToken),
+        body: JSON.stringify({
+          userId: principal,
+          kind: 'relay',
+          provider: 'cloudflare',
+          operation: 'relay-bytes',
+          operationId: `relay:${principal}:${day}:${offset}`,
+          providerPrice: { relayBytes: bytes },
+        }),
+      }),
+    );
+    const authorizationBody = (await authorization.json().catch(() => undefined)) as
+      | { reservationId?: unknown }
+      | undefined;
+    if (!authorization.ok || typeof authorizationBody?.reservationId !== 'string') {
+      return { kind: 'error', message: String(P2P_CLOSE_CODES.relayQuotaExceeded) };
+    }
+    const settled = await this.env.VGAI_BILLING.fetch(
+      new Request('https://billing.internal/usage/settle', {
+        method: 'POST',
+        headers: billingHeaders(serviceToken),
+        body: JSON.stringify({
+          reservationId: authorizationBody.reservationId,
+          kind: 'relay',
+          provider: 'cloudflare',
+          operation: 'relay-bytes',
+          providerUsage: { relayBytes: bytes },
+          externalId: `relay:${principal}:${day}:${offset}`,
+        }),
+      }),
+    );
+    if (settled.ok) return null;
+    await this.env.VGAI_BILLING.fetch(
+      new Request('https://billing.internal/usage/release', {
+        method: 'POST',
+        headers: billingHeaders(serviceToken),
+        body: JSON.stringify({
+          reservationId: authorizationBody.reservationId,
+          reason: 'relay_settlement_failed',
+        }),
+      }),
+    );
+    return { kind: 'error', message: String(P2P_CLOSE_CODES.relayQuotaExceeded) };
   }
 }
 
+function billingHeaders(token: string): HeadersInit {
+  return { 'Content-Type': 'application/json', 'X-VGAI-Service-Token': token };
+}
+
 function unauthorized(): ErrorSignalEnvelope {
   return { kind: 'error', message: String(P2P_CLOSE_CODES.unauthorized) };
 }