From 7cb0403faa07ea66c35e1cafae0661c4726bc3a5 Mon Sep 17 00:00:00 2001
From: Marcus Efraimsson <marcus.efraimsson@gmail.com>
Date: Mon, 28 May 2018 19:45:18 +0200
Subject: [PATCH] elasticsearch: handle if alert query contains template
 variable

If datasource handles targetContainsTemplate function it can evaluate if a certain
query contains template variables and this is used for show an error message that
template variables not is supported in alert queries.
---
 .../datasource/elasticsearch/datasource.ts    | 61 +++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/public/app/plugins/datasource/elasticsearch/datasource.ts b/public/app/plugins/datasource/elasticsearch/datasource.ts
index e3eccfb8029..5a8e83a16cb 100644
--- a/public/app/plugins/datasource/elasticsearch/datasource.ts
+++ b/public/app/plugins/datasource/elasticsearch/datasource.ts
@@ -408,4 +408,65 @@ export class ElasticDatasource {
   getTagValues(options) {
     return this.getTerms({ field: options.key, query: '*' });
   }
+
+  targetContainsTemplate(target) {
+    if (this.templateSrv.variableExists(target.query) || this.templateSrv.variableExists(target.alias)) {
+      return true;
+    }
+
+    for (let bucketAgg of target.bucketAggs) {
+      if (this.templateSrv.variableExists(bucketAgg.field) || this.objectContainsTemplate(bucketAgg.settings)) {
+        return true;
+      }
+    }
+
+    for (let metric of target.metrics) {
+      if (
+        this.templateSrv.variableExists(metric.field) ||
+        this.objectContainsTemplate(metric.settings) ||
+        this.objectContainsTemplate(metric.meta)
+      ) {
+        return true;
+      }
+    }
+
+    return false;
+  }
+
+  private isPrimitive(obj) {
+    if (obj === null || obj === undefined) {
+      return true;
+    }
+    if (['string', 'number', 'boolean'].some(type => type === typeof true)) {
+      return true;
+    }
+
+    return false;
+  }
+
+  private objectContainsTemplate(obj) {
+    if (!obj) {
+      return false;
+    }
+
+    for (let key of Object.keys(obj)) {
+      if (this.isPrimitive(obj[key])) {
+        if (this.templateSrv.variableExists(obj[key])) {
+          return true;
+        }
+      } else if (Array.isArray(obj[key])) {
+        for (let item of obj[key]) {
+          if (this.objectContainsTemplate(item)) {
+            return true;
+          }
+        }
+      } else {
+        if (this.objectContainsTemplate(obj[key])) {
+          return true;
+        }
+      }
+    }
+
+    return false;
+  }
 }