qw 0.5.0

QW
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

use anyhow::Result;
use inflector::Inflector;

pub struct FieldDef {
    pub name:     String,
    pub sql_type: String,
    pub not_null: bool,
}

pub fn generate_migration_sql(model_name: &str, default_pk: bool, fields: &[FieldDef]) -> String {
    let table_name = model_name.to_snake_case().to_plural();
    let mut columns = vec![];

    if default_pk {
        columns.push(r#"    "id" bigint NOT NULL PRIMARY KEY"#.to_string());
    }

    for f in fields {
        let constraint = if f.not_null { " NOT NULL" } else { "" };
        columns.push(format!(r#"    "{}" {}{}"#, f.name, f.sql_type, constraint));
    }

    format!(
        r#"CREATE TABLE "{table_name}"
(
{}
);
"#,
        columns.join(
            ",
"
        )
    )
}

pub fn generate_add_column_sql(table_name: &str, field: &FieldDef) -> String {
    let constraint = if field.not_null { " NOT NULL" } else { "" };
    format!(
        r#"ALTER TABLE "{table_name}" ADD COLUMN "{}" {}{};
"#,
        field.name, field.sql_type, constraint
    )
}

pub fn write_migration(migrations_path: &str, name: &str, sql: &str) -> Result<String> {
    let timestamp = chrono::Utc::now().format("%Y%m%d%H%M%S");
    let filename = format!("{timestamp}_{name}.sql");
    let path = std::path::Path::new(migrations_path).join(&filename);
    std::fs::write(&path, sql)?;
    Ok(filename)
}