@yume-chan/struct 0.0.9 → 0.0.10

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 (84) hide show
  1. package/CHANGELOG.json +23 -0
  2. package/CHANGELOG.md +16 -0
  3. package/LICENSE +21 -21
  4. package/README.md +720 -709
  5. package/dts/basic/context.d.ts +10 -18
  6. package/dts/basic/context.d.ts.map +1 -1
  7. package/dts/basic/definition.d.ts +4 -4
  8. package/dts/basic/definition.d.ts.map +1 -1
  9. package/dts/basic/field-value.d.ts +3 -5
  10. package/dts/basic/field-value.d.ts.map +1 -1
  11. package/dts/basic/struct-value.d.ts.map +1 -1
  12. package/dts/struct.d.ts +16 -17
  13. package/dts/struct.d.ts.map +1 -1
  14. package/dts/syncbird.d.ts +20 -0
  15. package/dts/syncbird.d.ts.map +1 -0
  16. package/dts/types/array-buffer.d.ts +13 -12
  17. package/dts/types/array-buffer.d.ts.map +1 -1
  18. package/dts/types/number.d.ts +4 -3
  19. package/dts/types/number.d.ts.map +1 -1
  20. package/dts/types/variable-length-array-buffer.d.ts +4 -4
  21. package/dts/types/variable-length-array-buffer.d.ts.map +1 -1
  22. package/dts/utils.d.ts +3 -2
  23. package/dts/utils.d.ts.map +1 -1
  24. package/esm/basic/context.js.map +1 -1
  25. package/esm/basic/definition.js.map +1 -1
  26. package/esm/basic/field-value.js +1 -2
  27. package/esm/basic/field-value.js.map +1 -1
  28. package/esm/basic/struct-value.js +0 -6
  29. package/esm/basic/struct-value.js.map +1 -1
  30. package/esm/struct.js +27 -17
  31. package/esm/struct.js.map +1 -1
  32. package/esm/syncbird.js +25 -0
  33. package/esm/syncbird.js.map +1 -0
  34. package/esm/types/array-buffer.js +21 -20
  35. package/esm/types/array-buffer.js.map +1 -1
  36. package/esm/types/number.js +9 -8
  37. package/esm/types/number.js.map +1 -1
  38. package/esm/types/variable-length-array-buffer.js +8 -8
  39. package/esm/types/variable-length-array-buffer.js.map +1 -1
  40. package/esm/utils.js +10 -0
  41. package/esm/utils.js.map +1 -1
  42. package/package.json +47 -50
  43. package/src/basic/context.ts +32 -42
  44. package/src/basic/definition.ts +72 -69
  45. package/src/basic/field-value.ts +67 -73
  46. package/src/basic/index.ts +4 -4
  47. package/src/basic/struct-value.ts +39 -45
  48. package/src/index.ts +5 -5
  49. package/src/struct.ts +595 -575
  50. package/src/syncbird.ts +53 -0
  51. package/src/types/array-buffer.ts +194 -184
  52. package/src/types/fixed-length-array-buffer.ts +17 -17
  53. package/src/types/index.ts +4 -4
  54. package/src/types/number.ts +114 -100
  55. package/src/types/variable-length-array-buffer.ts +142 -145
  56. package/src/utils.ts +62 -51
  57. package/cjs/basic/context.js +0 -7
  58. package/cjs/basic/context.js.map +0 -1
  59. package/cjs/basic/definition.js +0 -25
  60. package/cjs/basic/definition.js.map +0 -1
  61. package/cjs/basic/field-value.js +0 -40
  62. package/cjs/basic/field-value.js.map +0 -1
  63. package/cjs/basic/index.js +0 -8
  64. package/cjs/basic/index.js.map +0 -1
  65. package/cjs/basic/struct-value.js +0 -46
  66. package/cjs/basic/struct-value.js.map +0 -1
  67. package/cjs/index.js +0 -11
  68. package/cjs/index.js.map +0 -1
  69. package/cjs/struct.js +0 -177
  70. package/cjs/struct.js.map +0 -1
  71. package/cjs/types/array-buffer.js +0 -118
  72. package/cjs/types/array-buffer.js.map +0 -1
  73. package/cjs/types/fixed-length-array-buffer.js +0 -12
  74. package/cjs/types/fixed-length-array-buffer.js.map +0 -1
  75. package/cjs/types/index.js +0 -8
  76. package/cjs/types/index.js.map +0 -1
  77. package/cjs/types/number.js +0 -52
  78. package/cjs/types/number.js.map +0 -1
  79. package/cjs/types/variable-length-array-buffer.js +0 -77
  80. package/cjs/types/variable-length-array-buffer.js.map +0 -1
  81. package/cjs/utils.js +0 -11
  82. package/cjs/utils.js.map +0 -1
  83. package/tsconfig.cjs.json +0 -11
  84. package/tsconfig.esm.json +0 -15
package/src/struct.ts CHANGED
@@ -1,575 +1,595 @@
1
- import { StructDefaultOptions, StructDeserializationContext, StructFieldDefinition, StructFieldValue, StructOptions, StructSerializationContext, StructValue } from './basic';
2
- import { ArrayBufferFieldType, ArrayBufferLikeFieldType, FixedLengthArrayBufferLikeFieldDefinition, FixedLengthArrayBufferLikeFieldOptions, LengthField, NumberFieldDefinition, NumberFieldType, StringFieldType, Uint8ClampedArrayFieldType, VariableLengthArrayBufferLikeFieldDefinition, VariableLengthArrayBufferLikeFieldOptions } from './types';
3
- import { Awaited, Evaluate, Identity, Overwrite } from './utils';
4
-
5
- export interface StructLike<TValue> {
6
- deserialize(context: StructDeserializationContext): Promise<TValue>;
7
- }
8
-
9
- /**
10
- * Extract the value type of the specified `Struct`
11
- *
12
- * The lack of generic constraint is on purpose to allow `StructLike` types
13
- */
14
- export type StructValueType<T extends StructLike<any>> =
15
- Awaited<ReturnType<T['deserialize']>>;
16
-
17
- /**
18
- * Create a new `Struct` type with `TDescriptor` appended
19
- */
20
- type AddFieldDescriptor<
21
- TFields extends object,
22
- TOmitInitKey extends PropertyKey,
23
- TExtra extends object,
24
- TPostDeserialized,
25
- TFieldName extends PropertyKey,
26
- TDefinition extends StructFieldDefinition<any, any, any>
27
- > =
28
- Identity<Struct<
29
- // Merge two types
30
- // Evaluate immediately to optimize editor hover tooltip
31
- Evaluate<TFields & Record<TFieldName, TDefinition['TValue']>>,
32
- // Merge two `TOmitInitKey`s
33
- TOmitInitKey | TDefinition['TOmitInitKey'],
34
- TExtra,
35
- TPostDeserialized
36
- >>;
37
-
38
- /**
39
- * Overload methods to add an array buffer like field
40
- */
41
- interface ArrayBufferLikeFieldCreator<
42
- TFields extends object,
43
- TOmitInitKey extends PropertyKey,
44
- TExtra extends object,
45
- TPostDeserialized
46
- > {
47
- /**
48
- * Append a fixed-length array to the `Struct`
49
- *
50
- * @param name Name of the field
51
- * @param type `Array.SubType.ArrayBuffer` or `Array.SubType.String`
52
- * @param options Fixed-length array options
53
- * @param typescriptType Type of the field in TypeScript.
54
- * For example, if this field is a string, you can declare it as a string enum or literal union.
55
- */
56
- <
57
- TName extends PropertyKey,
58
- TType extends ArrayBufferLikeFieldType,
59
- TTypeScriptType = TType['TTypeScriptType'],
60
- >(
61
- name: TName,
62
- type: TType,
63
- options: FixedLengthArrayBufferLikeFieldOptions,
64
- typescriptType?: TTypeScriptType,
65
- ): AddFieldDescriptor<
66
- TFields,
67
- TOmitInitKey,
68
- TExtra,
69
- TPostDeserialized,
70
- TName,
71
- FixedLengthArrayBufferLikeFieldDefinition<
72
- TType,
73
- FixedLengthArrayBufferLikeFieldOptions
74
- >
75
- >;
76
-
77
- /**
78
- * Append a variable-length array to the `Struct`
79
- */
80
- <
81
- TName extends PropertyKey,
82
- TType extends ArrayBufferLikeFieldType,
83
- TOptions extends VariableLengthArrayBufferLikeFieldOptions<TFields>,
84
- TTypeScriptType = TType['TTypeScriptType'],
85
- >(
86
- name: TName,
87
- type: TType,
88
- options: TOptions,
89
- typescriptType?: TTypeScriptType,
90
- ): AddFieldDescriptor<
91
- TFields,
92
- TOmitInitKey,
93
- TExtra,
94
- TPostDeserialized,
95
- TName,
96
- VariableLengthArrayBufferLikeFieldDefinition<
97
- TType,
98
- TOptions
99
- >
100
- >;
101
- }
102
-
103
- /**
104
- * Similar to `ArrayBufferLikeFieldCreator`, but bind to a `ArrayBufferLikeFieldType`
105
- */
106
- interface ArrayBufferTypeFieldDefinitionCreator<
107
- TFields extends object,
108
- TOmitInitKey extends PropertyKey,
109
- TExtra extends object,
110
- TPostDeserialized,
111
- TType extends ArrayBufferLikeFieldType
112
- > {
113
- <
114
- TName extends PropertyKey,
115
- TTypeScriptType = TType['TTypeScriptType'],
116
- >(
117
- name: TName,
118
- options: FixedLengthArrayBufferLikeFieldOptions,
119
- typescriptType?: TTypeScriptType,
120
- ): AddFieldDescriptor<
121
- TFields,
122
- TOmitInitKey,
123
- TExtra,
124
- TPostDeserialized,
125
- TName,
126
- FixedLengthArrayBufferLikeFieldDefinition<
127
- TType,
128
- FixedLengthArrayBufferLikeFieldOptions
129
- >
130
- >;
131
-
132
- <
133
- TName extends PropertyKey,
134
- TLengthField extends LengthField<TFields>,
135
- TOptions extends VariableLengthArrayBufferLikeFieldOptions<TFields, TLengthField>,
136
- TTypeScriptType = TType['TTypeScriptType'],
137
- >(
138
- name: TName,
139
- options: TOptions,
140
- typescriptType?: TTypeScriptType,
141
- ): AddFieldDescriptor<
142
- TFields,
143
- TOmitInitKey,
144
- TExtra,
145
- TPostDeserialized,
146
- TName,
147
- VariableLengthArrayBufferLikeFieldDefinition<
148
- TType,
149
- TOptions
150
- >
151
- >;
152
- }
153
-
154
- export type StructPostDeserialized<TFields, TPostDeserialized> =
155
- (this: TFields, object: TFields) => TPostDeserialized;
156
-
157
- export type StructDeserializedResult<TFields extends object, TExtra extends object, TPostDeserialized> =
158
- TPostDeserialized extends undefined ? Overwrite<TExtra, TFields> : TPostDeserialized;
159
-
160
- export class Struct<
161
- TFields extends object = {},
162
- TOmitInitKey extends PropertyKey = never,
163
- TExtra extends object = {},
164
- TPostDeserialized = undefined,
165
- > implements StructLike<StructDeserializedResult<TFields, TExtra, TPostDeserialized>>{
166
- public readonly TFields!: TFields;
167
-
168
- public readonly TOmitInitKey!: TOmitInitKey;
169
-
170
- public readonly TExtra!: TExtra;
171
-
172
- public readonly TInit!: Evaluate<Omit<TFields, TOmitInitKey>>;
173
-
174
- public readonly TDeserializeResult!: StructDeserializedResult<TFields, TExtra, TPostDeserialized>;
175
-
176
- public readonly options: Readonly<StructOptions>;
177
-
178
- private _size = 0;
179
- /**
180
- * Gets the static size (exclude fields that can change size at runtime)
181
- */
182
- public get size() { return this._size; }
183
-
184
- private _fields: [name: PropertyKey, definition: StructFieldDefinition<any, any, any>][] = [];
185
-
186
- private _extra: PropertyDescriptorMap = {};
187
-
188
- private _postDeserialized?: StructPostDeserialized<any, any>;
189
-
190
- public constructor(options?: Partial<Readonly<StructOptions>>) {
191
- this.options = { ...StructDefaultOptions, ...options };
192
- }
193
-
194
- /**
195
- * Appends a `StructFieldDefinition` to the `Struct
196
- */
197
- public field<
198
- TName extends PropertyKey,
199
- TDefinition extends StructFieldDefinition<any, any, any>
200
- >(
201
- name: TName,
202
- definition: TDefinition,
203
- ): AddFieldDescriptor<
204
- TFields,
205
- TOmitInitKey,
206
- TExtra,
207
- TPostDeserialized,
208
- TName,
209
- TDefinition
210
- > {
211
- for (const field of this._fields) {
212
- if (field[0] === name) {
213
- throw new Error(`This struct already have a field with name '${name}'`);
214
- }
215
- }
216
-
217
- this._fields.push([name, definition]);
218
-
219
- const size = definition.getSize();
220
- this._size += size;
221
-
222
- // Force cast `this` to another type
223
- return this as any;
224
- }
225
-
226
- /**
227
- * Merges (flats) another `Struct`'s fields and extra fields into this one.
228
- */
229
- public fields<TOther extends Struct<any, any, any, any>>(
230
- other: TOther
231
- ): Struct<
232
- TFields & TOther['TFields'],
233
- TOmitInitKey | TOther['TOmitInitKey'],
234
- TExtra & TOther['TExtra'],
235
- TPostDeserialized
236
- > {
237
- for (const field of other._fields) {
238
- this._fields.push(field);
239
- }
240
- this._size += other._size;
241
- Object.assign(this._extra, other._extra);
242
- return this as any;
243
- }
244
-
245
- private number<
246
- TName extends PropertyKey,
247
- TType extends NumberFieldType = NumberFieldType,
248
- TTypeScriptType = TType['TTypeScriptType']
249
- >(
250
- name: TName,
251
- type: TType,
252
- _typescriptType?: TTypeScriptType,
253
- ) {
254
- return this.field(
255
- name,
256
- new NumberFieldDefinition(type, _typescriptType),
257
- );
258
- }
259
-
260
- /**
261
- * Appends an `int8` field to the `Struct`
262
- */
263
- public int8<
264
- TName extends PropertyKey,
265
- TTypeScriptType = (typeof NumberFieldType)['Uint8']['TTypeScriptType']
266
- >(
267
- name: TName,
268
- _typescriptType?: TTypeScriptType,
269
- ) {
270
- return this.number(
271
- name,
272
- NumberFieldType.Int8,
273
- _typescriptType
274
- );
275
- }
276
-
277
- /**
278
- * Appends an `uint8` field to the `Struct`
279
- */
280
- public uint8<
281
- TName extends PropertyKey,
282
- TTypeScriptType = (typeof NumberFieldType)['Uint8']['TTypeScriptType']
283
- >(
284
- name: TName,
285
- _typescriptType?: TTypeScriptType,
286
- ) {
287
- return this.number(
288
- name,
289
- NumberFieldType.Uint8,
290
- _typescriptType
291
- );
292
- }
293
-
294
- /**
295
- * Appends an `int16` field to the `Struct`
296
- */
297
- public int16<
298
- TName extends PropertyKey,
299
- TTypeScriptType = (typeof NumberFieldType)['Uint16']['TTypeScriptType']
300
- >(
301
- name: TName,
302
- _typescriptType?: TTypeScriptType,
303
- ) {
304
- return this.number(
305
- name,
306
- NumberFieldType.Int16,
307
- _typescriptType
308
- );
309
- }
310
-
311
- /**
312
- * Appends an `uint16` field to the `Struct`
313
- */
314
- public uint16<
315
- TName extends PropertyKey,
316
- TTypeScriptType = (typeof NumberFieldType)['Uint16']['TTypeScriptType']
317
- >(
318
- name: TName,
319
- _typescriptType?: TTypeScriptType,
320
- ) {
321
- return this.number(
322
- name,
323
- NumberFieldType.Uint16,
324
- _typescriptType
325
- );
326
- }
327
-
328
- /**
329
- * Appends an `int32` field to the `Struct`
330
- */
331
- public int32<
332
- TName extends PropertyKey,
333
- TTypeScriptType = (typeof NumberFieldType)['Int32']['TTypeScriptType']
334
- >(
335
- name: TName,
336
- _typescriptType?: TTypeScriptType,
337
- ) {
338
- return this.number(
339
- name,
340
- NumberFieldType.Int32,
341
- _typescriptType
342
- );
343
- }
344
-
345
- /**
346
- * Appends an `uint32` field to the `Struct`
347
- */
348
- public uint32<
349
- TName extends PropertyKey,
350
- TTypeScriptType = (typeof NumberFieldType)['Uint32']['TTypeScriptType']
351
- >(
352
- name: TName,
353
- typescriptType?: TTypeScriptType,
354
- ) {
355
- return this.number(
356
- name,
357
- NumberFieldType.Uint32,
358
- typescriptType
359
- );
360
- }
361
-
362
- /**
363
- * Appends an `int64` field to the `Struct`
364
- *
365
- * Requires native `BigInt` support
366
- */
367
- public int64<
368
- TName extends PropertyKey,
369
- TTypeScriptType = (typeof NumberFieldType)['Int64']['TTypeScriptType']
370
- >(
371
- name: TName,
372
- _typescriptType?: TTypeScriptType,
373
- ) {
374
- return this.number(
375
- name,
376
- NumberFieldType.Int64,
377
- _typescriptType
378
- );
379
- }
380
-
381
- /**
382
- * Appends an `uint64` field to the `Struct`
383
- *
384
- * Requires native `BigInt` support
385
- */
386
- public uint64<
387
- TName extends PropertyKey,
388
- TTypeScriptType = (typeof NumberFieldType)['Uint64']['TTypeScriptType']
389
- >(
390
- name: TName,
391
- _typescriptType?: TTypeScriptType,
392
- ) {
393
- return this.number(
394
- name,
395
- NumberFieldType.Uint64,
396
- _typescriptType
397
- );
398
- }
399
-
400
- private arrayBufferLike: ArrayBufferLikeFieldCreator<
401
- TFields,
402
- TOmitInitKey,
403
- TExtra,
404
- TPostDeserialized
405
- > = (
406
- name: PropertyKey,
407
- type: ArrayBufferLikeFieldType,
408
- options: FixedLengthArrayBufferLikeFieldOptions | VariableLengthArrayBufferLikeFieldOptions
409
- ): any => {
410
- if ('length' in options) {
411
- return this.field(
412
- name,
413
- new FixedLengthArrayBufferLikeFieldDefinition(type, options),
414
- );
415
- } else {
416
- return this.field(
417
- name,
418
- new VariableLengthArrayBufferLikeFieldDefinition(type, options),
419
- );
420
- }
421
- };
422
-
423
- public arrayBuffer: ArrayBufferTypeFieldDefinitionCreator<
424
- TFields,
425
- TOmitInitKey,
426
- TExtra,
427
- TPostDeserialized,
428
- ArrayBufferFieldType
429
- > = (
430
- name: PropertyKey,
431
- options: any
432
- ): any => {
433
- return this.arrayBufferLike(name, ArrayBufferFieldType.instance, options);
434
- };
435
-
436
- public uint8ClampedArray: ArrayBufferTypeFieldDefinitionCreator<
437
- TFields,
438
- TOmitInitKey,
439
- TExtra,
440
- TPostDeserialized,
441
- Uint8ClampedArrayFieldType
442
- > = (
443
- name: PropertyKey,
444
- options: any
445
- ): any => {
446
- return this.arrayBufferLike(name, Uint8ClampedArrayFieldType.instance, options);
447
- };
448
-
449
- public string: ArrayBufferTypeFieldDefinitionCreator<
450
- TFields,
451
- TOmitInitKey,
452
- TExtra,
453
- TPostDeserialized,
454
- StringFieldType
455
- > = (
456
- name: PropertyKey,
457
- options: any
458
- ): any => {
459
- return this.arrayBufferLike(name, StringFieldType.instance, options);
460
- };
461
-
462
- /**
463
- * Adds some extra fields into every Struct value.
464
- *
465
- * Extra fields will not affect serialize or deserialize process.
466
- *
467
- * Multiple calls to `extra` will merge all values together.
468
- *
469
- * @param value
470
- * An object containing anything you want to add to the result object. Accessors and methods are also allowed.
471
- */
472
- public extra<T extends Record<
473
- // This trick disallows any keys that are already in `TValue`
474
- Exclude<
475
- keyof T,
476
- Exclude<keyof T, keyof TFields>
477
- >,
478
- never
479
- >>(
480
- value: T & ThisType<Overwrite<Overwrite<TExtra, T>, TFields>>
481
- ): Struct<
482
- TFields,
483
- TOmitInitKey,
484
- Overwrite<TExtra, T>,
485
- TPostDeserialized
486
- > {
487
- Object.assign(this._extra, Object.getOwnPropertyDescriptors(value));
488
- return this as any;
489
- }
490
-
491
- /**
492
- * Registers (or replaces) a custom callback to be run after deserialized.
493
- *
494
- * A callback returning `never` (always throw an error)
495
- * will also change the return type of `deserialize` to `never`.
496
- */
497
- public postDeserialize(
498
- callback: StructPostDeserialized<TFields, never>
499
- ): Struct<TFields, TOmitInitKey, TExtra, never>;
500
- /**
501
- * Registers (or replaces) a custom callback to be run after deserialized.
502
- *
503
- * A callback returning `void` means it modify the result object in-place
504
- * (or doesn't modify it at all), so `deserialize` will still return the result object.
505
- */
506
- public postDeserialize(
507
- callback?: StructPostDeserialized<TFields, void>
508
- ): Struct<TFields, TOmitInitKey, TExtra, undefined>;
509
- /**
510
- * Registers (or replaces) a custom callback to be run after deserialized.
511
- *
512
- * A callback returning anything other than `undefined`
513
- * will `deserialize` to return that object instead.
514
- */
515
- public postDeserialize<TPostSerialize>(
516
- callback?: StructPostDeserialized<TFields, TPostSerialize>
517
- ): Struct<TFields, TOmitInitKey, TExtra, TPostSerialize>;
518
- public postDeserialize(
519
- callback?: StructPostDeserialized<TFields, any>
520
- ) {
521
- this._postDeserialized = callback;
522
- return this as any;
523
- }
524
-
525
- public async deserialize(
526
- context: StructDeserializationContext
527
- ): Promise<StructDeserializedResult<TFields, TExtra, TPostDeserialized>> {
528
- const value = new StructValue();
529
- Object.defineProperties(value.value, this._extra);
530
-
531
- for (const [name, definition] of this._fields) {
532
- const fieldValue = await definition.deserialize(this.options, context, value);
533
- value.set(name, fieldValue);
534
- }
535
-
536
- if (this._postDeserialized) {
537
- const object = value.value as TFields;
538
- const result = this._postDeserialized.call(object, object);
539
- if (result) {
540
- return result;
541
- }
542
- }
543
-
544
- return value.value as any;
545
- }
546
-
547
- public serialize(init: Evaluate<Omit<TFields, TOmitInitKey>>, context: StructSerializationContext): ArrayBuffer {
548
- const value = new StructValue();
549
-
550
- for (const [name, definition] of this._fields) {
551
- const fieldValue = definition.create(this.options, context, value, (init as any)[name]);
552
- value.set(name, fieldValue);
553
- }
554
-
555
- let structSize = 0;
556
- const fieldsInfo: { fieldValue: StructFieldValue, size: number; }[] = [];
557
-
558
- for (const [name] of this._fields) {
559
- const fieldValue = value.get(name);
560
- const size = fieldValue.getSize();
561
- fieldsInfo.push({ fieldValue, size });
562
- structSize += size;
563
- }
564
-
565
- const buffer = new ArrayBuffer(structSize);
566
- const dataView = new DataView(buffer);
567
- let offset = 0;
568
- for (const { fieldValue, size } of fieldsInfo) {
569
- fieldValue.serialize(dataView, offset, context);
570
- offset += size;
571
- }
572
-
573
- return buffer;
574
- }
575
- }
1
+ import type { StructAsyncDeserializeStream, StructDeserializeStream, StructFieldDefinition, StructFieldValue, StructOptions } from './basic';
2
+ import { StructDefaultOptions, StructValue } from './basic';
3
+ import { Syncbird } from "./syncbird";
4
+ import { ArrayBufferFieldType, ArrayBufferLikeFieldType, FixedLengthArrayBufferLikeFieldDefinition, FixedLengthArrayBufferLikeFieldOptions, LengthField, NumberFieldDefinition, NumberFieldType, StringFieldType, Uint8ClampedArrayFieldType, VariableLengthArrayBufferLikeFieldDefinition, VariableLengthArrayBufferLikeFieldOptions } from './types';
5
+ import { Evaluate, Identity, Overwrite, ValueOrPromise } from "./utils";
6
+
7
+ export interface StructLike<TValue> {
8
+ deserialize(stream: StructDeserializeStream | StructAsyncDeserializeStream): Promise<TValue>;
9
+ }
10
+
11
+ /**
12
+ * Extract the value type of the specified `Struct`
13
+ */
14
+ export type StructValueType<T extends StructLike<any>> =
15
+ Awaited<ReturnType<T['deserialize']>>;
16
+
17
+ /**
18
+ * Create a new `Struct` type with `TDefinition` appended
19
+ */
20
+ type AddFieldDescriptor<
21
+ TFields extends object,
22
+ TOmitInitKey extends PropertyKey,
23
+ TExtra extends object,
24
+ TPostDeserialized,
25
+ TFieldName extends PropertyKey,
26
+ TDefinition extends StructFieldDefinition<any, any, any>
27
+ > =
28
+ Identity<Struct<
29
+ // Merge two types
30
+ // Evaluate immediately to optimize editor hover tooltip
31
+ Evaluate<TFields & Record<TFieldName, TDefinition['TValue']>>,
32
+ // Merge two `TOmitInitKey`s
33
+ TOmitInitKey | TDefinition['TOmitInitKey'],
34
+ TExtra,
35
+ TPostDeserialized
36
+ >>;
37
+
38
+ /**
39
+ * Overload methods to add an array buffer like field
40
+ */
41
+ interface ArrayBufferLikeFieldCreator<
42
+ TFields extends object,
43
+ TOmitInitKey extends PropertyKey,
44
+ TExtra extends object,
45
+ TPostDeserialized
46
+ > {
47
+ /**
48
+ * Append a fixed-length array buffer like field to the `Struct`
49
+ *
50
+ * @param name Name of the field
51
+ * @param type `Array.SubType.ArrayBuffer` or `Array.SubType.String`
52
+ * @param options Fixed-length array options
53
+ * @param typescriptType Type of the field in TypeScript.
54
+ * For example, if this field is a string, you can declare it as a string enum or literal union.
55
+ */
56
+ <
57
+ TName extends PropertyKey,
58
+ TType extends ArrayBufferLikeFieldType<any, any>,
59
+ TTypeScriptType = TType['TTypeScriptType'],
60
+ >(
61
+ name: TName,
62
+ type: TType,
63
+ options: FixedLengthArrayBufferLikeFieldOptions,
64
+ typescriptType?: TTypeScriptType,
65
+ ): AddFieldDescriptor<
66
+ TFields,
67
+ TOmitInitKey,
68
+ TExtra,
69
+ TPostDeserialized,
70
+ TName,
71
+ FixedLengthArrayBufferLikeFieldDefinition<
72
+ TType,
73
+ FixedLengthArrayBufferLikeFieldOptions
74
+ >
75
+ >;
76
+
77
+ /**
78
+ * Append a variable-length array buffer like field to the `Struct`
79
+ */
80
+ <
81
+ TName extends PropertyKey,
82
+ TType extends ArrayBufferLikeFieldType<any, any>,
83
+ TOptions extends VariableLengthArrayBufferLikeFieldOptions<TFields>,
84
+ TTypeScriptType = TType['TTypeScriptType'],
85
+ >(
86
+ name: TName,
87
+ type: TType,
88
+ options: TOptions,
89
+ typescriptType?: TTypeScriptType,
90
+ ): AddFieldDescriptor<
91
+ TFields,
92
+ TOmitInitKey,
93
+ TExtra,
94
+ TPostDeserialized,
95
+ TName,
96
+ VariableLengthArrayBufferLikeFieldDefinition<
97
+ TType,
98
+ TOptions
99
+ >
100
+ >;
101
+ }
102
+
103
+ /**
104
+ * Similar to `ArrayBufferLikeFieldCreator`, but bind to `TType`
105
+ */
106
+ interface BindedArrayBufferLikeFieldDefinitionCreator<
107
+ TFields extends object,
108
+ TOmitInitKey extends PropertyKey,
109
+ TExtra extends object,
110
+ TPostDeserialized,
111
+ TType extends ArrayBufferLikeFieldType<any, any>
112
+ > {
113
+ <
114
+ TName extends PropertyKey,
115
+ TTypeScriptType = TType['TTypeScriptType'],
116
+ >(
117
+ name: TName,
118
+ options: FixedLengthArrayBufferLikeFieldOptions,
119
+ typescriptType?: TTypeScriptType,
120
+ ): AddFieldDescriptor<
121
+ TFields,
122
+ TOmitInitKey,
123
+ TExtra,
124
+ TPostDeserialized,
125
+ TName,
126
+ FixedLengthArrayBufferLikeFieldDefinition<
127
+ TType,
128
+ FixedLengthArrayBufferLikeFieldOptions
129
+ >
130
+ >;
131
+
132
+ <
133
+ TName extends PropertyKey,
134
+ TLengthField extends LengthField<TFields>,
135
+ TOptions extends VariableLengthArrayBufferLikeFieldOptions<TFields, TLengthField>,
136
+ TTypeScriptType = TType['TTypeScriptType'],
137
+ >(
138
+ name: TName,
139
+ options: TOptions,
140
+ typescriptType?: TTypeScriptType,
141
+ ): AddFieldDescriptor<
142
+ TFields,
143
+ TOmitInitKey,
144
+ TExtra,
145
+ TPostDeserialized,
146
+ TName,
147
+ VariableLengthArrayBufferLikeFieldDefinition<
148
+ TType,
149
+ TOptions
150
+ >
151
+ >;
152
+ }
153
+
154
+ export type StructPostDeserialized<TFields, TPostDeserialized> =
155
+ (this: TFields, object: TFields) => TPostDeserialized;
156
+
157
+ export type StructDeserializedResult<TFields extends object, TExtra extends object, TPostDeserialized> =
158
+ TPostDeserialized extends undefined ? Overwrite<TExtra, TFields> : TPostDeserialized;
159
+
160
+ export class Struct<
161
+ TFields extends object = {},
162
+ TOmitInitKey extends PropertyKey = never,
163
+ TExtra extends object = {},
164
+ TPostDeserialized = undefined,
165
+ > implements StructLike<StructDeserializedResult<TFields, TExtra, TPostDeserialized>>{
166
+ public readonly TFields!: TFields;
167
+
168
+ public readonly TOmitInitKey!: TOmitInitKey;
169
+
170
+ public readonly TExtra!: TExtra;
171
+
172
+ public readonly TInit!: Evaluate<Omit<TFields, TOmitInitKey>>;
173
+
174
+ public readonly TDeserializeResult!: StructDeserializedResult<TFields, TExtra, TPostDeserialized>;
175
+
176
+ public readonly options: Readonly<StructOptions>;
177
+
178
+ private _size = 0;
179
+ /**
180
+ * Gets the static size (exclude fields that can change size at runtime)
181
+ */
182
+ public get size() { return this._size; }
183
+
184
+ private _fields: [name: PropertyKey, definition: StructFieldDefinition<any, any, any>][] = [];
185
+
186
+ private _extra: PropertyDescriptorMap = {};
187
+
188
+ private _postDeserialized?: StructPostDeserialized<any, any>;
189
+
190
+ public constructor(options?: Partial<Readonly<StructOptions>>) {
191
+ this.options = { ...StructDefaultOptions, ...options };
192
+ }
193
+
194
+ /**
195
+ * Appends a `StructFieldDefinition` to the `Struct
196
+ */
197
+ public field<
198
+ TName extends PropertyKey,
199
+ TDefinition extends StructFieldDefinition<any, any, any>
200
+ >(
201
+ name: TName,
202
+ definition: TDefinition,
203
+ ): AddFieldDescriptor<
204
+ TFields,
205
+ TOmitInitKey,
206
+ TExtra,
207
+ TPostDeserialized,
208
+ TName,
209
+ TDefinition
210
+ > {
211
+ for (const field of this._fields) {
212
+ if (field[0] === name) {
213
+ throw new Error(`This struct already have a field with name '${name}'`);
214
+ }
215
+ }
216
+
217
+ this._fields.push([name, definition]);
218
+
219
+ const size = definition.getSize();
220
+ this._size += size;
221
+
222
+ // Force cast `this` to another type
223
+ return this as any;
224
+ }
225
+
226
+ /**
227
+ * Merges (flats) another `Struct`'s fields and extra fields into this one.
228
+ */
229
+ public fields<TOther extends Struct<any, any, any, any>>(
230
+ other: TOther
231
+ ): Struct<
232
+ TFields & TOther['TFields'],
233
+ TOmitInitKey | TOther['TOmitInitKey'],
234
+ TExtra & TOther['TExtra'],
235
+ TPostDeserialized
236
+ > {
237
+ for (const field of other._fields) {
238
+ this._fields.push(field);
239
+ }
240
+ this._size += other._size;
241
+ Object.assign(this._extra, other._extra);
242
+ return this as any;
243
+ }
244
+
245
+ private number<
246
+ TName extends PropertyKey,
247
+ TType extends NumberFieldType = NumberFieldType,
248
+ TTypeScriptType = TType['TTypeScriptType']
249
+ >(
250
+ name: TName,
251
+ type: TType,
252
+ _typescriptType?: TTypeScriptType,
253
+ ) {
254
+ return this.field(
255
+ name,
256
+ new NumberFieldDefinition(type, _typescriptType),
257
+ );
258
+ }
259
+
260
+ /**
261
+ * Appends an `int8` field to the `Struct`
262
+ */
263
+ public int8<
264
+ TName extends PropertyKey,
265
+ TTypeScriptType = (typeof NumberFieldType)['Uint8']['TTypeScriptType']
266
+ >(
267
+ name: TName,
268
+ _typescriptType?: TTypeScriptType,
269
+ ) {
270
+ return this.number(
271
+ name,
272
+ NumberFieldType.Int8,
273
+ _typescriptType
274
+ );
275
+ }
276
+
277
+ /**
278
+ * Appends an `uint8` field to the `Struct`
279
+ */
280
+ public uint8<
281
+ TName extends PropertyKey,
282
+ TTypeScriptType = (typeof NumberFieldType)['Uint8']['TTypeScriptType']
283
+ >(
284
+ name: TName,
285
+ _typescriptType?: TTypeScriptType,
286
+ ) {
287
+ return this.number(
288
+ name,
289
+ NumberFieldType.Uint8,
290
+ _typescriptType
291
+ );
292
+ }
293
+
294
+ /**
295
+ * Appends an `int16` field to the `Struct`
296
+ */
297
+ public int16<
298
+ TName extends PropertyKey,
299
+ TTypeScriptType = (typeof NumberFieldType)['Uint16']['TTypeScriptType']
300
+ >(
301
+ name: TName,
302
+ _typescriptType?: TTypeScriptType,
303
+ ) {
304
+ return this.number(
305
+ name,
306
+ NumberFieldType.Int16,
307
+ _typescriptType
308
+ );
309
+ }
310
+
311
+ /**
312
+ * Appends an `uint16` field to the `Struct`
313
+ */
314
+ public uint16<
315
+ TName extends PropertyKey,
316
+ TTypeScriptType = (typeof NumberFieldType)['Uint16']['TTypeScriptType']
317
+ >(
318
+ name: TName,
319
+ _typescriptType?: TTypeScriptType,
320
+ ) {
321
+ return this.number(
322
+ name,
323
+ NumberFieldType.Uint16,
324
+ _typescriptType
325
+ );
326
+ }
327
+
328
+ /**
329
+ * Appends an `int32` field to the `Struct`
330
+ */
331
+ public int32<
332
+ TName extends PropertyKey,
333
+ TTypeScriptType = (typeof NumberFieldType)['Int32']['TTypeScriptType']
334
+ >(
335
+ name: TName,
336
+ _typescriptType?: TTypeScriptType,
337
+ ) {
338
+ return this.number(
339
+ name,
340
+ NumberFieldType.Int32,
341
+ _typescriptType
342
+ );
343
+ }
344
+
345
+ /**
346
+ * Appends an `uint32` field to the `Struct`
347
+ */
348
+ public uint32<
349
+ TName extends PropertyKey,
350
+ TTypeScriptType = (typeof NumberFieldType)['Uint32']['TTypeScriptType']
351
+ >(
352
+ name: TName,
353
+ typescriptType?: TTypeScriptType,
354
+ ) {
355
+ return this.number(
356
+ name,
357
+ NumberFieldType.Uint32,
358
+ typescriptType
359
+ );
360
+ }
361
+
362
+ /**
363
+ * Appends an `int64` field to the `Struct`
364
+ *
365
+ * Requires native `BigInt` support
366
+ */
367
+ public int64<
368
+ TName extends PropertyKey,
369
+ TTypeScriptType = (typeof NumberFieldType)['Int64']['TTypeScriptType']
370
+ >(
371
+ name: TName,
372
+ _typescriptType?: TTypeScriptType,
373
+ ) {
374
+ return this.number(
375
+ name,
376
+ NumberFieldType.Int64,
377
+ _typescriptType
378
+ );
379
+ }
380
+
381
+ /**
382
+ * Appends an `uint64` field to the `Struct`
383
+ *
384
+ * Requires native `BigInt` support
385
+ */
386
+ public uint64<
387
+ TName extends PropertyKey,
388
+ TTypeScriptType = (typeof NumberFieldType)['Uint64']['TTypeScriptType']
389
+ >(
390
+ name: TName,
391
+ _typescriptType?: TTypeScriptType,
392
+ ) {
393
+ return this.number(
394
+ name,
395
+ NumberFieldType.Uint64,
396
+ _typescriptType
397
+ );
398
+ }
399
+
400
+ private arrayBufferLike: ArrayBufferLikeFieldCreator<
401
+ TFields,
402
+ TOmitInitKey,
403
+ TExtra,
404
+ TPostDeserialized
405
+ > = (
406
+ name: PropertyKey,
407
+ type: ArrayBufferLikeFieldType,
408
+ options: FixedLengthArrayBufferLikeFieldOptions | VariableLengthArrayBufferLikeFieldOptions
409
+ ): any => {
410
+ if ('length' in options) {
411
+ return this.field(
412
+ name,
413
+ new FixedLengthArrayBufferLikeFieldDefinition(type, options),
414
+ );
415
+ } else {
416
+ return this.field(
417
+ name,
418
+ new VariableLengthArrayBufferLikeFieldDefinition(type, options),
419
+ );
420
+ }
421
+ };
422
+
423
+ public arrayBuffer: BindedArrayBufferLikeFieldDefinitionCreator<
424
+ TFields,
425
+ TOmitInitKey,
426
+ TExtra,
427
+ TPostDeserialized,
428
+ ArrayBufferFieldType
429
+ > = (
430
+ name: PropertyKey,
431
+ options: any
432
+ ): any => {
433
+ return this.arrayBufferLike(name, ArrayBufferFieldType.instance, options);
434
+ };
435
+
436
+ public uint8ClampedArray: BindedArrayBufferLikeFieldDefinitionCreator<
437
+ TFields,
438
+ TOmitInitKey,
439
+ TExtra,
440
+ TPostDeserialized,
441
+ Uint8ClampedArrayFieldType
442
+ > = (
443
+ name: PropertyKey,
444
+ options: any
445
+ ): any => {
446
+ return this.arrayBufferLike(name, Uint8ClampedArrayFieldType.instance, options);
447
+ };
448
+
449
+ public string: BindedArrayBufferLikeFieldDefinitionCreator<
450
+ TFields,
451
+ TOmitInitKey,
452
+ TExtra,
453
+ TPostDeserialized,
454
+ StringFieldType
455
+ > = (
456
+ name: PropertyKey,
457
+ options: any
458
+ ): any => {
459
+ return this.arrayBufferLike(name, StringFieldType.instance, options);
460
+ };
461
+
462
+ /**
463
+ * Adds some extra properties into every `Struct` value.
464
+ *
465
+ * Extra properties will not affect serialize or deserialize process.
466
+ *
467
+ * Multiple calls to `extra` will merge all properties together.
468
+ *
469
+ * @param value
470
+ * An object containing properties to be added to the result value. Accessors and methods are also allowed.
471
+ */
472
+ public extra<T extends Record<
473
+ // This trick disallows any keys that are already in `TValue`
474
+ Exclude<
475
+ keyof T,
476
+ Exclude<keyof T, keyof TFields>
477
+ >,
478
+ never
479
+ >>(
480
+ value: T & ThisType<Overwrite<Overwrite<TExtra, T>, TFields>>
481
+ ): Struct<
482
+ TFields,
483
+ TOmitInitKey,
484
+ Overwrite<TExtra, T>,
485
+ TPostDeserialized
486
+ > {
487
+ Object.assign(this._extra, Object.getOwnPropertyDescriptors(value));
488
+ return this as any;
489
+ }
490
+
491
+ /**
492
+ * Registers (or replaces) a custom callback to be run after deserialized.
493
+ *
494
+ * A callback returning `never` (always throw an error)
495
+ * will also change the return type of `deserialize` to `never`.
496
+ */
497
+ public postDeserialize(
498
+ callback: StructPostDeserialized<TFields, never>
499
+ ): Struct<TFields, TOmitInitKey, TExtra, never>;
500
+ /**
501
+ * Registers (or replaces) a custom callback to be run after deserialized.
502
+ *
503
+ * A callback returning `void` means it modify the result object in-place
504
+ * (or doesn't modify it at all), so `deserialize` will still return the result object.
505
+ */
506
+ public postDeserialize(
507
+ callback?: StructPostDeserialized<TFields, void>
508
+ ): Struct<TFields, TOmitInitKey, TExtra, undefined>;
509
+ /**
510
+ * Registers (or replaces) a custom callback to be run after deserialized.
511
+ *
512
+ * A callback returning anything other than `undefined`
513
+ * will `deserialize` to return that object instead.
514
+ */
515
+ public postDeserialize<TPostSerialize>(
516
+ callback?: StructPostDeserialized<TFields, TPostSerialize>
517
+ ): Struct<TFields, TOmitInitKey, TExtra, TPostSerialize>;
518
+ public postDeserialize(
519
+ callback?: StructPostDeserialized<TFields, any>
520
+ ) {
521
+ this._postDeserialized = callback;
522
+ return this as any;
523
+ }
524
+
525
+ public deserialize(
526
+ stream: StructDeserializeStream,
527
+ ): StructDeserializedResult<TFields, TExtra, TPostDeserialized>;
528
+ public deserialize(
529
+ stream: StructAsyncDeserializeStream,
530
+ ): Promise<StructDeserializedResult<TFields, TExtra, TPostDeserialized>>;
531
+ public deserialize(
532
+ stream: StructDeserializeStream | StructAsyncDeserializeStream,
533
+ ): ValueOrPromise<StructDeserializedResult<TFields, TExtra, TPostDeserialized>> {
534
+ const value = new StructValue();
535
+ Object.defineProperties(value.value, this._extra);
536
+
537
+ return Syncbird.try(() => {
538
+ const iterator = this._fields[Symbol.iterator]();
539
+ const iterate: () => StructValue | Syncbird<StructValue> = () => {
540
+ const result = iterator.next();
541
+ if (result.done) {
542
+ return value;
543
+ }
544
+
545
+ const [name, definition] = result.value;
546
+ return Syncbird.resolve(
547
+ definition.deserialize(this.options, stream as any, value)
548
+ ).then(fieldValue => {
549
+ value.set(name, fieldValue);
550
+ return iterate();
551
+ });
552
+ };
553
+ return iterate();
554
+ }).then(value => {
555
+ if (this._postDeserialized) {
556
+ const object = value.value as TFields;
557
+ const result = this._postDeserialized.call(object, object);
558
+ if (result) {
559
+ return result;
560
+ }
561
+ }
562
+
563
+ return value.value;
564
+ }).valueOrPromise();
565
+ }
566
+
567
+ public serialize(init: Evaluate<Omit<TFields, TOmitInitKey>>): ArrayBuffer {
568
+ const value = new StructValue();
569
+
570
+ for (const [name, definition] of this._fields) {
571
+ const fieldValue = definition.create(this.options, value, (init as any)[name]);
572
+ value.set(name, fieldValue);
573
+ }
574
+
575
+ let structSize = 0;
576
+ const fieldsInfo: { fieldValue: StructFieldValue, size: number; }[] = [];
577
+
578
+ for (const [name] of this._fields) {
579
+ const fieldValue = value.get(name);
580
+ const size = fieldValue.getSize();
581
+ fieldsInfo.push({ fieldValue, size });
582
+ structSize += size;
583
+ }
584
+
585
+ const buffer = new ArrayBuffer(structSize);
586
+ const dataView = new DataView(buffer);
587
+ let offset = 0;
588
+ for (const { fieldValue, size } of fieldsInfo) {
589
+ fieldValue.serialize(dataView, offset);
590
+ offset += size;
591
+ }
592
+
593
+ return buffer;
594
+ }
595
+ }