wtx-ui 0.2.2

Different user interfaces for WTX
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use crate::clap::EmbedMigrations;
use std::{fmt::Write, path::Path};
use tokio::{fs::OpenOptions, io::AsyncWriteExt};
use wtx::database::schema_manager::misc::{group_and_migrations_from_path, parse_root_toml};

pub(crate) async fn embed_migrations(elem: EmbedMigrations) -> wtx::Result<()> {
  let (mut migration_groups, _) = parse_root_toml(Path::new(&elem.input))?;
  let mut buffer = String::new();

  migration_groups.sort();

  buffer.push_str(
    "#[rustfmt::skip]pub(crate) static GROUPS: wtx::database::schema_manager::EmbeddedMigrationsTy = &[",
  );

  for mg_path in migration_groups {
    let (mg, ms) = group_and_migrations_from_path(&mg_path, Ord::cmp)?;
    let mg_name = mg.name().to_ascii_uppercase();
    let mg_uid = mg.uid();

    buffer.write_fmt(format_args!(
      concat!(
        "{{",
        r#"const {mg_name}: &wtx::database::schema_manager::UserMigrationGroup<&'static str> = &wtx::database::schema_manager::UserMigrationGroup::new("{mg_name}",{mg_uid});"#,
        r#"const {mg_name}_MIGRATIONS: &[wtx::database::schema_manager::UserMigrationRef<'static, 'static>] = &["#
      ),
      mg_name = mg_name,
      mg_uid = mg_uid
    ))?;

    for rslt in ms {
      let migration = rslt?;
      let checksum = migration.checksum();
      let dbs = migration.dbs();
      let name = migration.name();
      let sql_down = migration.sql_down();
      let sql_up = migration.sql_up();
      let uid = migration.uid();

      buffer.write_fmt(format_args!(
        "wtx::database::schema_manager::UserMigrationRef::from_all_parts({checksum},&["
      ))?;
      for db in dbs {
        buffer.push_str("wtx::database::DatabaseTy::");
        buffer.push_str(db.strings().ident);
        buffer.push(',');
      }
      buffer.write_fmt(format_args!(r#"],"{name}","#))?;
      match migration.repeatability() {
        None => buffer.push_str("None"),
        Some(elem) => buffer.write_fmt(format_args!(
          "Some(wtx::database::schema_manager::Repeatability::{})",
          elem.strings().ident
        ))?,
      }
      buffer.write_fmt(format_args!(r#","{sql_down}","{sql_up}",{uid}),"#))?;
    }

    buffer.write_fmt(format_args!("];({mg_name},{mg_name}_MIGRATIONS)}},"))?;
  }

  buffer.push_str("];\n");

  OpenOptions::new()
    .create(true)
    .truncate(true)
    .write(true)
    .open(&elem.output)
    .await?
    .write_all(buffer.as_bytes())
    .await?;

  Ok(())
}