@elastic/esql 4.0.0 → 4.1.0

package/lib/ast/walker/walker.d.ts

@@ -50,7 +50,7 @@ export interface WalkerOptions {
     skipHeader?: boolean;
 }
 export type WalkerAstNode = types.ESQLAstNode | types.ESQLAstNode[];
-export type WalkerVisitorApi = Pick<Walker, 'abort'>;
+export type WalkerVisitorApi = Pick<Walker, 'abort' | 'skipChildren'>;
 /**
  * Iterates over all nodes in the AST and calls the appropriate visitor
  * functions.
@@ -237,8 +237,11 @@ export declare class Walker {
      */
     static visitAny: (tree: WalkerAstNode, visitAny: WalkerOptions["visitAny"], options?: WalkerOptions) => void;
     protected aborted: boolean;
+    protected skippedChildren: boolean;
     constructor(options: WalkerOptions);
     abort(): void;
+    skipChildren(): void;
+    protected readAndResetSkippedChildren(): boolean;
     walk(tree: WalkerAstNode | undefined, parent?: types.ESQLProperNode | undefined): void;
     protected walkList(list: types.ESQLAstNode[], parent: types.ESQLProperNode | undefined): void;
     walkCommand(node: types.ESQLAstCommand, parent: types.ESQLAstQueryExpression | undefined): void;

package/lib/ast/walker/walker.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"walker.d.ts","sourceRoot":"","sources":["../../../src/ast/walker/walker.ts"],"names":[],"mappings":"AASA,OAAO,EAAgB,KAAK,mBAAmB,EAAE,MAAM,4CAA4C,CAAC;AACpG,OAAO,KAAK,KAAK,KAAK,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,KAAK,MAAM,MAAM,uCAAuC,CAAC;AACrE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAEnD,KAAK,IAAI,GAAG,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;AAEpD,MAAM,WAAW,aAAa;IAG5B,MAAM,CAAC,EAAE,mBAAmB,CAAC;IAI7B,YAAY,CAAC,EAAE,CACb,IAAI,EAAE,KAAK,CAAC,WAAW,EACvB,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,EAChD,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,kBAAkB,CAAC,EAAE,CACnB,IAAI,EAAE,KAAK,CAAC,oBAAoB,EAChC,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,EAChD,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,kBAAkB,CAAC,EAAE,CACnB,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAC7B,MAAM,EAAE,KAAK,CAAC,WAAW,GAAG,SAAS,EACrC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,UAAU,CAAC,EAAE,CACX,IAAI,EAAE,KAAK,CAAC,sBAAsB,EAClC,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,aAAa,CAAC,EAAE,CACd,IAAI,EAAE,KAAK,CAAC,YAAY,EACxB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,KAAK,CAAC,UAAU,EACtB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,KAAK,CAAC,UAAU,EACtB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,UAAU,CAAC,EAAE,CACX,IAAI,EAAE,KAAK,CAAC,mBAAmB,EAC/B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,YAAY,CAAC,EAAE,CACb,IAAI,EAAE,KAAK,CAAC,WAAW,EACvB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,gBAAgB,CAAC,EAAE,CACjB,IAAI,EAAE,KAAK,CAAC,QAAQ,EACpB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,eAAe,CAAC,EAAE,CAChB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,YAAY,CAAC,EAAE,CACb,IAAI,EAAE,KAAK,CAAC,eAAe,EAC3B,OAAO,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACzC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,eAAe,CAAC,EAAE,CAChB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,QAAQ,CAAC,EAAE,CACT,IAAI,EAAE,KAAK,CAAC,OAAO,EACnB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,aAAa,CAAC,EAAE,CACd,IAAI,EAAE,KAAK,CAAC,YAAY,EACxB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,KAAK,CAAC,UAAU,EACtB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IAIV;;;;OAIG;IACH,kBAAkB,CAAC,EAAE,CACnB,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAC7B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IAEV;;;;OAIG;IACH,QAAQ,CAAC,EAAE,CACT,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IAEV;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC;IAE/B;;;;;OAKG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,MAAM,aAAa,GAAG,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;AAEpE,MAAM,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAErD;;;GAGG;AACH,qBAAa,MAAM;IAoYL,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,aAAa;IAnYrD;;OAEG;IACH,gBAAuB,IAAI,GAAI,MAAM,aAAa,EAAE,SAAS,aAAa,KAAG,MAAM,CAIjF;IAEF;;;;;;OAMG;IACH,gBAAuB,IAAI,GACzB,MAAM,aAAa,EACnB,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,EAClD,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,GAAG,SAAS,CAYjC;IAEF;;;;;;OAMG;IACH,gBAAuB,OAAO,GAC5B,MAAM,aAAa,EACnB,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,EAClD,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,EAAE,CAWvB;IAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAyCG;IACH,gBAAuB,KAAK,GAC1B,MAAM,aAAa,EACnB,UAAU,iBAAiB,EAC3B,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,GAAG,SAAS,CAGjC;IAEF;;;;;;;OAOG;IACH,gBAAuB,QAAQ,GAC7B,MAAM,aAAa,EACnB,UAAU,iBAAiB,EAC3B,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,EAAE,CAGvB;IAEF;;;;;;;;;;;;;;;;;;OAkBG;IACH,gBAAuB,OAAO,GAC5B,MAAM,aAAa,EACnB,SAAS,iBAAiB,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,CAAC,EACtE,UAAU,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAAC,KACtF,KAAK,CAAC,cAAc,GAAG,SAAS,CAOjC;IAEF;;;;;;;;OAQG;IACH,gBAAuB,UAAU,GAC/B,MAAM,aAAa,EACnB,SAAS,iBAAiB,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,CAAC,EACtE,UAAU,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAAC,KACtF,KAAK,CAAC,cAAc,EAAE,CAWvB;IAEF;;;;OAIG;IACH,gBAAuB,QAAQ,GAAI,MAAM,IAAI,EAAE,UAAU,aAAa,KAAG,KAAK,CAAC,WAAW,EAAE,CAE1F;IAEF;;;;OAIG;IACH,gBAAuB,MAAM,GAC3B,MAAM,aAAa,EACnB,UAAU,aAAa,KACtB,KAAK,CAAC,gBAAgB,EAAE,CASzB;IAEF;;;;;;;OAOG;IACH,gBAAuB,YAAY,GACjC,MAAM,aAAa,EACnB,iBAAiB,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,YAAY,KAAK,OAAO,CAAC,GAAG,MAAM,KAChE,KAAK,CAAC,YAAY,GAAG,SAAS,CAe/B;IAEF;;;;;;OAMG;IACH,gBAAuB,WAAW,GAAI,MAAM,aAAa,EAAE,MAAM,MAAM,KAAG,OAAO,CAE/E;IAEF;;;;;;;;;;;;;;;OAeG;IACH,gBAAuB,MAAM,GAC3B,MAAM,aAAa,EACnB,OAAO,KAAK,CAAC,cAAc,KAC1B,KAAK,CAAC,cAAc,GAAG,SAAS,CAWjC;IAEF;;;;;;;;;;OAUG;IACH,gBAAuB,OAAO,GAC5B,MAAM,aAAa,EACnB,OAAO,KAAK,CAAC,cAAc,KAC1B,KAAK,CAAC,cAAc,EAAE,CASvB;IAEF;;;;;;;;;OASG;IACH,gBAAuB,aAAa,GAClC,MAAM,aAAa,EACnB,UAAU,CACR,OAAO,EAAE,KAAK,CAAC,cAAc,EAC7B,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,UAAU,EAAE,MAAM,KAAK,CAAC,qBAAqB,KAC1C,IAAI,KACR,IAAI,CA2CL;IAEF;;;;;;OAMG;IACH,OAAc,QAAQ,GACpB,MAAM,aAAa,EACnB,UAAU,aAAa,CAAC,UAAU,CAAC,EACnC,UAAU,aAAa,KACtB,IAAI,CAEL;IAEF,SAAS,CAAC,OAAO,EAAE,OAAO,CAAS;gBAEJ,OAAO,EAAE,aAAa;IAE9C,KAAK,IAAI,IAAI;IAIb,IAAI,CACT,IAAI,EAAE,aAAa,GAAG,SAAS,EAC/B,MAAM,GAAE,KAAK,CAAC,cAAc,GAAG,SAAqB,GACnD,IAAI;IAoBP,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAiBtF,WAAW,CAChB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,GAC/C,IAAI;IAQA,iBAAiB,CACtB,IAAI,EAAE,KAAK,CAAC,oBAAoB,EAChC,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,GAC/C,IAAI;IAQA,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAAE,MAAM,EAAE,KAAK,CAAC,WAAW,GAAG,SAAS,GAAG,IAAI;IAMtF,cAAc,CACnB,IAAI,EAAE,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,iBAAiB,EACjD,MAAM,GAAE,KAAK,CAAC,cAAc,GAAG,SAAqB,GACnD,IAAI;IAUA,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAMrF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAoBlF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAWlF,SAAS,CACd,IAAI,EAAE,KAAK,CAAC,mBAAmB,EAC/B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAQA,cAAc,CACnB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAMA,YAAY,CACjB,IAAI,EAAE,KAAK,CAAC,YAAY,EACxB,MAAM,GAAE,KAAK,CAAC,cAAc,GAAG,SAAqB,GACnD,IAAI;IASA,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAM5E,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IActF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IASlF,SAAS,CACd,IAAI,EAAE,KAAK,CAAC,sBAAsB,EAClC,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAWP;;OAEG;IACI,cAAc,CACnB,IAAI,EAAE,MAAM,CAAC,aAAa,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAkBA,iBAAiB,CACtB,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAC7B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;CA0ER;AAED,eAAO,MAAM,IAAI,SA9qBsB,aAAa,WAAW,aAAa,KAAG,MA8qBhD,CAAC"}
+{"version":3,"file":"walker.d.ts","sourceRoot":"","sources":["../../../src/ast/walker/walker.ts"],"names":[],"mappings":"AASA,OAAO,EAAgB,KAAK,mBAAmB,EAAE,MAAM,4CAA4C,CAAC;AACpG,OAAO,KAAK,KAAK,KAAK,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,KAAK,MAAM,MAAM,uCAAuC,CAAC;AACrE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAEnD,KAAK,IAAI,GAAG,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;AAEpD,MAAM,WAAW,aAAa;IAG5B,MAAM,CAAC,EAAE,mBAAmB,CAAC;IAI7B,YAAY,CAAC,EAAE,CACb,IAAI,EAAE,KAAK,CAAC,WAAW,EACvB,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,EAChD,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,kBAAkB,CAAC,EAAE,CACnB,IAAI,EAAE,KAAK,CAAC,oBAAoB,EAChC,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,EAChD,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,kBAAkB,CAAC,EAAE,CACnB,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAC7B,MAAM,EAAE,KAAK,CAAC,WAAW,GAAG,SAAS,EACrC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,UAAU,CAAC,EAAE,CACX,IAAI,EAAE,KAAK,CAAC,sBAAsB,EAClC,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,aAAa,CAAC,EAAE,CACd,IAAI,EAAE,KAAK,CAAC,YAAY,EACxB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,KAAK,CAAC,UAAU,EACtB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,KAAK,CAAC,UAAU,EACtB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,UAAU,CAAC,EAAE,CACX,IAAI,EAAE,KAAK,CAAC,mBAAmB,EAC/B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,YAAY,CAAC,EAAE,CACb,IAAI,EAAE,KAAK,CAAC,WAAW,EACvB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,gBAAgB,CAAC,EAAE,CACjB,IAAI,EAAE,KAAK,CAAC,QAAQ,EACpB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,eAAe,CAAC,EAAE,CAChB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,YAAY,CAAC,EAAE,CACb,IAAI,EAAE,KAAK,CAAC,eAAe,EAC3B,OAAO,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACzC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,eAAe,CAAC,EAAE,CAChB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,QAAQ,CAAC,EAAE,CACT,IAAI,EAAE,KAAK,CAAC,OAAO,EACnB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,aAAa,CAAC,EAAE,CACd,IAAI,EAAE,KAAK,CAAC,YAAY,EACxB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IACV,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,KAAK,CAAC,UAAU,EACtB,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IAIV;;;;OAIG;IACH,kBAAkB,CAAC,EAAE,CACnB,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAC7B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IAEV;;;;OAIG;IACH,QAAQ,CAAC,EAAE,CACT,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,EACxC,MAAM,EAAE,gBAAgB,KACrB,IAAI,CAAC;IAEV;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC;IAE/B;;;;;OAKG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,MAAM,aAAa,GAAG,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;AAEpE,MAAM,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,EAAE,OAAO,GAAG,cAAc,CAAC,CAAC;AAEtE;;;GAGG;AACH,qBAAa,MAAM;IAqYL,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,aAAa;IApYrD;;OAEG;IACH,gBAAuB,IAAI,GAAI,MAAM,aAAa,EAAE,SAAS,aAAa,KAAG,MAAM,CAIjF;IAEF;;;;;;OAMG;IACH,gBAAuB,IAAI,GACzB,MAAM,aAAa,EACnB,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,EAClD,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,GAAG,SAAS,CAYjC;IAEF;;;;;;OAMG;IACH,gBAAuB,OAAO,GAC5B,MAAM,aAAa,EACnB,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,EAClD,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,EAAE,CAWvB;IAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAyCG;IACH,gBAAuB,KAAK,GAC1B,MAAM,aAAa,EACnB,UAAU,iBAAiB,EAC3B,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,GAAG,SAAS,CAGjC;IAEF;;;;;;;OAOG;IACH,gBAAuB,QAAQ,GAC7B,MAAM,aAAa,EACnB,UAAU,iBAAiB,EAC3B,UAAU,aAAa,KACtB,KAAK,CAAC,cAAc,EAAE,CAGvB;IAEF;;;;;;;;;;;;;;;;;;OAkBG;IACH,gBAAuB,OAAO,GAC5B,MAAM,aAAa,EACnB,SAAS,iBAAiB,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,CAAC,EACtE,UAAU,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAAC,KACtF,KAAK,CAAC,cAAc,GAAG,SAAS,CAOjC;IAEF;;;;;;;;OAQG;IACH,gBAAuB,UAAU,GAC/B,MAAM,aAAa,EACnB,SAAS,iBAAiB,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,OAAO,CAAC,EACtE,UAAU,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAAC,KACtF,KAAK,CAAC,cAAc,EAAE,CAWvB;IAEF;;;;OAIG;IACH,gBAAuB,QAAQ,GAAI,MAAM,IAAI,EAAE,UAAU,aAAa,KAAG,KAAK,CAAC,WAAW,EAAE,CAE1F;IAEF;;;;OAIG;IACH,gBAAuB,MAAM,GAC3B,MAAM,aAAa,EACnB,UAAU,aAAa,KACtB,KAAK,CAAC,gBAAgB,EAAE,CASzB;IAEF;;;;;;;OAOG;IACH,gBAAuB,YAAY,GACjC,MAAM,aAAa,EACnB,iBAAiB,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,YAAY,KAAK,OAAO,CAAC,GAAG,MAAM,KAChE,KAAK,CAAC,YAAY,GAAG,SAAS,CAe/B;IAEF;;;;;;OAMG;IACH,gBAAuB,WAAW,GAAI,MAAM,aAAa,EAAE,MAAM,MAAM,KAAG,OAAO,CAE/E;IAEF;;;;;;;;;;;;;;;OAeG;IACH,gBAAuB,MAAM,GAC3B,MAAM,aAAa,EACnB,OAAO,KAAK,CAAC,cAAc,KAC1B,KAAK,CAAC,cAAc,GAAG,SAAS,CAWjC;IAEF;;;;;;;;;;OAUG;IACH,gBAAuB,OAAO,GAC5B,MAAM,aAAa,EACnB,OAAO,KAAK,CAAC,cAAc,KAC1B,KAAK,CAAC,cAAc,EAAE,CASvB;IAEF;;;;;;;;;OASG;IACH,gBAAuB,aAAa,GAClC,MAAM,aAAa,EACnB,UAAU,CACR,OAAO,EAAE,KAAK,CAAC,cAAc,EAC7B,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,UAAU,EAAE,MAAM,KAAK,CAAC,qBAAqB,KAC1C,IAAI,KACR,IAAI,CA2CL;IAEF;;;;;;OAMG;IACH,OAAc,QAAQ,GACpB,MAAM,aAAa,EACnB,UAAU,aAAa,CAAC,UAAU,CAAC,EACnC,UAAU,aAAa,KACtB,IAAI,CAEL;IAEF,SAAS,CAAC,OAAO,EAAE,OAAO,CAAS;IACnC,SAAS,CAAC,eAAe,EAAE,OAAO,CAAS;gBAEZ,OAAO,EAAE,aAAa;IAE9C,KAAK,IAAI,IAAI;IAIb,YAAY,IAAI,IAAI;IAI3B,SAAS,CAAC,2BAA2B,IAAI,OAAO;IAMzC,IAAI,CACT,IAAI,EAAE,aAAa,GAAG,SAAS,EAC/B,MAAM,GAAE,KAAK,CAAC,cAAc,GAAG,SAAqB,GACnD,IAAI;IAoBP,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAiBtF,WAAW,CAChB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,GAC/C,IAAI;IAUA,iBAAiB,CACtB,IAAI,EAAE,KAAK,CAAC,oBAAoB,EAChC,MAAM,EAAE,KAAK,CAAC,sBAAsB,GAAG,SAAS,GAC/C,IAAI;IAUA,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAAE,MAAM,EAAE,KAAK,CAAC,WAAW,GAAG,SAAS,GAAG,IAAI;IAQtF,cAAc,CACnB,IAAI,EAAE,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,iBAAiB,EACjD,MAAM,GAAE,KAAK,CAAC,cAAc,GAAG,SAAqB,GACnD,IAAI;IAUA,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAQrF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAqBlF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAYlF,SAAS,CACd,IAAI,EAAE,KAAK,CAAC,mBAAmB,EAC/B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IASA,cAAc,CACnB,IAAI,EAAE,KAAK,CAAC,cAAc,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAQA,YAAY,CACjB,IAAI,EAAE,KAAK,CAAC,YAAY,EACxB,MAAM,GAAE,KAAK,CAAC,cAAc,GAAG,SAAqB,GACnD,IAAI;IAUA,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAQ5E,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAetF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GAAG,IAAI;IAUlF,SAAS,CACd,IAAI,EAAE,KAAK,CAAC,sBAAsB,EAClC,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAYP;;OAEG;IACI,cAAc,CACnB,IAAI,EAAE,MAAM,CAAC,aAAa,EAC1B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;IAkBA,iBAAiB,CACtB,IAAI,EAAE,KAAK,CAAC,iBAAiB,EAC7B,MAAM,EAAE,KAAK,CAAC,cAAc,GAAG,SAAS,GACvC,IAAI;CA6ER;AAED,eAAO,MAAM,IAAI,SA/sBsB,aAAa,WAAW,aAAa,KAAG,MA+sBhD,CAAC"}

package/lib/embedded_languages/promql/ast/find_node.d.ts

@@ -0,0 +1,24 @@
+import type { PromQLAstNode, PromQLAstQueryExpression } from '../types';
+/**
+ * Finds the deepest PromQL AST node at or after the given character position.
+ *
+ * - If `pos` falls inside a node, descends into that node looking for the most
+ * specific descendant containing `pos`.
+ * - If no node contains `pos`, returns the first node whose `location.min` is
+ * greater than `pos`.
+ * - Returns `null` if no such node exists (i.e. `pos` is past the end of the
+ * tree).
+ */
+export declare const findNodeAtOrAfter: (ast: PromQLAstQueryExpression, pos: number) => PromQLAstNode | null;
+/**
+ * Finds the deepest PromQL AST node at or before the given character position.
+ *
+ * - If `pos` falls inside a node, descends into that node looking for the most
+ * specific descendant containing `pos`.
+ * - If no node contains `pos`, returns the last node whose `location.max` is
+ * less than `pos`.
+ * - Returns `null` if no such node exists (i.e. `pos` is before the start of
+ * the tree).
+ */
+export declare const findNodeAtOrBefore: (ast: PromQLAstQueryExpression, pos: number) => PromQLAstNode | null;
+//# sourceMappingURL=find_node.d.ts.map

package/lib/embedded_languages/promql/ast/find_node.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"find_node.d.ts","sourceRoot":"","sources":["../../../../src/embedded_languages/promql/ast/find_node.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AAqDxE;;;;;;;;;GASG;AACH,eAAO,MAAM,iBAAiB,GAC5B,KAAK,wBAAwB,EAC7B,KAAK,MAAM,KACV,aAAa,GAAG,IAA6B,CAAC;AAEjD;;;;;;;;;GASG;AACH,eAAO,MAAM,kBAAkB,GAC7B,KAAK,wBAAwB,EAC7B,KAAK,MAAM,KACV,aAAa,GAAG,IAA8B,CAAC"}

package/lib/embedded_languages/promql/ast/traversal.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"traversal.d.ts","sourceRoot":"","sources":["../../../../src/embedded_languages/promql/ast/traversal.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAE9C,wBAAiB,oBAAoB,CAAC,IAAI,EAAE,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC,CAmClF"}
+{"version":3,"file":"traversal.d.ts","sourceRoot":"","sources":["../../../../src/embedded_languages/promql/ast/traversal.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAE9C,wBAAiB,oBAAoB,CAAC,IAAI,EAAE,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC,CAiFlF"}

package/lib/embedded_languages/promql/parser/decorations.d.ts

@@ -0,0 +1,39 @@
+import { type CommonTokenStream } from 'antlr4';
+import type { ESQLAstCommentSingleLine } from '../../../types';
+import type { PromQLAstQueryExpression } from '../types';
+/**
+ * A collected comment decoration with its positional context.
+ */
+export interface PromQLCommentDecoration {
+    type: 'comment';
+    /**
+     * Whether there is non-punctuation content to the left of this comment on
+     * the same line. When `true`, the comment is a trailing comment (e.g.
+     * `rate(x[5m]) # trailing`). When `false`, the comment occupies its own
+     * line (e.g. `# whole-line comment`).
+     */
+    hasContentToLeft: boolean;
+    /** The AST comment node. */
+    node: ESQLAstCommentSingleLine;
+}
+/**
+ * Collects all comments from the PromQL token stream.
+ *
+ * Iterates through every token (including hidden-channel tokens) and extracts
+ * comment decorations with their positional context.
+ *
+ * @param tokens ANTLR token stream from the PromQL lexer.
+ * @param offset Character offset to add to all locations (for embedded PromQL).
+ * @returns List of comment decorations in source order.
+ */
+export declare const collectPromQLDecorations: (tokens: CommonTokenStream, offset?: number) => PromQLCommentDecoration[];
+/**
+ * Attaches the comments collected by {@link collectPromQLDecorations} to the
+ * appropriate nodes in the PromQL AST. Similar attachment logic is used as
+ * in ES|QL.
+ *
+ * @param ast Root of the PromQL AST.
+ * @param comments Decorations collected from the token stream.
+ */
+export declare const attachPromQLDecorations: (ast: PromQLAstQueryExpression, comments: PromQLCommentDecoration[]) => void;
+//# sourceMappingURL=decorations.d.ts.map

package/lib/embedded_languages/promql/parser/decorations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"decorations.d.ts","sourceRoot":"","sources":["../../../../src/embedded_languages/promql/parser/decorations.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAGhD,OAAO,KAAK,EAAE,wBAAwB,EAAyB,MAAM,gBAAgB,CAAC;AACtF,OAAO,KAAK,EAAiB,wBAAwB,EAAE,MAAM,UAAU,CAAC;AAiBxE;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,SAAS,CAAC;IAEhB;;;;;OAKG;IACH,gBAAgB,EAAE,OAAO,CAAC;IAE1B,4BAA4B;IAC5B,IAAI,EAAE,wBAAwB,CAAC;CAChC;AAkBD;;;;;;;;;GASG;AACH,eAAO,MAAM,wBAAwB,GACnC,QAAQ,iBAAiB,EACzB,SAAQ,MAAU,KACjB,uBAAuB,EAsDzB,CAAC;AA2BF;;;;;;;GAOG;AACH,eAAO,MAAM,uBAAuB,GAClC,KAAK,wBAAwB,EAC7B,UAAU,uBAAuB,EAAE,KAClC,IAuBF,CAAC"}

package/lib/embedded_languages/promql/parser/parser.d.ts

@@ -11,6 +11,12 @@ export interface PromQLParseOptions {
      * The offset will be added to all location values in the AST.
      */
     offset?: number;
+    /**
+     * When `true`, comments are collected from the token stream and attached to
+     * AST nodes via the `formatting` property. Required for the pretty-printer
+     * to round-trip comments. Defaults to `false`.
+     */
+    withFormatting?: boolean;
 }
 /**
  * Parser for PromQL queries.

package/lib/embedded_languages/promql/parser/parser.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"parser.d.ts","sourceRoot":"","sources":["../../../../src/embedded_languages/promql/parser/parser.ts"],"names":[],"mappings":"AAOA,OAAO,EAAe,iBAAiB,EAAE,KAAK,UAAU,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,oCAAoC,CAAC;AAC5E,OAAO,EAAE,OAAO,IAAI,qBAAqB,EAAE,MAAM,qCAAqC,CAAC;AACvF,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAG9D,OAAO,KAAK,EAAE,wBAAwB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC5E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAElD,MAAM,WAAW,kBAAkB;IACjC;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;;GAKG;AACH,qBAAa,YAAY;aA2CL,GAAG,EAAE,MAAM;aACX,OAAO,EAAE,kBAAkB;IA3C7C;;OAEG;IACH,gBAAuB,MAAM,GAAI,KAAK,MAAM,EAAE,UAAU,kBAAkB,kBAExE;IAEF;;;;;;;;;;;;;OAaG;IACH,gBAAuB,KAAK,GAAI,KAAK,MAAM,EAAE,UAAU,kBAAkB,KAAG,iBAAiB,CAE3F;IAEF;;;;;OAKG;IACH,gBAAuB,WAAW,GAAI,KAAK,MAAM,KAAG,WAAW,EAAE,CAE/D;IAEF,SAAgB,OAAO,EAAE,UAAU,CAAC;IACpC,SAAgB,KAAK,EAAE,WAAW,CAAC;IACnC,SAAgB,MAAM,EAAE,iBAAiB,CAAC;IAC1C,SAAgB,MAAM,EAAE,qBAAqB,CAAC;IAC9C,SAAgB,MAAM,sBAA6B;gBAGjC,GAAG,EAAE,MAAM,EACX,OAAO,GAAE,kBAAuB;IAclD;;OAEG;IACI,KAAK,IAAI,iBAAiB,CAAC,wBAAwB,CAAC;IA2B3D;;OAEG;IACI,WAAW,IAAI,WAAW,EAAE;CAIpC"}
+{"version":3,"file":"parser.d.ts","sourceRoot":"","sources":["../../../../src/embedded_languages/promql/parser/parser.ts"],"names":[],"mappings":"AAOA,OAAO,EAAe,iBAAiB,EAAE,KAAK,UAAU,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,oCAAoC,CAAC;AAC5E,OAAO,EAAE,OAAO,IAAI,qBAAqB,EAAE,MAAM,qCAAqC,CAAC;AACvF,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAI9D,OAAO,KAAK,EAAE,wBAAwB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC5E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAElD,MAAM,WAAW,kBAAkB;IACjC;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;;OAIG;IACH,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED;;;;;GAKG;AACH,qBAAa,YAAY;aA2CL,GAAG,EAAE,MAAM;aACX,OAAO,EAAE,kBAAkB;IA3C7C;;OAEG;IACH,gBAAuB,MAAM,GAAI,KAAK,MAAM,EAAE,UAAU,kBAAkB,kBAExE;IAEF;;;;;;;;;;;;;OAaG;IACH,gBAAuB,KAAK,GAAI,KAAK,MAAM,EAAE,UAAU,kBAAkB,KAAG,iBAAiB,CAE3F;IAEF;;;;;OAKG;IACH,gBAAuB,WAAW,GAAI,KAAK,MAAM,KAAG,WAAW,EAAE,CAE/D;IAEF,SAAgB,OAAO,EAAE,UAAU,CAAC;IACpC,SAAgB,KAAK,EAAE,WAAW,CAAC;IACnC,SAAgB,MAAM,EAAE,iBAAiB,CAAC;IAC1C,SAAgB,MAAM,EAAE,qBAAqB,CAAC;IAC9C,SAAgB,MAAM,sBAA6B;gBAGjC,GAAG,EAAE,MAAM,EACX,OAAO,GAAE,kBAAuB;IAclD;;OAEG;IACI,KAAK,IAAI,iBAAiB,CAAC,wBAAwB,CAAC;IAiC3D;;OAEG;IACI,WAAW,IAAI,WAAW,EAAE;CAIpC"}

package/lib/index.js

@@ -311,10 +311,13 @@ function* childrenOfPromqlNode(node) {
       return;
     }
     case "function": {
-      if (node.grouping) {
+      if (node.grouping && node.groupingPosition !== "after") {
         yield node.grouping;
       }
       yield* node.args;
+      if (node.grouping && node.groupingPosition === "after") {
+        yield node.grouping;
+      }
       return;
     }
     case "selector": {
@@ -327,6 +330,49 @@ function* childrenOfPromqlNode(node) {
       if (value) yield value;
       return;
     }
+    case "binary-expression": {
+      yield node.left;
+      if (node.modifier) yield node.modifier;
+      yield node.right;
+      return;
+    }
+    case "unary-expression": {
+      yield node.arg;
+      return;
+    }
+    case "subquery": {
+      yield node.expr;
+      yield node.range;
+      if (node.resolution) yield node.resolution;
+      if (node.evaluation) yield node.evaluation;
+      return;
+    }
+    case "parens": {
+      if (node.child) yield node.child;
+      return;
+    }
+    case "evaluation": {
+      if (node.offset) yield node.offset;
+      if (node.at) yield node.at;
+      return;
+    }
+    case "offset": {
+      if (node.duration) yield node.duration;
+      return;
+    }
+    case "at": {
+      if (typeof node.value !== "string") yield node.value;
+      return;
+    }
+    case "modifier": {
+      yield* node.labels;
+      if (node.groupModifier) yield node.groupModifier;
+      return;
+    }
+    case "group-modifier": {
+      yield* node.labels;
+      return;
+    }
   }
   if ("args" in node && Array.isArray(node.args)) {
     yield* node.args;
@@ -1065,9 +1111,18 @@ var Walker = class _Walker {
     _Walker.walk(tree, { ...options, visitAny });
   };
   aborted = false;
+  skippedChildren = false;
   abort() {
     this.aborted = true;
   }
+  skipChildren() {
+    this.skippedChildren = true;
+  }
+  readAndResetSkippedChildren() {
+    const skipped = this.skippedChildren;
+    this.skippedChildren = false;
+    return skipped;
+  }
   walk(tree, parent = void 0) {
     if (this.aborted) return;
     if (!tree) return;
@@ -1105,17 +1160,20 @@ var Walker = class _Walker {
     if (this.aborted) return;
     const { options } = this;
     (options.visitCommand ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkList(node.args, node);
   }
   walkHeaderCommand(node, parent) {
     if (this.aborted) return;
     const { options } = this;
     (options.visitHeaderCommand ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkList(node.args, node);
   }
   walkOption(node, parent) {
     const { options } = this;
     (options.visitCommandOption ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkList(node.args, node);
   }
   walkExpression(node, parent = void 0) {
@@ -1130,11 +1188,13 @@ var Walker = class _Walker {
   walkListLiteral(node, parent) {
     const { options } = this;
     (options.visitListLiteral ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkList(node.values, node);
   }
   walkSource(node, parent) {
     const { options } = this;
     (options.visitSource ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     const children2 = [];
     if (node.prefix) {
       children2.push(node.prefix);
@@ -1151,6 +1211,7 @@ var Walker = class _Walker {
     const { options } = this;
     const { args } = node;
     (options.visitColumn ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     if (args) {
       this.walkList(args, node);
     }
@@ -1158,27 +1219,32 @@ var Walker = class _Walker {
   walkOrder(node, parent) {
     const { options } = this;
     (options.visitOrder ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkList(node.args, node);
   }
   walkInlineCast(node, parent) {
     const { options } = this;
     (options.visitInlineCast ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkExpression(node.value, node);
   }
   walkFunction(node, parent = void 0) {
     const { options } = this;
     (options.visitFunction ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     if (node.operator) this.walkSingleAstItem(node.operator, node);
     this.walkList(node.args, node);
   }
   walkMap(node, parent) {
     const { options } = this;
     (options.visitMap ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     this.walkList(node.entries, node);
   }
   walkMapEntry(node, parent) {
     const { options } = this;
     (options.visitMapEntry ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     if (options.order === "backward") {
       this.walkSingleAstItem(resolveItem(node.value), node);
       this.walkSingleAstItem(resolveItem(node.key), node);
@@ -1190,6 +1256,7 @@ var Walker = class _Walker {
   walkParens(node, parent) {
     const { options } = this;
     (options.visitParens ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     if (node.child) {
       this.walkSingleAstItem(node.child, node);
     }
@@ -1197,6 +1264,7 @@ var Walker = class _Walker {
   walkQuery(node, parent) {
     const { options } = this;
     (options.visitQuery ?? options.visitAny)?.(node, parent, this);
+    if (this.readAndResetSkippedChildren()) return;
     if (node.header && !options.skipHeader) {
       this.walkList(node.header, node);
     }
@@ -1265,6 +1333,7 @@ var Walker = class _Walker {
       }
       case "literal": {
         (options.visitLiteral ?? options.visitAny)?.(node, parent, this);
+        this.readAndResetSkippedChildren();
         break;
       }
       case "list": {
@@ -1281,10 +1350,12 @@ var Walker = class _Walker {
       }
       case "identifier": {
         (options.visitIdentifier ?? options.visitAny)?.(node, parent, this);
+        this.readAndResetSkippedChildren();
         break;
       }
       case "unknown": {
         (options.visitUnknown ?? options.visitAny)?.(node, parent, this);
+        this.readAndResetSkippedChildren();
         break;
       }
     }
@@ -61132,6 +61203,140 @@ var PromQLCstToAstConverter = class {
   }
 };
 
+// src/embedded_languages/promql/ast/find_node.ts
+var sortedLocatedChildren = (node) => {
+  const children2 = [];
+  for (const child of childrenOfPromqlNode(node)) {
+    if (child.location) children2.push(child);
+  }
+  children2.sort((a, b) => a.location.min - b.location.min);
+  return children2;
+};
+var findForward = (node, pos) => {
+  for (const child of sortedLocatedChildren(node)) {
+    const { min, max } = child.location;
+    if (min <= pos && max >= pos) {
+      return findForward(child, pos) ?? child;
+    }
+    if (min > pos) {
+      return child;
+    }
+  }
+  return null;
+};
+var findBackward = (node, pos) => {
+  const children2 = sortedLocatedChildren(node);
+  for (let i = children2.length - 1; i >= 0; i--) {
+    const child = children2[i];
+    const { min, max } = child.location;
+    if (min <= pos && max >= pos) {
+      return findBackward(child, pos) ?? child;
+    }
+    if (max < pos) {
+      return child;
+    }
+  }
+  return null;
+};
+var findNodeAtOrAfter = (ast, pos) => findForward(ast, pos);
+var findNodeAtOrBefore = (ast, pos) => findBackward(ast, pos);
+
+// src/embedded_languages/promql/parser/decorations.ts
+var HIDDEN_CHANNEL2 = 1;
+var punctuationChars2 = /* @__PURE__ */ new Set([".", ",", ";", ":", "(", ")", "[", "]", "{", "}"]);
+var isLikelyPunctuation2 = (text2) => text2.length === 1 && punctuationChars2.has(text2);
+var cleanCommentText = (text2) => {
+  let end = text2.length;
+  if (text2[end - 1] === "\n") {
+    end--;
+  }
+  if (end > 0 && text2[end - 1] === "\r") {
+    end--;
+  }
+  return text2.slice(1, end);
+};
+var collectPromQLDecorations = (tokens, offset = 0) => {
+  const result = [];
+  const list12 = tokens.tokens;
+  const length = list12.length;
+  let pos = 0;
+  let hasContentToLeft = false;
+  for (let i = 0; i < length - 1; i++) {
+    const token = list12[i];
+    const { channel, text: text2 } = token;
+    const min = pos;
+    const max = min + text2.length - 1;
+    pos = max + 1;
+    if (channel !== HIDDEN_CHANNEL2) {
+      if (!isLikelyPunctuation2(text2)) {
+        hasContentToLeft = true;
+      }
+      continue;
+    }
+    if (text2[0] !== "#") {
+      if (text2.indexOf("\n") !== -1) {
+        hasContentToLeft = false;
+      }
+      continue;
+    }
+    const cleanText = cleanCommentText(text2);
+    const endsWithNewline = text2[text2.length - 1] === "\n";
+    const maxOffset = endsWithNewline ? text2[text2.length - 2] === "\r" ? 2 : 1 : 0;
+    const node = Builder.comment("single-line", cleanText, {
+      min: min + offset,
+      max: max - maxOffset + offset
+    });
+    result.push({
+      type: "comment",
+      hasContentToLeft,
+      node
+    });
+    if (endsWithNewline) {
+      hasContentToLeft = false;
+    }
+  }
+  return result;
+};
+var ensureFormatting = (node) => {
+  if (!node.formatting) {
+    node.formatting = {};
+  }
+  return node.formatting;
+};
+var attachTopComment2 = (node, comment) => {
+  const formatting = ensureFormatting(node);
+  if (!formatting.top) formatting.top = [];
+  formatting.top.push(comment);
+};
+var attachBottomComment2 = (node, comment) => {
+  const formatting = ensureFormatting(node);
+  if (!formatting.bottom) formatting.bottom = [];
+  formatting.bottom.push(comment);
+};
+var attachTrailingComment = (node, comment) => {
+  const formatting = ensureFormatting(node);
+  formatting.rightSingleLine = comment;
+};
+var attachPromQLDecorations = (ast, comments) => {
+  for (const comment of comments) {
+    const { location } = comment.node;
+    if (!location) continue;
+    if (!comment.hasContentToLeft) {
+      const target2 = findNodeAtOrAfter(ast, location.max);
+      if (target2) {
+        attachTopComment2(target2, comment.node);
+      } else {
+        attachBottomComment2(ast, comment.node);
+      }
+      continue;
+    }
+    const target = findNodeAtOrBefore(ast, location.min);
+    if (target) {
+      attachTrailingComment(target, comment.node);
+    }
+  }
+};
+
 // src/embedded_languages/promql/parser/parser.ts
 var PromQLParser = class _PromQLParser {
   constructor(src2, options = {}) {
@@ -61197,6 +61402,11 @@ var PromQLParser = class _PromQLParser {
           errors: this.errors.getErrors()
         };
       }
+      if (this.options.withFormatting) {
+        this.tokens.fill();
+        const comments = collectPromQLDecorations(this.tokens, this.options.offset ?? 0);
+        attachPromQLDecorations(root, comments);
+      }
       return {
         root,
         errors: this.errors.getErrors()