Jędrzej Stuczyński
911b365609
* purge any temp databases on build * updated min rust version * fixed clippy::manual_abs_diff' in verloc due to updated msrv * wasm
105 lines
3.2 KiB
Rust
105 lines
3.2 KiB
Rust
use anyhow::Context;
|
|
use sqlx::{Connection, FromRow, SqliteConnection};
|
|
use std::env;
|
|
|
|
const SQLITE_DB_FILENAME: &str = "nym-api-example.sqlite";
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
let out_dir = env::var("OUT_DIR")?;
|
|
let database_path = format!("{out_dir}/{SQLITE_DB_FILENAME}");
|
|
|
|
// remove the db file if it already existed from previous build
|
|
// in case it was from a different branch
|
|
if std::fs::exists(&database_path)? {
|
|
std::fs::remove_file(&database_path)?;
|
|
}
|
|
|
|
#[cfg(target_family = "unix")]
|
|
write_db_path_to_file(&out_dir, SQLITE_DB_FILENAME)
|
|
.await
|
|
.ok();
|
|
|
|
let mut conn = SqliteConnection::connect(&format!("sqlite://{database_path}?mode=rwc"))
|
|
.await
|
|
.context("Failed to create SQLx database connection")?;
|
|
|
|
sqlx::migrate!("./migrations")
|
|
.run(&mut conn)
|
|
.await
|
|
.context("Failed to perform SQLx migrations")?;
|
|
|
|
#[derive(FromRow)]
|
|
struct Exists {
|
|
exists: bool,
|
|
}
|
|
|
|
// check if it was already run
|
|
let res: Exists = sqlx::query_as("SELECT EXISTS (SELECT 1 FROM v3_migration_info) AS 'exists'")
|
|
.fetch_one(&mut conn)
|
|
.await?;
|
|
|
|
let already_run = res.exists;
|
|
|
|
// execute the manual v3 migration
|
|
// it's performed on an empty storage, so we don't need to actually make any network queries
|
|
if !already_run {
|
|
sqlx::query(
|
|
r#"
|
|
CREATE TABLE gateway_details_temp
|
|
(
|
|
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
node_id INTEGER NOT NULL UNIQUE,
|
|
identity VARCHAR NOT NULL UNIQUE
|
|
);
|
|
|
|
DROP TABLE gateway_details;
|
|
ALTER TABLE gateway_details_temp RENAME TO gateway_details;
|
|
|
|
INSERT INTO v3_migration_info(id) VALUES (0);
|
|
"#,
|
|
)
|
|
.execute(&mut conn)
|
|
.await
|
|
.context("failed to update post v3 migration tables")?;
|
|
}
|
|
|
|
#[cfg(target_family = "unix")]
|
|
println!("cargo:rustc-env=DATABASE_URL=sqlite://{}", &database_path);
|
|
|
|
#[cfg(target_family = "windows")]
|
|
// for some strange reason we need to add a leading `/` to the windows path even though it's
|
|
// not a valid windows path... but hey, it works...
|
|
println!("cargo:rustc-env=DATABASE_URL=sqlite:///{}", &database_path);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// use `./enter_db.sh` to inspect DB
|
|
#[cfg(target_family = "unix")]
|
|
async fn write_db_path_to_file(out_dir: &str, db_filename: &str) -> anyhow::Result<()> {
|
|
use std::os::unix::fs::PermissionsExt;
|
|
use tokio::{fs::File, io::AsyncWriteExt};
|
|
|
|
if env::var("CI").is_ok() {
|
|
return Ok(());
|
|
}
|
|
let mut file = File::create("settings.sql").await?;
|
|
let settings = ".mode columns
|
|
.headers on";
|
|
file.write_all(settings.as_bytes()).await?;
|
|
|
|
let mut file = File::create("enter_db.sh").await?;
|
|
let contents = format!(
|
|
"#!/bin/sh\n\
|
|
sqlite3 -init settings.sql {out_dir}/{db_filename}",
|
|
);
|
|
file.write_all(contents.as_bytes()).await?;
|
|
|
|
file.set_permissions(std::fs::Permissions::from_mode(0o755))
|
|
.await
|
|
.map_err(anyhow::Error::from)?;
|
|
|
|
Ok(())
|
|
}
|