From ea372f042788f61a53ebbca3486090cc8577a20a Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@archive.org>
Date: Thu, 17 Dec 2020 12:13:02 -0800
Subject: add support for 'filters' aggregation

---
 src/parse.rs                                        | 18 ++++++++++++++++++
 tests/files/search/GET_search_agg_filters.txt       | 14 ++++++++++++++
 tests/files/search/GET_search_agg_filters_anon.txt  | 14 ++++++++++++++
 tests/files/search/GET_search_agg_filters_other.txt | 15 +++++++++++++++
 4 files changed, 61 insertions(+)
 create mode 100644 tests/files/search/GET_search_agg_filters.txt
 create mode 100644 tests/files/search/GET_search_agg_filters_anon.txt
 create mode 100644 tests/files/search/GET_search_agg_filters_other.txt

diff --git a/src/parse.rs b/src/parse.rs
index 8656daa..07c55a0 100644
--- a/src/parse.rs
+++ b/src/parse.rs
@@ -239,6 +239,14 @@ pub enum ApiQueryOrArray {
     Array(Vec<ApiQuery>),
 }
 
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(deny_unknown_fields)]
+#[serde(untagged)]
+pub enum ApiQueryNamedOrArray {
+    Named(HashMap<String, ApiQuery>),
+    Array(Vec<ApiQuery>),
+}
+
 #[derive(Serialize, Deserialize, Debug)]
 #[serde(deny_unknown_fields)]
 pub struct ApiHighlight {
@@ -644,6 +652,8 @@ pub struct ApiAggregation {
     #[serde(skip_serializing_if = "Option::is_none")]
     filter: Option<ApiQuery>,
     #[serde(skip_serializing_if = "Option::is_none")]
+    filters: Option<FiltersAggregation>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     histogram: Option<SimpleAggregation>,
     #[serde(skip_serializing_if = "Option::is_none")]
     terms: Option<TermsAggregation>,
@@ -713,6 +723,14 @@ pub struct DateHistogramAggregation {
     missing: Option<StringOrNum>,
 }
 
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(deny_unknown_fields)]
+pub struct FiltersAggregation {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    other_bucket_key: Option<String>,
+    filters: ApiQueryNamedOrArray,
+}
+
 #[derive(Serialize, Deserialize, Debug)]
 #[serde(deny_unknown_fields)]
 pub struct TermsAggregation {
diff --git a/tests/files/search/GET_search_agg_filters.txt b/tests/files/search/GET_search_agg_filters.txt
new file mode 100644
index 0000000..74b2976
--- /dev/null
+++ b/tests/files/search/GET_search_agg_filters.txt
@@ -0,0 +1,14 @@
+GET /logs/_search
+{
+  "size": 0,
+  "aggs" : {
+    "messages" : {
+      "filters" : {
+        "filters" : {
+          "errors" :   { "match" : { "body" : "error"   }},
+          "warnings" : { "match" : { "body" : "warning" }}
+        }
+      }
+    }
+  }
+}
diff --git a/tests/files/search/GET_search_agg_filters_anon.txt b/tests/files/search/GET_search_agg_filters_anon.txt
new file mode 100644
index 0000000..baa550d
--- /dev/null
+++ b/tests/files/search/GET_search_agg_filters_anon.txt
@@ -0,0 +1,14 @@
+GET /logs/_search
+{
+  "size": 0,
+  "aggs" : {
+    "messages" : {
+      "filters" : {
+        "filters" : [
+          { "match" : { "body" : "error"   }},
+          { "match" : { "body" : "warning" }}
+        ]
+      }
+    }
+  }
+}
diff --git a/tests/files/search/GET_search_agg_filters_other.txt b/tests/files/search/GET_search_agg_filters_other.txt
new file mode 100644
index 0000000..dd54e48
--- /dev/null
+++ b/tests/files/search/GET_search_agg_filters_other.txt
@@ -0,0 +1,15 @@
+GET /logs/_search
+{
+  "size": 0,
+  "aggs" : {
+    "messages" : {
+      "filters" : {
+        "other_bucket_key": "other_messages",
+        "filters" : {
+          "errors" :   { "match" : { "body" : "error"   }},
+          "warnings" : { "match" : { "body" : "warning" }}
+        }
+      }
+    }
+  }
+}
-- 
cgit v1.2.3