use mail_parser::HeaderName;
use crate::compiler::{
grammar::instruction::{CompilerState, MapLocalVars},
lexer::{word::Word, Token},
CompileError, ErrorType, Value,
};
use crate::compiler::grammar::test::Test;
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(
any(test, feature = "serde"),
derive(serde::Serialize, serde::Deserialize)
)]
#[cfg_attr(
feature = "rkyv",
derive(rkyv::Serialize, rkyv::Deserialize, rkyv::Archive)
)]
pub(crate) struct TestDuplicate {
pub handle: Option<Value>,
pub dup_match: DupMatch,
pub seconds: Option<u64>,
pub last: bool,
pub is_not: bool,
}
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(
any(test, feature = "serde"),
derive(serde::Serialize, serde::Deserialize)
)]
#[cfg_attr(
feature = "rkyv",
derive(rkyv::Serialize, rkyv::Deserialize, rkyv::Archive)
)]
pub(crate) enum DupMatch {
Header(Value),
UniqueId(Value),
Default,
}
impl CompilerState<'_> {
pub(crate) fn parse_test_duplicate(&mut self) -> Result<Test, CompileError> {
let mut handle = None;
let mut dup_match = DupMatch::Default;
let mut seconds = None;
let mut last = false;
while let Some(token_info) = self.tokens.peek() {
let token_info = token_info?;
let line_num = token_info.line_num;
let line_pos = token_info.line_pos;
match token_info.token {
Token::Tag(Word::Handle) => {
self.validate_argument(1, None, line_num, line_pos)?;
self.tokens.next();
handle = self.parse_string()?.into();
}
Token::Tag(Word::Header) => {
self.validate_argument(2, None, line_num, line_pos)?;
self.tokens.next();
let header = self.parse_string()?;
if let Value::Text(header_name) = &header {
if HeaderName::parse(header_name.as_ref()).is_none() {
return Err(self
.tokens
.unwrap_next()?
.custom(ErrorType::InvalidHeaderName));
}
}
dup_match = DupMatch::Header(header);
}
Token::Tag(Word::UniqueId) => {
self.validate_argument(2, None, line_num, line_pos)?;
self.tokens.next();
dup_match = DupMatch::UniqueId(self.parse_string()?);
}
Token::Tag(Word::Seconds) => {
self.validate_argument(3, None, line_num, line_pos)?;
self.tokens.next();
seconds = (self.tokens.expect_number(u64::MAX as usize)? as u64).into();
}
Token::Tag(Word::Last) => {
self.validate_argument(4, None, line_num, line_pos)?;
self.tokens.next();
last = true;
}
_ => break,
}
}
Ok(Test::Duplicate(TestDuplicate {
handle,
dup_match,
seconds,
last,
is_not: false,
}))
}
}
impl MapLocalVars for DupMatch {
fn map_local_vars(&mut self, last_id: usize) {
match self {
DupMatch::Header(header) => header.map_local_vars(last_id),
DupMatch::UniqueId(unique_id) => unique_id.map_local_vars(last_id),
DupMatch::Default => {}
}
}
}