feat!: Add support for rolling back migrations

cot-cli/Cargo.toml

@@ -20,7 +20,6 @@ path = "src/main.rs"
 workspace = true
 
 [dependencies]
-anstyle.workspace = true
 anyhow.workspace = true
 cargo_toml.workspace = true
 chrono.workspace = true

cot-cli/src/migration_generator.rs

@@ -7,6 +7,7 @@ use std::path::{Path, PathBuf};
 
 use anyhow::{Context, bail};
 use cot::db::migrations::{DynMigration, MigrationEngine};
+use cot::utils::cli::{StatusType, print_status_msg};
 use cot_codegen::model::{Field, Model, ModelArgs, ModelOpts, ModelType};
 use cot_codegen::symbol_resolver::SymbolResolver;
 use darling::FromMeta;
@@ -18,7 +19,7 @@ use quote::{ToTokens, format_ident, quote};
 use syn::{Meta, parse_quote};
 use tracing::{debug, trace};
 
-use crate::utils::{CargoTomlManager, PackageManager, StatusType, print_status_msg};
+use crate::utils::{CargoTomlManager, PackageManager};
 
 pub fn make_migrations(path: &Path, options: MigrationGeneratorOptions) -> anyhow::Result<()> {
     let Some(manager) = CargoTomlManager::from_path(path)? else {

cot-cli/src/new_project.rs

@@ -1,12 +1,11 @@
 use std::path::Path;
 
+use cot::utils::cli::{StatusType, print_status_msg};
 use heck::ToPascalCase;
 use rand::rngs::{StdRng, SysRng};
 use rand::{Rng, SeedableRng};
 use tracing::trace;
 
-use crate::utils::{StatusType, print_status_msg};
-
 macro_rules! project_file {
     ($name:literal) => {
         ($name, include_str!(concat!("project_template/", $name)))

cot-cli/src/utils.rs

@@ -1,78 +1,9 @@
 use std::collections::HashMap;
 use std::path::{Path, PathBuf};
 
-use anstyle::{AnsiColor, Color, Effects, Style};
 use anyhow::{Context, bail};
 use cargo_toml::Manifest;
 
-pub(crate) fn print_status_msg(status: StatusType, message: &str) {
-    let style = status.style();
-    let status_str = status.as_str();
-
-    eprintln!("{style}{status_str:>12}{style:#} {message}");
-}
-
-#[derive(Debug, Clone, Copy)]
-pub(crate) enum StatusType {
-    // In-Progress Ops
-    Creating,
-    Adding,
-    Modifying,
-    Removing,
-    // Completed Ops
-    Created,
-    Added,
-    Modified,
-    Removed,
-
-    // Status types
-    #[expect(dead_code)]
-    Error, // Should be used in Error handling inside remove operations
-    #[expect(dead_code)]
-    Warning, // Should be used as cautionary messages.
-    Notice,
-}
-
-impl StatusType {
-    fn style(self) -> Style {
-        let base_style = Style::new() | Effects::BOLD;
-
-        match self {
-            // In-Progress => Brighter colors
-            StatusType::Creating => base_style.fg_color(Some(Color::Ansi(AnsiColor::BrightGreen))),
-            StatusType::Adding => base_style.fg_color(Some(Color::Ansi(AnsiColor::BrightCyan))),
-            StatusType::Removing => {
-                base_style.fg_color(Some(Color::Ansi(AnsiColor::BrightMagenta)))
-            }
-            StatusType::Modifying => base_style.fg_color(Some(Color::Ansi(AnsiColor::BrightBlue))),
-            // Completed => Dimmed colors
-            StatusType::Created => base_style.fg_color(Some(Color::Ansi(AnsiColor::Green))),
-            StatusType::Added => base_style.fg_color(Some(Color::Ansi(AnsiColor::Cyan))),
-            StatusType::Removed => base_style.fg_color(Some(Color::Ansi(AnsiColor::Magenta))),
-            StatusType::Modified => base_style.fg_color(Some(Color::Ansi(AnsiColor::Blue))),
-            // Status types
-            StatusType::Warning => base_style.fg_color(Some(Color::Ansi(AnsiColor::Yellow))),
-            StatusType::Error => base_style.fg_color(Some(Color::Ansi(AnsiColor::Red))),
-            StatusType::Notice => base_style.fg_color(Some(Color::Ansi(AnsiColor::White))),
-        }
-    }
-    fn as_str(self) -> &'static str {
-        match self {
-            StatusType::Creating => "Creating",
-            StatusType::Adding => "Adding",
-            StatusType::Modifying => "Modifying",
-            StatusType::Removing => "Removing",
-            StatusType::Created => "Created",
-            StatusType::Added => "Added",
-            StatusType::Modified => "Modified",
-            StatusType::Removed => "Removed",
-            StatusType::Warning => "Warning",
-            StatusType::Error => "Error",
-            StatusType::Notice => "Notice",
-        }
-    }
-}
-
 #[derive(Debug)]
 pub(crate) enum CargoTomlManager {
     Workspace(WorkspaceManager),

cot/Cargo.toml

@@ -17,14 +17,15 @@ workspace = true
 
 [dependencies]
 aide = { workspace = true, optional = true }
+anstyle.workspace = true
 askama = { workspace = true, features = ["std"] }
 async-trait.workspace = true
 axum = { workspace = true, features = ["http1", "tokio"] }
 blake3.workspace = true
 bytes.workspace = true
 chrono = { workspace = true, features = ["alloc", "serde", "clock"] }
 chrono-tz.workspace = true
-clap.workspace = true
+clap = {workspace = true, features = ["string"] }
 cot_core.workspace = true
 cot_macros.workspace = true
 deadpool-redis = { workspace = true, features = ["tokio-comp", "rt_tokio_1"], optional = true }
@@ -72,6 +73,7 @@ url = { workspace = true, features = ["serde"] }
 criterion = { workspace = true, features = ["async_tokio"] }
 fake.workspace = true
 fantoccini.workspace = true
+insta.workspace = true
 mockall.workspace = true
 reqwest = { workspace = true, features = ["json"] }
 rustversion.workspace = true