ps_parser/parser/
token.rs

1use std::{collections::BTreeSet, 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 string_set(&self) -> BTreeSet<&String> {
43        let mut string_set = BTreeSet::new();
44        for token in self.0.iter() {
45            match token {
46                Token::String(deobfuscated) | Token::StringExpandable(_, deobfuscated) => {
47                    let _ = string_set.insert(deobfuscated);
48                }
49                _ => {}
50            }
51        }
52        string_set
53    }
54
55    pub fn expandable_strings(&self) -> Vec<Token> {
56        self.0
57            .iter()
58            .filter(|token| matches!(token, Token::StringExpandable(..)))
59            .cloned()
60            .collect()
61    }
62
63    pub fn expression(&self) -> Vec<Token> {
64        self.0
65            .iter()
66            .filter(|token| matches!(token, Token::Expression(..)))
67            .cloned()
68            .collect()
69    }
70
71    pub fn function(&self) -> Vec<Token> {
72        self.0
73            .iter()
74            .filter(|token| matches!(token, Token::Function(..)))
75            .cloned()
76            .collect()
77    }
78}
ps_parser/parser/token.rs

ps_parser/parser/
token.rs
