beehaven 0.8.16

package/dist/commands/explore.js

@@ -0,0 +1,264 @@
+/**
+ * beehaven explore — Inspect live Durable Object data.
+ *
+ * Subcommands:
+ *   beehaven explore ls                              List active DO types/instances
+ *   beehaven explore <cell-type> <cell-id> "SQL"     Run SQL against a DO
+ *   beehaven explore watch <cell-type> <cell-id>     Live tail of deltas (WebSocket)
+ */
+import WebSocket from 'ws';
+const DEFAULT_PORT = 8787;
+function baseUrl(port) {
+    return `http://127.0.0.1:${port}`;
+}
+function wsUrl(port) {
+    return `ws://127.0.0.1:${port}`;
+}
+// ── Helpers ──────────────────────────────────────────────────────────────
+async function fetchJSON(url) {
+    const res = await fetch(url);
+    if (!res.ok) {
+        const body = await res.text().catch(() => '');
+        throw new Error(`${res.status} ${res.statusText}${body ? ': ' + body : ''}`);
+    }
+    return res.json();
+}
+async function postJSON(url, body) {
+    const res = await fetch(url, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify(body),
+    });
+    if (!res.ok) {
+        const text = await res.text().catch(() => '');
+        throw new Error(`${res.status} ${res.statusText}${text ? ': ' + text : ''}`);
+    }
+    return res.json();
+}
+function printTable(rows) {
+    if (rows.length === 0) {
+        console.log('  (no rows)');
+        return;
+    }
+    const cols = Object.keys(rows[0]);
+    const widths = cols.map((col) => {
+        const max = rows.reduce((w, row) => {
+            const val = String(row[col] ?? '');
+            return Math.max(w, val.length);
+        }, col.length);
+        return Math.min(max, 40); // cap column width
+    });
+    // Header
+    const header = cols.map((c, i) => c.padEnd(widths[i])).join('  ');
+    const separator = widths.map((w) => '─'.repeat(w)).join('──');
+    console.log(`  ${header}`);
+    console.log(`  ${separator}`);
+    // Rows
+    for (const row of rows) {
+        const line = cols
+            .map((c, i) => {
+            const val = String(row[c] ?? '');
+            return val.length > widths[i] ? val.slice(0, widths[i] - 1) + '…' : val.padEnd(widths[i]);
+        })
+            .join('  ');
+        console.log(`  ${line}`);
+    }
+    console.log(`\n  ${rows.length} row(s)`);
+}
+// ── ls ───────────────────────────────────────────────────────────────────
+async function listInstances(port, cellType) {
+    if (cellType) {
+        // List instances of a specific cell type
+        const url = `${baseUrl(port)}/cell/${cellType}/instances`;
+        try {
+            const data = await fetchJSON(url);
+            const instances = Array.isArray(data) ? data : (data.instances ?? []);
+            if (instances.length === 0) {
+                console.log(`\n  No active instances for "${cellType}".\n`);
+                return;
+            }
+            console.log(`\n  Active instances of "${cellType}":\n`);
+            for (const id of instances) {
+                console.log(`    ${id}`);
+            }
+            console.log(`\n  ${instances.length} instance(s)\n`);
+        }
+        catch (err) {
+            console.error(`\n  Failed to list instances: ${err.message}\n`);
+            process.exit(1);
+        }
+    }
+    else {
+        // List all cell types — hit the health endpoint or /cell list
+        // Try listing known cell types from the dev server
+        const url = `${baseUrl(port)}/health`;
+        try {
+            const data = await fetchJSON(url);
+            const cells = data.cells ?? data.cellTypes ?? [];
+            if (cells.length === 0) {
+                console.log('\n  No cell types registered. Is the dev server running?\n');
+                return;
+            }
+            console.log('\n  Registered cell types:\n');
+            for (const cell of cells) {
+                // Try to get instance count
+                try {
+                    const instances = await fetchJSON(`${baseUrl(port)}/cell/${cell}/instances`);
+                    const count = Array.isArray(instances)
+                        ? instances.length
+                        : (instances.instances?.length ?? '?');
+                    console.log(`    ${cell}  (${count} instance${count !== 1 ? 's' : ''})`);
+                }
+                catch {
+                    console.log(`    ${cell}`);
+                }
+            }
+            console.log('');
+        }
+        catch (err) {
+            console.error(`\n  Could not reach dev server at 127.0.0.1:${port}.`);
+            console.error(`  Is "beehaven dev" running? (${err.message})\n`);
+            process.exit(1);
+        }
+    }
+}
+// ── SQL query ────────────────────────────────────────────────────────────
+async function runQuery(port, cellType, cellId, sql) {
+    const url = `${baseUrl(port)}/cell/${cellType}/${cellId}/sql/raw`;
+    try {
+        const data = await postJSON(url, { sql });
+        const rows = data.rows ?? data.results ?? data;
+        if (!Array.isArray(rows)) {
+            console.log('\n  Response:');
+            console.log(JSON.stringify(data, null, 2));
+            return;
+        }
+        console.log('');
+        printTable(rows);
+        console.log('');
+    }
+    catch (err) {
+        console.error(`\n  Query failed: ${err.message}\n`);
+        process.exit(1);
+    }
+}
+// ── watch (live delta tail) ──────────────────────────────────────────────
+function watchDeltas(port, cellType, cellId) {
+    const url = `${wsUrl(port)}/cell/${cellType}/${cellId}/ws`;
+    console.log(`\n  Connecting to ${url}...`);
+    const ws = new WebSocket(url);
+    let connected = false;
+    ws.on('open', () => {
+        connected = true;
+        console.log('  Connected. Streaming deltas (Ctrl+C to stop):\n');
+    });
+    ws.on('message', (data) => {
+        const raw = data.toString();
+        try {
+            const msg = JSON.parse(raw);
+            const ts = new Date().toISOString().slice(11, 23);
+            if (msg.type === 'sql:delta') {
+                const { table, op, id, delta, row } = msg;
+                const preview = delta ? JSON.stringify(delta) : row ? JSON.stringify(row) : '';
+                console.log(`  [${ts}] ${op.toUpperCase().padEnd(6)} ${table}/${id}  ${truncate(preview, 80)}`);
+            }
+            else if (msg.type === 'sql:hydration') {
+                const tables = msg.tables
+                    ? Object.keys(msg.tables).map((t) => `${t}(${msg.tables[t]?.length ?? '?'})`)
+                    : [];
+                console.log(`  [${ts}] HYDRATE ${tables.join(', ')}`);
+            }
+            else if (msg.type === 'sql:batch') {
+                console.log(`  [${ts}] BATCH   ${msg.deltas?.length ?? '?'} delta(s)`);
+            }
+            else if (msg.type === 'presence:update' || msg.type === 'presence:leave') {
+                console.log(`  [${ts}] ${msg.type.toUpperCase().padEnd(16)} ${msg.peerId ?? ''} ${JSON.stringify(msg.state ?? {})}`);
+            }
+            else {
+                // Generic message
+                console.log(`  [${ts}] ${msg.type ?? 'MSG'}  ${truncate(raw, 100)}`);
+            }
+        }
+        catch {
+            // Non-JSON message
+            const ts = new Date().toISOString().slice(11, 23);
+            console.log(`  [${ts}] RAW  ${truncate(raw, 100)}`);
+        }
+    });
+    ws.on('error', (err) => {
+        if (!connected) {
+            console.error(`\n  Could not connect to ${url}`);
+            console.error(`  Is "beehaven dev" running? (${err.message})\n`);
+            process.exit(1);
+        }
+        console.error(`  [error] ${err.message}`);
+    });
+    ws.on('close', (code, reason) => {
+        console.log(`\n  Connection closed (${code}${reason.length ? ': ' + reason.toString() : ''}).\n`);
+        process.exit(0);
+    });
+    // Clean shutdown
+    const cleanup = () => {
+        console.log('\n  Disconnecting...');
+        ws.close();
+        setTimeout(() => process.exit(0), 500);
+    };
+    process.on('SIGINT', cleanup);
+    process.on('SIGTERM', cleanup);
+}
+function truncate(s, max) {
+    return s.length > max ? s.slice(0, max - 1) + '…' : s;
+}
+// ── Main entry point ─────────────────────────────────────────────────────
+export async function explore(args, options = {}) {
+    const port = options.port || DEFAULT_PORT;
+    const sub = args[0];
+    if (!sub) {
+        printUsage();
+        return;
+    }
+    // beehaven explore ls [cell-type]
+    if (sub === 'ls') {
+        await listInstances(port, args[1]);
+        return;
+    }
+    // beehaven explore watch <cell-type> <cell-id>
+    if (sub === 'watch') {
+        const cellType = args[1];
+        const cellId = args[2];
+        if (!cellType || !cellId) {
+            console.error('\n  Usage: beehaven explore watch <cell-type> <cell-id>\n');
+            process.exit(1);
+        }
+        watchDeltas(port, cellType, cellId);
+        return;
+    }
+    // beehaven explore <cell-type> <cell-id> "SQL query"
+    const cellType = args[0];
+    const cellId = args[1];
+    const sql = args[2];
+    if (!cellId || !sql) {
+        console.error('\n  Usage: beehaven explore <cell-type> <cell-id> "SELECT * FROM table"\n');
+        process.exit(1);
+    }
+    await runQuery(port, cellType, cellId, sql);
+}
+function printUsage() {
+    console.log(`
+  beehaven explore — Inspect live Durable Object data
+
+  Usage:
+    beehaven explore ls [cell-type]                  List cell types or instances
+    beehaven explore <type> <id> "SQL"               Run SQL against a DO
+    beehaven explore watch <type> <id>               Live tail of deltas
+
+  Options:
+    --port <n>    Dev server port (default: 8787)
+
+  Examples:
+    beehaven explore ls                              List all cell types
+    beehaven explore ls notes                        List instances of "notes"
+    beehaven explore notes room-1 "SELECT * FROM notes"
+    beehaven explore watch notes room-1              Stream deltas in real time
+`);
+}