"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.handleRequest = void 0;
var _i18n = require("@kbn/i18n");
var _rxjs = require("rxjs");
var _operators = require("rxjs/operators");
var _ = require("../../..");
var _tabify = require("../../tabify");
/*
 * 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 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */

const handleRequest = ({
  abortSignal,
  aggs,
  filters,
  indexPattern,
  inspectorAdapters,
  query,
  searchSessionId,
  searchSourceService,
  timeFields,
  timeRange,
  disableShardWarnings,
  getNow,
  executionContext,
  title,
  description
}) => {
  return (0, _rxjs.defer)(async () => {
    var _indexPattern$getTime;
    const forceNow = getNow === null || getNow === void 0 ? void 0 : getNow();
    const searchSource = await searchSourceService.create();
    searchSource.setField('index', indexPattern);
    searchSource.setField('size', 0);
    const timeFilterSearchSource = searchSource.createChild({
      callParentStartHandlers: true
    });
    const requestSearchSource = timeFilterSearchSource.createChild({
      callParentStartHandlers: true
    });

    // If timeFields have been specified, use the specified ones, otherwise use primary time field of index
    // pattern if it's available.
    const defaultTimeField = indexPattern === null || indexPattern === void 0 ? void 0 : (_indexPattern$getTime = indexPattern.getTimeField) === null || _indexPattern$getTime === void 0 ? void 0 : _indexPattern$getTime.call(indexPattern);
    const defaultTimeFields = defaultTimeField ? [defaultTimeField.name] : [];
    const allTimeFields = timeFields !== null && timeFields !== void 0 && timeFields.length ? timeFields : defaultTimeFields;
    aggs.setTimeRange(timeRange);
    aggs.setForceNow(forceNow);
    aggs.setTimeFields(allTimeFields);

    // For now we need to mirror the history of the passed search source, since
    // the request inspector wouldn't work otherwise.
    Object.defineProperty(requestSearchSource, 'history', {
      get() {
        return searchSource.history;
      },
      set(history) {
        return searchSource.history = history;
      }
    });
    requestSearchSource.setField('aggs', aggs);
    requestSearchSource.onRequestStart((paramSearchSource, options) => {
      return aggs.onSearchRequestStart(paramSearchSource, options);
    });

    // If a timeRange has been specified and we had at least one timeField available, create range
    // filters for that those time fields
    if (timeRange && allTimeFields.length > 0) {
      timeFilterSearchSource.setField('filter', () => {
        return aggs.getSearchSourceTimeFilter(forceNow);
      });
    }
    requestSearchSource.setField('filter', filters);
    requestSearchSource.setField('query', query);
    return {
      allTimeFields,
      forceNow,
      requestSearchSource
    };
  }).pipe((0, _operators.switchMap)(({
    allTimeFields,
    forceNow,
    requestSearchSource
  }) => requestSearchSource.fetch$({
    abortSignal,
    disableShardFailureWarning: disableShardWarnings,
    sessionId: searchSessionId,
    inspector: {
      adapter: inspectorAdapters.requests,
      title: title !== null && title !== void 0 ? title : _i18n.i18n.translate('data.functions.esaggs.inspector.dataRequest.title', {
        defaultMessage: 'Data'
      }),
      description: description !== null && description !== void 0 ? description : _i18n.i18n.translate('data.functions.esaggs.inspector.dataRequest.description', {
        defaultMessage: 'This request queries Elasticsearch to fetch the data for the visualization.'
      })
    },
    executionContext
  }).pipe((0, _operators.map)(({
    rawResponse: response
  }) => {
    const parsedTimeRange = timeRange ? (0, _.calculateBounds)(timeRange, {
      forceNow
    }) : null;
    const tabifyParams = {
      metricsAtAllLevels: aggs.hierarchical,
      partialRows: aggs.partialRows,
      timeRange: parsedTimeRange ? {
        from: parsedTimeRange.min,
        to: parsedTimeRange.max,
        timeFields: allTimeFields
      } : undefined
    };
    return (0, _tabify.tabifyAggResponse)(aggs, response, tabifyParams);
  }))));
};
exports.handleRequest = handleRequest;