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

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

import "./bucket_agg";
import "./metric_agg";

import angular from "angular";
import _ from "lodash";
import * as queryDef from "./query_def";
import { QueryCtrl } from "app/plugins/sdk";

export class ElasticQueryCtrl extends QueryCtrl {
  static templateUrl = "partials/query.editor.html";

  esVersion: any;
  rawQueryOld: string;

  /** @ngInject **/
  constructor($scope, $injector, private $rootScope, private uiSegmentSrv) {
    super($scope, $injector);

    this.esVersion = this.datasource.esVersion;
    this.queryUpdated();
  }

  getFields(type) {
    var jsonStr = angular.toJson({ find: "fields", type: type });
    return this.datasource
      .metricFindQuery(jsonStr)
      .then(this.uiSegmentSrv.transformToSegments(false))
      .catch(this.handleQueryError.bind(this));
  }

  queryUpdated() {
    var newJson = angular.toJson(
      this.datasource.queryBuilder.build(this.target),
      true
    );
    if (this.rawQueryOld && newJson !== this.rawQueryOld) {
      this.refresh();
    }

    this.rawQueryOld = newJson;
    this.$rootScope.appEvent("elastic-query-updated");
  }

  getCollapsedText() {
    var metricAggs = this.target.metrics;
    var bucketAggs = this.target.bucketAggs;
    var metricAggTypes = queryDef.getMetricAggTypes(this.esVersion);
    var bucketAggTypes = queryDef.bucketAggTypes;
    var text = "";

    if (this.target.query) {
      text += "Query: " + this.target.query + ", ";
    }

    text += "Metrics: ";

    _.each(metricAggs, (metric, index) => {
      var aggDef = _.find(metricAggTypes, { value: metric.type });
      text += aggDef.text + "(";
      if (aggDef.requiresField) {
        text += metric.field;
      }
      text += "), ";
    });

    _.each(bucketAggs, (bucketAgg, index) => {
      if (index === 0) {
        text += " Group by: ";
      }

      var aggDef = _.find(bucketAggTypes, { value: bucketAgg.type });
      text += aggDef.text + "(";
      if (aggDef.requiresField) {
        text += bucketAgg.field;
      }
      text += "), ";
    });

    if (this.target.alias) {
      text += "Alias: " + this.target.alias;
    }

    return text;
  }

  handleQueryError(err) {
    this.error = err.message || "Failed to issue metric query";
    return [];
  }
}