Split select coins and get state

src/wallet/coin_store.rs

@@ -1,6 +1,6 @@
 use std::{collections::HashMap, future::Future};
 
-use chia_protocol::CoinState;
+use chia_protocol::{Coin, CoinState};
 use parking_lot::Mutex;
 
 /// Keeps track of the state of coins in a wallet.
@@ -9,7 +9,10 @@ pub trait CoinStore {
     fn update_coin_state(&self, coin_states: Vec<CoinState>) -> impl Future<Output = ()> + Send;
 
     /// Gets a list of unspent coins.
-    fn unspent_coins(&self) -> impl Future<Output = Vec<CoinState>> + Send;
+    fn unspent_coins(&self) -> impl Future<Output = Vec<Coin>> + Send;
+
+    /// Gets the current state of a coin.
+    fn coin_state(&self, coin_id: [u8; 32]) -> impl Future<Output = Option<CoinState>> + Send;
 
     /// Gets coin states for a given puzzle hash.
     fn is_used(&self, puzzle_hash: [u8; 32]) -> impl Future<Output = bool> + Send;
@@ -49,16 +52,25 @@ impl CoinStore for MemoryCoinStore {
         }
     }
 
-    async fn unspent_coins(&self) -> Vec<CoinState> {
+    async fn unspent_coins(&self) -> Vec<Coin> {
         self.coin_states
             .lock()
             .values()
             .flatten()
             .filter(|coin_state| coin_state.spent_height.is_none())
-            .cloned()
+            .map(|coin_state| coin_state.coin.clone())
             .collect()
     }
 
+    async fn coin_state(&self, coin_id: [u8; 32]) -> Option<CoinState> {
+        self.coin_states
+            .lock()
+            .values()
+            .flatten()
+            .find(|coin_state| coin_state.coin.coin_id() == coin_id)
+            .cloned()
+    }
+
     async fn is_used(&self, puzzle_hash: [u8; 32]) -> bool {
         self.coin_states
             .lock()