@elastic/esql 1.7.1 → 1.8.0

package/lib/index.mjs

@@ -25888,6 +25888,8 @@ var TsInfoCommandVisitorContext = class extends CommandVisitorContext {
 };
 var MetricsInfoCommandVisitorContext = class extends CommandVisitorContext {
 };
+var UserAgentCommandVisitorContext = class extends CommandVisitorContext {
+};
 var ExpressionVisitorContext = class extends VisitorContext {
 };
 var ColumnExpressionVisitorContext = class extends VisitorContext {
@@ -26157,6 +26159,14 @@ var GlobalVisitorContext = class {
           input
         );
       }
+      case "user_agent": {
+        if (!this.methods.visitUserAgentCommand) break;
+        return this.visitUserAgentCommand(
+          parent,
+          commandNode,
+          input
+        );
+      }
     }
     return this.visitCommandGeneric(parent, commandNode, input);
   }
@@ -26295,6 +26305,10 @@ var GlobalVisitorContext = class {
     const context = new MetricsInfoCommandVisitorContext(this, node, parent);
     return this.visitWithSpecificContext("visitMetricsInfoCommand", context, input);
   }
+  visitUserAgentCommand(parent, node, input) {
+    const context = new UserAgentCommandVisitorContext(this, node, parent);
+    return this.visitWithSpecificContext("visitUserAgentCommand", context, input);
+  }
   // #endregion
   // #region Expression visiting -------------------------------------------------------
   visitExpressionGeneric(parent, node, input) {
@@ -60499,6 +60513,10 @@ var CstToAstConverter = class {
     if (metricsInfoCommandCtx) {
       return this.fromMetricsInfoCommand(metricsInfoCommandCtx);
     }
+    const userAgentCommandCtx = ctx.userAgentCommand();
+    if (userAgentCommandCtx) {
+      return this.fromUserAgentCommand(userAgentCommandCtx);
+    }
   }
   createCommand(name, ctx, partial) {
     const parserFields = this.getParserFields(ctx);
@@ -61704,7 +61722,7 @@ var CstToAstConverter = class {
     const limitOption = this.fromMmrLimitOption(ctx);
     if (limitOption) args.push(limitOption);
     const limit = limitOption ? limitOption.args[0] : void 0;
-    const withOption = this.fromMmrWithOption(ctx.commandNamedParameters());
+    const withOption = this.fromOptionalNamedParametersWithOption(ctx.commandNamedParameters());
     if (withOption) args.push(withOption);
     const namedParameters = withOption ? withOption.args[0] : void 0;
     const command2 = this.createCommand("mmr", ctx, {
@@ -61760,17 +61778,6 @@ var CstToAstConverter = class {
     limitOption.location.max = limitValueCtx.stop?.stop ?? limitToken.symbol.stop;
     return limitOption;
   }
-  fromMmrWithOption(namedParametersCtx) {
-    if (!namedParametersCtx) return;
-    const withOption = this.fromCommandNamedParameters(namedParametersCtx);
-    const mapArg = withOption.args[0];
-    if (mapArg) {
-      const incomplete = mapArg.entries.some((entry) => entry.incomplete) || mapArg.entries.length === 0;
-      mapArg.incomplete = incomplete;
-      withOption.incomplete = incomplete;
-      return withOption;
-    }
-  }
   // --------------------------------------------------------------- URI_PARTS
   fromUriPartsCommand(ctx) {
     return this.fromQualifiedNameAssignmentCommand("uri_parts", ctx);
@@ -61811,6 +61818,58 @@ var CstToAstConverter = class {
   fromMetricsInfoCommand(ctx) {
     return this.createCommand("metrics_info", ctx);
   }
+  // --------------------------------------------------------------- USER_AGENT
+  fromUserAgentCommand(ctx) {
+    const args = [];
+    const qualifiedNameCtx = ctx.qualifiedName();
+    let targetField;
+    let expression2;
+    if (qualifiedNameCtx && ctx.ASSIGN()) {
+      targetField = this.toColumn(qualifiedNameCtx);
+      expression2 = this.fromPrimaryExpressionStrict(ctx.primaryExpression());
+      const assignment = this.toFunction(
+        ctx.ASSIGN().getText(),
+        ctx,
+        void 0,
+        "binary-expression"
+      );
+      assignment.args.push(targetField, expression2);
+      assignment.location = this.extendLocationToArgs(assignment);
+      args.push(assignment);
+    } else if (qualifiedNameCtx) {
+      targetField = this.toColumn(qualifiedNameCtx);
+      args.push(targetField);
+    }
+    const withOption = this.fromOptionalNamedParametersWithOption(ctx.commandNamedParameters());
+    if (withOption) {
+      args.push(withOption);
+    }
+    const namedParameters = withOption?.args[0];
+    const command2 = this.createCommand(
+      "user_agent",
+      ctx,
+      {
+        args,
+        ...namedParameters ? { namedParameters } : {}
+      }
+    );
+    if (targetField) {
+      command2.targetField = targetField;
+    }
+    if (expression2) {
+      command2.expression = expression2;
+    }
+    if (qualifiedNameCtx && ctx.ASSIGN()) {
+      if (expression2 && (expression2.incomplete || expression2.type === "unknown")) {
+        expression2.incomplete = true;
+        command2.incomplete = true;
+      }
+    } else if (qualifiedNameCtx) {
+      command2.incomplete = true;
+    }
+    command2.incomplete ||= withOption?.incomplete ?? false;
+    return command2;
+  }
   // -------------------------------------------------------------- expressions
   toColumnsFromCommand(ctx) {
     if (ctx instanceof MetadataContext) {
@@ -61970,6 +62029,24 @@ var CstToAstConverter = class {
     withOption.incomplete = map.incomplete;
     return withOption;
   }
+  /**
+   * Optional `WITH { ... }` from the `commandNamedParameters` rule. Returns undefined when
+   * there is no map (e.g. missing `WITH`). When a map is present, propagates incomplete
+   * state if any entry is incomplete or the map has no entries.
+   */
+  fromOptionalNamedParametersWithOption(namedParametersCtx) {
+    if (!namedParametersCtx) {
+      return;
+    }
+    const withOption = this.fromCommandNamedParameters(namedParametersCtx);
+    const mapArg = withOption.args[0];
+    if (mapArg) {
+      const incomplete = mapArg.entries.some((entry) => entry.incomplete) || mapArg.entries.length === 0;
+      mapArg.incomplete = incomplete;
+      withOption.incomplete = incomplete;
+      return withOption;
+    }
+  }
   collectInlineCast(ctx) {
     const value = this.fromPrimaryExpressionStrict(ctx.primaryExpression());
     return Builder.expression.inlineCast(
@@ -65569,6 +65646,7 @@ export {
   TsInfoCommandVisitorContext,
   UnaryExpressionGroup,
   UriPartsCommandVisitorContext,
+  UserAgentCommandVisitorContext,
   Visitor,
   VisitorContext,
   Walker,