@colyseus/schema 2.0.32 → 3.0.0-alpha.0

package/lib/encoder/ChangeTree.js

@@ -0,0 +1,384 @@
+"use strict";
+var _a;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChangeTree = exports.Root = void 0;
+const spec_1 = require("../encoding/spec");
+const symbols_1 = require("../types/symbols");
+const Metadata_1 = require("../Metadata");
+class Root {
+    constructor() {
+        this.nextUniqueId = 0;
+        this.refCount = new WeakMap();
+        // all changes
+        this.allChanges = new Map();
+        this.allFilteredChanges = new Map();
+        // pending changes to be encoded
+        this.changes = new Map();
+        this.filteredChanges = new Map();
+        this.views = [];
+    }
+    getNextUniqueId() {
+        return this.nextUniqueId++;
+    }
+    add(changeTree) {
+        const refCount = this.refCount.get(changeTree) || 0;
+        this.refCount.set(changeTree, refCount + 1);
+    }
+    remove(changeTree) {
+        const refCount = this.refCount.get(changeTree);
+        if (refCount <= 1) {
+            this.allChanges.delete(changeTree);
+            this.changes.delete(changeTree);
+            if (changeTree.isFiltered || changeTree.isPartiallyFiltered) {
+                this.allFilteredChanges.delete(changeTree);
+                this.filteredChanges.delete(changeTree);
+            }
+            this.refCount.delete(changeTree);
+        }
+        else {
+            this.refCount.set(changeTree, refCount - 1);
+        }
+        changeTree.forEachChild((child, _) => this.remove(child));
+    }
+    clear() {
+        this.changes.clear();
+    }
+}
+exports.Root = Root;
+class ChangeTree {
+    static { _a = symbols_1.$isNew; }
+    ;
+    constructor(ref) {
+        this.indexes = {}; // TODO: remove this, only used by MapSchema/SetSchema/CollectionSchema (`encodeKeyValueOperation`)
+        this.currentOperationIndex = 0;
+        this.allChanges = new Map();
+        this.allFilteredChanges = new Map();
+        this.changes = new Map();
+        this.filteredChanges = new Map();
+        this[_a] = true;
+        this.ref = ref;
+    }
+    setRoot(root) {
+        this.root = root;
+        this.root.add(this);
+        //
+        // At Schema initialization, the "root" structure might not be available
+        // yet, as it only does once the "Encoder" has been set up.
+        //
+        // So the "parent" may be already set without a "root".
+        //
+        this.checkIsFiltered(this.parent, this.parentIndex);
+        // unique refId for the ChangeTree.
+        this.ensureRefId();
+        if (!this.isFiltered) {
+            this.root.changes.set(this, this.changes);
+        }
+        if (this.isFiltered || this.isPartiallyFiltered) {
+            this.root.allFilteredChanges.set(this, this.allFilteredChanges);
+            this.root.filteredChanges.set(this, this.filteredChanges);
+            // } else {
+            //     this.root.allChanges.set(this, this.allChanges);
+        }
+        if (!this.isFiltered) {
+            this.root.allChanges.set(this, this.allChanges);
+        }
+        this.forEachChild((changeTree, _) => {
+            changeTree.setRoot(root);
+        });
+        // this.allChanges.forEach((_, index) => {
+        //     const childRef = this.ref[$getByIndex](index);
+        //     if (childRef && childRef[$changes]) {
+        //         childRef[$changes].setRoot(root);
+        //     }
+        // });
+    }
+    setParent(parent, root, parentIndex) {
+        this.parent = parent;
+        this.parentIndex = parentIndex;
+        // avoid setting parents with empty `root`
+        if (!root) {
+            return;
+        }
+        root.add(this);
+        // skip if parent is already set
+        if (root === this.root) {
+            this.forEachChild((changeTree, atIndex) => {
+                changeTree.setParent(this.ref, root, atIndex);
+            });
+            return;
+        }
+        this.root = root;
+        this.checkIsFiltered(parent, parentIndex);
+        if (!this.isFiltered) {
+            this.root.changes.set(this, this.changes);
+        }
+        if (this.isFiltered || this.isPartiallyFiltered) {
+            this.root.filteredChanges.set(this, this.filteredChanges);
+            this.root.allFilteredChanges.set(this, this.filteredChanges);
+        }
+        else {
+            this.root.allChanges.set(this, this.allChanges);
+        }
+        this.ensureRefId();
+        this.forEachChild((changeTree, atIndex) => {
+            changeTree.setParent(this.ref, root, atIndex);
+        });
+    }
+    forEachChild(callback) {
+        //
+        // assign same parent on child structures
+        //
+        if (Metadata_1.Metadata.isValidInstance(this.ref)) {
+            const metadata = this.ref['constructor'][Symbol.metadata];
+            // FIXME: need to iterate over parent metadata instead.
+            for (const field in metadata) {
+                const value = this.ref[field];
+                if (value && value[symbols_1.$changes]) {
+                    callback(value[symbols_1.$changes], metadata[field].index);
+                }
+            }
+        }
+        else if (typeof (this.ref) === "object") {
+            // MapSchema / ArraySchema, etc.
+            this.ref.forEach((value, key) => {
+                if (Metadata_1.Metadata.isValidInstance(value)) {
+                    callback(value[symbols_1.$changes], this.ref[symbols_1.$changes].indexes[key]);
+                }
+            });
+        }
+    }
+    operation(op) {
+        this.changes.set(--this.currentOperationIndex, op);
+        this.root?.changes.set(this, this.changes);
+    }
+    change(index, operation = spec_1.OPERATION.ADD) {
+        const metadata = this.ref['constructor'][Symbol.metadata];
+        const isFiltered = this.isFiltered || (metadata && metadata[metadata[index]].tag !== undefined);
+        const changeSet = (isFiltered)
+            ? this.filteredChanges
+            : this.changes;
+        const previousOperation = changeSet.get(index);
+        if (!previousOperation || previousOperation === spec_1.OPERATION.DELETE) {
+            const op = (!previousOperation)
+                ? operation
+                : (previousOperation === spec_1.OPERATION.DELETE)
+                    ? spec_1.OPERATION.DELETE_AND_ADD
+                    : operation;
+            changeSet.set(index, op);
+        }
+        //
+        // TODO: are DELETE operations being encoded as ADD here ??
+        //
+        if (isFiltered) {
+            this.allFilteredChanges.set(index, spec_1.OPERATION.ADD);
+            this.root?.filteredChanges.set(this, this.filteredChanges);
+        }
+        else {
+            this.allChanges.set(index, spec_1.OPERATION.ADD);
+            this.root?.changes.set(this, this.changes);
+        }
+    }
+    shiftChangeIndexes(shiftIndex) {
+        //
+        // Used only during:
+        //
+        // - ArraySchema#unshift()
+        //
+        const changeSet = (this.isFiltered)
+            ? this.filteredChanges
+            : this.changes;
+        const changeSetEntries = Array.from(changeSet.entries());
+        changeSet.clear();
+        // Re-insert each entry with the shifted index
+        for (const [index, op] of changeSetEntries) {
+            changeSet.set(index + shiftIndex, op);
+        }
+    }
+    shiftAllChangeIndexes(shiftIndex, startIndex = 0) {
+        //
+        // Used only during:
+        //
+        // - ArraySchema#splice()
+        //
+        if (this.isFiltered || this.isPartiallyFiltered) {
+            this._shiftAllChangeIndexes(shiftIndex, startIndex, this.allFilteredChanges);
+            this._shiftAllChangeIndexes(shiftIndex, startIndex, this.allChanges);
+        }
+        else {
+            this._shiftAllChangeIndexes(shiftIndex, startIndex, this.allChanges);
+        }
+    }
+    _shiftAllChangeIndexes(shiftIndex, startIndex = 0, allChangeSet) {
+        Array.from(allChangeSet.entries()).forEach(([index, op]) => {
+            // console.log('shiftAllChangeIndexes', index >= startIndex, { index, op, shiftIndex, startIndex })
+            if (index >= startIndex) {
+                allChangeSet.delete(index);
+                allChangeSet.set(index + shiftIndex, op);
+            }
+        });
+    }
+    indexedOperation(index, operation, allChangesIndex = index) {
+        const metadata = this.ref['constructor'][Symbol.metadata];
+        const isFiltered = this.isFiltered || (metadata && metadata[metadata[index]].tag !== undefined);
+        if (isFiltered) {
+            this.allFilteredChanges.set(allChangesIndex, spec_1.OPERATION.ADD);
+            this.filteredChanges.set(index, operation);
+            this.root?.filteredChanges.set(this, this.filteredChanges);
+        }
+        else {
+            this.allChanges.set(allChangesIndex, spec_1.OPERATION.ADD);
+            this.changes.set(index, operation);
+            this.root?.changes.set(this, this.changes);
+        }
+    }
+    getType(index) {
+        if (Metadata_1.Metadata.isValidInstance(this.ref)) {
+            const metadata = this.ref['constructor'][Symbol.metadata];
+            return metadata[metadata[index]].type;
+        }
+        else {
+            //
+            // Get the child type from parent structure.
+            // - ["string"] => "string"
+            // - { map: "string" } => "string"
+            // - { set: "string" } => "string"
+            //
+            return this.ref[symbols_1.$childType];
+        }
+    }
+    getChange(index) {
+        // TODO: optimize this. avoid checking against multiple instances
+        return this.changes.get(index) ?? this.filteredChanges.get(index);
+    }
+    //
+    // used during `.encode()`
+    //
+    getValue(index, isEncodeAll = false) {
+        //
+        // `isEncodeAll` param is only used by ArraySchema
+        //
+        return this.ref[symbols_1.$getByIndex](index, isEncodeAll);
+    }
+    delete(index, operation, allChangesIndex = index) {
+        if (index === undefined) {
+            try {
+                throw new Error(`@colyseus/schema ${this.ref.constructor.name}: trying to delete non-existing index '${index}'`);
+            }
+            catch (e) {
+                console.warn(e);
+            }
+            return;
+        }
+        const metadata = this.ref['constructor'][Symbol.metadata];
+        const isFiltered = this.isFiltered || (metadata && metadata[metadata[index]].tag !== undefined);
+        const changeSet = (isFiltered)
+            ? this.filteredChanges
+            : this.changes;
+        const previousValue = this.getValue(index);
+        changeSet.set(index, operation ?? spec_1.OPERATION.DELETE);
+        // remove `root` reference
+        if (previousValue && previousValue[symbols_1.$changes]) {
+            previousValue[symbols_1.$changes].root = undefined;
+            //
+            // FIXME: this.root is "undefined"
+            //
+            // This method is being called at decoding time when a DELETE operation is found.
+            //
+            // - This is due to using the concrete Schema class at decoding time.
+            // - "Reflected" structures do not have this problem.
+            //
+            // (the property descriptors should NOT be used at decoding time. only at encoding time.)
+            //
+            this.root?.remove(previousValue[symbols_1.$changes]);
+        }
+        //
+        // FIXME: this is looking a bit ugly (and repeated from `.change()`)
+        //
+        if (isFiltered) {
+            this.root?.filteredChanges.set(this, this.filteredChanges);
+            this.allFilteredChanges.delete(allChangesIndex);
+        }
+        else {
+            this.root?.changes.set(this, this.changes);
+            this.allChanges.delete(allChangesIndex);
+        }
+    }
+    endEncode() {
+        this.changes.clear();
+        this.ref[symbols_1.$onEncodeEnd]?.();
+        // Not a new instance anymore
+        delete this[symbols_1.$isNew];
+    }
+    discard(discardAll = false) {
+        //
+        // > MapSchema:
+        //      Remove cached key to ensure ADD operations is unsed instead of
+        //      REPLACE in case same key is used on next patches.
+        //
+        this.ref[symbols_1.$onEncodeEnd]?.();
+        this.changes.clear();
+        this.filteredChanges.clear();
+        // reset operation index
+        this.currentOperationIndex = 0;
+        if (discardAll) {
+            this.allChanges.clear();
+            this.allFilteredChanges.clear();
+            // remove children references
+            this.forEachChild((changeTree, _) => this.root?.remove(changeTree));
+        }
+    }
+    /**
+     * Recursively discard all changes from this, and child structures.
+     */
+    discardAll() {
+        this.changes.forEach((_, fieldIndex) => {
+            const value = this.getValue(fieldIndex);
+            if (value && value[symbols_1.$changes]) {
+                value[symbols_1.$changes].discardAll();
+            }
+        });
+        this.discard();
+    }
+    ensureRefId() {
+        // skip if refId is already set.
+        if (this.refId !== undefined) {
+            return;
+        }
+        this.refId = this.root.getNextUniqueId();
+    }
+    get changed() {
+        return this.changes.size > 0;
+    }
+    checkIsFiltered(parent, parentIndex) {
+        // Detect if current structure has "filters" declared
+        this.isPartiallyFiltered = this.ref['constructor']?.[Symbol.metadata]?.[-2];
+        // TODO: support "partially filtered", where the instance is visible, but only a field is not.
+        // Detect if parent has "filters" declared
+        while (parent && !this.isFiltered) {
+            const metadata = parent['constructor'][Symbol.metadata];
+            const fieldName = metadata?.[parentIndex];
+            const isParentOwned = metadata?.[fieldName]?.tag !== undefined;
+            this.isFiltered = isParentOwned || parent[symbols_1.$changes].isFiltered; // metadata?.[-2]
+            parent = parent[symbols_1.$changes].parent;
+        }
+        ;
+        //
+        // TODO: refactor this!
+        //
+        //      swapping `changes` and `filteredChanges` is required here
+        //      because "isFiltered" may not be imedialely available on `change()`
+        //
+        if (this.isFiltered && this.changes.size > 0) {
+            // swap changes reference
+            const changes = this.changes;
+            this.changes = this.filteredChanges;
+            this.filteredChanges = changes;
+            // swap "all changes" reference
+            const allFilteredChanges = this.allFilteredChanges;
+            this.allFilteredChanges = this.allChanges;
+            this.allChanges = allFilteredChanges;
+        }
+    }
+}
+exports.ChangeTree = ChangeTree;
+//# sourceMappingURL=ChangeTree.js.map

package/lib/encoder/ChangeTree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChangeTree.js","sourceRoot":"","sources":["../../src/encoder/ChangeTree.ts"],"names":[],"mappings":";;;;AAAA,2CAA6C;AAE7C,8CAA+G;AAO/G,0CAAuC;AAoBvC,MAAa,IAAI;IAAjB;QACc,iBAAY,GAAW,CAAC,CAAC;QACnC,aAAQ,GAAG,IAAI,OAAO,EAAsB,CAAC;QAE7C,cAAc;QACd,eAAU,GAAG,IAAI,GAAG,EAAsC,CAAC;QAC3D,uBAAkB,GAAG,IAAI,GAAG,EAAsC,CAAC;QAEnE,gCAAgC;QAChC,YAAO,GAAG,IAAI,GAAG,EAAsC,CAAC;QACxD,oBAAe,GAAG,IAAI,GAAG,EAAsC,CAAC;QAEhE,UAAK,GAAgB,EAAE,CAAC;IAmC5B,CAAC;IAjCG,eAAe;QACX,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC;IAC/B,CAAC;IAED,GAAG,CAAE,UAAsB;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,UAAsB;QACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YAChB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACnC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAEhC,IAAI,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,mBAAmB,EAAE,CAAC;gBAC1D,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAC3C,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC5C,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAErC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,UAAU,CAAC,YAAY,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CACjC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACJ;AA/CD,oBA+CC;AAED,MAAa,UAAU;kBAqBlB,gBAAM;IAFwC,CAAC;IAIhD,YAAY,GAAM;QAXlB,YAAO,GAA2B,EAAE,CAAC,CAAC,mGAAmG;QACzI,0BAAqB,GAAW,CAAC,CAAC;QAElC,eAAU,GAAG,IAAI,GAAG,EAAqB,CAAC;QAC1C,uBAAkB,GAAG,IAAI,GAAG,EAAqB,CAAC;QAElD,YAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;QACvC,oBAAe,GAAG,IAAI,GAAG,EAAqB,CAAC;QAE/C,QAAQ,GAAG,IAAI,CAAC;QAGZ,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACnB,CAAC;IAED,OAAO,CAAC,IAAU;QACd,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEpB,EAAE;QACF,wEAAwE;QACxE,2DAA2D;QAC3D,EAAE;QACF,uDAAuD;QACvD,EAAE;QACF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAEpD,mCAAmC;QACnC,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9C,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;YAE9D,WAAW;YACX,uDAAuD;QACvD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE;YAChC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,0CAA0C;QAC1C,qDAAqD;QACrD,4CAA4C;QAC5C,4CAA4C;QAC5C,QAAQ;QACR,MAAM;IACV,CAAC;IAED,SAAS,CACL,MAAW,EACX,IAAW,EACX,WAAoB;QAEpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAE/B,0CAA0C;QAC1C,IAAI,CAAC,IAAI,EAAE,CAAC;YAAC,OAAO;QAAC,CAAC;QAEtB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEf,gCAAgC;QAChC,IAAI,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;YACrB,IAAI,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE;gBACtC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;YACH,OAAO;QACX,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAE1C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9C,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;YAC1D,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QACjE,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,IAAI,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE;YACtC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACP,CAAC;IAED,YAAY,CAAC,QAAuD;QAChE,EAAE;QACF,yCAAyC;QACzC,EAAE;QACF,IAAI,mBAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAa,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAEpE,uDAAuD;YACvD,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAE9B,IAAI,KAAK,IAAI,KAAK,CAAC,kBAAQ,CAAC,EAAE,CAAC;oBAC3B,QAAQ,CAAC,KAAK,CAAC,kBAAQ,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;gBACrD,CAAC;YACL,CAAC;QAEL,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,QAAQ,EAAE,CAAC;YACxC,gCAAgC;YAC/B,IAAI,CAAC,GAAiB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;gBAC3C,IAAI,mBAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;oBAClC,QAAQ,CAAC,KAAK,CAAC,kBAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,kBAAQ,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC/D,CAAC;YACL,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED,SAAS,CAAC,EAAa;QACnB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,YAAuB,gBAAS,CAAC,GAAG;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAa,CAAC;QAEtE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC;QAChG,MAAM,SAAS,GAAG,CAAC,UAAU,CAAC;YAC1B,CAAC,CAAC,IAAI,CAAC,eAAe;YACtB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;QAEnB,MAAM,iBAAiB,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,iBAAiB,IAAI,iBAAiB,KAAK,gBAAS,CAAC,MAAM,EAAE,CAAC;YAC/D,MAAM,EAAE,GAAG,CAAC,CAAC,iBAAiB,CAAC;gBAC3B,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,CAAC,iBAAiB,KAAK,gBAAS,CAAC,MAAM,CAAC;oBACtC,CAAC,CAAC,gBAAS,CAAC,cAAc;oBAC1B,CAAC,CAAC,SAAS,CAAA;YACnB,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7B,CAAC;QAED,EAAE;QACF,2DAA2D;QAC3D,EAAE;QAEF,IAAI,UAAU,EAAE,CAAC;YACb,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,gBAAS,CAAC,GAAG,CAAC,CAAC;YAClD,IAAI,CAAC,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAE/D,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,gBAAS,CAAC,GAAG,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,UAAkB;QACjC,EAAE;QACF,oBAAoB;QACpB,EAAE;QACF,0BAA0B;QAC1B,EAAE;QACF,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC;YAC/B,CAAC,CAAC,IAAI,CAAC,eAAe;YACtB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;QAEnB,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;QACzD,SAAS,CAAC,KAAK,EAAE,CAAC;QAElB,8CAA8C;QAC9C,KAAK,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,gBAAgB,EAAE,CAAC;YACzC,SAAS,CAAC,GAAG,CAAC,KAAK,GAAG,UAAU,EAAE,EAAE,CAAC,CAAC;QAC1C,CAAC;IACL,CAAC;IAED,qBAAqB,CAAC,UAAkB,EAAE,aAAqB,CAAC;QAC5D,EAAE;QACF,oBAAoB;QACpB,EAAE;QACF,yBAAyB;QACzB,EAAE;QACF,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9C,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC7E,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAEzE,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACzE,CAAC;IACL,CAAC;IAEO,sBAAsB,CAAC,UAAkB,EAAE,aAAqB,CAAC,EAAE,YAAoC;QAC3G,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACvD,mGAAmG;YACnG,IAAI,KAAK,IAAI,UAAU,EAAE,CAAC;gBACtB,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC3B,YAAY,CAAC,GAAG,CAAC,KAAK,GAAG,UAAU,EAAE,EAAE,CAAC,CAAC;YAC7C,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,SAAoB,EAAE,eAAe,GAAG,KAAK;QACzE,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAa,CAAC;QACtE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC;QAEhG,IAAI,UAAU,EAAE,CAAC;YACb,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,eAAe,EAAE,gBAAS,CAAC,GAAG,CAAC,CAAC;YAC5D,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YAC3C,IAAI,CAAC,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAE/D,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,EAAE,gBAAS,CAAC,GAAG,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YACnC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC;IAED,OAAO,CAAC,KAAc;QAClB,IAAI,mBAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAa,CAAC;YACtE,OAAO,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;QAE1C,CAAC;aAAM,CAAC;YACJ,EAAE;YACF,4CAA4C;YAC5C,2BAA2B;YAC3B,kCAAkC;YAClC,kCAAkC;YAClC,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,oBAAU,CAAC,CAAC;QAChC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,KAAa;QACnB,iEAAiE;QACjE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACtE,CAAC;IAED,EAAE;IACF,0BAA0B;IAC1B,EAAE;IACF,QAAQ,CAAC,KAAa,EAAE,cAAuB,KAAK;QAChD,EAAE;QACF,kDAAkD;QAClD,EAAE;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,qBAAW,CAAC,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,SAAqB,EAAE,eAAe,GAAG,KAAK;QAChE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACtB,IAAI,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,0CAA0C,KAAK,GAAG,CAAC,CAAC;YACrH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpB,CAAC;YACD,OAAO;QACX,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAa,CAAC;QACtE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC;QAChG,MAAM,SAAS,GAAG,CAAC,UAAU,CAAC;YAC1B,CAAC,CAAC,IAAI,CAAC,eAAe;YACtB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;QAEnB,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAE3C,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,IAAI,gBAAS,CAAC,MAAM,CAAC,CAAC;QAEpD,0BAA0B;QAC1B,IAAI,aAAa,IAAI,aAAa,CAAC,kBAAQ,CAAC,EAAE,CAAC;YAC3C,aAAa,CAAC,kBAAQ,CAAC,CAAC,IAAI,GAAG,SAAS,CAAC;YAEzC,EAAE;YACF,kCAAkC;YAClC,EAAE;YACF,iFAAiF;YACjF,EAAE;YACF,qEAAqE;YACrE,qDAAqD;YACrD,EAAE;YACF,yFAAyF;YACzF,EAAE;YACF,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC,kBAAQ,CAAC,CAAC,CAAC;QAC/C,CAAC;QAED,EAAE;QACF,oEAAoE;QACpE,EAAE;QACF,IAAI,UAAU,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;YAC3D,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAEpD,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAC5C,CAAC;IACL,CAAC;IAED,SAAS;QACL,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,GAAG,CAAC,sBAAY,CAAC,EAAE,EAAE,CAAC;QAE3B,6BAA6B;QAC7B,OAAO,IAAI,CAAC,gBAAM,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,CAAC,aAAsB,KAAK;QAC/B,EAAE;QACF,eAAe;QACf,sEAAsE;QACtE,yDAAyD;QACzD,EAAE;QACF,IAAI,CAAC,GAAG,CAAC,sBAAY,CAAC,EAAE,EAAE,CAAC;QAE3B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,wBAAwB;QACxB,IAAI,CAAC,qBAAqB,GAAG,CAAC,CAAC;QAE/B,IAAI,UAAU,EAAE,CAAC;YACb,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;YAEhC,6BAA6B;YAC7B,IAAI,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAChC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;QACvC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,UAAU;QACN,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE;YACnC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YAExC,IAAI,KAAK,IAAI,KAAK,CAAC,kBAAQ,CAAC,EAAE,CAAC;gBAC3B,KAAK,CAAC,kBAAQ,CAAC,CAAC,UAAU,EAAE,CAAC;YACjC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;IAED,WAAW;QACP,gCAAgC;QAChC,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC3B,OAAO;QACX,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,OAAO;QACP,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;IACjC,CAAC;IAES,eAAe,CAAC,MAAW,EAAE,WAAmB;QACtD,qDAAqD;QACrD,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5E,8FAA8F;QAE9F,0CAA0C;QAC1C,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAa,MAAM,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAElE,MAAM,SAAS,GAAG,QAAQ,EAAE,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,aAAa,GAAG,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,GAAG,KAAK,SAAS,CAAC;YAE/D,IAAI,CAAC,UAAU,GAAG,aAAa,IAAI,MAAM,CAAC,kBAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,iBAAiB;YAEjF,MAAM,GAAG,MAAM,CAAC,kBAAQ,CAAC,CAAC,MAAM,CAAC;QACrC,CAAC;QAAA,CAAC;QAEF,EAAE;QACF,uBAAuB;QACvB,EAAE;QACF,iEAAiE;QACjE,0EAA0E;QAC1E,EAAE;QACF,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3C,yBAAyB;YACzB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAC7B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC;YAE/B,+BAA+B;YAC/B,MAAM,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;YACnD,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,UAAU,CAAC;YAC1C,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC;QACzC,CAAC;IACL,CAAC;CAEJ;AA7ZD,gCA6ZC","sourcesContent":["import { OPERATION } from \"../encoding/spec\";\nimport { Schema } from \"../Schema\";\nimport { $changes, $childType, $decoder, $onEncodeEnd, $encoder, $getByIndex, $isNew } from \"../types/symbols\";\n\nimport type { MapSchema } from \"../types/custom/MapSchema\";\nimport type { ArraySchema } from \"../types/custom/ArraySchema\";\nimport type { CollectionSchema } from \"../types/custom/CollectionSchema\";\nimport type { SetSchema } from \"../types/custom/SetSchema\";\n\nimport { Metadata } from \"../Metadata\";\nimport type { EncodeOperation } from \"./EncodeOperation\";\nimport type { DecodeOperation } from \"../decoder/DecodeOperation\";\nimport type { StateView } from \"./StateView\";\n\ndeclare global {\n    interface Object {\n        // FIXME: not a good practice to extend globals here\n        [$changes]?: ChangeTree;\n        [$encoder]?: EncodeOperation,\n        [$decoder]?: DecodeOperation,\n    }\n}\n\nexport type Ref = Schema\n    | ArraySchema\n    | MapSchema\n    | CollectionSchema\n    | SetSchema;\n\nexport class Root {\n    protected nextUniqueId: number = 0;\n    refCount = new WeakMap<ChangeTree, number>();\n\n    // all changes\n    allChanges = new Map<ChangeTree, Map<number, OPERATION>>();\n    allFilteredChanges = new Map<ChangeTree, Map<number, OPERATION>>();\n\n    // pending changes to be encoded\n    changes = new Map<ChangeTree, Map<number, OPERATION>>();\n    filteredChanges = new Map<ChangeTree, Map<number, OPERATION>>();\n\n    views: StateView[] = [];\n\n    getNextUniqueId() {\n        return this.nextUniqueId++;\n    }\n\n    add (changeTree: ChangeTree) {\n        const refCount = this.refCount.get(changeTree) || 0;\n        this.refCount.set(changeTree, refCount + 1);\n    }\n\n    remove(changeTree: ChangeTree) {\n        const refCount = this.refCount.get(changeTree);\n        if (refCount <= 1) {\n            this.allChanges.delete(changeTree);\n            this.changes.delete(changeTree);\n\n            if (changeTree.isFiltered || changeTree.isPartiallyFiltered) {\n                this.allFilteredChanges.delete(changeTree);\n                this.filteredChanges.delete(changeTree);\n            }\n\n            this.refCount.delete(changeTree);\n\n        } else {\n            this.refCount.set(changeTree, refCount - 1);\n        }\n\n        changeTree.forEachChild((child, _) =>\n            this.remove(child));\n    }\n\n    clear() {\n        this.changes.clear();\n    }\n}\n\nexport class ChangeTree<T extends Ref=any> {\n    ref: T;\n    refId: number;\n\n    root?: Root;\n\n    isFiltered?: boolean;\n    isPartiallyFiltered?: boolean;\n\n    parent?: Ref;\n    parentIndex?: number;\n\n    indexes: {[index: string]: any} = {}; // TODO: remove this, only used by MapSchema/SetSchema/CollectionSchema (`encodeKeyValueOperation`)\n    currentOperationIndex: number = 0;\n\n    allChanges = new Map<number, OPERATION>();\n    allFilteredChanges = new Map<number, OPERATION>();\n\n    changes = new Map<number, OPERATION>();\n    filteredChanges = new Map<number, OPERATION>();;\n\n    [$isNew] = true;\n\n    constructor(ref: T) {\n        this.ref = ref;\n    }\n\n    setRoot(root: Root) {\n        this.root = root;\n        this.root.add(this);\n\n        //\n        // At Schema initialization, the \"root\" structure might not be available\n        // yet, as it only does once the \"Encoder\" has been set up.\n        //\n        // So the \"parent\" may be already set without a \"root\".\n        //\n        this.checkIsFiltered(this.parent, this.parentIndex);\n\n        // unique refId for the ChangeTree.\n        this.ensureRefId();\n\n        if (!this.isFiltered) {\n            this.root.changes.set(this, this.changes);\n        }\n\n        if (this.isFiltered || this.isPartiallyFiltered) {\n            this.root.allFilteredChanges.set(this, this.allFilteredChanges);\n            this.root.filteredChanges.set(this, this.filteredChanges);\n\n        // } else {\n        //     this.root.allChanges.set(this, this.allChanges);\n        }\n\n        if (!this.isFiltered) {\n            this.root.allChanges.set(this, this.allChanges);\n        }\n\n        this.forEachChild((changeTree, _) => {\n            changeTree.setRoot(root);\n        });\n\n        // this.allChanges.forEach((_, index) => {\n        //     const childRef = this.ref[$getByIndex](index);\n        //     if (childRef && childRef[$changes]) {\n        //         childRef[$changes].setRoot(root);\n        //     }\n        // });\n    }\n\n    setParent(\n        parent: Ref,\n        root?: Root,\n        parentIndex?: number,\n    ) {\n        this.parent = parent;\n        this.parentIndex = parentIndex;\n\n        // avoid setting parents with empty `root`\n        if (!root) { return; }\n\n        root.add(this);\n\n        // skip if parent is already set\n        if (root === this.root) {\n            this.forEachChild((changeTree, atIndex) => {\n                changeTree.setParent(this.ref, root, atIndex);\n            });\n            return;\n        }\n\n        this.root = root;\n        this.checkIsFiltered(parent, parentIndex);\n\n        if (!this.isFiltered) {\n            this.root.changes.set(this, this.changes);\n        }\n\n        if (this.isFiltered || this.isPartiallyFiltered) {\n            this.root.filteredChanges.set(this, this.filteredChanges);\n            this.root.allFilteredChanges.set(this, this.filteredChanges);\n        } else {\n            this.root.allChanges.set(this, this.allChanges);\n        }\n\n        this.ensureRefId();\n\n        this.forEachChild((changeTree, atIndex) => {\n            changeTree.setParent(this.ref, root, atIndex);\n        });\n    }\n\n    forEachChild(callback: (change: ChangeTree, atIndex: number) => void) {\n        //\n        // assign same parent on child structures\n        //\n        if (Metadata.isValidInstance(this.ref)) {\n            const metadata: Metadata = this.ref['constructor'][Symbol.metadata];\n\n            // FIXME: need to iterate over parent metadata instead.\n            for (const field in metadata) {\n                const value = this.ref[field];\n\n                if (value && value[$changes]) {\n                    callback(value[$changes], metadata[field].index);\n                }\n            }\n\n        } else if (typeof (this.ref) === \"object\") {\n            // MapSchema / ArraySchema, etc.\n            (this.ref as MapSchema).forEach((value, key) => {\n                if (Metadata.isValidInstance(value)) {\n                    callback(value[$changes], this.ref[$changes].indexes[key]);\n                }\n            });\n        }\n    }\n\n    operation(op: OPERATION) {\n        this.changes.set(--this.currentOperationIndex, op);\n        this.root?.changes.set(this, this.changes);\n    }\n\n    change(index: number, operation: OPERATION = OPERATION.ADD) {\n        const metadata = this.ref['constructor'][Symbol.metadata] as Metadata;\n\n        const isFiltered = this.isFiltered || (metadata && metadata[metadata[index]].tag !== undefined);\n        const changeSet = (isFiltered)\n            ? this.filteredChanges\n            : this.changes;\n\n        const previousOperation = changeSet.get(index);\n        if (!previousOperation || previousOperation === OPERATION.DELETE) {\n            const op = (!previousOperation)\n                ? operation\n                : (previousOperation === OPERATION.DELETE)\n                    ? OPERATION.DELETE_AND_ADD\n                    : operation\n            changeSet.set(index, op);\n        }\n\n        //\n        // TODO: are DELETE operations being encoded as ADD here ??\n        //\n\n        if (isFiltered) {\n            this.allFilteredChanges.set(index, OPERATION.ADD);\n            this.root?.filteredChanges.set(this, this.filteredChanges);\n\n        } else {\n            this.allChanges.set(index, OPERATION.ADD);\n            this.root?.changes.set(this, this.changes);\n        }\n    }\n\n    shiftChangeIndexes(shiftIndex: number) {\n        //\n        // Used only during:\n        //\n        // - ArraySchema#unshift()\n        //\n        const changeSet = (this.isFiltered)\n            ? this.filteredChanges\n            : this.changes;\n\n        const changeSetEntries = Array.from(changeSet.entries());\n        changeSet.clear();\n\n        // Re-insert each entry with the shifted index\n        for (const [index, op] of changeSetEntries) {\n            changeSet.set(index + shiftIndex, op);\n        }\n    }\n\n    shiftAllChangeIndexes(shiftIndex: number, startIndex: number = 0) {\n        //\n        // Used only during:\n        //\n        // - ArraySchema#splice()\n        //\n        if (this.isFiltered || this.isPartiallyFiltered) {\n            this._shiftAllChangeIndexes(shiftIndex, startIndex, this.allFilteredChanges);\n            this._shiftAllChangeIndexes(shiftIndex, startIndex, this.allChanges);\n\n        } else {\n            this._shiftAllChangeIndexes(shiftIndex, startIndex, this.allChanges);\n        }\n    }\n\n    private _shiftAllChangeIndexes(shiftIndex: number, startIndex: number = 0, allChangeSet: Map<number, OPERATION>) {\n        Array.from(allChangeSet.entries()).forEach(([index, op]) => {\n            // console.log('shiftAllChangeIndexes', index >= startIndex, { index, op, shiftIndex, startIndex })\n            if (index >= startIndex) {\n                allChangeSet.delete(index);\n                allChangeSet.set(index + shiftIndex, op);\n            }\n        });\n    }\n\n    indexedOperation(index: number, operation: OPERATION, allChangesIndex = index) {\n        const metadata = this.ref['constructor'][Symbol.metadata] as Metadata;\n        const isFiltered = this.isFiltered || (metadata && metadata[metadata[index]].tag !== undefined);\n\n        if (isFiltered) {\n            this.allFilteredChanges.set(allChangesIndex, OPERATION.ADD);\n            this.filteredChanges.set(index, operation);\n            this.root?.filteredChanges.set(this, this.filteredChanges);\n\n        } else {\n            this.allChanges.set(allChangesIndex, OPERATION.ADD);\n            this.changes.set(index, operation);\n            this.root?.changes.set(this, this.changes);\n        }\n    }\n\n    getType(index?: number) {\n        if (Metadata.isValidInstance(this.ref)) {\n            const metadata = this.ref['constructor'][Symbol.metadata] as Metadata;\n            return metadata[metadata[index]].type;\n\n        } else {\n            //\n            // Get the child type from parent structure.\n            // - [\"string\"] => \"string\"\n            // - { map: \"string\" } => \"string\"\n            // - { set: \"string\" } => \"string\"\n            //\n            return this.ref[$childType];\n        }\n    }\n\n    getChange(index: number) {\n        // TODO: optimize this. avoid checking against multiple instances\n        return this.changes.get(index) ?? this.filteredChanges.get(index);\n    }\n\n    //\n    // used during `.encode()`\n    //\n    getValue(index: number, isEncodeAll: boolean = false) {\n        //\n        // `isEncodeAll` param is only used by ArraySchema\n        //\n        return this.ref[$getByIndex](index, isEncodeAll);\n    }\n\n    delete(index: number, operation?: OPERATION, allChangesIndex = index) {\n        if (index === undefined) {\n            try {\n                throw new Error(`@colyseus/schema ${this.ref.constructor.name}: trying to delete non-existing index '${index}'`);\n            } catch (e) {\n                console.warn(e);\n            }\n            return;\n        }\n\n        const metadata = this.ref['constructor'][Symbol.metadata] as Metadata;\n        const isFiltered = this.isFiltered || (metadata && metadata[metadata[index]].tag !== undefined);\n        const changeSet = (isFiltered)\n            ? this.filteredChanges\n            : this.changes;\n\n        const previousValue = this.getValue(index);\n\n        changeSet.set(index, operation ?? OPERATION.DELETE);\n\n        // remove `root` reference\n        if (previousValue && previousValue[$changes]) {\n            previousValue[$changes].root = undefined;\n\n            //\n            // FIXME: this.root is \"undefined\"\n            //\n            // This method is being called at decoding time when a DELETE operation is found.\n            //\n            // - This is due to using the concrete Schema class at decoding time.\n            // - \"Reflected\" structures do not have this problem.\n            //\n            // (the property descriptors should NOT be used at decoding time. only at encoding time.)\n            //\n            this.root?.remove(previousValue[$changes]);\n        }\n\n        //\n        // FIXME: this is looking a bit ugly (and repeated from `.change()`)\n        //\n        if (isFiltered) {\n            this.root?.filteredChanges.set(this, this.filteredChanges);\n            this.allFilteredChanges.delete(allChangesIndex);\n\n        } else {\n            this.root?.changes.set(this, this.changes);\n            this.allChanges.delete(allChangesIndex);\n        }\n    }\n\n    endEncode() {\n        this.changes.clear();\n        this.ref[$onEncodeEnd]?.();\n\n        // Not a new instance anymore\n        delete this[$isNew];\n    }\n\n    discard(discardAll: boolean = false) {\n        //\n        // > MapSchema:\n        //      Remove cached key to ensure ADD operations is unsed instead of\n        //      REPLACE in case same key is used on next patches.\n        //\n        this.ref[$onEncodeEnd]?.();\n\n        this.changes.clear();\n        this.filteredChanges.clear();\n\n        // reset operation index\n        this.currentOperationIndex = 0;\n\n        if (discardAll) {\n            this.allChanges.clear();\n            this.allFilteredChanges.clear();\n\n            // remove children references\n            this.forEachChild((changeTree, _) =>\n                this.root?.remove(changeTree));\n        }\n    }\n\n    /**\n     * Recursively discard all changes from this, and child structures.\n     */\n    discardAll() {\n        this.changes.forEach((_, fieldIndex) => {\n            const value = this.getValue(fieldIndex);\n\n            if (value && value[$changes]) {\n                value[$changes].discardAll();\n            }\n        });\n\n        this.discard();\n    }\n\n    ensureRefId() {\n        // skip if refId is already set.\n        if (this.refId !== undefined) {\n            return;\n        }\n\n        this.refId = this.root.getNextUniqueId();\n    }\n\n    get changed() {\n        return this.changes.size > 0;\n    }\n\n    protected checkIsFiltered(parent: Ref, parentIndex: number) {\n        // Detect if current structure has \"filters\" declared\n        this.isPartiallyFiltered = this.ref['constructor']?.[Symbol.metadata]?.[-2];\n\n        // TODO: support \"partially filtered\", where the instance is visible, but only a field is not.\n\n        // Detect if parent has \"filters\" declared\n        while (parent && !this.isFiltered) {\n            const metadata: Metadata = parent['constructor'][Symbol.metadata];\n\n            const fieldName = metadata?.[parentIndex];\n            const isParentOwned = metadata?.[fieldName]?.tag !== undefined;\n\n            this.isFiltered = isParentOwned || parent[$changes].isFiltered; // metadata?.[-2]\n\n            parent = parent[$changes].parent;\n        };\n\n        //\n        // TODO: refactor this!\n        //\n        //      swapping `changes` and `filteredChanges` is required here\n        //      because \"isFiltered\" may not be imedialely available on `change()`\n        //\n        if (this.isFiltered && this.changes.size > 0) {\n            // swap changes reference\n            const changes = this.changes;\n            this.changes = this.filteredChanges;\n            this.filteredChanges = changes;\n\n            // swap \"all changes\" reference\n            const allFilteredChanges = this.allFilteredChanges;\n            this.allFilteredChanges = this.allChanges;\n            this.allChanges = allFilteredChanges;\n        }\n    }\n\n}\n"]}

package/lib/encoder/EncodeOperation.d.ts

@@ -0,0 +1,25 @@
+/// <reference types="node" />
+import { OPERATION } from "../encoding/spec";
+import type { ChangeTree, Ref } from "./ChangeTree";
+import type { Encoder } from "./Encoder";
+import type { Schema } from "../Schema";
+import type { PrimitiveType } from "../annotations";
+import type { Iterator } from "../encoding/decode";
+export type EncodeOperation<T extends Ref = any> = (encoder: Encoder, bytes: Buffer, changeTree: ChangeTree<T>, index: number, operation: OPERATION, it: Iterator, isEncodeAll: boolean, hasView: boolean) => void;
+export declare function encodePrimitiveType(type: PrimitiveType, bytes: Buffer, value: any, klass: Schema, field: string | number, it: Iterator): void;
+export declare function encodeValue(encoder: Encoder, bytes: Buffer, ref: Ref, type: any, value: any, field: string | number, operation: OPERATION, it: Iterator): void;
+/**
+ * Used for Schema instances.
+ * @private
+ */
+export declare const encodeSchemaOperation: EncodeOperation;
+/**
+ * Used for collections (MapSchema, CollectionSchema, SetSchema)
+ * @private
+ */
+export declare const encodeKeyValueOperation: EncodeOperation;
+/**
+ * Used for collections (MapSchema, ArraySchema, etc.)
+ * @private
+ */
+export declare const encodeArray: EncodeOperation;

package/lib/encoder/EncodeOperation.js

@@ -0,0 +1,156 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.encodeArray = exports.encodeKeyValueOperation = exports.encodeSchemaOperation = exports.encodeValue = exports.encodePrimitiveType = void 0;
+const spec_1 = require("../encoding/spec");
+const symbols_1 = require("../types/symbols");
+const registry_1 = require("../types/registry");
+const encode = require("../encoding/encode");
+const assert_1 = require("../encoding/assert");
+function encodePrimitiveType(type, bytes, value, klass, field, it) {
+    (0, assert_1.assertType)(value, type, klass, field);
+    const encodeFunc = encode[type];
+    if (encodeFunc) {
+        encodeFunc(bytes, value, it);
+        // encodeFunc(bytes, value);
+    }
+    else {
+        throw new assert_1.EncodeSchemaError(`a '${type}' was expected, but ${value} was provided in ${klass.constructor.name}#${field}`);
+    }
+}
+exports.encodePrimitiveType = encodePrimitiveType;
+;
+function encodeValue(encoder, bytes, ref, type, value, field, operation, it) {
+    if (type[Symbol.metadata] !== undefined) {
+        // TODO: move this to the `@type()` annotation
+        (0, assert_1.assertInstanceType)(value, type, ref, field);
+        //
+        // Encode refId for this instance.
+        // The actual instance is going to be encoded on next `changeTree` iteration.
+        //
+        encode.number(bytes, value[symbols_1.$changes].refId, it);
+        // Try to encode inherited TYPE_ID if it's an ADD operation.
+        if ((operation & spec_1.OPERATION.ADD) === spec_1.OPERATION.ADD) {
+            encoder.tryEncodeTypeId(bytes, type, value.constructor, it);
+        }
+    }
+    else if (typeof (type) === "string") {
+        //
+        // Primitive values
+        //
+        encodePrimitiveType(type, bytes, value, ref, field, it);
+    }
+    else {
+        //
+        // Custom type (MapSchema, ArraySchema, etc)
+        //
+        const definition = (0, registry_1.getType)(Object.keys(type)[0]);
+        //
+        // ensure a ArraySchema has been provided
+        //
+        (0, assert_1.assertInstanceType)(ref[field], definition.constructor, ref, field);
+        //
+        // Encode refId for this instance.
+        // The actual instance is going to be encoded on next `changeTree` iteration.
+        //
+        encode.number(bytes, value[symbols_1.$changes].refId, it);
+    }
+}
+exports.encodeValue = encodeValue;
+/**
+ * Used for Schema instances.
+ * @private
+ */
+const encodeSchemaOperation = function (encoder, bytes, changeTree, index, operation, it) {
+    const ref = changeTree.ref;
+    const metadata = ref['constructor'][Symbol.metadata];
+    const field = metadata[index];
+    const type = metadata[field].type;
+    const value = ref[field];
+    // "compress" field index + operation
+    bytes[it.offset++] = (index | operation) & 255;
+    // Do not encode value for DELETE operations
+    if (operation === spec_1.OPERATION.DELETE) {
+        return;
+    }
+    // TODO: inline this function call small performance gain
+    encodeValue(encoder, bytes, ref, type, value, field, operation, it);
+};
+exports.encodeSchemaOperation = encodeSchemaOperation;
+/**
+ * Used for collections (MapSchema, CollectionSchema, SetSchema)
+ * @private
+ */
+const encodeKeyValueOperation = function (encoder, bytes, changeTree, field, operation, it) {
+    const ref = changeTree.ref;
+    // encode operation
+    bytes[it.offset++] = operation & 255;
+    // custom operations
+    if (operation === spec_1.OPERATION.CLEAR) {
+        return;
+    }
+    // encode index
+    encode.number(bytes, field, it);
+    // Do not encode value for DELETE operations
+    if (operation === spec_1.OPERATION.DELETE) {
+        return;
+    }
+    //
+    // encode "alias" for dynamic fields (maps)
+    //
+    if ((operation & spec_1.OPERATION.ADD) == spec_1.OPERATION.ADD) { // ADD or DELETE_AND_ADD
+        if (typeof (ref['set']) === "function") {
+            //
+            // MapSchema dynamic key
+            //
+            const dynamicIndex = changeTree.ref['$indexes'].get(field);
+            encode.string(bytes, dynamicIndex, it);
+        }
+    }
+    const type = changeTree.getType(field);
+    const value = changeTree.getValue(field);
+    // TODO: inline this function call small performance gain
+    encodeValue(encoder, bytes, ref, type, value, field, operation, it);
+};
+exports.encodeKeyValueOperation = encodeKeyValueOperation;
+/**
+ * Used for collections (MapSchema, ArraySchema, etc.)
+ * @private
+ */
+const encodeArray = function (encoder, bytes, changeTree, field, operation, it, isEncodeAll, hasView) {
+    const ref = changeTree.ref;
+    if (hasView &&
+        operation === spec_1.OPERATION.DELETE &&
+        typeof (changeTree.getType(field)) !== "string") {
+        // encode delete by refId (array of schemas)
+        bytes[it.offset++] = spec_1.OPERATION.DELETE_BY_REFID;
+        const value = ref['tmpItems'][field];
+        const refId = value[symbols_1.$changes].refId;
+        encode.number(bytes, refId, it);
+        return;
+    }
+    // encode operation
+    bytes[it.offset++] = operation & 255;
+    // custom operations
+    if (operation === spec_1.OPERATION.CLEAR) {
+        return;
+    }
+    // encode index
+    encode.number(bytes, field, it);
+    // Do not encode value for DELETE operations
+    if (operation === spec_1.OPERATION.DELETE) {
+        return;
+    }
+    const type = changeTree.getType(field);
+    const value = changeTree.getValue(field, isEncodeAll);
+    // console.log("encodeArray -> ", {
+    //     ref: changeTree.ref.constructor.name,
+    //     field,
+    //     operation: OPERATION[operation],
+    //     value: value?.toJSON(),
+    //     items: ref.toJSON(),
+    // });
+    // TODO: inline this function call small performance gain
+    encodeValue(encoder, bytes, ref, type, value, field, operation, it);
+};
+exports.encodeArray = encodeArray;
+//# sourceMappingURL=EncodeOperation.js.map

package/lib/encoder/EncodeOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EncodeOperation.js","sourceRoot":"","sources":["../../src/encoder/EncodeOperation.ts"],"names":[],"mappings":";;;AAAA,2CAA6C;AAC7C,8CAA4C;AAC5C,gDAA4C;AAE5C,6CAA6C;AAC7C,+CAAuF;AAqBvF,SAAgB,mBAAmB,CAC/B,IAAmB,EACnB,KAAa,EACb,KAAU,EACV,KAAa,EACb,KAAsB,EACtB,EAAY;IAEZ,IAAA,mBAAU,EAAC,KAAK,EAAE,IAAc,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IAEhD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAc,CAAC,CAAC;IAE1C,IAAI,UAAU,EAAE,CAAC;QACb,UAAU,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7B,4BAA4B;IAEhC,CAAC;SAAM,CAAC;QACJ,MAAM,IAAI,0BAAiB,CAAC,MAAM,IAAI,uBAAuB,KAAK,oBAAoB,KAAK,CAAC,WAAW,CAAC,IAAI,IAAI,KAAK,EAAE,CAAC,CAAC;IAC7H,CAAC;AACL,CAAC;AAnBD,kDAmBC;AAAA,CAAC;AAEF,SAAgB,WAAW,CACvB,OAAgB,EAChB,KAAa,EACb,GAAQ,EACR,IAAS,EACT,KAAU,EACV,KAAsB,EACtB,SAAoB,EACpB,EAAY;IAEZ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,SAAS,EAAE,CAAC;QACtC,8CAA8C;QAC9C,IAAA,2BAAkB,EAAC,KAAK,EAAE,IAAqB,EAAE,GAAa,EAAE,KAAK,CAAC,CAAC;QAEvE,EAAE;QACF,kCAAkC;QAClC,6EAA6E;QAC7E,EAAE;QACF,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,kBAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAEhD,4DAA4D;QAC5D,IAAI,CAAC,SAAS,GAAG,gBAAS,CAAC,GAAG,CAAC,KAAK,gBAAS,CAAC,GAAG,EAAE,CAAC;YAChD,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,IAAqB,EAAE,KAAK,CAAC,WAA4B,EAAE,EAAE,CAAC,CAAC;QAClG,CAAC;IAEL,CAAC;SAAM,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,QAAQ,EAAE,CAAC;QACpC,EAAE;QACF,mBAAmB;QACnB,EAAE;QACF,mBAAmB,CAAC,IAAqB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAa,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAEvF,CAAC;SAAM,CAAC;QACJ,EAAE;QACF,4CAA4C;QAC5C,EAAE;QACF,MAAM,UAAU,GAAG,IAAA,kBAAO,EAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEjD,EAAE;QACF,yCAAyC;QACzC,EAAE;QACF,IAAA,2BAAkB,EAAC,GAAG,CAAC,KAAK,CAAC,EAAE,UAAU,CAAC,WAAW,EAAE,GAAa,EAAE,KAAK,CAAC,CAAC;QAE7E,EAAE;QACF,kCAAkC;QAClC,6EAA6E;QAC7E,EAAE;QACF,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,kBAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACpD,CAAC;AACL,CAAC;AAhDD,kCAgDC;AAED;;;GAGG;AACI,MAAM,qBAAqB,GAAoB,UAClD,OAAgB,EAChB,KAAa,EACb,UAA8B,EAC9B,KAAa,EACb,SAAoB,EACpB,EAAY;IAEZ,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC;IAC3B,MAAM,QAAQ,GAAG,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAErD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC9B,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC;IAClC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;IAEzB,qCAAqC;IACrC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,GAAG,CAAC;IAE/C,4CAA4C;IAC5C,IAAI,SAAS,KAAK,gBAAS,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO;IACX,CAAC;IAED,yDAAyD;IACzD,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;AACxE,CAAC,CAAA;AAzBY,QAAA,qBAAqB,yBAyBjC;AAED;;;GAGG;AACI,MAAM,uBAAuB,GAAoB,UACpD,OAAgB,EAChB,KAAa,EACb,UAAsB,EACtB,KAAa,EACb,SAAoB,EACpB,EAAY;IAEZ,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC;IAE3B,mBAAmB;IACnB,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,SAAS,GAAG,GAAG,CAAC;IAErC,oBAAoB;IACpB,IAAI,SAAS,KAAK,gBAAS,CAAC,KAAK,EAAE,CAAC;QAChC,OAAO;IACX,CAAC;IAED,eAAe;IACf,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAEhC,4CAA4C;IAC5C,IAAI,SAAS,KAAK,gBAAS,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO;IACX,CAAC;IAED,EAAE;IACF,2CAA2C;IAC3C,EAAE;IACF,IAAI,CAAC,SAAS,GAAG,gBAAS,CAAC,GAAG,CAAC,IAAI,gBAAS,CAAC,GAAG,EAAE,CAAC,CAAC,wBAAwB;QACxE,IAAI,OAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;YACpC,EAAE;YACF,wBAAwB;YACxB,EAAE;YACF,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;IACL,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEzC,yDAAyD;IACzD,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;AACxE,CAAC,CAAA;AA5CY,QAAA,uBAAuB,2BA4CnC;AAED;;;GAGG;AACI,MAAM,WAAW,GAAoB,UACxC,OAAgB,EAChB,KAAa,EACb,UAAmC,EACnC,KAAa,EACb,SAAoB,EACpB,EAAY,EACZ,WAAoB,EACpB,OAAgB;IAEhB,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC;IAE3B,IACI,OAAO;QACP,SAAS,KAAK,gBAAS,CAAC,MAAM;QAC9B,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,QAAQ,EACjD,CAAC;QACC,4CAA4C;QAC5C,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,gBAAS,CAAC,eAAe,CAAC;QAC/C,MAAM,KAAK,GAAI,GAAG,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,KAAK,CAAC,kBAAQ,CAAC,CAAC,KAAK,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAChC,OAAO;IACX,CAAC;IAED,mBAAmB;IACnB,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,SAAS,GAAG,GAAG,CAAC;IAErC,oBAAoB;IACpB,IAAI,SAAS,KAAK,gBAAS,CAAC,KAAK,EAAE,CAAC;QAChC,OAAO;IACX,CAAC;IAED,eAAe;IACf,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAEhC,4CAA4C;IAC5C,IAAI,SAAS,KAAK,gBAAS,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO;IACX,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAEtD,mCAAmC;IACnC,4CAA4C;IAC5C,aAAa;IACb,uCAAuC;IACvC,8BAA8B;IAC9B,2BAA2B;IAC3B,MAAM;IAEN,yDAAyD;IACzD,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;AACxE,CAAC,CAAA;AAtDY,QAAA,WAAW,eAsDvB","sourcesContent":["import { OPERATION } from \"../encoding/spec\";\nimport { $changes } from \"../types/symbols\";\nimport { getType } from \"../types/registry\";\n\nimport * as encode from \"../encoding/encode\";\nimport { EncodeSchemaError, assertInstanceType, assertType } from \"../encoding/assert\";\n\nimport type { ChangeTree, Ref } from \"./ChangeTree\";\nimport type { Encoder } from \"./Encoder\";\nimport type { Schema } from \"../Schema\";\nimport type { PrimitiveType } from \"../annotations\";\n\nimport type { Iterator } from \"../encoding/decode\";\nimport type { ArraySchema } from \"../types/custom/ArraySchema\";\n\nexport type EncodeOperation<T extends Ref = any> = (\n    encoder: Encoder,\n    bytes: Buffer,\n    changeTree: ChangeTree<T>,\n    index: number,\n    operation: OPERATION,\n    it: Iterator,\n    isEncodeAll: boolean,\n    hasView: boolean,\n) => void;\n\nexport function encodePrimitiveType(\n    type: PrimitiveType,\n    bytes: Buffer,\n    value: any,\n    klass: Schema,\n    field: string | number,\n    it: Iterator,\n) {\n    assertType(value, type as string, klass, field);\n\n    const encodeFunc = encode[type as string];\n\n    if (encodeFunc) {\n        encodeFunc(bytes, value, it);\n        // encodeFunc(bytes, value);\n\n    } else {\n        throw new EncodeSchemaError(`a '${type}' was expected, but ${value} was provided in ${klass.constructor.name}#${field}`);\n    }\n};\n\nexport function encodeValue(\n    encoder: Encoder,\n    bytes: Buffer,\n    ref: Ref,\n    type: any,\n    value: any,\n    field: string | number,\n    operation: OPERATION,\n    it: Iterator,\n) {\n    if (type[Symbol.metadata] !== undefined) {\n        // TODO: move this to the `@type()` annotation\n        assertInstanceType(value, type as typeof Schema, ref as Schema, field);\n\n        //\n        // Encode refId for this instance.\n        // The actual instance is going to be encoded on next `changeTree` iteration.\n        //\n        encode.number(bytes, value[$changes].refId, it);\n\n        // Try to encode inherited TYPE_ID if it's an ADD operation.\n        if ((operation & OPERATION.ADD) === OPERATION.ADD) {\n            encoder.tryEncodeTypeId(bytes, type as typeof Schema, value.constructor as typeof Schema, it);\n        }\n\n    } else if (typeof (type) === \"string\") {\n        //\n        // Primitive values\n        //\n        encodePrimitiveType(type as PrimitiveType, bytes, value, ref as Schema, field, it);\n\n    } else {\n        //\n        // Custom type (MapSchema, ArraySchema, etc)\n        //\n        const definition = getType(Object.keys(type)[0]);\n\n        //\n        // ensure a ArraySchema has been provided\n        //\n        assertInstanceType(ref[field], definition.constructor, ref as Schema, field);\n\n        //\n        // Encode refId for this instance.\n        // The actual instance is going to be encoded on next `changeTree` iteration.\n        //\n        encode.number(bytes, value[$changes].refId, it);\n    }\n}\n\n/**\n * Used for Schema instances.\n * @private\n */\nexport const encodeSchemaOperation: EncodeOperation = function (\n    encoder: Encoder,\n    bytes: Buffer,\n    changeTree: ChangeTree<Schema>,\n    index: number,\n    operation: OPERATION,\n    it: Iterator,\n) {\n    const ref = changeTree.ref;\n    const metadata = ref['constructor'][Symbol.metadata];\n\n    const field = metadata[index];\n    const type = metadata[field].type;\n    const value = ref[field];\n\n    // \"compress\" field index + operation\n    bytes[it.offset++] = (index | operation) & 255;\n\n    // Do not encode value for DELETE operations\n    if (operation === OPERATION.DELETE) {\n        return;\n    }\n\n    // TODO: inline this function call small performance gain\n    encodeValue(encoder, bytes, ref, type, value, field, operation, it);\n}\n\n/**\n * Used for collections (MapSchema, CollectionSchema, SetSchema)\n * @private\n */\nexport const encodeKeyValueOperation: EncodeOperation = function (\n    encoder: Encoder,\n    bytes: Buffer,\n    changeTree: ChangeTree,\n    field: number,\n    operation: OPERATION,\n    it: Iterator,\n) {\n    const ref = changeTree.ref;\n\n    // encode operation\n    bytes[it.offset++] = operation & 255;\n\n    // custom operations\n    if (operation === OPERATION.CLEAR) {\n        return;\n    }\n\n    // encode index\n    encode.number(bytes, field, it);\n\n    // Do not encode value for DELETE operations\n    if (operation === OPERATION.DELETE) {\n        return;\n    }\n\n    //\n    // encode \"alias\" for dynamic fields (maps)\n    //\n    if ((operation & OPERATION.ADD) == OPERATION.ADD) { // ADD or DELETE_AND_ADD\n        if (typeof(ref['set']) === \"function\") {\n            //\n            // MapSchema dynamic key\n            //\n            const dynamicIndex = changeTree.ref['$indexes'].get(field);\n            encode.string(bytes, dynamicIndex, it);\n        }\n    }\n\n    const type = changeTree.getType(field);\n    const value = changeTree.getValue(field);\n\n    // TODO: inline this function call small performance gain\n    encodeValue(encoder, bytes, ref, type, value, field, operation, it);\n}\n\n/**\n * Used for collections (MapSchema, ArraySchema, etc.)\n * @private\n */\nexport const encodeArray: EncodeOperation = function (\n    encoder: Encoder,\n    bytes: Buffer,\n    changeTree: ChangeTree<ArraySchema>,\n    field: number,\n    operation: OPERATION,\n    it: Iterator,\n    isEncodeAll: boolean,\n    hasView: boolean,\n) {\n    const ref = changeTree.ref;\n\n    if (\n        hasView &&\n        operation === OPERATION.DELETE &&\n        typeof (changeTree.getType(field)) !== \"string\"\n    ) {\n        // encode delete by refId (array of schemas)\n        bytes[it.offset++] = OPERATION.DELETE_BY_REFID;\n        const value =  ref['tmpItems'][field];\n        const refId = value[$changes].refId;\n        encode.number(bytes, refId, it);\n        return;\n    }\n\n    // encode operation\n    bytes[it.offset++] = operation & 255;\n\n    // custom operations\n    if (operation === OPERATION.CLEAR) {\n        return;\n    }\n\n    // encode index\n    encode.number(bytes, field, it);\n\n    // Do not encode value for DELETE operations\n    if (operation === OPERATION.DELETE) {\n        return;\n    }\n\n    const type = changeTree.getType(field);\n    const value = changeTree.getValue(field, isEncodeAll);\n\n    // console.log(\"encodeArray -> \", {\n    //     ref: changeTree.ref.constructor.name,\n    //     field,\n    //     operation: OPERATION[operation],\n    //     value: value?.toJSON(),\n    //     items: ref.toJSON(),\n    // });\n\n    // TODO: inline this function call small performance gain\n    encodeValue(encoder, bytes, ref, type, value, field, operation, it);\n}"]}