Create mirror accounts for committee members

input-output-hk · May 13, 2021 · 257c2d7 · 257c2d7
1 parent 0eb9f65
commit 257c2d7
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/catalyst-toolbox-lib/src/recovery/tally.rs b/catalyst-toolbox-lib/src/recovery/tally.rs
@@ -284,19 +284,23 @@ impl FragmentReplayer {
 
         for initial in &mut config.initial {
             if let Initial::Fund(ref mut utxos) = initial {
+                let mut new_committee_accounts = Vec::new();
                 for utxo in utxos.iter_mut() {
                     let wallet = Wallet::new_account_with_discrimination(
                         &mut rng,
                         chain_addr::Discrimination::Production,
                     );
+                    let new_initial_utxo = wallet.to_initial_fund(utxo.value.into());
+                    wallets.insert(utxo.address.clone(), wallet);
                     if committee_members.contains(&utxo.address) {
                         eprintln!("Committee account found {}", &utxo.address);
-                        continue;
+                        // push new mirror address
+                        new_committee_accounts.push(new_initial_utxo);
+                    } else {
+                        *utxo = new_initial_utxo;
                     }
-                    let new_initial_utxo = wallet.to_initial_fund(utxo.value.into());
-                    wallets.insert(utxo.address.clone(), wallet);
-                    *utxo = new_initial_utxo;
                 }
+                utxos.append(&mut new_committee_accounts);
             }
         }