Spike for constraints

Cargo.toml

@@ -42,6 +42,8 @@ rand = "0.8.4"
 serde_json = "1.0.68"
 serde_plain = "1.0.0"
 surf = "2.3.1"
+semver = "1.0.5"
+enum_dispatch = "0.3.8"
 
 [dependencies.chrono]
 features = ["serde"]

src/api.rs

@@ -4,8 +4,11 @@ use std::collections::HashMap;
 use std::default::Default;
 
 use chrono::Utc;
+use enum_dispatch::enum_dispatch;
 use serde::{Deserialize, Serialize};
 
+use crate::{Context, Evaluate};
+
 #[derive(Serialize, Deserialize, Debug)]
 // #[serde(deny_unknown_fields)]
 pub struct Features {
@@ -40,24 +43,117 @@ pub struct Strategy {
     pub parameters: Option<HashMap<String, String>>,
 }
 
+#[derive(Clone, Serialize, Deserialize, Debug)]
+pub enum StrConstraint {
+    StrEndsWith(Vec<String>, bool),
+    StrStartsWith(Vec<String>, bool),
+    StrContains(Vec<String>, bool),
+}
+
+#[derive(Clone, Serialize, Deserialize, Debug)]
+pub enum NumericConstraint {
+    NumEq(String),
+    NumGt(String),
+    NumGte(String),
+    NumLt(String),
+    NumLte(String),
+}
+
+#[derive(Clone, Serialize, Deserialize, Debug)]
+pub enum DateConstraint {
+    DateAfter(String),
+    DateBefore(String),
+}
+
+#[derive(Clone, Serialize, Deserialize, Debug)]
+pub enum SemverConstraint {
+    SemverEq(String),
+    SemverGt(String),
+    SemverLt(String),
+}
+
+#[derive(Clone, Serialize, Deserialize, Debug)]
+pub enum ContainsConstraint {
+    In(Vec<String>),
+    NotIn(Vec<String>),
+}
+
+#[enum_dispatch]
+#[derive(Clone, Serialize, Deserialize, Debug)]
+pub enum ConstraintExpression {
+    NumericConstraint,
+    StrConstraint,
+    DateConstraint,
+    SemverConstraint,
+    ContainsConstraint,
+}
+
+#[enum_dispatch(ConstraintExpression)]
+pub trait EvaluatorConstructor {
+    fn yield_evaluator(self, getter: crate::strategy::Expr) -> Evaluate;
+}
+
+pub trait Expression: Fn(&Context) -> Option<&String> {
+    fn clone_boxed(&self) -> Box<dyn Expression + Send + Sync + 'static>;
+}
+
 #[derive(Clone, Serialize, Deserialize, Debug)]
 // #[serde(deny_unknown_fields)]
 pub struct Constraint {
     #[serde(rename = "contextName")]
     pub context_name: String,
+    pub inverted: Option<bool>,
     #[serde(flatten)]
     pub expression: ConstraintExpression,
 }
 
-#[derive(Clone, Serialize, Deserialize, Debug)]
-#[serde(tag = "operator", content = "values")]
-// #[serde(deny_unknown_fields)]
-pub enum ConstraintExpression {
-    #[serde(rename = "IN")]
-    In(Vec<String>),
-    #[serde(rename = "NOT_IN")]
-    NotIn(Vec<String>),
-}
+// #[derive(Clone, Serialize, Deserialize, Debug)]
+// #[serde(tag = "operator")]
+// // #[serde(deny_unknown_fields)]
+// pub enum ConstraintExpression {
+//     #[serde(rename = "IN")]
+//     In { values: Vec<String> },
+//     #[serde(rename = "NOT_IN")]
+//     NotIn { values: Vec<String> },
+//     #[serde(rename = "STR_ENDS_WITH")]
+//     StrEndsWith {
+//         values: Vec<String>,
+//         #[serde(rename = "caseInsensitive")]
+//         case_insensitive: Option<bool>,
+//     },
+//     #[serde(rename = "STR_STARTS_WITH")]
+//     StrStartsWith {
+//         values: Vec<String>,
+//         #[serde(rename = "caseInsensitive")]
+//         case_insensitive: Option<bool>,
+//     },
+//     #[serde(rename = "STR_CONTAINS")]
+//     StrContains {
+//         values: Vec<String>,
+//         #[serde(rename = "caseInsensitive")]
+//         case_insensitive: Option<bool>,
+//     },
+//     #[serde(rename = "NUM_EQ")]
+//     NumEq { value: String },
+//     #[serde(rename = "NUM_GT")]
+//     NumGt { value: String },
+//     #[serde(rename = "NUM_GTE")]
+//     NumGte { value: String },
+//     #[serde(rename = "NUM_LT")]
+//     NumLt { value: String },
+//     #[serde(rename = "NUM_LTE")]
+//     NumLte { value: String },
+//     #[serde(rename = "DATE_AFTER")]
+//     DateAfter { value: String },
+//     #[serde(rename = "DATE_BEFORE")]
+//     DateBefore { value: String },
+//     #[serde(rename = "SEMVER_EQ")]
+//     SemverEq { value: String },
+//     #[serde(rename = "SEMVER_GT")]
+//     SemverGt { value: String },
+//     #[serde(rename = "SEMVER_LT")]
+//     SemverLt { value: String },
+// }
 
 #[derive(Clone, Serialize, Deserialize, Debug)]
 // #[serde(deny_unknown_fields)]
@@ -134,6 +230,38 @@ pub struct MetricsBucket {
 #[cfg(test)]
 mod tests {
     use super::Registration;
+    use super::*;
+
+    #[test]
+    fn parses_advanced_constraint_structure() -> Result<(), serde_json::Error> {
+        let data = r#"
+    {
+      "contextName": "customField",
+      "operator": "STR_STARTS_WITH",
+      "values": ["some-string"]
+    }"#;
+        let _: super::Constraint = serde_json::from_str(data)?;
+
+        let data = r#"
+    {
+      "contextName": "customField",
+      "operator": "NUM_GTE",
+      "value": "7"
+    }"#;
+        let _: super::Constraint = serde_json::from_str(data)?;
+
+        let data = r#"
+    {
+      "contextName": "customField",
+      "operator": "STR_STARTS_WITH",
+      "values": ["some-string"],
+      "caseInsensitive": true,
+      "inverted": true
+    }"#;
+        let _: super::Constraint = serde_json::from_str(data)?;
+
+        Ok(())
+    }
 
     #[test]
     fn parse_reference_doc() -> Result<(), serde_json::Error> {

src/context.rs

@@ -2,6 +2,7 @@
 //! <https://docs.getunleash.io/user_guide/unleash_context>
 use std::{collections::HashMap, net::IpAddr};
 
+use chrono::Utc;
 use serde::{de, Deserialize};
 
 // Custom IP Address newtype that can be deserialised from strings e.g. 127.0.0.1 for use with tests.
@@ -40,4 +41,10 @@ pub struct Context {
     pub app_name: String,
     #[serde(default)]
     pub environment: String,
+    #[serde(default = "current_time", rename = "currentTime")]
+    pub current_time: String,
+}
+
+fn current_time() -> String {
+    Utc::now().to_rfc3339()
 }