lssa/wallet/src/config.rs

use std::{
    collections::HashMap,
    io::{BufReader, Write as _},
    path::Path,
    time::Duration,
};

use anyhow::{Context as _, Result};
use common::config::BasicAuth;
use humantime_serde;
use key_protocol::key_management::key_tree::{
    chain_index::ChainIndex, keys_private::ChildKeysPrivate, keys_public::ChildKeysPublic,
};
use log::warn;
use serde::{Deserialize, Serialize};
use testnet_initial_state::{
    PrivateAccountPrivateInitialData, PublicAccountPrivateInitialData,
    initial_priv_accounts_private_keys, initial_pub_accounts_private_keys,
};
use url::Url;

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PersistentAccountDataPublic {
    pub account_id: nssa::AccountId,
    pub chain_index: ChainIndex,
    pub data: Option<ChildKeysPublic>, // `None` when Keycard is used.
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PersistentAccountDataPrivate {
    pub identifiers: Vec<nssa_core::Identifier>,
    pub chain_index: ChainIndex,
    pub data: ChildKeysPrivate,
}

// Big difference in enum variants sizes
// however it is improbable, that we will have that much accounts, that it will substantialy affect
// memory
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum InitialAccountData {
    Public(PublicAccountPrivateInitialData),
    Private(Box<PrivateAccountPrivateInitialData>),
}

impl InitialAccountData {
    #[must_use]
    pub fn account_id(&self) -> nssa::AccountId {
        match &self {
            Self::Public(acc) => acc.account_id,
            Self::Private(acc) => acc.account_id(),
        }
    }

    pub(crate) fn create_initial_accounts_data() -> Vec<Self> {
        let pub_data = initial_pub_accounts_private_keys();
        let priv_data = initial_priv_accounts_private_keys();

        pub_data
            .into_iter()
            .map(Into::into)
            .chain(priv_data.into_iter().map(Into::into))
            .collect()
    }
}

// Big difference in enum variants sizes
// however it is improbable, that we will have that much accounts, that it will substantialy affect
// memory
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum PersistentAccountData {
    Public(PersistentAccountDataPublic),
    Private(Box<PersistentAccountDataPrivate>),
    Preconfigured(InitialAccountData),
}

/// A human-readable label for an account.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct Label(String);

impl Label {
    #[must_use]
    pub const fn new(label: String) -> Self {
        Self(label)
    }
}

impl std::fmt::Display for Label {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PersistentStorage {
    pub accounts: Vec<PersistentAccountData>,
    pub last_synced_block: u64,
    /// Account labels keyed by account ID string (e.g.,
    /// "2rnKprXqWGWJTkDZKsQbFXa4ctKRbapsdoTKQFnaVGG8").
    #[serde(default)]
    pub labels: HashMap<String, Label>,
}

impl PersistentStorage {
    pub fn from_path(path: &Path) -> Result<Self> {
        #[expect(
            clippy::wildcard_enum_match_arm,
            reason = "We want to provide a specific error message for not found case"
        )]
        match std::fs::File::open(path) {
            Ok(file) => {
                let storage_content = BufReader::new(file);
                Ok(serde_json::from_reader(storage_content)?)
            }
            Err(err) => match err.kind() {
                std::io::ErrorKind::NotFound => {
                    anyhow::bail!("Not found, please setup roots from config command beforehand");
                }
                _ => {
                    anyhow::bail!("IO error {err:#?}");
                }
            },
        }
    }
}

impl From<PublicAccountPrivateInitialData> for InitialAccountData {
    fn from(value: PublicAccountPrivateInitialData) -> Self {
        Self::Public(value)
    }
}

impl From<PrivateAccountPrivateInitialData> for InitialAccountData {
    fn from(value: PrivateAccountPrivateInitialData) -> Self {
        Self::Private(Box::new(value))
    }
}

impl From<PersistentAccountDataPublic> for PersistentAccountData {
    fn from(value: PersistentAccountDataPublic) -> Self {
        Self::Public(value)
    }
}

impl From<PersistentAccountDataPrivate> for PersistentAccountData {
    fn from(value: PersistentAccountDataPrivate) -> Self {
        Self::Private(Box::new(value))
    }
}

impl From<InitialAccountData> for PersistentAccountData {
    fn from(value: InitialAccountData) -> Self {
        Self::Preconfigured(value)
    }
}

#[cfg(test)]
mod tests {
    use key_protocol::key_management::key_tree::{
        chain_index::ChainIndex, keys_public::ChildKeysPublic,
    };
    use nssa::{AccountId, PrivateKey, PublicKey};

    use super::PersistentAccountDataPublic;

    // Root public account keys derived from a known test seed; see key_protocol's keys_public
    // tests.
    const CSK_BYTES: [u8; 32] = [
        40, 35, 239, 19, 53, 178, 250, 55, 115, 12, 34, 3, 153, 153, 72, 170, 190, 36, 172, 36,
        202, 148, 181, 228, 35, 222, 58, 84, 156, 24, 146, 86,
    ];
    const CPK_BYTES: [u8; 32] = [
        219, 141, 130, 105, 11, 203, 187, 124, 112, 75, 223, 22, 11, 164, 153, 127, 59, 247, 244,
        166, 75, 66, 242, 224, 35, 156, 161, 75, 41, 51, 76, 245,
    ];
    const CCC: [u8; 32] = [
        238, 94, 84, 154, 56, 224, 80, 218, 133, 249, 179, 222, 9, 24, 17, 252, 120, 127, 222, 13,
        146, 126, 232, 239, 113, 9, 194, 219, 190, 48, 187, 155,
    ];

    fn make_child_keys_public() -> ChildKeysPublic {
        ChildKeysPublic {
            csk: PrivateKey::try_new(CSK_BYTES).unwrap(),
            cpk: PublicKey::try_new(CPK_BYTES).unwrap(),
            ccc: CCC,
            cci: None,
        }
    }

    fn make_public_account_data(data: Option<ChildKeysPublic>) -> PersistentAccountDataPublic {
        PersistentAccountDataPublic {
            account_id: AccountId::new([0_u8; 32]),
            chain_index: ChainIndex::root(),
            data,
        }
    }

    #[test]
    fn persistent_account_data_public_roundtrip_some() {
        let original = make_public_account_data(Some(make_child_keys_public()));
        let json = serde_json::to_string(&original).unwrap();
        let decoded: PersistentAccountDataPublic = serde_json::from_str(&json).unwrap();
        let keys = decoded.data.expect("data should be Some after roundtrip");
        assert_eq!(keys.csk, PrivateKey::try_new(CSK_BYTES).unwrap());
        assert_eq!(keys.cpk, PublicKey::try_new(CPK_BYTES).unwrap());
        assert_eq!(keys.ccc, CCC);
        assert_eq!(keys.cci, None);
    }

    #[test]
    fn persistent_account_data_public_roundtrip_none() {
        let original = make_public_account_data(None);
        let json = serde_json::to_string(&original).unwrap();
        let decoded: PersistentAccountDataPublic = serde_json::from_str(&json).unwrap();
        assert!(decoded.data.is_none());
    }

    #[test]
    fn persistent_account_data_public_legacy_shape_deserializes() {
        let legacy_json = r#"{
            "account_id": "11111111111111111111111111111111",
            "chain_index": [],
            "data": {
                "csk": "2823ef1335b2fa37730c2203999948aabe24ac24ca94b5e423de3a549c189256",
                "cpk": "db8d82690bcbbb7c704bdf160ba4997f3bf7f4a64b42f2e0239ca14b29334cf5",
                "ccc": [238,94,84,154,56,224,80,218,133,249,179,222,9,24,17,252,120,127,222,13,146,126,232,239,113,9,194,219,190,48,187,155],
                "cci": null
            }
        }"#;
        let decoded: PersistentAccountDataPublic = serde_json::from_str(legacy_json).unwrap();
        let keys = decoded.data.expect("legacy shape deserializes as Some");
        assert_eq!(keys.csk, PrivateKey::try_new(CSK_BYTES).unwrap());
        assert_eq!(keys.cpk, PublicKey::try_new(CPK_BYTES).unwrap());
        assert_eq!(keys.ccc, CCC);
        assert_eq!(keys.cci, None);
    }
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GasConfig {
    /// Gas spent per deploying one byte of data.
    pub gas_fee_per_byte_deploy: u64,
    /// Gas spent per reading one byte of data in VM.
    pub gas_fee_per_input_buffer_runtime: u64,
    /// Gas spent per one byte of contract data in runtime.
    pub gas_fee_per_byte_runtime: u64,
    /// Cost of one gas of runtime in public balance.
    pub gas_cost_runtime: u64,
    /// Cost of one gas of deployment in public balance.
    pub gas_cost_deploy: u64,
    /// Gas limit for deployment.
    pub gas_limit_deploy: u64,
    /// Gas limit for runtime.
    pub gas_limit_runtime: u64,
}

#[optfield::optfield(pub WalletConfigOverrides, rewrap, attrs = (derive(Debug, Default, Clone)))]
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WalletConfig {
    /// Sequencer URL.
    pub sequencer_addr: Url,
    /// Sequencer polling duration for new blocks.
    #[serde(with = "humantime_serde")]
    pub seq_poll_timeout: Duration,
    /// Sequencer polling max number of blocks to find transaction.
    pub seq_tx_poll_max_blocks: usize,
    /// Sequencer polling max number error retries.
    pub seq_poll_max_retries: u64,
    /// Max amount of blocks to poll in one request.
    pub seq_block_poll_max_amount: u64,
    /// Basic authentication credentials
    #[serde(skip_serializing_if = "Option::is_none")]
    pub basic_auth: Option<BasicAuth>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub initial_accounts: Option<Vec<InitialAccountData>>,
}

impl Default for WalletConfig {
    fn default() -> Self {
        Self {
            sequencer_addr: "http://127.0.0.1:3040".parse().unwrap(),
            seq_poll_timeout: Duration::from_secs(12),
            seq_tx_poll_max_blocks: 5,
            seq_poll_max_retries: 5,
            seq_block_poll_max_amount: 100,
            basic_auth: None,
            initial_accounts: None,
        }
    }
}

impl WalletConfig {
    pub fn from_path_or_initialize_default(config_path: &Path) -> Result<Self> {
        match std::fs::File::open(config_path) {
            Ok(file) => {
                let reader = std::io::BufReader::new(file);
                Ok(serde_json::from_reader(reader)?)
            }
            Err(err) if err.kind() == std::io::ErrorKind::NotFound => {
                println!("Config not found, setting up default config");

                let config_home = config_path.parent().ok_or_else(|| {
                    anyhow::anyhow!(
                        "Could not get parent directory of config file at {}",
                        config_path.display()
                    )
                })?;
                std::fs::create_dir_all(config_home)?;

                println!("Created configs dir at path {}", config_home.display());

                let mut file = std::fs::OpenOptions::new()
                    .write(true)
                    .create(true)
                    .truncate(true)
                    .open(config_path)?;

                let config = Self::default();
                let default_config_serialized = serde_json::to_vec_pretty(&config).unwrap();

                file.write_all(&default_config_serialized)?;

                println!("Configs set up");
                Ok(config)
            }
            Err(err) => Err(err).context("IO error"),
        }
    }

    pub fn apply_overrides(&mut self, overrides: WalletConfigOverrides) {
        let Self {
            sequencer_addr,
            seq_poll_timeout,
            seq_tx_poll_max_blocks,
            seq_poll_max_retries,
            seq_block_poll_max_amount,
            basic_auth,
            initial_accounts,
        } = self;

        let WalletConfigOverrides {
            sequencer_addr: o_sequencer_addr,
            seq_poll_timeout: o_seq_poll_timeout,
            seq_tx_poll_max_blocks: o_seq_tx_poll_max_blocks,
            seq_poll_max_retries: o_seq_poll_max_retries,
            seq_block_poll_max_amount: o_seq_block_poll_max_amount,
            basic_auth: o_basic_auth,
            initial_accounts: o_initial_accounts,
        } = overrides;

        if let Some(v) = o_sequencer_addr {
            warn!("Overriding wallet config 'sequencer_addr' to {v}");
            *sequencer_addr = v;
        }
        if let Some(v) = o_seq_poll_timeout {
            warn!("Overriding wallet config 'seq_poll_timeout' to {v:?}");
            *seq_poll_timeout = v;
        }
        if let Some(v) = o_seq_tx_poll_max_blocks {
            warn!("Overriding wallet config 'seq_tx_poll_max_blocks' to {v}");
            *seq_tx_poll_max_blocks = v;
        }
        if let Some(v) = o_seq_poll_max_retries {
            warn!("Overriding wallet config 'seq_poll_max_retries' to {v}");
            *seq_poll_max_retries = v;
        }
        if let Some(v) = o_seq_block_poll_max_amount {
            warn!("Overriding wallet config 'seq_block_poll_max_amount' to {v}");
            *seq_block_poll_max_amount = v;
        }
        if let Some(v) = o_basic_auth {
            warn!("Overriding wallet config 'basic_auth' to {v:#?}");
            *basic_auth = v;
        }
        if let Some(v) = o_initial_accounts {
            warn!("Overriding wallet config 'initial_accounts' to {v:#?}");
            *initial_accounts = v;
        }
    }
}