qail_core/ast/
values.rs

1use crate::ast::Qail;
2use serde::{Deserialize, Serialize};
3use uuid::Uuid;
4
5/// Time interval unit for duration expressions
6#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
7pub enum IntervalUnit {
8    Second,
9    Minute,
10    Hour,
11    Day,
12    Week,
13    Month,
14    Year,
15}
16
17impl std::fmt::Display for IntervalUnit {
18    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
19        match self {
20            IntervalUnit::Second => write!(f, "seconds"),
21            IntervalUnit::Minute => write!(f, "minutes"),
22            IntervalUnit::Hour => write!(f, "hours"),
23            IntervalUnit::Day => write!(f, "days"),
24            IntervalUnit::Week => write!(f, "weeks"),
25            IntervalUnit::Month => write!(f, "months"),
26            IntervalUnit::Year => write!(f, "years"),
27        }
28    }
29}
30
31/// A value in a condition.
32#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
33pub enum Value {
34    Null,
35    Bool(bool),
36    Int(i64),
37    Float(f64),
38    String(String),
39    Param(usize),
40    /// Named parameter reference (:name, :id, etc.)
41    NamedParam(String),
42    Function(String),
43    Array(Vec<Value>),
44    Subquery(Box<Qail>),
45    Column(String),
46    Uuid(Uuid),
47    NullUuid,
48    /// Time interval (e.g., 24 hours, 7 days)
49    Interval { amount: i64, unit: IntervalUnit },
50    Timestamp(String),
51    /// Binary data (bytea)
52    Bytes(Vec<u8>),
53    /// AST Expression (for complex expression comparisons like col > NOW() - INTERVAL)
54    Expr(Box<crate::ast::Expr>),
55    /// Vector embedding for similarity search (Qdrant)
56    Vector(Vec<f32>),
57}
58
59impl std::fmt::Display for Value {
60    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
61        match self {
62            Value::Null => write!(f, "NULL"),
63            Value::Bool(b) => write!(f, "{}", b),
64            Value::Int(n) => write!(f, "{}", n),
65            Value::Float(n) => write!(f, "{}", n),
66            Value::String(s) => write!(f, "'{}'", s),
67            Value::Param(n) => write!(f, "${}", n),
68            Value::NamedParam(name) => write!(f, ":{}", name),
69            Value::Function(s) => write!(f, "{}", s),
70            Value::Array(arr) => {
71                write!(f, "(")?;
72                for (i, v) in arr.iter().enumerate() {
73                    if i > 0 {
74                        write!(f, ", ")?;
75                    }
76                    write!(f, "{}", v)?;
77                }
78                write!(f, ")")
79            }
80            Value::Subquery(_) => write!(f, "(SUBQUERY)"),
81            Value::Column(s) => write!(f, "{}", s),
82            Value::Uuid(u) => write!(f, "'{}'", u),
83            Value::NullUuid => write!(f, "NULL"),
84            Value::Interval { amount, unit } => write!(f, "INTERVAL '{} {}'", amount, unit),
85            Value::Timestamp(ts) => write!(f, "'{}'", ts),
86            Value::Bytes(bytes) => {
87                write!(f, "'\\x")?;
88                for byte in bytes {
89                    write!(f, "{:02x}", byte)?;
90                }
91                write!(f, "'")
92            }
93            Value::Expr(expr) => write!(f, "{}", expr),
94            Value::Vector(v) => {
95                write!(f, "[")?;
96                for (i, val) in v.iter().enumerate() {
97                    if i > 0 { write!(f, ", ")?; }
98                    write!(f, "{}", val)?;
99                }
100                write!(f, "]")
101            }
102        }
103    }
104}
105
106impl From<bool> for Value {
107    fn from(b: bool) -> Self {
108        Value::Bool(b)
109    }
110}
111
112impl From<i32> for Value {
113    fn from(n: i32) -> Self {
114        Value::Int(n as i64)
115    }
116}
117
118impl From<i64> for Value {
119    fn from(n: i64) -> Self {
120        Value::Int(n)
121    }
122}
123
124impl From<f64> for Value {
125    fn from(n: f64) -> Self {
126        Value::Float(n)
127    }
128}
129
130impl From<&str> for Value {
131    fn from(s: &str) -> Self {
132        Value::String(s.to_string())
133    }
134}
135
136impl From<String> for Value {
137    fn from(s: String) -> Self {
138        Value::String(s)
139    }
140}
141
142impl From<Uuid> for Value {
143    fn from(u: Uuid) -> Self {
144        Value::Uuid(u)
145    }
146}
147
148impl From<Option<Uuid>> for Value {
149    fn from(opt: Option<Uuid>) -> Self {
150        match opt {
151            Some(u) => Value::Uuid(u),
152            None => Value::NullUuid,
153        }
154    }
155}