63 lines
1.8 KiB
Rust
63 lines
1.8 KiB
Rust
|
mod create_user;
|
||
|
|
mod two;
|
||
|
|
|
||
|
|
use argh::FromArgs;
|
||
|
|
use tracing::info;
|
||
|
|
use web3_proxy::app::get_migrated_db;
|
||
|
|
|
||
|
|
#[derive(Debug, FromArgs)]
|
||
|
|
/// Command line interface for admins to interact with web3-proxy
|
||
|
|
pub struct TopConfig {
|
||
|
|
/// what host the client should connect to
|
||
|
|
#[argh(
|
||
|
|
option,
|
||
|
|
default = "\"mysql://root:dev_web3_proxy@127.0.0.1:3306/dev_web3_proxy\".to_string()"
|
||
|
|
)]
|
||
|
|
pub db_url: String,
|
||
|
|
|
||
|
|
/// this one cli can do multiple things
|
||
|
|
#[argh(subcommand)]
|
||
|
|
sub_command: SubCommand,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
||
|
|
#[argh(subcommand)]
|
||
|
|
enum SubCommand {
|
||
|
|
CreateUser(create_user::CreateUserSubCommand),
|
||
|
|
Two(two::SubCommandTwo),
|
||
|
|
// TODO: sub command to downgrade migrations?
|
||
|
|
// TODO: sub command to add new api keys to an existing user?
|
||
|
|
}
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn main() -> anyhow::Result<()> {
|
||
|
|
// if RUST_LOG isn't set, configure a default
|
||
|
|
// TODO: is there a better way to do this?
|
||
|
|
if std::env::var("RUST_LOG").is_err() {
|
||
|
|
// std::env::set_var("RUST_LOG", "info,web3_proxy=debug,web3_proxy_cli=debug");
|
||
|
|
std::env::set_var("RUST_LOG", "info,web3_proxy=debug,web3_proxy_cli=debug");
|
||
|
|
}
|
||
|
|
|
||
|
|
// install global collector configured based on RUST_LOG env var.
|
||
|
|
tracing_subscriber::fmt()
|
||
|
|
.with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
|
||
|
|
.compact()
|
||
|
|
.init();
|
||
|
|
|
||
|
|
// this probably won't matter for us in docker, but better safe than sorry
|
||
|
|
fdlimit::raise_fd_limit();
|
||
|
|
|
||
|
|
let cli_config: TopConfig = argh::from_env();
|
||
|
|
|
||
|
|
info!("hello, {}", cli_config.db_url);
|
||
|
|
|
||
|
|
match cli_config.sub_command {
|
||
|
|
SubCommand::CreateUser(x) => {
|
||
|
|
let db = get_migrated_db(cli_config.db_url, 1).await?;
|
||
|
|
|
||
|
|
x.main(&db).await
|
||
|
|
}
|
||
|
|
SubCommand::Two(x) => x.main().await,
|
||
|
|
}
|
||
|
|
}
|