asbru 0.0.12

A Data-oriented service mesh generator
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

use crate::codegen::render::graphql::enum_value_definition::EnumDefinitionExt;
use crate::codegen::render::graphql::field::FieldDefinitionExt;
use crate::codegen::{context::Context, generate::GenericErrors, render::render::Render};
use async_graphql_parser::types::{TypeDefinition, TypeKind};
use codegen::{Enum, Impl, Scope};

pub struct EnumWrapper<'a> {
    // We store the whole type definition because we might need directives but it's an enum, we
    // should refine this type later.
    pub doc: &'a TypeDefinition,
    pub context: &'a Context<'a>,
}

impl<'a> EnumWrapper<'a> {
    fn object_name(&self) -> &'a str {
        self.doc.name.node.as_str()
    }

    fn domain_name(&self) -> String {
        format!("{}.rs", self.object_name().to_lowercase())
    }

    /// Generate an enum file for the actual type.
    /// We create a representation for each fields with no arguments and no directive.
    pub fn generate_enum_file(&self) -> Result<(), GenericErrors> {
        let mut scope = Scope::new();
        scope.import("serde", "Serialize");
        scope.import("serde", "Deserialize");
        scope.import("async_graphql", "*");

        let enum_struct = scope
            .new_enum(&format!("{}", self.object_name()))
            .vis("pub")
            .derive("Enum")
            .derive("Debug")
            .derive("Clone")
            .derive("Deserialize")
            .derive("Serialize")
            .derive("Eq")
            .derive("PartialEq")
            .derive("Copy");

        // Add field for it.
        let _fields = match &self.doc.kind {
            TypeKind::Enum(enum_values) => &enum_values.values,
            _ => {
                return Err(GenericErrors::GenericGeneratorError);
            }
        }
        .iter()
        .filter_map(|x| {
            let enum_value = &x.node;
            enum_value.generate_enum_value(enum_struct);
            Some(())
        })
        .collect::<Vec<()>>();

        self.context.create_a_new_file(
            format!("domain/{}", &self.domain_name()),
            scope.to_string().as_bytes(),
        )?;

        Ok(())
    }
}

impl<'a> Render for EnumWrapper<'a> {
    fn generate(&self) -> Result<(), GenericErrors> {
        // let object_name = self.doc.name.node.as_str();

        self.generate_enum_file()?;

        Ok(())
    }
}