82 lines
2.3 KiB
Rust
82 lines
2.3 KiB
Rust
#[macro_use]
|
|
extern crate prettytable;
|
|
|
|
#[cfg(feature = "mysql-admutils-compatibility")]
|
|
use std::path::PathBuf;
|
|
|
|
#[cfg(feature = "mysql-admutils-compatibility")]
|
|
use crate::cli::mysql_admutils_compatibility::{mysql_dbadm, mysql_useradm};
|
|
|
|
use clap::Parser;
|
|
|
|
mod cli;
|
|
mod core;
|
|
|
|
#[cfg(feature = "tui")]
|
|
mod tui;
|
|
|
|
#[derive(Parser)]
|
|
struct Args {
|
|
#[command(subcommand)]
|
|
command: Command,
|
|
|
|
#[command(flatten)]
|
|
config_overrides: core::config::GlobalConfigArgs,
|
|
|
|
#[cfg(feature = "tui")]
|
|
#[arg(short, long, alias = "tui", global = true)]
|
|
interactive: bool,
|
|
}
|
|
|
|
/// Database administration tool for non-admin users to manage their own MySQL databases and users.
|
|
///
|
|
/// This tool allows you to manage users and databases in MySQL.
|
|
///
|
|
/// You are only allowed to manage databases and users that are prefixed with
|
|
/// either your username, or a group that you are a member of.
|
|
#[derive(Parser)]
|
|
#[command(version, about, disable_help_subcommand = true)]
|
|
enum Command {
|
|
#[command(flatten)]
|
|
Db(cli::database_command::DatabaseCommand),
|
|
|
|
#[command(flatten)]
|
|
User(cli::user_command::UserCommand),
|
|
}
|
|
|
|
#[tokio::main(flavor = "current_thread")]
|
|
async fn main() -> anyhow::Result<()> {
|
|
env_logger::init();
|
|
|
|
#[cfg(feature = "mysql-admutils-compatibility")]
|
|
{
|
|
let argv0 = std::env::args().next().and_then(|s| {
|
|
PathBuf::from(s)
|
|
.file_name()
|
|
.map(|s| s.to_string_lossy().to_string())
|
|
});
|
|
|
|
match argv0.as_deref() {
|
|
Some("mysql-dbadm") => return mysql_dbadm::main().await,
|
|
Some("mysql-useradm") => return mysql_useradm::main().await,
|
|
_ => { /* fall through */ }
|
|
}
|
|
}
|
|
|
|
let args: Args = Args::parse();
|
|
let config = core::config::get_config(args.config_overrides)?;
|
|
let connection = core::config::mysql_connection_from_config(config).await?;
|
|
|
|
let result = match args.command {
|
|
Command::Db(command) => cli::database_command::handle_command(command, connection).await,
|
|
Command::User(user_args) => cli::user_command::handle_command(user_args, connection).await,
|
|
};
|
|
|
|
match result {
|
|
Ok(_) => println!("Changes committed to database"),
|
|
Err(_) => println!("Changes reverted due to error"),
|
|
}
|
|
|
|
result
|
|
}
|