refactor typescript wiring

webclient/buf.gen.plugin.mjs

@@ -0,0 +1,99 @@
+// @ts-check
+/**
+ * Custom protoc-gen-es sibling plugin. Emits `src/generated/index.ts`, a
+ * single rollup that re-exports every generated `_pb` module and adds
+ * `MessageInitShape<T>` param aliases for every `Command_*` message.
+ *
+ * Wired into `buf.gen.yaml` as a second local plugin. Runs with the same
+ * descriptor set protoc-gen-es consumes, so output always tracks the protos.
+ */
+import { createEcmaScriptPlugin, runNodeJs } from '@bufbuild/protoplugin';
+
+const HEADER = [
+  '// @generated by protoc-gen-data. DO NOT EDIT.',
+  '// Rollup of all proto modules + MessageInitShape param aliases for every Command_*.',
+  '/* eslint-disable */',
+  '',
+  '',
+].join('\n');
+
+const inner = createEcmaScriptPlugin({
+  name: 'protoc-gen-data',
+  version: 'v0.1.0',
+  generateTs(schema) {
+    const f = schema.generateFile('index.ts');
+
+    const MessageInitShape = f.import('MessageInitShape', '@bufbuild/protobuf', true);
+    const MessageType = f.import('Message', '@bufbuild/protobuf', true);
+    const GenExtensionType = f.import('GenExtension', '@bufbuild/protobuf/codegenv2', true);
+
+    const sortedFiles = [...schema.files].sort((a, b) => a.name.localeCompare(b.name));
+
+    for (const file of sortedFiles) {
+      f.print('export * from ', f.string(`./proto/${file.name}_pb`), ';');
+    }
+    f.print();
+
+    const commandMessages = [];
+    for (const file of sortedFiles) {
+      for (const msg of file.messages) {
+        if (msg.name.startsWith('Command_')) {
+          commandMessages.push(msg);
+        }
+      }
+    }
+    commandMessages.sort((a, b) => a.name.localeCompare(b.name));
+
+    // importSchema() resolves paths relative to this plugin's `out` dir, which
+    // yields `./<name>_pb` — but the _pb files live under ./proto/ (protoc-gen-es's
+    // out). Build the import path explicitly so it points inside the proto subdir.
+    for (const msg of commandMessages) {
+      const alias = msg.name.slice('Command_'.length) + 'Params';
+      const schemaName = `${msg.name}Schema`;
+      const schemaSym = f.import(schemaName, `./proto/${msg.file.name}_pb`, true);
+      f.print('export type ', alias, ' = ', MessageInitShape, '<typeof ', schemaSym, '>;');
+    }
+    f.print();
+
+    // Generic extension registry infrastructure. Consolidates the three
+    // near-duplicate registry types and helpers that used to live in
+    // src/websocket/services/protobuf-types.ts into one generic pair.
+    // Specialised aliases (Session/Room/Game) still live in protobuf-types.ts
+    // because GameExtensionRegistry needs GameEventMeta — a hand-written
+    // domain type whose import would create a generated/ ↔ types/ cycle.
+    f.print('export type RegistryEntry<V, T extends ', MessageType, ', M = unknown> = [');
+    f.print('  ', GenExtensionType, '<T, V>,');
+    f.print('  (value: V, meta: M) => void,');
+    f.print('];');
+    f.print();
+    // Return type widens V to `unknown` so the heterogeneous entries that
+    // callers build can be stored in a homogeneous `RegistryEntry<unknown, T, M>[]`
+    // array. This is the actual value-add over a bare tuple literal.
+    f.print('export function makeEntry<T extends ', MessageType, ', V, M = unknown>(');
+    f.print('  ext: ', GenExtensionType, '<T, V>,');
+    f.print('  handler: (value: V, meta: M) => void,');
+    f.print('): RegistryEntry<unknown, T, M> {');
+    f.print('  return [ext, handler] as unknown as RegistryEntry<unknown, T, M>;');
+    f.print('}');
+  },
+});
+
+// Skip f.preamble() above and inject a custom rollup-aware header here instead —
+// preamble() would write "@generated from file X.proto" which is misleading for
+// a rollup file built from every input proto.
+/** @type {import('@bufbuild/protoplugin').Plugin} */
+const plugin = {
+  name: inner.name,
+  version: inner.version,
+  run(request) {
+    const response = inner.run(request);
+    for (const file of response.file) {
+      if (file.name === 'index.ts' && typeof file.content === 'string') {
+        file.content = HEADER + file.content;
+      }
+    }
+    return response;
+  },
+};
+
+runNodeJs(plugin);