pi-harness-runtime 1.1.375 → 1.1.377

package/scripts/relay-client.ts

@@ -0,0 +1,184 @@
+#!/usr/bin/env bun
+/**
+ * Real-time Callback Relay Client - Two-way streaming
+ * 
+ * Uses the relay package's typed streaming protocol:
+ * - Snapshot + Live event pattern (t3code style)
+ * - Connection state machine with automatic reconnection
+ * - Typed message handling
+ * 
+ * Usage:
+ *   RELAY_WS_URL=wss://webhook.bunchee.online:8444 \
+ *   RELAY_AUTH_TOKEN=change-me-in-production \
+ *   bun run scripts/relay-client.ts
+ */
+
+import { readFileSync, writeFileSync, existsSync } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+import { RelayClient } from "../packages/relay/src/client";
+import type { CallbackPayload, SnapshotMessage } from "../packages/relay/src/types";
+
+// Configuration
+const RELAY_WS_URL = process.env.RELAY_WS_URL || "wss://webhook.bunchee.online:8444";
+const RELAY_AUTH_TOKEN = process.env.RELAY_AUTH_TOKEN || "change-me-in-production";
+const JOB_PIPE_FILE = join(homedir(), ".pi-harness-runtime", "job-commands.jsonl");
+
+// ============================================================================
+// Telegram Integration
+// ============================================================================
+
+function loadKeys() {
+  const keysDir = join(homedir(), ".pi-harness-runtime", "keys");
+  return {
+    telegramBotToken: readFileSync(join(keysDir, "telegram-bot-token.txt"), "utf8").trim(),
+    telegramAllowedUsers: readFileSync(join(keysDir, "telegram-allowed-users.txt"), "utf8").trim(),
+  };
+}
+
+function writeToJobPipe(command: { command: string; jobId: string; userId: number; timestamp: string }): void {
+  const line = JSON.stringify(command) + "\n";
+  writeFileSync(JOB_PIPE_FILE, line, { flag: "a" });
+  console.log(`[Client] Wrote to job pipe: ${command.command} ${command.jobId}`);
+}
+
+async function sendTelegramMessage(chatId: number, text: string): Promise<void> {
+  const keys = loadKeys();
+  await fetch(`https://api.telegram.org/bot${keys.telegramBotToken}/sendMessage`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ chat_id: chatId, text }),
+  });
+}
+
+// ============================================================================
+// Callback Processing
+// ============================================================================
+
+async function processCallback(callback: CallbackPayload): Promise<void> {
+  console.log(`[Client] Processing: ${callback.type} - ${callback.action} ${callback.targetId}`);
+
+  switch (callback.action) {
+    case "approve":
+      writeToJobPipe({
+        command: "resume",
+        jobId: callback.targetId,
+        userId: callback.userId,
+        timestamp: new Date().toISOString(),
+      });
+      await sendTelegramMessage(callback.userId, `Approved! Job ${callback.targetId} will resume.`);
+      break;
+
+    case "reject":
+      writeToJobPipe({
+        command: "cancel",
+        jobId: callback.targetId,
+        userId: callback.userId,
+        timestamp: new Date().toISOString(),
+      });
+      await sendTelegramMessage(callback.userId, `Rejected. Job ${callback.targetId} cancelled.`);
+      break;
+
+    case "sayhi":
+      await sendTelegramMessage(
+        callback.userId,
+        "Hi! Two-way streaming communication working!"
+      );
+      break;
+
+    default:
+      console.log(`[Client] Unknown action: ${callback.action}`);
+  }
+}
+
+// ============================================================================
+// Main
+// ============================================================================
+
+async function main() {
+  console.log("===========================================");
+  console.log("Real-time Callback Relay Client - Streaming v2");
+  console.log("===========================================");
+  console.log(`Server: ${RELAY_WS_URL}`);
+  console.log("===========================================");
+
+  // Create client with typed callbacks
+  const client = new RelayClient(
+    {
+      serverUrl: RELAY_WS_URL,
+      token: RELAY_AUTH_TOKEN,
+      keepAlive: 30000,
+    },
+    {
+      onConnected: (sessionId) => {
+        console.log(`[Client] Connected! Session: ${sessionId}`);
+      },
+
+      onDisconnected: () => {
+        console.log("[Client] Disconnected");
+      },
+
+      onSnapshot: (snapshot: SnapshotMessage["payload"]) => {
+        console.log(`[Client] Snapshot received:`);
+        console.log(`  - Session: ${snapshot.sessionId}`);
+        console.log(`  - Version: ${snapshot.serverVersion}`);
+        console.log(`  - Connected clients: ${snapshot.connectedClients}`);
+        console.log(`  - Recent callbacks: ${snapshot.recentCallbacks.length}`);
+        
+        // Process any recent callbacks from snapshot
+        for (const callback of snapshot.recentCallbacks) {
+          processCallback(callback).catch(console.error);
+        }
+      },
+
+      onSynchronized: () => {
+        console.log("[Client] Synchronized - receiving live events");
+      },
+
+      onCallback: (callback: CallbackPayload) => {
+        console.log(`[Client] Live callback: ${callback.action} ${callback.targetId}`);
+        processCallback(callback).catch(console.error);
+      },
+
+      onStatus: (status) => {
+        console.log(`[Client] Status [${status.kind}]: ${status.code} - ${status.message}`);
+      },
+
+      onError: (error) => {
+        console.error("[Client] Error:", error.message);
+      },
+
+      onStateChange: (change) => {
+        const prev = change.previous.state;
+        const curr = change.current.state;
+        if (prev !== curr) {
+          console.log(`[Client] State: ${prev} → ${curr}`);
+        }
+      },
+    }
+  );
+
+  // Handle graceful shutdown
+  const shutdown = () => {
+    console.log("\nShutting down...");
+    client.disconnect();
+    process.exit(0);
+  };
+
+  process.on("SIGINT", shutdown);
+  process.on("SIGTERM", shutdown);
+
+  // Connect
+  try {
+    await client.connect();
+    console.log("[Client] Connection established!");
+    
+    // Keep running
+    await new Promise(() => {});
+  } catch (err) {
+    console.error("[Client] Failed to connect:", err);
+    process.exit(1);
+  }
+}
+
+main();

package/scripts/sync-callback-queue.ts

@@ -0,0 +1,57 @@
+#!/usr/bin/env bun
+/**
+ * Syncs callback queue from VPS to local for processing.
+ * Run this alongside telegram-callback-processor.ts
+ * 
+ * Usage:
+ *   bun run scripts/sync-callback-queue.ts
+ * 
+ * Or set up a cron job:
+ *   * * * * * bun /home/frappe/pi-harness-runtime/scripts/sync-callback-queue.ts
+ */
+
+import { execSync } from "child_process";
+import { readFileSync, writeFileSync, existsSync } from "fs";
+import { join } from "path";
+
+const VPS_HOST = process.env.VPS_HOST || "frappe@217.216.37.251";
+const VPS_QUEUE = "/home/frappe/.pi-harness-runtime/callback-queue.jsonl";
+const LOCAL_QUEUE = join(process.env.HOME || "/home/frappe", ".pi-harness-runtime", "callback-queue.jsonl");
+
+function syncQueue(): number {
+  try {
+    // SSH to get VPS queue content
+    const vpsQueueContent = execSync(`ssh -o BatchMode=yes ${VPS_HOST} "cat ${VPS_QUEUE}"`, { encoding: "utf8" });
+
+    if (!vpsQueueContent.trim()) {
+      return 0; // Nothing to sync
+    }
+
+    // Append to local queue
+    const existing = existsSync(LOCAL_QUEUE) ? readFileSync(LOCAL_QUEUE, "utf8") : "";
+    const newQueue = existing + vpsQueueContent;
+    writeFileSync(LOCAL_QUEUE, newQueue);
+
+    // Clear VPS queue
+    execSync(`ssh -o BatchMode=yes ${VPS_HOST} "truncate -s 0 ${VPS_QUEUE}"`);
+
+    const count = vpsQueueContent.split("\n").filter(l => l.trim()).length;
+    if (count > 0) {
+      console.log(`[Sync] Synced ${count} callback(s) from VPS`);
+    }
+    return count;
+  } catch (err) {
+    console.error("[Sync] Error:", err);
+    return 0;
+  }
+}
+
+// Run once and exit (for cron) unless --watch is passed
+const isWatch = process.argv.includes("--watch");
+
+if (isWatch) {
+  console.log("[Sync] Watching VPS queue every 500ms...");
+  setInterval(syncQueue, 500);
+} else {
+  syncQueue();
+}

package/scripts/telegram-callback-processor.ts

@@ -89,6 +89,22 @@ export async function processCallbacks(): Promise<number> {
           break;
         }
 
+        case "sayhi": {
+          console.log(`👋 SAYHI: ${item.targetId} from @${item.username || item.userId}`);
+          // Send greeting response via Telegram
+          const botToken = process.env.TELEGRAM_BOT_TOKEN || readFileSync(join(process.env.HOME || "", ".pi-harness-runtime", "keys", "telegram-bot-token.txt"), "utf-8").trim();
+          const chatId = item.userId;
+          await fetch(`https://api.telegram.org/bot${botToken}/sendMessage`, {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify({
+              chat_id: chatId,
+              text: "👋 Hi! Two-way communication is working! Click buttons to interact."
+            })
+          });
+          break;
+        }
+
         default: {
           console.log(`❓ UNKNOWN: ${item.action}_${item.targetId}`);
           if (onUnknown) {