rust-db-blueprint 0.1.0

A Rust code generator — reads YAML draft files and generates Axum + SQLx models, migrations, handlers, routes, requests, tests, and seeds
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

use indexmap::IndexMap;

use crate::tree::Tree;

pub struct SeederGenerator;

impl SeederGenerator {
    pub fn generate(tree: &Tree) -> IndexMap<String, String> {
        let mut files = IndexMap::new();

        let mut lines = vec![
            "//! Database seed functions.".to_string(),
            "//! Run these to populate your database with test data.".to_string(),
            String::new(),
            "use sqlx::PgPool;".to_string(),
            "use crate::db::factories::*;".to_string(),
            String::new(),
            "/// Run all seeders.".to_string(),
            "pub async fn run_all(pool: &PgPool) {".to_string(),
        ];

        for model_name in &tree.seeders {
            let lower = model_name.to_lowercase();
            if tree.models.contains_key(model_name) {
                if let Some(model) = tree.models.get(model_name) {
                    if model.pivot { continue; }
                }
                lines.push(format!("    create_{}_default(pool).await;", lower));
            }
        }

        lines.push("}".to_string());

        let content = lines.join("\n");
        files.insert("src/db/seeds.rs".to_string(), content);
        files
    }
}