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

use etwin_squirrel::SchemaResolver;
pub use etwin_squirrel::SchemaStateRef;
use include_dir::{include_dir, Dir};
use lazy_static::lazy_static;
use sqlx::PgPool;
use std::error::Error;

const DB_SCRIPTS: Dir = include_dir!("$CARGO_MANIFEST_DIR/scripts");

lazy_static! {
  static ref SQUIRREL: SchemaResolver = SchemaResolver::new(&DB_SCRIPTS);
}

pub async fn get_state(db: &PgPool) -> Result<SchemaStateRef<'static>, Box<dyn Error>> {
  SQUIRREL.get_state(db).await
}

pub async fn empty(db: &PgPool) -> Result<(), Box<dyn Error>> {
  SQUIRREL.empty(db).await
}

pub async fn force_create_latest(db: &PgPool, void: bool) -> Result<(), Box<dyn Error>> {
  SQUIRREL.force_create_latest(db, void).await
}

// pub async fn upgrade_to_latest(db: &PgPool) -> Result<(), Box<dyn Error>> {
//   let start = SQUIRREL.get_state(db).await?;
//   let end = SQUIRREL.get_latest();
//   let migration = SQUIRREL
//     .create_migration(start, end, MigrationDirection::UpgradeOnly);
//   let migration = migration.ok_or_else(|| "FailedToComputeMigrationToLatest".into())?;
//   SQUIRREL.apply_migration(db, &migration).await
// }

pub async fn force_create(db: &PgPool, state: SchemaStateRef<'static>, void: bool) -> Result<(), Box<dyn Error>> {
  SQUIRREL.force_create(db, state, void).await
}