1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
//! `Statement` represents an SQL statement,
//! which can be later sent to a database.

use super::Value;

/// SQL statement, possibly with bound parameters
pub struct Statement {
    pub(crate) q: String,
    pub(crate) params: Vec<Value>,
}

impl Statement {
    /// Creates a new simple statement without bound parameters
    ///
    /// # Examples
    ///
    /// ```
    /// let stmt = libsql_client::Statement::new("SELECT * FROM sqlite_master");
    /// ```
    pub fn new(q: impl Into<String>) -> Statement {
        Self {
            q: q.into(),
            params: vec![],
        }
    }

    /// Creates a statement with bound parameters
    ///
    /// # Examples
    ///
    /// ```
    /// let stmt = libsql_client::Statement::with_params("UPDATE t SET x = ? WHERE key = ?", &[3, 8]);
    /// ```
    pub fn with_params(q: impl Into<String>, params: &[impl Into<Value> + Clone]) -> Statement {
        Self {
            q: q.into(),
            params: params.iter().map(|p| p.clone().into()).collect(),
        }
    }
}

impl From<String> for Statement {
    fn from(q: String) -> Statement {
        Statement { q, params: vec![] }
    }
}

impl From<&str> for Statement {
    fn from(val: &str) -> Self {
        val.to_string().into()
    }
}

impl From<&&str> for Statement {
    fn from(val: &&str) -> Self {
        val.to_string().into()
    }
}

impl std::fmt::Display for Statement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        if self.params.is_empty() {
            write!(f, "{}", serde_json::json!(self.q))
        } else {
            let params: Vec<String> = self
                .params
                .iter()
                .map(|p| serde_json::json!(p).to_string())
                .collect();
            write!(
                f,
                "{{\"q\": {}, \"params\": [{}]}}",
                serde_json::json!(self.q),
                params.join(",")
            )
        }
    }
}