use super::super::tests::TestResult;
use crate::{Context, Sieve, bytecode::ops, runtime::RuntimeError};
use mail_parser::{
Encoding, Header, HeaderName, HeaderValue, MimeHeaders, PartType,
decoders::html::{html_to_text, text_to_html},
};
#[derive(Clone, Copy)]
enum Conversion {
TextToHtml,
TextPlainToHtml,
HtmlToText,
}
impl<'x> Context<'x> {
pub(crate) fn exec_convert(
&mut self,
script: &'x Sieve<'x>,
convert: &ops::Convert,
) -> Result<TestResult, RuntimeError> {
let from_media_type = self.eval_value(script, convert.from_media_type)?;
let to_media_type = self.eval_value(script, convert.to_media_type)?;
let from_media_type = from_media_type.to_string();
let to_media_type = to_media_type.to_string();
if from_media_type.eq_ignore_ascii_case(to_media_type.as_ref()) {
return Ok(TestResult::Bool(false ^ convert.is_not));
}
let conversion = if (from_media_type.eq_ignore_ascii_case("text")
|| from_media_type.starts_with("text/"))
&& to_media_type.eq_ignore_ascii_case("text/html")
{
if from_media_type.eq_ignore_ascii_case("text") {
Conversion::TextPlainToHtml
} else {
Conversion::TextToHtml
}
} else if from_media_type.eq_ignore_ascii_case("text/html")
&& to_media_type.eq_ignore_ascii_case("text/plain")
{
Conversion::HtmlToText
} else {
return Ok(TestResult::Bool(false ^ convert.is_not));
};
let mut did_convert = false;
for part in self.message.parts.iter_mut() {
let (new_body, ct) = match (&part.body, conversion) {
(PartType::Html(html), Conversion::HtmlToText) => (
PartType::Text(html_to_text(html.as_ref()).into()),
"text/plain; charset=utf8",
),
(PartType::Text(text), Conversion::TextToHtml) => (
PartType::Html(text_to_html(text.as_ref()).into()),
"text/html; charset=utf8",
),
(PartType::Text(text), Conversion::TextPlainToHtml)
if part
.content_type()
.and_then(|ct| ct.c_subtype.as_ref())
.is_some_and(|st| st.eq_ignore_ascii_case("plain")) =>
{
(
PartType::Html(text_to_html(text.as_ref()).into()),
"text/html; charset=utf8",
)
}
_ => {
continue;
}
};
part.headers = vec![Header {
name: HeaderName::Other("Content-Type".into()),
value: HeaderValue::Text(ct.into()),
offset_start: 0,
offset_end: 0,
offset_field: 0,
}];
self.message_size = self.message_size + ct.len() + new_body.len() + 16
- (if part.offset_body != 0 {
(part.offset_end - part.offset_header) as usize
} else {
part.body.len()
});
part.offset_body = 0;
part.body = new_body;
part.encoding = Encoding::QuotedPrintable;
did_convert = true;
}
if did_convert {
self.has_changes = true;
}
Ok(TestResult::Bool(did_convert ^ convert.is_not))
}
}