@onekeyfe/hd-core 1.2.0-alpha.10 → 1.2.0-alpha.12

package/src/events/call.ts

@@ -53,6 +53,12 @@ export interface IFrameCancelMessage {
   payload: { connectId?: string };
 }
 
+export interface IFrameCancelOperationMessage {
+  event: typeof IFRAME.CANCEL_OPERATION;
+  type: typeof IFRAME.CANCEL_OPERATION;
+  payload: { operationId: string };
+}
+
 export interface IFrameSwitchTransportMessage {
   event: typeof IFRAME.SWITCH_TRANSPORT;
   type: typeof IFRAME.SWITCH_TRANSPORT;

package/src/events/core.ts

@@ -5,6 +5,7 @@ import type {
   IFrameCallMessage,
   IFrameCallbackMessage,
   IFrameCancelMessage,
+  IFrameCancelOperationMessage,
   IFrameSwitchTransportMessage,
 } from './call';
 import type { DeviceEventMessage } from './device';
@@ -23,6 +24,7 @@ export type CoreMessage = {
   | IFrameEventMessage
   | IFrameCallMessage
   | IFrameCancelMessage
+  | IFrameCancelOperationMessage
   | IFrameSwitchTransportMessage
   | IFrameCallbackMessage
   | UiResponseMessage

package/src/events/iframe.ts

@@ -8,6 +8,7 @@ export const IFRAME = {
   INIT_BRIDGE: 'iframe-init-bridge',
   CALL: 'iframe-call',
   CANCEL: 'iframe-cancel',
+  CANCEL_OPERATION: 'iframe-cancel-operation',
   SWITCH_TRANSPORT: 'iframe-switch-transport',
   CALLBACK: 'iframe-callback',
 } as const;

package/src/index.ts

@@ -27,6 +27,7 @@ const HardwareSdk = ({
   eventEmitter,
   uiResponse,
   cancel,
+  cancelOperation,
   updateSettings,
   switchTransport,
 }: InjectApi): CoreApi =>
@@ -37,6 +38,7 @@ const HardwareSdk = ({
     eventEmitter,
     uiResponse,
     cancel,
+    cancelOperation,
     updateSettings,
     switchTransport,
   });
@@ -49,6 +51,7 @@ const HardwareSDKLowLevel = ({
   addHardwareGlobalEventListener,
   uiResponse,
   cancel,
+  cancelOperation,
   updateSettings,
   switchTransport,
 }: LowLevelInjectApi): LowLevelCoreApi =>
@@ -60,6 +63,7 @@ const HardwareSDKLowLevel = ({
     addHardwareGlobalEventListener,
     uiResponse,
     cancel,
+    cancelOperation,
     updateSettings,
     switchTransport,
   });

package/src/inject.ts

@@ -36,12 +36,14 @@ export interface InjectApi {
   dispose: CoreApi['dispose'];
   uiResponse: CoreApi['uiResponse'];
   cancel: CoreApi['cancel'];
+  cancelOperation: CoreApi['cancelOperation'];
   switchTransport: CoreApi['switchTransport'];
 }
 
 export const inject = ({
   call,
   cancel,
+  cancelOperation,
   dispose,
   eventEmitter,
   init,
@@ -73,6 +75,7 @@ export const inject = ({
     uiResponse,
 
     cancel,
+    cancelOperation,
 
     updateSettings,
 
@@ -96,6 +99,7 @@ export const createCoreApi = (
   | 'dispose'
   | 'uiResponse'
   | 'cancel'
+  | 'cancelOperation'
   | 'updateSettings'
   | 'switchTransport'
 > => ({
@@ -158,6 +162,8 @@ export const createCoreApi = (
   deviceReboot: (connectId, params) => call({ ...params, connectId, method: 'deviceReboot' }),
   deviceInfoGet: (connectId, params) => call({ ...params, connectId, method: 'deviceInfoGet' }),
   deviceStatusGet: (connectId, params) => call({ ...params, connectId, method: 'deviceStatusGet' }),
+  deviceGetOnboardingStatus: (connectId, params) =>
+    call({ ...params, connectId, method: 'deviceGetOnboardingStatus' }),
   deviceSessionGet: (connectId, params) =>
     call({ ...params, connectId, method: 'deviceSessionGet' }),
   deviceFirmwareUpdate: (connectId, params) =>
@@ -168,6 +174,14 @@ export const createCoreApi = (
     call({ ...params, connectId, method: 'deviceFactoryInfoSet' }),
   deviceFactoryInfoGet: (connectId, params) =>
     call({ ...params, connectId, method: 'deviceFactoryInfoGet' }),
+  deviceSettingsGet: (connectId, params) =>
+    call({ ...params, connectId, method: 'deviceSettingsGet' }),
+  deviceSettingsSet: (connectId, params) =>
+    call({ ...params, connectId, method: 'deviceSettingsSet' }),
+  deviceSettingsPageShow: (connectId, params) =>
+    call({ ...params, connectId, method: 'deviceSettingsPageShow' }),
+  deviceUploadWallpaper: (connectId, params) =>
+    call({ ...params, connectId, method: 'deviceUploadWallpaper' }),
   filesystemPermissionFix: (connectId, params) =>
     call({ ...params, connectId, method: 'filesystemPermissionFix' }),
   fileRead: (connectId, params) => call({ ...params, connectId, method: 'fileRead' }),
@@ -181,6 +195,7 @@ export const createCoreApi = (
     call({ ...params, connectId, method: 'filesystemFileRead' }),
   filesystemFileWrite: (connectId, params) =>
     call({ ...params, connectId, method: 'filesystemFileWrite' }),
+  uploadPortfolio: (connectId, params) => call({ ...params, connectId, method: 'uploadPortfolio' }),
   filesystemFileDelete: (connectId, params) =>
     call({ ...params, connectId, method: 'filesystemFileDelete' }),
   filesystemDirList: (connectId, params) =>

package/src/lowLevelInject.ts

@@ -13,6 +13,7 @@ export interface LowLevelInjectApi {
   dispose: CoreApi['dispose'];
   uiResponse: CoreApi['uiResponse'];
   cancel: CoreApi['cancel'];
+  cancelOperation: CoreApi['cancelOperation'];
   updateSettings: CoreApi['updateSettings'];
   switchTransport: CoreApi['switchTransport'];
   addHardwareGlobalEventListener: (listener: IAddHardwareGlobalEventListener) => void;
@@ -25,6 +26,7 @@ export type LowLevelCoreApi = Omit<CoreApi, 'on' | 'off'> & {
 export const lowLevelInject = ({
   call,
   cancel,
+  cancelOperation,
   dispose,
   eventEmitter,
   init,
@@ -48,6 +50,7 @@ export const lowLevelInject = ({
     uiResponse,
 
     cancel,
+    cancelOperation,
 
     updateSettings,
 

package/src/protocols/protocol-v2/features.ts

@@ -50,8 +50,26 @@ export const getProtocolV2SeState = (se?: DeviceSEInfo): ProtocolV2SeStateLabel
 export const getProtocolV2SeType = (se?: DeviceSEInfo): string | null =>
   normalizeEnumValue(DeviceSeType, se?.type);
 
+/**
+ * 兼容尚未提供显式 runtime mode 的 Protocol V2 固件。
+ *
+ * 当前 romloader 只上报 hw、fw.romloader 和 fw.bootloader；bootloader
+ * 还会上报 application/application_data、coprocessor 或 SE 信息。
+ */
+export const isProtocolV2RomloaderDeviceInfo = (deviceInfo?: ProtocolV2DeviceInfo | null) =>
+  !!deviceInfo &&
+  deviceInfo.status == null &&
+  deviceInfo.fw?.romloader != null &&
+  deviceInfo.fw?.application == null &&
+  deviceInfo.fw?.application_data == null &&
+  deviceInfo.coprocessor == null &&
+  deviceInfo.se1 == null &&
+  deviceInfo.se2 == null &&
+  deviceInfo.se3 == null &&
+  deviceInfo.se4 == null;
+
 export const isProtocolV2BootloaderDeviceInfo = (deviceInfo?: ProtocolV2DeviceInfo | null) =>
-  !!deviceInfo && deviceInfo.status == null;
+  !!deviceInfo && deviceInfo.status == null && !isProtocolV2RomloaderDeviceInfo(deviceInfo);
 
 export const PROTOCOL_V2_FEATURES_DEVICE_INFO_REQUEST = {
   targets: {

package/src/protocols/protocol-v2/index.ts

@@ -4,6 +4,8 @@ export {
   PROTOCOL_V2_FEATURES_DEVICE_INFO_REQUEST,
   PROTOCOL_V2_STATUS_DEVICE_INFO_REQUEST,
   PROTOCOL_V2_VERSIONS_DEVICE_INFO_REQUEST,
+  isProtocolV2BootloaderDeviceInfo,
+  isProtocolV2RomloaderDeviceInfo,
   getProtocolV2SeState,
   getProtocolV2SeType,
 } from './features';

package/src/protocols/protocol-v2/unlockRetry.ts

@@ -0,0 +1,46 @@
+import { HardwareErrorCode } from '@onekeyfe/hd-shared';
+
+import { LoggerNames, getLogger } from '../../utils';
+
+import type { BaseMethod } from '../../api/BaseMethod';
+import type { Device } from '../../device/Device';
+
+const Log = getLogger(LoggerNames.Core);
+
+type RunnableMethod = Pick<BaseMethod, 'run' | 'unlockPolicy'> & { name?: string };
+type UnlockableDevice = Pick<Device, 'isProtocolV2' | 'unlockDevice'>;
+
+function isDeviceLockedError(error: unknown): boolean {
+  return (
+    typeof error === 'object' &&
+    error !== null &&
+    'errorCode' in error &&
+    error.errorCode === HardwareErrorCode.DeviceLocked
+  );
+}
+
+export async function runMethodWithUnlockRetry(method: RunnableMethod, device: UnlockableDevice) {
+  try {
+    return await method.run();
+  } catch (error) {
+    if (
+      !device.isProtocolV2() ||
+      method.unlockPolicy !== 'retry-on-locked' ||
+      !isDeviceLockedError(error)
+    ) {
+      throw error;
+    }
+
+    Log.debug('Protocol V2 unlock retry triggered', { method: method.name });
+    await device.unlockDevice();
+    Log.debug('Protocol V2 unlock completed', { method: method.name });
+    try {
+      const response = await method.run();
+      Log.debug('Protocol V2 method retry completed', { method: method.name, success: true });
+      return response;
+    } catch (retryError) {
+      Log.debug('Protocol V2 method retry completed', { method: method.name, success: false });
+      throw retryError;
+    }
+  }
+}

package/src/protocols/protocol-v2/walletSession.ts

@@ -43,11 +43,20 @@ export async function getProtocolV2WalletSession(
     typeof device.getInternalState === 'function' ? device.getInternalState() : undefined;
 
   try {
-    const { message } = await device.commands.typedCall(
-      'DeviceSessionGet',
-      'DeviceSession',
-      cachedSessionId ? { session_id: cachedSessionId } : {}
-    );
+    const requestDeviceSession = (sessionId?: string) =>
+      device.commands.typedCall(
+        'DeviceSessionGet',
+        'DeviceSession',
+        sessionId ? { session_id: sessionId } : {}
+      );
+
+    const { message } = await requestDeviceSession(cachedSessionId).catch(async error => {
+      if (!cachedSessionId || !isProtocolV2InvalidSessionError(error)) {
+        throw error;
+      }
+      device.clearInternalState();
+      return requestDeviceSession();
+    });
 
     if (
       options?.expectedPassphraseState &&

package/src/topLevelInject.ts

@@ -52,6 +52,8 @@ export const topLevelInject = () => {
 
     cancel: (connectId?: string) => lowLevelApi?.cancel(connectId),
 
+    cancelOperation: (operationId: string) => lowLevelApi?.cancelOperation(operationId),
+
     updateSettings: settings => lowLevelApi?.updateSettings(settings) ?? Promise.resolve(false),
 
     switchTransport: (env: ConnectSettings['env']) =>

package/src/types/api/firmwareUpdate.ts

@@ -61,10 +61,22 @@ export interface FirmwareUpdateV3Params {
  * firmwareUpdateV4（Protocol V2）按 DeviceFirmwareTargetType 拆分的目标二进制。
  * 除 romloader 外，每个字段对应一个 bootloader 可接受的固件升级 target。
  */
+export type FirmwareUpdateV4Target =
+  | 'boot'
+  | 'app_v1'
+  | 'app_v2'
+  | 'coprocessor'
+  | 'resource'
+  | 'se01'
+  | 'se02'
+  | 'se03'
+  | 'se04';
+
 export interface FirmwareUpdateV4Params {
   platform: IPlatform;
   chunkSize?: number;
   firmwareType?: EFirmwareType;
+  targetsToUpdate?: FirmwareUpdateV4Target[];
 
   /** FW_MGMT_TARGET_ROMLOADER = 2；当前 Pro2 bootloader 不接受通过 firmwareUpdateV4 安装 */
   romloaderBinary?: ArrayBuffer;

package/src/types/api/getDeviceInfo.ts

@@ -16,7 +16,7 @@ export type GetDeviceInfoParams = {
   includeRaw?: boolean;
 };
 
-export type DeviceInfoMode = 'normal' | 'bootloader' | 'notInitialized' | 'unknown';
+export type DeviceInfoMode = 'normal' | 'bootloader' | 'romloader' | 'notInitialized' | 'unknown';
 
 export type DeviceInfoStatus = {
   mode: DeviceInfoMode;

package/src/types/api/getPassphraseState.ts

@@ -1,17 +1,8 @@
 import type { CommonParams, Response } from '../params';
 
-export type GetPassphraseStatePayload = {
-  passphraseState?: string;
-  sessionId?: string;
-  unlockedAttachPin?: boolean;
-  passphraseProtection?: boolean | null;
-};
-
-export type GetPassphraseStateParams = CommonParams & {
-  allowCreateAttachPin?: boolean;
-};
+export type GetPassphraseStateParams = CommonParams;
 
 export declare function getPassphraseState(
   connectId?: string,
   params?: GetPassphraseStateParams
-): Response<GetPassphraseStatePayload>;
+): Response<string | undefined>;

package/src/types/api/index.ts

@@ -3,10 +3,15 @@ import type {
   deviceFactoryInfoSet,
   deviceFirmwareUpdate,
   deviceGetFirmwareUpdateStatus,
+  deviceGetOnboardingStatus,
   deviceInfoGet,
   deviceReboot,
   deviceSessionGet,
+  deviceSettingsGet,
+  deviceSettingsPageShow,
+  deviceSettingsSet,
   deviceStatusGet,
+  deviceUploadWallpaper,
   dirList,
   dirMake,
   dirRemove,
@@ -26,6 +31,7 @@ import type {
   pathInfo,
   ping,
   protocolInfoRequest,
+  uploadPortfolio,
 } from './protocolV2';
 import type { off, on, removeAllListeners } from './event';
 import type { uiResponse } from './uiResponse';
@@ -179,7 +185,7 @@ export type {
   DeviceProfileVerify,
   DeviceProfileVersions,
 } from './getDeviceInfo';
-export type { GetPassphraseStateParams, GetPassphraseStatePayload } from './getPassphraseState';
+export type { GetPassphraseStateParams } from './getPassphraseState';
 export type { ClearSessionCacheParams, ClearSessionCachePayload } from './sessionCache';
 
 export type CoreApi = {
@@ -195,6 +201,7 @@ export type CoreApi = {
   call: (params: any) => Promise<any>;
   uiResponse: typeof uiResponse;
   cancel: (connectId?: string) => void;
+  cancelOperation: (operationId: string) => void;
   updateSettings: typeof updateSettings;
   switchTransport: (env: ConnectSettings['env']) => Promise<{ success: boolean }>;
   getLogs: typeof getLogs;
@@ -260,11 +267,16 @@ export type CoreApi = {
   deviceReboot: typeof deviceReboot;
   deviceInfoGet: typeof deviceInfoGet;
   deviceStatusGet: typeof deviceStatusGet;
+  deviceGetOnboardingStatus: typeof deviceGetOnboardingStatus;
   deviceSessionGet: typeof deviceSessionGet;
   deviceFirmwareUpdate: typeof deviceFirmwareUpdate;
   deviceGetFirmwareUpdateStatus: typeof deviceGetFirmwareUpdateStatus;
   deviceFactoryInfoSet: typeof deviceFactoryInfoSet;
   deviceFactoryInfoGet: typeof deviceFactoryInfoGet;
+  deviceSettingsGet: typeof deviceSettingsGet;
+  deviceSettingsSet: typeof deviceSettingsSet;
+  deviceSettingsPageShow: typeof deviceSettingsPageShow;
+  deviceUploadWallpaper: typeof deviceUploadWallpaper;
   filesystemPermissionFix: typeof filesystemPermissionFix;
   fileRead: typeof fileRead;
   fileWrite: typeof fileWrite;
@@ -275,6 +287,7 @@ export type CoreApi = {
   pathInfo: typeof pathInfo;
   filesystemFileRead: typeof filesystemFileRead;
   filesystemFileWrite: typeof filesystemFileWrite;
+  uploadPortfolio: typeof uploadPortfolio;
   filesystemFileDelete: typeof filesystemFileDelete;
   filesystemDirList: typeof filesystemDirList;
   filesystemDirMake: typeof filesystemDirMake;

package/src/types/api/protocolV2.ts

@@ -2,7 +2,9 @@ import type { CommonParams, Response } from '../params';
 import type {
   DeviceFactoryInfo,
   DeviceFirmwareUpdateStatus,
+  DevOnboardingStatus,
   DeviceSession,
+  DeviceSettings,
   DeviceStatus,
   ProtocolInfo,
   ProtocolV2DeviceInfo,
@@ -15,7 +17,12 @@ import type {
   DeviceRebootParams,
 } from '../../api/protocol-v2/helpers';
 import type { DeviceInfoGetParams } from '../../api/protocol-v2/DeviceInfoGet';
-import type { DeviceSessionGetParams } from '../../api/protocol-v2/DeviceSessionGet';
+import type { DeviceSettingsSetParams } from '../../api/protocol-v2/DeviceSettingsSet';
+import type { DeviceSettingsPageShowParams } from '../../api/protocol-v2/DeviceSettingsPageShow';
+import type {
+  DeviceUploadWallpaperParams,
+  DeviceUploadWallpaperResponse,
+} from '../../api/protocol-v2/DeviceUploadWallpaper';
 
 // 参数类型单源：以 api/protocol-v2 的实现为准（type-only re-export，无运行时依赖）
 export type {
@@ -31,7 +38,15 @@ export type {
   DeviceInfoGetTargets,
   DeviceInfoGetTypes,
 } from '../../api/protocol-v2/DeviceInfoGet';
-export type { DeviceSessionGetParams } from '../../api/protocol-v2/DeviceSessionGet';
+export type { DeviceSettingsSetParams } from '../../api/protocol-v2/DeviceSettingsSet';
+export type {
+  DeviceSettingsPageShowParams,
+  SupportedDeviceSettingsPage,
+} from '../../api/protocol-v2/DeviceSettingsPageShow';
+export type {
+  DeviceUploadWallpaperParams,
+  DeviceUploadWallpaperResponse,
+} from '../../api/protocol-v2/DeviceUploadWallpaper';
 
 // ── Shared response shapes (Protocol V2 file system) ────────────────────
 
@@ -97,9 +112,14 @@ export declare function deviceStatusGet(
   params?: CommonParams
 ): Response<DeviceStatus>;
 
+export declare function deviceGetOnboardingStatus(
+  connectId: string,
+  params?: CommonParams
+): Response<DevOnboardingStatus>;
+
 export declare function deviceSessionGet(
   connectId: string,
-  params?: CommonParams & DeviceSessionGetParams
+  params?: CommonParams
 ): Response<DeviceSession>;
 
 export declare function deviceFirmwareUpdate(
@@ -122,6 +142,26 @@ export declare function deviceFactoryInfoGet(
   params?: CommonParams
 ): Response<DeviceFactoryInfo>;
 
+export declare function deviceSettingsGet(
+  connectId: string,
+  params?: CommonParams
+): Response<DeviceSettings>;
+
+export declare function deviceSettingsSet(
+  connectId: string,
+  params: CommonParams & DeviceSettingsSetParams
+): Response<Success>;
+
+export declare function deviceSettingsPageShow(
+  connectId: string,
+  params: CommonParams & DeviceSettingsPageShowParams
+): Response<Success>;
+
+export declare function deviceUploadWallpaper(
+  connectId: string,
+  params: CommonParams & DeviceUploadWallpaperParams
+): Response<DeviceUploadWallpaperResponse>;
+
 export declare function filesystemPermissionFix(
   connectId: string,
   params?: CommonParams
@@ -181,6 +221,15 @@ export declare function pathInfo(
 
 export declare const filesystemFileRead: typeof fileRead;
 export declare const filesystemFileWrite: typeof fileWrite;
+
+export declare function uploadPortfolio(
+  connectId: string,
+  params: {
+    operationId?: string;
+    packageBytes: ArrayBuffer | Uint8Array | Blob;
+    timeoutMs?: number | string;
+  }
+): Response<FileInfo & { portfolioUpdated: true }>;
 export declare const filesystemFileDelete: typeof fileDelete;
 export declare const filesystemDirList: typeof dirList;
 export declare const filesystemDirMake: typeof dirMake;

package/src/types/device.ts

@@ -91,6 +91,7 @@ export type DeviceFeaturesProtocol = 'V1' | 'V2' | 'unknown';
 export type DeviceFeaturesMode =
   | 'normal'
   | 'bootloader'
+  | 'romloader'
   | 'notInitialized'
   | 'backupMode'
   | 'unknown';

package/src/utils/deviceFeaturesUtils.ts

@@ -89,7 +89,6 @@ export const getPassphraseStateWithRefreshDeviceInfo = async (
   options?: {
     expectPassphraseState?: string;
     onlyMainPin?: boolean;
-    allowCreateAttachPin?: boolean;
     initSession?: boolean;
   }
 ) => {
@@ -164,7 +163,6 @@ export const getPassphraseState = async (
   options?: {
     expectPassphraseState?: string;
     onlyMainPin?: boolean;
-    allowCreateAttachPin?: boolean;
     initSession?: boolean;
   }
 ): Promise<{
@@ -196,12 +194,9 @@ export const getPassphraseState = async (
     supportAttachPinCapability || supportProSeriesAttachPinPassphrase(deviceType, firmwareVersion);
 
   if (supportGetPassphraseState) {
-    const payload: GetPassphraseState = options?.onlyMainPin
-      ? { _only_main_pin: true }
-      : { passphrase_state: options?.expectPassphraseState };
-    if (options?.allowCreateAttachPin) {
-      payload.allow_create_attach_pin = true;
-    }
+    const payload: GetPassphraseState = {
+      passphrase_state: options?.onlyMainPin ? undefined : options?.expectPassphraseState,
+    };
 
     const { message, type } = await commands.typedCall(
       'GetPassphraseState',

package/src/utils/index.ts

@@ -38,6 +38,12 @@ export { getHDPath, getScriptType, getOutputScriptType } from '../api/helpers/pa
 export const isBleConnect = (env: string) => env === 'react-native' || env === 'lowlevel';
 
 export { getHomeScreenHex, getHomeScreenDefaultList, getHomeScreenSize } from './homescreen';
+export {
+  encodePro2Wallpaper,
+  PRO2_WALLPAPER_HEIGHT,
+  PRO2_WALLPAPER_WIDTH,
+} from './pro2Wallpaper';
+export type { Pro2WallpaperColorFormat } from './pro2Wallpaper';
 
 export const wait = (ms: number) =>
   new Promise(resolve => {

package/src/utils/pro2Wallpaper.ts

@@ -0,0 +1,111 @@
+import { ERRORS, HardwareErrorCode } from '@onekeyfe/hd-shared';
+
+export const PRO2_WALLPAPER_WIDTH = 604;
+export const PRO2_WALLPAPER_HEIGHT = 1024;
+
+export type Pro2WallpaperColorFormat = 'RGB565' | 'RGB565A8';
+
+const COLOR_FORMAT_RGB565 = 0x12;
+const COLOR_FORMAT_RGB565A8 = 0x14;
+
+const RED_THRESHOLD = [
+  1, 7, 3, 5, 0, 8, 2, 6, 7, 1, 5, 3, 8, 0, 6, 2, 3, 5, 0, 8, 2, 6, 1, 7, 5, 3, 8,
+  0, 6, 2, 7, 1, 0, 8, 2, 6, 1, 7, 3, 5, 8, 0, 6, 2, 7, 1, 5, 3, 2, 6, 1, 7, 3, 5, 0,
+  8, 6, 2, 7, 1, 5, 3, 8, 0,
+];
+const GREEN_THRESHOLD = [
+  1, 3, 2, 2, 3, 1, 2, 2, 2, 2, 0, 4, 2, 2, 4, 0, 3, 1, 2, 2, 1, 3, 2, 2, 2, 2, 4,
+  0, 2, 2, 0, 4, 1, 3, 2, 2, 3, 1, 2, 2, 2, 2, 0, 4, 2, 2, 4, 0, 3, 1, 2, 2, 1, 3, 2,
+  2, 2, 2, 4, 0, 2, 2, 0, 4,
+];
+const BLUE_THRESHOLD = [
+  5, 3, 8, 0, 6, 2, 7, 1, 3, 5, 0, 8, 2, 6, 1, 7, 8, 0, 6, 2, 7, 1, 5, 3, 0, 8, 2,
+  6, 1, 7, 3, 5, 6, 2, 7, 1, 5, 3, 8, 0, 2, 6, 1, 7, 3, 5, 0, 8, 7, 1, 5, 3, 8, 0, 6,
+  2, 1, 7, 3, 5, 0, 8, 2, 6,
+];
+
+function invalidParameter(message: string): Error {
+  return ERRORS.TypedError(HardwareErrorCode.CallMethodInvalidParameter, message);
+}
+
+function asBytes(rgba: Uint8Array | ArrayBuffer): Uint8Array {
+  return rgba instanceof Uint8Array ? rgba : new Uint8Array(rgba);
+}
+
+function align(value: number, boundary: number): number {
+  return Math.ceil(value / boundary) * boundary;
+}
+
+export function encodePro2Wallpaper(options: {
+  width: number;
+  height: number;
+  rgba: Uint8Array | ArrayBuffer;
+}): { data: Uint8Array; colorFormat: Pro2WallpaperColorFormat } {
+  const { width, height } = options;
+  if (!Number.isInteger(width) || width <= 0 || width > 0xffff) {
+    throw invalidParameter('Wallpaper width must be an integer between 1 and 65535.');
+  }
+  if (!Number.isInteger(height) || height <= 0 || height > 0xffff) {
+    throw invalidParameter('Wallpaper height must be an integer between 1 and 65535.');
+  }
+
+  const rgba = asBytes(options.rgba);
+  const expectedLength = width * height * 4;
+  if (rgba.byteLength !== expectedLength) {
+    throw invalidParameter(
+      `Wallpaper RGBA data length must be ${expectedLength} bytes, received ${rgba.byteLength}.`
+    );
+  }
+
+  let hasTransparency = false;
+  for (let index = 3; index < rgba.length; index += 4) {
+    if (rgba[index] !== 0xff) {
+      hasTransparency = true;
+      break;
+    }
+  }
+
+  const colorFormat: Pro2WallpaperColorFormat = hasTransparency ? 'RGB565A8' : 'RGB565';
+  const stride = align(width * 2, 4);
+  const alphaStride = stride / 2;
+  const rgbSize = stride * height;
+  const alphaSize = hasTransparency ? alphaStride * height : 0;
+  const data = new Uint8Array(12 + rgbSize + alphaSize);
+  const view = new DataView(data.buffer);
+
+  data[0] = 0x19;
+  data[1] = hasTransparency ? COLOR_FORMAT_RGB565A8 : COLOR_FORMAT_RGB565;
+  view.setUint16(2, 0, true);
+  view.setUint16(4, width, true);
+  view.setUint16(6, height, true);
+  view.setUint16(8, stride, true);
+  view.setUint16(10, 0, true);
+
+  for (let y = 0; y < height; y += 1) {
+    for (let x = 0; x < width; x += 1) {
+      const sourceOffset = (y * width + x) * 4;
+      const thresholdIndex = ((y & 7) << 3) + (x & 7);
+      let red = Math.min(rgba[sourceOffset] + RED_THRESHOLD[thresholdIndex], 0xff) & 0xf8;
+      let green =
+        Math.min(rgba[sourceOffset + 1] + GREEN_THRESHOLD[thresholdIndex], 0xff) & 0xfc;
+      let blue =
+        Math.min(rgba[sourceOffset + 2] + BLUE_THRESHOLD[thresholdIndex], 0xff) & 0xf8;
+      if (!hasTransparency) {
+        const alpha = rgba[sourceOffset + 3];
+        red = (red * alpha) >> 8;
+        green = (green * alpha) >> 8;
+        blue = (blue * alpha) >> 8;
+      }
+      const rgb565 = ((red >> 3) << 11) | ((green >> 2) << 5) | (blue >> 3);
+      const rgbOffset = 12 + y * stride + x * 2;
+      data[rgbOffset] = rgb565 & 0xff;
+      data[rgbOffset + 1] = rgb565 >> 8;
+
+      if (hasTransparency) {
+        data[12 + rgbSize + y * alphaStride + x] = rgba[sourceOffset + 3];
+      }
+    }
+  }
+
+  return { data, colorFormat };
+}