use crate::matcher::{Matcher, MatcherResult};
use crate::token::{TOKEN_TYPE_SYMBOL, Token};
use std::collections::HashMap;
#[derive(Clone, Debug, Copy)]
pub struct SymbolMatcher {
pub index: usize,
pub precedence: u8,
pub running: bool,
}
impl Matcher for SymbolMatcher {
fn reset(&mut self, _ctx: &mut Box<HashMap<String, i32>>) {
self.index = 0;
self.running = true;
}
fn find_match(
&mut self,
oc: Option<char>,
value: &[char],
_ctx: &mut Box<HashMap<String, i32>>,
) -> MatcherResult {
match oc {
Some(c) if !c.is_whitespace() && !c.is_alphanumeric() => {
self.index += 1;
MatcherResult::Running()
}
_ => self.generate_symbol_token(value),
}
}
fn is_running(&self) -> bool {
self.running
}
fn precedence(&self) -> u8 {
self.precedence
}
}
impl SymbolMatcher {
#[inline(always)]
fn generate_symbol_token(&mut self, value: &[char]) -> MatcherResult {
self.running = false;
if self.index > 0 {
MatcherResult::Matched(Token {
value: value[0..self.index].iter().collect(),
token_type: TOKEN_TYPE_SYMBOL,
len: self.index,
line: 0,
column: self.index,
precedence: self.precedence,
})
} else {
MatcherResult::Failed()
}
}
}
#[cfg(test)]
mod tests {
use crate::input::InputString;
use crate::matcher::Matcher;
use crate::matcher::symbol::SymbolMatcher;
use crate::matcher::whitespace::WhitespaceMatcher;
use crate::token::TOKEN_TYPE_SYMBOL;
use crate::{LexxError, Lexxer, Lexxor};
#[test]
fn matcher_symbol_matches_single_symbol() {
let mut lexxor: Box<dyn Lexxer> = Box::new(Lexxor::<512>::new(
Box::new(InputString::new(String::from("@"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
));
match lexxor.next_token() {
Err(e) => match e {
LexxError::TokenNotFound(_) => {
unreachable!("Should not have failed parsing file");
}
LexxError::Error(_) => {
unreachable!("Should not have failed parsing file");
}
},
Ok(Some(t)) => {
assert_eq!(t.value, "@");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
}
Ok(None) => {
unreachable!("Should not hit None");
}
}
}
#[test]
fn matcher_symbol_matches_multiple_symbols() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("!@#$%^&*"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
);
match lexxor.next_token() {
Err(e) => match e {
LexxError::TokenNotFound(_) => {
unreachable!("Should not have failed parsing file");
}
LexxError::Error(_) => {
unreachable!("Should not have failed parsing file");
}
},
Ok(Some(t)) => {
assert_eq!(t.value, "!@#$%^&*");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
}
Ok(None) => {
unreachable!("Should not hit None");
}
}
}
#[test]
fn matcher_symbol_stops_at_alphanumeric() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("@#$abc"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
);
match lexxor.next_token() {
Ok(Some(t)) => {
assert_eq!(t.value, "@#$");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
}
_ => {
unreachable!("Should have matched symbols");
}
}
}
#[test]
fn matcher_symbol_stops_at_whitespace() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("@#$ %^&"))),
vec![
Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
}),
Box::new(WhitespaceMatcher {
index: 0,
column: 0,
line: 0,
precedence: 0,
running: true,
}),
],
);
match lexxor.next_token() {
Ok(Some(t)) => {
assert_eq!(t.value, "@#$");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
}
_ => {
unreachable!("Should have matched symbols");
}
}
assert!(
matches!(lexxor.next_token(), Ok(Some(t)) if t.token_type == crate::token::TOKEN_TYPE_WHITESPACE)
);
match lexxor.next_token() {
Ok(Some(t)) => {
assert_eq!(t.value, "%^&");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
}
_ => {
unreachable!("Should have matched symbols");
}
}
}
#[test]
fn matcher_symbol_handles_empty_input() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from(""))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
);
assert!(matches!(lexxor.next_token(), Ok(None)));
}
#[test]
fn matcher_symbol_handles_non_symbol_input() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("abc123"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
);
match lexxor.next_token() {
Err(LexxError::TokenNotFound(_)) => {
}
_ => {
unreachable!("Should have failed to match");
}
}
}
#[test]
fn matcher_symbol_resets_properly() {
let mut matcher = SymbolMatcher {
index: 5,
precedence: 0,
running: false,
};
let mut ctx = Box::new(std::collections::HashMap::<String, i32>::new());
matcher.reset(&mut ctx);
assert_eq!(matcher.index, 0);
assert!(matcher.running);
}
#[test]
fn matcher_symbol_returns_correct_token_type() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("+-*/"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
);
match lexxor.next_token() {
Ok(Some(t)) => {
assert_eq!(t.value, "+-*/");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
assert_eq!(t.precedence, 0);
}
_ => {
unreachable!("Should have matched symbols");
}
}
}
#[test]
fn matcher_symbol_respects_precedence() {
let custom_precedence = 5;
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("++"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: custom_precedence,
running: true,
})],
);
match lexxor.next_token() {
Ok(Some(t)) => {
assert_eq!(t.precedence, custom_precedence);
}
_ => {
unreachable!("Should have matched symbols");
}
}
}
#[test]
fn matcher_symbol_handles_unicode_symbols() {
let mut lexxor = Lexxor::<512>::new(
Box::new(InputString::new(String::from("§¶†‡"))),
vec![Box::new(SymbolMatcher {
index: 0,
precedence: 0,
running: true,
})],
);
match lexxor.next_token() {
Ok(Some(t)) => {
assert_eq!(t.value, "§¶†‡");
assert_eq!(t.token_type, TOKEN_TYPE_SYMBOL);
}
_ => {
unreachable!("Should have matched symbols");
}
}
}
}