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

use crate::{ast, context::Context, ScalarFieldId, StringId};

pub(super) fn visit_model_field_native_type_attribute(
    id: ScalarFieldId,
    datasource_name: StringId,
    type_name: StringId,
    attr: &ast::Attribute,
    ctx: &mut Context<'_>,
) {
    let args = &attr.arguments;
    let args: Vec<String> = args.arguments.iter().map(|arg| arg.value.to_string()).collect();

    ctx.types[id].native_type = Some((datasource_name, type_name, args, attr.span))
}

pub(super) fn visit_composite_type_field_native_type_attribute(
    id: (ast::CompositeTypeId, ast::FieldId),
    datasource_name: StringId,
    type_name: StringId,
    attr: &ast::Attribute,
    ctx: &mut Context<'_>,
) {
    let args = &attr.arguments;
    let args: Vec<String> = args.arguments.iter().map(|arg| arg.value.to_string()).collect();

    if let Some(composite_type_field) = ctx.types.composite_type_fields.get_mut(&id) {
        composite_type_field.native_type = Some((datasource_name, type_name, args, attr.span))
    }
}