1 files changed, 132 insertions, 0 deletions
diff --git a/notmuch/src/query.rs b/notmuch/src/query.rs
new file mode 100644
index 0000000..50b56e5
--- /dev/null
+++ b/notmuch/src/query.rs
@@ -0,0 +1,132 @@
+use std::ops::Drop;
+use std::ptr;
+use std::ffi::{CStr, CString};
+
+use supercow::{Phantomcow, Supercow};
+
+use error::Result;
+use ffi;
+use ffi::{Sort, Exclude};
+use Database;
+use Messages;
+use MessageOwner;
+use Threads;
+use DatabaseExt;
+use utils::ScopedSupercow;
+
+#[derive(Debug)]
+pub struct Query<'d> {
+    pub(crate) ptr: *mut ffi::notmuch_query_t,
+    marker: Phantomcow<'d, Database>,
+}
+
+impl<'d> Drop for Query<'d> {
+    fn drop(&mut self) {
+        unsafe { ffi::notmuch_query_destroy(self.ptr) };
+    }
+}
+
+impl<'d> MessageOwner for Query<'d> {}
+
+impl<'d> Query<'d> {
+    pub(crate) fn from_ptr<O>(ptr: *mut ffi::notmuch_query_t, owner: O) -> Query<'d>
+    where
+        O: Into<Phantomcow<'d, Database>>,
+    {
+        Query {
+            ptr,
+            marker: owner.into(),
+        }
+    }
+
+    pub fn create<D>(db: D, query_string: &str) -> Result<Self>
+    where
+        D: Into<Supercow<'d, Database>>,
+    {
+        <Database as DatabaseExt>::create_query(db, query_string)
+    }
+
+    pub fn query_string(self: &Self) -> String {
+        let qstring = unsafe {
+            CStr::from_ptr(ffi::notmuch_query_get_query_string(self.ptr))
+        };
+        qstring.to_str().unwrap().to_string()
+    }
+
+    /// Specify the sorting desired for this query.
+    pub fn set_sort(self: &Self, sort: Sort) {
+        unsafe { ffi::notmuch_query_set_sort(self.ptr, sort.into()) }
+    }
+
+    /// Return the sort specified for this query. See
+    /// `set_sort`.
+    pub fn sort(self: &Self) -> Sort {
+        unsafe { ffi::notmuch_query_get_sort(self.ptr) }.into()
+    }
+
+    /// Filter messages according to the query and return
+    pub fn search_messages<'q>(self: &'d Self) -> Result<Messages<'q, Self>> {
+        <Query as QueryExt>::search_messages(self)
+    }
+
+    pub fn count_messages(self: &Self) -> Result<u32> {
+        let mut cnt = 0;
+        unsafe { ffi::notmuch_query_count_messages(self.ptr, &mut cnt) }.as_result()?;
+
+        Ok(cnt)
+    }
+
+    pub fn search_threads<'q>(self: &'d Self) -> Result<Threads<'d, 'q>> {
+        <Query<'d> as QueryExt>::search_threads(self)
+    }
+
+    pub fn count_threads(self: &Self) -> Result<u32> {
+        let mut cnt = 0;
+        unsafe { ffi::notmuch_query_count_threads(self.ptr, &mut cnt) }.as_result()?;
+
+        Ok(cnt)
+    }
+
+    pub fn add_tag_exclude(self: &Self, tag: &str) -> Result<()>
+    {
+        let tag_str = CString::new(tag).unwrap();
+        unsafe { ffi::notmuch_query_add_tag_exclude(self.ptr, tag_str.as_ptr()) }.as_result()
+    }
+
+    pub fn set_omit_excluded(self: &Self, omit_excluded: Exclude) {
+        unsafe { ffi::notmuch_query_set_omit_excluded(self.ptr, omit_excluded.into()) }
+    }
+}
+
+pub trait QueryExt<'d> {
+    fn search_threads<'q, Q>(query: Q) -> Result<Threads<'d, 'q>>
+    where
+        Q: Into<ScopedSupercow<'q, Query<'d>>>,
+    {
+        let queryref = query.into();
+
+        let mut thrds = ptr::null_mut();
+        unsafe { ffi::notmuch_query_search_threads(queryref.ptr, &mut thrds) }
+            .as_result()?;
+
+        Ok(Threads::from_ptr(thrds, ScopedSupercow::phantom(queryref)))
+    }
+
+    fn search_messages<'q, Q>(query: Q) -> Result<Messages<'q, Query<'d>>>
+    where
+        Q: Into<ScopedSupercow<'q, Query<'d>>>,
+    {
+        let queryref = query.into();
+
+        let mut msgs = ptr::null_mut();
+        unsafe { ffi::notmuch_query_search_messages(queryref.ptr, &mut msgs) }
+            .as_result()?;
+
+        Ok(Messages::from_ptr(msgs, ScopedSupercow::phantom(queryref)))
+    }
+}
+
+impl<'d> QueryExt<'d> for Query<'d> {}
+
+unsafe impl<'d> Send for Query<'d> {}
+unsafe impl<'d> Sync for Query<'d> {}