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

use std::collections::HashMap;

use gitql_ast::scope::Scope;
use gitql_ast::statement::AggregateFunction;

pub struct ParserContext {
    pub symbol_table: Scope,
    pub aggregations: HashMap<String, AggregateFunction>,

    pub selected_fields: Vec<String>,
    pub hidden_selections: Vec<String>,

    pub generated_field_count: i32,
    pub is_single_value_query: bool,
    pub has_group_by_statement: bool,
}

impl Default for ParserContext {
    fn default() -> Self {
        ParserContext {
            symbol_table: Scope::new(),
            aggregations: HashMap::new(),
            selected_fields: Vec::new(),
            hidden_selections: Vec::new(),
            generated_field_count: 0,
            is_single_value_query: false,
            has_group_by_statement: false,
        }
    }
}

impl ParserContext {
    pub fn generate_column_name(&mut self) -> String {
        self.generated_field_count += 1;
        format!("column_{}", self.generated_field_count)
    }
}