From bc03874d1e4e899647ed56e850fffd65751a9ccb Mon Sep 17 00:00:00 2001
From: vadimkibana <vadimkibana@gmail.com>
Date: Tue, 11 Jun 2024 17:14:56 +0200
Subject: [PATCH] use FROM command to use METRICS fields

---
 .../src/validation/helpers.ts                      | 14 +++++++++++---
 .../src/validation/validation.ts                   |  9 +++++----
 2 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/packages/kbn-esql-validation-autocomplete/src/validation/helpers.ts b/packages/kbn-esql-validation-autocomplete/src/validation/helpers.ts
index 20ec8990e1830..11cf5fd2d8f49 100644
--- a/packages/kbn-esql-validation-autocomplete/src/validation/helpers.ts
+++ b/packages/kbn-esql-validation-autocomplete/src/validation/helpers.ts
@@ -6,7 +6,13 @@
  * Side Public License, v 1.
  */
 
-import type { ESQLAst, ESQLAstItem, ESQLMessage, ESQLSingleAstItem } from '@kbn/esql-ast';
+import type {
+  ESQLAst,
+  ESQLAstItem,
+  ESQLAstMetricsCommand,
+  ESQLMessage,
+  ESQLSingleAstItem,
+} from '@kbn/esql-ast';
 import { FunctionDefinition } from '../definitions/types';
 import { getAllArrayTypes, getAllArrayValues } from '../shared/helpers';
 import { getMessageFromId } from './errors';
@@ -14,8 +20,10 @@ import type { ESQLPolicy, ReferenceMaps } from './types';
 
 export function buildQueryForFieldsFromSource(queryString: string, ast: ESQLAst) {
   const firstCommand = ast[0];
-  if (firstCommand == null) {
-    return '';
+  if (!firstCommand) return '';
+  if (firstCommand.name === 'metrics') {
+    const metrics = firstCommand as ESQLAstMetricsCommand;
+    return `FROM ${metrics.sources.map((source) => source.name).join(', ')}`;
   }
   return queryString.substring(0, firstCommand.location.max + 1);
 }
diff --git a/packages/kbn-esql-validation-autocomplete/src/validation/validation.ts b/packages/kbn-esql-validation-autocomplete/src/validation/validation.ts
index 5396896e112eb..2a24a99c14fab 100644
--- a/packages/kbn-esql-validation-autocomplete/src/validation/validation.ts
+++ b/packages/kbn-esql-validation-autocomplete/src/validation/validation.ts
@@ -205,8 +205,8 @@ function validateFunctionColumnArg(
   const messages: ESQLMessage[] = [];
   if (isColumnItem(actualArg)) {
     if (actualArg.name) {
-      const { hit: columnCheck, nameHit } = columnExists(actualArg, references);
-      if (!columnCheck) {
+      const { hit, nameHit } = columnExists(actualArg, references);
+      if (!hit) {
         if (argDef.constantOnly) {
           messages.push(
             getMessageFromId({
@@ -1150,13 +1150,14 @@ async function validateAst(
   messages.push(...validateUnsupportedTypeFields(availableFields));
 
   for (const command of ast) {
-    const commandMessages = validateCommand(command, {
+    const references: ReferenceMaps = {
       sources,
       fields: availableFields,
       policies: availablePolicies,
       variables,
       query: queryString,
-    });
+    };
+    const commandMessages = validateCommand(command, references);
     messages.push(...commandMessages);
   }