"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.TIME_SERIES_BUCKET_SELECTOR_PATH_NAME = exports.TIME_SERIES_BUCKET_SELECTOR_FIELD = void 0;
exports.getResultFromEs = getResultFromEs;
exports.timeSeriesQuery = timeSeriesQuery;
var _server = require("@kbn/alerting-plugin/server");
var _esQuery = require("@kbn/es-query");
var _common = require("../../../common");
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

const TIME_SERIES_BUCKET_SELECTOR_PATH_NAME = 'compareValue';
exports.TIME_SERIES_BUCKET_SELECTOR_PATH_NAME = TIME_SERIES_BUCKET_SELECTOR_PATH_NAME;
const TIME_SERIES_BUCKET_SELECTOR_FIELD = `params.${TIME_SERIES_BUCKET_SELECTOR_PATH_NAME}`;
exports.TIME_SERIES_BUCKET_SELECTOR_FIELD = TIME_SERIES_BUCKET_SELECTOR_FIELD;
async function timeSeriesQuery(params) {
  const {
    logger,
    esClient,
    query: queryParams,
    condition: conditionParams
  } = params;
  const {
    index,
    timeWindowSize,
    timeWindowUnit,
    interval,
    timeField,
    dateStart,
    dateEnd,
    filterKuery
  } = queryParams;
  const window = `${timeWindowSize}${timeWindowUnit}`;
  const dateRangeInfo = (0, _common.getDateRangeInfo)({
    dateStart,
    dateEnd,
    window,
    interval
  });
  const {
    aggType,
    aggField,
    termField,
    termSize
  } = queryParams;

  // core query
  // Constructing a typesafe ES query in JS is problematic, use any escapehatch for now
  // eslint-disable-next-line @typescript-eslint/no-explicit-any
  const esQuery = {
    index,
    body: {
      size: 0,
      query: {
        bool: {
          filter: [{
            range: {
              [timeField]: {
                gte: dateRangeInfo.dateStart,
                lt: dateRangeInfo.dateEnd,
                format: 'strict_date_time'
              }
            }
          }, ...(!!filterKuery ? [(0, _esQuery.toElasticsearchQuery)((0, _esQuery.fromKueryExpression)(filterKuery))] : [])]
        }
      },
      aggs: (0, _common.buildAggregation)({
        timeSeries: {
          timeField,
          timeWindowSize,
          timeWindowUnit,
          dateStart,
          dateEnd,
          interval
        },
        aggType,
        aggField,
        termField,
        termSize,
        condition: conditionParams
      })
    },
    ignore_unavailable: true,
    allow_no_indices: true
  };

  // add the aggregations

  const isCountAgg = (0, _common.isCountAggregation)(aggType);
  const isGroupAgg = (0, _common.isGroupAggregation)(termField);
  const includeConditionInQuery = !!conditionParams;
  const logPrefix = 'indexThreshold timeSeriesQuery: callCluster';
  logger.debug(`${logPrefix} call: ${JSON.stringify(esQuery)}`);
  let esResult;
  // note there are some commented out console.log()'s below, which are left
  // in, as they are VERY useful when debugging these queries; debug logging
  // isn't as nice since it's a single long JSON line.

  // console.log('time_series_query.ts request\n', JSON.stringify(esQuery, null, 4));
  try {
    esResult = (await esClient.search(esQuery, {
      ignore: [404],
      meta: true
    })).body;
  } catch (err) {
    // console.log('time_series_query.ts error\n', JSON.stringify(err, null, 4));
    logger.warn(`${logPrefix} error: ${(0, _server.getEsErrorMessage)(err)}`);
    return {
      results: [],
      truncated: false
    };
  }

  // console.log('time_series_query.ts response\n', JSON.stringify(esResult, null, 4));
  logger.debug(`${logPrefix} result: ${JSON.stringify(esResult)}`);
  return getResultFromEs({
    isCountAgg,
    isGroupAgg,
    isConditionInQuery: includeConditionInQuery,
    esResult,
    resultLimit: conditionParams === null || conditionParams === void 0 ? void 0 : conditionParams.resultLimit
  });
}
function getResultFromEs({
  isCountAgg,
  isGroupAgg,
  isConditionInQuery,
  esResult,
  resultLimit
}) {
  var _aggregations$groupAg, _aggregations$groupAg2, _aggregations$groupAg3;
  const aggregations = (esResult === null || esResult === void 0 ? void 0 : esResult.aggregations) || {};

  // add a fake 'all documents' group aggregation, if a group aggregation wasn't used
  if (!isGroupAgg && aggregations.dateAgg) {
    const dateAgg = aggregations.dateAgg;
    aggregations.groupAgg = {
      buckets: [{
        key: 'all documents',
        dateAgg
      }]
    };
    delete aggregations.dateAgg;
  }

  // @ts-expect-error specify aggregations type explicitly
  const groupBuckets = ((_aggregations$groupAg = aggregations.groupAgg) === null || _aggregations$groupAg === void 0 ? void 0 : _aggregations$groupAg.buckets) || [];
  // @ts-expect-error specify aggregations type explicitly
  const numGroupsTotal = (_aggregations$groupAg2 = (_aggregations$groupAg3 = aggregations.groupAggCount) === null || _aggregations$groupAg3 === void 0 ? void 0 : _aggregations$groupAg3.count) !== null && _aggregations$groupAg2 !== void 0 ? _aggregations$groupAg2 : 0;
  const result = {
    results: [],
    truncated: isConditionInQuery && resultLimit ? numGroupsTotal > resultLimit : false
  };
  for (const groupBucket of groupBuckets) {
    var _groupBucket$dateAgg;
    if (resultLimit && result.results.length === resultLimit) break;
    const groupName = `${groupBucket === null || groupBucket === void 0 ? void 0 : groupBucket.key}`;
    const dateBuckets = (groupBucket === null || groupBucket === void 0 ? void 0 : (_groupBucket$dateAgg = groupBucket.dateAgg) === null || _groupBucket$dateAgg === void 0 ? void 0 : _groupBucket$dateAgg.buckets) || [];
    const groupResult = {
      group: groupName,
      metrics: []
    };
    result.results.push(groupResult);
    for (const dateBucket of dateBuckets) {
      const date = dateBucket.to_as_string;
      const value = isCountAgg ? dateBucket.doc_count : dateBucket.metricAgg.value;
      groupResult.metrics.push([date, value]);
    }
  }
  return result;
}