#![allow(clippy::derive_ord_xor_partial_ord)] use crate::ctx::Context; use crate::dbs::Options; use crate::dbs::Transaction; use crate::err::Error; use crate::sql::array::{array, Array}; use crate::sql::block::{block, Block}; use crate::sql::common::commas; use crate::sql::constant::{constant, Constant}; use crate::sql::datetime::{datetime, Datetime}; use crate::sql::duration::{duration, Duration}; use crate::sql::edges::{edges, Edges}; use crate::sql::error::IResult; use crate::sql::expression::{expression, Expression}; use crate::sql::fmt::{Fmt, Pretty}; use crate::sql::function::{function, Function}; use crate::sql::future::{future, Future}; use crate::sql::geometry::{geometry, Geometry}; use crate::sql::id::Id; use crate::sql::idiom::{idiom, Idiom}; use crate::sql::kind::Kind; use crate::sql::model::{model, Model}; use crate::sql::number::{number, Number}; use crate::sql::object::{object, Object}; use crate::sql::operation::Operation; use crate::sql::param::{param, Param}; use crate::sql::part::Part; use crate::sql::range::{range, Range}; use crate::sql::regex::{regex, Regex}; use crate::sql::serde::is_internal_serialization; use crate::sql::strand::{strand, Strand}; use crate::sql::subquery::{subquery, Subquery}; use crate::sql::table::{table, Table}; use crate::sql::thing::{thing, Thing}; use crate::sql::uuid::{uuid as unique, Uuid}; use async_recursion::async_recursion; use bigdecimal::BigDecimal; use chrono::{DateTime, Utc}; use derive::Store; use fuzzy_matcher::skim::SkimMatcherV2; use fuzzy_matcher::FuzzyMatcher; use geo::Point; use nom::branch::alt; use nom::bytes::complete::tag_no_case; use nom::combinator::map; use nom::multi::separated_list1; use once_cell::sync::Lazy; use serde::{Deserialize, Serialize}; use std::cmp::Ordering; use std::collections::BTreeMap; use std::collections::HashMap; use std::fmt::{self, Display, Formatter, Write}; use std::ops; use std::ops::Deref; use std::str::FromStr; static MATCHER: Lazy = Lazy::new(|| SkimMatcherV2::default().ignore_case()); #[derive(Clone, Debug, Default, Eq, PartialEq, Serialize, Deserialize, Hash)] pub struct Values(pub Vec); impl Deref for Values { type Target = Vec; fn deref(&self) -> &Self::Target { &self.0 } } impl IntoIterator for Values { type Item = Value; type IntoIter = std::vec::IntoIter; fn into_iter(self) -> Self::IntoIter { self.0.into_iter() } } impl Display for Values { fn fmt(&self, f: &mut Formatter) -> fmt::Result { Display::fmt(&Fmt::comma_separated(&self.0), f) } } pub fn values(i: &str) -> IResult<&str, Values> { let (i, v) = separated_list1(commas, value)(i)?; Ok((i, Values(v))) } pub fn selects(i: &str) -> IResult<&str, Values> { let (i, v) = separated_list1(commas, select)(i)?; Ok((i, Values(v))) } pub fn whats(i: &str) -> IResult<&str, Values> { let (i, v) = separated_list1(commas, what)(i)?; Ok((i, Values(v))) } #[derive(Clone, Debug, PartialEq, PartialOrd, Deserialize, Store, Hash)] pub enum Value { None, Null, False, True, Number(Number), Strand(Strand), Duration(Duration), Datetime(Datetime), Uuid(Uuid), Array(Array), Object(Object), Geometry(Geometry), // --- Param(Param), Idiom(Idiom), Table(Table), Thing(Thing), Model(Model), Regex(Regex), Block(Box), Range(Box), Edges(Box), Future(Box), Constant(Constant), Function(Box), Subquery(Box), Expression(Box), } impl Eq for Value {} impl Ord for Value { fn cmp(&self, other: &Self) -> Ordering { self.partial_cmp(other).unwrap_or(Ordering::Equal) } } impl Default for Value { fn default() -> Value { Value::None } } impl From for Value { #[inline] fn from(v: bool) -> Self { match v { true => Value::True, false => Value::False, } } } impl From for Value { fn from(v: Uuid) -> Self { Value::Uuid(v) } } impl From for Value { fn from(v: Param) -> Self { Value::Param(v) } } impl From for Value { fn from(v: Idiom) -> Self { Value::Idiom(v) } } impl From for Value { fn from(v: Model) -> Self { Value::Model(v) } } impl From for Value { fn from(v: Table) -> Self { Value::Table(v) } } impl From for Value { fn from(v: Thing) -> Self { Value::Thing(v) } } impl From for Value { fn from(v: Regex) -> Self { Value::Regex(v) } } impl From for Value { fn from(v: Array) -> Self { Value::Array(v) } } impl From for Value { fn from(v: Object) -> Self { Value::Object(v) } } impl From for Value { fn from(v: Number) -> Self { Value::Number(v) } } impl From for Value { fn from(v: Strand) -> Self { Value::Strand(v) } } impl From for Value { fn from(v: Geometry) -> Self { Value::Geometry(v) } } impl From for Value { fn from(v: Datetime) -> Self { Value::Datetime(v) } } impl From for Value { fn from(v: Duration) -> Self { Value::Duration(v) } } impl From for Value { fn from(v: Constant) -> Self { Value::Constant(v) } } impl From for Value { fn from(v: Block) -> Self { Value::Block(Box::new(v)) } } impl From for Value { fn from(v: Range) -> Self { Value::Range(Box::new(v)) } } impl From for Value { fn from(v: Edges) -> Self { Value::Edges(Box::new(v)) } } impl From for Value { fn from(v: Future) -> Self { Value::Future(Box::new(v)) } } impl From for Value { fn from(v: Function) -> Self { Value::Function(Box::new(v)) } } impl From for Value { fn from(v: Subquery) -> Self { Value::Subquery(Box::new(v)) } } impl From for Value { fn from(v: Expression) -> Self { Value::Expression(Box::new(v)) } } impl From for Value { fn from(v: i8) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: i16) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: i32) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: i64) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: isize) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: u8) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: u16) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: u32) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: u64) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: usize) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: f32) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: f64) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: BigDecimal) -> Self { Value::Number(Number::from(v)) } } impl From for Value { fn from(v: String) -> Self { Value::Strand(Strand::from(v)) } } impl From<&str> for Value { fn from(v: &str) -> Self { Value::Strand(Strand::from(v)) } } impl From> for Value { fn from(v: DateTime) -> Self { Value::Datetime(Datetime::from(v)) } } impl From<(f64, f64)> for Value { fn from(v: (f64, f64)) -> Self { Value::Geometry(Geometry::from(v)) } } impl From<[f64; 2]> for Value { fn from(v: [f64; 2]) -> Self { Value::Geometry(Geometry::from(v)) } } impl From> for Value { fn from(v: Point) -> Self { Value::Geometry(Geometry::from(v)) } } impl From for Value { fn from(v: Operation) -> Self { Value::Object(Object::from(v)) } } impl From for Value { fn from(v: uuid::Uuid) -> Self { Value::Uuid(Uuid(v)) } } impl From> for Value { fn from(v: Vec<&str>) -> Self { Value::Array(Array::from(v)) } } impl From> for Value { fn from(v: Vec) -> Self { Value::Array(Array::from(v)) } } impl From> for Value { fn from(v: Vec) -> Self { Value::Array(Array::from(v)) } } impl From> for Value { fn from(v: Vec) -> Self { Value::Array(Array::from(v)) } } impl From> for Value { fn from(v: Vec) -> Self { Value::Array(Array::from(v)) } } impl From> for Value { fn from(v: Vec) -> Self { Value::Array(Array::from(v)) } } impl From> for Value { fn from(v: HashMap) -> Self { Value::Object(Object::from(v)) } } impl From> for Value { fn from(v: BTreeMap) -> Self { Value::Object(Object::from(v)) } } impl From> for Value { fn from(v: Option) -> Self { match v { Some(v) => v, None => Value::None, } } } impl From> for Value { fn from(v: Option) -> Self { match v { Some(v) => Value::from(v), None => Value::None, } } } impl From for Value { fn from(v: Id) -> Self { match v { Id::Number(v) => v.into(), Id::String(v) => v.into(), Id::Object(v) => v.into(), Id::Array(v) => v.into(), } } } impl TryFrom for i64 { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Number(x) => x.try_into(), _ => Err(Error::TryFromError(value.to_string(), "i64")), } } } impl TryFrom for f64 { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Number(x) => x.try_into(), _ => Err(Error::TryFromError(value.to_string(), "f64")), } } } impl TryFrom for BigDecimal { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Number(x) => x.try_into(), _ => Err(Error::TryFromError(value.to_string(), "BigDecimal")), } } } impl TryFrom for String { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Strand(x) => Ok(x.into()), _ => Err(Error::TryFromError(value.to_string(), "String")), } } } impl TryFrom for bool { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::True => Ok(true), Value::False => Ok(false), _ => Err(Error::TryFromError(value.to_string(), "bool")), } } } impl TryFrom for std::time::Duration { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Duration(x) => Ok(x.into()), _ => Err(Error::TryFromError(value.to_string(), "time::Duration")), } } } impl TryFrom for DateTime { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Datetime(x) => Ok(x.into()), _ => Err(Error::TryFromError(value.to_string(), "chrono::DateTime")), } } } impl TryFrom for uuid::Uuid { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Uuid(x) => Ok(x.into()), _ => Err(Error::TryFromError(value.to_string(), "uuid::Uuid")), } } } impl TryFrom for Vec { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Array(x) => Ok(x.into()), _ => Err(Error::TryFromError(value.to_string(), "Vec")), } } } impl TryFrom for Object { type Error = Error; fn try_from(value: Value) -> Result { match value { Value::Object(x) => Ok(x), _ => Err(Error::TryFromError(value.to_string(), "Object")), } } } impl Value { // ----------------------------------- // Initial record value // ----------------------------------- /// Create an empty Object Value pub fn base() -> Self { Value::Object(Object::default()) } // ----------------------------------- // Builtin types // ----------------------------------- /// Convert this Value to a Result pub fn ok(self) -> Result { Ok(self) } /// Convert this Value to an Option pub fn output(self) -> Option { match self { Value::None => None, _ => Some(self), } } // ----------------------------------- // Simple value detection // ----------------------------------- /// Check if this Value is NONE or NULL pub fn is_none_or_null(&self) -> bool { matches!(self, Value::None | Value::Null) } /// Check if this Value is NONE pub fn is_none(&self) -> bool { matches!(self, Value::None) } /// Check if this Value is NULL pub fn is_null(&self) -> bool { matches!(self, Value::Null) } /// Check if this Value not NONE or NULL pub fn is_some(&self) -> bool { !self.is_none() && !self.is_null() } /// Check if this Value is TRUE or 'true' pub fn is_true(&self) -> bool { match self { Value::True => true, Value::Strand(v) => v.eq_ignore_ascii_case("true"), _ => false, } } /// Check if this Value is FALSE or 'false' pub fn is_false(&self) -> bool { match self { Value::False => true, Value::Strand(v) => v.eq_ignore_ascii_case("false"), _ => false, } } /// Check if this Value is truthy pub fn is_truthy(&self) -> bool { match self { Value::True => true, Value::False => false, Value::Uuid(_) => true, Value::Thing(_) => true, Value::Geometry(_) => true, Value::Array(v) => !v.is_empty(), Value::Object(v) => !v.is_empty(), Value::Strand(v) => !v.is_empty() && !v.eq_ignore_ascii_case("false"), Value::Number(v) => v.is_truthy(), Value::Duration(v) => v.as_nanos() > 0, Value::Datetime(v) => v.timestamp() > 0, _ => false, } } /// Check if this Value is a UUID pub fn is_uuid(&self) -> bool { matches!(self, Value::Uuid(_)) } /// Check if this Value is a Thing pub fn is_thing(&self) -> bool { matches!(self, Value::Thing(_)) } /// Check if this Value is a Model pub fn is_model(&self) -> bool { matches!(self, Value::Model(_)) } /// Check if this Value is a Range pub fn is_range(&self) -> bool { matches!(self, Value::Range(_)) } /// Check if this Value is a Table pub fn is_table(&self) -> bool { matches!(self, Value::Table(_)) } /// Check if this Value is a Strand pub fn is_strand(&self) -> bool { matches!(self, Value::Strand(_)) } /// Check if this Value is an Array pub fn is_array(&self) -> bool { matches!(self, Value::Array(_)) } /// Check if this Value is an Object pub fn is_object(&self) -> bool { matches!(self, Value::Object(_)) } /// Check if this Value is a Number pub fn is_number(&self) -> bool { matches!(self, Value::Number(_)) } /// Check if this Value is an int Number pub fn is_int(&self) -> bool { matches!(self, Value::Number(Number::Int(_))) } /// Check if this Value is a float Number pub fn is_float(&self) -> bool { matches!(self, Value::Number(Number::Float(_))) } /// Check if this Value is a decimal Number pub fn is_decimal(&self) -> bool { matches!(self, Value::Number(Number::Decimal(_))) } /// Check if this Value is a Number and is an integer pub fn is_integer(&self) -> bool { matches!(self, Value::Number(v) if v.is_integer()) } /// Check if this Value is a Number and is positive pub fn is_positive(&self) -> bool { matches!(self, Value::Number(v) if v.is_positive()) } /// Check if this Value is a Number and is negative pub fn is_negative(&self) -> bool { matches!(self, Value::Number(v) if v.is_negative()) } /// Check if this Value is a Number and is zero or positive pub fn is_zero_or_positive(&self) -> bool { matches!(self, Value::Number(v) if v.is_zero_or_positive()) } /// Check if this Value is a Number and is zero or negative pub fn is_zero_or_negative(&self) -> bool { matches!(self, Value::Number(v) if v.is_zero_or_negative()) } /// Check if this Value is a Datetime pub fn is_datetime(&self) -> bool { matches!(self, Value::Datetime(_)) } /// Check if this Value is a Thing of a specific type pub fn is_type_record(&self, types: &[Table]) -> bool { match self { Value::Thing(v) => types.iter().any(|tb| tb.0 == v.tb), _ => false, } } /// Check if this Value is a Geometry of a specific type pub fn is_type_geometry(&self, types: &[String]) -> bool { match self { Value::Geometry(Geometry::Point(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "point")) } Value::Geometry(Geometry::Line(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "line")) } Value::Geometry(Geometry::Polygon(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "polygon")) } Value::Geometry(Geometry::MultiPoint(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "multipoint")) } Value::Geometry(Geometry::MultiLine(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "multiline")) } Value::Geometry(Geometry::MultiPolygon(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "multipolygon")) } Value::Geometry(Geometry::Collection(_)) => { types.iter().any(|t| matches!(t.as_str(), "feature" | "collection")) } _ => false, } } // ----------------------------------- // Simple conversion of value // ----------------------------------- /// Convert this Value into an i64 pub fn as_int(self) -> i64 { match self { Value::True => 1, Value::Strand(v) => v.parse::().unwrap_or(0), Value::Number(v) => v.as_int(), Value::Duration(v) => v.as_secs() as i64, Value::Datetime(v) => v.timestamp(), _ => 0, } } /// Convert this Value into a f64 pub fn as_float(self) -> f64 { match self { Value::True => 1.0, Value::Strand(v) => v.parse::().unwrap_or(0.0), Value::Number(v) => v.as_float(), Value::Duration(v) => v.as_secs() as f64, Value::Datetime(v) => v.timestamp() as f64, _ => 0.0, } } /// Convert this Value into a BigDecimal pub fn as_decimal(self) -> BigDecimal { match self { Value::True => BigDecimal::from(1), Value::Number(v) => v.as_decimal(), Value::Strand(v) => BigDecimal::from_str(v.as_str()).unwrap_or_default(), Value::Duration(v) => v.as_secs().into(), Value::Datetime(v) => v.timestamp().into(), _ => BigDecimal::default(), } } /// Convert this Value into a Number pub fn as_number(self) -> Number { match self { Value::True => Number::from(1), Value::Number(v) => v, Value::Strand(v) => Number::from(v.as_str()), Value::Duration(v) => v.as_secs().into(), Value::Datetime(v) => v.timestamp().into(), _ => Number::default(), } } /// Convert this Value into a Strand pub fn as_strand(self) -> Strand { match self { Value::Strand(v) => v, Value::Uuid(v) => v.to_raw().into(), Value::Datetime(v) => v.to_raw().into(), _ => self.to_string().into(), } } /// Convert this Value into a Datetime pub fn as_datetime(self) -> Datetime { match self { Value::Strand(v) => Datetime::from(v.as_str()), Value::Datetime(v) => v, _ => Datetime::default(), } } /// Convert this Value into a Duration pub fn as_duration(self) -> Duration { match self { Value::Strand(v) => Duration::from(v.as_str()), Value::Duration(v) => v, _ => Duration::default(), } } /// Convert this Value into a String pub fn as_string(self) -> String { match self { Value::Strand(v) => v.0, Value::Uuid(v) => v.to_raw(), Value::Datetime(v) => v.to_raw(), _ => self.to_string(), } } /// Convert this Value into a usize pub fn as_usize(self) -> usize { match self { Value::Number(v) => v.as_usize(), _ => 0, } } /// Converts this Value into an unquoted String pub fn as_raw_string(self) -> String { match self { Value::Strand(v) => v.0, Value::Uuid(v) => v.to_raw(), Value::Datetime(v) => v.to_raw(), _ => self.to_string(), } } // ----------------------------------- // Expensive conversion of value // ----------------------------------- /// Convert this Value into a Number pub fn to_number(&self) -> Number { match self { Value::True => Number::from(1), Value::Number(v) => v.clone(), Value::Strand(v) => Number::from(v.as_str()), Value::Duration(v) => v.as_secs().into(), Value::Datetime(v) => v.timestamp().into(), _ => Number::default(), } } /// Converts this Value into a Strand pub fn to_strand(&self) -> Strand { match self { Value::Strand(v) => v.clone(), Value::Uuid(v) => v.to_raw().into(), Value::Datetime(v) => v.to_raw().into(), _ => self.to_string().into(), } } /// Converts this Value into a Datetime pub fn to_datetime(&self) -> Datetime { match self { Value::Strand(v) => Datetime::from(v.as_str()), Value::Datetime(v) => v.clone(), _ => Datetime::default(), } } /// Converts this Value into a Duration pub fn to_duration(&self) -> Duration { match self { Value::Strand(v) => Duration::from(v.as_str()), Value::Duration(v) => v.clone(), _ => Duration::default(), } } /// Converts this Value into an unquoted String pub fn to_raw_string(&self) -> String { match self { Value::Strand(v) => v.0.to_owned(), Value::Uuid(v) => v.to_raw(), Value::Datetime(v) => v.to_raw(), _ => self.to_string(), } } /// Converts this Value into a field name pub fn to_idiom(&self) -> Idiom { match self { Value::Param(v) => v.simplify(), Value::Idiom(v) => v.simplify(), Value::Strand(v) => v.0.to_string().into(), Value::Datetime(v) => v.0.to_string().into(), Value::Future(_) => "fn::future".to_string().into(), Value::Function(v) => match v.as_ref() { Function::Script(_, _) => "fn::script".to_string().into(), Function::Normal(f, _) => f.to_string().into(), Function::Cast(_, v) => v.to_idiom(), }, _ => self.to_string().into(), } } /// Try to convert this Value into a set of JSONPatch operations pub fn to_operations(&self) -> Result, Error> { match self { Value::Array(v) => v .iter() .map(|v| match v { Value::Object(v) => v.to_operation(), _ => Err(Error::InvalidPatch { message: String::from("Operation must be an object"), }), }) .collect::, Error>>(), _ => Err(Error::InvalidPatch { message: String::from("Operations must be an array"), }), } } // ----------------------------------- // Simple conversion of value // ----------------------------------- pub fn make_bool(self) -> Value { match self { Value::True | Value::False => self, _ => self.is_truthy().into(), } } pub fn make_int(self) -> Value { match self { Value::Number(Number::Int(_)) => self, _ => self.as_int().into(), } } pub fn make_float(self) -> Value { match self { Value::Number(Number::Float(_)) => self, _ => self.as_float().into(), } } pub fn make_decimal(self) -> Value { match self { Value::Number(Number::Decimal(_)) => self, _ => self.as_decimal().into(), } } pub fn make_number(self) -> Value { match self { Value::Number(_) => self, _ => self.as_number().into(), } } pub fn make_strand(self) -> Value { match self { Value::Strand(_) => self, _ => self.as_strand().into(), } } pub fn make_datetime(self) -> Value { match self { Value::Datetime(_) => self, _ => self.as_datetime().into(), } } pub fn make_duration(self) -> Value { match self { Value::Duration(_) => self, _ => self.as_duration().into(), } } pub fn could_be_table(self) -> Value { match self { Value::Strand(v) => Table::from(v.0).into(), _ => self, } } pub fn convert_to(self, kind: &Kind) -> Value { match kind { Kind::Any => self, Kind::Bool => self.make_bool(), Kind::Int => self.make_int(), Kind::Float => self.make_float(), Kind::Decimal => self.make_decimal(), Kind::Number => self.make_number(), Kind::String => self.make_strand(), Kind::Datetime => self.make_datetime(), Kind::Duration => self.make_duration(), Kind::Array => match self { Value::Array(_) => self, _ => Value::None, }, Kind::Object => match self { Value::Object(_) => self, _ => Value::None, }, Kind::Record(t) => match self.is_type_record(t) { true => self, _ => Value::None, }, Kind::Geometry(t) => match self.is_type_geometry(t) { true => self, _ => Value::None, }, } } // ----------------------------------- // Record ID extraction // ----------------------------------- /// Fetch the record id if there is one pub fn record(self) -> Option { match self { // This is an object so look for the id field Value::Object(mut v) => match v.remove("id") { Some(Value::Thing(v)) => Some(v), _ => None, }, // This is an array so take the first item Value::Array(mut v) => match v.len() { 1 => v.remove(0).record(), _ => None, }, // This is a record id already Value::Thing(v) => Some(v), // There is no valid record id _ => None, } } // ----------------------------------- // JSON Path conversion // ----------------------------------- /// Converts this Value into a JSONPatch path pub(crate) fn jsonpath(&self) -> Idiom { self.to_strand() .as_str() .trim_start_matches('/') .split(&['.', '/'][..]) .map(Part::from) .collect::>() .into() } // ----------------------------------- // JSON Path conversion // ----------------------------------- /// Checkes whether this value is a static value pub(crate) fn is_static(&self) -> bool { match self { Value::None => true, Value::Null => true, Value::False => true, Value::True => true, Value::Uuid(_) => true, Value::Number(_) => true, Value::Strand(_) => true, Value::Duration(_) => true, Value::Datetime(_) => true, Value::Geometry(_) => true, Value::Array(v) => v.iter().all(Value::is_static), Value::Object(v) => v.values().all(Value::is_static), Value::Constant(_) => true, _ => false, } } // ----------------------------------- // Value operations // ----------------------------------- /// Check if this Value is equal to another Value pub fn equal(&self, other: &Value) -> bool { match self { Value::None => other.is_none(), Value::Null => other.is_null(), Value::True => other.is_true(), Value::False => other.is_false(), Value::Thing(v) => match other { Value::Thing(w) => v == w, Value::Regex(w) => match w.regex() { Some(ref r) => r.is_match(v.to_string().as_str()), None => false, }, _ => false, }, Value::Regex(v) => match other { Value::Regex(w) => v == w, Value::Number(w) => match v.regex() { Some(ref r) => r.is_match(w.to_string().as_str()), None => false, }, Value::Strand(w) => match v.regex() { Some(ref r) => r.is_match(w.as_str()), None => false, }, _ => false, }, Value::Uuid(v) => match other { Value::Uuid(w) => v == w, _ => false, }, Value::Array(v) => match other { Value::Array(w) => v == w, _ => false, }, Value::Object(v) => match other { Value::Object(w) => v == w, _ => false, }, Value::Strand(v) => match other { Value::Strand(w) => v == w, Value::Regex(w) => match w.regex() { Some(ref r) => r.is_match(v.as_str()), None => false, }, _ => v == &other.to_strand(), }, Value::Number(v) => match other { Value::Number(w) => v == w, Value::Strand(_) => v == &other.to_number(), Value::Regex(w) => match w.regex() { Some(ref r) => r.is_match(v.to_string().as_str()), None => false, }, _ => false, }, Value::Geometry(v) => match other { Value::Geometry(w) => v == w, _ => false, }, Value::Duration(v) => match other { Value::Duration(w) => v == w, Value::Strand(_) => v == &other.to_duration(), _ => false, }, Value::Datetime(v) => match other { Value::Datetime(w) => v == w, Value::Strand(_) => v == &other.to_datetime(), _ => false, }, _ => self == other, } } /// Check if all Values in an Array are equal to another Value pub fn all_equal(&self, other: &Value) -> bool { match self { Value::Array(v) => v.iter().all(|v| v.equal(other)), _ => self.equal(other), } } /// Check if any Values in an Array are equal to another Value pub fn any_equal(&self, other: &Value) -> bool { match self { Value::Array(v) => v.iter().any(|v| v.equal(other)), _ => self.equal(other), } } /// Fuzzy check if this Value is equal to another Value pub fn fuzzy(&self, other: &Value) -> bool { match self { Value::Strand(v) => match other { Value::Strand(w) => MATCHER.fuzzy_match(v.as_str(), w.as_str()).is_some(), _ => MATCHER.fuzzy_match(v.as_str(), other.to_raw_string().as_str()).is_some(), }, _ => self.equal(other), } } /// Fuzzy check if all Values in an Array are equal to another Value pub fn all_fuzzy(&self, other: &Value) -> bool { match self { Value::Array(v) => v.iter().all(|v| v.fuzzy(other)), _ => self.fuzzy(other), } } /// Fuzzy check if any Values in an Array are equal to another Value pub fn any_fuzzy(&self, other: &Value) -> bool { match self { Value::Array(v) => v.iter().any(|v| v.fuzzy(other)), _ => self.fuzzy(other), } } /// Check if this Value contains another Value pub fn contains(&self, other: &Value) -> bool { match self { Value::Array(v) => v.iter().any(|v| v.equal(other)), Value::Thing(v) => match other { Value::Strand(w) => v.to_string().contains(w.as_str()), _ => v.to_string().contains(other.to_raw_string().as_str()), }, Value::Strand(v) => match other { Value::Strand(w) => v.contains(w.as_str()), _ => v.contains(other.to_raw_string().as_str()), }, Value::Geometry(v) => match other { Value::Geometry(w) => v.contains(w), _ => false, }, _ => false, } } /// Check if all Values in an Array contain another Value pub fn contains_all(&self, other: &Value) -> bool { match other { Value::Array(v) => v.iter().all(|v| match self { Value::Array(w) => w.iter().any(|w| v.equal(w)), Value::Geometry(_) => self.contains(v), _ => false, }), _ => false, } } /// Check if any Values in an Array contain another Value pub fn contains_any(&self, other: &Value) -> bool { match other { Value::Array(v) => v.iter().any(|v| match self { Value::Array(w) => w.iter().any(|w| v.equal(w)), Value::Geometry(_) => self.contains(v), _ => false, }), _ => false, } } /// Check if this Value intersects another Value pub fn intersects(&self, other: &Value) -> bool { match self { Value::Geometry(v) => match other { Value::Geometry(w) => v.intersects(w), _ => false, }, _ => false, } } // ----------------------------------- // Sorting operations // ----------------------------------- /// Compare this Value to another Value lexicographically pub fn lexical_cmp(&self, other: &Value) -> Option { match (self, other) { (Value::Strand(a), Value::Strand(b)) => Some(lexical_sort::lexical_cmp(a, b)), _ => self.partial_cmp(other), } } /// Compare this Value to another Value using natrual numerical comparison pub fn natural_cmp(&self, other: &Value) -> Option { match (self, other) { (Value::Strand(a), Value::Strand(b)) => Some(lexical_sort::natural_cmp(a, b)), _ => self.partial_cmp(other), } } /// Compare this Value to another Value lexicographically and using natrual numerical comparison pub fn natural_lexical_cmp(&self, other: &Value) -> Option { match (self, other) { (Value::Strand(a), Value::Strand(b)) => Some(lexical_sort::natural_lexical_cmp(a, b)), _ => self.partial_cmp(other), } } // ----------------------------------- // Mathematical operations // ----------------------------------- pub fn pow(self, other: Value) -> Value { self.as_number().pow(other.as_number()).into() } } impl fmt::Display for Value { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { let mut f = Pretty::from(f); match self { Value::None => write!(f, "NONE"), Value::Null => write!(f, "NULL"), Value::True => write!(f, "true"), Value::False => write!(f, "false"), Value::Number(v) => write!(f, "{v}"), Value::Strand(v) => write!(f, "{v}"), Value::Duration(v) => write!(f, "{v}"), Value::Datetime(v) => write!(f, "{v}"), Value::Uuid(v) => write!(f, "{v}"), Value::Array(v) => write!(f, "{v}"), Value::Object(v) => write!(f, "{v}"), Value::Geometry(v) => write!(f, "{v}"), Value::Param(v) => write!(f, "{v}"), Value::Idiom(v) => write!(f, "{v}"), Value::Table(v) => write!(f, "{v}"), Value::Thing(v) => write!(f, "{v}"), Value::Model(v) => write!(f, "{v}"), Value::Regex(v) => write!(f, "{v}"), Value::Block(v) => write!(f, "{v}"), Value::Range(v) => write!(f, "{v}"), Value::Edges(v) => write!(f, "{v}"), Value::Future(v) => write!(f, "{v}"), Value::Constant(v) => write!(f, "{v}"), Value::Function(v) => write!(f, "{v}"), Value::Subquery(v) => write!(f, "{v}"), Value::Expression(v) => write!(f, "{v}"), } } } impl Value { pub(crate) fn writeable(&self) -> bool { match self { Value::Block(v) => v.writeable(), Value::Array(v) => v.iter().any(Value::writeable), Value::Object(v) => v.iter().any(|(_, v)| v.writeable()), Value::Function(v) => v.args().iter().any(Value::writeable), Value::Subquery(v) => v.writeable(), Value::Expression(v) => v.l.writeable() || v.r.writeable(), _ => false, } } #[cfg_attr(not(target_arch = "wasm32"), async_recursion)] #[cfg_attr(target_arch = "wasm32", async_recursion(?Send))] pub(crate) async fn compute( &self, ctx: &Context<'_>, opt: &Options, txn: &Transaction, doc: Option<&'async_recursion Value>, ) -> Result { match self { Value::None => Ok(Value::None), Value::Null => Ok(Value::Null), Value::True => Ok(Value::True), Value::False => Ok(Value::False), Value::Thing(v) => v.compute(ctx, opt, txn, doc).await, Value::Block(v) => v.compute(ctx, opt, txn, doc).await, Value::Range(v) => v.compute(ctx, opt, txn, doc).await, Value::Param(v) => v.compute(ctx, opt, txn, doc).await, Value::Idiom(v) => v.compute(ctx, opt, txn, doc).await, Value::Array(v) => v.compute(ctx, opt, txn, doc).await, Value::Object(v) => v.compute(ctx, opt, txn, doc).await, Value::Future(v) => v.compute(ctx, opt, txn, doc).await, Value::Constant(v) => v.compute(ctx, opt, txn, doc).await, Value::Function(v) => v.compute(ctx, opt, txn, doc).await, Value::Subquery(v) => v.compute(ctx, opt, txn, doc).await, Value::Expression(v) => v.compute(ctx, opt, txn, doc).await, _ => Ok(self.to_owned()), } } } impl Serialize for Value { fn serialize(&self, s: S) -> Result where S: serde::Serializer, { if is_internal_serialization() { match self { Value::None => s.serialize_unit_variant("Value", 0, "None"), Value::Null => s.serialize_unit_variant("Value", 1, "Null"), Value::False => s.serialize_unit_variant("Value", 2, "False"), Value::True => s.serialize_unit_variant("Value", 3, "True"), Value::Number(v) => s.serialize_newtype_variant("Value", 4, "Number", v), Value::Strand(v) => s.serialize_newtype_variant("Value", 5, "Strand", v), Value::Duration(v) => s.serialize_newtype_variant("Value", 6, "Duration", v), Value::Datetime(v) => s.serialize_newtype_variant("Value", 7, "Datetime", v), Value::Uuid(v) => s.serialize_newtype_variant("Value", 8, "Uuid", v), Value::Array(v) => s.serialize_newtype_variant("Value", 9, "Array", v), Value::Object(v) => s.serialize_newtype_variant("Value", 10, "Object", v), Value::Geometry(v) => s.serialize_newtype_variant("Value", 11, "Geometry", v), Value::Param(v) => s.serialize_newtype_variant("Value", 12, "Param", v), Value::Idiom(v) => s.serialize_newtype_variant("Value", 13, "Idiom", v), Value::Table(v) => s.serialize_newtype_variant("Value", 14, "Table", v), Value::Thing(v) => s.serialize_newtype_variant("Value", 15, "Thing", v), Value::Model(v) => s.serialize_newtype_variant("Value", 16, "Model", v), Value::Regex(v) => s.serialize_newtype_variant("Value", 17, "Regex", v), Value::Block(v) => s.serialize_newtype_variant("Value", 18, "Block", v), Value::Range(v) => s.serialize_newtype_variant("Value", 19, "Range", v), Value::Edges(v) => s.serialize_newtype_variant("Value", 20, "Edges", v), Value::Future(v) => s.serialize_newtype_variant("Value", 21, "Future", v), Value::Constant(v) => s.serialize_newtype_variant("Value", 22, "Constant", v), Value::Function(v) => s.serialize_newtype_variant("Value", 23, "Function", v), Value::Subquery(v) => s.serialize_newtype_variant("Value", 24, "Subquery", v), Value::Expression(v) => s.serialize_newtype_variant("Value", 25, "Expression", v), } } else { match self { Value::None => s.serialize_none(), Value::Null => s.serialize_none(), Value::True => s.serialize_bool(true), Value::False => s.serialize_bool(false), Value::Thing(v) => s.serialize_some(v), Value::Uuid(v) => s.serialize_some(v), Value::Array(v) => s.serialize_some(v), Value::Object(v) => s.serialize_some(v), Value::Number(v) => s.serialize_some(v), Value::Strand(v) => s.serialize_some(v), Value::Geometry(v) => s.serialize_some(v), Value::Duration(v) => s.serialize_some(v), Value::Datetime(v) => s.serialize_some(v), Value::Constant(v) => s.serialize_some(v), _ => s.serialize_none(), } } } } impl ops::Add for Value { type Output = Self; fn add(self, other: Self) -> Self { match (self, other) { (Value::Number(v), Value::Number(w)) => Value::Number(v + w), (Value::Strand(v), Value::Strand(w)) => Value::Strand(v + w), (Value::Datetime(v), Value::Duration(w)) => Value::Datetime(w + v), (Value::Duration(v), Value::Datetime(w)) => Value::Datetime(v + w), (Value::Duration(v), Value::Duration(w)) => Value::Duration(v + w), (v, w) => Value::from(v.as_number() + w.as_number()), } } } impl ops::Sub for Value { type Output = Self; fn sub(self, other: Self) -> Self { match (self, other) { (Value::Number(v), Value::Number(w)) => Value::Number(v - w), (Value::Datetime(v), Value::Datetime(w)) => Value::Duration(v - w), (Value::Datetime(v), Value::Duration(w)) => Value::Datetime(w - v), (Value::Duration(v), Value::Datetime(w)) => Value::Datetime(v - w), (Value::Duration(v), Value::Duration(w)) => Value::Duration(v - w), (v, w) => Value::from(v.as_number() - w.as_number()), } } } impl ops::Mul for Value { type Output = Self; fn mul(self, other: Self) -> Self { match (self, other) { (Value::Number(v), Value::Number(w)) => Value::Number(v * w), (v, w) => Value::from(v.as_number() * w.as_number()), } } } impl ops::Div for Value { type Output = Self; fn div(self, other: Self) -> Self { match (self.as_number(), other.as_number()) { (_, w) if w == Number::Int(0) => Value::None, (v, w) => Value::Number(v / w), } } } pub fn value(i: &str) -> IResult<&str, Value> { alt((double, single))(i) } pub fn double(i: &str) -> IResult<&str, Value> { map(expression, Value::from)(i) } pub fn single(i: &str) -> IResult<&str, Value> { alt(( alt(( map(tag_no_case("NONE"), |_| Value::None), map(tag_no_case("NULL"), |_| Value::Null), map(tag_no_case("true"), |_| Value::True), map(tag_no_case("false"), |_| Value::False), )), alt(( map(subquery, Value::from), map(function, Value::from), map(constant, Value::from), map(datetime, Value::from), map(duration, Value::from), map(geometry, Value::from), map(future, Value::from), map(unique, Value::from), map(number, Value::from), map(object, Value::from), map(array, Value::from), map(block, Value::from), map(param, Value::from), map(regex, Value::from), map(model, Value::from), map(idiom, Value::from), map(range, Value::from), map(thing, Value::from), map(strand, Value::from), )), ))(i) } pub fn select(i: &str) -> IResult<&str, Value> { alt(( alt(( map(tag_no_case("NONE"), |_| Value::None), map(tag_no_case("NULL"), |_| Value::Null), map(tag_no_case("true"), |_| Value::True), map(tag_no_case("false"), |_| Value::False), )), alt(( map(expression, Value::from), map(subquery, Value::from), map(function, Value::from), map(constant, Value::from), map(datetime, Value::from), map(duration, Value::from), map(geometry, Value::from), map(future, Value::from), map(unique, Value::from), map(number, Value::from), map(object, Value::from), map(array, Value::from), map(block, Value::from), map(param, Value::from), map(regex, Value::from), map(model, Value::from), map(edges, Value::from), map(range, Value::from), map(thing, Value::from), map(table, Value::from), map(strand, Value::from), )), ))(i) } pub fn what(i: &str) -> IResult<&str, Value> { alt(( map(subquery, Value::from), map(function, Value::from), map(constant, Value::from), map(future, Value::from), map(block, Value::from), map(param, Value::from), map(model, Value::from), map(edges, Value::from), map(range, Value::from), map(thing, Value::from), map(table, Value::from), ))(i) } pub fn json(i: &str) -> IResult<&str, Value> { alt(( map(tag_no_case("NULL"), |_| Value::Null), map(tag_no_case("true"), |_| Value::True), map(tag_no_case("false"), |_| Value::False), map(datetime, Value::from), map(duration, Value::from), map(geometry, Value::from), map(unique, Value::from), map(number, Value::from), map(object, Value::from), map(array, Value::from), map(thing, Value::from), map(strand, Value::from), ))(i) } #[cfg(test)] mod tests { use super::*; use crate::sql::test::Parse; use crate::sql::uuid::Uuid; #[test] fn check_none() { assert_eq!(true, Value::None.is_none()); assert_eq!(false, Value::Null.is_none()); assert_eq!(false, Value::from(1).is_none()); } #[test] fn check_null() { assert_eq!(true, Value::Null.is_null()); assert_eq!(false, Value::None.is_null()); assert_eq!(false, Value::from(1).is_null()); } #[test] fn check_true() { assert_eq!(false, Value::None.is_true()); assert_eq!(true, Value::True.is_true()); assert_eq!(false, Value::False.is_true()); assert_eq!(false, Value::from(1).is_true()); assert_eq!(true, Value::from("true").is_true()); assert_eq!(false, Value::from("false").is_true()); assert_eq!(false, Value::from("something").is_true()); } #[test] fn check_false() { assert_eq!(false, Value::None.is_false()); assert_eq!(false, Value::True.is_false()); assert_eq!(true, Value::False.is_false()); assert_eq!(false, Value::from(1).is_false()); assert_eq!(false, Value::from("true").is_false()); assert_eq!(true, Value::from("false").is_false()); assert_eq!(false, Value::from("something").is_false()); } #[test] fn convert_bool() { assert_eq!(false, Value::None.is_truthy()); assert_eq!(false, Value::Null.is_truthy()); assert_eq!(true, Value::True.is_truthy()); assert_eq!(false, Value::False.is_truthy()); assert_eq!(false, Value::from(0).is_truthy()); assert_eq!(true, Value::from(1).is_truthy()); assert_eq!(true, Value::from(-1).is_truthy()); assert_eq!(true, Value::from(1.1).is_truthy()); assert_eq!(true, Value::from(-1.1).is_truthy()); assert_eq!(true, Value::from("true").is_truthy()); assert_eq!(false, Value::from("false").is_truthy()); assert_eq!(true, Value::from("falsey").is_truthy()); assert_eq!(true, Value::from("something").is_truthy()); assert_eq!(true, Value::from(Uuid::new()).is_truthy()); } #[test] fn convert_int() { assert_eq!(0, Value::None.as_int()); assert_eq!(0, Value::Null.as_int()); assert_eq!(1, Value::True.as_int()); assert_eq!(0, Value::False.as_int()); assert_eq!(0, Value::from(0).as_int()); assert_eq!(1, Value::from(1).as_int()); assert_eq!(-1, Value::from(-1).as_int()); assert_eq!(1, Value::from(1.1).as_int()); assert_eq!(-1, Value::from(-1.1).as_int()); assert_eq!(3, Value::from("3").as_int()); assert_eq!(0, Value::from("true").as_int()); assert_eq!(0, Value::from("false").as_int()); assert_eq!(0, Value::from("something").as_int()); } #[test] fn convert_float() { assert_eq!(0.0, Value::None.as_float()); assert_eq!(0.0, Value::Null.as_float()); assert_eq!(1.0, Value::True.as_float()); assert_eq!(0.0, Value::False.as_float()); assert_eq!(0.0, Value::from(0).as_float()); assert_eq!(1.0, Value::from(1).as_float()); assert_eq!(-1.0, Value::from(-1).as_float()); assert_eq!(1.1, Value::from(1.1).as_float()); assert_eq!(-1.1, Value::from(-1.1).as_float()); assert_eq!(3.0, Value::from("3").as_float()); assert_eq!(0.0, Value::from("true").as_float()); assert_eq!(0.0, Value::from("false").as_float()); assert_eq!(0.0, Value::from("something").as_float()); } #[test] fn convert_number() { assert_eq!(Number::from(0), Value::None.as_number()); assert_eq!(Number::from(0), Value::Null.as_number()); assert_eq!(Number::from(1), Value::True.as_number()); assert_eq!(Number::from(0), Value::False.as_number()); assert_eq!(Number::from(0), Value::from(0).as_number()); assert_eq!(Number::from(1), Value::from(1).as_number()); assert_eq!(Number::from(-1), Value::from(-1).as_number()); assert_eq!(Number::from(1.1), Value::from(1.1).as_number()); assert_eq!(Number::from(-1.1), Value::from(-1.1).as_number()); assert_eq!(Number::from(3), Value::from("3").as_number()); assert_eq!(Number::from(0), Value::from("true").as_number()); assert_eq!(Number::from(0), Value::from("false").as_number()); assert_eq!(Number::from(0), Value::from("something").as_number()); } #[test] fn convert_strand() { assert_eq!(Strand::from("NONE"), Value::None.as_strand()); assert_eq!(Strand::from("NULL"), Value::Null.as_strand()); assert_eq!(Strand::from("true"), Value::True.as_strand()); assert_eq!(Strand::from("false"), Value::False.as_strand()); assert_eq!(Strand::from("0"), Value::from(0).as_strand()); assert_eq!(Strand::from("1"), Value::from(1).as_strand()); assert_eq!(Strand::from("-1"), Value::from(-1).as_strand()); assert_eq!(Strand::from("1.1"), Value::from(1.1).as_strand()); assert_eq!(Strand::from("-1.1"), Value::from(-1.1).as_strand()); assert_eq!(Strand::from("3"), Value::from("3").as_strand()); assert_eq!(Strand::from("true"), Value::from("true").as_strand()); assert_eq!(Strand::from("false"), Value::from("false").as_strand()); assert_eq!(Strand::from("something"), Value::from("something").as_strand()); } #[test] fn convert_string() { assert_eq!(String::from("NONE"), Value::None.as_string()); assert_eq!(String::from("NULL"), Value::Null.as_string()); assert_eq!(String::from("true"), Value::True.as_string()); assert_eq!(String::from("false"), Value::False.as_string()); assert_eq!(String::from("0"), Value::from(0).as_string()); assert_eq!(String::from("1"), Value::from(1).as_string()); assert_eq!(String::from("-1"), Value::from(-1).as_string()); assert_eq!(String::from("1.1"), Value::from(1.1).as_string()); assert_eq!(String::from("-1.1"), Value::from(-1.1).as_string()); assert_eq!(String::from("3"), Value::from("3").as_string()); assert_eq!(String::from("true"), Value::from("true").as_string()); assert_eq!(String::from("false"), Value::from("false").as_string()); assert_eq!(String::from("something"), Value::from("something").as_string()); } #[test] fn check_size() { assert_eq!(64, std::mem::size_of::()); assert_eq!(104, std::mem::size_of::>()); assert_eq!(40, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(16, std::mem::size_of::()); assert_eq!(12, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(56, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(56, std::mem::size_of::()); assert_eq!(48, std::mem::size_of::()); assert_eq!(24, std::mem::size_of::()); assert_eq!(8, std::mem::size_of::>()); assert_eq!(8, std::mem::size_of::>()); assert_eq!(8, std::mem::size_of::>()); assert_eq!(8, std::mem::size_of::>()); assert_eq!(8, std::mem::size_of::>()); } #[test] fn serialize_deserialize() { let val = Value::parse( "{ test: { something: [1, 'two', null, test:tobie, { something: false }] } }", ); let res = Value::parse( "{ test: { something: [1, 'two', null, test:tobie, { something: false }] } }", ); let enc: Vec = val.into(); let dec: Value = enc.into(); assert_eq!(res, dec); } }