@dimcool/dimclaw 0.1.14 → 0.1.18

package/dim-client.ts

@@ -3,6 +3,8 @@
  * Handles wallet-based authentication using a Solana keypair.
  */
 
+import { mkdir, writeFile, rename } from 'node:fs/promises';
+import path from 'node:path';
 import { SDK, NodeStorage } from '@dimcool/sdk';
 import { Keypair, Transaction } from '@solana/web3.js';
 import bs58 from 'bs58';
@@ -15,6 +17,14 @@ export interface DimClientConfig {
   apiUrl?: string;
   /** Referral code to use on first signup */
   referralCode?: string;
+  /** Path to HEARTBEAT.md for OpenClaw's heartbeat cycle */
+  heartbeatPath?: string;
+}
+
+export interface BufferedEvent {
+  event: string;
+  payload: unknown;
+  at: string;
 }
 
 export class DimClient {
@@ -23,6 +33,8 @@ export class DimClient {
   private config: DimClientConfig;
   private authenticated = false;
   private userId: string | null = null;
+  private eventQueue: BufferedEvent[] = [];
+  private unsubscribers: Array<() => void> = [];
 
   constructor(config: DimClientConfig) {
     this.config = config;
@@ -96,6 +108,106 @@ export class DimClient {
     await this.sdk.ensureWebSocketConnected(timeoutMs);
   }
 
+  /**
+   * Subscribe to key WS events and buffer them for agent consumption.
+   * Call after authenticate() when the WS transport is connected.
+   */
+  startEventListeners(): void {
+    const events = [
+      'chat:message',
+      'notification',
+      'lobby:matched',
+      'lobby:invitation',
+      'game:turn',
+      'game:completed',
+    ];
+    for (const event of events) {
+      this.unsubscribers.push(
+        this.sdk.events.subscribe(event, (payload: unknown) => {
+          this.eventQueue.push({
+            event,
+            payload,
+            at: new Date().toISOString(),
+          });
+          this.writeHeartbeat().catch(() => {});
+        }),
+      );
+    }
+  }
+
+  /** Write HEARTBEAT.md with pending event summary for OpenClaw's heartbeat cycle. */
+  private async writeHeartbeat(): Promise<void> {
+    if (!this.config.heartbeatPath) return;
+    const filePath = this.resolveHeartbeatPath();
+    const count = this.eventQueue.length;
+    if (count === 0) return;
+    const lines = ['# DIM Heartbeat', ''];
+    const eventTypes = new Set(this.eventQueue.map((e) => e.event));
+    if (eventTypes.has('chat:message')) {
+      lines.push('- You have new DMs — call dim_check_notifications');
+    }
+    if (eventTypes.has('notification')) {
+      lines.push(
+        '- New notifications (challenges, friend requests, game results) — call dim_check_notifications',
+      );
+    }
+    if (eventTypes.has('lobby:matched') || eventTypes.has('lobby:invitation')) {
+      lines.push('- A game match is ready — call dim_get_pending_events');
+    }
+    if (eventTypes.has('game:turn')) {
+      lines.push("- It's your turn in a game — call dim_get_pending_events");
+    }
+    if (eventTypes.has('game:completed')) {
+      lines.push('- A game has completed — call dim_get_pending_events');
+    }
+    lines.push('');
+    await mkdir(path.dirname(filePath), { recursive: true });
+    const tmp = `${filePath}.tmp`;
+    await writeFile(tmp, lines.join('\n'), 'utf8');
+    await rename(tmp, filePath);
+  }
+
+  private resolveHeartbeatPath(): string {
+    const p = this.config.heartbeatPath!;
+    if (p.startsWith('~/') && process.env.HOME) {
+      return path.join(process.env.HOME, p.slice(2));
+    }
+    return path.resolve(p);
+  }
+
+  /** Drain all buffered events since last call. */
+  drainEvents(): BufferedEvent[] {
+    return this.eventQueue.splice(0);
+  }
+
+  /** Peek at buffered event count without draining. */
+  get pendingEventCount(): number {
+    return this.eventQueue.length;
+  }
+
+  // ── Daily spend tracking ──────────────────────────────────────────────
+
+  private dailySpendMinor = 0;
+  private spendResetDate = '';
+
+  private resetDailySpendIfNeeded(): void {
+    const today = new Date().toISOString().slice(0, 10);
+    if (this.spendResetDate !== today) {
+      this.dailySpendMinor = 0;
+      this.spendResetDate = today;
+    }
+  }
+
+  recordSpend(amountMinor: number): void {
+    this.resetDailySpendIfNeeded();
+    this.dailySpendMinor += amountMinor;
+  }
+
+  get dailySpentDollars(): number {
+    this.resetDailySpendIfNeeded();
+    return this.dailySpendMinor / 1_000_000;
+  }
+
   getKeypair(): Keypair {
     return this.keypair;
   }