@colyseus/schema 3.0.48 → 3.0.49

package/build/cjs/index.js

@@ -3180,20 +3180,29 @@ function defineTypes(target, fields, options) {
     }
     return target;
 }
-function schema(fields, name, inherits = Schema) {
+function schema(fieldsAndMethods, name, inherits = Schema) {
+    const fields = {};
+    const methods = {};
     const defaultValues = {};
     const viewTagFields = {};
-    for (let fieldName in fields) {
-        const field = fields[fieldName];
-        if (typeof (field) === "object") {
-            if (field['default'] !== undefined) {
-                defaultValues[fieldName] = field['default'];
+    for (let fieldName in fieldsAndMethods) {
+        const value = fieldsAndMethods[fieldName];
+        if (typeof (value) === "object") {
+            if (value['default'] !== undefined) {
+                defaultValues[fieldName] = value['default'];
             }
-            if (field['view'] !== undefined) {
-                viewTagFields[fieldName] = (typeof (field['view']) === "boolean")
+            if (value['view'] !== undefined) {
+                viewTagFields[fieldName] = (typeof (value['view']) === "boolean")
                     ? DEFAULT_VIEW_TAG
-                    : field['view'];
+                    : value['view'];
             }
+            fields[fieldName] = value;
+        }
+        else if (typeof (value) === "function") {
+            methods[fieldName] = value;
+        }
+        else {
+            fields[fieldName] = value;
         }
     }
     const klass = Metadata.setFields(class extends inherits {
@@ -3205,6 +3214,9 @@ function schema(fields, name, inherits = Schema) {
     for (let fieldName in viewTagFields) {
         view(viewTagFields[fieldName])(klass.prototype, fieldName);
     }
+    for (let methodName in methods) {
+        klass.prototype[methodName] = methods[methodName];
+    }
     if (name) {
         Object.defineProperty(klass, "name", { value: name });
     }