itsscb/sqlx
1
0
Fork 0
mirror of https://github.com/launchbadge/sqlx.git synced 2025-10-02 07:21:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use dialoger for prompt

Browse Source
This commit is contained in:
Jesper Axelsson 2020-04-19 09:21:01 +02:00 committed by Ryan Leckey
parent f491198891
commit 38892c1012
1 changed files with 8 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
cargo-sqlx/src/main.rs
View File

@ -10,6 +10,7 @@ use structopt::StructOpt;
use anyhow::{anyhow, Context, Result};
use console::style;
use dialoguer::Confirmation;
mod database_migrator;
mod postgres;
@ -133,8 +134,6 @@ async fn run_create_database(migrator: &dyn DatabaseMigrator) -> Result<()> {
}
async fn run_drop_database(migrator: &dyn DatabaseMigrator) -> Result<()> {
use std::io;
if !migrator.can_drop_database() {
return Err(anyhow!(
"Database drop is not implemented for {}",
@ -146,24 +145,13 @@ async fn run_drop_database(migrator: &dyn DatabaseMigrator) -> Result<()> {
let db_exists = migrator.check_if_database_exists(&db_name).await?;
if db_exists {
loop {
println!(
"\nAre you sure you want to drop the database: {}? Y/n",
db_name
);
let mut input = String::new();
io::stdin()
.read_line(&mut input)
.context("Failed to read line")?;
match input.trim() {
"Y" => break,
"N" => return Ok(()),
"n" => return Ok(()),
_ => continue,
};
if !Confirmation::new()
.with_text("\nAre you sure you want to drop the database: {}?")
.default(false)
.interact()?
{
println!("Aborting");
return Ok(());
}
println!("Dropping database: {}", db_name);