seedgen 0.3.0

Zero-config database seed data generator with deterministic, constraint-safe fake data
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use anyhow::Context;
use sqlx::PgPool;

use crate::introspection::introspect;
use crate::output::truncate_tables;
use crate::resolver::resolve;

#[derive(Debug)]
pub struct Args {
    pub confirm: bool,
    pub only: Option<Vec<String>>,
    pub cascade: bool,
}

pub async fn run(args: Args, url: &str) -> anyhow::Result<()> {
    if !args.confirm {
        anyhow::bail!(
            "reset is destructive — pass --confirm to proceed (this TRUNCATEs target tables)"
        );
    }

    if url.to_ascii_lowercase().contains("prod") {
        anyhow::bail!(
            "refusing to reset a database whose URL contains `prod`. \
             override is intentional and not yet implemented."
        );
    }

    let pool = PgPool::connect(url)
        .await
        .context("failed to connect to database")?;

    let tables = match args.only {
        Some(list) if !list.is_empty() => list,
        _ => {
            let schema = introspect(&pool).await?;
            let plan = resolve(&schema)?;
            let mut all = plan.ordered_tables;
            // Children first for safer truncation (CASCADE handles the rest anyway).
            all.reverse();
            all
        }
    };

    if tables.is_empty() {
        println!("No tables to reset.");
        return Ok(());
    }

    truncate_tables(&pool, &tables).await?;
    println!(
        "Truncated {} table{}{}",
        tables.len(),
        if tables.len() == 1 { "" } else { "s" },
        if args.cascade { " (CASCADE)" } else { "" },
    );
    Ok(())
}