@onekeyfe/hd-web-sdk 0.0.10 → 0.1.0

package/build/onekey-js-sdk.js

@@ -4215,6 +4215,10 @@ const inject = ({
     stellarSignTransaction: (connectId, params) => call(Object.assign(Object.assign({}, params), {
       connectId,
       method: 'stellarSignTransaction'
+    })),
+    firmwareUpdate: (connectId, params) => call(Object.assign(Object.assign({}, params), {
+      connectId,
+      method: 'firmwareUpdate'
     }))
   };
   return api;
@@ -4772,6 +4776,7 @@ const ERROR_CODES = {
   Method_InvalidParameter: '',
   Call_API: '',
   Call_NotResponse: 'No response data',
+  Method_FirmwareUpdate_DownloadFailed: '',
   Transport_InvalidProtobuf: '',
   Device_FwException: '',
   Device_UnexpectedMode: '',
@@ -4904,7 +4909,8 @@ const enableLog = enabled => {
 const httpRequest$1 = (url, type = 'text') => __awaiter(void 0, void 0, void 0, function* () {
   const response = yield axios__default["default"].request({
     url,
-    withCredentials: false
+    withCredentials: false,
+    responseType: type === 'binary' ? 'arraybuffer' : 'json'
   });
 
   if (+response.status === 200) {
@@ -4913,7 +4919,7 @@ const httpRequest$1 = (url, type = 'text') => __awaiter(void 0, void 0, void 0,
     }
 
     if (type === 'binary') {
-      return response.data.arrayBuffer();
+      return response.data;
     }
 
     return response.data;
@@ -4922,7 +4928,7 @@ const httpRequest$1 = (url, type = 'text') => __awaiter(void 0, void 0, void 0,
   throw new Error(`httpRequest error: ${url} ${response.statusText}`);
 });
 
-const httpRequest = (url, _type) => httpRequest$1(url);
+const httpRequest = (url, type) => httpRequest$1(url, type);
 
 const getTimeStamp = () => new Date().getTime();
 
@@ -14142,7 +14148,8 @@ const UI_REQUEST$1 = {
   REQUEST_BUTTON: 'ui-button',
   CLOSE_UI_WINDOW: 'ui-close_window',
   BLUETOOTH_PERMISSION: 'ui-bluetooth_permission',
-  LOCATION_PERMISSION: 'ui-location_permission'
+  LOCATION_PERMISSION: 'ui-location_permission',
+  FIRMWARE_PROGRESS: 'ui-firmware-progress'
 };
 
 const createUiMessage = (type, payload) => ({
@@ -14614,6 +14621,10 @@ class Device extends events.exports {
     });
   }
 
+  getCommands() {
+    return this.commands;
+  }
+
   getInternalState() {
     return this.internalState[this.instance];
   }
@@ -18022,6 +18033,179 @@ class StellarSignTransaction extends BaseMethod {
 
 }
 
+const getBinary = ({
+  features,
+  updateType,
+  version
+}) => __awaiter(void 0, void 0, void 0, function* () {
+  const releaseInfo = getInfo({
+    features,
+    updateType
+  });
+
+  if (!releaseInfo) {
+    throw TypedError('Runtime', 'no firmware found for this device');
+  }
+
+  if (version && !semver__default["default"].eq(releaseInfo.version, version)) {
+    throw TypedError('Runtime', 'firmware version mismatch');
+  }
+
+  const url = updateType === 'ble' ? releaseInfo.webUpdate : releaseInfo.url;
+  const fw = yield httpRequest(url, 'binary');
+  return Object.assign(Object.assign({}, releaseInfo), {
+    binary: fw
+  });
+});
+
+const getInfo = ({
+  features,
+  updateType
+}) => {
+  var _a, _b, _c;
+
+  const deviceType = getDeviceType(features);
+  const {
+    deviceMap
+  } = DataManager;
+  const releaseInfo = (_c = (_b = (_a = deviceMap === null || deviceMap === void 0 ? void 0 : deviceMap[deviceType]) === null || _a === void 0 ? void 0 : _a[updateType]) === null || _b === void 0 ? void 0 : _b[0]) !== null && _c !== void 0 ? _c : null;
+  return releaseInfo;
+};
+
+const postConfirmationMessage = device => {
+  var _a;
+
+  if ((_a = device.features) === null || _a === void 0 ? void 0 : _a.firmware_present) {
+    device.emit(DEVICE.BUTTON, device, {
+      code: 'ButtonRequest_FirmwareUpdate'
+    });
+  }
+};
+
+const postProgressMessage = (device, progress, postMessage) => {
+  postMessage(createUiMessage(UI_REQUEST$1.FIRMWARE_PROGRESS, {
+    device: device.toMessageObject(),
+    progress
+  }));
+};
+
+const uploadFirmware = (updateType, typedCall, postMessage, device, {
+  payload
+}) => __awaiter(void 0, void 0, void 0, function* () {
+  var _a, _b;
+
+  if (((_a = device.features) === null || _a === void 0 ? void 0 : _a.major_version) === 1) {
+    postConfirmationMessage(device);
+    const eraseCommand = updateType === 'firmware' ? 'FirmwareErase' : 'FirmwareErase_ex';
+    yield typedCall(eraseCommand, 'Success', {});
+    postProgressMessage(device, 0, postMessage);
+    const {
+      message
+    } = yield typedCall('FirmwareUpload', 'Success', {
+      payload
+    });
+    return message;
+  }
+
+  if (((_b = device.features) === null || _b === void 0 ? void 0 : _b.major_version) === 2) {
+    postConfirmationMessage(device);
+    const length = payload.byteLength;
+    let response = yield typedCall('FirmwareErase', ['FirmwareRequest', 'Success'], {
+      length
+    });
+
+    while (response.type !== 'Success') {
+      const start = response.message.offset;
+      const end = response.message.offset + response.message.length;
+      const chunk = payload.slice(start, end);
+
+      if (start > 0) {
+        postProgressMessage(device, Math.round(start / length * 100), postMessage);
+      }
+
+      response = yield typedCall('FirmwareUpload', ['FirmwareRequest', 'Success'], {
+        payload: chunk
+      });
+    }
+
+    postProgressMessage(device, 100, postMessage);
+    return response.message;
+  }
+
+  throw TypedError('Runtime', 'uploadFirmware: unknown major_version');
+});
+
+class FirmwareUpdate extends BaseMethod {
+  init() {
+    this.allowDeviceMode = [UI_REQUEST.BOOTLOADER, UI_REQUEST.INITIALIZE];
+    this.requireDeviceMode = [UI_REQUEST.BOOTLOADER];
+    const {
+      payload
+    } = this;
+    validateParams(payload, [{
+      name: 'version',
+      type: 'array'
+    }, {
+      name: 'binary',
+      type: 'buffer'
+    }]);
+
+    if (!payload.updateType) {
+      throw TypedError('Method_InvalidParameter', 'updateType is required');
+    }
+
+    this.params = {
+      updateType: payload.updateType
+    };
+
+    if ('version' in payload) {
+      this.params = Object.assign(Object.assign({}, this.params), {
+        version: payload.version
+      });
+    }
+
+    if ('binary' in payload) {
+      this.params = Object.assign(Object.assign({}, this.params), {
+        binary: payload.binary
+      });
+    }
+  }
+
+  run() {
+    return __awaiter(this, void 0, void 0, function* () {
+      const {
+        device,
+        params
+      } = this;
+      let binary;
+
+      try {
+        if (params.binary) {
+          binary = this.params.binary;
+        } else {
+          if (!device.features) {
+            throw TypedError('Runtime', 'no features found for this device');
+          }
+
+          const firmware = yield getBinary({
+            features: device.features,
+            version: params.version,
+            updateType: params.updateType
+          });
+          binary = firmware.binary;
+        }
+      } catch (err) {
+        throw TypedError('Method_FirmwareUpdate_DownloadFailed', err);
+      }
+
+      return uploadFirmware(params.updateType, this.device.getCommands().typedCall.bind(this.device.getCommands()), this.postMessage, device, {
+        payload: binary
+      });
+    });
+  }
+
+}
+
 var ApiMethods = /*#__PURE__*/Object.freeze({
   __proto__: null,
   searchDevices: SearchDevices,
@@ -18061,7 +18245,8 @@ var ApiMethods = /*#__PURE__*/Object.freeze({
   solGetAddress: SolGetAddress,
   solSignTransaction: SolSignTransaction,
   stellarGetAddress: StellarGetAddress,
-  stellarSignTransaction: StellarSignTransaction
+  stellarSignTransaction: StellarSignTransaction,
+  firmwareUpdate: FirmwareUpdate
 });
 
 function findMethod(message) {
@@ -18267,6 +18452,7 @@ const callAPI = message => __awaiter(void 0, void 0, void 0, function* () {
   try {
     method = findMethod(message);
     method.connector = _connector;
+    method.postMessage = postMessage;
     method.init();
   } catch (error) {
     return Promise.reject(error);