From e442793a7dcbbd4a303bfa26c863db571477e0ac Mon Sep 17 00:00:00 2001
From: Dhanraj30 <dhanraj30102002@gmail.com>
Date: Wed, 20 Nov 2024 13:12:02 +0530
Subject: [PATCH] fix:support-classic-accounts-trustlines

---
 .../src/commands/snapshot/create.rs           | 34 ++++++++++++-------
 1 file changed, 21 insertions(+), 13 deletions(-)

diff --git a/cmd/soroban-cli/src/commands/snapshot/create.rs b/cmd/soroban-cli/src/commands/snapshot/create.rs
index 6ce48d3f2..5150304d5 100644
--- a/cmd/soroban-cli/src/commands/snapshot/create.rs
+++ b/cmd/soroban-cli/src/commands/snapshot/create.rs
@@ -20,7 +20,7 @@ use stellar_xdr::curr::{
     LedgerKeyClaimableBalance, LedgerKeyConfigSetting, LedgerKeyContractCode,
     LedgerKeyContractData, LedgerKeyData, LedgerKeyLiquidityPool, LedgerKeyOffer,
     LedgerKeyTrustLine, LedgerKeyTtl, Limited, Limits, ReadXdr, ScAddress, ScContractInstance,
-    ScVal,
+    ScVal, TrustLineAsset,
 };
 use tokio::fs::OpenOptions;
 use tokio::io::BufReader;
@@ -283,7 +283,14 @@ impl Cmd {
                     }
                     let keep = match &key {
                         LedgerKey::Account(k) => current.account_ids.contains(&k.account_id),
-                        LedgerKey::Trustline(k) => current.account_ids.contains(&k.account_id),
+                        LedgerKey::Trustline(k) => {
+                            current.account_ids.contains(&k.account_id) ||
+                            current.account_ids.iter().any(|id| match &k.asset {
+                                TrustLineAsset::CreditAlphanum4(a4) => a4.issuer == *id,
+                                TrustLineAsset::CreditAlphanum12(a12) => a12.issuer == *id,
+                                _ => false,  // Ignore non-credit assets
+                            })
+                        },
                         LedgerKey::ContractData(k) => current.contract_ids.contains(&k.contract),
                         LedgerKey::ContractCode(e) => current.wasm_hashes.contains(&e.hash),
                         _ => false,
@@ -293,7 +300,7 @@ impl Cmd {
                     }
                     seen.insert(key.clone());
                     let Some(val) = val else { continue };
-                    match &val.data {
+                    let should_save = match &val.data {
                         LedgerEntryData::ContractData(e) => {
                             // If a contract instance references contract
                             // executable stored in another ledger entry, add
@@ -313,7 +320,7 @@ impl Cmd {
                                                 hex::encode(hash)
                                             ));
                                         }
-                                    }
+                                    } 
                                     ScVal::ContractInstance(ScContractInstance {
                                         executable: ContractExecutable::StellarAsset,
                                         storage: Some(storage),
@@ -327,9 +334,7 @@ impl Cmd {
                                                 Asset::CreditAlphanum4(a4) => Some(a4.issuer),
                                                 Asset::CreditAlphanum12(a12) => Some(a12.issuer),
                                             } {
-                                                print.infoln(format!(
-                                                    "Adding asset issuer {issuer} to search"
-                                                ));
+                                                print.infoln(format!("Adding asset issuer {issuer} to search"));
                                                 next.account_ids.insert(issuer);
                                             }
                                         }
@@ -337,14 +342,17 @@ impl Cmd {
                                     _ => {}
                                 }
                             }
-                            keep
+                            true
                         }
-                        _ => false,
+                        LedgerEntryData::ContractCode(_) => true, 
+                        _ => false
                     };
-                    snapshot
-                        .ledger_entries
-                        .push((Box::new(key), (Box::new(val), Some(u32::MAX))));
-                    count_saved += 1;
+                    if should_save {
+                        snapshot
+                            .ledger_entries
+                            .push((Box::new(key), (Box::new(val), Some(u32::MAX))));
+                        count_saved += 1;
+                    }
                 }
                 if count_saved > 0 {
                     print.infoln(format!("Found {count_saved} entries"));