backbone-catalog 0.7.0

//! Database seeder binary for module: catalog
//!
//! Generated by metaphor-schema. Do not edit manually.
//!
//! Run with: cargo run --bin catalog-seeder
//! Or via CLI: backbone seed run --module catalog
//!
//! This binary runs all registered seeders in order.

use anyhow::Result;
use sqlx::postgres::PgPoolOptions;
use std::env;

// Import seeders
use backbone_catalog::seeders::SeedAttributeSeeder;
use backbone_catalog::seeders::SeedAttributeValueSeeder;
use backbone_catalog::seeders::SeedBrandSeeder;
use backbone_catalog::seeders::SeedItemSeeder;
use backbone_catalog::seeders::SeedItemGroupSeeder;
use backbone_catalog::seeders::SeedItemVariantSeeder;
use backbone_catalog::seeders::SeedUomSeeder;
use backbone_catalog::seeders::SeedUomConversionSeeder;
use backbone_catalog::seeders::Seeder;

#[tokio::main]
async fn main() -> Result<()> {
    let args: Vec<String> = env::args().collect();
    let force = args.iter().any(|a| a == "--force");
    let filter: Option<&str> = args.iter()
        .skip(1)
        .find(|a| !a.starts_with("-"))
        .map(|s| s.as_str());

    let database_url = env::var("DATABASE_URL")
        .map_err(|e| anyhow::anyhow!("DATABASE_URL must be set: {e}"))?;

    println!("Connecting to database...");

    let pool = PgPoolOptions::new()
        .max_connections(5)
        .connect(&database_url)
        .await?;

    println!("Connected to database");
    println!();
    println!("Running seeders for module: catalog");
    println!();

    // Register seeders in order
    let mut seeders: Vec<Box<dyn Seeder + Send + Sync>> = Vec::new();
    seeders.push(Box::new(SeedAttributeSeeder::new()));
    seeders.push(Box::new(SeedAttributeValueSeeder::new()));
    seeders.push(Box::new(SeedBrandSeeder::new()));
    seeders.push(Box::new(SeedItemSeeder::new()));
    seeders.push(Box::new(SeedItemGroupSeeder::new()));
    seeders.push(Box::new(SeedItemVariantSeeder::new()));
    seeders.push(Box::new(SeedUomSeeder::new()));
    seeders.push(Box::new(SeedUomConversionSeeder::new()));

    // Sort by order
    seeders.sort_by_key(|s| s.order());

    // Run seeders
    for seeder in &seeders {
        let name = seeder.name();

        // Apply filter if specified
        if let Some(f) = filter {
            if !name.contains(f) {
                continue;
            }
        }

        if force {
            println!("   Running {}...", name);
            seeder.run(&pool).await?;
            println!("   {} completed", name);
        } else {
            println!("   Checking {}...", name);
            if seeder.should_run(&pool).await? {
                println!("   Running {}...", name);
                seeder.run(&pool).await?;
                println!("   {} completed", name);
            } else {
                println!("   {} already applied, skipping", name);
            }
        }
    }

    println!();
    println!("All seeders completed successfully!");

    Ok(())
}