fraiseql-cli 2.3.2

CLI tools for FraiseQL v2 - Schema compilation and development utilities
Documentation 
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

//! Scala code generator.

use super::{
    super::init::Language,
    SchemaGenerator,
    utils::{infer_sql_source, map_graphql_to_lang, to_camel_case, wrap_nullable},
};
use crate::schema::intermediate::{
    IntermediateEnum, IntermediateQuery, IntermediateSchema, IntermediateType,
};

// =============================================================================
// Scala generator
// =============================================================================

pub(super) struct ScalaGenerator;

impl SchemaGenerator for ScalaGenerator {
    fn generate(&self, schema: &IntermediateSchema) -> String {
        let mut out = String::new();

        for enum_def in &schema.enums {
            generate_scala_enum(&mut out, enum_def);
        }

        for ty in &schema.types {
            generate_scala_type(&mut out, ty);
        }

        for query in &schema.queries {
            generate_scala_query(&mut out, query);
        }

        while out.ends_with("\n\n") {
            out.pop();
        }
        if !out.ends_with('\n') {
            out.push('\n');
        }
        out
    }
}

fn generate_scala_enum(out: &mut String, enum_def: &IntermediateEnum) {
    out.push_str(&format!("object {} extends Enumeration {{\n", enum_def.name));
    let names: Vec<String> = enum_def
        .values
        .iter()
        .map(|v| format!("val {} = Value(\"{}\")", v.name, v.name))
        .collect();
    out.push_str(&format!("  {}\n", names.join("; ")));
    out.push_str("}\n\n");
}

fn generate_scala_type(out: &mut String, ty: &IntermediateType) {
    let sql_source = infer_sql_source(&ty.name);
    out.push_str(&format!("@Type(sqlSource = \"{sql_source}\")\n"));
    out.push_str(&format!("case class {}(\n", ty.name));

    for (i, field) in ty.fields.iter().enumerate() {
        let lang_type = map_graphql_to_lang(Language::Scala, &field.field_type);
        let type_str = if field.nullable {
            wrap_nullable(Language::Scala, &lang_type)
        } else {
            lang_type
        };
        let comma = if i + 1 < ty.fields.len() { "," } else { "" };
        out.push_str(&format!("  {}: {type_str}{comma}\n", to_camel_case(&field.name)));
    }
    out.push_str(")\n\n");
}

fn generate_scala_query(out: &mut String, query: &IntermediateQuery) {
    let sql_source = query.sql_source.as_deref().unwrap_or("v_unknown");
    let mut params = vec![format!("returnType = classOf[{}]", query.return_type)];
    if query.returns_list {
        params.push("returnArray = true".to_string());
    }
    params.push(format!("sqlSource = \"{sql_source}\""));

    out.push_str(&format!("@Query({})\n", params.join(", ")));

    let ret_type = if query.returns_list {
        format!("List[{}]", query.return_type)
    } else {
        query.return_type.clone()
    };

    if query.arguments.is_empty() {
        out.push_str(&format!("def {}(): {ret_type} = ???\n\n", query.name));
    } else {
        let args: Vec<String> = query
            .arguments
            .iter()
            .map(|a| {
                let lang_type = map_graphql_to_lang(Language::Scala, &a.arg_type);
                format!("{}: {lang_type}", a.name)
            })
            .collect();
        out.push_str(&format!("def {}({}): {ret_type} = ???\n\n", query.name, args.join(", ")));
    }
}