ps_parser/parser/
token.rs

1use std::fmt::Display;
2
3use super::script_result::PsValue;
4
5#[derive(Debug, Clone, PartialEq)]
6pub enum Token {
7    StringExpandable(String, String),
8    String(String),
9    Expression(String, PsValue),
10    Function(String, String, Vec<PsValue>),
11}
12
13impl Display for Token {
14    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
15        write!(f, "{:?}", self)
16    }
17}
18
19#[derive(Default, Debug, Clone, PartialEq)]
20pub struct Tokens(Vec<Token>);
21impl Tokens {
22    pub fn new() -> Self {
23        Self(Vec::new())
24    }
25
26    pub fn push(&mut self, token: Token) {
27        self.0.push(token)
28    }
29
30    pub fn all(&self) -> Vec<Token> {
31        self.0.clone()
32    }
33
34    pub fn strings(&self) -> Vec<Token> {
35        self.0
36            .iter()
37            .filter(|token| matches!(token, Token::String(..)))
38            .cloned()
39            .collect()
40    }
41
42    pub fn expandable_strings(&self) -> Vec<Token> {
43        self.0
44            .iter()
45            .filter(|token| matches!(token, Token::StringExpandable(..)))
46            .cloned()
47            .collect()
48    }
49
50    pub fn expression(&self) -> Vec<Token> {
51        self.0
52            .iter()
53            .filter(|token| matches!(token, Token::Expression(..)))
54            .cloned()
55            .collect()
56    }
57
58    pub fn function(&self) -> Vec<Token> {
59        self.0
60            .iter()
61            .filter(|token| matches!(token, Token::Function(..)))
62            .cloned()
63            .collect()
64    }
65}