chore: extract crate utxo nodes

bob-collective · Dec 13, 2024 · 6de85a7 · 6de85a7
1 parent b9a8ce8
commit 6de85a7
Showing 1 changed file with 15 additions and 30 deletions.
diff --git a/sdk/src/wallet/utxo.ts b/sdk/src/wallet/utxo.ts
@@ -28,6 +28,18 @@ class TreeNode<T> {
     }
 }
 
+const createUtxoNodes = (utxos: UTXO[], cardinalOutputsSet: Set<string>) =>
+    utxos.reduce(
+        (acc, utxo) => {
+            if (!cardinalOutputsSet.has(OutPoint.toString(utxo)))
+                acc.push(new TreeNode<OutputNodeData>({ ...utxo, cardinal: true }));
+            else acc.push(null);
+
+            return acc;
+        },
+        [] as (TreeNode<OutputNodeData> | null)[]
+    );
+
 const processNodes = async (rootNodes: (TreeNode<OutputNodeData> | null)[], esploraClient: EsploraClient) => {
     const queue = Array.from(rootNodes);
 
@@ -159,16 +171,7 @@ export async function createBitcoinPsbt(
 
     const cardinalOutputsSet = new Set(cardinalOutputs.map((output) => output.outpoint));
 
-    const rootUtxoNodes = utxos.reduce(
-        (acc, utxo) => {
-            if (!cardinalOutputsSet.has(OutPoint.toString(utxo)))
-                acc.push(new TreeNode<OutputNodeData>({ ...utxo, cardinal: true }));
-            else acc.push(null);
-
-            return acc;
-        },
-        [] as (TreeNode<OutputNodeData> | null)[]
-    );
+    const rootUtxoNodes = createUtxoNodes(utxos, cardinalOutputsSet);
 
     await processNodes(rootUtxoNodes, esploraClient);
 
@@ -384,16 +387,7 @@ export async function estimateTxFee(
 
     const cardinalOutputsSet = new Set(cardinalOutputs.map((output) => output.outpoint));
 
-    const rootUtxoNodes = utxos.reduce(
-        (acc, utxo) => {
-            if (!cardinalOutputsSet.has(OutPoint.toString(utxo)))
-                acc.push(new TreeNode<OutputNodeData>({ ...utxo, cardinal: true }));
-            else acc.push(null);
-
-            return acc;
-        },
-        [] as (TreeNode<OutputNodeData> | null)[]
-    );
+    const rootUtxoNodes = createUtxoNodes(utxos, cardinalOutputsSet);
 
     await processNodes(rootUtxoNodes, esploraClient);
 
@@ -509,16 +503,7 @@ export async function getBalance(address?: string) {
 
     const cardinalOutputsSet = new Set(cardinalOutputs.map((output) => output.outpoint));
 
-    const rootUtxoNodes = utxos.reduce(
-        (acc, utxo) => {
-            if (!cardinalOutputsSet.has(OutPoint.toString(utxo)))
-                acc.push(new TreeNode<OutputNodeData>({ ...utxo, cardinal: true }));
-            else acc.push(null);
-
-            return acc;
-        },
-        [] as (TreeNode<OutputNodeData> | null)[]
-    );
+    const rootUtxoNodes = createUtxoNodes(utxos, cardinalOutputsSet);
 
     await processNodes(rootUtxoNodes, esploraClient);