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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

//! An SQLite statement

use crate::{
    api::{row::Row, sqlite::Sqlite, types::SqliteType},
    error,
    error::Error,
    ffi,
};

/// An SQLite statement
#[derive(Debug)]
pub struct Statement<'a> {
    /// The database
    pub(in crate::api) sqlite: &'a Sqlite,
    /// The statement
    pub(in crate::api) raw: *mut ffi::sqlite3_stmt,
}
impl<'a> Statement<'a> {
    /// Binds a value
    ///
    /// # Important
    /// Column indices for binding sadly start with `1`
    pub fn bind<T>(&self, column: std::ffi::c_int, value: T) -> Result<(), Error>
    where
        SqliteType: TryFrom<T>,
        <SqliteType as TryFrom<T>>::Error: std::error::Error + Send + 'static,
    {
        // Create intermediate value and bind it
        let value =
            SqliteType::try_from(value).map_err(|e| error!(with: e, "Failed to convert value into SQLite type"))?;
        match value {
            SqliteType::Null => self.bind_null(column),
            SqliteType::Integer(value) => self.bind_integer(column, value),
            SqliteType::Real(value) => self.bind_real(column, value),
            SqliteType::Text(value) => self.bind_text(column, value),
            SqliteType::Blob(value) => self.bind_blob(column, value),
        }
    }
    /// Binds a NULL value
    fn bind_null(&self, column: std::ffi::c_int) -> Result<(), Error> {
        let retval = unsafe { ffi::sqlite3_bind_null(self.raw, column) };
        unsafe { ffi::sqlite3_check_result(retval, self.sqlite.raw) }
    }
    /// Binds an INTEGER value
    fn bind_integer(&self, column: std::ffi::c_int, value: i64) -> Result<(), Error> {
        let retval = unsafe { ffi::sqlite3_bind_int64(self.raw, column, value) };
        unsafe { ffi::sqlite3_check_result(retval, self.sqlite.raw) }
    }
    /// Binds a REAL value
    fn bind_real(&self, column: std::ffi::c_int, value: f64) -> Result<(), Error> {
        let retval = unsafe { ffi::sqlite3_bind_double(self.raw, column, value) };
        unsafe { ffi::sqlite3_check_result(retval, self.sqlite.raw) }
    }
    /// Binds a TEXT value
    fn bind_text(&self, column: std::ffi::c_int, value: String) -> Result<(), Error> {
        let retval = unsafe {
            // Bind the text value and instruct SQLite to immediately copy the value
            ffi::sqlite3_bind_text64(
                self.raw,
                column,
                value.as_ptr() as _,
                value.len() as _,
                ffi::sqlite3_transient(),
                ffi::SQLITE_UTF8 as _,
            )
        };
        unsafe { ffi::sqlite3_check_result(retval, self.sqlite.raw) }
    }
    /// Binds a BLOB value
    fn bind_blob(&self, column: std::ffi::c_int, value: Vec<u8>) -> Result<(), Error> {
        let retval = unsafe {
            // Bind the blob value and instruct SQLite to immediately copy the value
            ffi::sqlite3_bind_blob64(self.raw, column, value.as_ptr() as _, value.len() as _, ffi::sqlite3_transient())
        };
        unsafe { ffi::sqlite3_check_result(retval, self.sqlite.raw) }
    }

    /// Executes the statement and gets the next result row if any
    pub fn step(&mut self) -> Result<Option<Row<'_, 'a>>, Error> {
        // Validate retval
        let retval = unsafe { ffi::sqlite3_step(self.raw) };
        if let ffi::SQLITE_ROW = retval {
            // Return row
            Ok(Some(Row { statement: self }))
        } else if let ffi::SQLITE_DONE = retval {
            // Return none
            Ok(None)
        } else {
            // Get the error
            let error = unsafe { ffi::sqlite3_check_result(retval, self.sqlite.raw) };
            match error {
                Ok(_) => Err(error!("Unknown result code for SQLite step: {retval}")),
                Err(e) => Err(e),
            }
        }
    }
}
impl<'a> Drop for Statement<'a> {
    fn drop(&mut self) {
        // Destroy statement
        unsafe { ffi::sqlite3_finalize(self.raw) };
    }
}