dan/Portfolio
1
0
Fork 0
mirror of https://github.com/danbulant/Portfolio synced 2026-05-27 22:12:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: require URL instead

Browse source
This commit is contained in:
EETagent 2022-10-30 12:52:09 +01:00
parent 4c8546f81e
commit ee4d137a7d
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
recovery/src/main.rs
View file

@ -11,13 +11,13 @@ use ::entity::parent::Entity as Parent;
#[tokio::main] #[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> { async fn main() -> Result<(), Box<dyn std::error::Error>> {
let clap = command!() let clap = command!()
.arg(arg!([name] "Path to the db .sql backup"))
.arg( .arg(
arg!( arg!(
-d --database <DATABASE> "Path to the database SQL backup file" -d --database <URL> "URL to the database or sql file with postgres:// or sqlite://"
) )
.alias("url")
.required(true) .required(true)
.value_parser(value_parser!(PathBuf)), .value_parser(value_parser!(Url)),
) )
.arg( .arg(
arg!( arg!(
@ -34,8 +34,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
) )
.get_matches(); .get_matches();
let mut sqlite_url = Url::from_file_path(clap.get_one::<PathBuf>("DATABASE").unwrap()).unwrap(); let sqlite_url = clap.get_one::<Url>("database").unwrap();
sqlite_url.set_scheme("sqlite").unwrap();
println!("Connecting to {:?}", sqlite_url);
if sqlite_url.scheme() != "sqlite" && sqlite_url.scheme() != "postgres" {
return Err("URL scheme postgres:// or sqlite:// required")?;
}
let db: DatabaseConnection = Database::connect(sqlite_url.as_str()).await?; let db: DatabaseConnection = Database::connect(sqlite_url.as_str()).await?;