"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.buildInput = buildInput;
var _saferLodashSet = require("@kbn/safer-lodash-set");
/*
 * 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.
 */

/*
watch.input.search.request.indices
 */
function buildIndices(index) {
  if (Array.isArray(index)) {
    return index;
  }
  return [index];
}

/*
watch.input.search.request.body.query.bool.filter.range
 */
function buildRange(timeWindowSize, timeWindowUnit, timeField) {
  return {
    [timeField]: {
      gte: `{{ctx.trigger.scheduled_time}}||-${timeWindowSize}${timeWindowUnit}`,
      lte: `{{ctx.trigger.scheduled_time}}`,
      format: 'strict_date_optional_time||epoch_millis'
    }
  };
}

/*
watch.input.search.request.body.query
 */
function buildQuery(timeWindowSize, timeWindowUnit, timeField) {
  //TODO: This is where a saved search would be applied
  return {
    bool: {
      filter: {
        range: buildRange(timeWindowSize, timeWindowUnit, timeField)
      }
    }
  };
}

/*
watch.input.search.request.body.aggs
 */
function buildAggs(aggType, aggField, termField, termSize, termOrder) {
  if (aggType === 'count' && !termField) {
    return null;
  }
  if (aggType === 'count' && termField) {
    return {
      bucketAgg: {
        terms: {
          field: termField,
          size: termSize,
          order: {
            _count: termOrder
          }
        }
      }
    };
  }
  if (aggType !== 'count' && !termField) {
    const result = {
      metricAgg: {}
    };
    (0, _saferLodashSet.set)(result, `metricAgg.${aggType}`, {
      field: aggField
    });
    return result;
  }
  if (aggType !== 'count' && termField) {
    const result = {
      bucketAgg: {
        terms: {
          field: termField,
          size: termSize,
          order: {
            metricAgg: termOrder
          }
        },
        aggs: {
          metricAgg: {}
        }
      }
    };
    (0, _saferLodashSet.set)(result, `bucketAgg.aggs.metricAgg.${aggType}`, {
      field: aggField
    });
    return result;
  }
}

/*
watch.input.search.request.body
 */
function buildBody(timeWindowSize, timeWindowUnit, timeField, aggType, aggField, termField, termSize, termOrder) {
  const result = {
    size: 0,
    query: buildQuery(timeWindowSize, timeWindowUnit, timeField)
  };
  const aggs = buildAggs(aggType, aggField, termField, termSize, termOrder);
  if (Boolean(aggs)) {
    result.aggs = aggs;
  }
  return result;
}

/*
watch.input
 */
function buildInput({
  index,
  timeWindowSize,
  timeWindowUnit,
  timeField,
  aggType,
  aggField,
  termField,
  termSize,
  termOrder
}) {
  return {
    search: {
      request: {
        body: buildBody(timeWindowSize, timeWindowUnit, timeField, aggType, aggField, termField, termSize, termOrder),
        indices: buildIndices(index)
      }
    }
  };
}