react-native-update/harmony/src/main/ets/UpdateModuleImpl.ts

import { TurboModuleContext } from 'rnoh/ts';
import dataPreferences from '@ohos.data.preferences';
import bundleManager from '@ohos.bundle.bundleManager';
import common from '@ohos.app.ability.common';
import { BusinessError } from '@ohos.base';
import { UpdateContext } from './UpdateContext';
import { DownloadTaskParams } from './DownloadTaskParams';
import logger from './Logger';

const TAG = "UpdateModuleImpl";

export class UpdateModuleImpl {
  static readonly NAME = "Pushy";

  static async downloadFullUpdate(
    updateContext: UpdateContext, 
    options: { updateUrl: string; hash: string }
  ): Promise<void> {
    try {
      await updateContext.downloadFullUpdate(options.updateUrl, options.hash, {
        onDownloadCompleted: (params: DownloadTaskParams) => {
          return Promise.resolve();
        },
        onDownloadFailed: (error: Error) => {
          return Promise.reject(error);
        }
      });
    } catch (error) {
      logger.error(TAG, `downloadFullUpdate failed: ${error}`);
      throw error;
    }
  }

  static async downloadAndInstallApk(
    context: common.UIAbilityContext,
    options: { url: string; hash: string; target: string }
  ): Promise<void> {
    try {
      const want = {
        action: 'action.system.home',
        parameters: {
          uri: 'appmarket://details'
        }
      };

      if (!context) {
        throw new Error('获取context失败');
      }

      await context.startAbility(want);
    } catch (error) {
      logger.error(TAG, `installApk failed: ${error}`);
      throw error;
    }
  }

  static async downloadPatchFromPackage(
    updateContext: UpdateContext,
    options: { updateUrl: string; hash: string }
  ): Promise<void> {
    try {
      await updateContext.downloadPatchFromPackage(options.updateUrl, options.hash, {
        onDownloadCompleted: (params: DownloadTaskParams) => {
          return Promise.resolve();
        },
        onDownloadFailed: (error: Error) => {
          return Promise.reject(error);
        }
      });
    } catch (error) {
      logger.error(TAG, `downloadPatchFromPackage failed: ${error}`);
      throw error;
    }
  }

  static async downloadPatchFromPpk(
    updateContext: UpdateContext,
    options: { updateUrl: string; hash: string; originHash: string }
  ): Promise<void> {
    try {
      await updateContext.downloadPatchFromPpk(
        options.updateUrl,
        options.hash,
        options.originHash,
        {
          onDownloadCompleted: (params: DownloadTaskParams) => {
            return Promise.resolve();
          },
          onDownloadFailed: (error: Error) => {
            return Promise.reject(error);
          }
        }
      );
    } catch (error) {
      logger.error(TAG, `downloadPatchFromPpk failed: ${error}`);
      throw new Error(`执行报错: ${error.message}`);
    }
  }

  static async reloadUpdate(
    updateContext: UpdateContext,
    context: common.UIAbilityContext,
    options: { hash: string }
  ): Promise<void> {
    const hash = options.hash;
    if (!hash) {
      throw new Error('hash不能为空');
    }

    try {
      await updateContext.switchVersion(hash);
      const bundleInfo = await bundleManager.getBundleInfoForSelf(
        bundleManager.BundleFlag.GET_BUNDLE_INFO_WITH_REQUESTED_PERMISSION
      );
      await context.terminateSelf();
      const want = {
        bundleName: bundleInfo.name,
        abilityName: context.abilityInfo?.name
      };
      await context.startAbility(want);
    } catch (error) {
      logger.error(TAG, `reloadUpdate failed: ${error}`);
      throw new Error(`pushy:switchVersion failed ${error.message}`);
    }
  }

  static async setNeedUpdate(
    updateContext: UpdateContext,
    options: { hash: string }
  ): Promise<boolean> {
    const hash = options.hash;
    if (!hash) {
      throw new Error('hash不能为空');
    }

    try {
      await updateContext.switchVersion(hash);
      return true;
    } catch (error) {
      logger.error(TAG, `setNeedUpdate failed: ${error}`);
      throw new Error(`switchVersionLater failed: ${error.message}`);
    }
  }

  static async markSuccess(updateContext: UpdateContext): Promise<boolean> {
    try {
      await updateContext.markSuccess();
      return true;
    } catch (error) {
      logger.error(TAG, `markSuccess failed: ${error}`);
      throw new Error(`执行报错: ${error.message}`);
    }
  }

  static async setUuid(
    updateContext: UpdateContext,
    uuid: string
  ): Promise<boolean> {
    try {
      await updateContext.setKv('uuid', uuid);
      return true;
    } catch (error) {
      logger.error(TAG, `setUuid failed: ${error}`);
      throw new Error(`执行报错: ${error.message}`);
    }
  }

  static checkJson(json: string): boolean {
    try {
      JSON.parse(json);
      return true;
    } catch {
      return false;
    }
  }

  static async setLocalHashInfo(
    updateContext: UpdateContext,
    hash: string,
    info: string
  ): Promise<boolean> {
    if (!this.checkJson(info)) {
      await updateContext.setKv(`hash_${hash}`, info);
      throw new Error('校验报错:json字符串格式错误');
    }
    await updateContext.setKv(`hash_${hash}`, info);
    return true;
  }

  static async getLocalHashInfo(
    updateContext: UpdateContext,
    hash: string
  ): Promise<string> {
    const value = await updateContext.getKv(`hash_${hash}`);
    if (!this.checkJson(value)) {
      throw new Error('校验报错:json字符串格式错误');
    }
    return value;
  }
}