cdk-signatory 0.16.0

//! Signatory CLI main logic
//!
//! This logic is in this file to be excluded for wasm
use std::path::PathBuf;

use anyhow::{bail, Result};
use clap::Parser;
#[cfg(feature = "sqlite")]
use {
    anyhow::anyhow,
    bip39::rand::{thread_rng, Rng},
    bip39::Mnemonic,
    cdk_common::database::MintKeysDatabase,
    cdk_common::CurrencyUnit,
    cdk_signatory::{db_signatory, start_grpc_server},
    cdk_sqlite::MintSqliteDatabase,
    std::collections::HashMap,
    std::net::SocketAddr,
    std::str::FromStr,
    std::sync::Arc,
    std::{env, fs},
    tracing_subscriber::EnvFilter,
};

/// Common CLI arguments for CDK binaries
#[derive(Parser, Debug)]
pub struct CommonArgs {
    /// Enable logging (default is false)
    #[arg(long, default_value_t = false)]
    pub enable_logging: bool,

    /// Logging level when enabled (default is debug)
    #[arg(long, default_value = "debug")]
    pub log_level: tracing::Level,
}

/// Initialize logging based on CLI arguments
#[cfg(feature = "sqlite")]
pub fn init_logging(enable_logging: bool, log_level: tracing::Level) {
    if enable_logging {
        let default_filter = log_level.to_string();

        // Common filters to reduce noise
        let sqlx_filter = "sqlx=warn";
        let hyper_filter = "hyper=warn";
        let h2_filter = "h2=warn";
        let rustls_filter = "rustls=warn";
        let reqwest_filter = "reqwest=warn";

        let env_filter = EnvFilter::new(format!(
            "{default_filter},{sqlx_filter},{hyper_filter},{h2_filter},{rustls_filter},{reqwest_filter}"
        ));

        // Ok if successful, Err if already initialized
        let _ = tracing_subscriber::fmt()
            .with_env_filter(env_filter)
            .with_ansi(false)
            .try_init();
    }
}

#[cfg(feature = "sqlite")]
const DEFAULT_WORK_DIR: &str = ".cdk-signatory";
#[cfg(feature = "sqlite")]
const ENV_MNEMONIC: &str = "CDK_MINTD_MNEMONIC";

/// Simple CLI application to interact with cashu
#[derive(Parser)]
#[command(name = "cashu-signatory")]
#[command(author = "thesimplekid <tsk@thesimplekid.com>")]
#[command(version = "0.1.0")]
#[command(author, version, about, long_about = None)]
struct Cli {
    #[command(flatten)]
    common: CommonArgs,

    /// Database engine to use (sqlite/redb)
    #[arg(short, long, default_value = "sqlite")]
    engine: String,
    /// Database password for sqlcipher
    #[arg(long)]
    password: Option<String>,
    /// Path to working dir
    #[arg(short, long)]
    work_dir: Option<PathBuf>,
    #[arg(long, default_value = "127.0.0.1")]
    listen_addr: String,
    #[arg(long, default_value = "15060")]
    listen_port: u32,
    #[arg(long, short)]
    certs: Option<String>,
    /// Supported units with the format of name,fee and max_order
    #[arg(long, short, default_value = "sat,0,32")]
    units: Vec<String>,
}

/// Main function for the signatory standalone binary
#[cfg(not(feature = "sqlite"))]
pub async fn cli_main() -> Result<()> {
    bail!("No database feature enabled. Enable the 'sqlite' feature to use this binary.");
}

/// Main function for the signatory standalone binary
#[cfg(feature = "sqlite")]
pub async fn cli_main() -> Result<()> {
    let args: Cli = Cli::parse();

    // Initialize logging based on CLI arguments
    init_logging(args.common.enable_logging, args.common.log_level);

    let work_dir = match &args.work_dir {
        Some(work_dir) => work_dir.clone(),
        None => {
            let home_dir = home::home_dir().ok_or(anyhow!("Unknown how"))?;
            home_dir.join(DEFAULT_WORK_DIR)
        }
    };

    fs::create_dir_all(&work_dir)?;

    let localstore: Arc<dyn MintKeysDatabase<Err = cdk_common::database::Error> + Send + Sync> =
        match args.engine.as_str() {
            "sqlite" => {
                let sql_path = work_dir.join("cdk-cli.sqlite");
                #[cfg(not(feature = "sqlcipher"))]
                let db = MintSqliteDatabase::new(&sql_path).await?;
                #[cfg(feature = "sqlcipher")]
                let db = {
                    match args.password {
                        Some(pass) => MintSqliteDatabase::new((sql_path.clone(), pass)).await?,
                        None => bail!("Missing database password"),
                    }
                };

                Arc::new(db)
            }
            _ => bail!("Unknown DB engine"),
        };

    let supported_units = args
        .units
        .into_iter()
        .map(|unit| {
            let mut parts = unit.split(",").collect::<Vec<_>>();
            parts.reverse();
            let unit: CurrencyUnit = parts.pop().unwrap_or_default().parse()?;
            let fee = parts
                .pop()
                .map(|x| x.parse())
                .transpose()?
                .unwrap_or_default();
            let max_order: u32 = parts.pop().map(|x| x.parse()).transpose()?.unwrap_or(32);
            // Convert max_order to amounts list (powers of 2)
            let amounts: Vec<u64> = (0..max_order).map(|i| 2u64.pow(i)).collect();
            Ok::<(_, (_, _)), anyhow::Error>((unit, (fee, amounts)))
        })
        .collect::<Result<HashMap<_, _>, _>>()?;

    let certs = Some(
        args.certs
            .map(|x| x.into())
            .unwrap_or_else(|| work_dir.clone()),
    );

    let seed_path = work_dir.join("seed");

    let mnemonic = if let Ok(mnemonic) = env::var(ENV_MNEMONIC) {
        Mnemonic::from_str(&mnemonic)?
    } else {
        match fs::metadata(seed_path.clone()) {
            Ok(_) => {
                let contents = fs::read_to_string(seed_path.clone())?;
                Mnemonic::from_str(&contents)?
            }
            Err(_e) => {
                let mut rng = thread_rng();
                let random_bytes: [u8; 32] = rng.gen();

                let mnemonic = Mnemonic::from_entropy(&random_bytes)?;
                tracing::info!("Creating new seed");

                fs::write(seed_path, mnemonic.to_string())?;

                mnemonic
            }
        }
    };
    let seed = mnemonic.to_seed_normalized("");

    let signatory =
        db_signatory::DbSignatory::new(localstore, &seed, supported_units, Default::default())
            .await?;

    let socket_addr = SocketAddr::from_str(&format!("{}:{}", args.listen_addr, args.listen_port))?;

    start_grpc_server(Arc::new(signatory), socket_addr, certs).await?;

    Ok(())
}