golem-schema-derive 1.6.0-dev.2

Derive macros for the Golem schema model
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

// Copyright 2024-2026 Golem Cloud
//
// Licensed under the Golem Source License v1.1 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://license.golem.cloud/LICENSE
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! Codegen entry points.

use crate::codegen::r#enum::{expand_enum_from_schema, expand_enum_into_schema};
use crate::codegen::r#struct::{expand_struct_from_schema, expand_struct_into_schema};
use crate::codegen::union::{expand_union_from_schema, expand_union_into_schema};
use crate::parse::parse_type_attrs;
use proc_macro2::TokenStream;
use syn::{Data, DeriveInput};

pub fn expand_into_schema(input: DeriveInput) -> syn::Result<TokenStream> {
    let type_attrs = parse_type_attrs(&input.attrs)?;
    match &input.data {
        Data::Struct(data) => expand_struct_into_schema(&input, &type_attrs, data),
        Data::Enum(data) => {
            if type_attrs.union {
                expand_union_into_schema(&input, &type_attrs, data)
            } else {
                expand_enum_into_schema(&input, &type_attrs, data)
            }
        }
        Data::Union(_) => Err(syn::Error::new_spanned(
            input.ident,
            "`union` types are not supported by `IntoSchema`",
        )),
    }
}

pub fn expand_from_schema(input: DeriveInput) -> syn::Result<TokenStream> {
    let type_attrs = parse_type_attrs(&input.attrs)?;
    match &input.data {
        Data::Struct(data) => expand_struct_from_schema(&input, &type_attrs, data),
        Data::Enum(data) => {
            if type_attrs.union {
                expand_union_from_schema(&input, &type_attrs, data)
            } else {
                expand_enum_from_schema(&input, &type_attrs, data)
            }
        }
        Data::Union(_) => Err(syn::Error::new_spanned(
            input.ident,
            "`union` types are not supported by `FromSchema`",
        )),
    }
}