bonescript-compiler 0.5.3 → 0.5.4

package/src/emit_websocket.ts

@@ -1,226 +1,229 @@
-/**
- * BoneScript WebSocket Runtime Emitter
- * Generates runnable WebSocket servers for `channel` declarations.
- */
-
-import * as IR from "./ir";
-import { toSnakeCase } from "./lowering_helpers";
-
-function toCamelCase(s: string): string {
-  return s.replace(/_([a-z])/g, (_, c) => c.toUpperCase());
-}
-
-// ─── WebSocket Server ────────────────────────────────────────────────────────
-
-export function emitWebSocketServer(system: IR.IRSystem): string {
-  const channels = system.modules.filter(m => m.kind === "realtime_service");
-  if (channels.length === 0) return "";
-
-  const lines: string[] = [];
-  lines.push(`// Generated by BoneScript compiler. DO NOT EDIT.`);
-  lines.push(`// WebSocket server for realtime channels`);
-  lines.push(``);
-  lines.push(`import { WebSocketServer, WebSocket } from "ws";`);
-  lines.push(`import { IncomingMessage, Server } from "http";`);
-  lines.push(`import jwt from "jsonwebtoken";`);
-  lines.push(`import { eventBus } from "./events";`);
-  lines.push(`import { logger } from "./logger";`);
-  lines.push(``);
-  lines.push(`const JWT_SECRET = process.env.JWT_SECRET || "bonescript-dev-secret-change-in-production";`);
-  lines.push(``);
-  // Redis pub/sub for multi-instance support
-  lines.push(`// Redis pub/sub for multi-instance WebSocket broadcasting`);
-  lines.push(`let redisSub: any = null;`);
-  lines.push(`let redisPub: any = null;`);
-  lines.push(`if (process.env.REDIS_URL) {`);
-  lines.push(`  try {`);
-  lines.push(`    const { createClient } = require("redis");`);
-  lines.push(`    redisSub = createClient({ url: process.env.REDIS_URL });`);
-  lines.push(`    redisPub = createClient({ url: process.env.REDIS_URL });`);
-  lines.push(`    Promise.all([redisSub.connect(), redisPub.connect()]).then(() => {`);
-  lines.push(`      logger.info("redis_connected", { event: "startup" });`);
-  lines.push(`    }).catch((e: any) => {`);
-  lines.push(`      logger.error("redis_connect_failed", { event: "startup", metadata: { error: e.message } });`);
-  lines.push(`      redisSub = null; redisPub = null;`);
-  lines.push(`    });`);
-  lines.push(`  } catch { redisSub = null; redisPub = null; }`);
-  lines.push(`}`);
-
-  // Per-channel client registry
-  lines.push(`interface Client {`);
-  lines.push(`  socket: WebSocket;`);
-  lines.push(`  user_id: string;`);
-  lines.push(`  channel: string;`);
-  lines.push(`  topics: Set<string>;`);
-  lines.push(`}`);
-  lines.push(``);
-  lines.push(`const clients: Map<string, Set<Client>> = new Map();`);
-  lines.push(``);
-
-  // Channel configs
-  lines.push(`const CHANNELS: Record<string, { ordering: string; persistence: string; max_size: number }> = {`);
-  for (const ch of channels) {
-    lines.push(`  "${ch.name}": {`);
-    lines.push(`    ordering: "${ch.config["ordering"] || "fifo"}",`);
-    lines.push(`    persistence: "${ch.config["persistence"] || "none"}",`);
-    lines.push(`    max_size: ${ch.config["max_size"] || 10000},`);
-    lines.push(`  },`);
-  }
-  lines.push(`};`);
-  lines.push(``);
-
-  // Message buffer for persistence
-  lines.push(`const messageBuffers: Map<string, any[]> = new Map();`);
-  lines.push(``);
-  lines.push(`function getBuffer(channel: string): any[] {`);
-  lines.push(`  const buf = messageBuffers.get(channel);`);
-  lines.push(`  if (buf) return buf;`);
-  lines.push(`  const fresh: any[] = [];`);
-  lines.push(`  messageBuffers.set(channel, fresh);`);
-  lines.push(`  return fresh;`);
-  lines.push(`}`);
-  lines.push(``);
-  lines.push(`function persistMessage(channel: string, msg: any) {`);
-  lines.push(`  const cfg = CHANNELS[channel];`);
-  lines.push(`  if (!cfg || cfg.persistence === "none") return;`);
-  lines.push(`  const buf = getBuffer(channel);`);
-  lines.push(`  buf.push(msg);`);
-  lines.push(``);
-  lines.push(`  // Honor persistence config (last_N, full)`);
-  lines.push(`  const match = cfg.persistence.match(/^last_(\\d+)$/);`);
-  lines.push(`  if (match) {`);
-  lines.push(`    const limit = parseInt(match[1], 10);`);
-  lines.push(`    while (buf.length > limit) buf.shift();`);
-  lines.push(`  } else if (buf.length > cfg.max_size) {`);
-  lines.push(`    buf.shift();`);
-  lines.push(`  }`);
-  lines.push(`}`);
-  lines.push(``);
-
-  // Setup function
-  lines.push(`export function setupWebSocketServer(httpServer: Server): WebSocketServer {`);
-  lines.push(`  const wss = new WebSocketServer({ server: httpServer, path: "/ws" });`);
-  lines.push(``);
-  lines.push(`  wss.on("connection", (socket: WebSocket, req: IncomingMessage) => {`);
-  lines.push(`    const url = new URL(req.url || "/", \`http://\${req.headers.host}\`);`);
-  lines.push(`    const channel = url.searchParams.get("channel") || "";`);
-  lines.push(`    const token = url.searchParams.get("token") || "";`);
-  lines.push(``);
-  lines.push(`    if (!CHANNELS[channel]) {`);
-  lines.push(`      socket.send(JSON.stringify({ type: "error", message: "Unknown channel: " + channel }));`);
-  lines.push(`      socket.close();`);
-  lines.push(`      return;`);
-  lines.push(`    }`);
-  lines.push(``);
-  lines.push(`    let userId: string;`);
-  lines.push(`    try {`);
-  lines.push(`      const decoded = jwt.verify(token, JWT_SECRET) as { sub: string };`);
-  lines.push(`      userId = decoded.sub;`);
-  lines.push(`    } catch {`);
-  lines.push(`      socket.send(JSON.stringify({ type: "error", message: "Authentication failed" }));`);
-  lines.push(`      socket.close();`);
-  lines.push(`      return;`);
-  lines.push(`    }`);
-  lines.push(``);
-  lines.push(`    const client: Client = { socket, user_id: userId, channel, topics: new Set() };`);
-  lines.push(`    const set = clients.get(channel) || new Set();`);
-  lines.push(`    set.add(client);`);
-  lines.push(`    clients.set(channel, set);`);
-  lines.push(``);
-  lines.push(`    console.log(\`[ws] User \${userId} connected to \${channel} (\${set.size} active)\`);`);
-  lines.push(``);
-  lines.push(`    // Send buffered history`);
-  lines.push(`    const history = messageBuffers.get(channel) || [];`);
-  lines.push(`    for (const msg of history) {`);
-  lines.push(`      socket.send(JSON.stringify(msg));`);
-  lines.push(`    }`);
-  lines.push(``);
-  lines.push(`    socket.send(JSON.stringify({ type: "connected", channel, history_size: history.length }));`);
-  lines.push(``);
-  lines.push(`    socket.on("message", (data) => {`);
-  lines.push(`      try {`);
-  lines.push(`        const msg = JSON.parse(data.toString());`);
-  lines.push(``);
-  lines.push(`        if (msg.type === "subscribe" && msg.topic) {`);
-  lines.push(`          client.topics.add(msg.topic);`);
-  lines.push(`          return;`);
-  lines.push(`        }`);
-  lines.push(`        if (msg.type === "unsubscribe" && msg.topic) {`);
-  lines.push(`          client.topics.delete(msg.topic);`);
-  lines.push(`          return;`);
-  lines.push(`        }`);
-  lines.push(``);
-  lines.push(`        // Broadcast`);
-  lines.push(`        const broadcast = {`);
-  lines.push(`          type: msg.type || "message",`);
-  lines.push(`          payload: msg.payload || msg,`);
-  lines.push(`          from: userId,`);
-  lines.push(`          timestamp: new Date().toISOString(),`);
-  lines.push(`        };`);
-  lines.push(``);
-  lines.push(`        persistMessage(channel, broadcast);`);
-  lines.push(`        broadcastToChannel(channel, broadcast, client);`);
-  lines.push(`      } catch (e: any) {`);
-  lines.push(`        socket.send(JSON.stringify({ type: "error", message: e.message }));`);
-  lines.push(`      }`);
-  lines.push(`    });`);
-  lines.push(``);
-  lines.push(`    socket.on("close", () => {`);
-  lines.push(`      const set = clients.get(channel);`);
-  lines.push(`      if (set) set.delete(client);`);
-  lines.push(`      console.log(\`[ws] User \${userId} disconnected from \${channel}\`);`);
-  lines.push(`    });`);
-  lines.push(``);
-  lines.push(`    // Heartbeat`);
-  lines.push(`    const heartbeat = setInterval(() => {`);
-  lines.push(`      if (socket.readyState === WebSocket.OPEN) {`);
-  lines.push(`        socket.ping();`);
-  lines.push(`      } else {`);
-  lines.push(`        clearInterval(heartbeat);`);
-  lines.push(`      }`);
-  lines.push(`    }, 30000);`);
-  lines.push(`  });`);
-  lines.push(``);
-  lines.push(`  // Bridge: forward eventBus events to WebSocket clients`);
-  for (const ch of channels) {
-    lines.push(`  // Channel '${ch.name}' bridge`);
-  }
-  lines.push(``);
-  // Redis subscription for cross-instance delivery
-  lines.push(`  // Subscribe to Redis channels for cross-instance delivery`);
-  lines.push(`  if (redisSub) {`);
-  for (const ch of channels) {
-    lines.push(`    redisSub.subscribe("ws:${ch.name}", (message: string) => {`);
-    lines.push(`      const set = clients.get("${ch.name}");`);
-    lines.push(`      if (!set) return;`);
-    lines.push(`      for (const client of set) {`);
-    lines.push(`        if (client.socket.readyState === WebSocket.OPEN) client.socket.send(message);`);
-    lines.push(`      }`);
-    lines.push(`    }).catch(() => {});`);
-  }
-  lines.push(`  }`);
-  lines.push(`  return wss;`);
-  lines.push(`}`);
-  lines.push(``);
-  lines.push(`function broadcastToChannel(channel: string, message: any, exclude?: Client) {`);
-  lines.push(`  const set = clients.get(channel);`);
-  lines.push(`  const data = JSON.stringify(message);`);
-  lines.push(`  // Broadcast to local clients`);
-  lines.push(`  if (set) {`);
-  lines.push(`    for (const client of set) {`);
-  lines.push(`      if (client === exclude) continue;`);
-  lines.push(`      if (client.socket.readyState === WebSocket.OPEN) {`);
-  lines.push(`        client.socket.send(data);`);
-  lines.push(`      }`);
-  lines.push(`    }`);
-  lines.push(`  }`);
-  lines.push(`  // Publish to Redis for cross-instance delivery`);
-  lines.push(`  if (redisPub) {`);
-  lines.push(`    redisPub.publish(\`ws:\${channel}\`, data).catch(() => {});`);
-  lines.push(`  }`);
-  lines.push(`}`);
-  lines.push(``);
-  lines.push(`export { broadcastToChannel };`);
-
-  return lines.join("\n");
-}
+/**
+ * BoneScript WebSocket Runtime Emitter
+ * Generates runnable WebSocket servers for `channel` declarations.
+ */
+
+import * as IR from "./ir";
+
+function toSnakeCase(s: string): string {
+  return s.replace(/([a-z])([A-Z])/g, "$1_$2").toLowerCase();
+}
+
+function toCamelCase(s: string): string {
+  return s.replace(/_([a-z])/g, (_, c) => c.toUpperCase());
+}
+
+// ─── WebSocket Server ────────────────────────────────────────────────────────
+
+export function emitWebSocketServer(system: IR.IRSystem): string {
+  const channels = system.modules.filter(m => m.kind === "realtime_service");
+  if (channels.length === 0) return "";
+
+  const lines: string[] = [];
+  lines.push(`// Generated by BoneScript compiler. DO NOT EDIT.`);
+  lines.push(`// WebSocket server for realtime channels`);
+  lines.push(``);
+  lines.push(`import { WebSocketServer, WebSocket } from "ws";`);
+  lines.push(`import { IncomingMessage, Server } from "http";`);
+  lines.push(`import jwt from "jsonwebtoken";`);
+  lines.push(`import { eventBus } from "./events";`);
+  lines.push(`import { logger } from "./logger";`);
+  lines.push(``);
+  lines.push(`const JWT_SECRET = process.env.JWT_SECRET || "bonescript-dev-secret-change-in-production";`);
+  lines.push(``);
+  // Redis pub/sub for multi-instance support
+  lines.push(`// Redis pub/sub for multi-instance WebSocket broadcasting`);
+  lines.push(`let redisSub: any = null;`);
+  lines.push(`let redisPub: any = null;`);
+  lines.push(`if (process.env.REDIS_URL) {`);
+  lines.push(`  try {`);
+  lines.push(`    const { createClient } = require("redis");`);
+  lines.push(`    redisSub = createClient({ url: process.env.REDIS_URL });`);
+  lines.push(`    redisPub = createClient({ url: process.env.REDIS_URL });`);
+  lines.push(`    Promise.all([redisSub.connect(), redisPub.connect()]).then(() => {`);
+  lines.push(`      logger.info("redis_connected", { event: "startup" });`);
+  lines.push(`    }).catch((e: any) => {`);
+  lines.push(`      logger.error("redis_connect_failed", { event: "startup", metadata: { error: e.message } });`);
+  lines.push(`      redisSub = null; redisPub = null;`);
+  lines.push(`    });`);
+  lines.push(`  } catch { redisSub = null; redisPub = null; }`);
+  lines.push(`}`);
+
+  // Per-channel client registry
+  lines.push(`interface Client {`);
+  lines.push(`  socket: WebSocket;`);
+  lines.push(`  user_id: string;`);
+  lines.push(`  channel: string;`);
+  lines.push(`  topics: Set<string>;`);
+  lines.push(`}`);
+  lines.push(``);
+  lines.push(`const clients: Map<string, Set<Client>> = new Map();`);
+  lines.push(``);
+
+  // Channel configs
+  lines.push(`const CHANNELS: Record<string, { ordering: string; persistence: string; max_size: number }> = {`);
+  for (const ch of channels) {
+    lines.push(`  "${ch.name}": {`);
+    lines.push(`    ordering: "${ch.config["ordering"] || "fifo"}",`);
+    lines.push(`    persistence: "${ch.config["persistence"] || "none"}",`);
+    lines.push(`    max_size: ${ch.config["max_size"] || 10000},`);
+    lines.push(`  },`);
+  }
+  lines.push(`};`);
+  lines.push(``);
+
+  // Message buffer for persistence
+  lines.push(`const messageBuffers: Map<string, any[]> = new Map();`);
+  lines.push(``);
+  lines.push(`function getBuffer(channel: string): any[] {`);
+  lines.push(`  const buf = messageBuffers.get(channel);`);
+  lines.push(`  if (buf) return buf;`);
+  lines.push(`  const fresh: any[] = [];`);
+  lines.push(`  messageBuffers.set(channel, fresh);`);
+  lines.push(`  return fresh;`);
+  lines.push(`}`);
+  lines.push(``);
+  lines.push(`function persistMessage(channel: string, msg: any) {`);
+  lines.push(`  const cfg = CHANNELS[channel];`);
+  lines.push(`  if (!cfg || cfg.persistence === "none") return;`);
+  lines.push(`  const buf = getBuffer(channel);`);
+  lines.push(`  buf.push(msg);`);
+  lines.push(``);
+  lines.push(`  // Honor persistence config (last_N, full)`);
+  lines.push(`  const match = cfg.persistence.match(/^last_(\\d+)$/);`);
+  lines.push(`  if (match) {`);
+  lines.push(`    const limit = parseInt(match[1], 10);`);
+  lines.push(`    while (buf.length > limit) buf.shift();`);
+  lines.push(`  } else if (buf.length > cfg.max_size) {`);
+  lines.push(`    buf.shift();`);
+  lines.push(`  }`);
+  lines.push(`}`);
+  lines.push(``);
+
+  // Setup function
+  lines.push(`export function setupWebSocketServer(httpServer: Server): WebSocketServer {`);
+  lines.push(`  const wss = new WebSocketServer({ server: httpServer, path: "/ws" });`);
+  lines.push(``);
+  lines.push(`  wss.on("connection", (socket: WebSocket, req: IncomingMessage) => {`);
+  lines.push(`    const url = new URL(req.url || "/", \`http://\${req.headers.host}\`);`);
+  lines.push(`    const channel = url.searchParams.get("channel") || "";`);
+  lines.push(`    const token = url.searchParams.get("token") || "";`);
+  lines.push(``);
+  lines.push(`    if (!CHANNELS[channel]) {`);
+  lines.push(`      socket.send(JSON.stringify({ type: "error", message: "Unknown channel: " + channel }));`);
+  lines.push(`      socket.close();`);
+  lines.push(`      return;`);
+  lines.push(`    }`);
+  lines.push(``);
+  lines.push(`    let userId: string;`);
+  lines.push(`    try {`);
+  lines.push(`      const decoded = jwt.verify(token, JWT_SECRET) as { sub: string };`);
+  lines.push(`      userId = decoded.sub;`);
+  lines.push(`    } catch {`);
+  lines.push(`      socket.send(JSON.stringify({ type: "error", message: "Authentication failed" }));`);
+  lines.push(`      socket.close();`);
+  lines.push(`      return;`);
+  lines.push(`    }`);
+  lines.push(``);
+  lines.push(`    const client: Client = { socket, user_id: userId, channel, topics: new Set() };`);
+  lines.push(`    const set = clients.get(channel) || new Set();`);
+  lines.push(`    set.add(client);`);
+  lines.push(`    clients.set(channel, set);`);
+  lines.push(``);
+  lines.push(`    console.log(\`[ws] User \${userId} connected to \${channel} (\${set.size} active)\`);`);
+  lines.push(``);
+  lines.push(`    // Send buffered history`);
+  lines.push(`    const history = messageBuffers.get(channel) || [];`);
+  lines.push(`    for (const msg of history) {`);
+  lines.push(`      socket.send(JSON.stringify(msg));`);
+  lines.push(`    }`);
+  lines.push(``);
+  lines.push(`    socket.send(JSON.stringify({ type: "connected", channel, history_size: history.length }));`);
+  lines.push(``);
+  lines.push(`    socket.on("message", (data) => {`);
+  lines.push(`      try {`);
+  lines.push(`        const msg = JSON.parse(data.toString());`);
+  lines.push(``);
+  lines.push(`        if (msg.type === "subscribe" && msg.topic) {`);
+  lines.push(`          client.topics.add(msg.topic);`);
+  lines.push(`          return;`);
+  lines.push(`        }`);
+  lines.push(`        if (msg.type === "unsubscribe" && msg.topic) {`);
+  lines.push(`          client.topics.delete(msg.topic);`);
+  lines.push(`          return;`);
+  lines.push(`        }`);
+  lines.push(``);
+  lines.push(`        // Broadcast`);
+  lines.push(`        const broadcast = {`);
+  lines.push(`          type: msg.type || "message",`);
+  lines.push(`          payload: msg.payload || msg,`);
+  lines.push(`          from: userId,`);
+  lines.push(`          timestamp: new Date().toISOString(),`);
+  lines.push(`        };`);
+  lines.push(``);
+  lines.push(`        persistMessage(channel, broadcast);`);
+  lines.push(`        broadcastToChannel(channel, broadcast, client);`);
+  lines.push(`      } catch (e: any) {`);
+  lines.push(`        socket.send(JSON.stringify({ type: "error", message: e.message }));`);
+  lines.push(`      }`);
+  lines.push(`    });`);
+  lines.push(``);
+  lines.push(`    socket.on("close", () => {`);
+  lines.push(`      const set = clients.get(channel);`);
+  lines.push(`      if (set) set.delete(client);`);
+  lines.push(`      console.log(\`[ws] User \${userId} disconnected from \${channel}\`);`);
+  lines.push(`    });`);
+  lines.push(``);
+  lines.push(`    // Heartbeat`);
+  lines.push(`    const heartbeat = setInterval(() => {`);
+  lines.push(`      if (socket.readyState === WebSocket.OPEN) {`);
+  lines.push(`        socket.ping();`);
+  lines.push(`      } else {`);
+  lines.push(`        clearInterval(heartbeat);`);
+  lines.push(`      }`);
+  lines.push(`    }, 30000);`);
+  lines.push(`  });`);
+  lines.push(``);
+  lines.push(`  // Bridge: forward eventBus events to WebSocket clients`);
+  for (const ch of channels) {
+    lines.push(`  // Channel '${ch.name}' bridge`);
+  }
+  lines.push(``);
+  // Redis subscription for cross-instance delivery
+  lines.push(`  // Subscribe to Redis channels for cross-instance delivery`);
+  lines.push(`  if (redisSub) {`);
+  for (const ch of channels) {
+    lines.push(`    redisSub.subscribe("ws:${ch.name}", (message: string) => {`);
+    lines.push(`      const set = clients.get("${ch.name}");`);
+    lines.push(`      if (!set) return;`);
+    lines.push(`      for (const client of set) {`);
+    lines.push(`        if (client.socket.readyState === WebSocket.OPEN) client.socket.send(message);`);
+    lines.push(`      }`);
+    lines.push(`    }).catch(() => {});`);
+  }
+  lines.push(`  }`);
+  lines.push(`  return wss;`);
+  lines.push(`}`);
+  lines.push(``);
+  lines.push(`function broadcastToChannel(channel: string, message: any, exclude?: Client) {`);
+  lines.push(`  const set = clients.get(channel);`);
+  lines.push(`  const data = JSON.stringify(message);`);
+  lines.push(`  // Broadcast to local clients`);
+  lines.push(`  if (set) {`);
+  lines.push(`    for (const client of set) {`);
+  lines.push(`      if (client === exclude) continue;`);
+  lines.push(`      if (client.socket.readyState === WebSocket.OPEN) {`);
+  lines.push(`        client.socket.send(data);`);
+  lines.push(`      }`);
+  lines.push(`    }`);
+  lines.push(`  }`);
+  lines.push(`  // Publish to Redis for cross-instance delivery`);
+  lines.push(`  if (redisPub) {`);
+  lines.push(`    redisPub.publish(\`ws:\${channel}\`, data).catch(() => {});`);
+  lines.push(`  }`);
+  lines.push(`}`);
+  lines.push(``);
+  lines.push(`export { broadcastToChannel };`);
+
+  return lines.join("\n");
+}