bonescript-compiler 0.9.0 → 0.11.0

package/src/index.ts

@@ -18,6 +18,7 @@ export { FullEmitter } from "./emit_full";
 export { NakamaEmitter } from "./emit_nakama";
 export { PrismaEmitter } from "./emit_prisma";
 export { SqliteEmitter } from "./emit_sqlite";
+export { ColyseusEmitter } from "./emit_colyseus";
 export type { NakamaEmittedFile } from "./emit_nakama";
 export type { EmittedFile } from "./emitter";
 export { Verifier } from "./verifier";

package/src/lowering.ts

@@ -17,6 +17,27 @@ function toSnakeCase(s: string): string {
   return s.replace(/([a-z])([A-Z])/g, "$1_$2").toLowerCase();
 }
 
+/**
+ * Extract the entity name from a `participants:` type expression.
+ *
+ * - `Player` → "Player"
+ * - `set<Player>` → "Player"
+ * - `list<Player>` → "Player"
+ * - `optional<Player>` → "Player"
+ * - anything else → null
+ */
+function extractParticipantEntity(t: AST.TypeExprNode | null): string | null {
+  if (!t) return null;
+  if (t.kind === "EntityRefType") return t.name;
+  if (t.kind === "GenericType") {
+    for (const arg of t.typeArgs) {
+      const inner = extractParticipantEntity(arg);
+      if (inner) return inner;
+    }
+  }
+  return null;
+}
+
 // ─── Deterministic ID Generation ─────────────────────────────────────────────
 
 function makeId(systemName: string, kind: string, name: string): string {
@@ -122,11 +143,30 @@ export class Lowering {
     }
 
     // Lower channels → realtime_service modules
+    // Lower channels — realtime_service modules.
+    //
+    // Capabilities tagged `sync: realtime` are routed into the channel that
+    // matches their participant entity, so each room can expose them as typed
+    // onMessage handlers. The matching rule: if any of the capability's
+    // parameters is the participant entity type (e.g. channel has
+    // `participants: set<Player>` and capability has `move(player: Player, ...)`),
+    // it gets dispatched on that room.
+    const realtimeCaps = capabilities.filter(c => c.sync === "realtime");
     for (const channel of channels) {
-      modules.push(this.lowerChannel(channel));
+      const channelCaps = realtimeCaps.filter(c => {
+        const partEntityName = extractParticipantEntity(channel.participants);
+        if (!partEntityName) return false;
+        return c.params.some(p => {
+          if (p.type.kind === "EntityRefType") return p.type.name === partEntityName;
+          // Also match generic types like `set<Player>` if they happen to appear as a param
+          if (p.type.kind === "GenericType") {
+            return p.type.typeArgs.some(a => a.kind === "EntityRefType" && a.name === partEntityName);
+          }
+          return false;
+        });
+      });
+      modules.push(this.lowerChannel(channel, channelCaps, entities));
     }
-
-    // Lower events
     for (const ev of eventDecls) {
       events.push(this.lowerEvent(ev));
     }
@@ -516,18 +556,48 @@ export class Lowering {
 
   // ─── Channel Lowering ──────────────────────────────────────────────────────
 
-  private lowerChannel(channel: AST.ChannelDeclNode): IR.IRModule {
+  private lowerChannel(
+    channel: AST.ChannelDeclNode,
+    realtimeCapabilities: AST.CapabilityDeclNode[] = [],
+    entities: AST.EntityDeclNode[] = [],
+  ): IR.IRModule {
+    // Default lifecycle methods that every Colyseus/WebSocket channel exposes.
+    const baseMethods: IR.IRMethod[] = [
+      { name: "connect", input: [], output: "connection", preconditions: [], effects: [], emissions: [], idempotent: false, authenticated: true, timeout_ms: 5000, retry: null, pipeline: null, algorithm: null, sync: null },
+      { name: "subscribe", input: [{ name: "topic", type: "string", nullable: false, unique: false, indexed: false, default_value: null }], output: "subscription", preconditions: [], effects: [], emissions: [], idempotent: true, authenticated: true, timeout_ms: 5000, retry: null, pipeline: null, algorithm: null, sync: null },
+      { name: "publish", input: [{ name: "message", type: "json", nullable: false, unique: false, indexed: false, default_value: null }], output: "void", preconditions: [], effects: [], emissions: [], idempotent: false, authenticated: true, timeout_ms: 5000, retry: null, pipeline: null, algorithm: null, sync: null },
+    ];
+
+    // Realtime capabilities lower as additional interface methods. The
+    // Colyseus emitter will turn these into typed onMessage handlers.
+    const capMethods = realtimeCapabilities.map(c => this.lowerCapability(c));
+
+    const participantEntity = extractParticipantEntity(channel.participants);
+
+    // If we have an entity reference, copy its field definitions into the
+    // channel's config so the Colyseus emitter can build a Player schema
+    // without re-resolving the AST. We only forward primitive-typed `owns:`
+    // fields — entity refs and complex types are skipped because Colyseus
+    // schemas need scalar @type decorators.
+    const participantFields: { name: string; type: string }[] = [];
+    if (participantEntity) {
+      const entity = entities.find(e => e.name === participantEntity);
+      if (entity) {
+        for (const f of entity.owns) {
+          if (f.type.kind === "PrimitiveType") {
+            participantFields.push({ name: f.name, type: f.type.name });
+          }
+        }
+      }
+    }
+
     return {
       id: makeId(this.systemName, "realtime_service", channel.name),
       kind: "realtime_service",
       name: channel.name,
       interfaces: [{
         name: `I${channel.name}Channel`,
-        methods: [
-          { name: "connect", input: [], output: "connection", preconditions: [], effects: [], emissions: [], idempotent: false, authenticated: true, timeout_ms: 5000, retry: null, pipeline: null, algorithm: null, sync: null },
-          { name: "subscribe", input: [{ name: "topic", type: "string", nullable: false, unique: false, indexed: false, default_value: null }], output: "subscription", preconditions: [], effects: [], emissions: [], idempotent: true, authenticated: true, timeout_ms: 5000, retry: null, pipeline: null, algorithm: null, sync: null },
-          { name: "publish", input: [{ name: "message", type: "json", nullable: false, unique: false, indexed: false, default_value: null }], output: "void", preconditions: [], effects: [], emissions: [], idempotent: false, authenticated: true, timeout_ms: 5000, retry: null, pipeline: null, algorithm: null, sync: null },
-        ],
+        methods: [...baseMethods, ...capMethods],
       }],
       models: [],
       events: [],
@@ -539,6 +609,8 @@ export class Lowering {
         ordering: channel.ordering || "fifo",
         persistence: channel.persistence || "none",
         max_size: channel.maxSize || 10000,
+        ...(participantEntity ? { participant_entity: participantEntity } : {}),
+        ...(participantFields.length > 0 ? { participant_fields: JSON.stringify(participantFields) } : {}),
       },
     };
   }