Fee estimator init and updates

Author: Jasonvdb

example/App.tsx

@@ -62,6 +62,34 @@ const App = () => {
               }
             }}
           />
+
+          <Button
+            title={'initFeeEstimator'}
+            onPress={async () => {
+              const res = await ldk.initFeeEstimator({
+                high: 1000,
+                normal: 500,
+                low: 100,
+              });
+              if (res.isOk()) {
+                alert(res.value);
+              }
+            }}
+          />
+
+          <Button
+            title={'updateFees'}
+            onPress={async () => {
+              const res = await ldk.updateFees({
+                high: 1000,
+                normal: 500,
+                low: 100,
+              });
+              if (res.isOk()) {
+                alert(res.value);
+              }
+            }}
+          />
         </ScrollView>
       </SafeAreaView>
     </>

ios/Classes/LdkFeeEstimator.swift

@@ -9,15 +9,14 @@ import Foundation
 import LDKFramework
 
 class LdkFeeEstimator: FeeEstimator {
-    var high: UInt32 = 0
-    var normal: UInt32 = 0
-    var low: UInt32 = 0
-
-    init(high: UInt32, normal: UInt32, low: UInt32) {
+    private var high: UInt32 = 0
+    private var normal: UInt32 = 0
+    private var low: UInt32 = 0
+    
+    func update(high: UInt32, normal: UInt32, low: UInt32) {
         self.high = high
         self.normal = normal
         self.low = low
-        super.init()
     }
 
     override func get_est_sat_per_1000_weight(confirmation_target: LDKConfirmationTarget) -> UInt32 {
@@ -35,6 +34,6 @@ class LdkFeeEstimator: FeeEstimator {
             return low
         }
 
-        return UInt32(1)
+        return normal
     }
 }

ios/Helpers.swift

@@ -7,7 +7,7 @@
 
 import Foundation
 
-//TODO replicate in typescript
+//TODO replicate in typescript and kotlin
 enum LdkEventNames: String {
     case register_tx = "register_tx"
     case register_output = "register_output"
@@ -18,6 +18,38 @@ enum LdkEventNames: String {
     case update_persisted_channel = "update_persisted_channel"
 }
 
+enum LdkError: Error {
+  case unknown_error
+  case init_fee_estimator
+}
+
+extension LdkError: LocalizedError {
+  public var errorDescription: String? {
+    switch self {
+    case .unknown_error:
+      return "unknown_error"
+    case .init_fee_estimator:
+      return "init_fee_estimator"
+    }
+  }
+}
+
+enum LdkCallbackResponses: String {
+    case fee_estimator_initialised = "fee_estimator_initialised"
+    case fees_updated = "fees_updated"
+}
+
+func handleResolve(_ resolve: RCTPromiseResolveBlock, _ res: LdkCallbackResponses) {
+    //TODO log
+    resolve(res.rawValue)
+}
+
+func handleReject(_ reject: RCTPromiseRejectBlock, _ error: LdkError) {
+    //TODO log
+    reject(error.errorDescription, error.localizedDescription, error)
+}
+
+
 func sendEvent(eventName: LdkEventNames, eventBody: [String: String]) {
 //    ReactEventEmitter.sharedInstance()?.sendEvent(withName: eventName, body: eventBody)
     print("\(eventName)")
@@ -35,3 +67,11 @@ extension Data {
         return self.map { String(format: format, $0) }.joined()
     }
 }
+
+
+extension Ldk {
+    @objc
+    static func requiresMainQueueSetup() -> Bool {
+        return true
+    }
+}

ios/Ldk.m

@@ -6,4 +6,14 @@ @interface RCT_EXTERN_MODULE(Ldk, NSObject)
 
 RCT_EXTERN_METHOD(startChainMonitor:(RCTPromiseResolveBlock)resolve reject:(RCTPromiseRejectBlock)reject)
 
+//Startup methods
+RCT_EXTERN_METHOD(initFeeEstimator:(RCTPromiseResolveBlock)resolve reject:(RCTPromiseRejectBlock)reject)
+
+//Update state methods
+RCT_EXTERN_METHOD(updateFees:(NSInteger *)high
+                  normal:(NSInteger *)normal
+                  low:(NSInteger *)low
+                  resolve:(RCTPromiseResolveBlock)resolve
+                  reject:(RCTPromiseRejectBlock)reject)
+
 @end

ios/Ldk.swift

@@ -3,7 +3,7 @@ import LDKFramework
 
 @objc(Ldk)
 class Ldk: NSObject {
-    var networkGraphPath = ""
+    var feeEstimator: LdkFeeEstimator?
 
     lazy var ldkStorage: URL = {
         let docsurl = try! FileManager.default.url(for:.documentDirectory, in: .userDomainMask, appropriateFor: nil, create: false)
@@ -26,14 +26,33 @@ class Ldk: NSObject {
         resolve(String(data: try! JSONEncoder().encode(res), encoding: .utf8)!)
     }
 
+    @objc
+    func initFeeEstimator(_ resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock) {
+        feeEstimator = LdkFeeEstimator()
+        handleResolve(resolve, .fee_estimator_initialised)
+    }
+    
+    @objc
+    func updateFees(_ high: NSInteger, normal: NSInteger, low: NSInteger, resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock) {
+        guard let feeEstimator = feeEstimator else {
+            return handleReject(reject, .init_fee_estimator)
+        }
+        
+        feeEstimator.update(high: UInt32(high), normal: UInt32(normal), low: UInt32(low))
+        handleResolve(resolve, .fees_updated)
+    }
 
     @objc
     func startChainMonitor(_ resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock) {
+        guard let feeEstimator = feeEstimator else {
+            return handleReject(reject, .init_fee_estimator)
+        }
+        
         let filter = LdkFilter()
         let broadcaster = LdkBroadcaster()
         let logger = LdkLogger()
         //TODO set these fees from the JS code. They should be able to be updated live.
-        let feeEstimator = LdkFeeEstimator(high: 1000, normal: 500, low: 100)
+        
         let persister = LdkPersister()
 
         let chainMonitor = ChainMonitor(
@@ -44,14 +63,25 @@ class Ldk: NSObject {
             persister: persister
         )
 
+        
+        
+        let seed: [UInt8] = [0] //TODO
+        
+        let seconds = UInt64(NSDate().timeIntervalSince1970)
+        let nanoSeconds = UInt32.init(truncating: NSNumber(value: seconds * 1000 * 1000))
+        
+        let keysManager = KeysManager(seed: seed, starting_time_secs: seconds, starting_time_nanos: nanoSeconds)
+        
+        let network = LDKNetwork_Bitcoin
+        
         let _ = ChannelManager(
             fee_est: feeEstimator,
             chain_monitor: Watch(),
             tx_broadcaster: broadcaster,
             logger: logger,
-            keys_manager: KeysInterface(),
+            keys_manager: keysManager.as_KeysInterface(),
             config: UserConfig(),
-            params: ChainParameters(network_arg: LDKNetwork_Bitcoin, best_block_arg: BestBlock(block_hash: [], height: 0))
+            params: ChainParameters(network_arg: network, best_block_arg: BestBlock(block_hash: [], height: 0))
         )
 
         resolve("Chain monitor started")

ios/Ldk.xcodeproj/project.xcworkspace/xcuserdata/jason.xcuserdatad/UserInterfaceState.xcuserstate

@@ -1,6 +1,6 @@
 import { NativeModules, Platform } from 'react-native';
 import { err, ok, Result } from './utils/result';
-import { TLogListener } from './utils/types';
+import { TFeeUpdateReq, TLogListener } from './utils/types';
 
 const LINKING_ERROR =
 	`The package 'react-native-ldk' doesn't seem to be linked. Make sure: \n\n` +
@@ -33,7 +33,7 @@ class LDK {
 	}
 
 	//TODO
-	// Step 1: Initialize the FeeEstimator
+	// Step 1: Initialize the FeeEstimator ✅
 	// Step 2: Initialize the Logger
 	// Step 3: Initialize the BroadcasterInterface
 	// Step 4: Initialize Persist
@@ -53,6 +53,17 @@ class LDK {
 	// Step 18: Persist ChannelManager and NetworkGraph
 	// Step 19: Background Processing
 
+	async initFeeEstimator(fees: TFeeUpdateReq): Promise<Result<string>> {
+		const res = await NativeLDK.initFeeEstimator();
+		await this.updateFees(fees);
+		return ok(res);
+	}
+
+	async updateFees({ high, normal, low }: TFeeUpdateReq): Promise<Result<string>> {
+		const res = await NativeLDK.updateFees(high, normal, low);
+		return ok(res);
+	}
+
 	/**
 	 * Starts the startChainMonitor service
 	 * @return {Promise<Err<unknown> | Ok<string>>}

src/ldk.ts

@@ -25,3 +25,9 @@ export type TLogListener = {
 	id: string;
 	callback: (log: string) => void;
 };
+
+export type TFeeUpdateReq = {
+	high: number,
+	normal: number,
+	low: number
+}