bsv-wallet-cli 0.2.8

use anyhow::{Context, Result};
use bsv_sdk::primitives::PrivateKey;
use bsv_wallet_toolbox::{
    ArcConfig, Chain, Services, ServicesOptions, StorageSqlx, Wallet, WalletStorageWriter,
};

use crate::cli::Cli;

pub struct WalletContext {
    pub wallet: Wallet<StorageSqlx, Services>,
    pub identity_key: String,
    pub root_key: PrivateKey,
    pub chain: Chain,
    pub json_output: bool,
}

impl WalletContext {
    pub async fn load(cli: &Cli) -> Result<Self> {
        let root_key_hex =
            std::env::var("ROOT_KEY").context("ROOT_KEY not set. Run `bsv-wallet init` first.")?;

        let root_key = PrivateKey::from_hex(&root_key_hex)?;
        let identity_key = root_key.public_key().to_hex();

        let chain = if cli.testnet {
            Chain::Test
        } else {
            Chain::Main
        };

        let storage = StorageSqlx::open(&cli.db).await?;
        storage.make_available().await?;

        let services = {
            let mut opts = match chain {
                Chain::Main => ServicesOptions::mainnet(),
                Chain::Test => ServicesOptions::testnet(),
            };
            if let Ok(url) = std::env::var("CHAINTRACKS_URL") {
                opts = opts.with_chaintracks_url(url);
            }
            if let Ok(key) = std::env::var("TAAL_API_KEY") {
                if !key.is_empty() {
                    let arc_url = opts.arc_url.clone();
                    opts = opts.with_arc(arc_url, Some(ArcConfig::with_api_key(key)));
                }
            }
            Services::with_options(chain, opts)?
        };

        // Wire ChainTracker into storage for merkle proof validation (Layer 1)
        if let Some(ref ct) = services.chaintracks {
            storage.set_chain_tracker(ct.clone()).await;
        }

        let wallet = Wallet::new(Some(root_key.clone()), storage, services).await?;

        Ok(Self {
            wallet,
            identity_key,
            root_key,
            chain,
            json_output: cli.json,
        })
    }
}