tensor_eigen/commands/
download.rs

1use anyhow::{anyhow, Result};
2use solana_sdk::pubkey::Pubkey;
3use std::fs::File;
4use std::io::Write;
5use std::path::PathBuf;
6use tensor_amm::accounts::{NftDepositReceipt, Pool};
7use tensor_marketplace::accounts::{BidState, ListState};
8use tensor_whitelist::accounts::{MintProof, MintProofV2, Whitelist, WhitelistV2};
9
10use crate::{commands::Discriminator, discriminators::deserialize_account, setup::CliConfig};
11
12pub struct DownloadParams {
13    pub rpc_url: Option<String>,
14    pub address: Pubkey,
15    pub output_dir: Option<PathBuf>,
16}
17
18pub fn handle_download(args: DownloadParams) -> Result<()> {
19    let config = CliConfig::new(None, args.rpc_url)?;
20
21    let data = config.client.get_account_data(&args.address)?;
22    let discriminator = &data[0..8];
23
24    let (account_data, file_name) = match discriminator {
25        d if d == Pool::discriminator() => {
26            let pool = deserialize_account::<Pool>(&data)?;
27            (serde_json::to_value(&pool)?, "pool.json")
28        }
29        d if d == NftDepositReceipt::discriminator() => {
30            let nft_deposit_receipt = deserialize_account::<NftDepositReceipt>(&data)?;
31            (
32                serde_json::to_value(&nft_deposit_receipt)?,
33                "nft_deposit_receipt.json",
34            )
35        }
36        d if d == Whitelist::discriminator() => {
37            let whitelist = deserialize_account::<Whitelist>(&data)?;
38            (serde_json::to_value(&whitelist)?, "whitelist.json")
39        }
40        d if d == WhitelistV2::discriminator() => {
41            let whitelist_v2 = deserialize_account::<WhitelistV2>(&data)?;
42            (serde_json::to_value(&whitelist_v2)?, "whitelist_v2.json")
43        }
44        d if d == MintProof::discriminator() => {
45            let mint_proof = deserialize_account::<MintProof>(&data)?;
46            (serde_json::to_value(&mint_proof)?, "mint_proof.json")
47        }
48        d if d == MintProofV2::discriminator() => {
49            let mint_proof_v2 = deserialize_account::<MintProofV2>(&data)?;
50            (serde_json::to_value(&mint_proof_v2)?, "mint_proof_v2.json")
51        }
52        d if d == BidState::discriminator() => {
53            let bid_state = deserialize_account::<BidState>(&data)?;
54            (serde_json::to_value(&bid_state)?, "bid_state.json")
55        }
56        d if d == ListState::discriminator() => {
57            let list_state = deserialize_account::<ListState>(&data)?;
58            (serde_json::to_value(&list_state)?, "list_state.json")
59        }
60        _ => return Err(anyhow!("Unsupported discriminator")),
61    };
62
63    let output_path = args.output_dir.unwrap_or_else(|| PathBuf::from("."));
64    let mut file = File::create(output_path.join(file_name))?;
65    let json_string = serde_json::to_string_pretty(&account_data)?;
66    file.write_all(json_string.as_bytes())?;
67
68    println!("Account data saved to {}", output_path.display());
69
70    Ok(())
71}
tensor_eigen/commands/download.rs

tensor_eigen/commands/
download.rs
