spacetimedb 2.4.0 → 2.5.0

package/dist/server/index.mjs

@@ -4782,6 +4782,9 @@ var RawModuleDefV10Section = t.enum("RawModuleDefV10Section", {
   },
   get HttpRoutes() {
     return t.array(RawHttpRouteDefV10);
+  },
+  get ViewPrimaryKeys() {
+    return t.array(RawViewPrimaryKeyDefV10);
   }
 });
 var RawModuleDefV8 = t.object("RawModuleDefV8", {
@@ -5019,6 +5022,10 @@ var RawViewDefV9 = t.object("RawViewDefV9", {
     return AlgebraicType2;
   }
 });
+var RawViewPrimaryKeyDefV10 = t.object("RawViewPrimaryKeyDefV10", {
+  viewSourceName: t.string(),
+  columns: t.array(t.string())
+});
 var ReducerDef = t.object("ReducerDef", {
   name: t.string(),
   get args() {
@@ -5129,6 +5136,7 @@ var ModuleContext = class {
     schedules: [],
     procedures: [],
     views: [],
+    viewPrimaryKeys: [],
     lifeCycleReducers: [],
     httpHandlers: [],
     httpRoutes: [],
@@ -5152,6 +5160,12 @@ var ModuleContext = class {
     push(module.reducers && { tag: "Reducers", value: module.reducers });
     push(module.procedures && { tag: "Procedures", value: module.procedures });
     push(module.views && { tag: "Views", value: module.views });
+    push(
+      module.viewPrimaryKeys && {
+        tag: "ViewPrimaryKeys",
+        value: module.viewPrimaryKeys
+      }
+    );
     push(module.schedules && { tag: "Schedules", value: module.schedules });
     push(
       module.lifeCycleReducers && {
@@ -6336,6 +6350,18 @@ function registerView(ctx, opts, exportName, anon, params, ret, fn) {
     params: paramType,
     returnType
   });
+  const primaryKeyColumns = viewPrimaryKeyColumns(ret);
+  if (primaryKeyColumns.length > 1) {
+    throw new TypeError(
+      `View '${exportName}' can have at most one primaryKey() column on its returned row type; found ${primaryKeyColumns.join(", ")}`
+    );
+  }
+  if (primaryKeyColumns.length === 1) {
+    ctx.moduleDef.viewPrimaryKeys.push({
+      viewSourceName: exportName,
+      columns: primaryKeyColumns
+    });
+  }
   if (opts.name != null) {
     ctx.moduleDef.explicitNames.entries.push({
       tag: "Function",
@@ -6362,6 +6388,24 @@ function registerView(ctx, opts, exportName, anon, params, ret, fn) {
     returnTypeBaseSize: bsatnBaseSize(typespace, returnType)
   });
 }
+function viewPrimaryKeyColumns(ret) {
+  const row = viewReturnRow(ret);
+  if (row == null) {
+    return [];
+  }
+  return Object.entries(row.row).filter(
+    (entry) => entry[1].columnMetadata.isPrimaryKey === true
+  ).map(([name]) => name);
+}
+function viewReturnRow(ret) {
+  if (ret instanceof ArrayBuilder && ret.element instanceof RowBuilder) {
+    return ret.element;
+  }
+  if (ret instanceof OptionBuilder && ret.value instanceof RowBuilder) {
+    return ret.value;
+  }
+  return void 0;
+}
 
 // src/lib/errors.ts
 var SenderError = class extends Error {
@@ -7898,7 +7942,7 @@ var Schema = class {
     }
     return makeReducerExport(this.#ctx, opts, {}, fn, Lifecycle.OnDisconnect);
   }
-  view(opts, ret, fn) {
+  view(opts, ret, fn, ..._) {
     return makeViewExport(this.#ctx, opts, {}, ret, fn);
   }
   // TODO: re-enable once parameterized views are supported in SQL
@@ -7925,7 +7969,7 @@ var Schema = class {
   //     defineView(name, false, paramsOrRet as Params, retOrFn, maybeFn!);
   //   }
   // }
-  anonymousView(opts, ret, fn) {
+  anonymousView(opts, ret, fn, ..._) {
     return makeAnonViewExport(this.#ctx, opts, {}, ret, fn);
   }
   procedure(...args) {