breez-sdk-liquid/cli/src/main.rs

mod commands;
mod persist;

use std::{fs, path::PathBuf};

use anyhow::{anyhow, Result};
use breez_sdk_liquid::prelude::*;
use clap::Parser;
use commands::{handle_command, CliHelper, Command, CommandResult};
use log::{error, info};
use persist::CliPersistence;
use rustyline::{error::ReadlineError, hint::HistoryHinter, Editor};

#[derive(Parser, Debug)]
pub(crate) struct Args {
    #[clap(short, long)]
    pub(crate) data_dir: Option<String>,

    #[clap(short, long)]
    pub(crate) cache_dir: Option<String>,

    #[clap(short, long)]
    pub(crate) log_file: Option<String>,

    #[clap(short, long, value_parser = parse_network_arg)]
    pub(crate) network: Option<LiquidNetwork>,
}

fn parse_network_arg(s: &str) -> Result<LiquidNetwork, String> {
    LiquidNetwork::try_from(s).map_err(|e| e.to_string())
}

fn show_results(result: Result<String>) -> Result<()> {
    let result_str = match result {
        Ok(r) => r,
        Err(err) => serde_json::to_string_pretty(&CommandResult {
            success: false,
            message: err.to_string(),
        })?,
    };

    Ok(println!("{result_str}"))
}

struct CliEventListener {}

impl EventListener for CliEventListener {
    fn on_event(&self, e: SdkEvent) {
        info!("Received event: {:?}", e);
    }
}

#[tokio::main]
async fn main() -> Result<()> {
    let args = Args::parse();

    let data_dir_str = args.data_dir.unwrap_or(DEFAULT_DATA_DIR.to_string());
    let data_dir = PathBuf::from(&data_dir_str);
    fs::create_dir_all(&data_dir)?;

    LiquidSdk::init_logging(&data_dir_str, None)?;

    let persistence = CliPersistence { data_dir };
    let history_file = &persistence.history_file();

    let rl = &mut Editor::new()?;
    rl.set_helper(Some(CliHelper {
        hinter: HistoryHinter {},
    }));

    if rl.load_history(history_file).is_err() {
        info!("No history found");
    }

    let mnemonic = persistence.get_or_create_mnemonic()?;
    let network = args.network.unwrap_or(LiquidNetwork::Testnet);
    let breez_api_key = std::env::var_os("BREEZ_API_KEY")
        .map(|var| var.into_string().expect("Expected valid API key string"));
    let mut config = LiquidSdk::default_config(network, breez_api_key)?;
    config.working_dir = data_dir_str;
    config.cache_dir = args.cache_dir;
    let sdk = LiquidSdk::connect(ConnectRequest {
        mnemonic: mnemonic.to_string(),
        config,
    })
    .await?;
    let listener_id = sdk
        .add_event_listener(Box::new(CliEventListener {}))
        .await?;

    let cli_prompt = match network {
        LiquidNetwork::Mainnet => "breez-liquid-cli [mainnet]> ",
        LiquidNetwork::Testnet => "breez-liquid-cli [testnet]> ",
    };

    loop {
        let readline = rl.readline(cli_prompt);
        match readline {
            Ok(line) => {
                rl.add_history_entry(line.as_str())?;
                let mut vec: Vec<&str> = line.as_str().split_whitespace().collect();
                vec.insert(0, "");
                let cli_res = Command::try_parse_from(vec);
                if cli_res.is_err() {
                    println!("{}", cli_res.unwrap_err());
                    continue;
                }
                let res = handle_command(rl, &sdk, cli_res.unwrap()).await;
                show_results(res)?;
            }
            Err(ReadlineError::Interrupted) => {
                info!("CTRL-C");
                sdk.disconnect().await?;
                break;
            }
            Err(ReadlineError::Eof) => {
                info!("CTRL-D");
                sdk.disconnect().await?;
                break;
            }
            Err(err) => {
                error!("Error: {:?}", err);
                break;
            }
        }
    }

    sdk.remove_event_listener(listener_id).await?;
    rl.save_history(history_file).map_err(|e| anyhow!(e))
}