@knapsack/design-token-utils 5.0.0--canary.6593.4738696.0 → 5.0.0

package/dist/token-src-updater/token-src-updater.js

@@ -1,15 +1,3 @@
-var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
-    if (kind === "m") throw new TypeError("Private method is not writable");
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
-    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
-};
-var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var _TokenSrcUpdater_instances, _TokenSrcUpdater__src, _TokenSrcUpdater_prefix, _TokenSrcUpdater_dataIsStale, _TokenSrcUpdater__data, _TokenSrcUpdater_src_set, _TokenSrcUpdater_assertIdDoesNotHavePrefixes;
 import { produce } from 'immer';
 import loSetWith from 'lodash/setWith.js';
 import loGet from 'lodash/get.js';
@@ -108,539 +96,48 @@ function makeTokenSrc({ tokenId, tokenSrcValueInfo, description: $description, }
  *
  */
 export class TokenSrcUpdater {
+    // Actual src
+    #_src = {};
+    // Hide at runtime
+    // eslint-disable-next-line no-unused-private-class-members -- Reserved for future use
+    #prefix = undefined;
+    #dataIsStale = true;
+    #_data = undefined;
+    /**
+     * This is the group key that all collections are nested under.
+     * Convenient prop instead of `this.data.globalConfig?.collectionsParentKey`
+     */
+    collectionsParentKey;
+    /**
+     * Set src should also process/set data. Remember: external cannot set
+     */
+    set #src(newSrc) {
+        this.#_src = newSrc;
+        this.#dataIsStale = true;
+        this.collectionsParentKey =
+            newSrc?.$extensions?.['cloud.knapsack']?.global?.collectionsParentKey ||
+                '';
+    }
     /**
      * Getter for #_src. Remember: external CAN get (the whole point)
      */
     get src() {
-        return __classPrivateFieldGet(this, _TokenSrcUpdater__src, "f");
+        return this.#_src;
     }
     /**
      * Calculate `TokenDataSummary` from current `src`
      */
     get data() {
-        if (__classPrivateFieldGet(this, _TokenSrcUpdater_dataIsStale, "f") || !__classPrivateFieldGet(this, _TokenSrcUpdater__data, "f")) {
-            __classPrivateFieldSet(this, _TokenSrcUpdater__data, convertTokenGroupToData(this.src), "f");
-            __classPrivateFieldSet(this, _TokenSrcUpdater_dataIsStale, false, "f");
+        if (this.#dataIsStale || !this.#_data) {
+            this.#_data = convertTokenGroupToData(this.src);
+            this.#dataIsStale = false;
         }
-        return __classPrivateFieldGet(this, _TokenSrcUpdater__data, "f");
+        return this.#_data;
     }
     constructor({ src, prefix }) {
-        _TokenSrcUpdater_instances.add(this);
-        // Actual src
-        _TokenSrcUpdater__src.set(this, {});
-        // Hide at runtime
-        _TokenSrcUpdater_prefix.set(this, undefined);
-        _TokenSrcUpdater_dataIsStale.set(this, true);
-        _TokenSrcUpdater__data.set(this, undefined);
-        /**
-         * Ensures that `id` does not start with the Collection Parent Key nor
-         * a Collection ID (if present)
-         */
-        _TokenSrcUpdater_assertIdDoesNotHavePrefixes.set(this, ({ id, collectionId, }) => {
-            const prefix = collectionId
-                ? `${this.collectionsParentKey}.${collectionId}`
-                : this.collectionsParentKey;
-            if (id === null || id === void 0 ? void 0 : id.startsWith(prefix)) {
-                throw new Error(`ID "${id}" should not start with "${prefix}"`);
-            }
-        });
-        /**
-         * Collection operations
-         */
-        this.collection = {
-            /**
-             * Collection -> Create
-             */
-            create: ({ collectionId, excludeInitialMode, }) => {
-                if (this.data.collectionsById[collectionId]) {
-                    throw new Error(`Collection ${collectionId} already exists.`);
-                }
-                this.createGroup({
-                    groupId: `${this.collectionsParentKey}.${collectionId}`,
-                    groupData: {
-                        config: {
-                            collectionInfo: {
-                                type: 'collection',
-                                modes: excludeInitialMode ? [] : [{ id: 'default' }],
-                            },
-                        },
-                    },
-                });
-                return { collectionId };
-            },
-            /**
-             * Collection -> Rename
-             */
-            rename: ({ collectionId, newName, }) => {
-                this.renameGroup({
-                    originalGroupId: `${this.collectionsParentKey}.${collectionId}`,
-                    destinationGroupId: `${this.collectionsParentKey}.${newName}`,
-                });
-            },
-            /**
-             * Collection -> Delete
-             */
-            delete: ({ collectionId }) => {
-                if (collectionId.startsWith(`${this.collectionsParentKey}.`)) {
-                    throw new Error(`Collection IDs should not start with ${this.collectionsParentKey} - those are Group IDs.`);
-                }
-                this.deleteGroup({
-                    groupId: `${this.collectionsParentKey}.${collectionId}`,
-                });
-            },
-        };
-        /**
-         * Collection -> Mode operations
-         */
-        this.mode = {
-            /**
-             * Collection -> Mode -> Create
-             */
-            create: ({ collectionId, modeId, }) => {
-                var _a;
-                const { groupId, modes } = this.data.collectionsById[collectionId];
-                const colGroup = this.data.groupsById[groupId];
-                this.updateGroup({
-                    groupId,
-                    groupData: {
-                        config: {
-                            ...colGroup.groupConfig,
-                            collectionInfo: {
-                                ...(((_a = colGroup.groupConfig) === null || _a === void 0 ? void 0 : _a.collectionInfo) || {}),
-                                type: 'collection',
-                                modes: [
-                                    ...modes,
-                                    {
-                                        id: modeId,
-                                    },
-                                ],
-                            },
-                        },
-                    },
-                });
-            },
-            /**
-             * Collection -> Mode -> Rename
-             * @TODO: ensure new Mode ID does not already exist
-             */
-            rename: ({ collectionId, modeId, newModeId, }) => {
-                // Why use produce here instead of just calling rename() on a loop?
-                // If we can run multiple operations in a single Immer produce(), it prevents
-                // thrashing the this.src object and thus unnecessary updates.
-                __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
-                    var _a;
-                    // Why not just use the processed this.data which is easier to look up on?
-                    // The answer is that the returned object we get here will need to be
-                    // modified within this Immer produce() anyway, so we might as well just
-                    // use the this.src directly.
-                    const originalColData = loGet(draft, [
-                        this.collectionsParentKey,
-                        collectionId,
-                        '$extensions',
-                        'cloud.knapsack',
-                    ]);
-                    const colInfo = (_a = originalColData === null || originalColData === void 0 ? void 0 : originalColData.group) === null || _a === void 0 ? void 0 : _a.collectionInfo;
-                    if (!colInfo || colInfo.type !== 'collection')
-                        throw new Error('Not a Collection');
-                    const { modes = [] } = colInfo;
-                    const originalMode = modes === null || modes === void 0 ? void 0 : modes.find((m) => m.id === modeId);
-                    if (!originalMode)
-                        throw new Error('Mode not found');
-                    // Find every single token with the mode name and change its modes. Use
-                    // this.data for much easier looping here.
-                    this.data.collectionsById[collectionId].vars.forEach((varObj) => {
-                        if (!varObj.modesById[modeId])
-                            return;
-                        // Get original mode data
-                        const original = loGet(draft, `${varObj.groupId}.${originalMode.id}`);
-                        // If values for mode in var not found, don't do anything
-                        if (!original) {
-                            return;
-                        }
-                        // Set new Mode key and value
-                        loSetWith(draft, ensureTokenPath(`${varObj.groupId}.${newModeId}`), original, Object);
-                        // Now nuke the old Mode within the Var group
-                        loUnset(draft, `${varObj.groupId}.${originalMode.id}`); // remove old mode
-                    });
-                    // Finally update the name of mode within the collection group modes array
-                    originalMode.id = newModeId;
-                }), "a", _TokenSrcUpdater_src_set);
-            },
-            /**
-             * Collection -> Mode -> Delete
-             */
-            delete: ({ collectionId, modeId, }) => {
-                __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
-                    var _a;
-                    const originalColData = loGet(draft, [
-                        this.collectionsParentKey,
-                        collectionId,
-                        '$extensions',
-                        'cloud.knapsack',
-                    ]);
-                    const colInfo = (_a = originalColData === null || originalColData === void 0 ? void 0 : originalColData.group) === null || _a === void 0 ? void 0 : _a.collectionInfo;
-                    if (!colInfo || colInfo.type !== 'collection')
-                        throw new Error('Not a Collection');
-                    const { modes = [] } = colInfo;
-                    const originalMode = modes === null || modes === void 0 ? void 0 : modes.find((m) => m.id === modeId);
-                    if (!originalMode)
-                        throw new Error('Mode not found');
-                    // Find every single token with the mode name and change its modes. Use
-                    // this.data for much easier looping here.
-                    this.data.collectionsById[collectionId].vars.forEach((varObj) => {
-                        if (!varObj.modesById[modeId])
-                            return;
-                        // Nuke the old Mode within the Var group
-                        loUnset(draft, `${varObj.groupId}.${originalMode.id}`); // remove old mode
-                    });
-                    // Finally update the name of mode within the collection group modes array
-                    loRemove(modes, (mode) => mode.id === modeId);
-                }), "a", _TokenSrcUpdater_src_set);
-            },
-            reorder: ({ collectionId, modeId, index, }) => {
-                __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
-                    var _a;
-                    const originalColData = loGet(draft, [
-                        this.collectionsParentKey,
-                        collectionId,
-                        '$extensions',
-                        'cloud.knapsack',
-                    ]);
-                    const colInfo = (_a = originalColData === null || originalColData === void 0 ? void 0 : originalColData.group) === null || _a === void 0 ? void 0 : _a.collectionInfo;
-                    if (!colInfo || colInfo.type !== 'collection')
-                        throw new Error('Not a Collection');
-                    const { modes = [] } = colInfo;
-                    const originalMode = modes === null || modes === void 0 ? void 0 : modes.find((m) => m.id === modeId);
-                    if (!originalMode)
-                        throw new Error('Mode not found');
-                    const fromIndex = modes.findIndex((m) => m.id === modeId);
-                    const [theModeToMove] = modes.splice(fromIndex, 1);
-                    modes.splice(index, 0, theModeToMove);
-                }), "a", _TokenSrcUpdater_src_set);
-            },
-        };
-        /**
-         * Collection -> Collection Group operations
-         */
-        this.collectionGroup = {
-            /**
-             * Collection -> Collection Group -> Create
-             */
-            create: ({ collectionId, groupId, parentGroupId, }) => {
-                if (parentGroupId) {
-                    __classPrivateFieldGet(this, _TokenSrcUpdater_assertIdDoesNotHavePrefixes, "f").call(this, { id: parentGroupId, collectionId });
-                }
-                if (this.data.collectionsById[collectionId].groupIds.indexOf(groupId) !== -1) {
-                    throw new Error(`Group ${groupId} already exists in ${collectionId}.`);
-                }
-                this.createGroup({
-                    groupId,
-                });
-                return {
-                    groupId,
-                };
-            },
-            /**
-             * Collection -> Collection Group -> Rename
-             */
-            rename: ({ collectionId, collectionGroupId, name, }) => {
-                assertDotlessId(name);
-                __classPrivateFieldGet(this, _TokenSrcUpdater_assertIdDoesNotHavePrefixes, "f").call(this, {
-                    id: collectionGroupId,
-                    collectionId,
-                });
-                const originalGroupId = `${this.collectionsParentKey}.${collectionId}.${collectionGroupId}`;
-                const destinationGroupId = originalGroupId
-                    .split('.')
-                    // replace the last segment of the ID with the new name
-                    .map((chunk, i, arr) => (i === arr.length - 1 ? name : chunk))
-                    .join('.');
-                this.renameGroup({
-                    originalGroupId,
-                    destinationGroupId,
-                });
-            },
-            /**
-             * Collection -> Collection Group -> Delete
-             */
-            delete: ({ collectionId, collectionGroupId, }) => {
-                __classPrivateFieldGet(this, _TokenSrcUpdater_assertIdDoesNotHavePrefixes, "f").call(this, {
-                    id: collectionGroupId,
-                    collectionId,
-                });
-                this.deleteGroup({
-                    groupId: `${this.collectionsParentKey}.${collectionId}.${collectionGroupId}`,
-                });
-            },
-            /**
-             * Collection -> Collection Group -> Move
-             */
-            move: ({ collectionId, collectionGroupId, newParentCollectionId, newParentCollectionGroupId, }) => {
-                __classPrivateFieldGet(this, _TokenSrcUpdater_assertIdDoesNotHavePrefixes, "f").call(this, {
-                    id: collectionGroupId,
-                    collectionId,
-                });
-                __classPrivateFieldGet(this, _TokenSrcUpdater_assertIdDoesNotHavePrefixes, "f").call(this, {
-                    id: newParentCollectionGroupId,
-                    collectionId,
-                });
-                const originalGroupId = `${this.collectionsParentKey}.${collectionId}.${collectionGroupId}`;
-                const finalSegment = originalGroupId.split('.').at(-1);
-                if (!finalSegment)
-                    throw new Error('Cannot extract final segment.');
-                const destinationCollectionId = newParentCollectionId !== null && newParentCollectionId !== void 0 ? newParentCollectionId : collectionId;
-                // Empty string newParentCollectionGroupId means move to root of collection
-                const destinationCollectionGroupId = newParentCollectionGroupId === ''
-                    ? ''
-                    : `.${newParentCollectionGroupId}`;
-                const destinationGroupId = `${this.collectionsParentKey}.${destinationCollectionId}${destinationCollectionGroupId}.${finalSegment}`;
-                this.renameGroup({
-                    originalGroupId,
-                    destinationGroupId,
-                });
-            },
-            /**
-             * CONVERT vanilla group to collection group
-             */
-            convert: ({ currentGroupId, collectionId, newParentCollectionGroupId, }) => {
-                // Get the vanilla group data
-                const vanillaGroup = this.data.groupsById[currentGroupId];
-                // Create var with default value for each mode
-                try {
-                    // Quick validation that mode we'll be adding later exists
-                    const modeId = this.data.collectionsById[collectionId].modes[0].id;
-                    if (!modeId)
-                        throw new Error('Collection must have at least one mode');
-                    // Make first collection group
-                    this.collectionGroup.create({
-                        collectionId,
-                        groupId: `collections.${collectionId}.${newParentCollectionGroupId}.${vanillaGroup.id}`,
-                    });
-                    // Recursively dive through token group. NOTE: Only a function definition,
-                    // actually called below.
-                    const dive = (group) => {
-                        // Now loop all children, creating var for each
-                        group.children.tokenIds.forEach((tokenId) => {
-                            this.var.create({
-                                collectionId,
-                                groupId: newParentCollectionGroupId,
-                                varId: tokenId,
-                                varType: this.data.tokensById[tokenId].type,
-                            });
-                            const tokenData = this.data.tokensById[tokenId];
-                            const tokenSrcValueInfo = getTokenSrcInfo(tokenData);
-                            // Set the default value for the var for *first* mode
-                            this.var.setModeValue({
-                                collectionId,
-                                varId: `${newParentCollectionGroupId}.${tokenId}`,
-                                modeId,
-                                tokenSrcValueInfo,
-                            });
-                        });
-                        // Yay recursion!
-                        group.children.groupIds.forEach((subGroupId) => {
-                            this.collectionGroup.create({
-                                collectionId,
-                                groupId: `collections.${collectionId}.${newParentCollectionGroupId}.${subGroupId}`,
-                            });
-                            dive(this.data.groupsById[subGroupId]);
-                        });
-                    };
-                    // Kick off the call to the above
-                    dive(vanillaGroup);
-                    // Nuke the original vanilla group
-                    this.deleteGroup({
-                        groupId: currentGroupId,
-                    });
-                }
-                catch (e) {
-                    if (e instanceof Error) {
-                        // @TODO: roll back any changes from the try above
-                        throw new Error(`Error converting vanilla group to collection group: ${e.message}`);
-                    }
-                }
-            },
-        };
-        /**
-         * Var operations
-         */
-        this.var = {
-            /**
-             * Var -> Create
-             */
-            create: ({ collectionId, groupId, varId, varType, }) => {
-                __classPrivateFieldGet(this, _TokenSrcUpdater_assertIdDoesNotHavePrefixes, "f").call(this, {
-                    id: groupId,
-                    collectionId,
-                });
-                const newVarId = groupId ? `${groupId}.${varId}` : varId;
-                const fullGroupId = `${this.collectionsParentKey}.${collectionId}.${newVarId}`;
-                // This is pretty fast
-                const existingVar = loGet(this.src, fullGroupId);
-                if (existingVar) {
-                    console.log(`Var ${newVarId} already exists in ${collectionId}. Skipping.`);
-                    return;
-                }
-                this.createGroup({
-                    groupId: fullGroupId,
-                    groupData: {
-                        type: varType,
-                        config: {
-                            collectionInfo: {
-                                type: 'var',
-                            },
-                        },
-                    },
-                });
-            },
-            /**
-             * Var -> Rename
-             */
-            rename: ({ collectionId, varId, newVarId, }) => {
-                this.renameGroup({
-                    originalGroupId: `${this.collectionsParentKey}.${collectionId}.${varId}`,
-                    destinationGroupId: `${this.collectionsParentKey}.${collectionId}.${newVarId}`,
-                });
-            },
-            /**
-             * Var -> Delete
-             */
-            delete: ({ collectionId, varId, }) => {
-                this.deleteGroup({
-                    groupId: `${this.collectionsParentKey}.${collectionId}.${varId}`,
-                });
-            },
-            /**
-             * Var -> Move/Rename (moving groups is the same as renaming)
-             * @TODO: needs work around existing Var groups
-             */
-            move: ({ collectionId, varId, newParentGroupId, }) => {
-                const varGroupPath = `${this.collectionsParentKey}.${collectionId}.${varId}`;
-                const lastPart = varGroupPath.split('.').at(-1);
-                const destinationGroupId = `${this.collectionsParentKey}.${collectionId}.${newParentGroupId}.${lastPart}`;
-                this.renameGroup({
-                    originalGroupId: varGroupPath,
-                    destinationGroupId,
-                });
-            },
-            /**
-             * Var -> setDescription
-             */
-            setDescription: ({ collectionId, varId, description, }) => {
-                const groupId = `${this.collectionsParentKey}.${collectionId}.${varId}`;
-                this.updateGroup({
-                    groupId,
-                    groupData: {
-                        description,
-                    },
-                });
-            },
-            /**
-             * Var -> setModeValue
-             */
-            /**
-             * Set one or many var mode values
-             */
-            setModeValue: (modeVals) => {
-                // Do everything inside a single Immer produce for batch efficiency
-                __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
-                    // Force the incoming args to be an array so we can handle one or many
-                    castArray(modeVals).forEach((val) => {
-                        var _a, _b;
-                        const { collectionId, varId, modeId, tokenSrcValueInfo, skipRefExistsCheck, } = val;
-                        const { vars } = this.data.collectionsById[collectionId];
-                        const varValue = vars.find((v) => v.id === varId);
-                        if (!varValue) {
-                            throw new Error(`Var ${varId} does not exist in collection ${collectionId}`);
-                        }
-                        const tokenId = `${this.collectionsParentKey}.${collectionId}.${varId}.${modeId}`;
-                        // Handle the different kinds of tokenSrcValueInfo: var refs and static
-                        switch (tokenSrcValueInfo.kind) {
-                            case 'ref': {
-                                const { isVarRef, referencedTokenId } = tokenSrcValueInfo;
-                                // Note that there are TWO kinds of "refs": referencing another var
-                                // (and all its modes therein) and just referencing a vanilla token.
-                                if (isVarRef) {
-                                    if (!skipRefExistsCheck) {
-                                        const ref = this.data.groupsById[referencedTokenId];
-                                        if (!ref) {
-                                            throw new Error(`No ref exists for "${referencedTokenId}"`);
-                                        }
-                                        if (((_b = (_a = ref.groupConfig) === null || _a === void 0 ? void 0 : _a.collectionInfo) === null || _b === void 0 ? void 0 : _b.type) !== 'var') {
-                                            throw new Error(`Referenced token "${referencedTokenId}" is not a var`);
-                                        }
-                                    }
-                                    // This is the fancypants ref to a whole collections var
-                                    const tokenSrc = {
-                                        $extensions: {
-                                            'cloud.knapsack': {
-                                                group: {
-                                                    collectionInfo: {
-                                                        type: 'var-ref',
-                                                        referencedId: referencedTokenId,
-                                                    },
-                                                },
-                                            },
-                                        },
-                                    };
-                                    const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
-                                    // @ts-expect-error - force removing key that should not exist
-                                    delete newTokenSrc.$value;
-                                    loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
-                                    break;
-                                }
-                                // Cannot call another produce function within a produce
-                                // this.updateToken({
-                                //   tokenId,
-                                //   tokenSrcValueInfo,
-                                // });
-                                // @TODO: find a way to call this.updateToken() instead of repeating below
-                                // This is the vanilla token ref
-                                const tokenSrc = makeTokenSrc({
-                                    tokenId,
-                                    tokenSrcValueInfo,
-                                });
-                                const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
-                                loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
-                                break;
-                            }
-                            case 'static': {
-                                // Cannot call another function that uses produce
-                                // this.updateToken({
-                                //   tokenId,
-                                //   tokenSrcValueInfo,
-                                // });
-                                // @TODO: find a way to call this.updateToken() instead of repeating below
-                                const tokenSrc = makeTokenSrc({
-                                    tokenId,
-                                    tokenSrcValueInfo,
-                                });
-                                const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
-                                loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
-                                break;
-                            }
-                            default: {
-                                const _exhaustiveCheck = tokenSrcValueInfo;
-                                throw new Error(`Unhandled kind: ${JSON.stringify(tokenSrcValueInfo)}`);
-                            }
-                        }
-                        // Token Mode Values should not have $type since the $type of the Mode is
-                        // set within the Var object above. This just guarantees $type is not set.
-                        loUnset(draft, `${tokenId}.$type`);
-                    });
-                }), "a", _TokenSrcUpdater_src_set);
-            },
-            /**
-             * Var -> unsetModeValue
-             */
-            unsetModeValue: ({ collectionId, varId, modeId, }) => {
-                const tokenId = `${this.collectionsParentKey}.${collectionId}.${varId}.${modeId}`;
-                this.deleteToken({ tokenId });
-            },
-        };
         this.collectionsParentKey = '';
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, src, "a", _TokenSrcUpdater_src_set);
-        __classPrivateFieldSet(this, _TokenSrcUpdater_prefix, prefix, "f");
+        this.#src = src;
+        this.#prefix = prefix;
     }
     /**
      * Single Token operations
@@ -655,9 +152,9 @@ export class TokenSrcUpdater {
             description,
         });
         // Reminder: mutate draft within produce(). Do NOT return.
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             loSetWith(draft, ensureTokenPath(tokenId), tokenSrc, Object);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
     }
     /**
      * Simple token data updates (value, description, etc)
@@ -669,10 +166,10 @@ export class TokenSrcUpdater {
             description,
         });
         // Reminder: mutate draft within produce(). Do NOT return.
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
             loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
     }
     /**
      * Remove token at dot.path. If other things ref it, update those to original
@@ -709,7 +206,6 @@ export class TokenSrcUpdater {
         // ];
         // @TODO: probably need to sort this by precedence of reference
         const tokensWithRefToDeletedTokens = this.data.tokens.filter((token) => {
-            var _a;
             switch (token.type) {
                 case 'shadow': {
                     if (token.kind === 'static') {
@@ -763,7 +259,7 @@ export class TokenSrcUpdater {
                 case 'fontWeight':
                 case 'cubicBezier':
                 case 'string': {
-                    return (_a = token === null || token === void 0 ? void 0 : token.references) === null || _a === void 0 ? void 0 : _a.some((refStack) => {
+                    return token?.references?.some((refStack) => {
                         // First ref in stack is **direct** reference to deleted token
                         const mostImmediateRef = refStack.at(0);
                         // Make TS happy that it's not undefined
@@ -795,7 +291,7 @@ export class TokenSrcUpdater {
             return x;
         }
         // Reminder: mutate draft within produce(). Do NOT return.
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             // Update these tokens to use original token value.
             tokensWithRefToDeletedTokens.forEach((tokenWithRef) => {
                 switch (tokenWithRef.type) {
@@ -846,7 +342,7 @@ export class TokenSrcUpdater {
             deletedTokens.forEach((deletedToken) => {
                 loUnset(draft, deletedToken.id);
             });
-        }), "a", _TokenSrcUpdater_src_set);
+        });
         // Return deleted tokens for any callee that wants verification
         return deletedTokens;
     }
@@ -877,15 +373,14 @@ export class TokenSrcUpdater {
                 : { $type: undefined }),
         });
         // Reminder: mutate draft within produce(). Do NOT return.
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             loSetWith(draft, ensureTokenPath(destinationTokenId), tokenProcessed, Object);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
         const tokensWithRefToRenamedToken = this.data.tokens.filter((token) => {
-            var _a;
-            return (_a = token === null || token === void 0 ? void 0 : token.references) === null || _a === void 0 ? void 0 : _a.some((refStack) => refStack.at(0) === originalTokenId);
+            return token?.references?.some((refStack) => refStack.at(0) === originalTokenId);
         });
         // Reminder: mutate draft within produce(). Do NOT return.
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             tokensWithRefToRenamedToken.forEach((token) => {
                 // @TODO: handle token values other than just strings; `color`, for
                 // instance, is a string.
@@ -895,7 +390,7 @@ export class TokenSrcUpdater {
                 loSetWith(draft, ensureTokenPath(`${token.id}.$value`), newVal, Object);
             });
             loUnset(draft, originalTokenId);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
         // Return original and new token name for any callee that wants verification
         return {
             originalTokenId,
@@ -907,19 +402,19 @@ export class TokenSrcUpdater {
      */
     createGroup({ groupId, groupData, }) {
         const srcGroup = {
-            ...((groupData === null || groupData === void 0 ? void 0 : groupData.description)
+            ...(groupData?.description
                 ? { $description: groupData.description }
                 : {}),
-            $type: (groupData === null || groupData === void 0 ? void 0 : groupData.type) || undefined,
+            $type: groupData?.type || undefined,
         };
-        if (groupData === null || groupData === void 0 ? void 0 : groupData.config) {
+        if (groupData?.config) {
             srcGroup.$extensions = {
                 'cloud.knapsack': {
                     group: groupData.config,
                 },
             };
         }
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             // What existed prior, spread to ensure COPY of object
             const originalSrcGroup = { ...loGet(draft, groupId, {}) };
             // Merge new into what existed already, but remove empty key/values
@@ -933,7 +428,7 @@ export class TokenSrcUpdater {
             // brackets to delimit object nesting. Thus splitting the groupId by dots
             // into an array means we can support square brackets in path names.
             ensureTokenPath(groupId), mergedSrcGroup, Object);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
     }
     /**
      * Update token fields, like description and type
@@ -944,21 +439,21 @@ export class TokenSrcUpdater {
             ...(groupData.description ? { $description: groupData.description } : {}),
             $type: groupData.type,
         };
-        if (groupData === null || groupData === void 0 ? void 0 : groupData.config) {
+        if (groupData?.config) {
             newSrcGroup.$extensions = {
                 'cloud.knapsack': {
                     group: groupData.config,
                 },
             };
         }
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             // What existed prior, spread to ensure COPY of object
             const originalSrcGroup = { ...loGet(draft, groupId, {}) };
             // Merge new into what existed already, but remove empty key/values
             const mergedSrcGroup = removeUndefinedProps(loMerge(originalSrcGroup, newSrcGroup));
             // Set it
             loSetWith(draft, ensureTokenPath(groupId), mergedSrcGroup, Object);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
     }
     /**
      * Removing a group nukes all tokens within
@@ -972,15 +467,14 @@ export class TokenSrcUpdater {
         this.deleteToken({
             tokenId: this.data.tokens.flatMap(({ id }) => id.startsWith(`${groupId}.`) ? id : []),
         });
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             loUnset(draft, groupId);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
     }
     /**
      * Renaming a group utilizes prior methods
      */
     renameGroup({ originalGroupId, destinationGroupId, }) {
-        var _a;
         const existingGroup = loGet(this.src, originalGroupId);
         const destinationGroup = loGet(this.src, destinationGroupId);
         if (destinationGroup) {
@@ -1016,18 +510,17 @@ export class TokenSrcUpdater {
                         }
                         : {}),
                     ...('$type' in existingGroup ? { type: existingGroup.$type } : {}),
-                    config: (_a = this.data.groupsById[originalGroupId]) === null || _a === void 0 ? void 0 : _a.groupConfig,
+                    config: this.data.groupsById[originalGroupId]?.groupConfig,
                 },
             });
         // Dig through all token values and replace references from the old group ID
         // to the new
         this.data.groups.forEach((group) => {
-            var _a, _b, _c, _d;
             // We're only looking for var-refs
-            if (((_b = (_a = group === null || group === void 0 ? void 0 : group.groupConfig) === null || _a === void 0 ? void 0 : _a.collectionInfo) === null || _b === void 0 ? void 0 : _b.type) !== 'var-ref')
+            if (group?.groupConfig?.collectionInfo?.type !== 'var-ref')
                 return;
             // And the referenced ID does NOT match, skip
-            if (((_d = (_c = group === null || group === void 0 ? void 0 : group.groupConfig) === null || _c === void 0 ? void 0 : _c.collectionInfo) === null || _d === void 0 ? void 0 : _d.referencedId) !== originalGroupId)
+            if (group?.groupConfig?.collectionInfo?.referencedId !== originalGroupId)
                 return;
             // We are a var-ref and it's referencing the group ID that just changed
             // i.e. 'collections.brand.color.primary.modeA'
@@ -1051,24 +544,23 @@ export class TokenSrcUpdater {
         // Finally we delete the original group. Note: we're using produce() here
         // directly instead of calling this.deleteGroup() due to how Immer holds onto
         // the proxy. This apparently keeps it in the same tick.
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
+        this.#src = produce(this.src, (draft) => {
             loUnset(draft, originalGroupId);
-        }), "a", _TokenSrcUpdater_src_set);
+        });
     }
     setCollectionsParentKey({ collectionsParentKey, }) {
-        __classPrivateFieldSet(this, _TokenSrcUpdater_instances, produce(this.src, (draft) => {
-            var _a, _b, _c;
+        this.#src = produce(this.src, (draft) => {
             draft.$extensions = {
                 ...(draft.$extensions || {}),
                 'cloud.knapsack': {
-                    ...(((_a = draft.$extensions) === null || _a === void 0 ? void 0 : _a['cloud.knapsack']) || {}),
+                    ...(draft.$extensions?.['cloud.knapsack'] || {}),
                     global: {
-                        ...(((_c = (_b = draft.$extensions) === null || _b === void 0 ? void 0 : _b['cloud.knapsack']) === null || _c === void 0 ? void 0 : _c.global) || {}),
+                        ...(draft.$extensions?.['cloud.knapsack']?.global || {}),
                         collectionsParentKey,
                     },
                 },
             };
-        }), "a", _TokenSrcUpdater_src_set);
+        });
         // Update the instance property to match the new value
         this.collectionsParentKey = collectionsParentKey;
         if (collectionsParentKey) {
@@ -1077,13 +569,506 @@ export class TokenSrcUpdater {
             });
         }
     }
+    /**
+     * Ensures that `id` does not start with the Collection Parent Key nor
+     * a Collection ID (if present)
+     */
+    #assertIdDoesNotHavePrefixes = ({ id, collectionId, }) => {
+        const prefix = collectionId
+            ? `${this.collectionsParentKey}.${collectionId}`
+            : this.collectionsParentKey;
+        if (id?.startsWith(prefix)) {
+            throw new Error(`ID "${id}" should not start with "${prefix}"`);
+        }
+    };
+    /**
+     * Collection operations
+     */
+    collection = {
+        /**
+         * Collection -> Create
+         */
+        create: ({ collectionId, excludeInitialMode, }) => {
+            if (this.data.collectionsById[collectionId]) {
+                throw new Error(`Collection ${collectionId} already exists.`);
+            }
+            this.createGroup({
+                groupId: `${this.collectionsParentKey}.${collectionId}`,
+                groupData: {
+                    config: {
+                        collectionInfo: {
+                            type: 'collection',
+                            modes: excludeInitialMode ? [] : [{ id: 'default' }],
+                        },
+                    },
+                },
+            });
+            return { collectionId };
+        },
+        /**
+         * Collection -> Rename
+         */
+        rename: ({ collectionId, newName, }) => {
+            this.renameGroup({
+                originalGroupId: `${this.collectionsParentKey}.${collectionId}`,
+                destinationGroupId: `${this.collectionsParentKey}.${newName}`,
+            });
+        },
+        /**
+         * Collection -> Delete
+         */
+        delete: ({ collectionId }) => {
+            if (collectionId.startsWith(`${this.collectionsParentKey}.`)) {
+                throw new Error(`Collection IDs should not start with ${this.collectionsParentKey} - those are Group IDs.`);
+            }
+            this.deleteGroup({
+                groupId: `${this.collectionsParentKey}.${collectionId}`,
+            });
+        },
+    };
+    /**
+     * Collection -> Mode operations
+     */
+    mode = {
+        /**
+         * Collection -> Mode -> Create
+         */
+        create: ({ collectionId, modeId, }) => {
+            const { groupId, modes } = this.data.collectionsById[collectionId];
+            const colGroup = this.data.groupsById[groupId];
+            this.updateGroup({
+                groupId,
+                groupData: {
+                    config: {
+                        ...colGroup.groupConfig,
+                        collectionInfo: {
+                            ...(colGroup.groupConfig?.collectionInfo || {}),
+                            type: 'collection',
+                            modes: [
+                                ...modes,
+                                {
+                                    id: modeId,
+                                },
+                            ],
+                        },
+                    },
+                },
+            });
+        },
+        /**
+         * Collection -> Mode -> Rename
+         * @TODO: ensure new Mode ID does not already exist
+         */
+        rename: ({ collectionId, modeId, newModeId, }) => {
+            // Why use produce here instead of just calling rename() on a loop?
+            // If we can run multiple operations in a single Immer produce(), it prevents
+            // thrashing the this.src object and thus unnecessary updates.
+            this.#src = produce(this.src, (draft) => {
+                // Why not just use the processed this.data which is easier to look up on?
+                // The answer is that the returned object we get here will need to be
+                // modified within this Immer produce() anyway, so we might as well just
+                // use the this.src directly.
+                const originalColData = loGet(draft, [
+                    this.collectionsParentKey,
+                    collectionId,
+                    '$extensions',
+                    'cloud.knapsack',
+                ]);
+                const colInfo = originalColData?.group?.collectionInfo;
+                if (!colInfo || colInfo.type !== 'collection')
+                    throw new Error('Not a Collection');
+                const { modes = [] } = colInfo;
+                const originalMode = modes?.find((m) => m.id === modeId);
+                if (!originalMode)
+                    throw new Error('Mode not found');
+                // Find every single token with the mode name and change its modes. Use
+                // this.data for much easier looping here.
+                this.data.collectionsById[collectionId].vars.forEach((varObj) => {
+                    if (!varObj.modesById[modeId])
+                        return;
+                    // Get original mode data
+                    const original = loGet(draft, `${varObj.groupId}.${originalMode.id}`);
+                    // If values for mode in var not found, don't do anything
+                    if (!original) {
+                        return;
+                    }
+                    // Set new Mode key and value
+                    loSetWith(draft, ensureTokenPath(`${varObj.groupId}.${newModeId}`), original, Object);
+                    // Now nuke the old Mode within the Var group
+                    loUnset(draft, `${varObj.groupId}.${originalMode.id}`); // remove old mode
+                });
+                // Finally update the name of mode within the collection group modes array
+                originalMode.id = newModeId;
+            });
+        },
+        /**
+         * Collection -> Mode -> Delete
+         */
+        delete: ({ collectionId, modeId, }) => {
+            this.#src = produce(this.src, (draft) => {
+                const originalColData = loGet(draft, [
+                    this.collectionsParentKey,
+                    collectionId,
+                    '$extensions',
+                    'cloud.knapsack',
+                ]);
+                const colInfo = originalColData?.group?.collectionInfo;
+                if (!colInfo || colInfo.type !== 'collection')
+                    throw new Error('Not a Collection');
+                const { modes = [] } = colInfo;
+                const originalMode = modes?.find((m) => m.id === modeId);
+                if (!originalMode)
+                    throw new Error('Mode not found');
+                // Find every single token with the mode name and change its modes. Use
+                // this.data for much easier looping here.
+                this.data.collectionsById[collectionId].vars.forEach((varObj) => {
+                    if (!varObj.modesById[modeId])
+                        return;
+                    // Nuke the old Mode within the Var group
+                    loUnset(draft, `${varObj.groupId}.${originalMode.id}`); // remove old mode
+                });
+                // Finally update the name of mode within the collection group modes array
+                loRemove(modes, (mode) => mode.id === modeId);
+            });
+        },
+        reorder: ({ collectionId, modeId, index, }) => {
+            this.#src = produce(this.src, (draft) => {
+                const originalColData = loGet(draft, [
+                    this.collectionsParentKey,
+                    collectionId,
+                    '$extensions',
+                    'cloud.knapsack',
+                ]);
+                const colInfo = originalColData?.group?.collectionInfo;
+                if (!colInfo || colInfo.type !== 'collection')
+                    throw new Error('Not a Collection');
+                const { modes = [] } = colInfo;
+                const originalMode = modes?.find((m) => m.id === modeId);
+                if (!originalMode)
+                    throw new Error('Mode not found');
+                const fromIndex = modes.findIndex((m) => m.id === modeId);
+                const [theModeToMove] = modes.splice(fromIndex, 1);
+                modes.splice(index, 0, theModeToMove);
+            });
+        },
+    };
+    /**
+     * Collection -> Collection Group operations
+     */
+    collectionGroup = {
+        /**
+         * Collection -> Collection Group -> Create
+         */
+        create: ({ collectionId, groupId, parentGroupId, }) => {
+            if (parentGroupId) {
+                this.#assertIdDoesNotHavePrefixes({ id: parentGroupId, collectionId });
+            }
+            if (this.data.collectionsById[collectionId].groupIds.indexOf(groupId) !== -1) {
+                throw new Error(`Group ${groupId} already exists in ${collectionId}.`);
+            }
+            this.createGroup({
+                groupId,
+            });
+            return {
+                groupId,
+            };
+        },
+        /**
+         * Collection -> Collection Group -> Rename
+         */
+        rename: ({ collectionId, collectionGroupId, name, }) => {
+            assertDotlessId(name);
+            this.#assertIdDoesNotHavePrefixes({
+                id: collectionGroupId,
+                collectionId,
+            });
+            const originalGroupId = `${this.collectionsParentKey}.${collectionId}.${collectionGroupId}`;
+            const destinationGroupId = originalGroupId
+                .split('.')
+                // replace the last segment of the ID with the new name
+                .map((chunk, i, arr) => (i === arr.length - 1 ? name : chunk))
+                .join('.');
+            this.renameGroup({
+                originalGroupId,
+                destinationGroupId,
+            });
+        },
+        /**
+         * Collection -> Collection Group -> Delete
+         */
+        delete: ({ collectionId, collectionGroupId, }) => {
+            this.#assertIdDoesNotHavePrefixes({
+                id: collectionGroupId,
+                collectionId,
+            });
+            this.deleteGroup({
+                groupId: `${this.collectionsParentKey}.${collectionId}.${collectionGroupId}`,
+            });
+        },
+        /**
+         * Collection -> Collection Group -> Move
+         */
+        move: ({ collectionId, collectionGroupId, newParentCollectionId, newParentCollectionGroupId, }) => {
+            this.#assertIdDoesNotHavePrefixes({
+                id: collectionGroupId,
+                collectionId,
+            });
+            this.#assertIdDoesNotHavePrefixes({
+                id: newParentCollectionGroupId,
+                collectionId,
+            });
+            const originalGroupId = `${this.collectionsParentKey}.${collectionId}.${collectionGroupId}`;
+            const finalSegment = originalGroupId.split('.').at(-1);
+            if (!finalSegment)
+                throw new Error('Cannot extract final segment.');
+            const destinationCollectionId = newParentCollectionId ?? collectionId;
+            // Empty string newParentCollectionGroupId means move to root of collection
+            const destinationCollectionGroupId = newParentCollectionGroupId === ''
+                ? ''
+                : `.${newParentCollectionGroupId}`;
+            const destinationGroupId = `${this.collectionsParentKey}.${destinationCollectionId}${destinationCollectionGroupId}.${finalSegment}`;
+            this.renameGroup({
+                originalGroupId,
+                destinationGroupId,
+            });
+        },
+        /**
+         * CONVERT vanilla group to collection group
+         */
+        convert: ({ currentGroupId, collectionId, newParentCollectionGroupId, }) => {
+            // Get the vanilla group data
+            const vanillaGroup = this.data.groupsById[currentGroupId];
+            // Create var with default value for each mode
+            try {
+                // Quick validation that mode we'll be adding later exists
+                const modeId = this.data.collectionsById[collectionId].modes[0].id;
+                if (!modeId)
+                    throw new Error('Collection must have at least one mode');
+                // Make first collection group
+                this.collectionGroup.create({
+                    collectionId,
+                    groupId: `collections.${collectionId}.${newParentCollectionGroupId}.${vanillaGroup.id}`,
+                });
+                // Recursively dive through token group. NOTE: Only a function definition,
+                // actually called below.
+                const dive = (group) => {
+                    // Now loop all children, creating var for each
+                    group.children.tokenIds.forEach((tokenId) => {
+                        this.var.create({
+                            collectionId,
+                            groupId: newParentCollectionGroupId,
+                            varId: tokenId,
+                            varType: this.data.tokensById[tokenId].type,
+                        });
+                        const tokenData = this.data.tokensById[tokenId];
+                        const tokenSrcValueInfo = getTokenSrcInfo(tokenData);
+                        // Set the default value for the var for *first* mode
+                        this.var.setModeValue({
+                            collectionId,
+                            varId: `${newParentCollectionGroupId}.${tokenId}`,
+                            modeId,
+                            tokenSrcValueInfo,
+                        });
+                    });
+                    // Yay recursion!
+                    group.children.groupIds.forEach((subGroupId) => {
+                        this.collectionGroup.create({
+                            collectionId,
+                            groupId: `collections.${collectionId}.${newParentCollectionGroupId}.${subGroupId}`,
+                        });
+                        dive(this.data.groupsById[subGroupId]);
+                    });
+                };
+                // Kick off the call to the above
+                dive(vanillaGroup);
+                // Nuke the original vanilla group
+                this.deleteGroup({
+                    groupId: currentGroupId,
+                });
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    // @TODO: roll back any changes from the try above
+                    throw new Error(`Error converting vanilla group to collection group: ${e.message}`);
+                }
+            }
+        },
+    };
+    /**
+     * Var operations
+     */
+    var = {
+        /**
+         * Var -> Create
+         */
+        create: ({ collectionId, groupId, varId, varType, }) => {
+            this.#assertIdDoesNotHavePrefixes({
+                id: groupId,
+                collectionId,
+            });
+            const newVarId = groupId ? `${groupId}.${varId}` : varId;
+            const fullGroupId = `${this.collectionsParentKey}.${collectionId}.${newVarId}`;
+            // This is pretty fast
+            const existingVar = loGet(this.src, fullGroupId);
+            if (existingVar) {
+                console.log(`Var ${newVarId} already exists in ${collectionId}. Skipping.`);
+                return;
+            }
+            this.createGroup({
+                groupId: fullGroupId,
+                groupData: {
+                    type: varType,
+                    config: {
+                        collectionInfo: {
+                            type: 'var',
+                        },
+                    },
+                },
+            });
+        },
+        /**
+         * Var -> Rename
+         */
+        rename: ({ collectionId, varId, newVarId, }) => {
+            this.renameGroup({
+                originalGroupId: `${this.collectionsParentKey}.${collectionId}.${varId}`,
+                destinationGroupId: `${this.collectionsParentKey}.${collectionId}.${newVarId}`,
+            });
+        },
+        /**
+         * Var -> Delete
+         */
+        delete: ({ collectionId, varId, }) => {
+            this.deleteGroup({
+                groupId: `${this.collectionsParentKey}.${collectionId}.${varId}`,
+            });
+        },
+        /**
+         * Var -> Move/Rename (moving groups is the same as renaming)
+         * @TODO: needs work around existing Var groups
+         */
+        move: ({ collectionId, varId, newParentGroupId, }) => {
+            const varGroupPath = `${this.collectionsParentKey}.${collectionId}.${varId}`;
+            const lastPart = varGroupPath.split('.').at(-1);
+            const destinationGroupId = `${this.collectionsParentKey}.${collectionId}.${newParentGroupId}.${lastPart}`;
+            this.renameGroup({
+                originalGroupId: varGroupPath,
+                destinationGroupId,
+            });
+        },
+        /**
+         * Var -> setDescription
+         */
+        setDescription: ({ collectionId, varId, description, }) => {
+            const groupId = `${this.collectionsParentKey}.${collectionId}.${varId}`;
+            this.updateGroup({
+                groupId,
+                groupData: {
+                    description,
+                },
+            });
+        },
+        /**
+         * Var -> setModeValue
+         */
+        /**
+         * Set one or many var mode values
+         */
+        setModeValue: (modeVals) => {
+            // Do everything inside a single Immer produce for batch efficiency
+            this.#src = produce(this.src, (draft) => {
+                // Force the incoming args to be an array so we can handle one or many
+                castArray(modeVals).forEach((val) => {
+                    const { collectionId, varId, modeId, tokenSrcValueInfo, skipRefExistsCheck, } = val;
+                    const { vars } = this.data.collectionsById[collectionId];
+                    const varValue = vars.find((v) => v.id === varId);
+                    if (!varValue) {
+                        throw new Error(`Var ${varId} does not exist in collection ${collectionId}`);
+                    }
+                    const tokenId = `${this.collectionsParentKey}.${collectionId}.${varId}.${modeId}`;
+                    // Handle the different kinds of tokenSrcValueInfo: var refs and static
+                    switch (tokenSrcValueInfo.kind) {
+                        case 'ref': {
+                            const { isVarRef, referencedTokenId } = tokenSrcValueInfo;
+                            // Note that there are TWO kinds of "refs": referencing another var
+                            // (and all its modes therein) and just referencing a vanilla token.
+                            if (isVarRef) {
+                                if (!skipRefExistsCheck) {
+                                    const ref = this.data.groupsById[referencedTokenId];
+                                    if (!ref) {
+                                        throw new Error(`No ref exists for "${referencedTokenId}"`);
+                                    }
+                                    if (ref.groupConfig?.collectionInfo?.type !== 'var') {
+                                        throw new Error(`Referenced token "${referencedTokenId}" is not a var`);
+                                    }
+                                }
+                                // This is the fancypants ref to a whole collections var
+                                const tokenSrc = {
+                                    $extensions: {
+                                        'cloud.knapsack': {
+                                            group: {
+                                                collectionInfo: {
+                                                    type: 'var-ref',
+                                                    referencedId: referencedTokenId,
+                                                },
+                                            },
+                                        },
+                                    },
+                                };
+                                const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
+                                // @ts-expect-error - force removing key that should not exist
+                                delete newTokenSrc.$value;
+                                loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
+                                break;
+                            }
+                            // Cannot call another produce function within a produce
+                            // this.updateToken({
+                            //   tokenId,
+                            //   tokenSrcValueInfo,
+                            // });
+                            // @TODO: find a way to call this.updateToken() instead of repeating below
+                            // This is the vanilla token ref
+                            const tokenSrc = makeTokenSrc({
+                                tokenId,
+                                tokenSrcValueInfo,
+                            });
+                            const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
+                            loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
+                            break;
+                        }
+                        case 'static': {
+                            // Cannot call another function that uses produce
+                            // this.updateToken({
+                            //   tokenId,
+                            //   tokenSrcValueInfo,
+                            // });
+                            // @TODO: find a way to call this.updateToken() instead of repeating below
+                            const tokenSrc = makeTokenSrc({
+                                tokenId,
+                                tokenSrcValueInfo,
+                            });
+                            const newTokenSrc = loMerge(loGet(draft, tokenId, {}), tokenSrc);
+                            loSetWith(draft, ensureTokenPath(tokenId), newTokenSrc, Object);
+                            break;
+                        }
+                        default: {
+                            const _exhaustiveCheck = tokenSrcValueInfo;
+                            throw new Error(`Unhandled kind: ${JSON.stringify(tokenSrcValueInfo)}`);
+                        }
+                    }
+                    // Token Mode Values should not have $type since the $type of the Mode is
+                    // set within the Var object above. This just guarantees $type is not set.
+                    loUnset(draft, `${tokenId}.$type`);
+                });
+            });
+        },
+        /**
+         * Var -> unsetModeValue
+         */
+        unsetModeValue: ({ collectionId, varId, modeId, }) => {
+            const tokenId = `${this.collectionsParentKey}.${collectionId}.${varId}.${modeId}`;
+            this.deleteToken({ tokenId });
+        },
+    };
 }
-_TokenSrcUpdater__src = new WeakMap(), _TokenSrcUpdater_prefix = new WeakMap(), _TokenSrcUpdater_dataIsStale = new WeakMap(), _TokenSrcUpdater__data = new WeakMap(), _TokenSrcUpdater_assertIdDoesNotHavePrefixes = new WeakMap(), _TokenSrcUpdater_instances = new WeakSet(), _TokenSrcUpdater_src_set = function _TokenSrcUpdater_src_set(newSrc) {
-    var _a, _b, _c;
-    __classPrivateFieldSet(this, _TokenSrcUpdater__src, newSrc, "f");
-    __classPrivateFieldSet(this, _TokenSrcUpdater_dataIsStale, true, "f");
-    this.collectionsParentKey =
-        ((_c = (_b = (_a = newSrc === null || newSrc === void 0 ? void 0 : newSrc.$extensions) === null || _a === void 0 ? void 0 : _a['cloud.knapsack']) === null || _b === void 0 ? void 0 : _b.global) === null || _c === void 0 ? void 0 : _c.collectionsParentKey) ||
-            '';
-};
 //# sourceMappingURL=token-src-updater.js.map