mirror of https://git.meli.delivery/meli/meli
melib/notmuch: add Message,TagIterator,Thread types
parent
8c6c9806b5
commit
341ff9164b
@ -0,0 +1,229 @@
|
|||||||
|
/*
|
||||||
|
* melib - notmuch backend
|
||||||
|
*
|
||||||
|
* Copyright 2020 Manos Pitsidianakis
|
||||||
|
*
|
||||||
|
* This file is part of meli.
|
||||||
|
*
|
||||||
|
* meli is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* meli is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with meli. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
use crate::thread::{ThreadHash, ThreadNode, ThreadNodeHash};
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Message<'m> {
|
||||||
|
pub lib: Arc<libloading::Library>,
|
||||||
|
pub message: *mut notmuch_message_t,
|
||||||
|
pub is_from_thread: bool,
|
||||||
|
pub _ph: std::marker::PhantomData<&'m notmuch_message_t>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'m> Message<'m> {
|
||||||
|
pub fn find_message(
|
||||||
|
lib: Arc<libloading::Library>,
|
||||||
|
db: &'m DbConnection,
|
||||||
|
msg_id: &CStr,
|
||||||
|
) -> Result<Message<'m>> {
|
||||||
|
let mut message: *mut notmuch_message_t = std::ptr::null_mut();
|
||||||
|
unsafe {
|
||||||
|
call!(lib, notmuch_database_find_message)(
|
||||||
|
*db.inner.read().unwrap(),
|
||||||
|
msg_id.as_ptr(),
|
||||||
|
&mut message as *mut _,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
if message.is_null() {
|
||||||
|
return Err(MeliError::new(format!(
|
||||||
|
"Message with message id {:?} not found in notmuch database.",
|
||||||
|
msg_id
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
Ok(Message {
|
||||||
|
lib,
|
||||||
|
message,
|
||||||
|
is_from_thread: false,
|
||||||
|
_ph: std::marker::PhantomData,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn env_hash(&self) -> EnvelopeHash {
|
||||||
|
let msg_id = unsafe { call!(self.lib, notmuch_message_get_message_id)(self.message) };
|
||||||
|
let c_str = unsafe { CStr::from_ptr(msg_id) };
|
||||||
|
{
|
||||||
|
let mut hasher = DefaultHasher::default();
|
||||||
|
c_str.hash(&mut hasher);
|
||||||
|
hasher.finish()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn msg_id(&self) -> &[u8] {
|
||||||
|
let c_str = self.msg_id_cstr();
|
||||||
|
c_str.to_bytes()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn msg_id_cstr(&self) -> &CStr {
|
||||||
|
let msg_id = unsafe { call!(self.lib, notmuch_message_get_message_id)(self.message) };
|
||||||
|
unsafe { CStr::from_ptr(msg_id) }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn date(&self) -> crate::datetime::UnixTimestamp {
|
||||||
|
(unsafe { call!(self.lib, notmuch_message_get_date)(self.message) }) as u64
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn into_envelope(
|
||||||
|
self,
|
||||||
|
index: Arc<RwLock<HashMap<EnvelopeHash, CString>>>,
|
||||||
|
tag_index: Arc<RwLock<BTreeMap<u64, String>>>,
|
||||||
|
) -> Result<Envelope> {
|
||||||
|
let mut contents = Vec::new();
|
||||||
|
let path = self.get_filename().to_os_string();
|
||||||
|
let mut f = std::fs::File::open(&path)?;
|
||||||
|
f.read_to_end(&mut contents)?;
|
||||||
|
let env_hash = self.env_hash();
|
||||||
|
let mut env = Envelope::from_bytes(&contents, None).chain_err_summary(|| {
|
||||||
|
index.write().unwrap().remove(&env_hash);
|
||||||
|
format!("could not parse path {:?}", path)
|
||||||
|
})?;
|
||||||
|
env.set_hash(env_hash);
|
||||||
|
index
|
||||||
|
.write()
|
||||||
|
.unwrap()
|
||||||
|
.insert(env_hash, self.msg_id_cstr().into());
|
||||||
|
let mut tag_lock = tag_index.write().unwrap();
|
||||||
|
let (flags, tags) = TagIterator::new(self).collect_flags_and_tags();
|
||||||
|
for tag in tags {
|
||||||
|
let mut hasher = DefaultHasher::new();
|
||||||
|
hasher.write(tag.as_bytes());
|
||||||
|
let num = hasher.finish();
|
||||||
|
if !tag_lock.contains_key(&num) {
|
||||||
|
tag_lock.insert(num, tag);
|
||||||
|
}
|
||||||
|
env.labels_mut().push(num);
|
||||||
|
}
|
||||||
|
env.set_flags(flags);
|
||||||
|
Ok(env)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn replies_iter(&self) -> Option<MessageIterator> {
|
||||||
|
if self.is_from_thread {
|
||||||
|
let messages = unsafe { call!(self.lib, notmuch_message_get_replies)(self.message) };
|
||||||
|
if messages.is_null() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(MessageIterator {
|
||||||
|
lib: self.lib.clone(),
|
||||||
|
messages,
|
||||||
|
_ph: std::marker::PhantomData,
|
||||||
|
is_from_thread: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn into_thread_node(&self) -> (ThreadNodeHash, ThreadNode) {
|
||||||
|
(
|
||||||
|
ThreadNodeHash::from(self.msg_id()),
|
||||||
|
ThreadNode {
|
||||||
|
message: Some(self.env_hash()),
|
||||||
|
parent: None,
|
||||||
|
children: vec![],
|
||||||
|
date: self.date(),
|
||||||
|
show_subject: true,
|
||||||
|
group: ThreadHash::new(),
|
||||||
|
unseen: false,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn add_tag(&self, tag: &CStr) -> Result<()> {
|
||||||
|
if let Err(err) = unsafe {
|
||||||
|
try_call!(
|
||||||
|
self.lib,
|
||||||
|
call!(self.lib, notmuch_message_add_tag)(self.message, tag.as_ptr())
|
||||||
|
)
|
||||||
|
} {
|
||||||
|
return Err(MeliError::new("Could not set tag.").set_source(Some(Arc::new(err))));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn remove_tag(&self, tag: &CStr) -> Result<()> {
|
||||||
|
if let Err(err) = unsafe {
|
||||||
|
try_call!(
|
||||||
|
self.lib,
|
||||||
|
call!(self.lib, notmuch_message_remove_tag)(self.message, tag.as_ptr())
|
||||||
|
)
|
||||||
|
} {
|
||||||
|
return Err(MeliError::new("Could not set tag.").set_source(Some(Arc::new(err))));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn tags_to_maildir_flags(&self) -> Result<()> {
|
||||||
|
if let Err(err) = unsafe {
|
||||||
|
try_call!(
|
||||||
|
self.lib,
|
||||||
|
call!(self.lib, notmuch_message_tags_to_maildir_flags)(self.message)
|
||||||
|
)
|
||||||
|
} {
|
||||||
|
return Err(MeliError::new("Could not set flags.").set_source(Some(Arc::new(err))));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_filename(&self) -> &OsStr {
|
||||||
|
let fs_path = unsafe { call!(self.lib, notmuch_message_get_filename)(self.message) };
|
||||||
|
let c_str = unsafe { CStr::from_ptr(fs_path) };
|
||||||
|
&OsStr::from_bytes(c_str.to_bytes())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Drop for Message<'_> {
|
||||||
|
fn drop(&mut self) {
|
||||||
|
unsafe { call!(self.lib, notmuch_message_destroy)(self.message) };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct MessageIterator<'query> {
|
||||||
|
pub lib: Arc<libloading::Library>,
|
||||||
|
pub messages: *mut notmuch_messages_t,
|
||||||
|
pub is_from_thread: bool,
|
||||||
|
pub _ph: std::marker::PhantomData<*const Query<'query>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'q> Iterator for MessageIterator<'q> {
|
||||||
|
type Item = Message<'q>;
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
if self.messages.is_null() {
|
||||||
|
None
|
||||||
|
} else if unsafe { call!(self.lib, notmuch_messages_valid)(self.messages) } == 1 {
|
||||||
|
let message = unsafe { call!(self.lib, notmuch_messages_get)(self.messages) };
|
||||||
|
unsafe {
|
||||||
|
call!(self.lib, notmuch_messages_move_to_next)(self.messages);
|
||||||
|
}
|
||||||
|
Some(Message {
|
||||||
|
lib: self.lib.clone(),
|
||||||
|
message,
|
||||||
|
is_from_thread: self.is_from_thread,
|
||||||
|
_ph: std::marker::PhantomData,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
self.messages = std::ptr::null_mut();
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,129 @@
|
|||||||
|
/*
|
||||||
|
* melib - notmuch backend
|
||||||
|
*
|
||||||
|
* Copyright 2020 Manos Pitsidianakis
|
||||||
|
*
|
||||||
|
* This file is part of meli.
|
||||||
|
*
|
||||||
|
* meli is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* meli is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with meli. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
pub struct TagIterator<'m> {
|
||||||
|
pub tags: *mut notmuch_tags_t,
|
||||||
|
pub message: Message<'m>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Drop for TagIterator<'_> {
|
||||||
|
fn drop(&mut self) {
|
||||||
|
unsafe { call!(self.message.lib, notmuch_tags_destroy)(self.tags) };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'m> TagIterator<'m> {
|
||||||
|
pub fn new(message: Message<'m>) -> TagIterator<'m> {
|
||||||
|
TagIterator {
|
||||||
|
tags: unsafe { call!(message.lib, notmuch_message_get_tags)(message.message) },
|
||||||
|
message,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn collect_flags_and_tags(self) -> (Flag, Vec<String>) {
|
||||||
|
fn flags(path: &CStr) -> Flag {
|
||||||
|
let mut flag = Flag::default();
|
||||||
|
let mut ptr = path.to_bytes().len().saturating_sub(1);
|
||||||
|
let mut is_valid = true;
|
||||||
|
while !path.to_bytes()[..ptr + 1].ends_with(b":2,") {
|
||||||
|
match path.to_bytes()[ptr] {
|
||||||
|
b'D' => flag |= Flag::DRAFT,
|
||||||
|
b'F' => flag |= Flag::FLAGGED,
|
||||||
|
b'P' => flag |= Flag::PASSED,
|
||||||
|
b'R' => flag |= Flag::REPLIED,
|
||||||
|
b'S' => flag |= Flag::SEEN,
|
||||||
|
b'T' => flag |= Flag::TRASHED,
|
||||||
|
_ => {
|
||||||
|
is_valid = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if ptr == 0 {
|
||||||
|
is_valid = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
ptr -= 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if !is_valid {
|
||||||
|
return Flag::default();
|
||||||
|
}
|
||||||
|
|
||||||
|
flag
|
||||||
|
}
|
||||||
|
let fs_path =
|
||||||
|
unsafe { call!(self.message.lib, notmuch_message_get_filename)(self.message.message) };
|
||||||
|
let c_str = unsafe { CStr::from_ptr(fs_path) };
|
||||||
|
|
||||||
|
let tags = self.collect::<Vec<&CStr>>();
|
||||||
|
let mut flag = Flag::default();
|
||||||
|
let mut vec = vec![];
|
||||||
|
for t in tags {
|
||||||
|
match t.to_bytes() {
|
||||||
|
b"draft" => {
|
||||||
|
flag.set(Flag::DRAFT, true);
|
||||||
|
}
|
||||||
|
b"flagged" => {
|
||||||
|
flag.set(Flag::FLAGGED, true);
|
||||||
|
}
|
||||||
|
b"passed" => {
|
||||||
|
flag.set(Flag::PASSED, true);
|
||||||
|
}
|
||||||
|
b"replied" => {
|
||||||
|
flag.set(Flag::REPLIED, true);
|
||||||
|
}
|
||||||
|
b"unread" => {
|
||||||
|
flag.set(Flag::SEEN, false);
|
||||||
|
}
|
||||||
|
b"trashed" => {
|
||||||
|
flag.set(Flag::TRASHED, true);
|
||||||
|
}
|
||||||
|
_other => {
|
||||||
|
vec.push(t.to_string_lossy().into_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
(flag | flags(c_str), vec)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'m> Iterator for TagIterator<'m> {
|
||||||
|
type Item = &'m CStr;
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
if self.tags.is_null() {
|
||||||
|
None
|
||||||
|
} else if unsafe { call!(self.message.lib, notmuch_tags_valid)(self.tags) } == 1 {
|
||||||
|
let ret = Some(unsafe {
|
||||||
|
CStr::from_ptr(call!(self.message.lib, notmuch_tags_get)(self.tags))
|
||||||
|
});
|
||||||
|
unsafe {
|
||||||
|
call!(self.message.lib, notmuch_tags_move_to_next)(self.tags);
|
||||||
|
}
|
||||||
|
ret
|
||||||
|
} else {
|
||||||
|
self.tags = std::ptr::null_mut();
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,88 @@
|
|||||||
|
/*
|
||||||
|
* melib - notmuch backend
|
||||||
|
*
|
||||||
|
* Copyright 2020 Manos Pitsidianakis
|
||||||
|
*
|
||||||
|
* This file is part of meli.
|
||||||
|
*
|
||||||
|
* meli is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* meli is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with meli. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
use super::*;
|
||||||
|
use crate::thread::ThreadHash;
|
||||||
|
|
||||||
|
pub struct Thread<'query> {
|
||||||
|
pub lib: Arc<libloading::Library>,
|
||||||
|
pub ptr: *mut notmuch_thread_t,
|
||||||
|
pub _ph: std::marker::PhantomData<*const Query<'query>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'q> Thread<'q> {
|
||||||
|
pub fn id(&self) -> ThreadHash {
|
||||||
|
let thread_id = unsafe { call!(self.lib, notmuch_thread_get_thread_id)(self.ptr) };
|
||||||
|
let c_str = unsafe { CStr::from_ptr(thread_id) };
|
||||||
|
ThreadHash::from(c_str.to_bytes())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn date(&self) -> crate::datetime::UnixTimestamp {
|
||||||
|
(unsafe { call!(self.lib, notmuch_thread_get_newest_date)(self.ptr) }) as u64
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn len(&self) -> usize {
|
||||||
|
(unsafe { call!(self.lib, notmuch_thread_get_total_messages)(self.ptr) }) as usize
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn iter(&'q self) -> MessageIterator<'q> {
|
||||||
|
let ptr = unsafe { call!(self.lib, notmuch_thread_get_messages)(self.ptr) };
|
||||||
|
MessageIterator {
|
||||||
|
lib: self.lib.clone(),
|
||||||
|
messages: ptr,
|
||||||
|
is_from_thread: true,
|
||||||
|
_ph: std::marker::PhantomData,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Drop for Thread<'_> {
|
||||||
|
fn drop(&mut self) {
|
||||||
|
unsafe { call!(self.lib, notmuch_thread_destroy)(self.ptr) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct ThreadsIterator<'query> {
|
||||||
|
pub lib: Arc<libloading::Library>,
|
||||||
|
pub threads: *mut notmuch_threads_t,
|
||||||
|
pub _ph: std::marker::PhantomData<*const Query<'query>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'q> Iterator for ThreadsIterator<'q> {
|
||||||
|
type Item = Thread<'q>;
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
if self.threads.is_null() {
|
||||||
|
None
|
||||||
|
} else if unsafe { call!(self.lib, notmuch_threads_valid)(self.threads) } == 1 {
|
||||||
|
let thread = unsafe { call!(self.lib, notmuch_threads_get)(self.threads) };
|
||||||
|
unsafe {
|
||||||
|
call!(self.lib, notmuch_threads_move_to_next)(self.threads);
|
||||||
|
}
|
||||||
|
Some(Thread {
|
||||||
|
lib: self.lib.clone(),
|
||||||
|
ptr: thread,
|
||||||
|
_ph: std::marker::PhantomData,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
self.threads = std::ptr::null_mut();
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue