mirror of https://git.meli.delivery/meli/meli
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
4.2 KiB
Rust
121 lines
4.2 KiB
Rust
/*
|
|
* Copyright (c) 2020-2023, Stalwart Labs Ltd.
|
|
*
|
|
* This file is part of the Stalwart Sieve Interpreter.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This program 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 Affero General Public License for more details.
|
|
* in the LICENSE file at the top-level directory of this distribution.
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* You can be released from the requirements of the AGPLv3 license by
|
|
* purchasing a commercial license. Please contact licensing@stalw.art
|
|
* for more details.
|
|
*/
|
|
|
|
use mail_parser::HeaderName;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::sieve::compiler::{
|
|
grammar::instruction::{CompilerState, MapLocalVars},
|
|
lexer::{word::Word, Token},
|
|
CompileError, ErrorType, Value,
|
|
};
|
|
|
|
use crate::sieve::compiler::grammar::test::Test;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
|
|
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, Serialize, Deserialize)]
|
|
pub(crate) enum DupMatch {
|
|
Header(Value),
|
|
UniqueId(Value),
|
|
Default,
|
|
}
|
|
|
|
impl<'x> CompilerState<'x> {
|
|
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).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 => {}
|
|
}
|
|
}
|
|
}
|