sqlite-wasm-connector/src/database.rs

use libc::{c_char, c_int, c_void};
use raw;
use std::marker::PhantomData;
use std::path::Path;

use {Result, Statement};

/// A database.
pub struct Database {
    raw: *mut raw::sqlite3,
    phantom: PhantomData<raw::sqlite3>,
}

/// A callback triggered for each row of an executed SQL query. If the callback
/// returns `false`, no more rows will be processed.
pub type ExecuteCallback<'l> = FnMut(Vec<(String, String)>) -> bool + 'l;

impl Database {
    /// Open a database connect.
    pub fn open(path: &Path) -> Result<Database> {
        let mut raw = 0 as *mut _;
        unsafe {
            success!(raw::sqlite3_open(path_to_c_str!(path), &mut raw));
        }
        Ok(Database { raw: raw, phantom: PhantomData })
    }

    /// Execute an SQL query.
    pub fn execute<'l>(&self, sql: &str, callback: Option<&mut ExecuteCallback<'l>>)
                       -> Result<()> {

        match callback {
            Some(callback) => unsafe {
                let mut callback = Box::new(callback);
                success!(self, raw::sqlite3_exec(self.raw, str_to_c_str!(sql),
                                                 Some(execute_callback),
                                                 &mut callback as *mut _ as *mut _,
                                                 0 as *mut _));
            },
            None => unsafe {
                success!(self, raw::sqlite3_exec(self.raw, str_to_c_str!(sql), None,
                                                 0 as *mut _, 0 as *mut _));
            },
        }
        Ok(())
    }

    /// Compile an SQL statement.
    #[inline]
    pub fn prepare<'l>(&'l self, sql: &str) -> Result<Statement<'l>> {
        ::statement::new(self, sql)
    }

    /// Set a callback to handle rejected operations when the database is busy.
    ///
    /// The callback is triggered when the database cannot perform an operation
    /// due to processing of a request from another client. If the callback
    /// returns `true`, the operation will be repeated.
    pub fn set_busy_handler<F: FnMut(usize) -> bool>(&mut self, callback: Option<F>)
                                                     -> Result<()> {

        match callback {
            Some(callback) => unsafe {
                use std::mem::transmute;
                let callback = Box::new(callback);
                success!(raw::sqlite3_busy_handler(self.raw, Some(busy_callback::<F>),
                                                   transmute::<_, *mut c_void>(callback)));
            },
            None => unsafe {
                success!(raw::sqlite3_busy_handler(self.raw, None, 0 as *mut _));
            },
        }
        Ok(())
    }

    /// Set a timeout before rejecting an operation when the database is busy.
    #[inline]
    pub fn set_busy_timeout(&mut self, milliseconds: usize) -> Result<()> {
        unsafe { success!(self, raw::sqlite3_busy_timeout(self.raw, milliseconds as c_int)) };
        Ok(())
    }
}

impl Drop for Database {
    #[inline]
    fn drop(&mut self) {
        unsafe { raw::sqlite3_close(self.raw) };
    }
}

#[inline]
pub fn as_raw(database: &Database) -> *mut raw::sqlite3 {
    database.raw
}

extern fn busy_callback<F: FnMut(usize) -> bool>(callback: *mut c_void, attempts: c_int) -> c_int {
    if unsafe { (*(callback as *mut F))(attempts as usize) } { 1 } else { 0 }
}

extern fn execute_callback(callback: *mut c_void, count: c_int, values: *mut *mut c_char,
                           columns: *mut *mut c_char) -> c_int {

    unsafe {
        let mut pairs = Vec::with_capacity(count as usize);

        for i in 0..(count as isize) {
            let column = c_str_to_string!(*columns.offset(i));
            let value = c_str_to_string!(*values.offset(i));
            pairs.push((column, value));
        }

        let ref mut callback = *(callback as *mut Box<&mut ExecuteCallback>);
        if callback(pairs) { 0 } else { 1 }
    }
}

#[cfg(test)]
mod tests {
    use super::Database;
    use tests::setup;

    macro_rules! ok(
        ($result:expr) => ($result.unwrap());
    );

    #[test]
    fn execute() {
        let (path, _directory) = setup();
        let database = ok!(Database::open(&path));
        match database.execute(":)", None) {
            Err(error) => assert_eq!(error.message, Some(String::from(r#"unrecognized token: ":""#))),
            _ => assert!(false),
        }
    }

    #[test]
    fn set_busy_handler() {
        let (path, _directory) = setup();
        let mut database = ok!(Database::open(&path));
        do_set_busy_handler(&mut database);
    }

    fn do_set_busy_handler(database: &mut Database) {
        ok!(database.set_busy_handler(Some(|_| true)));
    }
}