germanubis/sqlx: 🧰 The Rust SQL Toolkit. An async, pure Rust SQL crate featuring compile-time checked queries without a DSL. Supports PostgreSQL, MySQL, and SQLite.

sqlx-cli/src/bin/cargo-sqlx.rs

24 строки745 B

David James

Fix issue #1431 (#1432)

11 сен 2021, 01:06

11 сен 2021, 01:06bb33a29

use clap::{crate_version, AppSettings, FromArgMatches, IntoApp};
use console::style;
use dotenv::dotenv;
use sqlx_cli::Opt;
use std::{env, process};

#[tokio::main]
async fn main() {
    // when invoked as `cargo sqlx [...]` the args we see are `[...]/sqlx-cli sqlx prepare`
    // so we want to notch out that superfluous "sqlx"
    let args = env::args_os().skip(2);

    dotenv().ok();
    let matches = Opt::into_app()
        .version(crate_version!())
        .bin_name("cargo sqlx")
        .setting(AppSettings::NoBinaryName)
        .get_matches_from(args);

    if let Err(error) = sqlx_cli::run(Opt::from_arg_matches(&matches)).await {
        println!("{} {}", style("error:").bold().red(), error);
        process::exit(1);
    }
}