From c4ac2c15c584dd781989cbf496461fe96b66b0cb Mon Sep 17 00:00:00 2001
From: Jason <jayvdb1@gmail.com>
Date: Thu, 21 Mar 2024 11:49:52 +0200
Subject: [PATCH] feat: ios list all channel monitors

---
 example/Dev.tsx        |  17 +++++++
 lib/ios/Helpers.swift  | 113 ++++++++++++++++++++++-------------------
 lib/ios/Ldk.m          |   3 ++
 lib/ios/Ldk.swift      |  45 +++++++++++++++-
 lib/src/ldk.ts         |  19 +++++++
 lib/src/utils/types.ts |   7 +++
 6 files changed, 152 insertions(+), 52 deletions(-)

diff --git a/example/Dev.tsx b/example/Dev.tsx
index 4d8da97b..ab2086f6 100644
--- a/example/Dev.tsx
+++ b/example/Dev.tsx
@@ -587,6 +587,23 @@ const Dev = (): ReactElement => {
 						}}
 					/>
 
+					<Button
+						title={'List channel monitors'}
+						onPress={async (): Promise<void> => {
+							const monitorsRes = await ldk.listChannelMonitors(true);
+							if (monitorsRes.isErr()) {
+								return setMessage(monitorsRes.error.message);
+							}
+
+							let msg = `Channel Monitors (${monitorsRes.value.length}): \n`;
+							monitorsRes.value.forEach((monitor) => {
+								msg += `\n\n${JSON.stringify(monitor)}`;
+							});
+
+							setMessage(msg);
+						}}
+					/>
+
 					<Button
 						title={'Show version'}
 						onPress={async (): Promise<void> => {
diff --git a/lib/ios/Helpers.swift b/lib/ios/Helpers.swift
index a8b1705c..f27a59c1 100644
--- a/lib/ios/Helpers.swift
+++ b/lib/ios/Helpers.swift
@@ -188,6 +188,17 @@ extension LightningDevKit.RouteHop {
     }
 }
 
+extension ChannelMonitor {
+    func asJson(channelId: String) -> [String: Any?] {
+        return [
+            "channel_id": channelId,
+            "funding_txo": Data(getFundingTxo().1).hexEncodedString(),
+            "counterparty_node_id": Data(getCounterpartyNodeId() ?? []).hexEncodedString(),
+            "claimable_balances": getClaimableBalances().map({ $0.asJson })
+        ]
+    }
+}
+
 extension Data {
     struct HexEncodingOptions: OptionSet {
         let rawValue: Int
@@ -402,63 +413,63 @@ extension UserConfig {
     }
 }
 
+extension Balance {
+    var asJson: [String: Any] {
+        switch getValueType() {
+        case .ClaimableAwaitingConfirmations:
+            let b = getValueAsClaimableAwaitingConfirmations()!
+            return [
+                "amount_satoshis": b.getAmountSatoshis(),
+                "confirmation_height": b.getConfirmationHeight(),
+                "type": "ClaimableAwaitingConfirmations"
+            ] as [String : Any]
+        case .ClaimableOnChannelClose:
+            let b = getValueAsClaimableOnChannelClose()!
+            return [
+                "amount_satoshis": b.getAmountSatoshis(),
+                "type": "ClaimableOnChannelClose",
+            ] as [String : Any]
+        case .ContentiousClaimable:
+            let b = getValueAsContentiousClaimable()!
+            return [
+                "amount_satoshis": b.getAmountSatoshis(),
+                "timeout_height": b.getTimeoutHeight(),
+                "type": "ContentiousClaimable"
+            ] as [String : Any]
+        case .CounterpartyRevokedOutputClaimable:
+            let b = getValueAsCounterpartyRevokedOutputClaimable()!
+            return [
+                "amount_satoshis": b.getAmountSatoshis(),
+                "type": "CounterpartyRevokedOutputClaimable"
+            ] as [String : Any]
+        case .MaybePreimageClaimableHTLC:
+            let b = getValueAsMaybePreimageClaimableHtlc()!
+            return [
+                "amount_satoshis": b.getAmountSatoshis(),
+                "expiry_height": b.getExpiryHeight(),
+                "type": "MaybePreimageClaimableHTLC"
+            ] as [String : Any]
+        case .MaybeTimeoutClaimableHTLC:
+            let b = getValueAsMaybeTimeoutClaimableHtlc()!
+            return [
+                "amount_satoshis": b.getAmountSatoshis(),
+                "claimable_height": b.getClaimableHeight(),
+                "type": "MaybeTimeoutClaimableHTLC"
+            ] as [String : Any]
+        default:
+            LdkEventEmitter.shared.send(withEvent: .native_log, body: "Unknown balance type type in claimableBalances() \(getValueType())")
+            return ["amount_satoshis": 0, "type": "Unknown"] as [String : Any]
+        }
+    }
+}
+
 extension ChainMonitor {
     func getClaimableBalancesAsJson(ignoredChannels: [Bindings.ChannelDetails]) -> [[String: Any]] {
         var result: [[String: Any]] = []
 
         let claimableBalances = self.getClaimableBalances(ignoredChannels: ignoredChannels)
         for balance in claimableBalances {
-            switch balance.getValueType() {
-            case .ClaimableAwaitingConfirmations:
-                let b = balance.getValueAsClaimableAwaitingConfirmations()!
-                result.append([
-                    "amount_satoshis": b.getAmountSatoshis(),
-                    "confirmation_height": b.getConfirmationHeight(),
-                    "type": "ClaimableAwaitingConfirmations"
-                ] as [String : Any])
-                break
-            case .ClaimableOnChannelClose:
-                let b = balance.getValueAsClaimableOnChannelClose()!
-                result.append([
-                    "amount_satoshis": b.getAmountSatoshis(),
-                    "type": "ClaimableOnChannelClose",
-                ] as [String : Any])
-                break
-            case .ContentiousClaimable:
-                let b = balance.getValueAsContentiousClaimable()!
-                result.append([
-                    "amount_satoshis": b.getAmountSatoshis(),
-                    "timeout_height": b.getTimeoutHeight(),
-                    "type": "ContentiousClaimable"
-                ] as [String : Any])
-                break
-            case .CounterpartyRevokedOutputClaimable:
-                let b = balance.getValueAsCounterpartyRevokedOutputClaimable()!
-                result.append([
-                    "amount_satoshis": b.getAmountSatoshis(),
-                    "type": "CounterpartyRevokedOutputClaimable"
-                ] as [String : Any])
-                break
-            case .MaybePreimageClaimableHTLC:
-                let b = balance.getValueAsMaybePreimageClaimableHtlc()!
-                result.append([
-                    "amount_satoshis": b.getAmountSatoshis(),
-                    "expiry_height": b.getExpiryHeight(),
-                    "type": "MaybePreimageClaimableHTLC"
-                ] as [String : Any])
-                break
-            case .MaybeTimeoutClaimableHTLC:
-                let b = balance.getValueAsMaybeTimeoutClaimableHtlc()!
-                result.append([
-                    "amount_satoshis": b.getAmountSatoshis(),
-                    "claimable_height": b.getClaimableHeight(),
-                    "type": "MaybeTimeoutClaimableHTLC"
-                ] as [String : Any])
-                break
-            default:
-                LdkEventEmitter.shared.send(withEvent: .native_log, body: "Unknown balance type type in claimableBalances() \(balance.getValueType())")
-                result.append(["amount_satoshis": 0, "type": "Unknown"] as [String : Any])
-            }
+            result.append(balance.asJson)
         }
         
         return result
diff --git a/lib/ios/Ldk.m b/lib/ios/Ldk.m
index a8444df2..f1d781c9 100644
--- a/lib/ios/Ldk.m
+++ b/lib/ios/Ldk.m
@@ -109,6 +109,9 @@ @interface RCT_EXTERN_MODULE(Ldk, NSObject)
                   reject:(RCTPromiseRejectBlock)reject)
 RCT_EXTERN_METHOD(listChannelFiles:(RCTPromiseResolveBlock)resolve
                   reject:(RCTPromiseRejectBlock)reject)
+RCT_EXTERN_METHOD(listChannelMonitors:(BOOL *)ignoreOpenChannels
+                  resolve:(RCTPromiseResolveBlock)resolve
+                  reject:(RCTPromiseRejectBlock)reject)
 RCT_EXTERN_METHOD(networkGraphListNodeIds:(RCTPromiseResolveBlock)resolve
                   reject:(RCTPromiseRejectBlock)reject)
 RCT_EXTERN_METHOD(networkGraphListChannels:(RCTPromiseResolveBlock)resolve
diff --git a/lib/ios/Ldk.swift b/lib/ios/Ldk.swift
index e0667477..46a89397 100644
--- a/lib/ios/Ldk.swift
+++ b/lib/ios/Ldk.swift
@@ -1014,6 +1014,49 @@ class Ldk: NSObject {
         return resolve(try! FileManager.default.contentsOfDirectory(at: channelStoragePath, includingPropertiesForKeys: nil).map { $0.lastPathComponent })
     }
     
+    @objc
+    func listChannelMonitors(_ ignoreOpenChannels: Bool, resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock) {
+        guard let channelManager else {
+            return handleReject(reject, .init_channel_manager)
+        }
+        
+        guard let keysManager else {
+            return handleReject(reject, .init_keys_manager)
+        }
+        
+        guard let channelStoragePath = Ldk.channelStoragePath else {
+            return handleReject(reject, .init_storage_path)
+        }
+        
+        let excludeChannelIds = ignoreOpenChannels ? channelManager.listChannels().map { Data($0.getChannelId() ?? []).hexEncodedString() }.filter { $0 != "" } : []
+
+        let channelFiles = try! FileManager.default.contentsOfDirectory(at: channelStoragePath, includingPropertiesForKeys: nil)
+        
+        var result: [[String: Any?]] = []
+        for channelFile in channelFiles {
+            let channelId = channelFile.lastPathComponent.replacingOccurrences(of: ".bin", with: "")
+            
+            guard !excludeChannelIds.contains(channelId) else {
+                continue
+            }
+            
+            let channelMonitorResult = Bindings.readThirtyTwoBytesChannelMonitor(
+                ser: [UInt8](try! Data(contentsOf: channelFile.standardizedFileURL)),
+                argA: keysManager.inner.asEntropySource(),
+                argB: keysManager.signerProvider
+            )
+            
+            guard let (channelId, channelMonitor) = channelMonitorResult.getValue() else {
+                LdkEventEmitter.shared.send(withEvent: .native_log, body: "Loading channel error. No channel value.")
+                continue
+            }
+            
+            result.append(channelMonitor.asJson(channelId: Data(channelId).hexEncodedString()))
+        }
+        
+        return resolve(result)
+    }
+    
     @objc
     func networkGraphListNodeIds(_ resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock) {
         guard let networkGraph = networkGraph?.readOnly() else {
@@ -1203,7 +1246,7 @@ class Ldk: NSObject {
     
     @objc
     func spendRecoveredForceCloseOutputs(_ transaction: NSString, confirmationHeight: NSInteger, changeDestinationScript: NSString, resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock) {
-        //TODO check which ones are not open channels and try spend them again
+        
         guard let channelStoragePath = Ldk.channelStoragePath, let keysManager, let channelManager else {
             return handleReject(reject, .init_storage_path)
         }
diff --git a/lib/src/ldk.ts b/lib/src/ldk.ts
index 0fd5a2ca..92fc9fcc 100644
--- a/lib/src/ldk.ts
+++ b/lib/src/ldk.ts
@@ -38,6 +38,7 @@ import {
 	TDownloadScorer,
 	TInitKeysManager,
 	TSpendRecoveredForceCloseOutputsReq,
+	TChannelMonitor,
 } from './utils/types';
 import { extractPaymentRequest } from './utils/helpers';
 
@@ -957,6 +958,24 @@ class LDK {
 		}
 	}
 
+	/**
+	 * Lists all channel monitors from storage
+	 * @param ignoreOpenChannels
+	 * @returns {Promise<Ok<Ok<string[]> | Err<string[]>> | Err<unknown>>}
+	 */
+	async listChannelMonitors(
+		ignoreOpenChannels: boolean,
+	): Promise<Result<TChannelMonitor[]>> {
+		try {
+			const res = await NativeLDK.listChannelMonitors(ignoreOpenChannels);
+			this.writeDebugToLog('listChannelMonitors', `Monitors: ${res.length}`);
+			return ok(res);
+		} catch (e) {
+			this.writeErrorToLog('listChannelMonitors', e);
+			return err(e);
+		}
+	}
+
 	/**
 	 * Fetches list of all node IDs in network graph
 	 * https://docs.rs/lightning/latest/lightning/routing/gossip/struct.ReadOnlyNetworkGraph.html#method.nodes
diff --git a/lib/src/utils/types.ts b/lib/src/utils/types.ts
index 7aeb44db..52ab9cd6 100644
--- a/lib/src/utils/types.ts
+++ b/lib/src/utils/types.ts
@@ -164,6 +164,13 @@ export type TChannel = {
 	confirmations: number;
 };
 
+export type TChannelMonitor = {
+	channel_id: string;
+	funding_txo: string;
+	counterparty_node_id: string;
+	claimable_balances: [TClaimableBalance];
+};
+
 export type TNetworkGraphChannelInfo = {
 	shortChannelId: string;
 	capacity_sats?: number;