use anyhow::Result;
use sqlx::postgres::PgPoolOptions;
use std::env;
use backbone_digest::seeders::SeedDigestDigestSeeder;
use backbone_digest::seeders::SeedDigestSubscriptionSeeder;
use backbone_digest::seeders::SeedDigestDigestKpiSeeder;
use backbone_digest::seeders::SeedDigestTipSeeder;
use backbone_digest::seeders::SeedDigestTipUserSeeder;
use backbone_digest::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: digest");
println!();
let mut seeders: Vec<Box<dyn Seeder + Send + Sync>> = Vec::new();
seeders.push(Box::new(SeedDigestDigestSeeder::new()));
seeders.push(Box::new(SeedDigestSubscriptionSeeder::new()));
seeders.push(Box::new(SeedDigestDigestKpiSeeder::new()));
seeders.push(Box::new(SeedDigestTipSeeder::new()));
seeders.push(Box::new(SeedDigestTipUserSeeder::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(())
}