Add new command to cli

Author: owjs3901

Cargo.lock

@@ -8,6 +8,7 @@ publish = false
 anyhow = "1"
 clap = { version = "4", features = ["derive"] }
 serde_json = "1"
+serde_yaml = "0.9"
 vespertide-config = { path = "../vespertide-config" }
 vespertide-core = { path = "../vespertide-core" }
 vespertide-planner = { path = "../vespertide-planner" }

crates/vespertide-cli/Cargo.toml

@@ -1,12 +1,14 @@
 pub mod diff;
 pub mod init;
+pub mod new;
 pub mod log;
 pub mod sql;
 pub mod revision;
 pub mod status;
 
 pub use diff::cmd_diff;
 pub use init::cmd_init;
+pub use new::cmd_new;
 pub use log::cmd_log;
 pub use sql::cmd_sql;
 pub use revision::cmd_revision;

crates/vespertide-cli/src/commands/mod.rs

@@ -0,0 +1,60 @@
+use std::fs;
+
+use anyhow::{Context, Result, bail};
+use serde_json::Value;
+use vespertide_core::TableDef;
+
+use crate::{ModelFormat, utils::load_config};
+
+pub fn cmd_new(name: String, format: ModelFormat) -> Result<()> {
+    let config = load_config()?;
+    let dir = config.models_dir();
+    if !dir.exists() {
+        fs::create_dir_all(dir).context("create models directory")?;
+    }
+
+    let ext = match format {
+        ModelFormat::Json => "json",
+        ModelFormat::Yaml => "yaml",
+        ModelFormat::Yml => "yml",
+    };
+
+    let path = dir.join(format!("{name}.{ext}"));
+    if path.exists() {
+        bail!("model file already exists: {}", path.display());
+    }
+
+    let table = TableDef {
+        name: name.clone(),
+        columns: Vec::new(),
+        constraints: Vec::new(),
+        indexes: Vec::new(),
+    };
+
+    match format {
+        ModelFormat::Json => write_json_with_schema(&path, &table)?,
+        ModelFormat::Yaml | ModelFormat::Yml => write_yaml(&path, &table)?,
+    }
+
+    println!("Created model template: {}", path.display());
+    Ok(())
+}
+
+fn write_json_with_schema(path: &std::path::Path, table: &TableDef) -> Result<()> {
+    let mut value = serde_json::to_value(table).context("serialize table to json")?;
+    if let Value::Object(ref mut map) = value {
+        map.insert(
+            "$schema".to_string(),
+            Value::String("https://json-schema.org/draft/2020-12/schema".to_string()),
+        );
+    }
+    let text = serde_json::to_string_pretty(&value).context("stringify json with schema")?;
+    fs::write(path, text).with_context(|| format!("write file: {}", path.display()))?;
+    Ok(())
+}
+
+fn write_yaml(path: &std::path::Path, table: &TableDef) -> Result<()> {
+    let text = serde_yaml::to_string(table).context("serialize table to yaml")?;
+    fs::write(path, text).with_context(|| format!("write file: {}", path.display()))?;
+    Ok(())
+}

crates/vespertide-cli/src/commands/new.rs

@@ -3,7 +3,8 @@ use clap::{Parser, Subcommand};
 
 mod commands;
 mod utils;
-use commands::{cmd_diff, cmd_init, cmd_log, cmd_revision, cmd_sql, cmd_status};
+use commands::{cmd_diff, cmd_init, cmd_log, cmd_new, cmd_revision, cmd_sql, cmd_status};
+use clap::ValueEnum;
 
 /// vespertide command-line interface.
 #[derive(Parser, Debug)]
@@ -21,6 +22,14 @@ enum Commands {
     Sql,
     /// Show SQL per applied migration (chronological log).
     Log,
+    /// Create a new model file from template.
+    New {
+        /// Model name (table name).
+        name: String,
+        /// Output format: json|yaml|yml (default: json).
+        #[arg(short = 'f', long = "format", default_value = "json", value_enum)]
+        format: ModelFormat,
+    },
     /// Show current status.
     Status,
     /// Create a new revision with a message.
@@ -32,12 +41,20 @@ enum Commands {
     Init,
 }
 
+#[derive(Copy, Clone, Debug, ValueEnum)]
+pub enum ModelFormat {
+    Json,
+    Yaml,
+    Yml,
+}
+
 fn main() -> Result<()> {
     let cli = Cli::parse();
     match cli.command {
         Commands::Diff => cmd_diff(),
         Commands::Sql => cmd_sql(),
         Commands::Log => cmd_log(),
+        Commands::New { name, format } => cmd_new(name, format),
         Commands::Status => cmd_status(),
         Commands::Revision { message } => cmd_revision(message),
         Commands::Init => cmd_init(),