@yume-chan/struct 0.0.15 → 0.0.17

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