energylink/public/app/plugins/datasource/elasticsearch/query_def.ts

///<reference path="../../../headers/common.d.ts" />

import _ from "lodash";

export const metricAggTypes = [
  { text: "Count", value: "count", requiresField: false },
  {
    text: "Average",
    value: "avg",
    requiresField: true,
    supportsInlineScript: true,
    supportsMissing: true
  },
  {
    text: "Sum",
    value: "sum",
    requiresField: true,
    supportsInlineScript: true,
    supportsMissing: true
  },
  {
    text: "Max",
    value: "max",
    requiresField: true,
    supportsInlineScript: true,
    supportsMissing: true
  },
  {
    text: "Min",
    value: "min",
    requiresField: true,
    supportsInlineScript: true,
    supportsMissing: true
  },
  {
    text: "Extended Stats",
    value: "extended_stats",
    requiresField: true,
    supportsMissing: true,
    supportsInlineScript: true
  },
  {
    text: "Percentiles",
    value: "percentiles",
    requiresField: true,
    supportsMissing: true,
    supportsInlineScript: true
  },
  {
    text: "Unique Count",
    value: "cardinality",
    requiresField: true,
    supportsMissing: true
  },
  {
    text: "Moving Average",
    value: "moving_avg",
    requiresField: false,
    isPipelineAgg: true,
    minVersion: 2
  },
  {
    text: "Derivative",
    value: "derivative",
    requiresField: false,
    isPipelineAgg: true,
    minVersion: 2
  },
  { text: "Raw Document", value: "raw_document", requiresField: false }
];

export const bucketAggTypes = [
  { text: "Terms", value: "terms", requiresField: true },
  { text: "Filters", value: "filters" },
  { text: "Geo Hash Grid", value: "geohash_grid", requiresField: true },
  { text: "Date Histogram", value: "date_histogram", requiresField: true },
  { text: "Histogram", value: "histogram", requiresField: true }
];

export const orderByOptions = [
  { text: "Doc Count", value: "_count" },
  { text: "Term value", value: "_term" }
];

export const orderOptions = [
  { text: "Top", value: "desc" },
  { text: "Bottom", value: "asc" }
];

export const sizeOptions = [
  { text: "No limit", value: "0" },
  { text: "1", value: "1" },
  { text: "2", value: "2" },
  { text: "3", value: "3" },
  { text: "5", value: "5" },
  { text: "10", value: "10" },
  { text: "15", value: "15" },
  { text: "20", value: "20" }
];

export const extendedStats = [
  { text: "Avg", value: "avg" },
  { text: "Min", value: "min" },
  { text: "Max", value: "max" },
  { text: "Sum", value: "sum" },
  { text: "Count", value: "count" },
  { text: "Std Dev", value: "std_deviation" },
  { text: "Std Dev Upper", value: "std_deviation_bounds_upper" },
  { text: "Std Dev Lower", value: "std_deviation_bounds_lower" }
];

export const intervalOptions = [
  { text: "auto", value: "auto" },
  { text: "10s", value: "10s" },
  { text: "1m", value: "1m" },
  { text: "5m", value: "5m" },
  { text: "10m", value: "10m" },
  { text: "20m", value: "20m" },
  { text: "1h", value: "1h" },
  { text: "1d", value: "1d" }
];

export const movingAvgModelOptions = [
  { text: "Simple", value: "simple" },
  { text: "Linear", value: "linear" },
  { text: "Exponentially Weighted", value: "ewma" },
  { text: "Holt Linear", value: "holt" },
  { text: "Holt Winters", value: "holt_winters" }
];

export const pipelineOptions = {
  moving_avg: [
    { text: "window", default: 5 },
    { text: "model", default: "simple" },
    { text: "predict", default: undefined },
    { text: "minimize", default: false }
  ],
  derivative: [{ text: "unit", default: undefined }]
};

export const movingAvgModelSettings = {
  simple: [],
  linear: [],
  ewma: [{ text: "Alpha", value: "alpha", default: undefined }],
  holt: [
    { text: "Alpha", value: "alpha", default: undefined },
    { text: "Beta", value: "beta", default: undefined }
  ],
  holt_winters: [
    { text: "Alpha", value: "alpha", default: undefined },
    { text: "Beta", value: "beta", default: undefined },
    { text: "Gamma", value: "gamma", default: undefined },
    { text: "Period", value: "period", default: undefined },
    { text: "Pad", value: "pad", default: undefined, isCheckbox: true }
  ]
};

export function getMetricAggTypes(esVersion) {
  return _.filter(metricAggTypes, function(f) {
    if (f.minVersion) {
      return f.minVersion <= esVersion;
    } else {
      return true;
    }
  });
}

export function getPipelineOptions(metric) {
  if (!isPipelineAgg(metric.type)) {
    return [];
  }

  return pipelineOptions[metric.type];
}

export function isPipelineAgg(metricType) {
  if (metricType) {
    var po = pipelineOptions[metricType];
    return po !== null && po !== undefined;
  }

  return false;
}

export function getPipelineAggOptions(targets) {
  var result = [];
  _.each(targets.metrics, function(metric) {
    if (!isPipelineAgg(metric.type)) {
      result.push({ text: describeMetric(metric), value: metric.id });
    }
  });

  return result;
}

export function getMovingAvgSettings(model, filtered) {
  var filteredResult = [];
  if (filtered) {
    _.each(movingAvgModelSettings[model], function(setting) {
      if (!setting.isCheckbox) {
        filteredResult.push(setting);
      }
    });
    return filteredResult;
  }
  return movingAvgModelSettings[model];
}

export function getOrderByOptions(target) {
  var metricRefs = [];
  _.each(target.metrics, function(metric) {
    if (metric.type !== "count") {
      metricRefs.push({ text: describeMetric(metric), value: metric.id });
    }
  });

  return orderByOptions.concat(metricRefs);
}

export function describeOrder(order) {
  var def = _.find(orderOptions, { value: order });
  return def.text;
}

export function describeMetric(metric) {
  var def = _.find(metricAggTypes, { value: metric.type });
  return def.text + " " + metric.field;
}

export function describeOrderBy(orderBy, target) {
  var def = _.find(orderByOptions, { value: orderBy });
  if (def) {
    return def.text;
  }
  var metric = _.find(target.metrics, { id: orderBy });
  if (metric) {
    return describeMetric(metric);
  } else {
    return "metric not found";
  }
}