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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use std::str::FromStr;

mod json_types;
pub mod types;

/// Parses `output/function_names.json` and returns a vector of function_names.
pub fn get_bigquery_function_names() -> Vec<String> {
    let contents = include_str!("../output/function_names.json");
    let function_names: Vec<String> = serde_json::from_str(&contents).unwrap();
    return function_names;
}

/// Parses `output/functions.json` and returns a vector of categories.
pub fn get_bigquery_function_categories() -> Vec<String> {
    let contents = include_str!("../output/categories.json");
    let categories: Vec<String> = serde_json::from_str(&contents).unwrap();
    return categories;
}

/// Parses `output/functions.json` and returns a vector of functions.
pub fn get_bigquery_functions() -> Vec<types::Function> {
    let contents = include_str!("../output/functions.json");
    let functions: Vec<json_types::Function> = serde_json::from_str(&contents).unwrap();

    let converted_functions = functions
        .into_iter()
        .map(|function| {
            types::Function::new(
                function.name,
                function
                    .arguments
                    .into_iter()
                    .map(|argument| {
                        types::Argument::new(argument.name, argument.supported_argument_type)
                    })
                    .collect(),
                types::Category::from_str(&function.category)
                    .unwrap_or(types::Category::NoCategory),
                function.description,
                function.description_markdown,
            )
        })
        .collect();

    return converted_functions;
}

pub fn get_distinct_allowed_categories() -> [types::Category; 3] {
    [
        types::Category::Aggregate,
        types::Category::Approximate_aggregate,
        types::Category::HyperLogLog,
    ]
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::types::Category;

    #[test]
    fn test_get_bigquery_function_names() {
        let function_names = get_bigquery_function_names();
        assert_eq!(function_names.len(), 300);
    }

    #[test]
    fn test_get_bigquery_function_categories() {
        let categories = get_bigquery_function_categories();
        assert_eq!(categories.len(), 25);
    }

    #[test]
    fn test_get_bigquery_functions() {
        let functions = get_bigquery_functions();
        assert_eq!(functions.len(), 300);
    }

    #[test]
    fn test_get_distinct_allowed_categories() {
        let categories = get_distinct_allowed_categories();
        assert_eq!(categories.len(), 3);
        assert!(categories.contains(&Category::Aggregate));
        assert!(categories.contains(&Category::Approximate_aggregate));
        assert!(categories.contains(&Category::HyperLogLog));
    }
}