1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use crate::lexer::{Token, TokenList, tokenize_internal};
use crate::keywords::{RegexToken, sql_regex, init_trie};
use crate::filters::{Filter, StmtFilter, TokenListFilter};
use crate::trie::Trie;

// 'a
pub struct FilterStack {
    regs: Vec<RegexToken>,
    trie: Trie,
    pub preprocess: Vec<Box<dyn Filter>>,
    pub stmtprocess: Vec<Box<dyn StmtFilter>>,
    pub tlistprocess: Vec<Box<dyn TokenListFilter>>,
    pub postprocess: Vec<Box<dyn StmtFilter>>,
}


impl FilterStack {

    pub fn new() -> Self {
        Self { 
            regs: sql_regex(),
            trie: init_trie(),
            preprocess: vec![],
            stmtprocess: vec![],
            postprocess: vec![],
            tlistprocess: vec![],
        }
    }

    // TODO: support more than one sql
    pub fn run(&self, sql: &str, grouping: bool) -> Vec<Token> {
        let mut tokens = tokenize_internal(sql, &self.regs, &self.trie);
        if grouping {
            tokens = super::grouping::group(tokens);
        }
        tokens
    }

    // format sql
    pub fn format(&mut self, sql: &str, grouping: bool) -> Vec<Token> {
        let mut tokens = tokenize_internal(sql, &self.regs, &self.trie);
        for token in tokens.iter_mut() {
            self.preprocess.iter().for_each(|filter| filter.process(token));
        }
        if grouping {
            tokens = super::grouping::group(tokens);
        }
        self.stmtprocess.iter().for_each(|filter| filter.process(&mut tokens));
        // for token in tokens.iter() {
        //     println!("{:?}", token);
        // }
        let mut token_list = TokenList{ tokens: tokens };
        self.tlistprocess.iter_mut().for_each(|filter| filter.process(&mut token_list));
        tokens = std::mem::replace(&mut token_list.tokens, vec![]);
        self.postprocess.iter().for_each(|filter| filter.process(&mut tokens));
        tokens
    }
}