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

pub mod config_engines;
pub mod types;

use types::{Stack, StackType};

use lazy_static::lazy_static;

lazy_static! {
    pub static ref API: Stack =
        serde_yaml::from_str(include_str!("templates/api.yaml")).expect("api.yaml not found");
    pub static ref DATAWAREHOUSE: Stack =
        serde_yaml::from_str(include_str!("templates/data_warehouse.yaml"))
            .expect("data_warehouse.yaml not found");
    pub static ref GEOSPATIAL: Stack =
        serde_yaml::from_str(include_str!("templates/gis.yaml")).expect("gis.yaml not found");
    pub static ref ML: Stack =
        serde_yaml::from_str(include_str!("templates/machine_learning.yaml"))
            .expect("machine_learning.yaml not found");
    pub static ref MONGO_ALTERNATIVE: Stack =
        serde_yaml::from_str(include_str!("templates/mongo_alternative.yaml"))
            .expect("mongo_alternative.yaml not found");
    pub static ref MQ: Stack = serde_yaml::from_str(include_str!("templates/message_queue.yaml"))
        .expect("message_queue.yaml not found");
    pub static ref OLAP: Stack =
        serde_yaml::from_str(include_str!("templates/olap.yaml")).expect("olap.yaml not found");
    pub static ref OLTP: Stack =
        serde_yaml::from_str(include_str!("templates/oltp.yaml")).expect("oltp.yaml not found");
    pub static ref RAG: Stack =
        serde_yaml::from_str(include_str!("templates/rag.yaml")).expect("rag.yaml not found");
    pub static ref STANDARD: Stack = serde_yaml::from_str(include_str!("templates/standard.yaml"))
        .expect("standard.yaml not found");
    pub static ref VECTOR_DB: Stack = serde_yaml::from_str(include_str!("templates/vectordb.yaml"))
        .expect("vectordb.yaml not found");
}

pub fn get_stack(entity: StackType) -> types::Stack {
    match entity {
        StackType::API => API.clone(),
        StackType::DataWarehouse => DATAWAREHOUSE.clone(),
        StackType::Geospatial => GEOSPATIAL.clone(),
        StackType::MachineLearning => ML.clone(),
        StackType::MessageQueue => MQ.clone(),
        StackType::MongoAlternative => MONGO_ALTERNATIVE.clone(),
        StackType::OLAP => OLAP.clone(),
        StackType::OLTP => OLTP.clone(),
        StackType::RAG => RAG.clone(),
        StackType::Standard => STANDARD.clone(),
        StackType::VectorDB => VECTOR_DB.clone(),
    }
}