Merge pull request #30 from PlayerData/erase-command

Expose ability to erase secondary image

Author: ball-hayden

android/src/main/java/uk/co/playerdata/reactnativemcumanager/McuManagerModule.kt

@@ -6,8 +6,10 @@ import android.net.Uri
 import android.util.Log
 import com.facebook.react.bridge.*
 import com.facebook.react.modules.core.DeviceEventManagerModule.RCTDeviceEventEmitter
+import io.runtime.mcumgr.ble.McuMgrBleTransport
+import io.runtime.mcumgr.managers.ImageManager
 
-class McuManagerModule(val reactContext: ReactApplicationContext) : ReactContextBaseJavaModule(reactContext) {
+class McuManagerModule(private val reactContext: ReactApplicationContext) : ReactContextBaseJavaModule(reactContext) {
     private val TAG = "McuManagerModule"
     private val bluetoothAdapter: BluetoothAdapter? = BluetoothAdapter.getDefaultAdapter()
     private val upgrades: MutableMap<String, DeviceUpgrade> = mutableMapOf()
@@ -16,13 +18,35 @@ class McuManagerModule(val reactContext: ReactApplicationContext) : ReactContext
         return "McuManager"
     }
 
+    @ReactMethod
+    fun eraseImage(macAddress: String?, promise: Promise) {
+        if (this.bluetoothAdapter == null) {
+            throw Exception("No bluetooth adapter")
+        }
+
+        try {
+            val device: BluetoothDevice = bluetoothAdapter.getRemoteDevice(macAddress)
+
+            var transport = McuMgrBleTransport(reactContext, device)
+            transport.connect(device).timeout(60000).await()
+
+            val imageManager = ImageManager(transport);
+            imageManager.erase()
+
+            promise.resolve(null)
+        } catch (e: Throwable) {
+            promise.reject(e)
+        }
+    }
+
     @ReactMethod
     fun createUpgrade(id: String, macAddress: String?, updateFileUriString: String?, updateOptions: ReadableMap) {
         if (this.bluetoothAdapter == null) {
-            throw Exception("no bluetooth adapter")
+            throw Exception("No bluetooth adapter")
         }
+
         if (upgrades.contains(id)){
-            throw Exception("update ID already present")
+            throw Exception("Update ID already present")
         }
 
         val device: BluetoothDevice = bluetoothAdapter.getRemoteDevice(macAddress)

ios/McuManager.m

@@ -12,6 +12,13 @@ + (BOOL)requiresMainQueueSetup
                   supportedEvents
                   )
 
+RCT_EXTERN_METHOD(
+                  eraseImage:
+                  NSString
+                  resolver:(RCTPromiseResolveBlock)resolve
+                  rejecter:(RCTPromiseRejectBlock)reject
+                  )
+
 RCT_EXTERN_METHOD(
                   createUpgrade:
                   NSString

ios/McuManager.swift

@@ -1,5 +1,6 @@
 import os
 import CoreBluetooth
+import iOSMcuManagerLibrary
 
 
 @objc(RNMcuManager)
@@ -19,6 +20,27 @@ class RNMcuManager: RCTEventEmitter {
         ]
     }
 
+    @objc
+    func eraseImage(_ bleId: String, resolver resolve: @escaping RCTPromiseResolveBlock, rejecter reject: @escaping RCTPromiseRejectBlock) -> Void {
+        guard let bleUuid = UUID(uuidString: bleId) else {
+            let error = NSError(domain: "", code: 200, userInfo: nil)
+            return reject("error", "failed to parse uuid", error);
+        }
+
+        let bleTransport = McuMgrBleTransport(bleUuid)
+        let imageManager = ImageManager(transporter: bleTransport)
+
+        imageManager.erase { (response: McuMgrResponse?, err: Error?) in
+            if (err != nil) {
+                reject("ERASE_ERR", err?.localizedDescription, err)
+                return
+            }
+
+            resolve(nil)
+            return
+        }
+    }
+
     @objc
     func createUpgrade(_ id: String, bleId: String, updateFileUriString: String, updateOptions: Dictionary<String, Any>) -> Void {
         upgrades[id] = DeviceUpgrade(id: id, bleId: bleId, fileURI: updateFileUriString, options: updateOptions, eventEmitter: self)

src/index.tsx

@@ -1,7 +1,15 @@
+import { NativeModules } from 'react-native';
+
+const { McuManager } = NativeModules;
+
 import Upgrade, {
   FirmwareUpgradeState,
   UpgradeOptions,
   UpgradeMode,
 } from './Upgrade';
 
+export const eraseImage = McuManager.eraseImage as (
+  bleId: string
+) => Promise<void>;
+
 export { Upgrade, FirmwareUpgradeState, UpgradeOptions, UpgradeMode };