@traqula/core 0.0.15 → 0.0.17

package/lib/{CoreFactory.d.ts → AstCoreFactory.d.ts}

@@ -7,7 +7,7 @@ export type SubTyped<Type extends PropertyKey, Subtype extends PropertyKey> = {
     type: Type;
     subType: Subtype;
 };
-export declare class CoreFactory {
+export declare class AstCoreFactory {
     wrap<T>(val: T, loc: SourceLocation): Wrap<T>;
     isLocalized(obj: unknown): obj is Localized;
     sourceLocation(...elements: (undefined | IToken | Localized)[]): SourceLocation;

package/lib/{CoreFactory.js → AstCoreFactory.js}

@@ -1,4 +1,4 @@
-export class CoreFactory {
+export class AstCoreFactory {
     wrap(val, loc) {
         return { val, loc };
     }
@@ -132,4 +132,4 @@ export class CoreFactory {
         return temp.type === type && temp.subType === subType;
     }
 }
-//# sourceMappingURL=CoreFactory.js.map
+//# sourceMappingURL=AstCoreFactory.js.map

package/lib/AstCoreFactory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AstCoreFactory.js","sourceRoot":"","sources":["AstCoreFactory.ts"],"names":[],"mappings":"AAiBA,MAAM,OAAO,cAAc;IAClB,IAAI,CAAI,GAAM,EAAE,GAAmB;QACxC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;IACtB,CAAC;IAEM,WAAW,CAAC,GAAY;QAC7B,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,IAAI,KAAK,IAAI,GAAG;YAC5D,OAAO,GAAG,CAAC,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,GAAG,KAAK,IAAI,IAAI,oBAAoB,IAAI,GAAG,CAAC,GAAG,CAAC;IACvF,CAAC;IAEM,cAAc,CAAC,GAAG,QAA4C;QACnE,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;QAC3D,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC5C,CAAC;QACD,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAC7C,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC;YACtE,IAAI,CAAC,6BAA6B,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;QACpG,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;QAC9B,OAAO;YACL,kBAAkB,EAAE,QAAQ;YAC5B,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC0B,KAAK,CAAC,GAAI,CAAC,KAAK,CAAC,CAAC;gBAC1E,KAAK,CAAC,WAAW;YACnB,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC6B,IAAI,CAAC,GAAI,CAAC,GAAG,CAAC,CAAC;gBACrE,CAAC,IAAI,CAAC,SAAU,GAAG,CAAC,CAAC;SAC1B,CAAC;IACJ,CAAC;IAEM,2BAA2B;QAChC,OAAO,EAAE,kBAAkB,EAAE,eAAe,EAAE,CAAC;IACjD,CAAC;IAED;;OAEG;IACI,cAAc,CAAiB,GAAM;QAC1C,OAAO,EAAE,GAAG,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,2BAA2B,EAAE,EAAE,CAAC;IAC7D,CAAC;IAEM,mBAAmB,CAAC,KAAc,EAAE,MAA6B;QACtE,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,6GAA6G;YAC7G,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;YAC7D,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEM,iBAAiB,CAAmB,GAAM;QAC/C,MAAM,IAAI,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;QACxB,KAAK,MAAM,CAAE,GAAG,EAAE,KAAK,CAAE,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,IAAK,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7G,CAAC;QACD,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACxB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,iBAAiB,CAAiB,GAAM;QAC7C,IAAI,IAAI,CAAC,6BAA6B,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAChD,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,EAAE,GAAG,GAAG,EAAE,CAAC;IACpB,CAAC;IAEM,gBAAgB,CAAC,GAAW;QACjC,OAAO,oBAAoB,IAAI,GAAG,CAAC;IACrC,CAAC;IAEM,oBAAoB,CAAC,KAAa,EAAE,GAAW;QACpD,OAAO;YACL,kBAAkB,EAAE,QAAQ;YAC5B,KAAK;YACL,GAAG;SACJ,CAAC;IACJ,CAAC;IAEM,GAAG;QACR,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,CAAC;IAChD,CAAC;IAEM,sBAAsB,CAAC,GAAW;QACvC,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,kBAAkB,KAAK,QAAQ,CAAC;IAC3E,CAAC;IAEM,6BAA6B,CAAC,GAAW;QAC9C,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,kBAAkB,KAAK,eAAe,CAAC;IAClF,CAAC;IAEM,+BAA+B,CAAC,GAAmB;QACxD,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;IAClC,CAAC;IAIM,yBAAyB,CAAC,UAAyC,EAAE,GAAY;QACtF,IAAI,QAAQ,CAAC;QACb,IAAI,MAAM,CAAC;QACX,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;YACnC,QAAQ,GAAG,UAAU,CAAC;YACtB,MAAM,GAAG,GAAI,CAAC;QAChB,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC;YAC5B,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC;QAC1B,CAAC;QACD,OAAO;YACL,kBAAkB,EAAE,aAAa;YACjC,KAAK,EAAE,QAAQ;YACf,GAAG,EAAE,MAAM;SACZ,CAAC;IACJ,CAAC;IAEM,2BAA2B,CAAC,GAAW;QAC5C,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,kBAAkB,KAAK,aAAa,CAAC;IAChF,CAAC;IAEM,gCAAgC,CAAC,GAAW;QACjD,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,kBAAkB,KAAK,cAAc,CAAC;IACjF,CAAC;IAEM,eAAe,CAAC,IAAe;QACpC,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,CAAC;YAC/C,IAAI,CAAC,gCAAgC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACpD,CAAC;IAEM,WAAW,CAAC,IAAe,EAAE,QAAoB;QACtD,IAAI,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/B,QAAQ,EAAE,CAAC;QACb,CAAC;IACH,CAAC;IAEM,6BAA6B,CAAC,GAAW;QAC9C,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,kBAAkB,KAAK,eAAe,CAAC;IAClF,CAAC;IAEM,QAAQ,CAAsB,GAAW,EAAE,IAAU;QAC1D,MAAM,MAAM,GAAmB,GAAG,CAAC;QACnC,OAAO,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC;IAC9B,CAAC;IAEM,WAAW,CAA8C,GAAW,EAAE,IAAU,EAAE,OAAgB;QAEvG,MAAM,IAAI,GAA0C,GAAG,CAAC;QACxD,OAAO,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,OAAO,KAAK,OAAO,CAAC;IACxD,CAAC;CACF","sourcesContent":["import type { IToken } from 'chevrotain';\n\nimport type {\n  SourceLocation,\n  SourceLocationNodeAutoGenerate,\n  SourceLocationNodeReplace,\n  SourceLocationNoMaterialize,\n  SourceLocationSource,\n  SourceLocationStringReplace,\n  Node,\n  Localized,\n  Wrap,\n} from './nodeTypings.js';\n\nexport type Typed<Type extends PropertyKey> = { type: Type };\nexport type SubTyped<Type extends PropertyKey, Subtype extends PropertyKey> = { type: Type; subType: Subtype };\n\nexport class AstCoreFactory {\n  public wrap<T>(val: T, loc: SourceLocation): Wrap<T> {\n    return { val, loc };\n  }\n\n  public isLocalized(obj: unknown): obj is Localized {\n    return typeof obj === 'object' && obj !== null && 'loc' in obj &&\n      typeof obj.loc === 'object' && obj.loc !== null && 'sourceLocationType' in obj.loc;\n  }\n\n  public sourceLocation(...elements: (undefined | IToken | Localized)[]): SourceLocation {\n    const pureElements = elements.filter(x => x !== undefined);\n    if (pureElements.length === 0) {\n      return this.sourceLocationNoMaterialize();\n    }\n    const filtered = pureElements.filter(element =>\n      !this.isLocalized(element) || this.isSourceLocationSource(element.loc) ||\n      this.isSourceLocationStringReplace(element.loc) || this.isSourceLocationNodeReplace(element.loc));\n    if (filtered.length === 0) {\n      return this.gen();\n    }\n    const first = filtered[0];\n    const last = filtered.at(-1)!;\n    return {\n      sourceLocationType: 'source',\n      start: this.isLocalized(first) ?\n          (<SourceLocationSource | SourceLocationStringReplace> first.loc).start :\n        first.startOffset,\n      end: this.isLocalized(last) ?\n          (<SourceLocationSource | SourceLocationStringReplace> last.loc).end :\n          (last.endOffset! + 1),\n    };\n  }\n\n  public sourceLocationNoMaterialize(): SourceLocationNoMaterialize {\n    return { sourceLocationType: 'noMaterialize' };\n  }\n\n  /**\n   * Returns a copy of the argument that is not materialized\n   */\n  public dematerialized<T extends Node>(arg: T): T & { loc: SourceLocationNoMaterialize } {\n    return { ...arg, loc: this.sourceLocationNoMaterialize() };\n  }\n\n  public safeObjectTransform(value: unknown, mapper: (some: object) => any): any {\n    if (value && typeof value === 'object') {\n      // If you wonder why this is all so hard, this is the reason. We cannot lose the methods of our Array objects\n      if (Array.isArray(value)) {\n        return value.map(x => this.safeObjectTransform(x, mapper));\n      }\n      return mapper(value);\n    }\n    return value;\n  }\n\n  public forcedAutoGenTree<T extends object>(obj: T): T {\n    const copy = { ...obj };\n    for (const [ key, value ] of Object.entries(copy)) {\n      (<Record<string, object>> copy)[key] = this.safeObjectTransform(value, obj => this.forcedAutoGenTree(obj));\n    }\n    if (this.isLocalized(copy)) {\n      copy.loc = this.gen();\n    }\n    return copy;\n  }\n\n  public forceMaterialized<T extends Node>(arg: T): T {\n    if (this.isSourceLocationNoMaterialize(arg.loc)) {\n      return this.forcedAutoGenTree(arg);\n    }\n    return { ...arg };\n  }\n\n  public isSourceLocation(loc: object): loc is SourceLocation {\n    return 'sourceLocationType' in loc;\n  }\n\n  public sourceLocationSource(start: number, end: number): SourceLocationSource {\n    return {\n      sourceLocationType: 'source',\n      start,\n      end,\n    };\n  }\n\n  public gen(): SourceLocationNodeAutoGenerate {\n    return { sourceLocationType: 'autoGenerate' };\n  }\n\n  public isSourceLocationSource(loc: object): loc is SourceLocationSource {\n    return this.isSourceLocation(loc) && loc.sourceLocationType === 'source';\n  }\n\n  public isSourceLocationStringReplace(loc: object): loc is SourceLocationStringReplace {\n    return this.isSourceLocation(loc) && loc.sourceLocationType === 'stringReplace';\n  }\n\n  public sourceLocationNodeReplaceUnsafe(loc: SourceLocation): SourceLocationNodeReplace {\n    if (this.isSourceLocationSource(loc)) {\n      return this.sourceLocationNodeReplace(loc);\n    }\n    throw new Error('not possible');\n  }\n\n  public sourceLocationNodeReplace(location: SourceLocationSource): SourceLocationNodeReplace;\n  public sourceLocationNodeReplace(start: number, end: number): SourceLocationNodeReplace;\n  public sourceLocationNodeReplace(startOrLoc: number | SourceLocationSource, end?: number): SourceLocationNodeReplace {\n    let starting;\n    let ending;\n    if (typeof startOrLoc === 'number') {\n      starting = startOrLoc;\n      ending = end!;\n    } else {\n      starting = startOrLoc.start;\n      ending = startOrLoc.end;\n    }\n    return {\n      sourceLocationType: 'nodeReplace',\n      start: starting,\n      end: ending,\n    };\n  }\n\n  public isSourceLocationNodeReplace(loc: object): loc is SourceLocationNodeReplace {\n    return this.isSourceLocation(loc) && loc.sourceLocationType === 'nodeReplace';\n  }\n\n  public isSourceLocationNodeAutoGenerate(loc: object): loc is SourceLocationNodeAutoGenerate {\n    return this.isSourceLocation(loc) && loc.sourceLocationType === 'autoGenerate';\n  }\n\n  public nodeShouldPrint(node: Localized): boolean {\n    return this.isSourceLocationNodeReplace(node.loc) ||\n      this.isSourceLocationNodeAutoGenerate(node.loc);\n  }\n\n  public printFilter(node: Localized, callback: () => void): void {\n    if (this.nodeShouldPrint(node)) {\n      callback();\n    }\n  }\n\n  public isSourceLocationNoMaterialize(loc: object): loc is SourceLocationNoMaterialize {\n    return this.isSourceLocation(loc) && loc.sourceLocationType === 'noMaterialize';\n  }\n\n  public isOfType<Type extends string>(obj: object, type: Type): obj is Typed<Type> {\n    const casted: { type?: any } = obj;\n    return casted.type === type;\n  }\n\n  public isOfSubType<Type extends string, SubType extends string>(obj: object, type: Type, subType: SubType):\n    obj is SubTyped<Type, SubType> {\n    const temp: { type?: unknown; subType?: unknown } = obj;\n    return temp.type === type && temp.subType === subType;\n  }\n}\n"]}

package/lib/Transformers.d.ts

@@ -3,35 +3,68 @@ type Safeness = 'safe' | 'unsafe';
 type SafeWrap<Safe extends Safeness, obj extends object> = Safe extends 'safe' ? {
     [key in keyof obj]: unknown;
 } : obj;
+export interface VisitContext {
+    shortcut?: boolean;
+    continue?: boolean;
+}
+export interface SelectiveTraversalContext<Nodes> {
+    next?: Nodes[];
+    shortcut?: boolean;
+}
 export declare class TransformerType<Nodes extends Pick<Node, 'type'>> {
+    protected clone<T>(obj: T): T;
     protected safeObjectVisit(value: unknown, mapper: (some: object) => any): unknown;
-    transformNode<Safe extends 'safe' | 'unsafe' = 'safe'>(curObject: object, nodeCallBacks: {
+    /**
+     * Recursively transforms all objects that are not arrays. Mapper is called on deeper objects first.
+     * @param startObject object to start iterating from
+     * @param mapper mapper to transform the various objects - argument is a copy of the original
+     * @param preVisitor callback that is evaluated before iterating deeper.
+     *   If continues is false, we do not iterate deeper, current object is still mapped. - default: true
+     *   If shortcut is true, we do not iterate deeper, nor do we branch out, this mapper will be the last one called.
+     *    - Default false
+     */
+    transformObject(startObject: object, mapper: (some: object) => any, preVisitor?: (some: object) => VisitContext): unknown;
+    /**
+     * Visitor that visits all objects. Visits deeper objects first.
+     */
+    visitObject(startObject: object, visitor: (some: object) => void, preVisitor?: (some: object) => VisitContext): void;
+    transformNode<Safe extends 'safe' | 'unsafe' = 'safe'>(startObject: object, nodeCallBacks: {
         [T in Nodes['type']]?: {
             transform?: (op: SafeWrap<Safe, Extract<Nodes, {
                 type: T;
             }>>) => any;
-            continue?: (op: Extract<Nodes, {
+            preVisitor?: (op: Extract<Nodes, {
                 type: T;
-            }>) => boolean;
+            }>) => VisitContext;
         };
     }): any;
-    visitNode(curObject: object, nodeCallBacks: {
+    visitNode(startObject: object, nodeCallBacks: {
+        [T in Nodes['type']]?: {
+            visitor?: (op: Extract<Nodes, {
+                type: T;
+            }>) => void;
+            preVisitor?: (op: Extract<Nodes, {
+                type: T;
+            }>) => VisitContext;
+        };
+    }): void;
+    traverseNodes(currentNode: Nodes, traverse: {
         [T in Nodes['type']]?: (op: Extract<Nodes, {
             type: T;
-        }>) => boolean;
+        }>) => SelectiveTraversalContext<Nodes>;
     }): void;
 }
 export declare class TransformerSubType<Nodes extends Pick<Node, 'type' | 'subType'>> extends TransformerType<Nodes> {
-    transformNodeSpecific<Safe extends 'safe' | 'unsafe' = 'safe'>(curObject: object, nodeCallBacks: {
+    transformNodeSpecific<Safe extends 'safe' | 'unsafe' = 'safe'>(startObject: object, nodeCallBacks: {
         [T in Nodes['type']]?: {
             transform?: (op: SafeWrap<Safe, Extract<Nodes, {
                 type: T;
             }>>) => any;
-            continue?: (op: Extract<Nodes, {
+            preVisitor?: (op: Extract<Nodes, {
                 type: T;
-            }>) => boolean;
+            }>) => VisitContext;
         };
-    }, nodeSpecificCallBacks?: {
+    }, nodeSpecificCallBacks: {
         [Type in Nodes['type']]?: {
             [SubType in Extract<Nodes, {
                 type: Type;
@@ -41,21 +74,47 @@ export declare class TransformerSubType<Nodes extends Pick<Node, 'type' | 'subTy
                     type: Type;
                     subType: SubType;
                 }>>) => any;
-                continue?: (op: Extract<Nodes, {
+                preVisitor?: (op: Extract<Nodes, {
                     type: Type;
                     subType: SubType;
-                }>) => boolean;
+                }>) => VisitContext;
             };
         };
     }): any;
     /**
      * When both nodeCallBack and NodeSpecific callBack are matched, will only look at nodeSpecifCallback
      */
-    visitNodeSpecific(curObject: object, nodeCallBacks: {
-        [T in Nodes['type']]?: (op: Extract<Nodes, {
-            type: T;
-        }>) => boolean | void;
-    }, nodeSpecificCallBacks?: {
+    visitNodeSpecific(startObject: object, nodeCallBacks: {
+        [T in Nodes['type']]?: {
+            visitor?: (op: Extract<Nodes, {
+                type: T;
+            }>) => void;
+            preVisitor?: (op: Extract<Nodes, {
+                type: T;
+            }>) => VisitContext;
+        };
+    }, nodeSpecificCallBacks: {
+        [Type in Nodes['type']]?: {
+            [Subtype in Extract<Nodes, {
+                type: Type;
+                subType: string;
+            }>['subType']]?: {
+                visitor?: (op: Extract<Nodes, {
+                    type: Type;
+                    subType: Subtype;
+                }>) => void;
+                preVisitor?: (op: Extract<Nodes, {
+                    type: Type;
+                    subType: Subtype;
+                }>) => VisitContext;
+            };
+        };
+    }): void;
+    traverseSubNodes(currentNode: Nodes, traverseNode: {
+        [Type in Nodes['type']]?: (op: Extract<Nodes, {
+            type: Type;
+        }>) => SelectiveTraversalContext<Nodes>;
+    }, traverseSubNode: {
         [Type in Nodes['type']]?: {
             [Subtype in Extract<Nodes, {
                 type: Type;
@@ -63,7 +122,7 @@ export declare class TransformerSubType<Nodes extends Pick<Node, 'type' | 'subTy
             }>['subType']]?: (op: Extract<Nodes, {
                 type: Type;
                 subType: Subtype;
-            }>) => boolean | void;
+            }>) => SelectiveTraversalContext<Nodes>;
         };
     }): void;
 }

package/lib/Transformers.js

@@ -1,7 +1,11 @@
 export class TransformerType {
+    clone(obj) {
+        const newObj = Object.create(Object.getPrototypeOf(obj));
+        Object.defineProperties(newObj, Object.getOwnPropertyDescriptors(obj));
+        return newObj;
+    }
     safeObjectVisit(value, mapper) {
         if (value && typeof value === 'object') {
-            // If you wonder why this is all so hard, this is the reason. We cannot lose the methods of our Array objects
             if (Array.isArray(value)) {
                 return value.map(x => this.safeObjectVisit(x, mapper));
             }
@@ -9,106 +13,214 @@ export class TransformerType {
         }
         return value;
     }
-    transformNode(curObject, nodeCallBacks) {
-        let continueCheck;
-        let transformation;
-        const casted = curObject;
-        if (casted.type) {
-            continueCheck = nodeCallBacks[casted.type]?.continue;
-            transformation = nodeCallBacks[casted.type]?.transform;
-        }
-        let shouldContinue = true;
-        if (continueCheck) {
-            shouldContinue = continueCheck(curObject);
-        }
-        if (!shouldContinue) {
+    /**
+     * Recursively transforms all objects that are not arrays. Mapper is called on deeper objects first.
+     * @param startObject object to start iterating from
+     * @param mapper mapper to transform the various objects - argument is a copy of the original
+     * @param preVisitor callback that is evaluated before iterating deeper.
+     *   If continues is false, we do not iterate deeper, current object is still mapped. - default: true
+     *   If shortcut is true, we do not iterate deeper, nor do we branch out, this mapper will be the last one called.
+     *    - Default false
+     */
+    transformObject(startObject, mapper, preVisitor = () => ({})) {
+        let didShortCut = false;
+        const recurse = (curObject) => {
+            const copy = this.clone(curObject);
+            const context = preVisitor(copy);
+            didShortCut = context.shortcut ?? false;
+            const continues = context.continue ?? true;
+            if (continues && !didShortCut) {
+                for (const [key, value] of Object.entries(copy)) {
+                    if (didShortCut) {
+                        return copy;
+                    }
+                    copy[key] =
+                        this.safeObjectVisit(value, obj => recurse(obj));
+                }
+            }
+            return mapper(copy);
+        };
+        return recurse(startObject);
+    }
+    /**
+     * Visitor that visits all objects. Visits deeper objects first.
+     */
+    visitObject(startObject, visitor, preVisitor = () => ({})) {
+        let didShortCut = false;
+        const recurse = (curObject) => {
+            const context = preVisitor(curObject);
+            didShortCut = context.shortcut ?? false;
+            const continues = context.continue ?? true;
+            if (continues && !didShortCut) {
+                for (const value of Object.values(curObject)) {
+                    if (didShortCut) {
+                        return;
+                    }
+                    this.safeObjectVisit(value, obj => recurse(obj));
+                }
+            }
+            visitor(curObject);
+        };
+        recurse(startObject);
+    }
+    transformNode(startObject, nodeCallBacks) {
+        const transformWrapper = (curObject) => {
+            const casted = curObject;
+            if (casted.type) {
+                const ogFunc = nodeCallBacks[casted.type]?.transform;
+                if (ogFunc) {
+                    return ogFunc(casted);
+                }
+            }
             return curObject;
-        }
-        const copy = { ...curObject };
-        for (const [key, value] of Object.entries(copy)) {
-            copy[key] =
-                this.safeObjectVisit(value, obj => this.transformNode(obj, nodeCallBacks));
-        }
-        if (transformation) {
-            return transformation(copy);
-        }
-        return copy;
+        };
+        const preTransformWrapper = (curObject) => {
+            const casted = curObject;
+            if (casted.type) {
+                const ogFunc = nodeCallBacks[casted.type]?.preVisitor;
+                if (ogFunc) {
+                    return ogFunc(casted);
+                }
+            }
+            return {};
+        };
+        return this.transformObject(startObject, transformWrapper, preTransformWrapper);
     }
-    visitNode(curObject, nodeCallBacks) {
-        let callback;
-        const casted = curObject;
-        if (casted.type) {
-            callback = nodeCallBacks[casted.type];
-        }
-        let shouldIterate = true;
-        if (callback) {
-            shouldIterate = callback(curObject);
-        }
-        if (shouldIterate) {
-            for (const value of Object.values(curObject)) {
-                this.safeObjectVisit(value, obj => this.visitNode(obj, nodeCallBacks));
+    visitNode(startObject, nodeCallBacks) {
+        const visitWrapper = (curObject) => {
+            const casted = curObject;
+            if (casted.type) {
+                const ogFunc = nodeCallBacks[casted.type]?.visitor;
+                if (ogFunc) {
+                    ogFunc(casted);
+                }
             }
-        }
+        };
+        const preVisitWrapper = (curObject) => {
+            const casted = curObject;
+            if (casted.type) {
+                const ogFunc = nodeCallBacks[casted.type]?.preVisitor;
+                if (ogFunc) {
+                    return ogFunc(casted);
+                }
+            }
+            return {};
+        };
+        this.visitObject(startObject, visitWrapper, preVisitWrapper);
+    }
+    traverseNodes(currentNode, traverse) {
+        let didShortCut = false;
+        const recurse = (curNode) => {
+            const traverser = traverse[curNode.type];
+            if (traverser) {
+                const { next, shortcut } = traverser(curNode);
+                didShortCut = shortcut ?? false;
+                if (!didShortCut) {
+                    for (const node of next ?? []) {
+                        if (didShortCut) {
+                            return;
+                        }
+                        recurse(node);
+                    }
+                }
+            }
+        };
+        recurse(currentNode);
     }
 }
 export class TransformerSubType extends TransformerType {
-    // Public visitObjects(curObject: object, visitor: (current: object) => void): void {
-    //   for (const value of Object.values(curObject)) {
-    //     this.safeObjectVisit(value, obj => this.visitObjects(obj, visitor));
-    //   }
-    // }
-    transformNodeSpecific(curObject, nodeCallBacks, nodeSpecificCallBacks = {}) {
-        let continueCheck;
-        let transformation;
-        const casted = curObject;
-        if (casted.type && casted.subType) {
-            const specific = nodeSpecificCallBacks[casted.type];
-            if (specific) {
-                continueCheck = specific[casted.subType]?.continue;
-                transformation = specific[casted.subType]?.transform;
+    transformNodeSpecific(startObject, nodeCallBacks, nodeSpecificCallBacks) {
+        const transformWrapper = (curObject) => {
+            let ogTransform;
+            const casted = curObject;
+            if (casted.type && casted.subType) {
+                const specific = nodeSpecificCallBacks[casted.type];
+                if (specific) {
+                    ogTransform = specific[casted.subType]?.transform;
+                }
+                if (!ogTransform) {
+                    ogTransform = nodeCallBacks[casted.type]?.transform;
+                }
             }
-        }
-        let shouldContinue = true;
-        if (continueCheck) {
-            shouldContinue = continueCheck(curObject);
-        }
-        if (!shouldContinue) {
-            return curObject;
-        }
-        const copy = { ...curObject };
-        for (const [key, value] of Object.entries(copy)) {
-            copy[key] =
-                this.safeObjectVisit(value, obj => this.transformNodeSpecific(obj, nodeCallBacks, nodeSpecificCallBacks));
-        }
-        if (transformation) {
-            return transformation(copy);
-        }
-        return copy;
+            return ogTransform ? ogTransform(casted) : curObject;
+        };
+        const preVisitWrapper = (curObject) => {
+            let ogPreVisit;
+            const casted = curObject;
+            if (casted.type && casted.subType) {
+                const specific = nodeSpecificCallBacks[casted.type];
+                if (specific) {
+                    ogPreVisit = specific[casted.subType]?.preVisitor;
+                }
+                if (!ogPreVisit) {
+                    ogPreVisit = nodeCallBacks[casted.type]?.preVisitor;
+                }
+            }
+            return ogPreVisit ? ogPreVisit(casted) : curObject;
+        };
+        return this.transformObject(startObject, transformWrapper, preVisitWrapper);
     }
     /**
      * When both nodeCallBack and NodeSpecific callBack are matched, will only look at nodeSpecifCallback
      */
-    visitNodeSpecific(curObject, nodeCallBacks, nodeSpecificCallBacks = {}) {
-        let callback;
-        const casted = curObject;
-        if (casted.type && casted.subType) {
-            const specific = nodeSpecificCallBacks[casted.type];
-            if (specific) {
-                callback = specific[casted.subType];
+    visitNodeSpecific(startObject, nodeCallBacks, nodeSpecificCallBacks) {
+        const visitWrapper = (curObject) => {
+            let ogTransform;
+            const casted = curObject;
+            if (casted.type && casted.subType) {
+                const specific = nodeSpecificCallBacks[casted.type];
+                if (specific) {
+                    ogTransform = specific[casted.subType]?.visitor;
+                }
+                if (!ogTransform) {
+                    ogTransform = nodeCallBacks[casted.type]?.visitor;
+                }
             }
-        }
-        if (!callback && casted.type) {
-            callback = nodeCallBacks[curObject.type];
-        }
-        let shouldIterate = true;
-        if (callback) {
-            shouldIterate = callback(curObject) ?? true;
-        }
-        if (shouldIterate) {
-            for (const value of Object.values(curObject)) {
-                this.safeObjectVisit(value, obj => this.visitNodeSpecific(obj, nodeCallBacks, nodeSpecificCallBacks));
+            if (ogTransform) {
+                ogTransform(casted);
             }
-        }
+        };
+        const preVisitWrapper = (curObject) => {
+            let ogPreVisit;
+            const casted = curObject;
+            if (casted.type && casted.subType) {
+                const specific = nodeSpecificCallBacks[casted.type];
+                if (specific) {
+                    ogPreVisit = specific[casted.subType]?.preVisitor;
+                }
+                if (!ogPreVisit) {
+                    ogPreVisit = nodeCallBacks[casted.type]?.preVisitor;
+                }
+            }
+            return ogPreVisit ? ogPreVisit(casted) : curObject;
+        };
+        this.visitObject(startObject, visitWrapper, preVisitWrapper);
+    }
+    traverseSubNodes(currentNode, traverseNode, traverseSubNode) {
+        let didShortCut = false;
+        const recurse = (curNode) => {
+            let traverser;
+            const subObj = traverseSubNode[curNode.type];
+            if (subObj) {
+                traverser = subObj[curNode.subType];
+            }
+            if (!traverser) {
+                traverser = traverseNode[curNode.type];
+            }
+            if (traverser) {
+                const { next, shortcut } = traverser(curNode);
+                didShortCut = shortcut ?? false;
+                if (!didShortCut) {
+                    for (const node of next ?? []) {
+                        if (didShortCut) {
+                            return;
+                        }
+                        recurse(node);
+                    }
+                }
+            }
+        };
+        recurse(currentNode);
     }
 }
 //# sourceMappingURL=Transformers.js.map

package/lib/Transformers.js.map

@@ -1 +1 @@
-{"version":3,"file":"Transformers.js","sourceRoot":"","sources":["Transformers.ts"],"names":[],"mappings":"AAKA,MAAM,OAAO,eAAe;IAChB,eAAe,CAAC,KAAc,EAAE,MAA6B;QACrE,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,6GAA6G;YAC7G,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;YACzD,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEM,aAAa,CAClB,SAAiB,EACjB,aAGE;QAEF,IAAI,aAAmD,CAAC;QACxD,IAAI,cAAoD,CAAC;QACzD,MAAM,MAAM,GAA6B,SAAS,CAAC;QACnD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC;YACrD,cAAc,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC;QACzD,CAAC;QACD,IAAI,cAAc,GAAG,IAAI,CAAC;QAC1B,IAAI,aAAa,EAAE,CAAC;YAClB,cAAc,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,IAAI,GAAuB,EAAE,GAAG,SAAS,EAAE,CAAC;QAClD,KAAK,MAAM,CAAE,GAAG,EAAE,KAAK,CAAE,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,IAAK,CAAC,GAAG,CAAC;gBACnC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAC/E,CAAC;QACD,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,SAAS,CACd,SAAiB,EACjB,aAAqF;QAErF,IAAI,QAA8C,CAAC;QACnD,MAAM,MAAM,GAA6B,SAAS,CAAC;QACnD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,aAAa,GAAG,IAAI,CAAC;QACzB,IAAI,QAAQ,EAAE,CAAC;YACb,aAAa,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7C,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,kBAAiE,SAAQ,eAAsB;IAC1G,qFAAqF;IACrF,oDAAoD;IACpD,2EAA2E;IAC3E,MAAM;IACN,IAAI;IAEG,qBAAqB,CAC1B,SAAiB,EACjB,aAGE,EACF,wBAIQ,EAAE;QAEV,IAAI,aAAmD,CAAC;QACxD,IAAI,cAAoD,CAAC;QACzD,MAAM,MAAM,GAA+C,SAAS,CAAC;QACrE,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpD,IAAI,QAAQ,EAAE,CAAC;gBACb,aAAa,GAAG,QAAQ,CAAyB,MAAM,CAAC,OAAO,CAAC,EAAE,QAAQ,CAAC;gBAC3E,cAAc,GAAG,QAAQ,CAAyB,MAAM,CAAC,OAAO,CAAC,EAAE,SAAS,CAAC;YAC/E,CAAC;QACH,CAAC;QACD,IAAI,cAAc,GAAG,IAAI,CAAC;QAC1B,IAAI,aAAa,EAAE,CAAC;YAClB,cAAc,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,IAAI,GAAuB,EAAE,GAAG,SAAS,EAAE,CAAC;QAClD,KAAK,MAAM,CAAE,GAAG,EAAE,KAAK,CAAE,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,IAAK,CAAC,GAAG,CAAC;gBACnC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,aAAa,EAAE,qBAAqB,CAAC,CAAC,CAAC;QAC9G,CAAC;QACD,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACI,iBAAiB,CACtB,SAAiB,EACjB,aAA4F,EAC5F,wBAEkF,EAAE;QAEpF,IAAI,QAAqD,CAAC;QAC1D,MAAM,MAAM,GAAgD,SAAS,CAAC;QACtE,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpD,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,GAAG,QAAQ,CAAwB,MAAM,CAAC,OAAO,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QACD,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAC7B,QAAQ,GAAG,aAAa,CAA4B,SAAU,CAAC,IAAI,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,aAAa,GAAG,IAAI,CAAC;QACzB,IAAI,QAAQ,EAAE,CAAC;YACb,aAAa,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC;QAC9C,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7C,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,aAAa,EAAE,qBAAqB,CAAC,CAAC,CAAC;YACxG,CAAC;QACH,CAAC;IACH,CAAC;CACF","sourcesContent":["import type { Node } from './nodeTypings.js';\n\ntype Safeness = 'safe' | 'unsafe';\ntype SafeWrap<Safe extends Safeness, obj extends object> = Safe extends 'safe' ? {[key in keyof obj]: unknown } : obj;\n\nexport class TransformerType<Nodes extends Pick<Node, 'type'>> {\n  protected safeObjectVisit(value: unknown, mapper: (some: object) => any): unknown {\n    if (value && typeof value === 'object') {\n      // If you wonder why this is all so hard, this is the reason. We cannot lose the methods of our Array objects\n      if (Array.isArray(value)) {\n        return value.map(x => this.safeObjectVisit(x, mapper));\n      }\n      return mapper(value);\n    }\n    return value;\n  }\n\n  public transformNode<Safe extends 'safe' | 'unsafe' = 'safe'>(\n    curObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: {\n      transform?: (op: SafeWrap<Safe, Extract<Nodes, { type: T }>>) => any;\n      continue?: (op: Extract<Nodes, { type: T }>) => boolean;\n    }},\n  ): any {\n    let continueCheck: ((node: any) => boolean) | undefined;\n    let transformation: ((node: any) => unknown) | undefined;\n    const casted = <{ type?: Nodes['type'] }>curObject;\n    if (casted.type) {\n      continueCheck = nodeCallBacks[casted.type]?.continue;\n      transformation = nodeCallBacks[casted.type]?.transform;\n    }\n    let shouldContinue = true;\n    if (continueCheck) {\n      shouldContinue = continueCheck(curObject);\n    }\n    if (!shouldContinue) {\n      return curObject;\n    }\n    const copy: { type?: unknown } = { ...curObject };\n    for (const [ key, value ] of Object.entries(copy)) {\n      (<Record<string, unknown>> copy)[key] =\n        this.safeObjectVisit(value, obj => this.transformNode(obj, nodeCallBacks));\n    }\n    if (transformation) {\n      return transformation(copy);\n    }\n    return copy;\n  }\n\n  public visitNode(\n    curObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: (op: Extract<Nodes, { type: T }>) => boolean },\n  ): void {\n    let callback: ((node: any) => boolean) | undefined;\n    const casted = <{ type?: Nodes['type'] }>curObject;\n    if (casted.type) {\n      callback = nodeCallBacks[casted.type];\n    }\n    let shouldIterate = true;\n    if (callback) {\n      shouldIterate = callback(curObject);\n    }\n    if (shouldIterate) {\n      for (const value of Object.values(curObject)) {\n        this.safeObjectVisit(value, obj => this.visitNode(obj, nodeCallBacks));\n      }\n    }\n  }\n}\n\nexport class TransformerSubType<Nodes extends Pick<Node, 'type' | 'subType'>> extends TransformerType<Nodes> {\n  // Public visitObjects(curObject: object, visitor: (current: object) => void): void {\n  //   for (const value of Object.values(curObject)) {\n  //     this.safeObjectVisit(value, obj => this.visitObjects(obj, visitor));\n  //   }\n  // }\n\n  public transformNodeSpecific<Safe extends 'safe' | 'unsafe' = 'safe'>(\n    curObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: {\n      transform?: (op: SafeWrap<Safe, Extract<Nodes, { type: T }>>) => any;\n      continue?: (op: Extract<Nodes, { type: T }>) => boolean;\n    }},\n    nodeSpecificCallBacks: {[Type in Nodes['type']]?: {\n      [SubType in Extract<Nodes, { type: Type; subType: string }>['subType']]?: {\n        transform?: (op: SafeWrap<Safe, Extract<Nodes, { type: Type; subType: SubType }>>) => any;\n        continue?: (op: Extract<Nodes, { type: Type; subType: SubType }>) => boolean;\n      }}} = {},\n  ): any {\n    let continueCheck: ((node: any) => boolean) | undefined;\n    let transformation: ((node: any) => unknown) | undefined;\n    const casted = <{ type?: Nodes['type']; subType?: string }>curObject;\n    if (casted.type && casted.subType) {\n      const specific = nodeSpecificCallBacks[casted.type];\n      if (specific) {\n        continueCheck = specific[<keyof typeof specific> casted.subType]?.continue;\n        transformation = specific[<keyof typeof specific> casted.subType]?.transform;\n      }\n    }\n    let shouldContinue = true;\n    if (continueCheck) {\n      shouldContinue = continueCheck(curObject);\n    }\n    if (!shouldContinue) {\n      return curObject;\n    }\n    const copy: { type?: unknown } = { ...curObject };\n    for (const [ key, value ] of Object.entries(copy)) {\n      (<Record<string, unknown>> copy)[key] =\n        this.safeObjectVisit(value, obj => this.transformNodeSpecific(obj, nodeCallBacks, nodeSpecificCallBacks));\n    }\n    if (transformation) {\n      return transformation(copy);\n    }\n    return copy;\n  }\n\n  /**\n   * When both nodeCallBack and NodeSpecific callBack are matched, will only look at nodeSpecifCallback\n   */\n  public visitNodeSpecific(\n    curObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: (op: Extract<Nodes, { type: T }>) => boolean | void },\n    nodeSpecificCallBacks: {[Type in Nodes['type']]?:\n      {[Subtype in Extract<Nodes, { type: Type; subType: string }>['subType']]?:\n        (op: Extract<Nodes, { type: Type; subType: Subtype }>) => boolean | void }} = {},\n  ): void {\n    let callback: ((node: any) => boolean | void) | undefined;\n    const casted = <{ type?: Nodes['type']; subType?: string }> curObject;\n    if (casted.type && casted.subType) {\n      const specific = nodeSpecificCallBacks[casted.type];\n      if (specific) {\n        callback = specific[<keyof typeof specific>casted.subType];\n      }\n    }\n    if (!callback && casted.type) {\n      callback = nodeCallBacks[(<{ type: Nodes['type'] }> curObject).type];\n    }\n    let shouldIterate = true;\n    if (callback) {\n      shouldIterate = callback(curObject) ?? true;\n    }\n    if (shouldIterate) {\n      for (const value of Object.values(curObject)) {\n        this.safeObjectVisit(value, obj => this.visitNodeSpecific(obj, nodeCallBacks, nodeSpecificCallBacks));\n      }\n    }\n  }\n}\n"]}
+{"version":3,"file":"Transformers.js","sourceRoot":"","sources":["Transformers.ts"],"names":[],"mappings":"AAeA,MAAM,OAAO,eAAe;IAChB,KAAK,CAAI,GAAM;QACvB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,gBAAgB,CACrB,MAAM,EACN,MAAM,CAAC,yBAAyB,CAAC,GAAG,CAAC,CACtC,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAES,eAAe,CAAC,KAAc,EAAE,MAA6B;QACrE,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;YACzD,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;;;;OAQG;IACI,eAAe,CACpB,WAAmB,EACnB,MAA6B,EAC7B,aAA6C,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;QAEvD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,MAAM,OAAO,GAAG,CAAC,SAAiB,EAAW,EAAE;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACnC,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YACjC,WAAW,GAAG,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC;YACxC,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;YAC3C,IAAI,SAAS,IAAI,CAAC,WAAW,EAAE,CAAC;gBAC9B,KAAK,MAAM,CAAE,GAAG,EAAE,KAAK,CAAE,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;oBAClD,IAAI,WAAW,EAAE,CAAC;wBAChB,OAAO,IAAI,CAAC;oBACd,CAAC;oBAC0B,IAAK,CAAC,GAAG,CAAC;wBACnC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC,CAAC;QAEF,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACI,WAAW,CAChB,WAAmB,EACnB,OAA+B,EAC/B,aAA6C,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;QAEvD,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,MAAM,OAAO,GAAG,CAAC,SAAiB,EAAQ,EAAE;YAC1C,MAAM,OAAO,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;YACtC,WAAW,GAAG,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC;YACxC,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;YAC3C,IAAI,SAAS,IAAI,CAAC,WAAW,EAAE,CAAC;gBAC9B,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC7C,IAAI,WAAW,EAAE,CAAC;wBAChB,OAAO;oBACT,CAAC;oBACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;YACD,OAAO,CAAC,SAAS,CAAC,CAAC;QACrB,CAAC,CAAC;QACF,OAAO,CAAC,WAAW,CAAC,CAAC;IACvB,CAAC;IAEM,aAAa,CAClB,WAAmB,EACnB,aAGE;QAEF,MAAM,gBAAgB,GAAG,CAAC,SAAiB,EAAW,EAAE;YACtD,MAAM,MAAM,GAA6B,SAAS,CAAC;YACnD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAChB,MAAM,MAAM,GAAyC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC;gBAC3F,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC;QACF,MAAM,mBAAmB,GAAG,CAAC,SAAiB,EAAgB,EAAE;YAC9D,MAAM,MAAM,GAA6B,SAAS,CAAC;YACnD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAChB,MAAM,MAAM,GAA8C,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;gBACjG,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,OAAO,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;IAClF,CAAC;IAEM,SAAS,CACd,WAAmB,EACnB,aAGE;QAEF,MAAM,YAAY,GAAG,CAAC,SAAiB,EAAQ,EAAE;YAC/C,MAAM,MAAM,GAA6B,SAAS,CAAC;YACnD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAChB,MAAM,MAAM,GAAyC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;gBACzF,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,CAAC,MAAM,CAAC,CAAC;gBACjB,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QACF,MAAM,eAAe,GAAG,CAAC,SAAiB,EAAgB,EAAE;YAC1D,MAAM,MAAM,GAA6B,SAAS,CAAC;YACnD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAChB,MAAM,MAAM,GAA8C,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;gBACjG,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,YAAY,EAAE,eAAe,CAAC,CAAC;IAC/D,CAAC;IAEM,aAAa,CAClB,WAAkB,EAClB,QAAyG;QAEzG,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,MAAM,OAAO,GAAG,CAAC,OAAc,EAAQ,EAAE;YACvC,MAAM,SAAS,GAAG,QAAQ,CAAgB,OAAO,CAAC,IAAI,CAAC,CAAC;YACxD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAM,OAAO,CAAC,CAAC;gBACnD,WAAW,GAAG,QAAQ,IAAI,KAAK,CAAC;gBAChC,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,KAAK,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,EAAE,CAAC;wBAC9B,IAAI,WAAW,EAAE,CAAC;4BAChB,OAAO;wBACT,CAAC;wBACD,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,CAAC,WAAW,CAAC,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,kBAAiE,SAAQ,eAAsB;IACnG,qBAAqB,CAC1B,WAAmB,EACnB,aAGE,EACF,qBAIK;QAEL,MAAM,gBAAgB,GAAG,CAAC,SAAiB,EAAW,EAAE;YACtD,IAAI,WAAiD,CAAC;YACtD,MAAM,MAAM,GAA+C,SAAS,CAAC;YACrE,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,WAAW,GAAG,QAAQ,CAAyB,MAAM,CAAC,OAAO,CAAC,EAAE,SAAS,CAAC;gBAC5E,CAAC;gBACD,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,WAAW,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC;gBACtD,CAAC;YACH,CAAC;YACD,OAAO,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACvD,CAAC,CAAC;QACF,MAAM,eAAe,GAAG,CAAC,SAAiB,EAAgB,EAAE;YAC1D,IAAI,UAAqD,CAAC;YAC1D,MAAM,MAAM,GAA+C,SAAS,CAAC;YACrE,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,UAAU,GAAG,QAAQ,CAAyB,MAAM,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC;gBAC5E,CAAC;gBACD,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;gBACtD,CAAC;YACH,CAAC;YACD,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrD,CAAC,CAAC;QACF,OAAO,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,gBAAgB,EAAE,eAAe,CAAC,CAAC;IAC9E,CAAC;IAED;;OAEG;IACI,iBAAiB,CACtB,WAAmB,EACnB,aAGE,EACF,qBAIK;QAEL,MAAM,YAAY,GAAG,CAAC,SAAiB,EAAQ,EAAE;YAC/C,IAAI,WAA8C,CAAC;YACnD,MAAM,MAAM,GAA+C,SAAS,CAAC;YACrE,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,WAAW,GAAG,QAAQ,CAAyB,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC;gBAC1E,CAAC;gBACD,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,WAAW,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;gBACpD,CAAC;YACH,CAAC;YACD,IAAI,WAAW,EAAE,CAAC;gBAChB,WAAW,CAAC,MAAM,CAAC,CAAC;YACtB,CAAC;QACH,CAAC,CAAC;QACF,MAAM,eAAe,GAAG,CAAC,SAAiB,EAAgB,EAAE;YAC1D,IAAI,UAAqD,CAAC;YAC1D,MAAM,MAAM,GAA+C,SAAS,CAAC;YACrE,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,UAAU,GAAG,QAAQ,CAAyB,MAAM,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC;gBAC5E,CAAC;gBACD,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;gBACtD,CAAC;YACH,CAAC;YACD,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrD,CAAC,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,YAAY,EAAE,eAAe,CAAC,CAAC;IAC/D,CAAC;IAEM,gBAAgB,CACrB,WAAkB,EAClB,YAC4E,EAC5E,eAEiG;QAEjG,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,MAAM,OAAO,GAAG,CAAC,OAAc,EAAQ,EAAE;YACvC,IAAI,SAAwE,CAAC;YAC7E,MAAM,MAAM,GAAG,eAAe,CAAgB,OAAO,CAAC,IAAI,CAAC,CAAC;YAC5D,IAAI,MAAM,EAAE,CAAC;gBACX,SAAS,GAAG,MAAM,CAAsB,OAAO,CAAC,OAAO,CAAC,CAAC;YAC3D,CAAC;YACD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS,GAAG,YAAY,CAAgB,OAAO,CAAC,IAAI,CAAC,CAAC;YACxD,CAAC;YACD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAM,OAAO,CAAC,CAAC;gBACnD,WAAW,GAAG,QAAQ,IAAI,KAAK,CAAC;gBAChC,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,KAAK,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,EAAE,CAAC;wBAC9B,IAAI,WAAW,EAAE,CAAC;4BAChB,OAAO;wBACT,CAAC;wBACD,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,CAAC,WAAW,CAAC,CAAC;IACvB,CAAC;CACF","sourcesContent":["import type { Node } from './nodeTypings.js';\n\ntype Safeness = 'safe' | 'unsafe';\ntype SafeWrap<Safe extends Safeness, obj extends object> = Safe extends 'safe' ? {[key in keyof obj]: unknown } : obj;\n\nexport interface VisitContext {\n  shortcut?: boolean;\n  continue?: boolean;\n}\n\nexport interface SelectiveTraversalContext<Nodes> {\n  next?: Nodes[];\n  shortcut?: boolean;\n}\n\nexport class TransformerType<Nodes extends Pick<Node, 'type'>> {\n  protected clone<T>(obj: T): T {\n    const newObj = Object.create(Object.getPrototypeOf(obj));\n    Object.defineProperties(\n      newObj,\n      Object.getOwnPropertyDescriptors(obj),\n    );\n    return newObj;\n  }\n\n  protected safeObjectVisit(value: unknown, mapper: (some: object) => any): unknown {\n    if (value && typeof value === 'object') {\n      if (Array.isArray(value)) {\n        return value.map(x => this.safeObjectVisit(x, mapper));\n      }\n      return mapper(value);\n    }\n    return value;\n  }\n\n  /**\n   * Recursively transforms all objects that are not arrays. Mapper is called on deeper objects first.\n   * @param startObject object to start iterating from\n   * @param mapper mapper to transform the various objects - argument is a copy of the original\n   * @param preVisitor callback that is evaluated before iterating deeper.\n   *   If continues is false, we do not iterate deeper, current object is still mapped. - default: true\n   *   If shortcut is true, we do not iterate deeper, nor do we branch out, this mapper will be the last one called.\n   *    - Default false\n   */\n  public transformObject(\n    startObject: object,\n    mapper: (some: object) => any,\n    preVisitor: (some: object) => VisitContext = () => ({}),\n  ): unknown {\n    let didShortCut = false;\n    const recurse = (curObject: object): unknown => {\n      const copy = this.clone(curObject);\n      const context = preVisitor(copy);\n      didShortCut = context.shortcut ?? false;\n      const continues = context.continue ?? true;\n      if (continues && !didShortCut) {\n        for (const [ key, value ] of Object.entries(copy)) {\n          if (didShortCut) {\n            return copy;\n          }\n          (<Record<string, unknown>> copy)[key] =\n            this.safeObjectVisit(value, obj => recurse(obj));\n        }\n      }\n      return mapper(copy);\n    };\n\n    return recurse(startObject);\n  }\n\n  /**\n   * Visitor that visits all objects. Visits deeper objects first.\n   */\n  public visitObject(\n    startObject: object,\n    visitor: (some: object) => void,\n    preVisitor: (some: object) => VisitContext = () => ({}),\n  ): void {\n    let didShortCut = false;\n\n    const recurse = (curObject: object): void => {\n      const context = preVisitor(curObject);\n      didShortCut = context.shortcut ?? false;\n      const continues = context.continue ?? true;\n      if (continues && !didShortCut) {\n        for (const value of Object.values(curObject)) {\n          if (didShortCut) {\n            return;\n          }\n          this.safeObjectVisit(value, obj => recurse(obj));\n        }\n      }\n      visitor(curObject);\n    };\n    recurse(startObject);\n  }\n\n  public transformNode<Safe extends 'safe' | 'unsafe' = 'safe'>(\n    startObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: {\n      transform?: (op: SafeWrap<Safe, Extract<Nodes, { type: T }>>) => any;\n      preVisitor?: (op: Extract<Nodes, { type: T }>) => VisitContext;\n    }},\n  ): any {\n    const transformWrapper = (curObject: object): unknown => {\n      const casted = <{ type?: Nodes['type'] }>curObject;\n      if (casted.type) {\n        const ogFunc: ((node: any) => unknown) | undefined = nodeCallBacks[casted.type]?.transform;\n        if (ogFunc) {\n          return ogFunc(casted);\n        }\n      }\n      return curObject;\n    };\n    const preTransformWrapper = (curObject: object): VisitContext => {\n      const casted = <{ type?: Nodes['type'] }>curObject;\n      if (casted.type) {\n        const ogFunc: ((node: any) => VisitContext) | undefined = nodeCallBacks[casted.type]?.preVisitor;\n        if (ogFunc) {\n          return ogFunc(casted);\n        }\n      }\n      return {};\n    };\n    return this.transformObject(startObject, transformWrapper, preTransformWrapper);\n  }\n\n  public visitNode(\n    startObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: {\n      visitor?: (op: Extract<Nodes, { type: T }>) => void;\n      preVisitor?: (op: Extract<Nodes, { type: T }>) => VisitContext;\n    }},\n  ): void {\n    const visitWrapper = (curObject: object): void => {\n      const casted = <{ type?: Nodes['type'] }>curObject;\n      if (casted.type) {\n        const ogFunc: ((node: any) => unknown) | undefined = nodeCallBacks[casted.type]?.visitor;\n        if (ogFunc) {\n          ogFunc(casted);\n        }\n      }\n    };\n    const preVisitWrapper = (curObject: object): VisitContext => {\n      const casted = <{ type?: Nodes['type'] }>curObject;\n      if (casted.type) {\n        const ogFunc: ((node: any) => VisitContext) | undefined = nodeCallBacks[casted.type]?.preVisitor;\n        if (ogFunc) {\n          return ogFunc(casted);\n        }\n      }\n      return {};\n    };\n    this.visitObject(startObject, visitWrapper, preVisitWrapper);\n  }\n\n  public traverseNodes(\n    currentNode: Nodes,\n    traverse: {[T in Nodes['type']]?: (op: Extract<Nodes, { type: T }>) => SelectiveTraversalContext<Nodes> },\n  ): void {\n    let didShortCut = false;\n\n    const recurse = (curNode: Nodes): void => {\n      const traverser = traverse[<Nodes['type']>curNode.type];\n      if (traverser) {\n        const { next, shortcut } = traverser(<any>curNode);\n        didShortCut = shortcut ?? false;\n        if (!didShortCut) {\n          for (const node of next ?? []) {\n            if (didShortCut) {\n              return;\n            }\n            recurse(node);\n          }\n        }\n      }\n    };\n\n    recurse(currentNode);\n  }\n}\n\nexport class TransformerSubType<Nodes extends Pick<Node, 'type' | 'subType'>> extends TransformerType<Nodes> {\n  public transformNodeSpecific<Safe extends 'safe' | 'unsafe' = 'safe'>(\n    startObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: {\n      transform?: (op: SafeWrap<Safe, Extract<Nodes, { type: T }>>) => any;\n      preVisitor?: (op: Extract<Nodes, { type: T }>) => VisitContext;\n    }},\n    nodeSpecificCallBacks: {[Type in Nodes['type']]?: {\n      [SubType in Extract<Nodes, { type: Type; subType: string }>['subType']]?: {\n        transform?: (op: SafeWrap<Safe, Extract<Nodes, { type: Type; subType: SubType }>>) => any;\n        preVisitor?: (op: Extract<Nodes, { type: Type; subType: SubType }>) => VisitContext;\n      }}},\n  ): any {\n    const transformWrapper = (curObject: object): unknown => {\n      let ogTransform: ((node: any) => unknown) | undefined;\n      const casted = <{ type?: Nodes['type']; subType?: string }>curObject;\n      if (casted.type && casted.subType) {\n        const specific = nodeSpecificCallBacks[casted.type];\n        if (specific) {\n          ogTransform = specific[<keyof typeof specific> casted.subType]?.transform;\n        }\n        if (!ogTransform) {\n          ogTransform = nodeCallBacks[casted.type]?.transform;\n        }\n      }\n      return ogTransform ? ogTransform(casted) : curObject;\n    };\n    const preVisitWrapper = (curObject: object): VisitContext => {\n      let ogPreVisit: ((node: any) => VisitContext) | undefined;\n      const casted = <{ type?: Nodes['type']; subType?: string }>curObject;\n      if (casted.type && casted.subType) {\n        const specific = nodeSpecificCallBacks[casted.type];\n        if (specific) {\n          ogPreVisit = specific[<keyof typeof specific> casted.subType]?.preVisitor;\n        }\n        if (!ogPreVisit) {\n          ogPreVisit = nodeCallBacks[casted.type]?.preVisitor;\n        }\n      }\n      return ogPreVisit ? ogPreVisit(casted) : curObject;\n    };\n    return this.transformObject(startObject, transformWrapper, preVisitWrapper);\n  }\n\n  /**\n   * When both nodeCallBack and NodeSpecific callBack are matched, will only look at nodeSpecifCallback\n   */\n  public visitNodeSpecific(\n    startObject: object,\n    nodeCallBacks: {[T in Nodes['type']]?: {\n      visitor?: (op: Extract<Nodes, { type: T }>) => void;\n      preVisitor?: (op: Extract<Nodes, { type: T }>) => VisitContext;\n    }},\n    nodeSpecificCallBacks: {[Type in Nodes['type']]?:\n      {[Subtype in Extract<Nodes, { type: Type; subType: string }>['subType']]?: {\n        visitor?: (op: Extract<Nodes, { type: Type; subType: Subtype }>) => void;\n        preVisitor?: (op: Extract<Nodes, { type: Type; subType: Subtype }>) => VisitContext;\n      }}},\n  ): void {\n    const visitWrapper = (curObject: object): void => {\n      let ogTransform: ((node: any) => void) | undefined;\n      const casted = <{ type?: Nodes['type']; subType?: string }>curObject;\n      if (casted.type && casted.subType) {\n        const specific = nodeSpecificCallBacks[casted.type];\n        if (specific) {\n          ogTransform = specific[<keyof typeof specific> casted.subType]?.visitor;\n        }\n        if (!ogTransform) {\n          ogTransform = nodeCallBacks[casted.type]?.visitor;\n        }\n      }\n      if (ogTransform) {\n        ogTransform(casted);\n      }\n    };\n    const preVisitWrapper = (curObject: object): VisitContext => {\n      let ogPreVisit: ((node: any) => VisitContext) | undefined;\n      const casted = <{ type?: Nodes['type']; subType?: string }>curObject;\n      if (casted.type && casted.subType) {\n        const specific = nodeSpecificCallBacks[casted.type];\n        if (specific) {\n          ogPreVisit = specific[<keyof typeof specific> casted.subType]?.preVisitor;\n        }\n        if (!ogPreVisit) {\n          ogPreVisit = nodeCallBacks[casted.type]?.preVisitor;\n        }\n      }\n      return ogPreVisit ? ogPreVisit(casted) : curObject;\n    };\n    this.visitObject(startObject, visitWrapper, preVisitWrapper);\n  }\n\n  public traverseSubNodes(\n    currentNode: Nodes,\n    traverseNode: {[Type in Nodes['type']]?:\n      (op: Extract<Nodes, { type: Type }>) => SelectiveTraversalContext<Nodes> },\n    traverseSubNode: {[Type in Nodes['type']]?:\n      {[Subtype in Extract<Nodes, { type: Type; subType: string }>['subType']]?:\n        (op: Extract<Nodes, { type: Type; subType: Subtype }>) => SelectiveTraversalContext<Nodes> }},\n  ): void {\n    let didShortCut = false;\n\n    const recurse = (curNode: Nodes): void => {\n      let traverser: ((call: any) => SelectiveTraversalContext<Nodes>) | undefined;\n      const subObj = traverseSubNode[<Nodes['type']>curNode.type];\n      if (subObj) {\n        traverser = subObj[<keyof typeof subObj>curNode.subType];\n      }\n      if (!traverser) {\n        traverser = traverseNode[<Nodes['type']>curNode.type];\n      }\n      if (traverser) {\n        const { next, shortcut } = traverser(<any>curNode);\n        didShortCut = shortcut ?? false;\n        if (!didShortCut) {\n          for (const node of next ?? []) {\n            if (didShortCut) {\n              return;\n            }\n            recurse(node);\n          }\n        }\n      }\n    };\n\n    recurse(currentNode);\n  }\n}\n"]}

package/lib/generator-builder/dynamicGenerator.d.ts

@@ -1,9 +1,10 @@
-import { CoreFactory } from '../CoreFactory.js';
+import { AstCoreFactory } from '../AstCoreFactory.js';
+import { traqulaIndentation } from '../utils.js';
 import type { GenRuleMap } from './builderTypes.js';
 import type { RuleDefArg } from './generatorTypes.js';
 export declare class DynamicGenerator<Context, Names extends string, RuleDefs extends GenRuleMap<Names>> {
     protected rules: RuleDefs;
-    protected readonly factory: CoreFactory;
+    protected readonly factory: AstCoreFactory;
     protected __context: Context | undefined;
     protected origSource: string;
     protected generatedUntil: number;
@@ -11,7 +12,9 @@ export declare class DynamicGenerator<Context, Names extends string, RuleDefs ex
     protected readonly stringBuilder: string[];
     constructor(rules: RuleDefs);
     setContext(context: Context): void;
-    protected getSafeContext(): Context;
+    protected getSafeContext(): Context & {
+        [traqulaIndentation]?: unknown;
+    };
     protected readonly subrule: RuleDefArg['SUBRULE'];
     protected readonly handleLoc: RuleDefArg['HANDLE_LOC'];
     protected readonly catchup: RuleDefArg['CATCHUP'];
@@ -19,4 +22,5 @@ export declare class DynamicGenerator<Context, Names extends string, RuleDefs ex
     private readonly printWord;
     private readonly printSpaceLeft;
     private readonly printWords;
+    private readonly printOnEmpty;
 }