spacetimedb 2.5.0 → 2.6.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (180) hide show
  1. package/LICENSE.txt +759 -759
  2. package/README.md +211 -211
  3. package/dist/angular/index.cjs.map +1 -1
  4. package/dist/angular/index.mjs.map +1 -1
  5. package/dist/browser/angular/index.mjs.map +1 -1
  6. package/dist/browser/react/index.mjs +129 -57
  7. package/dist/browser/react/index.mjs.map +1 -1
  8. package/dist/browser/solid/index.mjs +120 -50
  9. package/dist/browser/solid/index.mjs.map +1 -1
  10. package/dist/browser/svelte/index.mjs.map +1 -1
  11. package/dist/browser/vue/index.mjs.map +1 -1
  12. package/dist/index.browser.mjs +10 -2
  13. package/dist/index.browser.mjs.map +1 -1
  14. package/dist/index.cjs +10 -2
  15. package/dist/index.cjs.map +1 -1
  16. package/dist/index.mjs +10 -2
  17. package/dist/index.mjs.map +1 -1
  18. package/dist/min/index.browser.mjs +1 -1
  19. package/dist/min/index.browser.mjs.map +1 -1
  20. package/dist/min/react/index.mjs +1 -1
  21. package/dist/min/react/index.mjs.map +1 -1
  22. package/dist/min/sdk/index.browser.mjs +1 -1
  23. package/dist/min/sdk/index.browser.mjs.map +1 -1
  24. package/dist/react/index.cjs +129 -57
  25. package/dist/react/index.cjs.map +1 -1
  26. package/dist/react/index.mjs +129 -57
  27. package/dist/react/index.mjs.map +1 -1
  28. package/dist/react/useTable.d.ts.map +1 -1
  29. package/dist/sdk/connection_manager.d.ts +8 -0
  30. package/dist/sdk/connection_manager.d.ts.map +1 -1
  31. package/dist/sdk/db_connection_impl.d.ts +7 -0
  32. package/dist/sdk/db_connection_impl.d.ts.map +1 -1
  33. package/dist/sdk/index.browser.mjs +10 -2
  34. package/dist/sdk/index.browser.mjs.map +1 -1
  35. package/dist/sdk/index.cjs +10 -2
  36. package/dist/sdk/index.cjs.map +1 -1
  37. package/dist/sdk/index.mjs +10 -2
  38. package/dist/sdk/index.mjs.map +1 -1
  39. package/dist/sdk/websocket_test_adapter.d.ts +2 -1
  40. package/dist/sdk/websocket_test_adapter.d.ts.map +1 -1
  41. package/dist/server/index.mjs.map +1 -1
  42. package/dist/server/runtime.d.ts.map +1 -1
  43. package/dist/solid/index.cjs +120 -50
  44. package/dist/solid/index.cjs.map +1 -1
  45. package/dist/solid/index.mjs +120 -50
  46. package/dist/solid/index.mjs.map +1 -1
  47. package/dist/svelte/index.cjs.map +1 -1
  48. package/dist/svelte/index.mjs.map +1 -1
  49. package/dist/tanstack/index.cjs +120 -50
  50. package/dist/tanstack/index.cjs.map +1 -1
  51. package/dist/tanstack/index.mjs +120 -50
  52. package/dist/tanstack/index.mjs.map +1 -1
  53. package/dist/vue/index.cjs.map +1 -1
  54. package/dist/vue/index.mjs.map +1 -1
  55. package/package.json +1 -1
  56. package/src/angular/connection_state.ts +19 -19
  57. package/src/angular/index.ts +3 -3
  58. package/src/angular/injectors/index.ts +4 -4
  59. package/src/angular/injectors/inject-reducer.ts +62 -62
  60. package/src/angular/injectors/inject-spacetimedb-connected.ts +13 -13
  61. package/src/angular/injectors/inject-spacetimedb.ts +10 -10
  62. package/src/angular/injectors/inject-table.ts +234 -234
  63. package/src/angular/providers/index.ts +1 -1
  64. package/src/angular/providers/provide-spacetimedb.ts +96 -96
  65. package/src/index.ts +16 -16
  66. package/src/lib/algebraic_type.ts +819 -819
  67. package/src/lib/algebraic_type_variants.ts +26 -26
  68. package/src/lib/algebraic_value.ts +10 -10
  69. package/src/lib/autogen/types.ts +746 -746
  70. package/src/lib/binary_reader.ts +188 -188
  71. package/src/lib/binary_writer.ts +213 -213
  72. package/src/lib/connection_id.ts +102 -102
  73. package/src/lib/constraints.ts +48 -48
  74. package/src/lib/errors.ts +26 -26
  75. package/src/lib/filter.ts +195 -195
  76. package/src/lib/identity.ts +83 -83
  77. package/src/lib/indexes.ts +251 -251
  78. package/src/lib/option.ts +34 -34
  79. package/src/lib/query.ts +1019 -1019
  80. package/src/lib/reducer_schema.ts +38 -38
  81. package/src/lib/reducers.ts +116 -116
  82. package/src/lib/result.ts +36 -36
  83. package/src/lib/schedule_at.ts +86 -86
  84. package/src/lib/schema.ts +420 -420
  85. package/src/lib/table.ts +548 -548
  86. package/src/lib/table_schema.ts +64 -64
  87. package/src/lib/time_duration.ts +77 -77
  88. package/src/lib/timestamp.ts +148 -148
  89. package/src/lib/type_builders.test-d.ts +128 -128
  90. package/src/lib/type_builders.ts +4014 -4014
  91. package/src/lib/type_util.ts +124 -124
  92. package/src/lib/util.ts +196 -196
  93. package/src/lib/uuid.ts +337 -337
  94. package/src/react/SpacetimeDBProvider.ts +84 -84
  95. package/src/react/connection_state.ts +6 -6
  96. package/src/react/index.ts +5 -5
  97. package/src/react/useProcedure.ts +60 -60
  98. package/src/react/useReducer.ts +53 -53
  99. package/src/react/useSpacetimeDB.ts +18 -18
  100. package/src/react/useTable.ts +256 -251
  101. package/src/sdk/client_api/index.ts +114 -114
  102. package/src/sdk/client_api/types/procedures.ts +8 -8
  103. package/src/sdk/client_api/types/reducers.ts +8 -8
  104. package/src/sdk/client_api/types.ts +288 -288
  105. package/src/sdk/client_cache.ts +129 -129
  106. package/src/sdk/client_table.ts +179 -179
  107. package/src/sdk/connection_manager.ts +352 -237
  108. package/src/sdk/db_connection_builder.ts +290 -290
  109. package/src/sdk/db_connection_impl.ts +1356 -1347
  110. package/src/sdk/db_context.ts +28 -28
  111. package/src/sdk/db_view.ts +12 -12
  112. package/src/sdk/decompress.ts +51 -51
  113. package/src/sdk/event.ts +18 -18
  114. package/src/sdk/event_context.ts +51 -51
  115. package/src/sdk/event_emitter.ts +32 -32
  116. package/src/sdk/index.ts +14 -14
  117. package/src/sdk/internal.ts +2 -2
  118. package/src/sdk/json_api.ts +46 -46
  119. package/src/sdk/logger.ts +134 -134
  120. package/src/sdk/message_types.ts +46 -46
  121. package/src/sdk/procedures.ts +83 -83
  122. package/src/sdk/reducer_event.ts +20 -20
  123. package/src/sdk/reducer_handle.ts +12 -12
  124. package/src/sdk/reducers.ts +159 -159
  125. package/src/sdk/schema.ts +45 -45
  126. package/src/sdk/spacetime_module.ts +28 -28
  127. package/src/sdk/subscription_builder_impl.ts +275 -275
  128. package/src/sdk/table_cache.ts +581 -581
  129. package/src/sdk/type_utils.ts +19 -19
  130. package/src/sdk/version.ts +133 -133
  131. package/src/sdk/websocket_decompress_adapter.ts +63 -63
  132. package/src/sdk/websocket_protocols.ts +25 -25
  133. package/src/sdk/websocket_test_adapter.ts +107 -100
  134. package/src/sdk/websocket_v3_frames.ts +126 -126
  135. package/src/sdk/ws.ts +105 -105
  136. package/src/server/console.ts +81 -81
  137. package/src/server/db_view.ts +21 -21
  138. package/src/server/errors.ts +138 -138
  139. package/src/server/http.test-d.ts +80 -80
  140. package/src/server/http.ts +14 -14
  141. package/src/server/http_handlers.ts +413 -413
  142. package/src/server/http_internal.ts +79 -79
  143. package/src/server/http_shared.ts +186 -186
  144. package/src/server/index.ts +37 -37
  145. package/src/server/polyfills.ts +4 -4
  146. package/src/server/procedures.ts +239 -239
  147. package/src/server/query.ts +1 -1
  148. package/src/server/range.ts +53 -53
  149. package/src/server/reducers.ts +113 -113
  150. package/src/server/rng.ts +113 -113
  151. package/src/server/runtime.ts +1102 -1102
  152. package/src/server/schema.test-d.ts +99 -99
  153. package/src/server/schema.ts +663 -663
  154. package/src/server/sys.d.ts +125 -125
  155. package/src/server/view.test-d.ts +194 -194
  156. package/src/server/views.ts +340 -340
  157. package/src/solid/SpacetimeDBProvider.ts +97 -97
  158. package/src/solid/connection_state.ts +6 -6
  159. package/src/solid/index.ts +5 -5
  160. package/src/solid/useProcedure.ts +57 -57
  161. package/src/solid/useReducer.ts +50 -50
  162. package/src/solid/useSpacetimeDB.ts +18 -18
  163. package/src/solid/useTable.ts +203 -203
  164. package/src/svelte/SpacetimeDBProvider.ts +101 -101
  165. package/src/svelte/connection_state.ts +16 -16
  166. package/src/svelte/index.ts +4 -4
  167. package/src/svelte/useReducer.ts +61 -61
  168. package/src/svelte/useSpacetimeDB.ts +22 -22
  169. package/src/svelte/useTable.ts +218 -218
  170. package/src/tanstack/SpacetimeDBQueryClient.ts +330 -330
  171. package/src/tanstack/hooks.ts +83 -83
  172. package/src/tanstack/index.ts +16 -16
  173. package/src/util-stub.ts +1 -1
  174. package/src/vue/SpacetimeDBProvider.ts +157 -157
  175. package/src/vue/connection_state.ts +19 -19
  176. package/src/vue/index.ts +5 -5
  177. package/src/vue/useProcedure.ts +62 -62
  178. package/src/vue/useReducer.ts +55 -55
  179. package/src/vue/useSpacetimeDB.ts +18 -18
  180. package/src/vue/useTable.ts +229 -229
package/src/lib/schema.ts CHANGED
@@ -1,420 +1,420 @@
1
- import {
2
- AlgebraicType,
3
- ProductType,
4
- SumType,
5
- type AlgebraicTypeType,
6
- type AlgebraicTypeVariants,
7
- } from './algebraic_type';
8
- import type {
9
- CaseConversionPolicy,
10
- RawModuleDefV10,
11
- RawModuleDefV10Section,
12
- RawScopedTypeNameV10,
13
- RawTableDefV10,
14
- } from './autogen/types';
15
- import type { UntypedIndex } from './indexes';
16
- import type { UntypedTableDef } from './table';
17
- import type { UntypedTableSchema } from './table_schema';
18
- import {
19
- ArrayBuilder,
20
- OptionBuilder,
21
- ProductBuilder,
22
- RefBuilder,
23
- ResultBuilder,
24
- RowBuilder,
25
- SumBuilder,
26
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
27
- TypeBuilder,
28
- type ElementsObj,
29
- type Infer,
30
- type InferSpacetimeTypeOfTypeBuilder,
31
- type RowObj,
32
- type VariantsObj,
33
- } from './type_builders';
34
- import type { Values } from './type_util';
35
-
36
- export type TableNamesOf<S extends UntypedSchemaDef> = Values<
37
- S['tables']
38
- >['accessorName'];
39
-
40
- /**
41
- * An untyped representation of the database schema.
42
- */
43
- export type UntypedSchemaDef = {
44
- tables: Record<string, UntypedTableDef>;
45
- };
46
-
47
- /**
48
- * Helper type to convert an array of TableSchema into a schema definition
49
- */
50
- export interface TablesToSchema<T extends Record<string, UntypedTableSchema>>
51
- extends UntypedSchemaDef {
52
- tables: {
53
- readonly [AccName in keyof T & string]: TableToSchema<AccName, T[AccName]>;
54
- };
55
- }
56
-
57
- export interface TableToSchema<
58
- AccName extends string,
59
- T extends UntypedTableSchema,
60
- > extends UntypedTableDef {
61
- accessorName: AccName;
62
- columns: T['rowType']['row'];
63
- rowType: T['rowSpacetimeType'];
64
- // Declarative user-provided table-level indexes.
65
- indexes: T['idxs'];
66
- // Resolved runtime index metadata used by runtime consumers (e.g. TableCache).
67
- resolvedIndexes: readonly UntypedIndex<keyof T['rowType']['row'] & string>[];
68
- constraints: T['constraints'];
69
- }
70
-
71
- export function tablesToSchema<
72
- const T extends Record<string, UntypedTableSchema>,
73
- >(ctx: ModuleContext, tables: T): TablesToSchema<T> {
74
- // `TablesToSchema<T>['tables']` is intentionally readonly in the public type,
75
- // but we need a mutable builder while materializing it from entries.
76
- type MutableTableDefs = {
77
- -readonly [AccName in keyof TablesToSchema<T>['tables']]: TablesToSchema<T>['tables'][AccName];
78
- };
79
- const tableDefs = Object.create(null) as MutableTableDefs;
80
- for (const [accName, schema] of Object.entries(tables) as [
81
- keyof T & string,
82
- T[keyof T & string],
83
- ][]) {
84
- tableDefs[accName] = tableToSchema(
85
- accName,
86
- schema,
87
- schema.tableDef(ctx, accName)
88
- ) as TablesToSchema<T>['tables'][typeof accName];
89
- }
90
-
91
- return {
92
- tables: tableDefs as TablesToSchema<T>['tables'],
93
- };
94
- }
95
-
96
- export function tableToSchema<
97
- AccName extends string,
98
- const T extends UntypedTableSchema,
99
- >(
100
- accName: AccName,
101
- schema: T,
102
- tableDef: RawTableDefV10
103
- ): TableToSchema<AccName, T> {
104
- const getColName = (i: number) =>
105
- schema.rowType.algebraicType.value.elements[i].name;
106
-
107
- type AllowedCol = keyof T['rowType']['row'] & string;
108
- // Build fully-resolved runtime index metadata from the host-facing RawTableDef.
109
- // This is intentionally separate from `schema.idxs`, which keeps the original
110
- // user-declared `IndexOpts` shape for type-level inference.
111
- const resolvedIndexes: UntypedIndex<AllowedCol>[] = tableDef.indexes.map(
112
- idx => {
113
- const accessorName = idx.accessorName;
114
- if (typeof accessorName !== 'string' || accessorName.length === 0) {
115
- throw new TypeError(
116
- `Index '${idx.sourceName ?? '<unknown>'}' on table '${tableDef.sourceName}' is missing accessor name`
117
- );
118
- }
119
-
120
- const columnIds =
121
- idx.algorithm.tag === 'Direct'
122
- ? [idx.algorithm.value]
123
- : idx.algorithm.value;
124
-
125
- const unique = tableDef.constraints.some(
126
- c =>
127
- c.data.tag === 'Unique' &&
128
- c.data.value.columns.every(col => columnIds.includes(col))
129
- );
130
-
131
- const algorithm = (
132
- {
133
- BTree: 'btree',
134
- Hash: 'hash',
135
- Direct: 'direct',
136
- } as const
137
- )[idx.algorithm.tag];
138
-
139
- return {
140
- name: accessorName,
141
- unique,
142
- algorithm,
143
- columns: columnIds.map(getColName) as AllowedCol[],
144
- };
145
- }
146
- );
147
-
148
- return {
149
- // For client,`schama.tableName` will always be there as canonical name.
150
- // For module, if explicit name is not provided via `name`, accessor name will
151
- // be used, it is stored as alias in database, hence works in query builder.
152
- sourceName: schema.tableName || accName,
153
- accessorName: accName,
154
- columns: schema.rowType.row, // typed as T[i]['rowType']['row'] under TablesToSchema<T>
155
- rowType: schema.rowSpacetimeType,
156
- // Keep declarative indexes in their original shape for type-level consumers.
157
- indexes: schema.idxs,
158
- constraints: tableDef.constraints.map(c => ({
159
- name: c.sourceName,
160
- constraint: 'unique',
161
- columns: c.data.value.columns.map(getColName) as [string],
162
- })),
163
- // Expose resolved runtime indexes separately so runtime users don't have to
164
- // reinterpret `indexes` with unsafe casts.
165
- resolvedIndexes,
166
- tableDef,
167
- ...(tableDef.isEvent ? { isEvent: true } : {}),
168
- };
169
- }
170
-
171
- type CompoundTypeCache = Map<
172
- AlgebraicTypeVariants.Product | AlgebraicTypeVariants.Sum,
173
- RefBuilder<any, any>
174
- >;
175
-
176
- export type ModuleDef = {
177
- [S in RawModuleDefV10Section as Uncapitalize<S['tag']>]: S['value'];
178
- };
179
-
180
- type Section = RawModuleDefV10Section;
181
-
182
- export class ModuleContext {
183
- #compoundTypes: CompoundTypeCache = new Map();
184
-
185
- /**
186
- * The global module definition that gets populated by calls to `reducer()` and lifecycle hooks.
187
- */
188
- #moduleDef: ModuleDef = {
189
- typespace: { types: [] },
190
- tables: [],
191
- reducers: [],
192
- types: [],
193
- rowLevelSecurity: [],
194
- schedules: [],
195
- procedures: [],
196
- views: [],
197
- viewPrimaryKeys: [],
198
- lifeCycleReducers: [],
199
- httpHandlers: [],
200
- httpRoutes: [],
201
- caseConversionPolicy: { tag: 'SnakeCase' },
202
- explicitNames: {
203
- entries: [],
204
- },
205
- };
206
-
207
- get moduleDef(): ModuleDef {
208
- return this.#moduleDef;
209
- }
210
-
211
- rawModuleDefV10(): RawModuleDefV10 {
212
- const sections: Section[] = [];
213
-
214
- const push = <T extends Section>(s: T | undefined) => {
215
- if (s) sections.push(s);
216
- };
217
-
218
- const module = this.#moduleDef;
219
-
220
- push(module.typespace && { tag: 'Typespace', value: module.typespace });
221
- push(module.types && { tag: 'Types', value: module.types });
222
- push(module.tables && { tag: 'Tables', value: module.tables });
223
- push(module.reducers && { tag: 'Reducers', value: module.reducers });
224
- push(module.procedures && { tag: 'Procedures', value: module.procedures });
225
- push(module.views && { tag: 'Views', value: module.views });
226
- push(
227
- module.viewPrimaryKeys && {
228
- tag: 'ViewPrimaryKeys',
229
- value: module.viewPrimaryKeys,
230
- }
231
- );
232
- push(module.schedules && { tag: 'Schedules', value: module.schedules });
233
- push(
234
- module.lifeCycleReducers && {
235
- tag: 'LifeCycleReducers',
236
- value: module.lifeCycleReducers,
237
- }
238
- );
239
- push(
240
- module.httpHandlers && {
241
- tag: 'HttpHandlers',
242
- value: module.httpHandlers,
243
- }
244
- );
245
- push(
246
- module.httpRoutes && {
247
- tag: 'HttpRoutes',
248
- value: module.httpRoutes,
249
- }
250
- );
251
- push(
252
- module.rowLevelSecurity && {
253
- tag: 'RowLevelSecurity',
254
- value: module.rowLevelSecurity,
255
- }
256
- );
257
- push(
258
- module.explicitNames && {
259
- tag: 'ExplicitNames',
260
- value: module.explicitNames,
261
- }
262
- );
263
- push(
264
- module.caseConversionPolicy && {
265
- tag: 'CaseConversionPolicy',
266
- value: module.caseConversionPolicy,
267
- }
268
- );
269
- return { sections };
270
- }
271
-
272
- /**
273
- * Set the case conversion policy for this module.
274
- * Called by the settings mechanism.
275
- */
276
- setCaseConversionPolicy(policy: CaseConversionPolicy) {
277
- this.#moduleDef.caseConversionPolicy = policy;
278
- }
279
-
280
- get typespace() {
281
- return this.#moduleDef.typespace;
282
- }
283
-
284
- /**
285
- * Resolves the actual type of a TypeBuilder by following its references until it reaches a non-ref type.
286
- * @param typespace The typespace to resolve types against.
287
- * @param typeBuilder The TypeBuilder to resolve.
288
- * @returns The resolved algebraic type.
289
- */
290
- public resolveType<AT extends AlgebraicTypeType>(
291
- typeBuilder: RefBuilder<any, AT>
292
- ): AT {
293
- let ty: AlgebraicType = typeBuilder.algebraicType;
294
- while (ty.tag === 'Ref') {
295
- ty = this.typespace.types[ty.value];
296
- }
297
- return ty as AT;
298
- }
299
-
300
- /**
301
- * Adds a type to the module definition's typespace as a `Ref` if it is a named compound type (Product or Sum).
302
- * Otherwise, returns the type as is.
303
- * @param name
304
- * @param ty
305
- * @returns
306
- */
307
- public registerTypesRecursively<T extends TypeBuilder<any, AlgebraicType>>(
308
- typeBuilder: T
309
- ): T extends SumBuilder<any> | ProductBuilder<any> | RowBuilder<any>
310
- ? RefBuilder<Infer<T>, InferSpacetimeTypeOfTypeBuilder<T>>
311
- : T {
312
- if (
313
- (typeBuilder instanceof ProductBuilder && !isUnit(typeBuilder)) ||
314
- typeBuilder instanceof SumBuilder ||
315
- typeBuilder instanceof RowBuilder
316
- ) {
317
- return this.#registerCompoundTypeRecursively(typeBuilder) as any;
318
- } else if (typeBuilder instanceof OptionBuilder) {
319
- return new OptionBuilder(
320
- this.registerTypesRecursively(typeBuilder.value)
321
- ) as any;
322
- } else if (typeBuilder instanceof ResultBuilder) {
323
- return new ResultBuilder(
324
- this.registerTypesRecursively(typeBuilder.ok),
325
- this.registerTypesRecursively(typeBuilder.err)
326
- ) as any;
327
- } else if (typeBuilder instanceof ArrayBuilder) {
328
- return new ArrayBuilder(
329
- this.registerTypesRecursively(typeBuilder.element)
330
- ) as any;
331
- } else {
332
- return typeBuilder as any;
333
- }
334
- }
335
-
336
- #registerCompoundTypeRecursively<
337
- T extends
338
- | SumBuilder<VariantsObj>
339
- | ProductBuilder<ElementsObj>
340
- | RowBuilder<RowObj>,
341
- >(typeBuilder: T): RefBuilder<Infer<T>, InferSpacetimeTypeOfTypeBuilder<T>> {
342
- const ty = typeBuilder.algebraicType;
343
- // NB! You must ensure that all TypeBuilder passed into this function
344
- // have a name. This function ensures that nested types always have a
345
- // name by assigning them one if they are missing it.
346
- const name = typeBuilder.typeName;
347
- if (name === undefined) {
348
- throw new Error(
349
- `Missing type name for ${typeBuilder.constructor.name ?? 'TypeBuilder'} ${JSON.stringify(typeBuilder)}`
350
- );
351
- }
352
-
353
- let r = this.#compoundTypes.get(ty);
354
- if (r != null) {
355
- // Already added to typespace
356
- return r;
357
- }
358
-
359
- // Recursively register nested compound types
360
- const newTy =
361
- typeBuilder instanceof RowBuilder || typeBuilder instanceof ProductBuilder
362
- ? ({
363
- tag: 'Product',
364
- value: { elements: [] },
365
- } as AlgebraicTypeVariants.Product)
366
- : ({
367
- tag: 'Sum',
368
- value: { variants: [] },
369
- } as AlgebraicTypeVariants.Sum);
370
-
371
- r = new RefBuilder(this.#moduleDef.typespace.types.length);
372
- this.#moduleDef.typespace.types.push(newTy);
373
-
374
- this.#compoundTypes.set(ty, r);
375
-
376
- if (typeBuilder instanceof RowBuilder) {
377
- for (const [name, elem] of Object.entries(typeBuilder.row)) {
378
- (newTy.value as ProductType).elements.push({
379
- name,
380
- algebraicType: this.registerTypesRecursively(elem.typeBuilder)
381
- .algebraicType,
382
- });
383
- }
384
- } else if (typeBuilder instanceof ProductBuilder) {
385
- for (const [name, elem] of Object.entries(typeBuilder.elements)) {
386
- (newTy.value as ProductType).elements.push({
387
- name,
388
- algebraicType: this.registerTypesRecursively(elem).algebraicType,
389
- });
390
- }
391
- } else if (typeBuilder instanceof SumBuilder) {
392
- for (const [name, variant] of Object.entries(typeBuilder.variants)) {
393
- (newTy.value as SumType).variants.push({
394
- name,
395
- algebraicType: this.registerTypesRecursively(variant).algebraicType,
396
- });
397
- }
398
- }
399
-
400
- this.#moduleDef.types.push({
401
- sourceName: splitName(name),
402
- ty: r.ref,
403
- customOrdering: true,
404
- });
405
-
406
- return r;
407
- }
408
- }
409
-
410
- function isUnit(typeBuilder: ProductBuilder<ElementsObj>): boolean {
411
- return (
412
- typeBuilder.typeName == null &&
413
- typeBuilder.algebraicType.value.elements.length === 0
414
- );
415
- }
416
-
417
- export function splitName(name: string): RawScopedTypeNameV10 {
418
- const scope = name.split('.');
419
- return { sourceName: scope.pop()!, scope };
420
- }
1
+ import {
2
+ AlgebraicType,
3
+ ProductType,
4
+ SumType,
5
+ type AlgebraicTypeType,
6
+ type AlgebraicTypeVariants,
7
+ } from './algebraic_type';
8
+ import type {
9
+ CaseConversionPolicy,
10
+ RawModuleDefV10,
11
+ RawModuleDefV10Section,
12
+ RawScopedTypeNameV10,
13
+ RawTableDefV10,
14
+ } from './autogen/types';
15
+ import type { UntypedIndex } from './indexes';
16
+ import type { UntypedTableDef } from './table';
17
+ import type { UntypedTableSchema } from './table_schema';
18
+ import {
19
+ ArrayBuilder,
20
+ OptionBuilder,
21
+ ProductBuilder,
22
+ RefBuilder,
23
+ ResultBuilder,
24
+ RowBuilder,
25
+ SumBuilder,
26
+ // eslint-disable-next-line @typescript-eslint/no-unused-vars
27
+ TypeBuilder,
28
+ type ElementsObj,
29
+ type Infer,
30
+ type InferSpacetimeTypeOfTypeBuilder,
31
+ type RowObj,
32
+ type VariantsObj,
33
+ } from './type_builders';
34
+ import type { Values } from './type_util';
35
+
36
+ export type TableNamesOf<S extends UntypedSchemaDef> = Values<
37
+ S['tables']
38
+ >['accessorName'];
39
+
40
+ /**
41
+ * An untyped representation of the database schema.
42
+ */
43
+ export type UntypedSchemaDef = {
44
+ tables: Record<string, UntypedTableDef>;
45
+ };
46
+
47
+ /**
48
+ * Helper type to convert an array of TableSchema into a schema definition
49
+ */
50
+ export interface TablesToSchema<T extends Record<string, UntypedTableSchema>>
51
+ extends UntypedSchemaDef {
52
+ tables: {
53
+ readonly [AccName in keyof T & string]: TableToSchema<AccName, T[AccName]>;
54
+ };
55
+ }
56
+
57
+ export interface TableToSchema<
58
+ AccName extends string,
59
+ T extends UntypedTableSchema,
60
+ > extends UntypedTableDef {
61
+ accessorName: AccName;
62
+ columns: T['rowType']['row'];
63
+ rowType: T['rowSpacetimeType'];
64
+ // Declarative user-provided table-level indexes.
65
+ indexes: T['idxs'];
66
+ // Resolved runtime index metadata used by runtime consumers (e.g. TableCache).
67
+ resolvedIndexes: readonly UntypedIndex<keyof T['rowType']['row'] & string>[];
68
+ constraints: T['constraints'];
69
+ }
70
+
71
+ export function tablesToSchema<
72
+ const T extends Record<string, UntypedTableSchema>,
73
+ >(ctx: ModuleContext, tables: T): TablesToSchema<T> {
74
+ // `TablesToSchema<T>['tables']` is intentionally readonly in the public type,
75
+ // but we need a mutable builder while materializing it from entries.
76
+ type MutableTableDefs = {
77
+ -readonly [AccName in keyof TablesToSchema<T>['tables']]: TablesToSchema<T>['tables'][AccName];
78
+ };
79
+ const tableDefs = Object.create(null) as MutableTableDefs;
80
+ for (const [accName, schema] of Object.entries(tables) as [
81
+ keyof T & string,
82
+ T[keyof T & string],
83
+ ][]) {
84
+ tableDefs[accName] = tableToSchema(
85
+ accName,
86
+ schema,
87
+ schema.tableDef(ctx, accName)
88
+ ) as TablesToSchema<T>['tables'][typeof accName];
89
+ }
90
+
91
+ return {
92
+ tables: tableDefs as TablesToSchema<T>['tables'],
93
+ };
94
+ }
95
+
96
+ export function tableToSchema<
97
+ AccName extends string,
98
+ const T extends UntypedTableSchema,
99
+ >(
100
+ accName: AccName,
101
+ schema: T,
102
+ tableDef: RawTableDefV10
103
+ ): TableToSchema<AccName, T> {
104
+ const getColName = (i: number) =>
105
+ schema.rowType.algebraicType.value.elements[i].name;
106
+
107
+ type AllowedCol = keyof T['rowType']['row'] & string;
108
+ // Build fully-resolved runtime index metadata from the host-facing RawTableDef.
109
+ // This is intentionally separate from `schema.idxs`, which keeps the original
110
+ // user-declared `IndexOpts` shape for type-level inference.
111
+ const resolvedIndexes: UntypedIndex<AllowedCol>[] = tableDef.indexes.map(
112
+ idx => {
113
+ const accessorName = idx.accessorName;
114
+ if (typeof accessorName !== 'string' || accessorName.length === 0) {
115
+ throw new TypeError(
116
+ `Index '${idx.sourceName ?? '<unknown>'}' on table '${tableDef.sourceName}' is missing accessor name`
117
+ );
118
+ }
119
+
120
+ const columnIds =
121
+ idx.algorithm.tag === 'Direct'
122
+ ? [idx.algorithm.value]
123
+ : idx.algorithm.value;
124
+
125
+ const unique = tableDef.constraints.some(
126
+ c =>
127
+ c.data.tag === 'Unique' &&
128
+ c.data.value.columns.every(col => columnIds.includes(col))
129
+ );
130
+
131
+ const algorithm = (
132
+ {
133
+ BTree: 'btree',
134
+ Hash: 'hash',
135
+ Direct: 'direct',
136
+ } as const
137
+ )[idx.algorithm.tag];
138
+
139
+ return {
140
+ name: accessorName,
141
+ unique,
142
+ algorithm,
143
+ columns: columnIds.map(getColName) as AllowedCol[],
144
+ };
145
+ }
146
+ );
147
+
148
+ return {
149
+ // For client,`schama.tableName` will always be there as canonical name.
150
+ // For module, if explicit name is not provided via `name`, accessor name will
151
+ // be used, it is stored as alias in database, hence works in query builder.
152
+ sourceName: schema.tableName || accName,
153
+ accessorName: accName,
154
+ columns: schema.rowType.row, // typed as T[i]['rowType']['row'] under TablesToSchema<T>
155
+ rowType: schema.rowSpacetimeType,
156
+ // Keep declarative indexes in their original shape for type-level consumers.
157
+ indexes: schema.idxs,
158
+ constraints: tableDef.constraints.map(c => ({
159
+ name: c.sourceName,
160
+ constraint: 'unique',
161
+ columns: c.data.value.columns.map(getColName) as [string],
162
+ })),
163
+ // Expose resolved runtime indexes separately so runtime users don't have to
164
+ // reinterpret `indexes` with unsafe casts.
165
+ resolvedIndexes,
166
+ tableDef,
167
+ ...(tableDef.isEvent ? { isEvent: true } : {}),
168
+ };
169
+ }
170
+
171
+ type CompoundTypeCache = Map<
172
+ AlgebraicTypeVariants.Product | AlgebraicTypeVariants.Sum,
173
+ RefBuilder<any, any>
174
+ >;
175
+
176
+ export type ModuleDef = {
177
+ [S in RawModuleDefV10Section as Uncapitalize<S['tag']>]: S['value'];
178
+ };
179
+
180
+ type Section = RawModuleDefV10Section;
181
+
182
+ export class ModuleContext {
183
+ #compoundTypes: CompoundTypeCache = new Map();
184
+
185
+ /**
186
+ * The global module definition that gets populated by calls to `reducer()` and lifecycle hooks.
187
+ */
188
+ #moduleDef: ModuleDef = {
189
+ typespace: { types: [] },
190
+ tables: [],
191
+ reducers: [],
192
+ types: [],
193
+ rowLevelSecurity: [],
194
+ schedules: [],
195
+ procedures: [],
196
+ views: [],
197
+ viewPrimaryKeys: [],
198
+ lifeCycleReducers: [],
199
+ httpHandlers: [],
200
+ httpRoutes: [],
201
+ caseConversionPolicy: { tag: 'SnakeCase' },
202
+ explicitNames: {
203
+ entries: [],
204
+ },
205
+ };
206
+
207
+ get moduleDef(): ModuleDef {
208
+ return this.#moduleDef;
209
+ }
210
+
211
+ rawModuleDefV10(): RawModuleDefV10 {
212
+ const sections: Section[] = [];
213
+
214
+ const push = <T extends Section>(s: T | undefined) => {
215
+ if (s) sections.push(s);
216
+ };
217
+
218
+ const module = this.#moduleDef;
219
+
220
+ push(module.typespace && { tag: 'Typespace', value: module.typespace });
221
+ push(module.types && { tag: 'Types', value: module.types });
222
+ push(module.tables && { tag: 'Tables', value: module.tables });
223
+ push(module.reducers && { tag: 'Reducers', value: module.reducers });
224
+ push(module.procedures && { tag: 'Procedures', value: module.procedures });
225
+ push(module.views && { tag: 'Views', value: module.views });
226
+ push(
227
+ module.viewPrimaryKeys && {
228
+ tag: 'ViewPrimaryKeys',
229
+ value: module.viewPrimaryKeys,
230
+ }
231
+ );
232
+ push(module.schedules && { tag: 'Schedules', value: module.schedules });
233
+ push(
234
+ module.lifeCycleReducers && {
235
+ tag: 'LifeCycleReducers',
236
+ value: module.lifeCycleReducers,
237
+ }
238
+ );
239
+ push(
240
+ module.httpHandlers && {
241
+ tag: 'HttpHandlers',
242
+ value: module.httpHandlers,
243
+ }
244
+ );
245
+ push(
246
+ module.httpRoutes && {
247
+ tag: 'HttpRoutes',
248
+ value: module.httpRoutes,
249
+ }
250
+ );
251
+ push(
252
+ module.rowLevelSecurity && {
253
+ tag: 'RowLevelSecurity',
254
+ value: module.rowLevelSecurity,
255
+ }
256
+ );
257
+ push(
258
+ module.explicitNames && {
259
+ tag: 'ExplicitNames',
260
+ value: module.explicitNames,
261
+ }
262
+ );
263
+ push(
264
+ module.caseConversionPolicy && {
265
+ tag: 'CaseConversionPolicy',
266
+ value: module.caseConversionPolicy,
267
+ }
268
+ );
269
+ return { sections };
270
+ }
271
+
272
+ /**
273
+ * Set the case conversion policy for this module.
274
+ * Called by the settings mechanism.
275
+ */
276
+ setCaseConversionPolicy(policy: CaseConversionPolicy) {
277
+ this.#moduleDef.caseConversionPolicy = policy;
278
+ }
279
+
280
+ get typespace() {
281
+ return this.#moduleDef.typespace;
282
+ }
283
+
284
+ /**
285
+ * Resolves the actual type of a TypeBuilder by following its references until it reaches a non-ref type.
286
+ * @param typespace The typespace to resolve types against.
287
+ * @param typeBuilder The TypeBuilder to resolve.
288
+ * @returns The resolved algebraic type.
289
+ */
290
+ public resolveType<AT extends AlgebraicTypeType>(
291
+ typeBuilder: RefBuilder<any, AT>
292
+ ): AT {
293
+ let ty: AlgebraicType = typeBuilder.algebraicType;
294
+ while (ty.tag === 'Ref') {
295
+ ty = this.typespace.types[ty.value];
296
+ }
297
+ return ty as AT;
298
+ }
299
+
300
+ /**
301
+ * Adds a type to the module definition's typespace as a `Ref` if it is a named compound type (Product or Sum).
302
+ * Otherwise, returns the type as is.
303
+ * @param name
304
+ * @param ty
305
+ * @returns
306
+ */
307
+ public registerTypesRecursively<T extends TypeBuilder<any, AlgebraicType>>(
308
+ typeBuilder: T
309
+ ): T extends SumBuilder<any> | ProductBuilder<any> | RowBuilder<any>
310
+ ? RefBuilder<Infer<T>, InferSpacetimeTypeOfTypeBuilder<T>>
311
+ : T {
312
+ if (
313
+ (typeBuilder instanceof ProductBuilder && !isUnit(typeBuilder)) ||
314
+ typeBuilder instanceof SumBuilder ||
315
+ typeBuilder instanceof RowBuilder
316
+ ) {
317
+ return this.#registerCompoundTypeRecursively(typeBuilder) as any;
318
+ } else if (typeBuilder instanceof OptionBuilder) {
319
+ return new OptionBuilder(
320
+ this.registerTypesRecursively(typeBuilder.value)
321
+ ) as any;
322
+ } else if (typeBuilder instanceof ResultBuilder) {
323
+ return new ResultBuilder(
324
+ this.registerTypesRecursively(typeBuilder.ok),
325
+ this.registerTypesRecursively(typeBuilder.err)
326
+ ) as any;
327
+ } else if (typeBuilder instanceof ArrayBuilder) {
328
+ return new ArrayBuilder(
329
+ this.registerTypesRecursively(typeBuilder.element)
330
+ ) as any;
331
+ } else {
332
+ return typeBuilder as any;
333
+ }
334
+ }
335
+
336
+ #registerCompoundTypeRecursively<
337
+ T extends
338
+ | SumBuilder<VariantsObj>
339
+ | ProductBuilder<ElementsObj>
340
+ | RowBuilder<RowObj>,
341
+ >(typeBuilder: T): RefBuilder<Infer<T>, InferSpacetimeTypeOfTypeBuilder<T>> {
342
+ const ty = typeBuilder.algebraicType;
343
+ // NB! You must ensure that all TypeBuilder passed into this function
344
+ // have a name. This function ensures that nested types always have a
345
+ // name by assigning them one if they are missing it.
346
+ const name = typeBuilder.typeName;
347
+ if (name === undefined) {
348
+ throw new Error(
349
+ `Missing type name for ${typeBuilder.constructor.name ?? 'TypeBuilder'} ${JSON.stringify(typeBuilder)}`
350
+ );
351
+ }
352
+
353
+ let r = this.#compoundTypes.get(ty);
354
+ if (r != null) {
355
+ // Already added to typespace
356
+ return r;
357
+ }
358
+
359
+ // Recursively register nested compound types
360
+ const newTy =
361
+ typeBuilder instanceof RowBuilder || typeBuilder instanceof ProductBuilder
362
+ ? ({
363
+ tag: 'Product',
364
+ value: { elements: [] },
365
+ } as AlgebraicTypeVariants.Product)
366
+ : ({
367
+ tag: 'Sum',
368
+ value: { variants: [] },
369
+ } as AlgebraicTypeVariants.Sum);
370
+
371
+ r = new RefBuilder(this.#moduleDef.typespace.types.length);
372
+ this.#moduleDef.typespace.types.push(newTy);
373
+
374
+ this.#compoundTypes.set(ty, r);
375
+
376
+ if (typeBuilder instanceof RowBuilder) {
377
+ for (const [name, elem] of Object.entries(typeBuilder.row)) {
378
+ (newTy.value as ProductType).elements.push({
379
+ name,
380
+ algebraicType: this.registerTypesRecursively(elem.typeBuilder)
381
+ .algebraicType,
382
+ });
383
+ }
384
+ } else if (typeBuilder instanceof ProductBuilder) {
385
+ for (const [name, elem] of Object.entries(typeBuilder.elements)) {
386
+ (newTy.value as ProductType).elements.push({
387
+ name,
388
+ algebraicType: this.registerTypesRecursively(elem).algebraicType,
389
+ });
390
+ }
391
+ } else if (typeBuilder instanceof SumBuilder) {
392
+ for (const [name, variant] of Object.entries(typeBuilder.variants)) {
393
+ (newTy.value as SumType).variants.push({
394
+ name,
395
+ algebraicType: this.registerTypesRecursively(variant).algebraicType,
396
+ });
397
+ }
398
+ }
399
+
400
+ this.#moduleDef.types.push({
401
+ sourceName: splitName(name),
402
+ ty: r.ref,
403
+ customOrdering: true,
404
+ });
405
+
406
+ return r;
407
+ }
408
+ }
409
+
410
+ function isUnit(typeBuilder: ProductBuilder<ElementsObj>): boolean {
411
+ return (
412
+ typeBuilder.typeName == null &&
413
+ typeBuilder.algebraicType.value.elements.length === 0
414
+ );
415
+ }
416
+
417
+ export function splitName(name: string): RawScopedTypeNameV10 {
418
+ const scope = name.split('.');
419
+ return { sourceName: scope.pop()!, scope };
420
+ }