fuel-core 0.15.3

use crate::cli::DEFAULT_DB_PATH;
use clap::Parser;
use std::path::PathBuf;

#[derive(Debug, Clone, Parser)]
pub struct Command {
    #[clap(
        name = "DB_PATH",
        long = "db-path",
        parse(from_os_str),
        default_value = (*DEFAULT_DB_PATH).to_str().unwrap()
    )]
    pub database_path: PathBuf,

    /// Specify either an alias to a built-in configuration or filepath to a JSON file.
    #[clap(name = "CHAIN_CONFIG", long = "chain", default_value = "local_testnet")]
    pub chain_config: String,
}

#[cfg(not(feature = "rocksdb"))]
pub async fn exec(command: Command) -> anyhow::Result<()> {
    Err(anyhow::anyhow!(
        "Rocksdb must be enabled to use the database at {}",
        command.database_path.display()
    ))
}

#[cfg(feature = "rocksdb")]
pub async fn exec(command: Command) -> anyhow::Result<()> {
    use anyhow::Context;
    use fuel_chain_config::{
        ChainConfig,
        StateConfig,
    };
    use fuel_core::database::Database;

    let path = command.database_path;
    let config: ChainConfig = command.chain_config.parse()?;
    let db = Database::open(&path).context(format!(
        "failed to open database at path {}",
        path.display()
    ))?;

    let state_conf = StateConfig::generate_state_config(db)?;

    let chain_conf = ChainConfig {
        initial_state: Some(state_conf),
        ..config
    };

    let stdout = std::io::stdout().lock();

    serde_json::to_writer(stdout, &chain_conf)
        .context("failed to dump snapshot to JSON")?;
    Ok(())
}