@colyseus/schema 3.0.0-alpha.40 → 3.0.0-alpha.42

package/src/index.ts

@@ -14,8 +14,8 @@ export { CollectionSchema };
 import { SetSchema } from "./types/custom/SetSchema";
 export { SetSchema };
 
-import { registerType } from "./types/registry";
-export { registerType };
+import { registerType, defineCustomTypes } from "./types/registry";
+export { registerType, defineCustomTypes };
 
 registerType("map", { constructor: MapSchema });
 registerType("array", { constructor: ArraySchema });
@@ -27,10 +27,8 @@ export { dumpChanges } from "./utils";
 
 // Encoder / Decoder
 export { $track, $encoder, $decoder, $filter, $getByIndex, $deleteByIndex, $changes, $childType } from "./types/symbols";
-export type { Iterator } from "./encoding/decode";
-import * as encode from "./encoding/encode";
-import * as decode from "./encoding/decode";
-export { encode, decode };
+export { encode } from "./encoding/encode";
+export { decode, type Iterator } from "./encoding/decode";
 
 // Reflection
 export {

package/src/types/registry.ts

@@ -1,13 +1,24 @@
+import { DefinitionType, type } from "../annotations";
+import { BufferLike, encode } from "../encoding/encode";
+import { decode, Iterator } from "../encoding/decode";
+
 export interface TypeDefinition {
-    constructor: any,
+    constructor?: any,
+    encode?: (bytes: BufferLike, value: any, it: Iterator) => any;
+    decode?: (bytes: BufferLike, it: Iterator) => any;
 }
 
 const registeredTypes: {[identifier: string] : TypeDefinition} = {};
 const identifiers = new Map<any, string>();
 
 export function registerType(identifier: string, definition: TypeDefinition) {
-    identifiers.set(definition.constructor, identifier);
-    registeredTypes[identifier] = definition;
+    if (definition.constructor) {
+        identifiers.set(definition.constructor, identifier);
+        registeredTypes[identifier] = definition;
+    }
+
+    if (definition.encode) { encode[identifier] = definition.encode; }
+    if (definition.decode) { decode[identifier] = definition.decode; }
 }
 
 export function getIdentifier(klass: any): string {
@@ -17,3 +28,11 @@ export function getIdentifier(klass: any): string {
 export function getType(identifier: string): TypeDefinition {
     return registeredTypes[identifier];
 }
+
+export function defineCustomTypes<T extends {[key: string]: TypeDefinition}>(types: T) {
+    for (const identifier in types) {
+        registerType(identifier, types[identifier]);
+    }
+
+    return (t: keyof T) => type(t as DefinitionType);
+}