diff --git a/.changepacks/changepack_log_vKVGmtRxkfW3nE7rIhMcz.json b/.changepacks/changepack_log_vKVGmtRxkfW3nE7rIhMcz.json new file mode 100644 index 00000000..0eea7ab9 --- /dev/null +++ b/.changepacks/changepack_log_vKVGmtRxkfW3nE7rIhMcz.json @@ -0,0 +1 @@ +{"changes":{"crates/vespertide-exporter/Cargo.toml":"Minor","crates/vespertide-cli/Cargo.toml":"Minor"},"note":"Prisma ORM exporter 추가","date":"2026-07-09T14:25:30.6412773Z"} \ No newline at end of file diff --git a/Cargo.lock b/Cargo.lock index d102c03a..866612ce 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -93,9 +93,9 @@ dependencies = [ [[package]] name = "anyhow" -version = "1.0.102" +version = "1.0.103" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c" +checksum = "2a4385e2e34eb35d6b3efe798b9eb88096925d87726c0798709bf56d9ed84af3" [[package]] name = "ar_archive_writer" @@ -462,9 +462,9 @@ dependencies = [ [[package]] name = "crossbeam-epoch" -version = "0.9.18" +version = "0.9.20" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "5b82ac4a3c2ca9c3460964f020e1402edd5753411d7737aa39c3714ad1b5420e" +checksum = "2d6914041f254d6e9176c01941b21115dcfb7089e55135a35411081bd106ef3f" dependencies = [ "crossbeam-utils", ] @@ -2836,7 +2836,7 @@ checksum = "0b928f33d975fc6ad9f86c8f283853ad26bdd5b10b7f1542aa2fa15e2289105a" [[package]] name = "vespertide" -version = "0.1.61" +version = "0.2.1" dependencies = [ "sea-orm", "tokio", @@ -2846,7 +2846,7 @@ dependencies = [ [[package]] name = "vespertide-cli" -version = "0.1.61" +version = "0.2.1" dependencies = [ "anyhow", "assert_cmd", @@ -2878,7 +2878,7 @@ dependencies = [ [[package]] name = "vespertide-config" -version = "0.1.61" +version = "0.2.1" dependencies = [ "clap", "schemars", @@ -2888,7 +2888,7 @@ dependencies = [ [[package]] name = "vespertide-core" -version = "0.1.61" +version = "0.2.1" dependencies = [ "criterion", "proptest", @@ -2903,13 +2903,14 @@ dependencies = [ [[package]] name = "vespertide-exporter" -version = "0.1.61" +version = "0.2.1" dependencies = [ "criterion", "insta", "proptest", "rayon", "rstest", + "serde_json", "thiserror 2.0.18", "vespertide-config", "vespertide-core", @@ -2932,7 +2933,7 @@ dependencies = [ [[package]] name = "vespertide-loader" -version = "0.1.61" +version = "0.2.1" dependencies = [ "anyhow", "rayon", @@ -2948,7 +2949,7 @@ dependencies = [ [[package]] name = "vespertide-lsp" -version = "0.1.61" +version = "0.2.1" dependencies = [ "dashmap", "insta", @@ -2977,7 +2978,7 @@ dependencies = [ [[package]] name = "vespertide-macro" -version = "0.1.61" +version = "0.2.1" dependencies = [ "proc-macro-crate", "proc-macro2", @@ -2996,7 +2997,7 @@ dependencies = [ [[package]] name = "vespertide-naming" -version = "0.1.61" +version = "0.2.1" dependencies = [ "criterion", "proptest", @@ -3004,7 +3005,7 @@ dependencies = [ [[package]] name = "vespertide-planner" -version = "0.1.61" +version = "0.2.1" dependencies = [ "criterion", "insta", @@ -3019,7 +3020,7 @@ dependencies = [ [[package]] name = "vespertide-query" -version = "0.1.61" +version = "0.2.1" dependencies = [ "criterion", "insta", diff --git a/crates/vespertide-cli/src/commands/erd/tests/mod.rs b/crates/vespertide-cli/src/commands/erd/tests/mod.rs index 2d39a862..108d16f0 100644 --- a/crates/vespertide-cli/src/commands/erd/tests/mod.rs +++ b/crates/vespertide-cli/src/commands/erd/tests/mod.rs @@ -664,6 +664,24 @@ fn malformed_inline_fk_reference_is_ignored() { assert!(collect_foreign_key_relations(&[users, bad]).is_empty()); } +#[test] +fn table_level_fk_to_missing_table_is_ignored() { + let mut orphan = table( + "orphan", + vec![primary_key("id", integer()), column("ghost_id", integer())], + ); + orphan.constraints.push(TableConstraint::ForeignKey { + name: Some("fk_orphan_ghost".into()), + columns: vec!["ghost_id".into()], + ref_table: "ghost".into(), + ref_columns: vec!["id".into()], + on_delete: None, + on_update: None, + orphan_strategy: Default::default(), + }); + assert!(collect_foreign_key_relations(&[orphan]).is_empty()); +} + #[test] fn inline_unique_group_variants_drive_one_to_one_detection() { let users = normalize(&table("users", vec![primary_key("id", integer())])); diff --git a/crates/vespertide-cli/src/commands/export.rs b/crates/vespertide-cli/src/commands/export.rs deleted file mode 100644 index 7eec2f53..00000000 --- a/crates/vespertide-cli/src/commands/export.rs +++ /dev/null @@ -1,975 +0,0 @@ -use std::collections::HashMap; -use std::path::{Path, PathBuf}; - -use anyhow::{Context, Result}; -use clap::ValueEnum; -use futures::future::try_join_all; -use rayon::prelude::*; -use tokio::fs; -use vespertide_config::VespertideConfig; -use vespertide_core::TableDef; -use vespertide_exporter::{Orm, render_entity_with_schema, seaorm::SeaOrmExporterWithConfig}; - -use crate::parallel_config::{EXPORT_RENDER_PAR_MIN_LEN, EXPORT_RENDER_PAR_THRESHOLD}; -use crate::utils::load_config; - -#[derive(Copy, Clone, Debug, ValueEnum)] -pub enum OrmArg { - Seaorm, - Sqlalchemy, - Sqlmodel, - Jpa, -} - -impl From for Orm { - fn from(value: OrmArg) -> Self { - match value { - OrmArg::Seaorm => Orm::SeaOrm, - OrmArg::Sqlalchemy => Orm::SqlAlchemy, - OrmArg::Sqlmodel => Orm::SqlModel, - OrmArg::Jpa => Orm::Jpa, - } - } -} - -pub async fn cmd_export(orm: OrmArg, export_dir: Option) -> Result<()> { - let config = load_config()?; - let models = load_models_recursive(config.models_dir()) - .await - .context("load models recursively")?; - - // Normalize tables to convert inline constraints (primary_key, foreign_key, etc.) to table-level constraints - let normalized_models: Vec<(TableDef, PathBuf)> = models - .into_iter() - .map(|(table, rel_path)| { - table - .normalize() - .map_err(|e| anyhow::anyhow!("Failed to normalize table '{}': {}", table.name, e)) - .map(|normalized| (normalized, rel_path)) - }) - .collect::, _>>()?; - - let target_root = resolve_export_dir(export_dir, &config); - - // Clean the export directory before regenerating - let orm_kind: Orm = orm.into(); - clean_export_dir(&target_root, orm_kind).await?; - - if !target_root.exists() { - fs::create_dir_all(&target_root) - .await - .with_context(|| format!("create export dir {}", target_root.display()))?; - } - - // Extract all tables for schema context (used for FK chain resolution) - let all_tables: Vec = normalized_models.iter().map(|(t, _)| t.clone()).collect(); - - // Build module path mappings for SeaORM cross-directory relation resolution. - // Maps table_name -> module path segments (e.g., "admin" -> ["admin", "admin"]) - let module_paths: HashMap> = normalized_models - .iter() - .map(|(table, rel_path)| { - let segments = rel_path_to_module_segments(rel_path); - (table.name.to_string(), segments) - }) - .collect(); - - // Derive crate:: prefix from export directory (e.g., "src/models" -> "crate::models") - let crate_prefix = export_dir_to_crate_prefix(&target_root); - - // Create SeaORM exporter with config if needed - let seaorm_exporter = SeaOrmExporterWithConfig::new(config.seaorm(), config.prefix()); - let render_context = ExportRenderContext { - target_root: &target_root, - all_tables: &all_tables, - module_paths: &module_paths, - crate_prefix: &crate_prefix, - seaorm_exporter: &seaorm_exporter, - orm_kind, - }; - - // Generate all entity code (CPU-bound, done synchronously) - let entities: Vec<(String, PathBuf, String)> = - if normalized_models.len() < EXPORT_RENDER_PAR_THRESHOLD { - normalized_models - .iter() - .map(|model| render_export_entity(model, &render_context)) - .collect::>>()? - } else { - normalized_models - .par_iter() - .with_min_len(EXPORT_RENDER_PAR_MIN_LEN) - .map(|model| render_export_entity(model, &render_context)) - .collect::>>()? - }; - - // Write all files in parallel - let write_futures: Vec<_> = entities - .iter() - .map(|(name, out_path, code)| { - let name = name.clone(); - let out_path = out_path.clone(); - let code = code.clone(); - async move { - if let Some(parent) = out_path.parent() { - fs::create_dir_all(parent) - .await - .with_context(|| format!("create parent dir {}", parent.display()))?; - } - fs::write(&out_path, &code) - .await - .with_context(|| format!("write {}", out_path.display()))?; - println!("Exported {} -> {}", name, out_path.display()); - Ok::<(), anyhow::Error>(()) - } - }) - .collect(); - - try_join_all(write_futures).await?; - - // Ensure mod chain for SeaORM (must be done after all files are written) - if matches!(orm_kind, Orm::SeaOrm) { - for (_table, rel_path) in &normalized_models { - let out_path = build_output_path(&target_root, rel_path, orm_kind); - ensure_mod_chain(&target_root, rel_path) - .await - .with_context(|| format!("ensure mod chain for {}", out_path.display()))?; - } - } - - Ok(()) -} - -struct ExportRenderContext<'a> { - target_root: &'a Path, - all_tables: &'a [TableDef], - module_paths: &'a HashMap>, - crate_prefix: &'a str, - seaorm_exporter: &'a SeaOrmExporterWithConfig<'a>, - orm_kind: Orm, -} - -fn render_export_entity( - (table, rel_path): &(TableDef, PathBuf), - context: &ExportRenderContext<'_>, -) -> Result<(String, PathBuf, String)> { - let code = match context.orm_kind { - Orm::SeaOrm => context - .seaorm_exporter - .render_entity_with_schema_and_paths( - table, - context.all_tables, - context.module_paths, - context.crate_prefix, - ) - .map_err(|e| anyhow::anyhow!(e)), - _ => render_entity_with_schema(context.orm_kind, table, context.all_tables) - .map_err(|e| anyhow::anyhow!(e)), - }?; - let out_path = build_output_path(context.target_root, rel_path, context.orm_kind); - - Ok((table.name.to_string(), out_path, code)) -} - -/// Derive `crate::` prefix from the export directory path. -/// -/// For example: `src/models` → `crate::models`, `src/db/entities` → `crate::db::entities`. -/// If the path doesn't start with `src/`, returns empty string (fallback to `super::` behavior). -fn export_dir_to_crate_prefix(export_dir: &Path) -> String { - let normalized = export_dir.to_string_lossy().replace('\\', "/"); - let stripped = normalized.strip_prefix("./").unwrap_or(&normalized); - - if let Some(after_src) = stripped.strip_prefix("src/") { - let module_path = after_src.trim_end_matches('/').replace('/', "::"); - format!("crate::{module_path}") - } else { - String::new() - } -} - -/// Convert a relative model file path to Rust module path segments. -/// -/// For example: `admin/admin.json` → `["admin", "admin"]` -/// `estimate/estimate_checker.vespertide.json` → `["estimate", "estimate_checker"]` -fn rel_path_to_module_segments(rel_path: &Path) -> Vec { - let mut segments = Vec::new(); - - // Add directory components - if let Some(parent) = rel_path.parent() { - for component in parent.components() { - if let std::path::Component::Normal(name) = component - && let Some(s) = name.to_str() - { - segments.push(sanitize_filename(s).clone()); - } - } - } - - // Add file stem (strip extensions and .vespertide suffix) - if let Some(file_name) = rel_path.file_name().and_then(|n| n.to_str()) { - let (stem, _) = if let Some(dot_idx) = file_name.rfind('.') { - file_name.split_at(dot_idx) - } else { - (file_name, "") - }; - let stem = stem.strip_suffix(".vespertide").unwrap_or(stem); - segments.push(sanitize_filename(stem).clone()); - } - - segments -} - -fn resolve_export_dir(export_dir: Option, config: &VespertideConfig) -> PathBuf { - if let Some(dir) = export_dir { - return dir; - } - // Prefer explicit model_export_dir from config, fallback to default inside config. - config.model_export_dir().to_path_buf() -} - -/// Clean the export directory by removing all generated files. -/// This ensures no stale files remain from previous exports. -async fn clean_export_dir(root: &Path, orm: Orm) -> Result<()> { - if !root.exists() { - return Ok(()); - } - - let ext = match orm { - Orm::SeaOrm => "rs", - Orm::SqlAlchemy | Orm::SqlModel => "py", - Orm::Jpa => "java", - }; - - clean_dir_recursive(root, ext).await?; - Ok(()) -} - -/// Recursively remove files with the given extension and empty directories. -/// Uses async I/O for parallel file operations. -async fn clean_dir_recursive(dir: &Path, ext: &str) -> Result<()> { - if !dir.is_dir() { - return Ok(()); - } - - let mut entries = fs::read_dir(dir) - .await - .with_context(|| format!("read dir {}", dir.display()))?; - - let mut subdirs = Vec::new(); - let mut files_to_remove = Vec::new(); - - while let Some(entry) = entries.next_entry().await? { - let path = entry.path(); - if path.is_dir() { - subdirs.push(path); - } else if path.extension().and_then(|e| e.to_str()) == Some(ext) { - files_to_remove.push(path); - } - } - - // Remove files in parallel - let remove_futures: Vec<_> = files_to_remove - .into_iter() - .map(|path| async move { - fs::remove_file(&path) - .await - .with_context(|| format!("remove file {}", path.display())) - }) - .collect(); - - try_join_all(remove_futures).await?; - - // Recursively clean subdirectories - let subdir_futures: Vec<_> = subdirs - .iter() - .map(|subdir| clean_dir_recursive(subdir, ext)) - .collect(); - - try_join_all(subdir_futures).await?; - - // Remove empty directories - for subdir in subdirs { - let mut entries = fs::read_dir(&subdir).await?; - if entries.next_entry().await?.is_none() { - fs::remove_dir(&subdir) - .await - .with_context(|| format!("remove empty dir {}", subdir.display()))?; - } - } - - Ok(()) -} - -fn build_output_path(root: &Path, rel_path: &Path, orm: Orm) -> PathBuf { - // Sanitize file name: replace spaces with underscores - let mut out = root.to_path_buf(); - - // Reconstruct path with sanitized file name - for component in rel_path.components() { - if let std::path::Component::Normal(name) = component { - out.push(name); - } else { - out.push(component.as_os_str()); - } - } - - // Sanitize the file name (last component) - if let Some(file_name) = out.file_name().and_then(|n| n.to_str()) { - // Remove extension, sanitize, then add new extension - let (stem, _ext) = if let Some(dot_idx) = file_name.rfind('.') { - file_name.split_at(dot_idx) - } else { - (file_name, "") - }; - - // Strip ".vespertide" suffix if present (e.g., "user.vespertide" -> "user") - let stem = stem.strip_suffix(".vespertide").unwrap_or(stem); - - let sanitized = sanitize_filename(stem); - let ext = match orm { - Orm::SeaOrm => "rs", - Orm::SqlAlchemy | Orm::SqlModel => "py", - Orm::Jpa => "java", - }; - // Java requires filename to match PascalCase class name - let file_stem = if matches!(orm, Orm::Jpa) { - to_pascal_case(&sanitized) - } else { - sanitized - }; - out.set_file_name(format!("{file_stem}.{ext}")); - } - - out -} - -fn to_pascal_case(s: &str) -> String { - s.split('_') - .map(|word| { - let mut chars = word.chars(); - match chars.next() { - None => String::new(), - Some(first) => first.to_uppercase().chain(chars).collect(), - } - }) - .collect() -} - -fn sanitize_filename(name: &str) -> String { - name.chars() - .map(|ch| { - if ch.is_alphanumeric() || ch == '_' || ch == '-' { - ch - } else { - '_' - } - }) - .collect::() -} - -async fn load_models_recursive(base: &Path) -> Result> { - let mut out = Vec::new(); - if !base.exists() { - return Ok(out); - } - walk_models(base, base, &mut out).await?; - Ok(out) -} - -async fn ensure_mod_chain(root: &Path, rel_path: &Path) -> Result<()> { - // SeaORM exports use a standard nested Rust module tree. - let path_without_ext = rel_path.with_extension(""); - let path_stripped = if let Some(stem) = path_without_ext.file_stem().and_then(|s| s.to_str()) { - let stripped_stem = stem.strip_suffix(".vespertide").unwrap_or(stem); - if let Some(parent) = path_without_ext.parent() { - parent.join(stripped_stem) - } else { - PathBuf::from(stripped_stem) - } - } else { - path_without_ext - }; - let mut comps: Vec = path_stripped - .components() - .filter_map(|c| c.as_os_str().to_str().map(|s| sanitize_filename(s).clone())) - .collect(); - if comps.is_empty() { - return Ok(()); - } - - while let Some(child) = comps.pop() { - let dir = root.join(comps.join(std::path::MAIN_SEPARATOR_STR)); - let mod_path = dir.join("mod.rs"); - if let Some(parent) = mod_path.parent() - && !parent.exists() - { - fs::create_dir_all(parent).await?; - } - - let mut content = if mod_path.exists() { - fs::read_to_string(&mod_path).await? - } else { - String::new() - }; - - let decl = format!("pub mod {child};"); - if !content.lines().any(|line| line.trim() == decl) { - if !content.is_empty() && !content.ends_with('\n') { - content.push('\n'); - } - content.push_str(&decl); - content.push('\n'); - fs::write(mod_path, content).await?; - } - } - - Ok(()) -} - -#[async_recursion::async_recursion] -async fn walk_models( - root: &Path, - current: &Path, - acc: &mut Vec<(TableDef, PathBuf)>, -) -> Result<()> { - let mut entries = fs::read_dir(current) - .await - .with_context(|| format!("read {}", current.display()))?; - - while let Some(entry) = entries.next_entry().await? { - let path = entry.path(); - if path.is_dir() { - walk_models(root, &path, acc).await?; - continue; - } - let ext = path.extension().and_then(|s| s.to_str()); - if !matches!(ext, Some("json" | "yaml" | "yml")) { - continue; - } - let content = fs::read_to_string(&path) - .await - .with_context(|| format!("read model file: {}", path.display()))?; - let table: TableDef = if ext == Some("json") { - serde_json::from_str(&content) - .with_context(|| format!("parse JSON model: {}", path.display()))? - } else { - serde_yaml::from_str(&content) - .with_context(|| format!("parse YAML model: {}", path.display()))? - }; - let rel = path.strip_prefix(root).unwrap_or(&path).to_path_buf(); - acc.push((table, rel)); - } - Ok(()) -} - -#[cfg(test)] -mod tests { - use super::*; - use crate::test_support::CwdGuard; - use rstest::rstest; - use serial_test::serial; - use std::fs as std_fs; - use tempfile::tempdir; - use vespertide_core::{ColumnDef, ColumnType, SimpleColumnType, TableConstraint}; - - fn write_config() { - let cfg = VespertideConfig::default(); - let text = serde_json::to_string_pretty(&cfg).unwrap(); - std_fs::write("vespertide.json", text).unwrap(); - } - - fn write_model(path: &Path, table: &TableDef) { - if let Some(parent) = path.parent() { - std_fs::create_dir_all(parent).unwrap(); - } - std_fs::write(path, serde_json::to_string_pretty(table).unwrap()).unwrap(); - } - - fn sample_table(name: &str) -> TableDef { - TableDef { - name: name.into(), - description: None, - columns: vec![ColumnDef { - name: "id".into(), - r#type: ColumnType::Simple(SimpleColumnType::Integer), - nullable: false, - default: None, - comment: None, - primary_key: None, - unique: None, - index: None, - foreign_key: None, - }], - constraints: vec![TableConstraint::PrimaryKey { - auto_increment: false, - columns: vec!["id".into()], - strategy: vespertide_core::PrimaryKeyAdditionStrategy::default(), - }], - } - } - - #[tokio::test] - #[serial] - async fn export_writes_seaorm_files_to_default_dir() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - - let model = sample_table("users"); - write_model(Path::new("models/users.json"), &model); - - cmd_export(OrmArg::Seaorm, None).await.unwrap(); - - let out = PathBuf::from("src/models/users.rs"); - assert!(out.exists()); - let content = std_fs::read_to_string(out).unwrap(); - assert!(content.contains("#[sea_orm(table_name = \"users\")]")); - - // mod.rs wiring at root - let root_mod = PathBuf::from("src/models/mod.rs"); - assert!(root_mod.exists()); - let root_mod_content = std_fs::read_to_string(root_mod).unwrap(); - assert!(root_mod_content.contains("pub mod users;")); - } - - #[tokio::test] - #[serial] - async fn export_respects_custom_output_dir() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - - let model = sample_table("posts"); - write_model(Path::new("models/blog/posts.json"), &model); - - let custom = PathBuf::from("out_dir"); - cmd_export(OrmArg::Seaorm, Some(custom.clone())) - .await - .unwrap(); - - let out = custom.join("blog/posts.rs"); - assert!(out.exists()); - let content = std_fs::read_to_string(out).unwrap(); - assert!(content.contains("#[sea_orm(table_name = \"posts\")]")); - - // nested mod.rs wiring - let root_mod = custom.join("mod.rs"); - let blog_mod = custom.join("blog/mod.rs"); - assert!(root_mod.exists()); - assert!(blog_mod.exists()); - let root_mod_content = std_fs::read_to_string(root_mod).unwrap(); - let blog_mod_content = std_fs::read_to_string(blog_mod).unwrap(); - assert!(root_mod_content.contains("pub mod blog;")); - assert!(blog_mod_content.contains("pub mod posts;")); - } - - #[tokio::test] - #[serial] - async fn export_with_sqlalchemy_sets_py_extension() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - - let model = sample_table("items"); - write_model(Path::new("models/items.json"), &model); - - cmd_export(OrmArg::Sqlalchemy, None).await.unwrap(); - - let out = PathBuf::from("src/models/items.py"); - assert!(out.exists()); - let content = std_fs::read_to_string(out).unwrap(); - assert!(content.contains("items")); - } - - #[tokio::test] - #[serial] - async fn export_with_sqlmodel_sets_py_extension() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - - let model = sample_table("orders"); - write_model(Path::new("models/orders.json"), &model); - - cmd_export(OrmArg::Sqlmodel, None).await.unwrap(); - - let out = PathBuf::from("src/models/orders.py"); - assert!(out.exists()); - let content = std_fs::read_to_string(out).unwrap(); - assert!(content.contains("orders")); - } - - #[tokio::test] - #[serial] - async fn load_models_recursive_returns_empty_when_absent() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - let models = load_models_recursive(Path::new("no_models")).await.unwrap(); - assert!(models.is_empty()); - } - - #[tokio::test] - #[serial] - async fn load_models_recursive_ignores_non_model_files() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - - std_fs::create_dir_all("models").unwrap(); - std_fs::write("models/ignore.txt", "hello").unwrap(); - write_model(Path::new("models/valid.json"), &sample_table("valid")); - - let models = load_models_recursive(Path::new("models")).await.unwrap(); - assert_eq!(models.len(), 1); - assert_eq!(models[0].0.name, "valid"); - } - - #[tokio::test] - #[serial] - async fn load_models_recursive_parses_yaml_branch() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - - std_fs::create_dir_all("models").unwrap(); - let table = sample_table("yaml_table"); - let yaml = serde_yaml::to_string(&table).unwrap(); - std_fs::write("models/yaml_table.yaml", yaml).unwrap(); - - let models = load_models_recursive(Path::new("models")).await.unwrap(); - assert_eq!(models.len(), 1); - assert_eq!(models[0].0.name, "yaml_table"); - } - - #[tokio::test] - #[serial] - async fn ensure_mod_chain_adds_to_existing_file_without_trailing_newline() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("src/models"); - std_fs::create_dir_all(&root).unwrap(); - std_fs::write(root.join("mod.rs"), "pub mod existing;").unwrap(); - - ensure_mod_chain(&root, Path::new("blog/posts.rs")) - .await - .unwrap(); - - let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); - let blog_mod = std_fs::read_to_string(root.join("blog/mod.rs")).unwrap(); - assert!(root_mod.contains("pub mod existing;")); - assert!(root_mod.contains("pub mod blog;")); - assert!(blog_mod.contains("pub mod posts;")); - // ensure newline appended if missing - assert!(root_mod.ends_with('\n')); - } - - #[tokio::test] - async fn ensure_mod_chain_no_components_is_noop() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("src/models"); - std_fs::create_dir_all(&root).unwrap(); - // empty path should not error - assert!(ensure_mod_chain(&root, Path::new("")).await.is_ok()); - } - - #[test] - #[serial] - fn resolve_export_dir_prefers_override() { - let tmp = tempdir().unwrap(); - let _guard = CwdGuard::new(&tmp.path().to_path_buf()); - write_config(); - let cfg = VespertideConfig::default(); - let override_dir = PathBuf::from("custom_out"); - let resolved = super::resolve_export_dir(Some(override_dir.clone()), &cfg); - assert_eq!(resolved, override_dir); - } - - #[rstest] - #[case(OrmArg::Seaorm, Orm::SeaOrm)] - #[case(OrmArg::Sqlalchemy, Orm::SqlAlchemy)] - #[case(OrmArg::Sqlmodel, Orm::SqlModel)] - #[case(OrmArg::Jpa, Orm::Jpa)] - fn orm_arg_maps_to_enum(#[case] arg: OrmArg, #[case] expected: Orm) { - assert_eq!(Orm::from(arg), expected); - } - - #[rstest] - #[case("normal_name", "normal_name")] - #[case("user copy", "user_copy")] - #[case("user copy", "user__copy")] - #[case("user-copy", "user-copy")] - #[case("user.copy", "user_copy")] - #[case("user copy.json", "user_copy_json")] - fn test_sanitize_filename(#[case] input: &str, #[case] expected: &str) { - assert_eq!(sanitize_filename(input), expected); - } - - #[test] - fn build_output_path_sanitizes_spaces() { - use std::path::Path; - let root = Path::new("src/models"); - let rel_path = Path::new("user copy.json"); - let out = build_output_path(root, rel_path, Orm::SeaOrm); - assert_eq!(out, Path::new("src/models/user_copy.rs")); - - let rel_path2 = Path::new("blog/post name.yaml"); - let out2 = build_output_path(root, rel_path2, Orm::SeaOrm); - assert_eq!(out2, Path::new("src/models/blog/post_name.rs")); - } - - #[test] - fn build_output_path_handles_file_without_extension() { - use std::path::Path; - let root = Path::new("src/models"); - // File without extension - covers line 88 (else branch) - let rel_path = Path::new("users"); - let out = build_output_path(root, rel_path, Orm::SeaOrm); - assert_eq!(out, Path::new("src/models/users.rs")); - - let out_py = build_output_path(root, rel_path, Orm::SqlAlchemy); - assert_eq!(out_py, Path::new("src/models/users.py")); - } - - #[test] - fn build_output_path_handles_special_path_components() { - use std::path::Path; - let root = Path::new("src/models"); - // Path with CurDir component (.) - covers line 78 (non-Normal component branch) - let rel_path = Path::new("./blog/posts.json"); - let out = build_output_path(root, rel_path, Orm::SeaOrm); - // The . component gets pushed via the else branch - assert!(out.to_string_lossy().contains("posts")); - - // Path with ParentDir component (..) - let rel_path2 = Path::new("../other/items.yaml"); - let out2 = build_output_path(root, rel_path2, Orm::SeaOrm); - assert!(out2.to_string_lossy().contains("items")); - } - - #[test] - fn build_output_path_strips_vespertide_suffix() { - use std::path::Path; - let root = Path::new("src/models"); - - // .vespertide.json -> .rs (strips ".vespertide" from stem) - let rel_path = Path::new("user.vespertide.json"); - let out = build_output_path(root, rel_path, Orm::SeaOrm); - assert_eq!(out, Path::new("src/models/user.rs")); - - // Nested path with .vespertide.json - let rel_path2 = Path::new("blog/post.vespertide.json"); - let out2 = build_output_path(root, rel_path2, Orm::SeaOrm); - assert_eq!(out2, Path::new("src/models/blog/post.rs")); - - // .vespertide.yaml -> .py - let rel_path3 = Path::new("order.vespertide.yaml"); - let out3 = build_output_path(root, rel_path3, Orm::SqlAlchemy); - assert_eq!(out3, Path::new("src/models/order.py")); - - // Regular .json without .vespertide suffix still works - let rel_path4 = Path::new("item.json"); - let out4 = build_output_path(root, rel_path4, Orm::SeaOrm); - assert_eq!(out4, Path::new("src/models/item.rs")); - } - - #[tokio::test] - #[serial] - async fn ensure_mod_chain_strips_vespertide_suffix() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("src/models"); - std_fs::create_dir_all(&root).unwrap(); - - // File with .vespertide suffix should produce mod declaration without it - ensure_mod_chain(&root, Path::new("user.vespertide.json")) - .await - .unwrap(); - - let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); - // Should be "pub mod user;" not "pub mod user_vespertide;" - assert!(root_mod.contains("pub mod user;")); - assert!(!root_mod.contains("user_vespertide")); - - // Nested path with .vespertide suffix - ensure_mod_chain(&root, Path::new("blog/post.vespertide.json")) - .await - .unwrap(); - let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); - let blog_mod = std_fs::read_to_string(root.join("blog/mod.rs")).unwrap(); - assert!(root_mod.contains("pub mod blog;")); - assert!(blog_mod.contains("pub mod post;")); - assert!(!root_mod.contains("post_vespertide")); - assert!(!blog_mod.contains("post_vespertide")); - } - - #[tokio::test] - async fn clean_export_dir_removes_rs_files_for_seaorm() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("export_dir"); - std_fs::create_dir_all(&root).unwrap(); - - // Create some .rs files that should be cleaned - std_fs::write(root.join("old_model.rs"), "// old rust file").unwrap(); - std_fs::write(root.join("another.rs"), "// another rust file").unwrap(); - // Create a non-.rs file that should NOT be cleaned - std_fs::write(root.join("keep.txt"), "keep this").unwrap(); - - clean_export_dir(&root, Orm::SeaOrm).await.unwrap(); - - // .rs files should be gone - assert!(!root.join("old_model.rs").exists()); - assert!(!root.join("another.rs").exists()); - // .txt file should remain - assert!(root.join("keep.txt").exists()); - } - - #[tokio::test] - async fn clean_export_dir_removes_py_files_for_sqlalchemy() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("export_dir"); - std_fs::create_dir_all(&root).unwrap(); - - // Create some .py files that should be cleaned - std_fs::write(root.join("old_model.py"), "# old python file").unwrap(); - // Create a .rs file that should NOT be cleaned - std_fs::write(root.join("keep.rs"), "// keep this").unwrap(); - - clean_export_dir(&root, Orm::SqlAlchemy).await.unwrap(); - - // .py files should be gone - assert!(!root.join("old_model.py").exists()); - // .rs file should remain - assert!(root.join("keep.rs").exists()); - } - - #[tokio::test] - async fn clean_export_dir_removes_py_files_for_sqlmodel() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("export_dir"); - std_fs::create_dir_all(&root).unwrap(); - - std_fs::write(root.join("model.py"), "# python file").unwrap(); - - clean_export_dir(&root, Orm::SqlModel).await.unwrap(); - - assert!(!root.join("model.py").exists()); - } - - #[tokio::test] - async fn clean_export_dir_removes_java_files_for_jpa() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("export_dir"); - std_fs::create_dir_all(&root).unwrap(); - - std_fs::write(root.join("User.java"), "// java entity").unwrap(); - std_fs::write(root.join("Order.java"), "// java entity").unwrap(); - std_fs::write(root.join("keep.rs"), "// keep this").unwrap(); - - clean_export_dir(&root, Orm::Jpa).await.unwrap(); - - assert!(!root.join("User.java").exists()); - assert!(!root.join("Order.java").exists()); - assert!(root.join("keep.rs").exists()); - } - - #[tokio::test] - async fn clean_export_dir_handles_missing_directory() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("nonexistent_dir"); - - // Should not error on missing directory - let result = clean_export_dir(&root, Orm::SeaOrm).await; - assert!(result.is_ok()); - } - - #[tokio::test] - async fn clean_dir_recursive_cleans_subdirectories() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("export_dir"); - let subdir = root.join("nested"); - std_fs::create_dir_all(&subdir).unwrap(); - - // Create files in root and subdirectory - std_fs::write(root.join("root.rs"), "// root").unwrap(); - std_fs::write(subdir.join("nested.rs"), "// nested").unwrap(); - std_fs::write(subdir.join("keep.txt"), "keep").unwrap(); - - clean_dir_recursive(&root, "rs").await.unwrap(); - - // .rs files should be gone - assert!(!root.join("root.rs").exists()); - assert!(!subdir.join("nested.rs").exists()); - // .txt file should remain - assert!(subdir.join("keep.txt").exists()); - // subdir should still exist (has .txt file) - assert!(subdir.exists()); - } - - #[tokio::test] - async fn clean_dir_recursive_removes_empty_subdirectories() { - let tmp = tempdir().unwrap(); - let root = tmp.path().join("export_dir"); - let subdir = root.join("empty_after_clean"); - std_fs::create_dir_all(&subdir).unwrap(); - - // Create only .rs files in subdirectory - std_fs::write(subdir.join("only.rs"), "// only").unwrap(); - - clean_dir_recursive(&root, "rs").await.unwrap(); - - // .rs file should be gone - assert!(!subdir.join("only.rs").exists()); - // Empty subdirectory should be removed - assert!(!subdir.exists()); - } - - #[tokio::test] - async fn clean_dir_recursive_handles_non_directory() { - let tmp = tempdir().unwrap(); - let file_path = tmp.path().join("not_a_dir.txt"); - std_fs::write(&file_path, "content").unwrap(); - - // Should not error when called on a file instead of directory - let result = clean_dir_recursive(&file_path, "rs").await; - assert!(result.is_ok()); - } - - #[test] - fn build_output_path_jpa_uses_pascal_case_java_extension() { - use std::path::Path; - let root = Path::new("src/models"); - - // snake_case model → PascalCase .java - let rel_path = Path::new("order_item.json"); - let out = build_output_path(root, rel_path, Orm::Jpa); - assert_eq!(out, Path::new("src/models/OrderItem.java")); - - // Single word - let rel_path2 = Path::new("users.json"); - let out2 = build_output_path(root, rel_path2, Orm::Jpa); - assert_eq!(out2, Path::new("src/models/Users.java")); - - // Nested path - let rel_path3 = Path::new("blog/post_comment.yaml"); - let out3 = build_output_path(root, rel_path3, Orm::Jpa); - assert_eq!(out3, Path::new("src/models/blog/PostComment.java")); - } - - #[test] - fn build_output_path_jpa_strips_vespertide_suffix() { - use std::path::Path; - let root = Path::new("src/models"); - - let rel_path = Path::new("user.vespertide.json"); - let out = build_output_path(root, rel_path, Orm::Jpa); - assert_eq!(out, Path::new("src/models/User.java")); - } - - #[rstest] - #[case("order_item", "OrderItem")] - #[case("users", "Users")] - #[case("a", "A")] - #[case("user_profile_image", "UserProfileImage")] - #[case("a__b", "AB")] - fn test_to_pascal_case(#[case] input: &str, #[case] expected: &str) { - assert_eq!(to_pascal_case(input), expected); - } -} diff --git a/crates/vespertide-cli/src/commands/export/mod.rs b/crates/vespertide-cli/src/commands/export/mod.rs new file mode 100644 index 00000000..05fec433 --- /dev/null +++ b/crates/vespertide-cli/src/commands/export/mod.rs @@ -0,0 +1,506 @@ +use std::collections::HashMap; +use std::path::{Path, PathBuf}; + +use anyhow::{Context, Result}; +use clap::ValueEnum; +use futures::future::try_join_all; +use rayon::prelude::*; +use tokio::fs; +use vespertide_config::VespertideConfig; +use vespertide_core::TableDef; +use vespertide_exporter::{ + Orm, prisma, render_entity_with_schema, seaorm::SeaOrmExporterWithConfig, +}; + +use crate::parallel_config::{EXPORT_RENDER_PAR_MIN_LEN, EXPORT_RENDER_PAR_THRESHOLD}; +use crate::utils::load_config; + +#[derive(Copy, Clone, Debug, ValueEnum)] +pub enum OrmArg { + Seaorm, + Sqlalchemy, + Sqlmodel, + Jpa, + Prisma, +} + +impl From for Orm { + fn from(value: OrmArg) -> Self { + match value { + OrmArg::Seaorm => Orm::SeaOrm, + OrmArg::Sqlalchemy => Orm::SqlAlchemy, + OrmArg::Sqlmodel => Orm::SqlModel, + OrmArg::Jpa => Orm::Jpa, + OrmArg::Prisma => Orm::Prisma, + } + } +} + +pub async fn cmd_export(orm: OrmArg, export_dir: Option) -> Result<()> { + let config = load_config()?; + let models = load_models_recursive(config.models_dir()) + .await + .context("load models recursively")?; + + // Normalize tables to convert inline constraints (primary_key, foreign_key, etc.) to table-level constraints + let normalized_models: Vec<(TableDef, PathBuf)> = models + .into_iter() + .map(|(table, rel_path)| { + table + .normalize() + .map_err(|e| anyhow::anyhow!("Failed to normalize table '{}': {}", table.name, e)) + .map(|normalized| (normalized, rel_path)) + }) + .collect::, _>>()?; + + let target_root = resolve_export_dir(export_dir, &config); + + // Prisma uses a single-file output strategy + if matches!(orm, OrmArg::Prisma) { + return cmd_export_prisma(normalized_models, target_root).await; + } + + // Clean the export directory before regenerating + let orm_kind: Orm = orm.into(); + clean_export_dir(&target_root, orm_kind).await?; + + if !target_root.exists() { + fs::create_dir_all(&target_root) + .await + .with_context(|| format!("create export dir {}", target_root.display()))?; + } + + // Extract all tables for schema context (used for FK chain resolution) + let all_tables: Vec = normalized_models.iter().map(|(t, _)| t.clone()).collect(); + + // Build module path mappings for SeaORM cross-directory relation resolution. + // Maps table_name -> module path segments (e.g., "admin" -> ["admin", "admin"]) + let module_paths: HashMap> = normalized_models + .iter() + .map(|(table, rel_path)| { + let segments = rel_path_to_module_segments(rel_path); + (table.name.to_string(), segments) + }) + .collect(); + + // Derive crate:: prefix from export directory (e.g., "src/models" -> "crate::models") + let crate_prefix = export_dir_to_crate_prefix(&target_root); + + // Create SeaORM exporter with config if needed + let seaorm_exporter = SeaOrmExporterWithConfig::new(config.seaorm(), config.prefix()); + let render_context = ExportRenderContext { + target_root: &target_root, + all_tables: &all_tables, + module_paths: &module_paths, + crate_prefix: &crate_prefix, + seaorm_exporter: &seaorm_exporter, + orm_kind, + }; + + // Generate all entity code (CPU-bound, done synchronously) + let entities: Vec<(String, PathBuf, String)> = + if normalized_models.len() < EXPORT_RENDER_PAR_THRESHOLD { + normalized_models + .iter() + .map(|model| render_export_entity(model, &render_context)) + .collect::>>()? + } else { + normalized_models + .par_iter() + .with_min_len(EXPORT_RENDER_PAR_MIN_LEN) + .map(|model| render_export_entity(model, &render_context)) + .collect::>>()? + }; + + // Write all files in parallel + let write_futures: Vec<_> = entities + .iter() + .map(|(name, out_path, code)| { + let name = name.clone(); + let out_path = out_path.clone(); + let code = code.clone(); + async move { + if let Some(parent) = out_path.parent() { + fs::create_dir_all(parent) + .await + .with_context(|| format!("create parent dir {}", parent.display()))?; + } + fs::write(&out_path, &code) + .await + .with_context(|| format!("write {}", out_path.display()))?; + println!("Exported {} -> {}", name, out_path.display()); + Ok::<(), anyhow::Error>(()) + } + }) + .collect(); + + try_join_all(write_futures).await?; + + // Ensure mod chain for SeaORM (must be done after all files are written) + if matches!(orm_kind, Orm::SeaOrm) { + for (_table, rel_path) in &normalized_models { + let out_path = build_output_path(&target_root, rel_path, orm_kind); + ensure_mod_chain(&target_root, rel_path) + .await + .with_context(|| format!("ensure mod chain for {}", out_path.display()))?; + } + } + + Ok(()) +} + +struct ExportRenderContext<'a> { + target_root: &'a Path, + all_tables: &'a [TableDef], + module_paths: &'a HashMap>, + crate_prefix: &'a str, + seaorm_exporter: &'a SeaOrmExporterWithConfig<'a>, + orm_kind: Orm, +} + +fn render_export_entity( + (table, rel_path): &(TableDef, PathBuf), + context: &ExportRenderContext<'_>, +) -> Result<(String, PathBuf, String)> { + let code = match context.orm_kind { + Orm::SeaOrm => context + .seaorm_exporter + .render_entity_with_schema_and_paths( + table, + context.all_tables, + context.module_paths, + context.crate_prefix, + ) + .map_err(|e| anyhow::anyhow!(e)), + _ => render_entity_with_schema(context.orm_kind, table, context.all_tables) + .map_err(|e| anyhow::anyhow!(e)), + }?; + let out_path = build_output_path(context.target_root, rel_path, context.orm_kind); + + Ok((table.name.to_string(), out_path, code)) +} + +/// Derive `crate::` prefix from the export directory path. +/// +/// For example: `src/models` → `crate::models`, `src/db/entities` → `crate::db::entities`. +/// If the path doesn't start with `src/`, returns empty string (fallback to `super::` behavior). +fn export_dir_to_crate_prefix(export_dir: &Path) -> String { + let normalized = export_dir.to_string_lossy().replace('\\', "/"); + let stripped = normalized.strip_prefix("./").unwrap_or(&normalized); + + if let Some(after_src) = stripped.strip_prefix("src/") { + let module_path = after_src.trim_end_matches('/').replace('/', "::"); + format!("crate::{module_path}") + } else { + String::new() + } +} + +/// Convert a relative model file path to Rust module path segments. +/// +/// For example: `admin/admin.json` → `["admin", "admin"]` +/// `estimate/estimate_checker.vespertide.json` → `["estimate", "estimate_checker"]` +fn rel_path_to_module_segments(rel_path: &Path) -> Vec { + let mut segments = Vec::new(); + + // Add directory components + if let Some(parent) = rel_path.parent() { + for component in parent.components() { + if let std::path::Component::Normal(name) = component + && let Some(s) = name.to_str() + { + segments.push(sanitize_filename(s).clone()); + } + } + } + + // Add file stem (strip extensions and .vespertide suffix) + if let Some(file_name) = rel_path.file_name().and_then(|n| n.to_str()) { + let (stem, _) = if let Some(dot_idx) = file_name.rfind('.') { + file_name.split_at(dot_idx) + } else { + (file_name, "") + }; + let stem = stem.strip_suffix(".vespertide").unwrap_or(stem); + segments.push(sanitize_filename(stem).clone()); + } + + segments +} + +fn resolve_export_dir(export_dir: Option, config: &VespertideConfig) -> PathBuf { + if let Some(dir) = export_dir { + return dir; + } + // Prefer explicit model_export_dir from config, fallback to default inside config. + config.model_export_dir().to_path_buf() +} + +/// Clean the export directory by removing all generated files. +/// This ensures no stale files remain from previous exports. +async fn clean_export_dir(root: &Path, orm: Orm) -> Result<()> { + if !root.exists() { + return Ok(()); + } + + let ext = match orm { + Orm::SeaOrm => "rs", + Orm::SqlAlchemy | Orm::SqlModel => "py", + Orm::Jpa => "java", + Orm::Prisma => "prisma", + }; + + clean_dir_recursive(root, ext).await?; + Ok(()) +} + +/// Recursively remove files with the given extension and empty directories. +/// Uses async I/O for parallel file operations. +async fn clean_dir_recursive(dir: &Path, ext: &str) -> Result<()> { + if !dir.is_dir() { + return Ok(()); + } + + let mut entries = fs::read_dir(dir) + .await + .with_context(|| format!("read dir {}", dir.display()))?; + + let mut subdirs = Vec::new(); + let mut files_to_remove = Vec::new(); + + while let Some(entry) = entries.next_entry().await? { + let path = entry.path(); + if path.is_dir() { + subdirs.push(path); + } else if path.extension().and_then(|e| e.to_str()) == Some(ext) { + files_to_remove.push(path); + } + } + + // Remove files in parallel + let remove_futures: Vec<_> = files_to_remove + .into_iter() + .map(|path| async move { + fs::remove_file(&path) + .await + .with_context(|| format!("remove file {}", path.display())) + }) + .collect(); + + try_join_all(remove_futures).await?; + + // Recursively clean subdirectories + let subdir_futures: Vec<_> = subdirs + .iter() + .map(|subdir| clean_dir_recursive(subdir, ext)) + .collect(); + + try_join_all(subdir_futures).await?; + + // Remove empty directories + for subdir in subdirs { + let mut entries = fs::read_dir(&subdir).await?; + if entries.next_entry().await?.is_none() { + fs::remove_dir(&subdir) + .await + .with_context(|| format!("remove empty dir {}", subdir.display()))?; + } + } + + Ok(()) +} + +fn build_output_path(root: &Path, rel_path: &Path, orm: Orm) -> PathBuf { + // Sanitize file name: replace spaces with underscores + let mut out = root.to_path_buf(); + + // Reconstruct path with sanitized file name + for component in rel_path.components() { + if let std::path::Component::Normal(name) = component { + out.push(name); + } else { + out.push(component.as_os_str()); + } + } + + // Sanitize the file name (last component) + if let Some(file_name) = out.file_name().and_then(|n| n.to_str()) { + // Remove extension, sanitize, then add new extension + let (stem, _ext) = if let Some(dot_idx) = file_name.rfind('.') { + file_name.split_at(dot_idx) + } else { + (file_name, "") + }; + + // Strip ".vespertide" suffix if present (e.g., "user.vespertide" -> "user") + let stem = stem.strip_suffix(".vespertide").unwrap_or(stem); + + let sanitized = sanitize_filename(stem); + let ext = match orm { + Orm::SeaOrm => "rs", + Orm::SqlAlchemy | Orm::SqlModel => "py", + Orm::Jpa => "java", + Orm::Prisma => "prisma", + }; + // Java requires filename to match PascalCase class name + let file_stem = if matches!(orm, Orm::Jpa) { + to_pascal_case(&sanitized) + } else { + sanitized + }; + out.set_file_name(format!("{file_stem}.{ext}")); + } + + out +} + +fn to_pascal_case(s: &str) -> String { + s.split('_') + .map(|word| { + let mut chars = word.chars(); + match chars.next() { + None => String::new(), + Some(first) => first.to_uppercase().chain(chars).collect(), + } + }) + .collect() +} + +fn sanitize_filename(name: &str) -> String { + name.chars() + .map(|ch| { + if ch.is_alphanumeric() || ch == '_' || ch == '-' { + ch + } else { + '_' + } + }) + .collect::() +} + +async fn load_models_recursive(base: &Path) -> Result> { + let mut out = Vec::new(); + if !base.exists() { + return Ok(out); + } + walk_models(base, base, &mut out).await?; + Ok(out) +} + +async fn ensure_mod_chain(root: &Path, rel_path: &Path) -> Result<()> { + // SeaORM exports use a standard nested Rust module tree. + let path_without_ext = rel_path.with_extension(""); + let path_stripped = if let Some(stem) = path_without_ext.file_stem().and_then(|s| s.to_str()) { + let stripped_stem = stem.strip_suffix(".vespertide").unwrap_or(stem); + if let Some(parent) = path_without_ext.parent() { + parent.join(stripped_stem) + } else { + PathBuf::from(stripped_stem) + } + } else { + path_without_ext + }; + let mut comps: Vec = path_stripped + .components() + .filter_map(|c| c.as_os_str().to_str().map(|s| sanitize_filename(s).clone())) + .collect(); + if comps.is_empty() { + return Ok(()); + } + + while let Some(child) = comps.pop() { + let dir = root.join(comps.join(std::path::MAIN_SEPARATOR_STR)); + let mod_path = dir.join("mod.rs"); + if let Some(parent) = mod_path.parent() + && !parent.exists() + { + fs::create_dir_all(parent).await?; + } + + let mut content = if mod_path.exists() { + fs::read_to_string(&mod_path).await? + } else { + String::new() + }; + + let decl = format!("pub mod {child};"); + if !content.lines().any(|line| line.trim() == decl) { + if !content.is_empty() && !content.ends_with('\n') { + content.push('\n'); + } + content.push_str(&decl); + content.push('\n'); + fs::write(mod_path, content).await?; + } + } + + Ok(()) +} + +async fn cmd_export_prisma( + normalized_models: Vec<(TableDef, PathBuf)>, + target_root: PathBuf, +) -> Result<()> { + let all_tables: Vec = normalized_models.iter().map(|(t, _)| t.clone()).collect(); + let content = prisma::render_schema(&all_tables); + + clean_export_dir(&target_root, Orm::Prisma).await?; + + if !target_root.exists() { + fs::create_dir_all(&target_root) + .await + .with_context(|| format!("create export dir {}", target_root.display()))?; + } + + let out_path = target_root.join("schema.prisma"); + fs::write(&out_path, &content) + .await + .with_context(|| format!("write {}", out_path.display()))?; + + println!( + "Exported {} model(s) -> {}", + normalized_models.len(), + out_path.display() + ); + + Ok(()) +} + +#[async_recursion::async_recursion] +async fn walk_models( + root: &Path, + current: &Path, + acc: &mut Vec<(TableDef, PathBuf)>, +) -> Result<()> { + let mut entries = fs::read_dir(current) + .await + .with_context(|| format!("read {}", current.display()))?; + + while let Some(entry) = entries.next_entry().await? { + let path = entry.path(); + if path.is_dir() { + walk_models(root, &path, acc).await?; + continue; + } + let ext = path.extension().and_then(|s| s.to_str()); + if !matches!(ext, Some("json" | "yaml" | "yml")) { + continue; + } + let content = fs::read_to_string(&path) + .await + .with_context(|| format!("read model file: {}", path.display()))?; + let table: TableDef = if ext == Some("json") { + serde_json::from_str(&content) + .with_context(|| format!("parse JSON model: {}", path.display()))? + } else { + serde_yaml::from_str(&content) + .with_context(|| format!("parse YAML model: {}", path.display()))? + }; + let rel = path.strip_prefix(root).unwrap_or(&path).to_path_buf(); + acc.push((table, rel)); + } + Ok(()) +} + +#[cfg(test)] +mod tests; diff --git a/crates/vespertide-cli/src/commands/export/tests/mod.rs b/crates/vespertide-cli/src/commands/export/tests/mod.rs new file mode 100644 index 00000000..3462f530 --- /dev/null +++ b/crates/vespertide-cli/src/commands/export/tests/mod.rs @@ -0,0 +1,576 @@ +use super::*; +pub(super) use crate::test_support::CwdGuard; +pub(super) use rstest::rstest; +pub(super) use serial_test::serial; +pub(super) use std::fs as std_fs; +pub(super) use tempfile::tempdir; +pub(super) use vespertide_core::{ColumnDef, ColumnType, SimpleColumnType, TableConstraint}; + +mod prisma; + +fn write_config() { + let cfg = VespertideConfig::default(); + let text = serde_json::to_string_pretty(&cfg).unwrap(); + std_fs::write("vespertide.json", text).unwrap(); +} + +fn write_model(path: &Path, table: &TableDef) { + if let Some(parent) = path.parent() { + std_fs::create_dir_all(parent).unwrap(); + } + std_fs::write(path, serde_json::to_string_pretty(table).unwrap()).unwrap(); +} + +fn sample_table(name: &str) -> TableDef { + TableDef { + name: name.into(), + description: None, + columns: vec![ColumnDef { + name: "id".into(), + r#type: ColumnType::Simple(SimpleColumnType::Integer), + nullable: false, + default: None, + comment: None, + primary_key: None, + unique: None, + index: None, + foreign_key: None, + }], + constraints: vec![TableConstraint::PrimaryKey { + auto_increment: false, + columns: vec!["id".into()], + strategy: vespertide_core::PrimaryKeyAdditionStrategy::default(), + }], + } +} + +#[tokio::test] +#[serial] +async fn export_writes_seaorm_files_to_default_dir() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + let model = sample_table("users"); + write_model(Path::new("models/users.json"), &model); + + cmd_export(OrmArg::Seaorm, None).await.unwrap(); + + let out = PathBuf::from("src/models/users.rs"); + assert!(out.exists()); + let content = std_fs::read_to_string(out).unwrap(); + assert!(content.contains("#[sea_orm(table_name = \"users\")]")); + + // mod.rs wiring at root + let root_mod = PathBuf::from("src/models/mod.rs"); + assert!(root_mod.exists()); + let root_mod_content = std_fs::read_to_string(root_mod).unwrap(); + assert!(root_mod_content.contains("pub mod users;")); +} + +#[tokio::test] +#[serial] +async fn export_respects_custom_output_dir() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + let model = sample_table("posts"); + write_model(Path::new("models/blog/posts.json"), &model); + + let custom = PathBuf::from("out_dir"); + cmd_export(OrmArg::Seaorm, Some(custom.clone())) + .await + .unwrap(); + + let out = custom.join("blog/posts.rs"); + assert!(out.exists()); + let content = std_fs::read_to_string(out).unwrap(); + assert!(content.contains("#[sea_orm(table_name = \"posts\")]")); + + // nested mod.rs wiring + let root_mod = custom.join("mod.rs"); + let blog_mod = custom.join("blog/mod.rs"); + assert!(root_mod.exists()); + assert!(blog_mod.exists()); + let root_mod_content = std_fs::read_to_string(root_mod).unwrap(); + let blog_mod_content = std_fs::read_to_string(blog_mod).unwrap(); + assert!(root_mod_content.contains("pub mod blog;")); + assert!(blog_mod_content.contains("pub mod posts;")); +} + +#[tokio::test] +#[serial] +async fn export_with_sqlalchemy_sets_py_extension() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + let model = sample_table("items"); + write_model(Path::new("models/items.json"), &model); + + cmd_export(OrmArg::Sqlalchemy, None).await.unwrap(); + + let out = PathBuf::from("src/models/items.py"); + assert!(out.exists()); + let content = std_fs::read_to_string(out).unwrap(); + assert!(content.contains("items")); +} + +#[tokio::test] +#[serial] +async fn export_with_sqlmodel_sets_py_extension() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + let model = sample_table("orders"); + write_model(Path::new("models/orders.json"), &model); + + cmd_export(OrmArg::Sqlmodel, None).await.unwrap(); + + let out = PathBuf::from("src/models/orders.py"); + assert!(out.exists()); + let content = std_fs::read_to_string(out).unwrap(); + assert!(content.contains("orders")); +} + +#[tokio::test] +#[serial] +async fn export_seaorm_uses_absolute_crate_paths_for_cross_directory_fk() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + let user = sample_table("user"); + write_model(Path::new("models/admin/user.json"), &user); + + let mut post = sample_table("post"); + post.columns.push(ColumnDef { + name: "user_id".into(), + r#type: ColumnType::Simple(SimpleColumnType::Integer), + nullable: false, + default: None, + comment: None, + primary_key: None, + unique: None, + index: None, + foreign_key: None, + }); + post.constraints.push(TableConstraint::ForeignKey { + name: Some("fk_post__user_id".into()), + columns: vec!["user_id".into()], + ref_table: "user".into(), + ref_columns: vec!["id".into()], + on_delete: None, + on_update: None, + orphan_strategy: Default::default(), + }); + write_model(Path::new("models/blog/post.json"), &post); + + cmd_export(OrmArg::Seaorm, None).await.unwrap(); + + // Cross-directory FK relations must use the absolute `crate::` module path + // derived from the export dir, not a sibling `super::` path. + let content = std_fs::read_to_string("src/models/blog/post.rs").unwrap(); + assert!(content.contains("crate::models::admin::user")); +} + +#[tokio::test] +#[serial] +async fn export_creates_output_dir_even_without_models() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + cmd_export(OrmArg::Seaorm, None).await.unwrap(); + + assert!(PathBuf::from("src/models").exists()); +} + +#[tokio::test] +#[serial] +async fn load_models_recursive_returns_empty_when_absent() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + let models = load_models_recursive(Path::new("no_models")).await.unwrap(); + assert!(models.is_empty()); +} + +#[tokio::test] +#[serial] +async fn load_models_recursive_ignores_non_model_files() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + std_fs::create_dir_all("models").unwrap(); + std_fs::write("models/ignore.txt", "hello").unwrap(); + write_model(Path::new("models/valid.json"), &sample_table("valid")); + + let models = load_models_recursive(Path::new("models")).await.unwrap(); + assert_eq!(models.len(), 1); + assert_eq!(models[0].0.name, "valid"); +} + +#[tokio::test] +#[serial] +async fn load_models_recursive_parses_yaml_branch() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + std_fs::create_dir_all("models").unwrap(); + let table = sample_table("yaml_table"); + let yaml = serde_yaml::to_string(&table).unwrap(); + std_fs::write("models/yaml_table.yaml", yaml).unwrap(); + + let models = load_models_recursive(Path::new("models")).await.unwrap(); + assert_eq!(models.len(), 1); + assert_eq!(models[0].0.name, "yaml_table"); +} + +#[tokio::test] +#[serial] +async fn ensure_mod_chain_adds_to_existing_file_without_trailing_newline() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("src/models"); + std_fs::create_dir_all(&root).unwrap(); + std_fs::write(root.join("mod.rs"), "pub mod existing;").unwrap(); + + ensure_mod_chain(&root, Path::new("blog/posts.rs")) + .await + .unwrap(); + + let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); + let blog_mod = std_fs::read_to_string(root.join("blog/mod.rs")).unwrap(); + // exact contents: a newline separates the existing decl from the new one + assert_eq!(root_mod, "pub mod existing;\npub mod blog;\n"); + assert_eq!(blog_mod, "pub mod posts;\n"); +} + +#[tokio::test] +async fn ensure_mod_chain_creates_fresh_mod_rs_without_leading_newline() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("src/models"); + std_fs::create_dir_all(&root).unwrap(); + + ensure_mod_chain(&root, Path::new("users.json")) + .await + .unwrap(); + + let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); + assert_eq!(root_mod, "pub mod users;\n"); +} + +#[tokio::test] +async fn ensure_mod_chain_no_components_is_noop() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("src/models"); + std_fs::create_dir_all(&root).unwrap(); + // empty path should not error + assert!(ensure_mod_chain(&root, Path::new("")).await.is_ok()); +} + +#[test] +#[serial] +fn resolve_export_dir_prefers_override() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + let cfg = VespertideConfig::default(); + let override_dir = PathBuf::from("custom_out"); + let resolved = super::resolve_export_dir(Some(override_dir.clone()), &cfg); + assert_eq!(resolved, override_dir); +} + +#[rstest] +#[case(OrmArg::Seaorm, Orm::SeaOrm)] +#[case(OrmArg::Sqlalchemy, Orm::SqlAlchemy)] +#[case(OrmArg::Sqlmodel, Orm::SqlModel)] +#[case(OrmArg::Jpa, Orm::Jpa)] +#[case(OrmArg::Prisma, Orm::Prisma)] +fn orm_arg_maps_to_enum(#[case] arg: OrmArg, #[case] expected: Orm) { + assert_eq!(Orm::from(arg), expected); +} + +#[rstest] +#[case("normal_name", "normal_name")] +#[case("user copy", "user_copy")] +#[case("user copy", "user__copy")] +#[case("user-copy", "user-copy")] +#[case("user.copy", "user_copy")] +#[case("user copy.json", "user_copy_json")] +fn test_sanitize_filename(#[case] input: &str, #[case] expected: &str) { + assert_eq!(sanitize_filename(input), expected); +} + +#[test] +fn build_output_path_sanitizes_spaces() { + use std::path::Path; + let root = Path::new("src/models"); + let rel_path = Path::new("user copy.json"); + let out = build_output_path(root, rel_path, Orm::SeaOrm); + assert_eq!(out, Path::new("src/models/user_copy.rs")); + + let rel_path2 = Path::new("blog/post name.yaml"); + let out2 = build_output_path(root, rel_path2, Orm::SeaOrm); + assert_eq!(out2, Path::new("src/models/blog/post_name.rs")); +} + +#[test] +fn build_output_path_handles_file_without_extension() { + use std::path::Path; + let root = Path::new("src/models"); + // File without extension - covers line 88 (else branch) + let rel_path = Path::new("users"); + let out = build_output_path(root, rel_path, Orm::SeaOrm); + assert_eq!(out, Path::new("src/models/users.rs")); + + let out_py = build_output_path(root, rel_path, Orm::SqlAlchemy); + assert_eq!(out_py, Path::new("src/models/users.py")); +} + +#[test] +fn build_output_path_handles_special_path_components() { + use std::path::Path; + let root = Path::new("src/models"); + // Path with CurDir component (.) - covers line 78 (non-Normal component branch) + let rel_path = Path::new("./blog/posts.json"); + let out = build_output_path(root, rel_path, Orm::SeaOrm); + // The . component gets pushed via the else branch + assert!(out.to_string_lossy().contains("posts")); + + // Path with ParentDir component (..) + let rel_path2 = Path::new("../other/items.yaml"); + let out2 = build_output_path(root, rel_path2, Orm::SeaOrm); + assert!(out2.to_string_lossy().contains("items")); +} + +#[test] +fn build_output_path_strips_vespertide_suffix() { + use std::path::Path; + let root = Path::new("src/models"); + + // .vespertide.json -> .rs (strips ".vespertide" from stem) + let rel_path = Path::new("user.vespertide.json"); + let out = build_output_path(root, rel_path, Orm::SeaOrm); + assert_eq!(out, Path::new("src/models/user.rs")); + + // Nested path with .vespertide.json + let rel_path2 = Path::new("blog/post.vespertide.json"); + let out2 = build_output_path(root, rel_path2, Orm::SeaOrm); + assert_eq!(out2, Path::new("src/models/blog/post.rs")); + + // .vespertide.yaml -> .py + let rel_path3 = Path::new("order.vespertide.yaml"); + let out3 = build_output_path(root, rel_path3, Orm::SqlAlchemy); + assert_eq!(out3, Path::new("src/models/order.py")); + + // Regular .json without .vespertide suffix still works + let rel_path4 = Path::new("item.json"); + let out4 = build_output_path(root, rel_path4, Orm::SeaOrm); + assert_eq!(out4, Path::new("src/models/item.rs")); +} + +#[tokio::test] +#[serial] +async fn ensure_mod_chain_strips_vespertide_suffix() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("src/models"); + std_fs::create_dir_all(&root).unwrap(); + + // File with .vespertide suffix should produce mod declaration without it + ensure_mod_chain(&root, Path::new("user.vespertide.json")) + .await + .unwrap(); + + let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); + // Should be "pub mod user;" not "pub mod user_vespertide;" + assert!(root_mod.contains("pub mod user;")); + assert!(!root_mod.contains("user_vespertide")); + + // Nested path with .vespertide suffix + ensure_mod_chain(&root, Path::new("blog/post.vespertide.json")) + .await + .unwrap(); + let root_mod = std_fs::read_to_string(root.join("mod.rs")).unwrap(); + let blog_mod = std_fs::read_to_string(root.join("blog/mod.rs")).unwrap(); + assert!(root_mod.contains("pub mod blog;")); + assert!(blog_mod.contains("pub mod post;")); + assert!(!root_mod.contains("post_vespertide")); + assert!(!blog_mod.contains("post_vespertide")); +} + +#[tokio::test] +async fn clean_export_dir_removes_rs_files_for_seaorm() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("export_dir"); + std_fs::create_dir_all(&root).unwrap(); + + // Create some .rs files that should be cleaned + std_fs::write(root.join("old_model.rs"), "// old rust file").unwrap(); + std_fs::write(root.join("another.rs"), "// another rust file").unwrap(); + // Create a non-.rs file that should NOT be cleaned + std_fs::write(root.join("keep.txt"), "keep this").unwrap(); + + clean_export_dir(&root, Orm::SeaOrm).await.unwrap(); + + // .rs files should be gone + assert!(!root.join("old_model.rs").exists()); + assert!(!root.join("another.rs").exists()); + // .txt file should remain + assert!(root.join("keep.txt").exists()); +} + +#[tokio::test] +async fn clean_export_dir_removes_py_files_for_sqlalchemy() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("export_dir"); + std_fs::create_dir_all(&root).unwrap(); + + // Create some .py files that should be cleaned + std_fs::write(root.join("old_model.py"), "# old python file").unwrap(); + // Create a .rs file that should NOT be cleaned + std_fs::write(root.join("keep.rs"), "// keep this").unwrap(); + + clean_export_dir(&root, Orm::SqlAlchemy).await.unwrap(); + + // .py files should be gone + assert!(!root.join("old_model.py").exists()); + // .rs file should remain + assert!(root.join("keep.rs").exists()); +} + +#[tokio::test] +async fn clean_export_dir_removes_py_files_for_sqlmodel() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("export_dir"); + std_fs::create_dir_all(&root).unwrap(); + + std_fs::write(root.join("model.py"), "# python file").unwrap(); + + clean_export_dir(&root, Orm::SqlModel).await.unwrap(); + + assert!(!root.join("model.py").exists()); +} + +#[tokio::test] +async fn clean_export_dir_removes_java_files_for_jpa() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("export_dir"); + std_fs::create_dir_all(&root).unwrap(); + + std_fs::write(root.join("User.java"), "// java entity").unwrap(); + std_fs::write(root.join("Order.java"), "// java entity").unwrap(); + std_fs::write(root.join("keep.rs"), "// keep this").unwrap(); + + clean_export_dir(&root, Orm::Jpa).await.unwrap(); + + assert!(!root.join("User.java").exists()); + assert!(!root.join("Order.java").exists()); + assert!(root.join("keep.rs").exists()); +} + +#[tokio::test] +async fn clean_export_dir_handles_missing_directory() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("nonexistent_dir"); + + // Should not error on missing directory + let result = clean_export_dir(&root, Orm::SeaOrm).await; + assert!(result.is_ok()); +} + +#[tokio::test] +async fn clean_dir_recursive_cleans_subdirectories() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("export_dir"); + let subdir = root.join("nested"); + std_fs::create_dir_all(&subdir).unwrap(); + + // Create files in root and subdirectory + std_fs::write(root.join("root.rs"), "// root").unwrap(); + std_fs::write(subdir.join("nested.rs"), "// nested").unwrap(); + std_fs::write(subdir.join("keep.txt"), "keep").unwrap(); + + clean_dir_recursive(&root, "rs").await.unwrap(); + + // .rs files should be gone + assert!(!root.join("root.rs").exists()); + assert!(!subdir.join("nested.rs").exists()); + // .txt file should remain + assert!(subdir.join("keep.txt").exists()); + // subdir should still exist (has .txt file) + assert!(subdir.exists()); +} + +#[tokio::test] +async fn clean_dir_recursive_removes_empty_subdirectories() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("export_dir"); + let subdir = root.join("empty_after_clean"); + std_fs::create_dir_all(&subdir).unwrap(); + + // Create only .rs files in subdirectory + std_fs::write(subdir.join("only.rs"), "// only").unwrap(); + + clean_dir_recursive(&root, "rs").await.unwrap(); + + // .rs file should be gone + assert!(!subdir.join("only.rs").exists()); + // Empty subdirectory should be removed + assert!(!subdir.exists()); +} + +#[tokio::test] +async fn clean_dir_recursive_handles_non_directory() { + let tmp = tempdir().unwrap(); + let file_path = tmp.path().join("not_a_dir.txt"); + std_fs::write(&file_path, "content").unwrap(); + + // Should not error when called on a file instead of directory + let result = clean_dir_recursive(&file_path, "rs").await; + assert!(result.is_ok()); +} + +#[test] +fn build_output_path_jpa_uses_pascal_case_java_extension() { + use std::path::Path; + let root = Path::new("src/models"); + + // snake_case model → PascalCase .java + let rel_path = Path::new("order_item.json"); + let out = build_output_path(root, rel_path, Orm::Jpa); + assert_eq!(out, Path::new("src/models/OrderItem.java")); + + // Single word + let rel_path2 = Path::new("users.json"); + let out2 = build_output_path(root, rel_path2, Orm::Jpa); + assert_eq!(out2, Path::new("src/models/Users.java")); + + // Nested path + let rel_path3 = Path::new("blog/post_comment.yaml"); + let out3 = build_output_path(root, rel_path3, Orm::Jpa); + assert_eq!(out3, Path::new("src/models/blog/PostComment.java")); +} + +#[test] +fn build_output_path_jpa_strips_vespertide_suffix() { + use std::path::Path; + let root = Path::new("src/models"); + + let rel_path = Path::new("user.vespertide.json"); + let out = build_output_path(root, rel_path, Orm::Jpa); + assert_eq!(out, Path::new("src/models/User.java")); +} + +#[rstest] +#[case("order_item", "OrderItem")] +#[case("users", "Users")] +#[case("a", "A")] +#[case("user_profile_image", "UserProfileImage")] +#[case("a__b", "AB")] +fn test_to_pascal_case(#[case] input: &str, #[case] expected: &str) { + assert_eq!(to_pascal_case(input), expected); +} diff --git a/crates/vespertide-cli/src/commands/export/tests/prisma.rs b/crates/vespertide-cli/src/commands/export/tests/prisma.rs new file mode 100644 index 00000000..da450719 --- /dev/null +++ b/crates/vespertide-cli/src/commands/export/tests/prisma.rs @@ -0,0 +1,55 @@ +use super::*; + +#[tokio::test] +#[serial] +async fn export_prisma_writes_backend_neutral_schema() { + let tmp = tempdir().unwrap(); + let _guard = CwdGuard::new(&tmp.path().to_path_buf()); + write_config(); + + let mut model = sample_table("events"); + model.columns.push(ColumnDef { + name: "occurred_at".into(), + r#type: ColumnType::Simple(SimpleColumnType::Timestamptz), + nullable: false, + default: None, + comment: None, + primary_key: None, + unique: None, + index: None, + foreign_key: None, + }); + write_model(Path::new("models/events.json"), &model); + + cmd_export(OrmArg::Prisma, None).await.unwrap(); + + let out = PathBuf::from("src/models/schema.prisma"); + assert!(out.exists()); + let content = std_fs::read_to_string(out).unwrap(); + assert!(content.contains("provider = \"postgresql\"")); + assert!(content.contains("occurred_at DateTime")); + assert!(!content.contains("@db.")); +} + +#[test] +fn build_output_path_prisma_uses_prisma_extension() { + use std::path::Path; + let root = Path::new("src/models"); + let out = build_output_path(root, Path::new("user.json"), Orm::Prisma); + assert_eq!(out, Path::new("src/models/user.prisma")); +} + +#[tokio::test] +#[serial] +async fn clean_export_dir_removes_stale_prisma_files() { + let tmp = tempdir().unwrap(); + let root = tmp.path().join("out"); + std_fs::create_dir_all(&root).unwrap(); + std_fs::write(root.join("schema.prisma"), "stale").unwrap(); + std_fs::write(root.join("keep.txt"), "keep").unwrap(); + + clean_export_dir(&root, Orm::Prisma).await.unwrap(); + + assert!(!root.join("schema.prisma").exists()); + assert!(root.join("keep.txt").exists()); +} diff --git a/crates/vespertide-core/src/schema/table/tests/mod.rs b/crates/vespertide-core/src/schema/table/tests/mod.rs index 8f7fa584..2b195c93 100644 --- a/crates/vespertide-core/src/schema/table/tests/mod.rs +++ b/crates/vespertide-core/src/schema/table/tests/mod.rs @@ -774,7 +774,7 @@ fn normalize_inline_unique_with_array_existing_constraint() { let unique_constraint = &normalized.constraints[0]; assert!(matches!( unique_constraint, - TableConstraint::Unique { name: Some(n), columns: _, .. } + TableConstraint::Unique { name: Some(n), .. } if n == "uq_group" )); if let TableConstraint::Unique { columns, .. } = unique_constraint { diff --git a/crates/vespertide-exporter/Cargo.toml b/crates/vespertide-exporter/Cargo.toml index 35f13948..98addd72 100644 --- a/crates/vespertide-exporter/Cargo.toml +++ b/crates/vespertide-exporter/Cargo.toml @@ -23,6 +23,7 @@ criterion = { version = "0.8", features = ["html_reports"] } rstest = "0.26" insta = { version = "1.47", features = ["yaml"] } proptest = "1" +serde_json = "1" [[bench]] name = "codegen_benchmarks" diff --git a/crates/vespertide-exporter/src/lib.rs b/crates/vespertide-exporter/src/lib.rs index 0d4732c5..c1d5069d 100644 --- a/crates/vespertide-exporter/src/lib.rs +++ b/crates/vespertide-exporter/src/lib.rs @@ -1,9 +1,10 @@ //! Helpers to convert `TableDef` models into ORM-specific representations -//! such as `SeaORM`, `SQLAlchemy`, `SQLModel`, and JPA. +//! such as `SeaORM`, `SQLAlchemy`, `SQLModel`, JPA, and Prisma. pub mod jpa; pub mod orm; mod parallel_config; +pub mod prisma; pub mod seaorm; pub mod sqlalchemy; pub mod sqlmodel; @@ -13,6 +14,7 @@ mod utils; pub use jpa::JpaExporter; pub use orm::{Orm, OrmExporter, render_entity, render_entity_with_schema}; +pub use prisma::PrismaExporter; pub use seaorm::{SeaOrmExporter, render_entity as render_seaorm_entity}; pub use sqlalchemy::SqlAlchemyExporter; pub use sqlmodel::SqlModelExporter; diff --git a/crates/vespertide-exporter/src/orm.rs b/crates/vespertide-exporter/src/orm.rs index ea18829f..d1d68d2a 100644 --- a/crates/vespertide-exporter/src/orm.rs +++ b/crates/vespertide-exporter/src/orm.rs @@ -1,8 +1,8 @@ use vespertide_core::TableDef; use crate::{ - jpa::JpaExporter, seaorm::SeaOrmExporter, sqlalchemy::SqlAlchemyExporter, - sqlmodel::SqlModelExporter, + jpa::JpaExporter, prisma::PrismaExporter, seaorm::SeaOrmExporter, + sqlalchemy::SqlAlchemyExporter, sqlmodel::SqlModelExporter, }; /// Supported ORM targets. @@ -12,6 +12,7 @@ pub enum Orm { SqlAlchemy, SqlModel, Jpa, + Prisma, } /// Standardized exporter interface for all supported ORMs. @@ -36,6 +37,7 @@ pub fn render_entity(orm: Orm, table: &TableDef) -> Result { Orm::SqlAlchemy => SqlAlchemyExporter.render_entity(table), Orm::SqlModel => SqlModelExporter.render_entity(table), Orm::Jpa => JpaExporter.render_entity(table), + Orm::Prisma => PrismaExporter.render_entity(table), } } @@ -50,6 +52,7 @@ pub fn render_entity_with_schema( Orm::SqlAlchemy => SqlAlchemyExporter.render_entity_with_schema(table, schema), Orm::SqlModel => SqlModelExporter.render_entity_with_schema(table, schema), Orm::Jpa => JpaExporter.render_entity_with_schema(table, schema), + Orm::Prisma => PrismaExporter.render_entity_with_schema(table, schema), } } @@ -64,6 +67,7 @@ mod tests { #[case::sqlalchemy(Orm::SqlAlchemy)] #[case::sqlmodel(Orm::SqlModel)] #[case::jpa(Orm::Jpa)] + #[case::prisma(Orm::Prisma)] fn dispatch_render_entity_succeeds(#[case] orm: Orm) { let table = basic_single_pk(); assert!(render_entity(orm, &table).is_ok()); @@ -74,6 +78,7 @@ mod tests { #[case::sqlalchemy(Orm::SqlAlchemy)] #[case::sqlmodel(Orm::SqlModel)] #[case::jpa(Orm::Jpa)] + #[case::prisma(Orm::Prisma)] fn dispatch_render_entity_with_schema_succeeds(#[case] orm: Orm) { let table = basic_single_pk(); let schema = vec![table.clone()]; diff --git a/crates/vespertide-exporter/src/prisma/enums.rs b/crates/vespertide-exporter/src/prisma/enums.rs new file mode 100644 index 00000000..f38c5436 --- /dev/null +++ b/crates/vespertide-exporter/src/prisma/enums.rs @@ -0,0 +1,65 @@ +use vespertide_core::schema::column::EnumValues; + +pub(super) fn render_enum(name: &str, values: &EnumValues) -> String { + let enum_name = to_pascal_case(name); + let mut lines = Vec::new(); + lines.push(format!("enum {enum_name} {{")); + match values { + EnumValues::String(vals) => { + for val in vals { + let variant = to_screaming_snake(val); + if variant == *val { + lines.push(format!(" {variant}")); + } else { + lines.push(format!(" {variant} @map(\"{val}\")")); + } + } + } + EnumValues::Integer(vals) => { + // Prisma doesn't support integer enums natively; emit as string variants with comment + for val in vals { + let variant = to_screaming_snake(&val.name); + let value = val.value; + lines.push(format!(" {variant} // = {value}")); + } + } + } + lines.push("}".into()); + lines.join("\n") +} + +pub(super) fn to_pascal_case(s: &str) -> String { + s.split('_') + .map(|word| { + let mut chars = word.chars(); + match chars.next() { + None => String::new(), + Some(first) => first.to_uppercase().chain(chars).collect(), + } + }) + .collect() +} + +pub(super) fn to_screaming_snake(s: &str) -> String { + let mut result = String::new(); + let mut prev_lower = false; + for ch in s.chars() { + if ch.is_uppercase() && prev_lower { + result.push('_'); + } + if ch.is_alphanumeric() { + result.push(ch.to_ascii_uppercase()); + prev_lower = ch.is_lowercase(); + } else { + result.push('_'); + prev_lower = false; + } + } + let result = result.trim_end_matches('_').to_string(); + // Prisma identifiers cannot start with a digit; prefix with '_' to keep it valid. + if result.starts_with(|c: char| c.is_ascii_digit()) { + format!("_{result}") + } else { + result + } +} diff --git a/crates/vespertide-exporter/src/prisma/mod.rs b/crates/vespertide-exporter/src/prisma/mod.rs new file mode 100644 index 00000000..540437a2 --- /dev/null +++ b/crates/vespertide-exporter/src/prisma/mod.rs @@ -0,0 +1,140 @@ +mod enums; +mod render; +mod types; + +use std::collections::HashSet; + +use crate::orm::OrmExporter; +use vespertide_core::TableDef; +use vespertide_core::schema::column::{ColumnType, ComplexColumnType, EnumValues}; + +pub struct PrismaExporter; + +impl OrmExporter for PrismaExporter { + fn render_entity(&self, table: &TableDef) -> Result { + Ok(render_entity(table)) + } + + fn render_entity_with_schema( + &self, + table: &TableDef, + schema: &[TableDef], + ) -> Result { + Ok(render_entity_with_schema(table, schema)) + } +} + +/// Render a complete `schema.prisma` file for all tables. +/// +/// Output order: datasource → generator → (globally deduped) enum blocks → model blocks. +/// +/// The output is backend-neutral: model bodies carry no `@db.*` native +/// attributes, so the file stays valid regardless of which backend vespertide +/// targets. The datasource provider is fixed to `postgresql` (Prisma requires +/// declaring exactly one); users on another backend only change that one line. +pub fn render_schema(tables: &[TableDef]) -> String { + let mut seen_enums: HashSet = HashSet::new(); + let mut enum_blocks: Vec = Vec::new(); + for table in tables { + for (name, values) in collect_table_enums(table) { + if seen_enums.insert(name.to_string()) { + enum_blocks.push(enums::render_enum(name, values)); + } + } + } + + let mut parts: Vec = Vec::new(); + + // No `url` here: current Prisma moved connection config out of the + // schema file (`prisma.config.ts`), which also matches vespertide's + // models-only scope. + parts.push("datasource db {\n provider = \"postgresql\"\n}".to_string()); + + parts.push("generator client {\n provider = \"prisma-client-js\"\n}".to_string()); + + parts.extend(enum_blocks); + + for table in tables { + parts.push(render::render_model(table, tables)); + } + + parts.join("\n\n") + "\n" +} + +fn collect_table_enums(table: &TableDef) -> Vec<(&str, &EnumValues)> { + let mut seen = HashSet::new(); + let mut result = Vec::new(); + for col in &table.columns { + if let ColumnType::Complex(ComplexColumnType::Enum { name, values }) = &col.r#type + && seen.insert(name.as_str()) + { + result.push((name.as_str(), values)); + } + } + result +} + +/// Render enum blocks + model block without schema context. +/// +/// Passes the table itself as a one-element schema so that self-referential FK +/// back-relations are always emitted (Prisma requires both sides of a relation to +/// be present in the model, including self-referential ones). +pub fn render_entity(table: &TableDef) -> String { + render_entity_with_schema(table, std::slice::from_ref(table)) +} + +/// Render enum blocks + model block with full schema context (includes back-relations). +pub fn render_entity_with_schema(table: &TableDef, schema: &[TableDef]) -> String { + let mut parts: Vec = Vec::new(); + for (name, values) in collect_table_enums(table) { + parts.push(enums::render_enum(name, values)); + } + parts.push(render::render_model(table, schema)); + parts.join("\n\n") +} + +/// Multi-table entry point: render every table (enum + model blocks) with full +/// schema context and join them. Mirrors the other ORMs' `export` so the +/// cross-ORM test harness can dispatch Prisma through a single call. The +/// `datasource`/`generator` wrapper lives in [`render_schema`]. +pub fn export(schema: &[TableDef]) -> Result { + Ok(schema + .iter() + .map(|table| render_entity_with_schema(table, schema)) + .collect::>() + .join("\n\n")) +} + +/// Test-only accessor for the internal `to_pascal_case` helper, mirroring the +/// other ORM backends so the cross-ORM consolidation test can exercise it +/// without making the helper generally public. +#[cfg(test)] +pub fn to_pascal_case_for_tests(s: &str) -> String { + enums::to_pascal_case(s) +} + +#[cfg(test)] +mod tests { + use super::*; + use crate::tests::fixtures::basic_single_pk; + + #[test] + fn render_schema_emits_postgresql_provider() { + let tables = vec![basic_single_pk()]; + let schema = render_schema(&tables); + + assert!(schema.contains("provider = \"postgresql\"")); + assert!(!schema.contains("@db.")); + assert!(!schema.contains("output")); + } + + #[test] + fn render_schema_emits_shared_enum_block_once() { + let t1 = crate::tests::fixtures::enum_shared(); + let mut t2 = crate::tests::fixtures::enum_shared(); + t2.name = "archived_documents".into(); + let schema = render_schema(&[t1, t2]); + + assert_eq!(schema.matches("enum DocStatus {").count(), 1); + } +} diff --git a/crates/vespertide-exporter/src/prisma/render.rs b/crates/vespertide-exporter/src/prisma/render.rs new file mode 100644 index 00000000..262c67af --- /dev/null +++ b/crates/vespertide-exporter/src/prisma/render.rs @@ -0,0 +1,513 @@ +use std::collections::HashMap; + +use vespertide_core::TableDef; +use vespertide_core::schema::column::{ColumnType, ComplexColumnType, EnumValues}; +use vespertide_core::schema::constraint::TableConstraint; +use vespertide_core::schema::names::ColumnName; +use vespertide_core::schema::reference::ReferenceAction; + +use super::enums::{to_pascal_case, to_screaming_snake}; +use super::types::column_type_to_prisma; + +struct PkInfo { + columns: Vec, + auto_increment: bool, +} + +fn extract_pk_info(constraints: &[TableConstraint]) -> PkInfo { + for c in constraints { + if let TableConstraint::PrimaryKey { + auto_increment, + columns, + .. + } = c + { + return PkInfo { + columns: columns.iter().map(ToString::to_string).collect(), + auto_increment: *auto_increment, + }; + } + } + PkInfo { + columns: Vec::new(), + auto_increment: false, + } +} + +struct FkInfo<'a> { + ref_table: &'a str, + ref_cols: &'a [ColumnName], + on_delete: Option<&'a ReferenceAction>, + on_update: Option<&'a ReferenceAction>, +} + +pub(super) struct BackRelation { + pub(super) source_table: String, + pub(super) fk_col: String, + pub(super) is_one_to_one: bool, + pub(super) relation_name: Option, +} + +pub(super) fn back_rel_field(br: &BackRelation) -> (String, String) { + let source_pascal = to_pascal_case(&br.source_table); + let rel_type = if br.is_one_to_one { + format!("{source_pascal}?") + } else { + format!("{source_pascal}[]") + }; + + let field_name = if br.relation_name.is_some() { + let rel_field = infer_relation_field_name(&br.fk_col); + format!("{rel_field}_{}", br.source_table) + } else { + br.source_table.clone() + }; + + (field_name, rel_type) +} + +pub(super) fn collect_back_relations(target_table: &str, schema: &[TableDef]) -> Vec { + let mut result = Vec::new(); + + for source in schema { + let fks_to_target: Vec<(&str, &[ColumnName])> = source + .constraints + .iter() + .filter_map(|c| { + if let TableConstraint::ForeignKey { + columns, + ref_table, + ref_columns, + .. + } = c + { + if ref_table.as_str() == target_table && columns.len() == 1 { + Some((columns[0].as_str(), ref_columns.as_slice())) + } else { + None + } + } else { + None + } + }) + .collect(); + + if fks_to_target.is_empty() { + continue; + } + + let multi_fk = fks_to_target.len() > 1; + let is_self_ref = source.name.as_str() == target_table; + + for (fk_col, _) in &fks_to_target { + let is_unique = source.constraints.iter().any(|c| { + matches!(c, TableConstraint::Unique { columns, .. } + if columns.len() == 1 && columns[0].as_str() == *fk_col) + }); + + let needs_name = multi_fk || is_self_ref; + let relation_name = if needs_name { + let rel_field = infer_relation_field_name(fk_col); + let source_pascal = to_pascal_case(&source.name); + let rel_pascal = to_pascal_case(&rel_field); + Some(format!("{source_pascal}{rel_pascal}")) + } else { + None + }; + + result.push(BackRelation { + source_table: source.name.as_str().to_string(), + fk_col: fk_col.to_string(), + is_one_to_one: is_unique, + relation_name, + }); + } + } + + result +} + +pub(super) fn render_model(table: &TableDef, schema: &[TableDef]) -> String { + let mut lines: Vec = Vec::new(); + + if let Some(desc) = &table.description { + for line in desc.lines() { + lines.push(format!("/// {line}")); + } + } + + let model_name = to_pascal_case(&table.name); + lines.push(format!("model {model_name} {{")); + + let pk_info = extract_pk_info(&table.constraints); + let pk_columns: std::collections::HashSet<&str> = + pk_info.columns.iter().map(String::as_str).collect(); + let is_composite_pk = pk_info.columns.len() > 1; + + let unique_single: HashMap<&str, Option<&str>> = table + .constraints + .iter() + .filter_map(|c| { + if let TableConstraint::Unique { name, columns, .. } = c { + if columns.len() == 1 { + Some((columns[0].as_str(), name.as_deref())) + } else { + None + } + } else { + None + } + }) + .collect(); + + // FK lookup by column + let fk_by_col: HashMap<&str, FkInfo<'_>> = table + .constraints + .iter() + .filter_map(|c| { + if let TableConstraint::ForeignKey { + columns, + ref_table, + ref_columns, + on_delete, + on_update, + .. + } = c + { + if columns.len() == 1 { + Some(( + columns[0].as_str(), + FkInfo { + ref_table: ref_table.as_str(), + ref_cols: ref_columns.as_slice(), + on_delete: on_delete.as_ref(), + on_update: on_update.as_ref(), + }, + )) + } else { + None + } + } else { + None + } + }) + .collect(); + + // Count FKs per ref_table for disambiguation detection + let mut ref_table_fk_count: HashMap<&str, usize> = HashMap::new(); + for fk in fk_by_col.values() { + *ref_table_fk_count.entry(fk.ref_table).or_default() += 1; + } + + // Render scalar fields + inline relation fields + for col in &table.columns { + let col_name = col.name.as_str(); + let in_pk = pk_columns.contains(col_name); + let is_single_pk = in_pk && !is_composite_pk; + let auto_inc = is_single_pk && pk_info.auto_increment; + let is_unique = unique_single.get(col_name).copied(); + + if let Some(ref comment) = col.comment { + let comment = comment.replace('\n', " "); + lines.push(format!(" /// {comment}")); + } + + let (type_str, trailing_comment) = column_type_to_prisma(&col.r#type, col.nullable); + let mut attrs: Vec = Vec::new(); + + if is_single_pk { + attrs.push("@id".into()); + if auto_inc { + attrs.push("@default(autoincrement())".into()); + } + } + + if !auto_inc && let Some(ref default) = col.default { + attrs.push(prisma_default_attr(&default.to_sql(), &col.r#type)); + } + + if let Some(unique_name) = is_unique + && !is_single_pk + { + match unique_name { + Some(n) => attrs.push(format!("@unique(map: \"{n}\")")), + None => attrs.push("@unique".into()), + } + } + + let attrs_str = if attrs.is_empty() { + String::new() + } else { + format!(" {}", attrs.join(" ")) + }; + + let comment_str = trailing_comment + .map(|c| format!(" // {c}")) + .unwrap_or_default(); + + lines.push(format!(" {col_name} {type_str}{attrs_str}{comment_str}")); + + // Emit inline relation field for FK columns + if let Some(fk) = fk_by_col.get(col_name) { + // `rel_name_segment` drives @relation("…") naming — must stay consistent + // with the segment computed by collect_back_relations on the other side. + let rel_name_segment = infer_relation_field_name(col_name); + // When no `_id` was stripped the inferred name equals the FK column name, + // which would collide with the scalar field. Append `_rel` to deduplicate. + let rel_field_name = if rel_name_segment == col_name { + format!("{col_name}_rel") + } else { + rel_name_segment.clone() + }; + let rel_model = to_pascal_case(fk.ref_table); + let rel_type = if col.nullable { + format!("{rel_model}?") + } else { + rel_model.clone() + }; + + let multi_fk = ref_table_fk_count.get(fk.ref_table).copied().unwrap_or(0) > 1; + let is_self_ref = fk.ref_table == table.name.as_str(); + let needs_name = multi_fk || is_self_ref; + + let mut rel_args: Vec = Vec::new(); + if needs_name { + // Use rel_name_segment (pre-dedup) so the name matches back-relations. + let table_pascal = to_pascal_case(&table.name); + let field_pascal = to_pascal_case(&rel_name_segment); + rel_args.push(format!("\"{table_pascal}{field_pascal}\"")); + } + rel_args.push(format!("fields: [{col_name}]")); + rel_args.push(format!( + "references: [{}]", + fk.ref_cols + .iter() + .map(ColumnName::as_str) + .collect::>() + .join(", ") + )); + if let Some(od) = fk.on_delete { + let action = reference_action_to_prisma(od); + rel_args.push(format!("onDelete: {action}")); + } + if let Some(ou) = fk.on_update { + let action = reference_action_to_prisma(ou); + rel_args.push(format!("onUpdate: {action}")); + } + + let rel_args_str = rel_args.join(", "); + lines.push(format!( + " {rel_field_name} {rel_type} @relation({rel_args_str})" + )); + } + } + + // Back-relations from schema context + if !schema.is_empty() { + let back_rels = collect_back_relations(&table.name, schema); + for br in &back_rels { + let (field_name, rel_type) = back_rel_field(br); + let rel_attr = match &br.relation_name { + Some(name) => format!(" @relation(\"{name}\")"), + None => String::new(), + }; + lines.push(format!(" {field_name} {rel_type}{rel_attr}")); + } + } + + // Blank line before model-level attributes + lines.push(String::new()); + + // Composite PK + if is_composite_pk { + let pk_cols = pk_info.columns.join(", "); + lines.push(format!(" @@id([{pk_cols}])")); + } + + // Composite unique constraints + for c in &table.constraints { + if let TableConstraint::Unique { name, columns, .. } = c + && columns.len() > 1 + { + let cols = columns.join(", "); + if let Some(n) = name { + lines.push(format!(" @@unique([{cols}], map: \"{n}\")")); + } else { + lines.push(format!(" @@unique([{cols}])")); + } + } + } + + // All index constraints + for c in &table.constraints { + if let TableConstraint::Index { name, columns } = c { + let cols = columns.join(", "); + // `match` instead of `if let Some`: LLVM coverage attributes match + // arms reliably where this if/else was misattributed as uncovered. + match name { + Some(n) => lines.push(format!(" @@index([{cols}], map: \"{n}\")")), + None => lines.push(format!(" @@index([{cols}])")), + } + } + } + + // @@map (always present since model is PascalCase but table is snake_case) + let table_name = table.name.as_str(); + lines.push(format!(" @@map(\"{table_name}\")")); + lines.push("}".into()); + + lines.join("\n") +} + +fn prisma_default_attr(default_sql: &str, col_type: &ColumnType) -> String { + // Integer-backed enum: resolve to a variant identifier (SCREAMING_SNAKE), never a bare int. + if let ColumnType::Complex(ComplexColumnType::Enum { + values: EnumValues::Integer(int_values), + .. + }) = col_type + { + let key = default_sql.trim_matches(|c: char| c == '\'' || c == '"'); + // 1) numeric value match → variant name + if let Ok(n) = key.parse::() + && let Some(v) = int_values.iter().find(|v| v.value == n) + { + return format!("@default({})", to_screaming_snake(&v.name)); + } + // 2) exact variant-name match → variant name + if let Some(v) = int_values.iter().find(|v| v.name == key) { + return format!("@default({})", to_screaming_snake(&v.name)); + } + // 3) no match → dbgenerated fallback (valid PSL; avoids bare-int type error) + let escaped = key.replace('"', "\\\""); + return format!("@default(dbgenerated(\"{escaped}\"))"); + } + + if default_sql == "true" { + return "@default(true)".into(); + } + if default_sql == "false" { + return "@default(false)".into(); + } + + let lower = default_sql.to_lowercase(); + if lower.contains("now()") || lower.starts_with("current_timestamp") { + return "@default(now())".into(); + } + if lower.contains("gen_random_uuid()") + || lower.contains("uuid_generate_v4()") + || lower.contains("newid()") + { + return "@default(uuid())".into(); + } + + // Any remaining function call → dbgenerated + if default_sql.contains('(') { + let escaped = default_sql.replace('"', "\\\""); + return format!("@default(dbgenerated(\"{escaped}\"))"); + } + + // String literal with quotes — may be an enum value + if default_sql.starts_with('\'') || default_sql.starts_with('"') { + let stripped = default_sql.trim_matches(|c| c == '\'' || c == '"'); + if let ColumnType::Complex(ComplexColumnType::Enum { + values: EnumValues::String(variants), + .. + }) = col_type + && variants.iter().any(|v| v.as_str() == stripped) + { + let variant = to_screaming_snake(stripped); + return format!("@default({variant})"); + } + let s = stripped.replace('\\', "\\\\").replace('"', "\\\""); + return format!("@default(\"{s}\")"); + } + + // Numeric + if default_sql.parse::().is_ok() { + return format!("@default({default_sql})"); + } + + // Fallback + let escaped = default_sql.replace('"', "\\\""); + format!("@default(dbgenerated(\"{escaped}\"))") +} + +fn reference_action_to_prisma(action: &ReferenceAction) -> &'static str { + match action { + ReferenceAction::Cascade => "Cascade", + ReferenceAction::Restrict => "Restrict", + ReferenceAction::SetNull => "SetNull", + ReferenceAction::SetDefault => "SetDefault", + // Includes NoAction and unknown/future referential actions. + _ => "NoAction", + } +} + +fn infer_relation_field_name(fk_col: &str) -> String { + fk_col.strip_suffix("_id").unwrap_or(fk_col).to_string() +} + +#[cfg(test)] +mod tests { + use rstest::rstest; + use vespertide_core::schema::column::{NumValue, SimpleColumnType}; + + use super::*; + + #[rstest] + #[case::cascade(ReferenceAction::Cascade, "Cascade")] + #[case::restrict(ReferenceAction::Restrict, "Restrict")] + #[case::set_null(ReferenceAction::SetNull, "SetNull")] + #[case::set_default(ReferenceAction::SetDefault, "SetDefault")] + #[case::no_action(ReferenceAction::NoAction, "NoAction")] + fn reference_actions_map_to_prisma(#[case] action: ReferenceAction, #[case] expected: &str) { + assert_eq!(reference_action_to_prisma(&action), expected); + } + + #[test] + fn integer_enum_default_resolves_numeric_value_to_variant_name() { + let ty = ColumnType::Complex(ComplexColumnType::Enum { + name: "priority".into(), + values: EnumValues::Integer(vec![ + NumValue { + name: "low".into(), + value: 100, + }, + NumValue { + name: "high".into(), + value: 200, + }, + ]), + }); + assert_eq!(prisma_default_attr("100", &ty), "@default(LOW)"); + } + + #[test] + fn fk_on_update_action_is_rendered() { + let mut table = crate::tests::fixtures::table_with_fk(); + for c in &mut table.constraints { + if let TableConstraint::ForeignKey { on_update, .. } = c { + *on_update = Some(ReferenceAction::Cascade); + } + } + let rendered = render_model(&table, std::slice::from_ref(&table)); + assert!(rendered.contains("onUpdate: Cascade")); + } + + #[test] + fn named_index_is_rendered_with_map() { + let table = crate::tests::fixtures::table_with_indexes(); + let rendered = render_model(&table, &[]); + assert!(rendered.contains("@@index([created_at], map: \"idx_articles_created_at\")")); + assert!(rendered.contains("@@index([title])")); + } + + #[rstest] + #[case::postgres_gen_random_uuid("gen_random_uuid()")] + #[case::postgres_uuid_generate_v4("uuid_generate_v4()")] + #[case::mssql_newid("NEWID()")] + fn uuid_generating_default_maps_to_client_uuid(#[case] default_sql: &str) { + let ty = ColumnType::Simple(SimpleColumnType::Uuid); + assert_eq!(prisma_default_attr(default_sql, &ty), "@default(uuid())"); + } +} diff --git a/crates/vespertide-exporter/src/prisma/types.rs b/crates/vespertide-exporter/src/prisma/types.rs new file mode 100644 index 00000000..d83e6c67 --- /dev/null +++ b/crates/vespertide-exporter/src/prisma/types.rs @@ -0,0 +1,176 @@ +use vespertide_core::schema::column::{ColumnType, ComplexColumnType, SimpleColumnType}; + +use super::enums::to_pascal_case; + +/// Maps a vespertide column type to a `(Prisma scalar type, optional trailing +/// `// ...` comment)` pair. +/// +/// The output is backend-neutral: no `@db.*` native attributes are emitted, so +/// the same model body is valid under every Prisma provider. Physical column +/// types are owned by vespertide's own DDL generation, not the Prisma schema. +pub(super) fn column_type_to_prisma(ty: &ColumnType, nullable: bool) -> (String, Option) { + let q = if nullable { "?" } else { "" }; + + match ty { + // vespertide's MySQL DDL stores uuid as BINARY(16) — the one physical + // type that diverges from the neutral scalar, so the field carries a + // warning comment instead of a backend-specific type. + ColumnType::Simple(SimpleColumnType::Uuid) => ( + format!("String{q}"), + Some("stored as binary(16) on MySQL backends".to_string()), + ), + ColumnType::Simple(simple) => { + let base = match simple { + SimpleColumnType::SmallInt | SimpleColumnType::Integer => "Int", + SimpleColumnType::BigInt => "BigInt", + SimpleColumnType::Real | SimpleColumnType::DoublePrecision => "Float", + SimpleColumnType::Boolean => "Boolean", + SimpleColumnType::Date + | SimpleColumnType::Time + | SimpleColumnType::Timestamp + | SimpleColumnType::Timestamptz => "DateTime", + SimpleColumnType::Bytea => "Bytes", + SimpleColumnType::Json => "Json", + SimpleColumnType::Text + | SimpleColumnType::Interval + | SimpleColumnType::Inet + | SimpleColumnType::Cidr + | SimpleColumnType::Macaddr + | SimpleColumnType::Xml => "String", + // Unknown/future simple types fall back to a plain String column. + // `SimpleColumnType` is #[non_exhaustive]; every current variant is + // handled above, so this arm is currently unreachable. + #[cfg(not(tarpaulin_include))] + _ => "String", + }; + (format!("{base}{q}"), None) + } + ColumnType::Complex(complex) => match complex { + ComplexColumnType::Varchar { .. } | ComplexColumnType::Char { .. } => { + (format!("String{q}"), None) + } + ComplexColumnType::Numeric { .. } => (format!("Decimal{q}"), None), + ComplexColumnType::Custom { custom_type } => { + (format!("Unsupported(\"{custom_type}\"){q}"), None) + } + ComplexColumnType::Enum { name, .. } => { + let pascal = to_pascal_case(name); + (format!("{pascal}{q}"), None) + } + // Unknown/future complex types fall back to a plain String column. + // `ComplexColumnType` is #[non_exhaustive]; every current variant is + // matched above, so this arm is currently unreachable. + #[cfg(not(tarpaulin_include))] + _ => (format!("String{q}"), None), + }, + } +} + +#[cfg(test)] +mod tests { + use rstest::rstest; + use vespertide_core::schema::column::{ColumnType, ComplexColumnType, SimpleColumnType}; + + use super::*; + + /// Exhaustive oracle for every simple type. Each row is independent test + /// data, so flipping any production mapping arm fails here immediately. + #[rstest] + #[case::small_int(SimpleColumnType::SmallInt, "Int")] + #[case::integer(SimpleColumnType::Integer, "Int")] + #[case::big_int(SimpleColumnType::BigInt, "BigInt")] + #[case::real(SimpleColumnType::Real, "Float")] + #[case::double_precision(SimpleColumnType::DoublePrecision, "Float")] + #[case::boolean(SimpleColumnType::Boolean, "Boolean")] + #[case::date(SimpleColumnType::Date, "DateTime")] + #[case::time(SimpleColumnType::Time, "DateTime")] + #[case::timestamp(SimpleColumnType::Timestamp, "DateTime")] + #[case::timestamptz(SimpleColumnType::Timestamptz, "DateTime")] + #[case::bytea(SimpleColumnType::Bytea, "Bytes")] + #[case::json(SimpleColumnType::Json, "Json")] + #[case::text(SimpleColumnType::Text, "String")] + #[case::interval(SimpleColumnType::Interval, "String")] + #[case::inet(SimpleColumnType::Inet, "String")] + #[case::cidr(SimpleColumnType::Cidr, "String")] + #[case::macaddr(SimpleColumnType::Macaddr, "String")] + #[case::xml(SimpleColumnType::Xml, "String")] + fn simple_types_map_to_neutral_scalars(#[case] simple: SimpleColumnType, #[case] scalar: &str) { + let (rendered, comment) = column_type_to_prisma(&ColumnType::Simple(simple), false); + assert_eq!(rendered, scalar); + assert_eq!(comment, None); + } + + #[test] + fn uuid_maps_to_string_with_mysql_binary16_note() { + let ty = ColumnType::Simple(SimpleColumnType::Uuid); + let (rendered, comment) = column_type_to_prisma(&ty, false); + assert_eq!(rendered, "String"); + assert_eq!( + comment.as_deref(), + Some("stored as binary(16) on MySQL backends") + ); + + let (rendered, _) = column_type_to_prisma(&ty, true); + assert_eq!(rendered, "String?"); + } + + #[test] + fn nullable_appends_question_mark() { + let ty = ColumnType::Simple(SimpleColumnType::Timestamptz); + let (rendered, comment) = column_type_to_prisma(&ty, true); + assert_eq!(rendered, "DateTime?"); + assert_eq!(comment, None); + } + + #[test] + fn sized_complex_types_drop_size_info() { + let (rendered, comment) = column_type_to_prisma( + &ColumnType::Complex(ComplexColumnType::Varchar { length: 255 }), + false, + ); + assert_eq!(rendered, "String"); + assert_eq!(comment, None); + + let (rendered, comment) = column_type_to_prisma( + &ColumnType::Complex(ComplexColumnType::Char { length: 3 }), + false, + ); + assert_eq!(rendered, "String"); + assert_eq!(comment, None); + + let (rendered, comment) = column_type_to_prisma( + &ColumnType::Complex(ComplexColumnType::Numeric { + precision: 10, + scale: 2, + }), + true, + ); + assert_eq!(rendered, "Decimal?"); + assert_eq!(comment, None); + } + + #[test] + fn custom_and_enum_types_render_without_comment() { + let (rendered, comment) = column_type_to_prisma( + &ColumnType::Complex(ComplexColumnType::Custom { + custom_type: "ltree".into(), + }), + false, + ); + assert_eq!(rendered, "Unsupported(\"ltree\")"); + assert_eq!(comment, None); + + let (rendered, comment) = column_type_to_prisma( + &ColumnType::Complex(ComplexColumnType::Enum { + name: "order_status".into(), + values: vespertide_core::schema::column::EnumValues::String(vec![ + "open".into(), + "closed".into(), + ]), + }), + false, + ); + assert_eq!(rendered, "OrderStatus"); + assert_eq!(comment, None); + } +} diff --git a/crates/vespertide-exporter/src/tests/mod.rs b/crates/vespertide-exporter/src/tests/mod.rs index 5d99b907..fc6cc638 100644 --- a/crates/vespertide-exporter/src/tests/mod.rs +++ b/crates/vespertide-exporter/src/tests/mod.rs @@ -17,6 +17,7 @@ fn render_schema(orm: Orm, schema: &[TableDef]) -> Result { Orm::SqlAlchemy => crate::sqlalchemy::export(schema), Orm::SqlModel => crate::sqlmodel::render_entities(schema), Orm::Jpa => crate::jpa::render_entities(schema).map(|entities| entities.join("\n")), + Orm::Prisma => crate::prisma::export(schema), } } @@ -29,6 +30,7 @@ macro_rules! orm_cases { #[case::sqlalchemy(Orm::SqlAlchemy)] #[case::sqlmodel(Orm::SqlModel)] #[case::jpa(Orm::Jpa)] + #[case::prisma(Orm::Prisma)] fn $test_name(#[case] orm: Orm) { let table = $fixture(); let rendered = render_entity(orm, &table).unwrap(); @@ -45,6 +47,7 @@ macro_rules! orm_cases { #[case::sqlalchemy(Orm::SqlAlchemy)] #[case::sqlmodel(Orm::SqlModel)] #[case::jpa(Orm::Jpa)] + #[case::prisma(Orm::Prisma)] fn $test_name(#[case] orm: Orm) { let schema: Vec = $fixture(); let rendered = render_schema(orm, &schema).unwrap(); @@ -299,6 +302,7 @@ fn to_pascal_case_for(orm: Orm, s: &str) -> String { Orm::SqlAlchemy => crate::sqlalchemy::to_pascal_case_for_tests(s), Orm::SqlModel => crate::sqlmodel::to_pascal_case_for_tests(s), Orm::Jpa => crate::jpa::to_pascal_case_for_tests(s), + Orm::Prisma => crate::prisma::to_pascal_case_for_tests(s), } } @@ -318,6 +322,7 @@ fn to_pascal_case_for(orm: Orm, s: &str) -> String { #[case::sqlalchemy(Orm::SqlAlchemy)] #[case::sqlmodel(Orm::SqlModel)] #[case::jpa(Orm::Jpa)] +#[case::prisma(Orm::Prisma)] fn to_pascal_case_shared_semantics( #[values( ("", ""), @@ -344,6 +349,7 @@ fn to_pascal_case_shared_semantics( #[case::sqlalchemy(Orm::SqlAlchemy)] #[case::sqlmodel(Orm::SqlModel)] #[case::jpa(Orm::Jpa)] +#[case::prisma(Orm::Prisma)] fn render_entity_with_schema_snapshots( #[values( "many_to_many_article", diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__all_simple_types_snapshot@all_simple_types_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__all_simple_types_snapshot@all_simple_types_Prisma.snap new file mode 100644 index 00000000..d5f217f0 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__all_simple_types_snapshot@all_simple_types_Prisma.snap @@ -0,0 +1,28 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 138 +expression: rendered +--- +model AllTypes { + id Int @id + small Int + big BigInt + real_num Float + double_num Float + text_col String + bool_col Boolean + date_col DateTime + time_col DateTime + ts_col DateTime + tstz_col DateTime + interval_col String + bytea_col Bytes + uuid_col String // stored as binary(16) on MySQL backends + json_col Json + inet_col String + cidr_col String + macaddr_col String + xml_col String + + @@map("all_types") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__basic_single_pk_snapshot@basic_single_pk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__basic_single_pk_snapshot@basic_single_pk_Prisma.snap new file mode 100644 index 00000000..11514319 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__basic_single_pk_snapshot@basic_single_pk_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 66 +expression: rendered +--- +model Users { + id Int @id + display_name String? + + @@map("users") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__basic_table_with_description_snapshot@basic_table_with_description_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__basic_table_with_description_snapshot@basic_table_with_description_Prisma.snap new file mode 100644 index 00000000..132723fe --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__basic_table_with_description_snapshot@basic_table_with_description_Prisma.snap @@ -0,0 +1,15 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 61 +expression: rendered +--- +/// User accounts table +model Users { + /// Primary key + id Int @id @default(autoincrement()) + /// User email address + email String @unique + name String? + + @@map("users") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__complex_types_snapshot@complex_types_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__complex_types_snapshot@complex_types_Prisma.snap new file mode 100644 index 00000000..1d5d1cc8 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__complex_types_snapshot@complex_types_Prisma.snap @@ -0,0 +1,14 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 143 +expression: rendered +--- +model ComplexTypes { + id Int @id + varchar_col String + char_col String + numeric_col Decimal + custom_col Unsupported("CUSTOM_TYPE") + + @@map("complex_types") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_constraints_snapshot@composite_constraints_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_constraints_snapshot@composite_constraints_Prisma.snap new file mode 100644 index 00000000..94c7016a --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_constraints_snapshot@composite_constraints_Prisma.snap @@ -0,0 +1,17 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 172 +expression: rendered +--- +model OrderItems { + order_id Int + order Orders @relation(fields: [order_id], references: [id]) + product_id Int + product Products @relation(fields: [product_id], references: [id]) + quantity Int + + @@id([order_id, product_id]) + @@unique([order_id, product_id], map: "uq_order_items__order_product") + @@index([order_id], map: "ix_order_items__order_id") + @@map("order_items") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_index_snapshot@composite_index_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_index_snapshot@composite_index_Prisma.snap new file mode 100644 index 00000000..f579d991 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_index_snapshot@composite_index_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 179 +expression: rendered +--- +model CompositeIndex { + id Int @id + tenant_id Int + name String + + @@index([tenant_id, name], map: "idx_tenant_name") + @@map("composite_index") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_pk_snapshot@composite_pk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_pk_snapshot@composite_pk_Prisma.snap new file mode 100644 index 00000000..94e0e483 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_pk_snapshot@composite_pk_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Accounts { + id Int + tenant_id BigInt + + @@id([id, tenant_id]) + @@map("accounts") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_primary_key_snapshot@composite_primary_key_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_primary_key_snapshot@composite_primary_key_Prisma.snap new file mode 100644 index 00000000..7defd856 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_primary_key_snapshot@composite_primary_key_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 244 +expression: rendered +--- +model Membership { + tenant_id Int + user_id Int + role String + + @@id([tenant_id, user_id]) + @@map("membership") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_unique_constraint_snapshot@composite_unique_constraint_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_unique_constraint_snapshot@composite_unique_constraint_Prisma.snap new file mode 100644 index 00000000..3b529783 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_unique_constraint_snapshot@composite_unique_constraint_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 249 +expression: rendered +--- +model AccountAliases { + id Int @id + tenant_id Int + slug String + + @@unique([tenant_id, slug], map: "uq_account_aliases__tenant_slug") + @@map("account_aliases") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_unique_snapshot@composite_unique_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_unique_snapshot@composite_unique_Prisma.snap new file mode 100644 index 00000000..11c04b69 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__composite_unique_snapshot@composite_unique_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 174 +expression: rendered +--- +model CompositeUnique { + id Int @id + tenant_id Int + name String + + @@unique([tenant_id, name], map: "uq_tenant_name") + @@map("composite_unique") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__defaults_snapshot@defaults_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__defaults_snapshot@defaults_Prisma.snap new file mode 100644 index 00000000..33cc1046 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__defaults_snapshot@defaults_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 153 +expression: rendered +--- +model Articles { + id Int @id @default(autoincrement()) + published Boolean @default(false) + view_count Int @default(0) + status String @default("draft") + + @@map("articles") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_multiple_columns_snapshot@enum_multiple_columns_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_multiple_columns_snapshot@enum_multiple_columns_Prisma.snap new file mode 100644 index 00000000..85a18060 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_multiple_columns_snapshot@enum_multiple_columns_Prisma.snap @@ -0,0 +1,23 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum ProductCategory { + ELECTRONICS @map("electronics") + CLOTHING @map("clothing") + FOOD @map("food") +} + +enum AvailabilityStatus { + IN_STOCK @map("in_stock") + OUT_OF_STOCK @map("out_of_stock") + PRE_ORDER @map("pre_order") +} + +model Products { + id Int + category ProductCategory + availability AvailabilityStatus + + @@map("products") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_shared_snapshot@enum_shared_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_shared_snapshot@enum_shared_Prisma.snap new file mode 100644 index 00000000..86f0234f --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_shared_snapshot@enum_shared_Prisma.snap @@ -0,0 +1,17 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum DocStatus { + DRAFT @map("draft") + PUBLISHED @map("published") + ARCHIVED @map("archived") +} + +model Documents { + id Int + status DocStatus + review_status DocStatus? + + @@map("documents") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_special_values_snapshot@enum_special_values_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_special_values_snapshot@enum_special_values_Prisma.snap new file mode 100644 index 00000000..1123cf65 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_special_values_snapshot@enum_special_values_Prisma.snap @@ -0,0 +1,17 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum EventSeverity { + INFO_LEVEL @map("info-level") + WARNING_LEVEL @map("warning_level") + ERROR_LEVEL + _1CRITICAL @map("1critical") +} + +model Events { + id Int + severity EventSeverity + + @@map("events") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_with_default_snapshot@enum_with_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_with_default_snapshot@enum_with_default_Prisma.snap new file mode 100644 index 00000000..29cd4ceb --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__enum_with_default_snapshot@enum_with_default_Prisma.snap @@ -0,0 +1,18 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum TaskStatus { + PENDING @map("pending") + IN_PROGRESS @map("in_progress") + COMPLETED @map("completed") +} + +model Tasks { + id Int + status TaskStatus @default(PENDING) + priority Int @default(0) + is_archived Boolean @default(false) + + @@map("tasks") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__false_boolean_default_snapshot@false_boolean_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__false_boolean_default_snapshot@false_boolean_default_Prisma.snap new file mode 100644 index 00000000..a738b4e9 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__false_boolean_default_snapshot@false_boolean_default_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model BoolDefaults { + id Int @id + is_deleted Boolean @default(false) + + @@map("bool_defaults") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__fk_with_comment_and_auto_increment_snapshot@fk_with_comment_and_auto_increment_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__fk_with_comment_and_auto_increment_snapshot@fk_with_comment_and_auto_increment_Prisma.snap new file mode 100644 index 00000000..c7b043c2 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__fk_with_comment_and_auto_increment_snapshot@fk_with_comment_and_auto_increment_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 224 +expression: rendered +--- +model Child { + /// References parent table + parent_id Int @id @default(autoincrement()) + parent Parent @relation(fields: [parent_id], references: [id]) + value String + + @@map("child") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__inline_pk_snapshot@inline_pk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__inline_pk_snapshot@inline_pk_Prisma.snap new file mode 100644 index 00000000..367e721a --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__inline_pk_snapshot@inline_pk_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 86 +expression: rendered +--- +model Users { + id String @default(uuid()) // stored as binary(16) on MySQL backends + email String + + @@map("users") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_all_variant_types_snapshot@integer_enum_all_variant_types_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_all_variant_types_snapshot@integer_enum_all_variant_types_Prisma.snap new file mode 100644 index 00000000..ff867cfc --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_all_variant_types_snapshot@integer_enum_all_variant_types_Prisma.snap @@ -0,0 +1,17 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum EdgeState { + UNKNOWN // = -1 + NOT_STARTED // = 0 + IN_PROGRESS // = 10 + HTTP_500 // = 500 +} + +model WorkflowRuns { + id Int @id + state EdgeState + + @@map("workflow_runs") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_with_default_snapshot@integer_enum_with_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_with_default_snapshot@integer_enum_with_default_Prisma.snap new file mode 100644 index 00000000..dec9f410 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_with_default_snapshot@integer_enum_with_default_Prisma.snap @@ -0,0 +1,15 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum TaskStatus { + PENDING // = 0 + COMPLETED // = 100 +} + +model Tasks { + id Int + status TaskStatus @default(dbgenerated("1")) + + @@map("tasks") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_with_variant_default_snapshot@integer_enum_with_variant_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_with_variant_default_snapshot@integer_enum_with_variant_default_Prisma.snap new file mode 100644 index 00000000..8a245154 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__integer_enum_with_variant_default_snapshot@integer_enum_with_variant_default_Prisma.snap @@ -0,0 +1,15 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum TaskRunStatus { + PENDING // = 0 + COMPLETED // = 100 +} + +model TaskRuns { + id Int + status TaskRunStatus @default(COMPLETED) + + @@map("task_runs") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__json_default_snapshot@json_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__json_default_snapshot@json_default_Prisma.snap new file mode 100644 index 00000000..bb69f71f --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__json_default_snapshot@json_default_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Configs { + id Int @id + data Json @default(dbgenerated("{\"hello\": \"world\"}")) + + @@map("configs") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__jsonb_custom_type_snapshot@jsonb_custom_type_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__jsonb_custom_type_snapshot@jsonb_custom_type_Prisma.snap new file mode 100644 index 00000000..2b61c89d --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__jsonb_custom_type_snapshot@jsonb_custom_type_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model JsonStruct { + id Int + json_data Json + jsonb_data Unsupported("JSONB") + jsonb_nullable Unsupported("jsonb")? + + @@map("json_struct") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__no_description_snapshot@no_description_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__no_description_snapshot@no_description_Prisma.snap new file mode 100644 index 00000000..63eb7850 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__no_description_snapshot@no_description_Prisma.snap @@ -0,0 +1,9 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model NoDesc { + id Int @id + + @@map("no_desc") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__nullable_columns_snapshot@nullable_columns_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__nullable_columns_snapshot@nullable_columns_Prisma.snap new file mode 100644 index 00000000..a1ec5b80 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__nullable_columns_snapshot@nullable_columns_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 164 +expression: rendered +--- +model Profiles { + id Int @id @default(autoincrement()) + bio String? + avatar_url String? + + @@map("profiles") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__nullable_enum_snapshot@nullable_enum_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__nullable_enum_snapshot@nullable_enum_Prisma.snap new file mode 100644 index 00000000..6600f6ae --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__nullable_enum_snapshot@nullable_enum_Prisma.snap @@ -0,0 +1,15 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum StatusType { + ACTIVE @map("active") + INACTIVE @map("inactive") +} + +model NullableEnum { + id Int @id + status StatusType? + + @@map("nullable_enum") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__numeric_default_value_snapshot@numeric_default_value_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__numeric_default_value_snapshot@numeric_default_value_Prisma.snap new file mode 100644 index 00000000..f28edde8 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__numeric_default_value_snapshot@numeric_default_value_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 259 +expression: rendered +--- +model Products { + id Int + price Decimal @default(0.00) + + @@map("products") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__pk_and_fk_together_snapshot@pk_and_fk_together_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__pk_and_fk_together_snapshot@pk_and_fk_together_Prisma.snap new file mode 100644 index 00000000..ac9b83a3 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__pk_and_fk_together_snapshot@pk_and_fk_together_Prisma.snap @@ -0,0 +1,20 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 87 +expression: rendered +--- +model ArticleUser { + article_id String // stored as binary(16) on MySQL backends + article Article @relation(fields: [article_id], references: [id], onDelete: Cascade) + user_id String // stored as binary(16) on MySQL backends + user User @relation(fields: [user_id], references: [id], onDelete: Cascade) + author_order Int @default(1) + role String @default("contributor") + is_lead Boolean @default(false) + created_at DateTime @default(now()) + + @@id([article_id, user_id]) + @@index([article_id]) + @@index([user_id]) + @@map("article_user") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@composite_fk_parent_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@composite_fk_parent_Prisma.snap new file mode 100644 index 00000000..d6e65fa3 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@composite_fk_parent_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Parent { + id1 Int + id2 Int + + @@id([id1, id2]) + @@map("parent") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@dual_reverse_relations_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@dual_reverse_relations_Prisma.snap new file mode 100644 index 00000000..2adcc9ab --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@dual_reverse_relations_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model Dual { + username String @id + username_dual_rel DualRel[] @relation("DualRelUsername") + checker_username_dual_rel DualRel[] @relation("DualRelCheckerUsername") + + @@map("dual") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_article_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_article_Prisma.snap new file mode 100644 index 00000000..5fc9573f --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_article_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Article { + id BigInt @id + article_user ArticleUser[] + + @@map("article") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_missing_target_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_missing_target_Prisma.snap new file mode 100644 index 00000000..5fc9573f --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_missing_target_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Article { + id BigInt @id + article_user ArticleUser[] + + @@map("article") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_multiple_junctions_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_multiple_junctions_Prisma.snap new file mode 100644 index 00000000..c381d22d --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_multiple_junctions_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model User { + id String @id // stored as binary(16) on MySQL backends + user_media_role UserMediaRole[] + user_media_favorite UserMediaFavorite[] + + @@map("user") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_user_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_user_Prisma.snap new file mode 100644 index 00000000..ad510a77 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@many_to_many_user_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model User { + id String @id // stored as binary(16) on MySQL backends + article_user ArticleUser[] + + @@map("user") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_fk_same_table_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_fk_same_table_Prisma.snap new file mode 100644 index 00000000..dfdbf79b --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_fk_same_table_Prisma.snap @@ -0,0 +1,14 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model Post { + id String @id // stored as binary(16) on MySQL backends + creator_user_id String // stored as binary(16) on MySQL backends + creator_user User @relation("PostCreatorUser", fields: [creator_user_id], references: [id]) + used_by_user_id String // stored as binary(16) on MySQL backends + used_by_user User @relation("PostUsedByUser", fields: [used_by_user_id], references: [id]) + + @@map("post") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_has_one_relations_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_has_one_relations_Prisma.snap new file mode 100644 index 00000000..c038ac0e --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_has_one_relations_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model User { + id String @id // stored as binary(16) on MySQL backends + created_by_user_settings Settings? @relation("SettingsCreatedByUser") + updated_by_user_settings Settings? @relation("SettingsUpdatedByUser") + + @@map("user") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_reverse_relations_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_reverse_relations_Prisma.snap new file mode 100644 index 00000000..1f3dee9b --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@multiple_reverse_relations_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model User { + id String @id // stored as binary(16) on MySQL backends + preferred_user_profile Profile[] @relation("ProfilePreferredUser") + backup_user_profile Profile[] @relation("ProfileBackupUser") + + @@map("user") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_fk_not_in_pk_another_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_fk_not_in_pk_another_Prisma.snap new file mode 100644 index 00000000..849af190 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_fk_not_in_pk_another_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Another { + id Int @id + not_junction NotJunction[] + + @@map("another") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_fk_not_in_pk_other_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_fk_not_in_pk_other_Prisma.snap new file mode 100644 index 00000000..ae8747af --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_fk_not_in_pk_other_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Other { + id Int @id + not_junction NotJunction[] + + @@map("other") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_single_pk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_single_pk_Prisma.snap new file mode 100644 index 00000000..68f8850b --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@not_junction_single_pk_Prisma.snap @@ -0,0 +1,10 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Other { + id Int @id + regular Regular[] + + @@map("other") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@triple_reverse_relations_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@triple_reverse_relations_Prisma.snap new file mode 100644 index 00000000..10832fe0 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@triple_reverse_relations_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model Dual { + username String @id + username_triple_rel TripleRel[] @relation("TripleRelUsername") + checker_username_triple_rel TripleRel[] @relation("TripleRelCheckerUsername") + other_username_triple_rel TripleRel[] @relation("TripleRelOtherUsername") + + @@map("dual") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@username_fk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@username_fk_Prisma.snap new file mode 100644 index 00000000..2b4b4f10 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__render_entity_with_schema_snapshots@username_fk_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 376 +expression: rendered +--- +model Session { + id String @id // stored as binary(16) on MySQL backends + username String + username_rel User @relation(fields: [username], references: [username]) + + @@map("session") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__reserved_word_identifiers_snapshot@reserved_word_identifiers_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__reserved_word_identifiers_snapshot@reserved_word_identifiers_Prisma.snap new file mode 100644 index 00000000..fa9c2aaf --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__reserved_word_identifiers_snapshot@reserved_word_identifiers_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 239 +expression: rendered +--- +model Order { + id Int @id + user String + select Int + + @@map("order") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__self_referencing_fk_snapshot@self_referencing_fk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__self_referencing_fk_snapshot@self_referencing_fk_Prisma.snap new file mode 100644 index 00000000..ebb58630 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__self_referencing_fk_snapshot@self_referencing_fk_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model Employees { + id Int @id + manager_id Int? + manager Employees? @relation("EmployeesManager", fields: [manager_id], references: [id], onDelete: SetNull) + manager_employees Employees[] @relation("EmployeesManager") + + @@map("employees") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__server_default_and_true_boolean_snapshot@server_default_and_true_boolean_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__server_default_and_true_boolean_snapshot@server_default_and_true_boolean_Prisma.snap new file mode 100644 index 00000000..8163a502 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__server_default_and_true_boolean_snapshot@server_default_and_true_boolean_Prisma.snap @@ -0,0 +1,14 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 159 +expression: rendered +--- +model Logs { + id Int @id @default(autoincrement()) + active Boolean @default(true) + created_at DateTime @default(now()) + score Float @default(1.5) + tag String @default(dbgenerated("UNKNOWN_EXPR")) + + @@map("logs") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__server_defaults_snapshot@server_defaults_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__server_defaults_snapshot@server_defaults_Prisma.snap new file mode 100644 index 00000000..f62f1246 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__server_defaults_snapshot@server_defaults_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 154 +expression: rendered +--- +model WithDefaults { + id Int @id + created_at DateTime @default(now()) + status String @default("active") + count Int @default(0) + + @@map("with_defaults") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__small_multi_schema_sequential_snapshot@small_multi_schema_sequential_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__small_multi_schema_sequential_snapshot@small_multi_schema_sequential_Prisma.snap new file mode 100644 index 00000000..899faae3 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__small_multi_schema_sequential_snapshot@small_multi_schema_sequential_Prisma.snap @@ -0,0 +1,21 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 290 +expression: rendered +--- +model Users { + id Int @id + display_name String? + posts Posts[] + + @@map("users") +} + +model Posts { + id Int @id + user_id Int + user Users @relation(fields: [user_id], references: [id]) + title String + + @@map("posts") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__string_default_snapshot@string_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__string_default_snapshot@string_default_Prisma.snap new file mode 100644 index 00000000..61eabf4a --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__string_default_snapshot@string_default_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 204 +expression: rendered +--- +model StringDefaults { + id Int @id + status String @default("active") + + @@map("string_defaults") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_level_pk_snapshot@table_level_pk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_level_pk_snapshot@table_level_pk_Prisma.snap new file mode 100644 index 00000000..750ae86f --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_level_pk_snapshot@table_level_pk_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 133 +expression: rendered +--- +model Orders { + id String @id // stored as binary(16) on MySQL backends + customer_id String // stored as binary(16) on MySQL backends + total Float + + @@map("orders") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_composite_fk_snapshot@table_with_composite_fk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_composite_fk_snapshot@table_with_composite_fk_Prisma.snap new file mode 100644 index 00000000..ffaee027 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_composite_fk_snapshot@table_with_composite_fk_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 81 +expression: rendered +--- +model LineItems { + id Int @id + order_id Int + order_version Int + sku String + + @@map("line_items") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_enum_snapshot@table_with_enum_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_enum_snapshot@table_with_enum_Prisma.snap new file mode 100644 index 00000000..c069ed49 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_enum_snapshot@table_with_enum_Prisma.snap @@ -0,0 +1,16 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum OrderStatus { + PENDING @map("pending") + SHIPPED @map("shipped") + DELIVERED @map("delivered") +} + +model Orders { + id Int + status OrderStatus + + @@map("orders") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_fk_snapshot@table_with_fk_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_fk_snapshot@table_with_fk_Prisma.snap new file mode 100644 index 00000000..4b8bfc56 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_fk_snapshot@table_with_fk_Prisma.snap @@ -0,0 +1,13 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 76 +expression: rendered +--- +model Posts { + id Int @id + user_id Int + user Users @relation(fields: [user_id], references: [id]) + title String + + @@map("posts") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_indexes_snapshot@table_with_indexes_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_indexes_snapshot@table_with_indexes_Prisma.snap new file mode 100644 index 00000000..1d8c6af7 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_indexes_snapshot@table_with_indexes_Prisma.snap @@ -0,0 +1,14 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 128 +expression: rendered +--- +model Articles { + id Int @id + title String + created_at DateTime + + @@index([created_at], map: "idx_articles_created_at") + @@index([title]) + @@map("articles") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_integer_enum_snapshot@table_with_integer_enum_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_integer_enum_snapshot@table_with_integer_enum_Prisma.snap new file mode 100644 index 00000000..367e1ee1 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__table_with_integer_enum_snapshot@table_with_integer_enum_Prisma.snap @@ -0,0 +1,16 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +enum PriorityLevel { + LOW // = 0 + MEDIUM // = 10 + HIGH // = 20 +} + +model Tasks { + id Int @id + priority PriorityLevel + + @@map("tasks") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unique_and_indexed_snapshot@unique_and_indexed_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unique_and_indexed_snapshot@unique_and_indexed_Prisma.snap new file mode 100644 index 00000000..25520b55 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unique_and_indexed_snapshot@unique_and_indexed_Prisma.snap @@ -0,0 +1,15 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 123 +expression: rendered +--- +model Users { + id Int + email String @unique + username String @unique(map: "uq_username") + department String? + status String @default("active") + + @@index([department], map: "idx_department") + @@map("users") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unknown_constant_default_snapshot@unknown_constant_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unknown_constant_default_snapshot@unknown_constant_default_Prisma.snap new file mode 100644 index 00000000..507e34de --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unknown_constant_default_snapshot@unknown_constant_default_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 219 +expression: rendered +--- +model UnknownDefault { + id Int @id + value String @default(dbgenerated("SOME_CONSTANT")) + + @@map("unknown_default") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unknown_function_default_snapshot@unknown_function_default_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unknown_function_default_snapshot@unknown_function_default_Prisma.snap new file mode 100644 index 00000000..02377a87 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unknown_function_default_snapshot@unknown_function_default_Prisma.snap @@ -0,0 +1,11 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 214 +expression: rendered +--- +model UnknownDefaults { + id Int @id + code String @default(dbgenerated("gen_code()")) + + @@map("unknown_defaults") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_composite_index_snapshot@unnamed_composite_index_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_composite_index_snapshot@unnamed_composite_index_Prisma.snap new file mode 100644 index 00000000..31120619 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_composite_index_snapshot@unnamed_composite_index_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model UnnamedIndex { + id Int @id + col_a Int + col_b Int + + @@index([col_a, col_b]) + @@map("unnamed_index") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_composite_unique_snapshot@unnamed_composite_unique_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_composite_unique_snapshot@unnamed_composite_unique_Prisma.snap new file mode 100644 index 00000000..895a9617 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_composite_unique_snapshot@unnamed_composite_unique_Prisma.snap @@ -0,0 +1,12 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +expression: rendered +--- +model UnnamedUnique { + id Int @id + col_a Int + col_b Int + + @@unique([col_a, col_b]) + @@map("unnamed_unique") +} diff --git a/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_index_and_unique_snapshot@unnamed_index_and_unique_Prisma.snap b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_index_and_unique_snapshot@unnamed_index_and_unique_Prisma.snap new file mode 100644 index 00000000..0f4f1229 --- /dev/null +++ b/crates/vespertide-exporter/src/tests/snapshots/vespertide_exporter__tests__unnamed_index_and_unique_snapshot@unnamed_index_and_unique_Prisma.snap @@ -0,0 +1,14 @@ +--- +source: crates/vespertide-exporter/src/tests/mod.rs +assertion_line: 184 +expression: rendered +--- +model Events { + id Int @id + venue_id Int + date DateTime + + @@unique([venue_id, date]) + @@index([venue_id, date]) + @@map("events") +} diff --git a/crates/vespertide-exporter/tests/parallel_consolidated.rs b/crates/vespertide-exporter/tests/parallel_consolidated.rs index f93674b4..423b5486 100644 --- a/crates/vespertide-exporter/tests/parallel_consolidated.rs +++ b/crates/vespertide-exporter/tests/parallel_consolidated.rs @@ -37,6 +37,7 @@ fn render_schema(orm: Orm, schema: &[TableDef]) -> Result { Orm::Jpa => { vespertide_exporter::jpa::render_entities(schema).map(|entities| entities.join("\n")) } + Orm::Prisma => vespertide_exporter::prisma::export(schema), } }