feat: adds visitor context for metrics_info

Author: momovdg

src/ast/visitor/contexts.ts

@@ -18,6 +18,7 @@ import type {
   ESQLAstHeaderCommand,
   ESQLAstItem,
   ESQLAstJoinCommand,
+  ESQLAstMetricsInfoCommand,
   ESQLAstQueryExpression,
   ESQLAstRegisteredDomainCommand,
   ESQLAstRerankCommand,
@@ -593,6 +594,12 @@ export class UriPartsCommandVisitorContext<
   Data extends SharedData = SharedData,
 > extends CommandVisitorContext<Methods, Data, ESQLAstUriPartsCommand> {}
 
+// METRICS_INFO
+export class MetricsInfoCommandVisitorContext<
+  Methods extends VisitorMethods = VisitorMethods,
+  Data extends SharedData = SharedData,
+> extends CommandVisitorContext<Methods, Data, ESQLAstMetricsInfoCommand> {}
+
 // Expressions -----------------------------------------------------------------
 
 export class ExpressionVisitorContext<

src/ast/visitor/global_visitor_context.ts

@@ -12,6 +12,7 @@ import type {
   ESQLAstCompletionCommand,
   ESQLAstHeaderCommand,
   ESQLAstJoinCommand,
+  ESQLAstMetricsInfoCommand,
   ESQLAstQueryExpression,
   ESQLAstRegisteredDomainCommand,
   ESQLAstRerankCommand,
@@ -239,6 +240,14 @@ export class GlobalVisitorContext<
           input as any
         );
       }
+      case 'metrics_info': {
+        if (!this.methods.visitMetricsInfoCommand) break;
+        return this.visitMetricsInfoCommand(
+          parent,
+          commandNode as ESQLAstMetricsInfoCommand,
+          input as any
+        );
+      }
     }
     return this.visitCommandGeneric(parent, commandNode, input as any);
   }
@@ -527,6 +536,15 @@ export class GlobalVisitorContext<
     return this.visitWithSpecificContext('visitUriPartsCommand', context, input);
   }
 
+  public visitMetricsInfoCommand(
+    parent: contexts.VisitorContext | null,
+    node: ESQLAstMetricsInfoCommand,
+    input: types.VisitorInput<Methods, 'visitMetricsInfoCommand'>
+  ): types.VisitorOutput<Methods, 'visitMetricsInfoCommand'> {
+    const context = new contexts.MetricsInfoCommandVisitorContext(this, node, parent);
+    return this.visitWithSpecificContext('visitMetricsInfoCommand', context, input);
+  }
+
   // #endregion
 
   // #region Expression visiting -------------------------------------------------------

src/ast/visitor/types.ts

@@ -112,6 +112,7 @@ export type CommandVisitorInput<Methods extends VisitorMethods> = AnyToVoid<
     VisitorInput<Methods, 'visitRerankCommand'> &
     VisitorInput<Methods, 'visitChangePointCommand'> &
     VisitorInput<Methods, 'visitUriPartsCommand'> &
+    VisitorInput<Methods, 'visitMetricsInfoCommand'> &
     VisitorInput<Methods, 'visitRegisteredDomainCommand'>
 >;
 
@@ -145,6 +146,7 @@ export type CommandVisitorOutput<Methods extends VisitorMethods> =
   | VisitorOutput<Methods, 'visitChangePointCommand'>
   | VisitorOutput<Methods, 'visitCompletionCommand'>
   | VisitorOutput<Methods, 'visitUriPartsCommand'>
+  | VisitorOutput<Methods, 'visitMetricsInfoCommand'>
   | VisitorOutput<Methods, 'visitRegisteredDomainCommand'>;
 
 /* eslint-disable @typescript-eslint/no-explicit-any */
@@ -206,6 +208,11 @@ export interface VisitorMethods<
   visitFuseCommand?: Visitor<contexts.FuseCommandVisitorContext<Visitors, Data>, any, any>;
   visitMmrCommand?: Visitor<contexts.MmrCommandVisitorContext<Visitors, Data>, any, any>;
   visitUriPartsCommand?: Visitor<contexts.UriPartsCommandVisitorContext<Visitors, Data>, any, any>;
+  visitMetricsInfoCommand?: Visitor<
+    contexts.MetricsInfoCommandVisitorContext<Visitors, Data>,
+    any,
+    any
+  >;
   visitExpression?: Visitor<contexts.ExpressionVisitorContext<Visitors, Data>, any, any>;
   visitSourceExpression?: Visitor<
     contexts.SourceExpressionVisitorContext<Visitors, Data>,