//! Data Change Notification Callbacks #![allow(non_camel_case_types)] use std::mem; use std::ptr; use std::os::raw::{c_int, c_char, c_void}; use ffi; use {Connection, InnerConnection}; // Commit And Rollback Notification Callbacks // http://sqlite.org/c3ref/commit_hook.html /* void *sqlite3_commit_hook(sqlite3*, int(*)(void*), void*); void *sqlite3_rollback_hook(sqlite3*, void(*)(void *), void*); */ /// Authorizer Action Codes #[derive(Debug, PartialEq)] pub enum Action { UNKNOWN = -1, SQLITE_CREATE_INDEX = ffi::SQLITE_CREATE_INDEX as isize, SQLITE_CREATE_TABLE = ffi::SQLITE_CREATE_TABLE as isize, SQLITE_CREATE_TEMP_INDEX = ffi::SQLITE_CREATE_TEMP_INDEX as isize, SQLITE_CREATE_TEMP_TABLE = ffi::SQLITE_CREATE_TEMP_TABLE as isize, SQLITE_CREATE_TEMP_TRIGGER = ffi::SQLITE_CREATE_TEMP_TRIGGER as isize, SQLITE_CREATE_TEMP_VIEW = ffi::SQLITE_CREATE_TEMP_VIEW as isize, SQLITE_CREATE_TRIGGER = ffi::SQLITE_CREATE_TRIGGER as isize, SQLITE_CREATE_VIEW = ffi::SQLITE_CREATE_VIEW as isize, SQLITE_DELETE = ffi::SQLITE_DELETE as isize, SQLITE_DROP_INDEX = ffi::SQLITE_DROP_INDEX as isize, SQLITE_DROP_TABLE = ffi::SQLITE_DROP_TABLE as isize, SQLITE_DROP_TEMP_INDEX = ffi::SQLITE_DROP_TEMP_INDEX as isize, SQLITE_DROP_TEMP_TABLE = ffi::SQLITE_DROP_TEMP_TABLE as isize, SQLITE_DROP_TEMP_TRIGGER = ffi::SQLITE_DROP_TEMP_TRIGGER as isize, SQLITE_DROP_TEMP_VIEW = ffi::SQLITE_DROP_TEMP_VIEW as isize, SQLITE_DROP_TRIGGER = ffi::SQLITE_DROP_TRIGGER as isize, SQLITE_DROP_VIEW = ffi::SQLITE_DROP_VIEW as isize, SQLITE_INSERT = ffi::SQLITE_INSERT as isize, SQLITE_PRAGMA = ffi::SQLITE_PRAGMA as isize, SQLITE_READ = ffi::SQLITE_READ as isize, SQLITE_SELECT = ffi::SQLITE_SELECT as isize, SQLITE_TRANSACTION = ffi::SQLITE_TRANSACTION as isize, SQLITE_UPDATE = ffi::SQLITE_UPDATE as isize, SQLITE_ATTACH = ffi::SQLITE_ATTACH as isize, SQLITE_DETACH = ffi::SQLITE_DETACH as isize, SQLITE_ALTER_TABLE = ffi::SQLITE_ALTER_TABLE as isize, SQLITE_REINDEX = ffi::SQLITE_REINDEX as isize, SQLITE_ANALYZE = ffi::SQLITE_ANALYZE as isize, SQLITE_CREATE_VTABLE = ffi::SQLITE_CREATE_VTABLE as isize, SQLITE_DROP_VTABLE = ffi::SQLITE_DROP_VTABLE as isize, SQLITE_FUNCTION = ffi::SQLITE_FUNCTION as isize, SQLITE_SAVEPOINT = ffi::SQLITE_SAVEPOINT as isize, SQLITE_COPY = ffi::SQLITE_COPY as isize, SQLITE_RECURSIVE = 33, } impl From for Action { fn from(code: i32) -> Action { match code { ffi::SQLITE_CREATE_INDEX => Action::SQLITE_CREATE_INDEX, ffi::SQLITE_CREATE_TABLE => Action::SQLITE_CREATE_TABLE, ffi::SQLITE_CREATE_TEMP_INDEX => Action::SQLITE_CREATE_TEMP_INDEX, ffi::SQLITE_CREATE_TEMP_TABLE => Action::SQLITE_CREATE_TEMP_TABLE, ffi::SQLITE_CREATE_TEMP_TRIGGER => Action::SQLITE_CREATE_TEMP_TRIGGER, ffi::SQLITE_CREATE_TEMP_VIEW => Action::SQLITE_CREATE_TEMP_VIEW, ffi::SQLITE_CREATE_TRIGGER => Action::SQLITE_CREATE_TRIGGER, ffi::SQLITE_CREATE_VIEW => Action::SQLITE_CREATE_VIEW, ffi::SQLITE_DELETE => Action::SQLITE_DELETE, ffi::SQLITE_DROP_INDEX => Action::SQLITE_DROP_INDEX, ffi::SQLITE_DROP_TABLE => Action::SQLITE_DROP_TABLE, ffi::SQLITE_DROP_TEMP_INDEX => Action::SQLITE_DROP_TEMP_INDEX, ffi::SQLITE_DROP_TEMP_TABLE => Action::SQLITE_DROP_TEMP_TABLE, ffi::SQLITE_DROP_TEMP_TRIGGER => Action::SQLITE_DROP_TEMP_TRIGGER, ffi::SQLITE_DROP_TEMP_VIEW => Action::SQLITE_DROP_TEMP_VIEW, ffi::SQLITE_DROP_TRIGGER => Action::SQLITE_DROP_TRIGGER, ffi::SQLITE_DROP_VIEW => Action::SQLITE_DROP_VIEW, ffi::SQLITE_INSERT => Action::SQLITE_INSERT, ffi::SQLITE_PRAGMA => Action::SQLITE_PRAGMA, ffi::SQLITE_READ => Action::SQLITE_READ, ffi::SQLITE_SELECT => Action::SQLITE_SELECT, ffi::SQLITE_TRANSACTION => Action::SQLITE_TRANSACTION, ffi::SQLITE_UPDATE => Action::SQLITE_UPDATE, ffi::SQLITE_ATTACH => Action::SQLITE_ATTACH, ffi::SQLITE_DETACH => Action::SQLITE_DETACH, ffi::SQLITE_ALTER_TABLE => Action::SQLITE_ALTER_TABLE, ffi::SQLITE_REINDEX => Action::SQLITE_REINDEX, ffi::SQLITE_ANALYZE => Action::SQLITE_ANALYZE, ffi::SQLITE_CREATE_VTABLE => Action::SQLITE_CREATE_VTABLE, ffi::SQLITE_DROP_VTABLE => Action::SQLITE_DROP_VTABLE, ffi::SQLITE_FUNCTION => Action::SQLITE_FUNCTION, ffi::SQLITE_SAVEPOINT => Action::SQLITE_SAVEPOINT, ffi::SQLITE_COPY => Action::SQLITE_COPY, 33 => Action::SQLITE_RECURSIVE, _ => Action::UNKNOWN, } } } impl Connection { /// Register a callback function to be invoked whenever a row is updated, inserted or deleted in a rowid table. pub fn update_hook(&self, hook: F) where F: FnMut(Action, &str, &str, i64) { self.db.borrow_mut().update_hook(hook); } pub fn remove_update_hook(&self) { self.db.borrow_mut().remove_update_hook(); } } impl InnerConnection { pub fn remove_hooks(&mut self) { self.remove_update_hook(); } fn update_hook(&mut self, hook: F) where F: FnMut(Action, &str, &str, i64) { unsafe extern "C" fn call_boxed_closure(p_arg: *mut c_void, action_code: c_int, db_str: *const c_char, tbl_str: *const c_char, row_id: i64) where F: FnMut(Action, &str, &str, i64) { use std::ffi::CStr; use std::str; let boxed_hook: *mut F = mem::transmute(p_arg); assert!(!boxed_hook.is_null(), "Internal error - null function pointer"); let action = Action::from(action_code); let db_name = { let c_slice = CStr::from_ptr(db_str).to_bytes(); str::from_utf8_unchecked(c_slice) }; let tbl_name = { let c_slice = CStr::from_ptr(tbl_str).to_bytes(); str::from_utf8_unchecked(c_slice) }; (*boxed_hook)(action, db_name, tbl_name, row_id); } let previous_hook = { let boxed_hook: *mut F = Box::into_raw(Box::new(hook)); unsafe { ffi::sqlite3_update_hook(self.db(), Some(call_boxed_closure::), boxed_hook as *mut _) } }; free_boxed_update_hook(previous_hook); } fn remove_update_hook(&mut self) { let previous_hook = unsafe { ffi::sqlite3_update_hook(self.db(), None, ptr::null_mut()) }; free_boxed_update_hook(previous_hook); } } fn free_boxed_update_hook(hook: *mut c_void) { if !hook.is_null() { // TODO make sure that size_of::<*mut F>() is always equal to size_of::<*mut c_void>() let _: Box<*mut c_void> = unsafe { Box::from_raw(hook as *mut _) }; } } #[cfg(test)] mod test { use super::Action; use Connection; #[test] fn test_update_hook() { let db = Connection::open_in_memory().unwrap(); let mut called = false; db.update_hook(|action, db, tbl, row_id| { assert_eq!(Action::SQLITE_INSERT, action); assert_eq!("main", db); assert_eq!("foo", tbl); assert_eq!(1, row_id); called = true; }); db.execute_batch("CREATE TABLE foo (t TEXT)").unwrap(); db.execute_batch("INSERT INTO foo VALUES ('lisa')") .unwrap(); assert!(called); } }