cratestack-macros 0.3.7

Rust-native schema-first framework for typed HTTP APIs, generated clients, and backend services.
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

mod include;

use std::collections::BTreeSet;

use cratestack_core::{Field, Model};
use quote::quote;

use crate::shared::{ident, relation_model_fields, rust_type_tokens, to_snake_case};

use include::build_selection_include_accessor;

pub(super) struct SelectionRelationEntry {
    pub(super) include_methods: proc_macro2::TokenStream,
    pub(super) include_field: proc_macro2::TokenStream,
    pub(super) include_query_step: proc_macro2::TokenStream,
    pub(super) include_accessor: proc_macro2::TokenStream,
}

pub(super) fn build_selection_field_methods(fields: &[&Field]) -> Vec<proc_macro2::TokenStream> {
    fields
        .iter()
        .map(|field| {
            let method_ident = ident(&field.name);
            let field_name = &field.name;
            quote! {
                #[allow(non_snake_case)]
                pub fn #method_ident(mut self) -> Self {
                    self.fields
                        .get_or_insert_with(::std::collections::BTreeSet::new)
                        .insert(#field_name);
                    self
                }
            }
        })
        .collect()
}

pub(super) fn build_selected_scalar_accessors(
    fields: &[&Field],
    model_name: &str,
) -> Vec<proc_macro2::TokenStream> {
    fields
        .iter()
        .map(|field| {
            let method_ident = ident(&field.name);
            let field_name = &field.name;
            let field_type = rust_type_tokens(&field.ty);
            quote! {
                #[allow(non_snake_case)]
                pub fn #method_ident(&self) -> Result<#field_type, ::cratestack::CoolError> {
                    decode_projected_field::<#field_type>(
                        &self.fields,
                        self.allows_field(#field_name),
                        #model_name,
                        #field_name,
                    )
                }
            }
        })
        .collect()
}

pub(super) fn build_selection_relation_entries(
    model: &Model,
    model_names: &BTreeSet<&str>,
    model_name: &str,
) -> Result<Vec<SelectionRelationEntry>, String> {
    relation_model_fields(model, model_names)
        .into_iter()
        .map(|field| build_selection_relation_entry(field, model_name))
        .collect()
}

fn build_selection_relation_entry(
    field: &Field,
    model_name: &str,
) -> Result<SelectionRelationEntry, String> {
    let include_method_ident = ident(&format!("include_{}", field.name));
    let include_selected_method_ident = ident(&format!("include_{}_selected", field.name));
    let include_name = field.name.clone();
    let include_field_ident = ident(&field.name);
    let target_module_ident = ident(&to_snake_case(&field.ty.name));
    let target_include_selection =
        quote! { super::super::#target_module_ident::selection::IncludeSelection };

    let include_methods = quote! {
        #[allow(non_snake_case)]
        pub fn #include_method_ident(mut self) -> Self {
            self.includes.#include_field_ident = Some(Box::new(#target_include_selection::default()));
            self
        }

        #[allow(non_snake_case)]
        pub fn #include_selected_method_ident(
            mut self,
            selection: #target_include_selection,
        ) -> Self {
            self.includes.#include_field_ident = Some(Box::new(selection));
            self
        }
    };
    let include_field =
        quote! { pub #include_field_ident: Option<Box<#target_include_selection>>, };
    let include_query_step = quote! {
        if let Some(selection) = &self.includes.#include_field_ident {
            let prefix = #include_name;
            query.includes.push(prefix.to_owned());
            let include_query = selection.to_query();
            if !include_query.fields.is_empty() {
                query.include_fields.insert(prefix.to_owned(), include_query.fields);
            }
            for nested_include in include_query.includes {
                query.includes.push(format!("{prefix}.{nested_include}"));
            }
            for (path, fields) in include_query.include_fields {
                query.include_fields.insert(format!("{prefix}.{path}"), fields);
            }
        }
    };
    let include_accessor = build_selection_include_accessor(
        field,
        model_name,
        &include_name,
        &include_field_ident,
        &target_module_ident,
    );

    Ok(SelectionRelationEntry {
        include_methods,
        include_field,
        include_query_step,
        include_accessor,
    })
}