lwk_app 0.17.0

use crate::{blockchain_client::BlockchainClient, consts, Error};
use lwk_common::electrum_ssl::LIQUID_SOCKET;
use lwk_common::electrum_ssl::LIQUID_TESTNET_SOCKET;
use lwk_common::Network as JadeNetwork;
use lwk_jade::TIMEOUT;
use lwk_wollet::amp2::Amp2;
use lwk_wollet::clients::blocking::EsploraClient;
use lwk_wollet::elements::AssetId;
use lwk_wollet::ElementsNetwork;
use lwk_wollet::{amp2, ElectrumClient};
use std::fs;
use std::net::SocketAddr;
use std::path::PathBuf;
use std::str::FromStr;
use std::time::Duration;

#[derive(Clone, Debug)]
pub struct Config {
    /// The address where the RPC server is listening or the client is connecting to
    pub addr: SocketAddr,
    pub datadir: PathBuf,
    pub server_url: String,
    pub server_type: String,
    pub network: ElementsNetwork,

    pub explorer_url: String,

    pub registry_url: String,
    pub timeout: Duration,
    pub scanning_interval: Duration,

    pub amp2_url: String,
    pub amp2_keyorigin_xpub: String,
}

impl Config {
    pub fn default_testnet(datadir: PathBuf) -> Self {
        Self {
            addr: consts::DEFAULT_ADDR.into(),
            datadir,
            server_url: format!("ssl://{LIQUID_TESTNET_SOCKET}"),
            server_type: "electrum".into(),
            network: ElementsNetwork::LiquidTestnet,
            explorer_url: "https://blockstream.info/liquidtestnet/".into(),
            registry_url: "https://assets-testnet.blockstream.info/".into(),
            timeout: TIMEOUT,
            scanning_interval: consts::SCANNING_INTERVAL,
            amp2_url: amp2::URL_TESTNET.into(),
            amp2_keyorigin_xpub: amp2::KEYORIGIN_XPUB_TESTNET.into(),
        }
    }

    pub fn default_mainnet(datadir: PathBuf) -> Self {
        Self {
            addr: consts::DEFAULT_ADDR.into(),
            datadir,
            server_url: format!("ssl://{LIQUID_SOCKET}"),
            server_type: "electrum".into(),
            network: ElementsNetwork::Liquid,
            explorer_url: "https://blockstream.info/liquid/".into(),
            registry_url: "https://assets.blockstream.info/".into(),
            timeout: TIMEOUT,
            scanning_interval: consts::SCANNING_INTERVAL,
            amp2_url: "".into(),
            amp2_keyorigin_xpub: "".into(),
        }
    }

    /// For regtest there are no reasonable default for `electrum_url`, `explorer_url`, `registry_url`, `amp2_url` and `amp2_keyorigin_xpub`
    /// It will be caller responsability to mutate them according to regtest env
    pub fn default_regtest(datadir: PathBuf) -> Self {
        let policy_asset = "5ac9f65c0efcc4775e0baec4ec03abdde22473cd3cf33c0419ca290e0751b225";
        let policy_asset = AssetId::from_str(policy_asset).expect("static");
        Self {
            addr: consts::DEFAULT_ADDR.into(),
            datadir,
            server_url: "".into(),
            server_type: "electrum".into(),
            network: ElementsNetwork::ElementsRegtest { policy_asset },
            explorer_url: "".into(),
            registry_url: "".into(),
            timeout: TIMEOUT,
            // Scan more frequently while testing
            scanning_interval: Duration::from_secs(1),
            amp2_url: "".into(),
            amp2_keyorigin_xpub: "".into(),
        }
    }

    pub fn jade_network(&self) -> JadeNetwork {
        match self.network {
            ElementsNetwork::Liquid => JadeNetwork::Liquid,
            ElementsNetwork::LiquidTestnet => JadeNetwork::TestnetLiquid,
            ElementsNetwork::ElementsRegtest { policy_asset } => JadeNetwork::CustomElements(
                lwk_common::ElementsParamsBuilder::new()
                    .with_policy_asset(policy_asset)
                    .build()
                    .expect("static"),
            ),
        }
    }

    pub fn default_home() -> Result<PathBuf, Error> {
        let mut path = home::home_dir().ok_or(Error::Generic("Cannot get home dir".into()))?;
        path.push(".lwk");
        fs::create_dir_all(&path)?;
        Ok(path)
    }

    /// Appends the network to the given datadir
    pub fn datadir(&self) -> Result<PathBuf, Error> {
        let mut path: PathBuf = self.datadir.clone();
        path.push(self.network.as_str());
        fs::create_dir_all(&path)?;
        Ok(path)
    }

    /// Returns the path of the state file under datadir
    pub fn state_path(&self) -> Result<PathBuf, Error> {
        let mut path = self.datadir()?;
        path.push("state.json");
        Ok(path)
    }

    /// True if Liquid mainnet
    pub fn is_mainnet(&self) -> bool {
        matches!(self.network, ElementsNetwork::Liquid)
    }

    pub fn blockchain_client(&self) -> Result<BlockchainClient, Error> {
        // TODO cache it instead of recreating every time
        match self.server_type.as_ref() {
            "electrum" => {
                let electrum_url = self.server_url.parse().map_err(lwk_wollet::Error::Url)?;
                let electrum_client = ElectrumClient::new(&electrum_url)?;
                Ok(BlockchainClient::Electrum(electrum_client))
            }
            "esplora" => {
                let esplora_client = EsploraClient::new(&self.server_url, self.network)?;
                Ok(BlockchainClient::Esplora(esplora_client))
            }
            "waterfalls" => {
                let waterfalls_client =
                    EsploraClient::new_waterfalls(&self.server_url, self.network)?;
                Ok(BlockchainClient::Esplora(waterfalls_client))
            }
            _ => Err(Error::Generic(format!(
                "Unsupported server type: {}",
                self.server_type
            ))),
        }
    }

    pub fn amp2_client(&self) -> Result<Amp2, Error> {
        if self.is_mainnet() {
            return Err(Error::Generic(
                "AMP2 methods are not available for mainnet".into(),
            ));
        }

        if self.amp2_url.is_empty() {
            return Err(Error::Generic(
                "on regtest you have to specify --amp2-url".into(),
            ));
        }

        if self.amp2_keyorigin_xpub.is_empty() {
            return Err(Error::Generic(
                "on regtest you have to specify --amp2-keyorigin-xpub".into(),
            ));
        }

        let amp2_keyorigin_xpub = self.amp2_keyorigin_xpub.clone();
        let amp2_url = self.amp2_url.clone();

        let amp_client = Amp2::new(amp2_keyorigin_xpub, amp2_url)?;

        Ok(amp_client)
    }
}