rapina-macros 0.12.0

Procedural macros for the Rapina web framework
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

//! Type mapping for schema fields to Rust/SeaORM types.

use proc_macro2::TokenStream;
use quote::quote;

/// Scalar types supported in schema definitions.
#[derive(Debug, Clone, PartialEq)]
pub enum ScalarType {
    String,
    Text,
    I32,
    I64,
    F32,
    F64,
    Bool,
    Uuid,
    DateTime,
    NaiveDateTime,
    Date,
    Time,
    Decimal,
    Json,
    Bytes,
}

impl ScalarType {
    /// Parse a type identifier into a scalar type.
    ///
    /// Unsigned integer types (`u8`, `u16`, `u32`, `u64`) are intentionally
    /// rejected here so callers can emit a focused compile-time error.
    /// Most SQL backends (including Postgres) lack a native unsigned
    /// integer type, and silently widening to a signed column truncates
    /// values above the signed maximum. Use [`Self::is_unsupported_unsigned`]
    /// to detect them in the parser.
    pub fn from_ident(ident: &str) -> Option<Self> {
        match ident {
            "String" => Some(ScalarType::String),
            "Text" => Some(ScalarType::Text),
            "i32" | "i16" | "i8" | "integer" | "int" => Some(ScalarType::I32),
            "i64" | "bigint" => Some(ScalarType::I64),
            "f32" | "float" => Some(ScalarType::F32),
            "f64" | "double" => Some(ScalarType::F64),
            "bool" | "boolean" => Some(ScalarType::Bool),
            "Uuid" | "uuid" => Some(ScalarType::Uuid),
            "DateTime" | "DateTimeUtc" | "timestamptz" => Some(ScalarType::DateTime),
            "NaiveDateTime" | "timestamp" => Some(ScalarType::NaiveDateTime),
            "Date" | "date" => Some(ScalarType::Date),
            "Time" | "time" => Some(ScalarType::Time),
            "Decimal" | "numeric" | "money" => Some(ScalarType::Decimal),
            "Json" | "json" | "jsonb" => Some(ScalarType::Json),
            "Bytes" | "Blob" | "binary" | "bytea" | "varbinary" => Some(ScalarType::Bytes),
            _ => None,
        }
    }

    /// Returns true for Rust unsigned integer types that would silently
    /// be coerced to a signed column type if accepted as a scalar.
    ///
    /// The parser uses this to surface a clear compile-time error rather
    /// than generating an `i32`/`i64` field that overflows above the
    /// signed maximum (issue #549).
    pub fn is_unsupported_unsigned(ident: &str) -> bool {
        matches!(ident, "u8" | "u16" | "u32" | "u64")
    }

    /// Generate the Rust type for this scalar.
    pub fn rust_type(&self) -> TokenStream {
        match self {
            ScalarType::String | ScalarType::Text => quote! { String },
            ScalarType::I32 => quote! { i32 },
            ScalarType::I64 => quote! { i64 },
            ScalarType::F32 => quote! { f32 },
            ScalarType::F64 => quote! { f64 },
            ScalarType::Bool => quote! { bool },
            ScalarType::Uuid => quote! { rapina::uuid::Uuid },
            ScalarType::DateTime => quote! { DateTimeUtc },
            ScalarType::NaiveDateTime => quote! { DateTime },
            ScalarType::Date => quote! { Date },
            ScalarType::Time => quote! { Time },
            ScalarType::Decimal => quote! { rapina::rust_decimal::Decimal },
            ScalarType::Json => quote! { Json },
            ScalarType::Bytes => quote! { Vec<u8> },
        }
    }

    /// Generate SeaORM column type attribute if needed.
    /// Returns None if the default column type is correct.
    pub fn column_type_attr(&self) -> Option<TokenStream> {
        match self {
            ScalarType::Text => Some(quote! { #[sea_orm(column_type = "Text")] }),
            ScalarType::Decimal => {
                Some(quote! { #[sea_orm(column_type = "Decimal(Some((19, 4)))")] })
            }
            ScalarType::Json => Some(quote! { #[sea_orm(column_type = "Json")] }),
            _ => None,
        }
    }
}

/// Field type classification.
#[derive(Debug, Clone)]
pub enum FieldType {
    /// A scalar database column (String, i32, etc.)
    Scalar { scalar: ScalarType, optional: bool },
    /// A has_many relationship (Vec<Entity>)
    HasMany { target: syn::Ident },
    /// A belongs_to relationship (Entity or Option<Entity>)
    BelongsTo { target: syn::Ident, optional: bool },
}