From 780ca2a819a6d1c931580c8e8e31657072971876 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@archive.org>
Date: Mon, 24 Aug 2020 14:13:56 -0700
Subject: more progress and tests

---
 src/lib.rs | 94 +++++++++++++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 74 insertions(+), 20 deletions(-)

(limited to 'src/lib.rs')

diff --git a/src/lib.rs b/src/lib.rs
index 615e4ed..7d26360 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -52,6 +52,7 @@ URL query parameters:
 
 // https://www.elastic.co/guide/en/elasticsearch/reference/current/search-search.html
 #[derive(Serialize, Deserialize, Debug)]
+#[serde(deny_unknown_fields)]
 pub struct SearchBody {
     pub query: Option<ApiQuery>,
     pub highlight: Option<ApiHighlight>,
@@ -61,13 +62,13 @@ pub struct SearchBody {
     // script_fields disabled
 
     // https://www.elastic.co/guide/en/elasticsearch/reference/current/sort-search-results.html
-    // pub sort: Option<Vec<SortThing>>, // XXX: array of object or string
+    pub sort: Option<Vec<SortElement>>,
 
     pub slice: Option<ApiSlice>,
     pub stored_fields: Option<String>, // array of strings, or "_none_"
 
     // overlap with URL query parameters
-    pub docvalue_fields: Option<Vec<String>>, // XXX: array of strings or objects? {field, format} in the objects
+    pub docvalue_fields: Option<Vec<DocValOrString>>,
     pub explain: Option<bool>,
     pub from: Option<u32>,
     pub min_score: Option<f64>,
@@ -80,6 +81,13 @@ pub struct SearchBody {
     // not enumerated in direct docs, but seems to be allowed?
 }
 
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(deny_unknown_fields)]
+pub struct ScrollBody {
+    pub scroll_id: String,
+    pub scroll: Option<String>,
+}
+
 #[derive(Serialize, Deserialize, Debug)]
 pub struct ApiSlice {
     id: u32,
@@ -93,31 +101,26 @@ pub struct ApiRescore{
     pub window_size: Option<u32>,
 }
 
+// TODO: could revert to having query types as an enum, with flattening
 #[derive(Serialize, Deserialize, Debug)]
-pub enum ApiQuery {
+pub struct ApiQuery {
     // compound queries
     #[serde(rename = "bool")]
-    Bool(HashMap<String, QueryFieldOrString>),
-    #[serde(rename = "boosting")]
-    Boosting {positive: Box<ApiQuery>, negative: Box<ApiQuery>, negative_boost: f64},
-    #[serde(rename = "constant_score")]
-    ConstantScore {filter: Box<ApiQuery>, boost: Option<f64>},
+    bool_query: Option<BoolQuery>,
+    boosting: Option<BoostingQuery>,
+    constant_score: Option<ConstantScoreQuery>,
 
     // fulltext (leaf) queries
 
     // term-level (leaf) queries
     #[serde(rename = "match")]
-    Match(HashMap<String, QueryFieldOrString>),
-    #[serde(rename = "match_phrase")]
-    MatchPhrase(HashMap<String, QueryFieldOrString>),
-    #[serde(rename = "query_string")]
-    QueryString(QueryField),
+    match_query: Option<HashMap<String, QueryFieldOrString>>,
+    match_phrase: Option<HashMap<String, QueryFieldOrString>>,
+    query_string: Option<QueryField>,
 
     // other
-    #[serde(rename = "nested")]
-    Nested(NestedQuery),
-    #[serde(rename = "rescore_query")]
-    Rescore(Box<ApiQuery>),
+    nested: Option<NestedQuery>,
+    rescore_query: Option<Box<ApiQuery>>,
 }
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -129,6 +132,27 @@ pub struct ApiHighlight{
     settings: HighlightField,
 }
 
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(untagged)]
+pub enum SortMapValue {
+    String(String),
+    Object { order: String, mode: Option<String> },
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(untagged)]
+pub enum SortElement{
+    String(String),
+    Object(HashMap<String, SortMapValue>),
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(untagged)]
+pub enum DocValOrString {
+    String(String),
+    Object {field: String, format: Option<String>},
+}
+
 #[derive(Serialize, Deserialize, Debug)]
 #[serde(untagged)]
 pub enum QueryFieldOrString {
@@ -145,13 +169,36 @@ pub struct QueryField{
 }
 
 #[derive(Serialize, Deserialize, Debug)]
-pub struct NestedQuery{
+pub struct BoolQuery {
+    must: Option<Box<ApiQuery>>,
+    filter: Option<Box<ApiQuery>>,
+    should: Option<Box<ApiQuery>>,
+    must_not: Option<Box<ApiQuery>>,
+    minimum_should_match: Option<u32>,
+    boost: Option<f64>,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+pub struct NestedQuery {
     path: String,
     query: Box<ApiQuery>,
     score_mode: Option<String>,
     ignore_unmapped: Option<bool>,
 }
 
+#[derive(Serialize, Deserialize, Debug)]
+pub struct BoostingQuery {
+    positive: Box<ApiQuery>,
+    negative: Box<ApiQuery>,
+    negative_boost: f64,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+pub struct ConstantScoreQuery {
+    filter: Box<ApiQuery>,
+    boost: Option<f64>,
+}
+
 // https://www.elastic.co/guide/en/elasticsearch/reference/current/highlighting.html
 #[derive(Serialize, Deserialize, Debug)]
 pub struct HighlightField{
@@ -181,13 +228,20 @@ pub struct HighlightField{
 #[derive(Serialize, Deserialize, Debug)]
 pub struct ApiCollapse{
     field: String,
-    inner_hits: Option<InnerHits>,
+    inner_hits: Option<InnerHitsOneOrMore>,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(untagged)]
+pub enum InnerHitsOneOrMore {
+    Single(InnerHits),
+    Multiple(Vec<InnerHits>),
 }
 
 #[derive(Serialize, Deserialize, Debug)]
 pub struct InnerHits {
     from: Option<u32>,
     size: Option<u32>,
-    sort: Option<Vec<String>>,
+    sort: Option<Vec<SortElement>>,
     name: Option<String>,
 }
-- 
cgit v1.2.3