forge_core/schema/
field.rs

1use serde::{Deserialize, Serialize};
2
3use super::types::{RustType, SqlType};
4
5/// Definition of a model field.
6#[derive(Debug, Clone, Serialize, Deserialize)]
7pub struct FieldDef {
8    /// Field name in Rust (snake_case).
9    pub name: String,
10
11    /// Column name in SQL (may differ from field name).
12    pub column_name: String,
13
14    /// Rust type.
15    pub rust_type: RustType,
16
17    /// SQL type.
18    pub sql_type: SqlType,
19
20    /// Whether the field is nullable.
21    pub nullable: bool,
22
23    /// Documentation comment.
24    pub doc: Option<String>,
25}
26
27impl FieldDef {
28    /// Create a new field definition.
29    pub fn new(name: &str, rust_type: RustType) -> Self {
30        let sql_type = rust_type.to_sql_type();
31        let nullable = rust_type.is_nullable();
32        let column_name = to_snake_case(name);
33
34        Self {
35            name: name.to_string(),
36            column_name,
37            rust_type,
38            sql_type,
39            nullable,
40            doc: None,
41        }
42    }
43
44    pub fn to_typescript(&self) -> String {
45        let (ts_type, optional) = if self.nullable {
46            let inner_type = match &self.rust_type {
47                super::types::RustType::Option(inner) => inner.to_typescript(),
48                other => other.to_typescript(),
49            };
50            (inner_type, "?")
51        } else {
52            (self.rust_type.to_typescript(), "")
53        };
54        format!("  {}{}: {};", self.name, optional, ts_type)
55    }
56}
57
58/// Convert a string to snake_case.
59fn to_snake_case(s: &str) -> String {
60    let mut result = String::new();
61    for (i, c) in s.chars().enumerate() {
62        if c.is_uppercase() {
63            if i > 0 {
64                result.push('_');
65            }
66            result.push(c.to_lowercase().next().unwrap());
67        } else {
68            result.push(c);
69        }
70    }
71    result
72}
73
74/// Convert a string to camelCase.
75#[allow(dead_code)]
76fn to_camel_case(s: &str) -> String {
77    let mut result = String::new();
78    let mut capitalize_next = false;
79    for c in s.chars() {
80        if c == '_' {
81            capitalize_next = true;
82        } else if capitalize_next {
83            result.push(c.to_uppercase().next().unwrap());
84            capitalize_next = false;
85        } else {
86            result.push(c);
87        }
88    }
89    result
90}
91
92#[cfg(test)]
93mod tests {
94    use super::*;
95
96    #[test]
97    fn test_field_def_basic() {
98        let field = FieldDef::new("email", RustType::String);
99        assert_eq!(field.name, "email");
100        assert_eq!(field.column_name, "email");
101        assert!(!field.nullable);
102    }
103
104    #[test]
105    fn test_field_def_nullable() {
106        let field = FieldDef::new("avatar_url", RustType::Option(Box::new(RustType::String)));
107        assert!(field.nullable);
108    }
109
110    #[test]
111    fn test_to_snake_case() {
112        assert_eq!(to_snake_case("createdAt"), "created_at");
113        assert_eq!(to_snake_case("userId"), "user_id");
114        assert_eq!(to_snake_case("HTTPServer"), "h_t_t_p_server");
115    }
116
117    #[test]
118    fn test_to_camel_case() {
119        assert_eq!(to_camel_case("created_at"), "createdAt");
120        assert_eq!(to_camel_case("user_id"), "userId");
121    }
122}
forge_core/schema/field.rs

forge_core/schema/
field.rs
