web & server: plumb debugging info for content type hierarchy.

Also cleanup Email trait.
This commit is contained in:
Bill Thiede 2023-11-27 13:47:02 -08:00
parent 87dfe4ace7
commit 48466808d3
4 changed files with 128 additions and 43 deletions

View File

@ -77,6 +77,7 @@ impl UnhandledContentType {
#[derive(Debug)] #[derive(Debug)]
pub struct PlainText { pub struct PlainText {
text: String, text: String,
content_tree: String,
} }
#[Object] #[Object]
@ -84,11 +85,15 @@ impl PlainText {
async fn contents(&self) -> &str { async fn contents(&self) -> &str {
&self.text &self.text
} }
async fn content_tree(&self) -> &str {
&self.content_tree
}
} }
#[derive(Debug)] #[derive(Debug)]
pub struct Html { pub struct Html {
html: String, html: String,
content_tree: String,
} }
#[Object] #[Object]
@ -96,6 +101,9 @@ impl Html {
async fn contents(&self) -> &str { async fn contents(&self) -> &str {
&self.html &self.html
} }
async fn content_tree(&self) -> &str {
&self.content_tree
}
} }
#[derive(Debug, Union)] #[derive(Debug, Union)]
@ -105,6 +113,21 @@ pub enum Body {
Html(Html), Html(Html),
} }
impl Body {
fn html(html: String) -> Body {
Body::Html(Html {
html,
content_tree: "".to_string(),
})
}
fn text(text: String) -> Body {
Body::PlainText(PlainText {
text,
content_tree: "".to_string(),
})
}
}
#[derive(Debug, SimpleObject)] #[derive(Debug, SimpleObject)]
pub struct Email { pub struct Email {
pub name: Option<String>, pub name: Option<String>,
@ -216,6 +239,12 @@ impl QueryRoot {
// TODO(wathiede): normalize all email addresses through an address book with preferred // TODO(wathiede): normalize all email addresses through an address book with preferred
// display names (that default to the most commonly seen name). // display names (that default to the most commonly seen name).
let nm = ctx.data_unchecked::<Notmuch>(); let nm = ctx.data_unchecked::<Notmuch>();
let debug_content_tree = ctx
.look_ahead()
.field("messages")
.field("body")
.field("contentTree")
.exists();
let mut messages = Vec::new(); let mut messages = Vec::new();
for path in nm.files(&thread_id)? { for path in nm.files(&thread_id)? {
let path = path?; let path = path?;
@ -243,8 +272,21 @@ impl QueryRoot {
.get_first_value("date") .get_first_value("date")
.and_then(|d| mailparse::dateparse(&d).ok()); .and_then(|d| mailparse::dateparse(&d).ok());
let body = match extract_body(&m)? { let body = match extract_body(&m)? {
Body::Html(Html { html }) => Body::Html(Html { Body::PlainText(PlainText { text, content_tree }) => Body::PlainText(PlainText {
text,
content_tree: if debug_content_tree {
render_content_type_tree(&m)
} else {
content_tree
},
}),
Body::Html(Html { html, content_tree }) => Body::Html(Html {
html: ammonia::clean(&html), html: ammonia::clean(&html),
content_tree: if debug_content_tree {
render_content_type_tree(&m)
} else {
content_tree
},
}), }),
b => b, b => b,
}; };
@ -274,8 +316,8 @@ impl QueryRoot {
fn extract_body(m: &ParsedMail) -> Result<Body, Error> { fn extract_body(m: &ParsedMail) -> Result<Body, Error> {
let body = m.get_body()?; let body = m.get_body()?;
let ret = match m.ctype.mimetype.as_str() { let ret = match m.ctype.mimetype.as_str() {
"text/plain" => return Ok(Body::PlainText(PlainText { text: body })), "text/plain" => return Ok(Body::text(body)),
"text/html" => return Ok(Body::Html(Html { html: body })), "text/html" => return Ok(Body::html(body)),
"multipart/mixed" => extract_mixed(m), "multipart/mixed" => extract_mixed(m),
"multipart/alternative" => extract_alternative(m), "multipart/alternative" => extract_alternative(m),
_ => extract_unhandled(m), _ => extract_unhandled(m),
@ -301,13 +343,13 @@ fn extract_alternative(m: &ParsedMail) -> Result<Body, Error> {
for sp in &m.subparts { for sp in &m.subparts {
if sp.ctype.mimetype == "text/html" { if sp.ctype.mimetype == "text/html" {
let body = sp.get_body()?; let body = sp.get_body()?;
return Ok(Body::Html(Html { html: body })); return Ok(Body::html(body));
} }
} }
for sp in &m.subparts { for sp in &m.subparts {
if sp.ctype.mimetype == "text/plain" { if sp.ctype.mimetype == "text/plain" {
let body = sp.get_body()?; let body = sp.get_body()?;
return Ok(Body::PlainText(PlainText { text: body })); return Ok(Body::text(body));
} }
} }
Err("extract_alternative".into()) Err("extract_alternative".into())
@ -327,8 +369,8 @@ fn extract_mixed(m: &ParsedMail) -> Result<Body, Error> {
for sp in &m.subparts { for sp in &m.subparts {
let body = sp.get_body()?; let body = sp.get_body()?;
match sp.ctype.mimetype.as_str() { match sp.ctype.mimetype.as_str() {
"text/plain" => return Ok(Body::PlainText(PlainText { text: body })), "text/plain" => return Ok(Body::text(body)),
"text/html" => return Ok(Body::Html(Html { html: body })), "text/html" => return Ok(Body::html(body)),
_ => (), _ => (),
} }
} }
@ -340,13 +382,13 @@ fn extract_related(m: &ParsedMail) -> Result<Body, Error> {
for sp in &m.subparts { for sp in &m.subparts {
if sp.ctype.mimetype == "text/html" { if sp.ctype.mimetype == "text/html" {
let body = sp.get_body()?; let body = sp.get_body()?;
return Ok(Body::Html(Html { html: body })); return Ok(Body::html(body));
} }
} }
for sp in &m.subparts { for sp in &m.subparts {
if sp.ctype.mimetype == "text/plain" { if sp.ctype.mimetype == "text/plain" {
let body = sp.get_body()?; let body = sp.get_body()?;
return Ok(Body::PlainText(PlainText { text: body })); return Ok(Body::text(body));
} }
} }
Err("extract_related".into()) Err("extract_related".into())

View File

@ -159,6 +159,22 @@
"ofType": null "ofType": null
} }
} }
},
{
"args": [],
"deprecationReason": null,
"description": null,
"isDeprecated": false,
"name": "contentTree",
"type": {
"kind": "NON_NULL",
"name": null,
"ofType": {
"kind": "SCALAR",
"name": "String",
"ofType": null
}
}
} }
], ],
"inputFields": null, "inputFields": null,
@ -400,6 +416,22 @@
"ofType": null "ofType": null
} }
} }
},
{
"args": [],
"deprecationReason": null,
"description": null,
"isDeprecated": false,
"name": "contentTree",
"type": {
"kind": "NON_NULL",
"name": null,
"ofType": {
"kind": "SCALAR",
"name": "String",
"ofType": null
}
}
} }
], ],
"inputFields": null, "inputFields": null,

View File

@ -23,9 +23,11 @@ query ShowThreadQuery($threadId: String!) {
} }
... on PlainText { ... on PlainText {
contents contents
contentTree
} }
... on Html { ... on Html {
contents contents
contentTree
} }
} }
path path

View File

@ -985,58 +985,57 @@ fn view_search_pager_legacy(start: usize, count: usize, total: usize) -> Node<Ms
} }
trait Email { trait Email {
fn name(&self) -> &Option<String>; fn name(&self) -> Option<&str>;
fn addr(&self) -> &Option<String>; fn addr(&self) -> Option<&str>;
} }
impl<T: Email> Email for &'_ T { impl<T: Email> Email for &'_ T {
fn name(&self) -> &Option<String> { fn name(&self) -> Option<&str> {
return (*self).name(); return (*self).name();
} }
fn addr(&self) -> &Option<String> { fn addr(&self) -> Option<&str> {
return (*self).addr(); return (*self).addr();
} }
} }
impl Email for ShowThreadQueryThreadMessagesCc { impl Email for ShowThreadQueryThreadMessagesCc {
fn name(&self) -> &Option<String> { fn name(&self) -> Option<&str> {
return &self.name; self.name.as_deref()
} }
fn addr(&self) -> &Option<String> { fn addr(&self) -> Option<&str> {
return &self.addr; self.addr.as_deref()
} }
} }
impl Email for ShowThreadQueryThreadMessagesFrom { impl Email for ShowThreadQueryThreadMessagesFrom {
fn name(&self) -> &Option<String> { fn name(&self) -> Option<&str> {
return &self.name; self.name.as_deref()
} }
fn addr(&self) -> &Option<String> { fn addr(&self) -> Option<&str> {
return &self.addr; self.addr.as_deref()
} }
} }
impl Email for ShowThreadQueryThreadMessagesTo { impl Email for ShowThreadQueryThreadMessagesTo {
fn name(&self) -> &Option<String> { fn name(&self) -> Option<&str> {
return &self.name; self.name.as_deref()
} }
fn addr(&self) -> &Option<String> { fn addr(&self) -> Option<&str> {
return &self.addr; self.addr.as_deref()
} }
} }
fn view_addresses<E: Email>(addrs: &[E]) -> Vec<Node<Msg>> { fn view_address(email: impl Email) -> Node<Msg> {
addrs span![
.into_iter() C!["tag", "is-black"],
.map(|address| { email.addr().as_ref().map(|a| attrs! {At::Title=>a}),
span![ email
C!["tag", "is-black"], .name()
address.addr().as_ref().map(|a| attrs! {At::Title=>a}), .as_ref()
address .unwrap_or(&email.addr().unwrap_or("(UNKNOWN)"))
.name() ]
.as_ref() }
.unwrap_or(address.addr().as_ref().unwrap_or(&"(UNKNOWN)".to_string()))
] fn view_addresses(addrs: &[impl Email]) -> Vec<Node<Msg>> {
}) addrs.into_iter().map(view_address).collect::<Vec<_>>()
.collect::<Vec<_>>()
} }
fn view_thread(thread: &ShowThreadQueryThread) -> Node<Msg> { fn view_thread(thread: &ShowThreadQueryThread) -> Node<Msg> {
@ -1061,11 +1060,21 @@ fn view_thread(thread: &ShowThreadQueryThread) -> Node<Msg> {
ShowThreadQueryThreadMessagesBodyOnUnhandledContentType { contents }, ShowThreadQueryThreadMessagesBodyOnUnhandledContentType { contents },
) => pre![C!["error"], contents], ) => pre![C!["error"], contents],
ShowThreadQueryThreadMessagesBody::PlainText( ShowThreadQueryThreadMessagesBody::PlainText(
ShowThreadQueryThreadMessagesBodyOnPlainText { contents }, ShowThreadQueryThreadMessagesBodyOnPlainText {
) => div![C!["view-part-text-plain"], contents], contents,
content_tree,
},
) => div![C!["view-part-text-plain"], contents, pre![content_tree]],
ShowThreadQueryThreadMessagesBody::Html( ShowThreadQueryThreadMessagesBody::Html(
ShowThreadQueryThreadMessagesBodyOnHtml { contents }, ShowThreadQueryThreadMessagesBodyOnHtml {
) => div![C!["view-part-text-html"], raw![contents]], contents,
content_tree,
},
) => div![
C!["view-part-text-html"],
raw![contents],
pre![content_tree]
],
} }
], ],
] ]