"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.getLogstashPipelineIds = getLogstashPipelineIds;
var _moment = _interopRequireDefault(require("moment"));
var _lodash = require("lodash");
var _create_query = require("../create_query");
var _metrics = require("../metrics");
var _get_index_patterns = require("../cluster/get_index_patterns");
var _static_globals = require("../../static_globals");
/*
 * 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.
 */

async function getLogstashPipelineIds({
  req,
  clusterUuid,
  logstashUuid,
  size,
  ccs
}) {
  const start = _moment.default.utc(req.payload.timeRange.min).valueOf();
  const end = _moment.default.utc(req.payload.timeRange.max).valueOf();
  const filters = [];
  if (logstashUuid) {
    filters.push({
      term: {
        'logstash_stats.logstash.uuid': logstashUuid
      }
    });
  }
  const dataset = 'node_stats';
  const moduleType = 'logstash';
  const indexPatterns = (0, _get_index_patterns.getIndexPatterns)({
    config: _static_globals.Globals.app.config,
    ccs: ccs || req.payload.ccs,
    moduleType,
    dataset
  });
  const params = {
    index: indexPatterns,
    size: 0,
    ignore_unavailable: true,
    filter_path: ['aggregations.nest.id.buckets', 'aggregations.nest_mb.id.buckets'],
    body: {
      query: (0, _create_query.createQuery)({
        start,
        end,
        metric: _metrics.LogstashMetric.getMetricFields(),
        clusterUuid,
        filters
      }),
      aggs: {
        nest: {
          nested: {
            path: 'logstash_stats.pipelines'
          },
          aggs: {
            id: {
              terms: {
                field: 'logstash_stats.pipelines.id',
                size
              },
              aggs: {
                unnest: {
                  reverse_nested: {},
                  aggs: {
                    nodes: {
                      terms: {
                        field: 'logstash_stats.logstash.uuid',
                        size
                      }
                    }
                  }
                }
              }
            }
          }
        },
        nest_mb: {
          nested: {
            path: 'logstash.node.stats.pipelines'
          },
          aggs: {
            id: {
              terms: {
                field: 'logstash.node.stats.pipelines.id',
                size
              },
              aggs: {
                unnest_mb: {
                  reverse_nested: {},
                  aggs: {
                    nodes: {
                      terms: {
                        field: 'logstash.node.stats.logstash.uuid',
                        size
                      }
                    }
                  }
                }
              }
            }
          }
        }
      }
    }
  };
  const {
    callWithRequest
  } = req.server.plugins.elasticsearch.getCluster('monitoring');
  const response = await callWithRequest(req, 'search', params);
  let buckets = (0, _lodash.get)(response, 'aggregations.nest_mb.id.buckets', []);
  if (!buckets || buckets.length === 0) {
    buckets = (0, _lodash.get)(response, 'aggregations.nest.id.buckets', []);
  }
  return buckets.map(bucket => {
    let nodeBuckets = (0, _lodash.get)(bucket, 'unnest_mb.nodes.buckets', []);
    if (!nodeBuckets || nodeBuckets.length === 0) {
      nodeBuckets = (0, _lodash.get)(bucket, 'unnest.nodes.buckets', []);
    }
    return {
      id: bucket.key,
      nodeIds: nodeBuckets.map(item => item.key)
    };
  });
}