use anyhow::Result;
use sqlx::postgres::PgPoolOptions;
use std::env;
use backbone_payroll::seeders::SeedCompensationChangeSeeder;
use backbone_payroll::seeders::SeedPayrollEntrySeeder;
use backbone_payroll::seeders::SeedSalarySlipSeeder;
use backbone_payroll::seeders::SeedSalarySlipLineSeeder;
use backbone_payroll::seeders::SeedSalaryStructureSeeder;
use backbone_payroll::seeders::SeedSalaryComponentSeeder;
use backbone_payroll::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: payroll");
println!();
let mut seeders: Vec<Box<dyn Seeder + Send + Sync>> = Vec::new();
seeders.push(Box::new(SeedCompensationChangeSeeder::new()));
seeders.push(Box::new(SeedPayrollEntrySeeder::new()));
seeders.push(Box::new(SeedSalarySlipSeeder::new()));
seeders.push(Box::new(SeedSalarySlipLineSeeder::new()));
seeders.push(Box::new(SeedSalaryStructureSeeder::new()));
seeders.push(Box::new(SeedSalaryComponentSeeder::new()));
seeders.sort_by_key(|s| s.order());
for seeder in &seeders {
let name = seeder.name();
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(())
}