melib: Fix incorrect thread child linking
parent
e3457c40d6
commit
2fcd014bfe
|
@ -27,7 +27,8 @@ use mailbox::email::*;
|
||||||
|
|
||||||
extern crate fnv;
|
extern crate fnv;
|
||||||
use self::fnv::{FnvHashMap, FnvHashSet};
|
use self::fnv::{FnvHashMap, FnvHashSet};
|
||||||
use std::cell::RefCell;
|
use std::cell::{Ref, RefCell};
|
||||||
|
use std::cmp::Ordering;
|
||||||
use std::iter::FromIterator;
|
use std::iter::FromIterator;
|
||||||
use std::ops::Index;
|
use std::ops::Index;
|
||||||
use std::result::Result as StdResult;
|
use std::result::Result as StdResult;
|
||||||
|
@ -161,7 +162,7 @@ impl ThreadNode {
|
||||||
#[derive(Clone, Debug, Default, Deserialize)]
|
#[derive(Clone, Debug, Default, Deserialize)]
|
||||||
pub struct Threads {
|
pub struct Threads {
|
||||||
thread_nodes: Vec<ThreadNode>,
|
thread_nodes: Vec<ThreadNode>,
|
||||||
root_set: Vec<usize>,
|
root_set: RefCell<Vec<usize>>,
|
||||||
|
|
||||||
message_ids: FnvHashMap<String, usize>,
|
message_ids: FnvHashMap<String, usize>,
|
||||||
hash_set: FnvHashSet<EnvelopeHash>,
|
hash_set: FnvHashSet<EnvelopeHash>,
|
||||||
|
@ -178,6 +179,24 @@ impl PartialEq for ThreadNode {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct RootIterator<'a> {
|
||||||
|
pos: usize,
|
||||||
|
root_set: Ref<'a, Vec<usize>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> Iterator for RootIterator<'a> {
|
||||||
|
type Item = usize;
|
||||||
|
fn next(&mut self) -> Option<usize> {
|
||||||
|
{
|
||||||
|
if self.pos == self.root_set.len() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
self.pos += 1;
|
||||||
|
return Some(self.root_set[self.pos - 1]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl Threads {
|
impl Threads {
|
||||||
pub fn new(collection: &mut FnvHashMap<EnvelopeHash, Envelope>) -> Threads {
|
pub fn new(collection: &mut FnvHashMap<EnvelopeHash, Envelope>) -> Threads {
|
||||||
/* To reconstruct thread information from the mails we need: */
|
/* To reconstruct thread information from the mails we need: */
|
||||||
|
@ -226,11 +245,9 @@ impl Threads {
|
||||||
root_set.push(*v);
|
root_set.push(*v);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
root_set.sort_by(|a, b| thread_nodes[*b].date.cmp(&thread_nodes[*a].date));
|
|
||||||
|
|
||||||
let mut t = Threads {
|
let mut t = Threads {
|
||||||
thread_nodes,
|
thread_nodes,
|
||||||
root_set,
|
root_set: RefCell::new(root_set),
|
||||||
message_ids,
|
message_ids,
|
||||||
hash_set,
|
hash_set,
|
||||||
..Default::default()
|
..Default::default()
|
||||||
|
@ -259,7 +276,7 @@ impl Threads {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn build_collection(&mut self, collection: &FnvHashMap<EnvelopeHash, Envelope>) {
|
fn build_collection(&mut self, collection: &FnvHashMap<EnvelopeHash, Envelope>) {
|
||||||
for i in &self.root_set {
|
for i in self.root_set.borrow().iter() {
|
||||||
node_build(
|
node_build(
|
||||||
*i,
|
*i,
|
||||||
&mut self.thread_nodes,
|
&mut self.thread_nodes,
|
||||||
|
@ -284,6 +301,41 @@ impl Threads {
|
||||||
sort: (SortField, SortOrder),
|
sort: (SortField, SortOrder),
|
||||||
collection: &FnvHashMap<EnvelopeHash, Envelope>,
|
collection: &FnvHashMap<EnvelopeHash, Envelope>,
|
||||||
) {
|
) {
|
||||||
|
let root_set = &mut self.root_set.borrow_mut();
|
||||||
|
root_set.sort_by(|a, b| match sort {
|
||||||
|
(SortField::Date, SortOrder::Desc) => {
|
||||||
|
let a = &self.thread_nodes[*a];
|
||||||
|
let b = &self.thread_nodes[*b];
|
||||||
|
b.date.cmp(&a.date)
|
||||||
|
}
|
||||||
|
(SortField::Date, SortOrder::Asc) => {
|
||||||
|
let a = &self.thread_nodes[*a];
|
||||||
|
let b = &self.thread_nodes[*b];
|
||||||
|
a.date.cmp(&b.date)
|
||||||
|
}
|
||||||
|
(SortField::Subject, SortOrder::Desc) => {
|
||||||
|
let a = &self.thread_nodes[*a].message();
|
||||||
|
let b = &self.thread_nodes[*b].message();
|
||||||
|
|
||||||
|
if a.is_none() || b.is_none() {
|
||||||
|
return Ordering::Equal;
|
||||||
|
}
|
||||||
|
let ma = &collection[&a.unwrap()];
|
||||||
|
let mb = &collection[&b.unwrap()];
|
||||||
|
ma.subject().cmp(&mb.subject())
|
||||||
|
}
|
||||||
|
(SortField::Subject, SortOrder::Asc) => {
|
||||||
|
let a = &self.thread_nodes[*a].message();
|
||||||
|
let b = &self.thread_nodes[*b].message();
|
||||||
|
|
||||||
|
if a.is_none() || b.is_none() {
|
||||||
|
return Ordering::Equal;
|
||||||
|
}
|
||||||
|
let ma = &collection[&a.unwrap()];
|
||||||
|
let mb = &collection[&b.unwrap()];
|
||||||
|
mb.subject().cmp(&ma.subject())
|
||||||
|
}
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn sort_by(
|
pub fn sort_by(
|
||||||
|
@ -292,7 +344,6 @@ impl Threads {
|
||||||
subsort: (SortField, SortOrder),
|
subsort: (SortField, SortOrder),
|
||||||
collection: &FnvHashMap<EnvelopeHash, Envelope>,
|
collection: &FnvHashMap<EnvelopeHash, Envelope>,
|
||||||
) {
|
) {
|
||||||
/*
|
|
||||||
if *self.sort.borrow() != sort {
|
if *self.sort.borrow() != sort {
|
||||||
self.inner_sort_by(sort, collection);
|
self.inner_sort_by(sort, collection);
|
||||||
*self.sort.borrow_mut() = sort;
|
*self.sort.borrow_mut() = sort;
|
||||||
|
@ -301,11 +352,10 @@ impl Threads {
|
||||||
self.inner_subsort_by(subsort, collection);
|
self.inner_subsort_by(subsort, collection);
|
||||||
*self.subsort.borrow_mut() = subsort;
|
*self.subsort.borrow_mut() = subsort;
|
||||||
}
|
}
|
||||||
*/
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn thread_to_mail(&self, i: usize) -> EnvelopeHash {
|
pub fn thread_to_mail(&self, i: usize) -> EnvelopeHash {
|
||||||
let thread = &self.thread_nodes[self.root_set[i]];
|
let thread = &self.thread_nodes[self.root_set.borrow()[i]];
|
||||||
thread.message().unwrap()
|
thread.message().unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -313,8 +363,19 @@ impl Threads {
|
||||||
&self.thread_nodes
|
&self.thread_nodes
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn root_set(&self) -> &Vec<usize> {
|
pub fn root_len(&self) -> usize {
|
||||||
&self.root_set
|
self.root_set.borrow().len()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn root_set(&self, idx: usize) -> usize {
|
||||||
|
self.root_set.borrow()[idx]
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn root_iter<'a>(&'a self) -> RootIterator<'a> {
|
||||||
|
RootIterator {
|
||||||
|
pos: 0,
|
||||||
|
root_set: self.root_set.borrow(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn has_sibling(&self, i: usize) -> bool {
|
pub fn has_sibling(&self, i: usize) -> bool {
|
||||||
|
@ -381,13 +442,7 @@ fn link_envelope(
|
||||||
/* The index of the reference we are currently examining, start from current message */
|
/* The index of the reference we are currently examining, start from current message */
|
||||||
let mut ref_ptr = t_idx;
|
let mut ref_ptr = t_idx;
|
||||||
|
|
||||||
let mut iasf = 0;
|
|
||||||
for &refn in envelope.references().iter().rev() {
|
for &refn in envelope.references().iter().rev() {
|
||||||
if iasf == 1 {
|
|
||||||
/*FIXME: Skips anything other than direct parents */
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
iasf += 1;
|
|
||||||
let r_id = String::from_utf8_lossy(refn.raw()).into();
|
let r_id = String::from_utf8_lossy(refn.raw()).into();
|
||||||
let parent_id = if message_ids.contains_key(&r_id) {
|
let parent_id = if message_ids.contains_key(&r_id) {
|
||||||
let parent_id = message_ids[&r_id];
|
let parent_id = message_ids[&r_id];
|
||||||
|
@ -395,7 +450,7 @@ fn link_envelope(
|
||||||
|| thread_nodes[ref_ptr].is_descendant(thread_nodes, &thread_nodes[parent_id]))
|
|| thread_nodes[ref_ptr].is_descendant(thread_nodes, &thread_nodes[parent_id]))
|
||||||
{
|
{
|
||||||
thread_nodes[ref_ptr].parent = Some(parent_id);
|
thread_nodes[ref_ptr].parent = Some(parent_id);
|
||||||
if thread_nodes[parent_id].children.is_empty() {
|
if !thread_nodes[parent_id].children.contains(&ref_ptr) {
|
||||||
thread_nodes[parent_id].children.push(ref_ptr);
|
thread_nodes[parent_id].children.push(ref_ptr);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -503,7 +558,7 @@ fn node_build(
|
||||||
|
|
||||||
let mut has_unseen = thread_nodes[idx].has_unseen;
|
let mut has_unseen = thread_nodes[idx].has_unseen;
|
||||||
for c in thread_nodes[idx].children.clone().iter() {
|
for c in thread_nodes[idx].children.clone().iter() {
|
||||||
node_build(*c, thread_nodes, indentation, *c, collection);
|
node_build(*c, thread_nodes, indentation, idx, collection);
|
||||||
has_unseen |= thread_nodes[*c].has_unseen;
|
has_unseen |= thread_nodes[*c].has_unseen;
|
||||||
}
|
}
|
||||||
thread_nodes[idx].has_unseen = has_unseen;
|
thread_nodes[idx].has_unseen = has_unseen;
|
||||||
|
|
|
@ -121,7 +121,7 @@ impl CompactListing {
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
self.length = mailbox.collection.threads.root_set().len();
|
self.length = mailbox.collection.threads.root_len();
|
||||||
self.content = CellBuffer::new(MAX_COLS, self.length + 1, Cell::with_char(' '));
|
self.content = CellBuffer::new(MAX_COLS, self.length + 1, Cell::with_char(' '));
|
||||||
if self.length == 0 {
|
if self.length == 0 {
|
||||||
write_string_to_grid(
|
write_string_to_grid(
|
||||||
|
@ -136,14 +136,16 @@ impl CompactListing {
|
||||||
}
|
}
|
||||||
let threads = &mailbox.collection.threads;
|
let threads = &mailbox.collection.threads;
|
||||||
threads.sort_by(self.sort, self.subsort, &mailbox.collection);
|
threads.sort_by(self.sort, self.subsort, &mailbox.collection);
|
||||||
for (idx, root_idx) in threads.root_set().iter().enumerate() {
|
for (idx, root_idx) in threads.root_iter().enumerate() {
|
||||||
let thread_node = &threads.thread_nodes()[*root_idx];
|
let thread_node = &threads.thread_nodes()[root_idx];
|
||||||
let i = if let Some(i) = thread_node.message() {
|
let i = if let Some(i) = thread_node.message() {
|
||||||
i
|
i
|
||||||
} else {
|
} else {
|
||||||
threads.thread_nodes()[thread_node.children()[0]]
|
let mut iter_ptr = thread_node.children()[0];
|
||||||
.message()
|
while threads.thread_nodes()[iter_ptr].message().is_none() {
|
||||||
.unwrap()
|
iter_ptr = threads.thread_nodes()[iter_ptr].children()[0];
|
||||||
|
}
|
||||||
|
threads.thread_nodes()[iter_ptr].message().unwrap()
|
||||||
};
|
};
|
||||||
let root_envelope: &Envelope = &mailbox.collection[&i];
|
let root_envelope: &Envelope = &mailbox.collection[&i];
|
||||||
let fg_color = if thread_node.has_unseen() {
|
let fg_color = if thread_node.has_unseen() {
|
||||||
|
@ -180,14 +182,16 @@ impl CompactListing {
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
let threads = &mailbox.collection.threads;
|
let threads = &mailbox.collection.threads;
|
||||||
let thread_node = threads.root_set()[idx];
|
let thread_node = threads.root_set(idx);
|
||||||
let thread_node = &threads.thread_nodes()[thread_node];
|
let thread_node = &threads.thread_nodes()[thread_node];
|
||||||
let i = if let Some(i) = thread_node.message() {
|
let i = if let Some(i) = thread_node.message() {
|
||||||
i
|
i
|
||||||
} else {
|
} else {
|
||||||
threads.thread_nodes()[thread_node.children()[0]]
|
let mut iter_ptr = thread_node.children()[0];
|
||||||
.message()
|
while threads.thread_nodes()[iter_ptr].message().is_none() {
|
||||||
.unwrap()
|
iter_ptr = threads.thread_nodes()[iter_ptr].children()[0];
|
||||||
|
}
|
||||||
|
threads.thread_nodes()[iter_ptr].message().unwrap()
|
||||||
};
|
};
|
||||||
let root_envelope: &Envelope = &mailbox.collection[&i];
|
let root_envelope: &Envelope = &mailbox.collection[&i];
|
||||||
let fg_color = if !root_envelope.is_seen() {
|
let fg_color = if !root_envelope.is_seen() {
|
||||||
|
|
|
@ -98,7 +98,7 @@ impl ThreadListing {
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
self.length = mailbox.collection.threads.root_set().len();
|
self.length = mailbox.collection.threads.root_len();
|
||||||
self.content = CellBuffer::new(MAX_COLS, self.length + 1, Cell::with_char(' '));
|
self.content = CellBuffer::new(MAX_COLS, self.length + 1, Cell::with_char(' '));
|
||||||
if self.length == 0 {
|
if self.length == 0 {
|
||||||
write_string_to_grid(
|
write_string_to_grid(
|
||||||
|
@ -118,12 +118,12 @@ impl ThreadListing {
|
||||||
let threads = &mailbox.collection.threads;
|
let threads = &mailbox.collection.threads;
|
||||||
threads.sort_by(self.sort, self.subsort, &mailbox.collection);
|
threads.sort_by(self.sort, self.subsort, &mailbox.collection);
|
||||||
let thread_nodes: &Vec<ThreadNode> = &threads.thread_nodes();
|
let thread_nodes: &Vec<ThreadNode> = &threads.thread_nodes();
|
||||||
let mut iter = threads.root_set().into_iter().peekable();
|
let mut iter = threads.root_iter().peekable();
|
||||||
let len = threads.root_set().len().to_string().chars().count();
|
let len = threads.root_len().to_string().chars().count();
|
||||||
/* This is just a desugared for loop so that we can use .peek() */
|
/* This is just a desugared for loop so that we can use .peek() */
|
||||||
let mut idx = 0;
|
let mut idx = 0;
|
||||||
while let Some(i) = iter.next() {
|
while let Some(i) = iter.next() {
|
||||||
let thread_node = &thread_nodes[*i];
|
let thread_node = &thread_nodes[i];
|
||||||
|
|
||||||
if !thread_node.has_message() {
|
if !thread_node.has_message() {
|
||||||
continue;
|
continue;
|
||||||
|
@ -136,7 +136,7 @@ impl ThreadListing {
|
||||||
}
|
}
|
||||||
|
|
||||||
match iter.peek() {
|
match iter.peek() {
|
||||||
Some(&x) if thread_nodes[*x].indentation() == indentation => {
|
Some(&x) if thread_nodes[x].indentation() == indentation => {
|
||||||
indentations.pop();
|
indentations.pop();
|
||||||
indentations.push(true);
|
indentations.push(true);
|
||||||
}
|
}
|
||||||
|
@ -145,7 +145,7 @@ impl ThreadListing {
|
||||||
indentations.push(false);
|
indentations.push(false);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if threads.has_sibling(*i) {
|
if threads.has_sibling(i) {
|
||||||
indentations.pop();
|
indentations.pop();
|
||||||
indentations.push(true);
|
indentations.push(true);
|
||||||
}
|
}
|
||||||
|
@ -167,7 +167,7 @@ impl ThreadListing {
|
||||||
envelope,
|
envelope,
|
||||||
idx,
|
idx,
|
||||||
indentation,
|
indentation,
|
||||||
*i,
|
i,
|
||||||
threads,
|
threads,
|
||||||
&indentations,
|
&indentations,
|
||||||
len,
|
len,
|
||||||
|
@ -185,11 +185,11 @@ impl ThreadListing {
|
||||||
}
|
}
|
||||||
|
|
||||||
match iter.peek() {
|
match iter.peek() {
|
||||||
Some(&x) if thread_nodes[*x].indentation() > indentation => {
|
Some(&x) if thread_nodes[x].indentation() > indentation => {
|
||||||
indentations.push(false);
|
indentations.push(false);
|
||||||
}
|
}
|
||||||
Some(&x) if thread_nodes[*x].indentation() < indentation => {
|
Some(&x) if thread_nodes[x].indentation() < indentation => {
|
||||||
for _ in 0..(indentation - thread_nodes[*x].indentation()) {
|
for _ in 0..(indentation - thread_nodes[x].indentation()) {
|
||||||
indentations.pop();
|
indentations.pop();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -63,12 +63,14 @@ impl ThreadView {
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
let threads = &mailbox.collection.threads;
|
let threads = &mailbox.collection.threads;
|
||||||
let thread_node = &threads.thread_nodes()[threads.root_set()[coordinates.2]];
|
let thread_node = &threads.thread_nodes()[threads.root_set(coordinates.2)];
|
||||||
|
|
||||||
if thread_node.message().is_some() {
|
if thread_node.message().is_some() {
|
||||||
stack.push((0, threads.root_set()[coordinates.2]));
|
stack.push((0, threads.root_set(coordinates.2)));
|
||||||
} else {
|
} else {
|
||||||
stack.push((1, thread_node.children()[0]));
|
for &c in thread_node.children().iter() {
|
||||||
|
stack.push((1, c));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
let mut view = ThreadView {
|
let mut view = ThreadView {
|
||||||
dirty: true,
|
dirty: true,
|
||||||
|
@ -313,7 +315,7 @@ impl ThreadView {
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
let threads = &mailbox.collection.threads;
|
let threads = &mailbox.collection.threads;
|
||||||
let thread_node = &threads.thread_nodes()[threads.root_set()[self.coordinates.2]];
|
let thread_node = &threads.thread_nodes()[threads.root_set(self.coordinates.2)];
|
||||||
let i = if let Some(i) = thread_node.message() {
|
let i = if let Some(i) = thread_node.message() {
|
||||||
i
|
i
|
||||||
} else {
|
} else {
|
||||||
|
@ -432,7 +434,7 @@ impl ThreadView {
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
let threads = &mailbox.collection.threads;
|
let threads = &mailbox.collection.threads;
|
||||||
let thread_node = &threads.thread_nodes()[threads.root_set()[self.coordinates.2]];
|
let thread_node = &threads.thread_nodes()[threads.root_set(self.coordinates.2)];
|
||||||
let i = if let Some(i) = thread_node.message() {
|
let i = if let Some(i) = thread_node.message() {
|
||||||
i
|
i
|
||||||
} else {
|
} else {
|
||||||
|
|
Loading…
Reference in New Issue