use std::ffi::{CStr, CString}; use std::ops::Drop; use std::path::Path; use std::ptr; use libc; use error::Result; use ffi; use utils::ToStr; use Directory; use Query; use query::QueryPtr; use Tags; use TagsOwner; // Re-exported under database module for pretty namespacin'. pub use ffi::DatabaseMode; #[derive(Copy, Clone, Debug)] pub struct Version(libc::c_uint); #[derive(Clone, Debug)] pub struct Revision { pub revision: libc::c_ulong, pub uuid: String, } #[derive(Debug)] pub(crate) struct DatabasePtr { pub ptr: *mut ffi::notmuch_database_t, } impl Drop for DatabasePtr { fn drop(&mut self) { unsafe { ffi::notmuch_database_destroy(self.ptr) }; } } impl DatabasePtr { pub(crate) fn create_query(&self, query_string: &str) -> Result { let query_str = CString::new(query_string).unwrap(); let query = unsafe { ffi::notmuch_query_create(self.ptr, query_str.as_ptr()) }; Ok(QueryPtr { ptr: query }) } } #[derive(Debug)] pub struct Database { pub(crate) handle: DatabasePtr, } impl TagsOwner for Database {} impl Database { pub fn create>(path: &P) -> Result { let path_str = CString::new(path.as_ref().to_str().unwrap()).unwrap(); let mut db = ptr::null_mut(); try!(unsafe { ffi::notmuch_database_create(path_str.as_ptr(), &mut db) }.as_result()); Ok(Database { handle: DatabasePtr { ptr: db }, }) } pub fn open>(path: &P, mode: DatabaseMode) -> Result { let path_str = CString::new(path.as_ref().to_str().unwrap()).unwrap(); let mut db = ptr::null_mut(); try!( unsafe { ffi::notmuch_database_open(path_str.as_ptr(), mode.into(), &mut db,) } .as_result() ); Ok(Database { handle: DatabasePtr { ptr: db }, }) } pub fn close(&mut self) -> Result<()> { try!(unsafe { ffi::notmuch_database_close(self.handle.ptr) }.as_result()); Ok(()) } pub fn compact, F: FnMut(&str)>( path: &P, backup_path: Option<&P>, ) -> Result<()> { let status: Option = None; Database::_compact(path, backup_path, status) } pub fn compact_with_status, F: FnMut(&str)>( path: &P, backup_path: Option<&P>, status: F, ) -> Result<()> { Database::_compact(path, backup_path, Some(status)) } fn _compact, F: FnMut(&str)>( path: &P, backup_path: Option<&P>, status: Option, ) -> Result<()> { extern "C" fn wrapper( message: *const libc::c_char, closure: *mut libc::c_void, ) { let closure = closure as *mut F; unsafe { (*closure)(message.to_str().unwrap()) } } let path_str = CString::new(path.as_ref().to_str().unwrap()).unwrap(); let backup_path = backup_path.map(|p| CString::new(p.as_ref().to_str().unwrap()).unwrap()); try!( unsafe { ffi::notmuch_database_compact( path_str.as_ptr(), backup_path.map_or(ptr::null(), |p| p.as_ptr()), if status.is_some() { Some(wrapper::) } else { None }, status.map_or(ptr::null_mut(), |f| &f as *const _ as *mut libc::c_void), ) } .as_result() ); Ok(()) } pub fn path(&self) -> &Path { Path::new( unsafe { ffi::notmuch_database_get_path(self.handle.ptr) } .to_str() .unwrap(), ) } pub fn version(&self) -> Version { Version(unsafe { ffi::notmuch_database_get_version(self.handle.ptr) }) } #[cfg(feature = "v0_21")] pub fn revision(&self) -> Revision { let uuid_p: *const libc::c_char = ptr::null(); let revision = unsafe { ffi::notmuch_database_get_revision( self.handle.ptr, (&uuid_p) as *const _ as *mut *const libc::c_char, ) }; let uuid = unsafe { CStr::from_ptr(uuid_p) }; Revision { revision, uuid: uuid.to_string_lossy().into_owned(), } } pub fn needs_upgrade(&self) -> bool { unsafe { ffi::notmuch_database_needs_upgrade(self.handle.ptr) == 1 } } pub fn upgrade(&mut self) -> Result<()> { let status: Option = None; self._upgrade(status) } pub fn upgrade_with_status(&mut self, status: F) -> Result<()> { self._upgrade(Some(status)) } fn _upgrade(&mut self, status: Option) -> Result<()> { #[allow(trivial_numeric_casts)] extern "C" fn wrapper(closure: *mut libc::c_void, progress: libc::c_double) { let closure = closure as *mut F; unsafe { (*closure)(progress as f64) } } try!( unsafe { ffi::notmuch_database_upgrade( self.handle.ptr, if status.is_some() { Some(wrapper::) } else { None }, status.map_or(ptr::null_mut(), |f| &f as *const _ as *mut libc::c_void), ) } .as_result() ); Ok(()) } pub fn directory<'d, P: AsRef>(&'d self, path: &P) -> Result>> { let path_str = CString::new(path.as_ref().to_str().unwrap()).unwrap(); let mut dir = ptr::null_mut(); try!( unsafe { ffi::notmuch_database_get_directory(self.handle.ptr, path_str.as_ptr(), &mut dir) } .as_result() ); if dir.is_null() { Ok(None) } else { Ok(Some(Directory::from_ptr(dir, self))) } } pub fn create_query<'d>(&'d self, query_string: &str) -> Result> { self.handle.create_query(query_string).map(move |handle|{ Query::from_handle(handle, self) }) } pub fn all_tags<'d>(&'d self) -> Result> { let tags = unsafe { ffi::notmuch_database_get_all_tags(self.handle.ptr) }; Ok(Tags::from_ptr(tags, self)) } } pub trait DatabaseExt{ } unsafe impl Send for Database {} unsafe impl Sync for Database {}