xref: /applications/standard/dlp_manager/entry/src/main/ets/feature/HomeFeature.ets

/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { BusinessError } from '@ohos.base';
import common from '@ohos.app.ability.common';
import IDLDLPProperty from '../serviceExtensionAbility/sequenceable/dlpClass';
import GlobalContext from '../common/GlobalContext';
import {
  closeDlpFileCallback,
  sandBoxLinkFileCallback,
  fileOpenHistoryCallback,
  linkSetCallback,
  genDlpFileCallback,
  openDlpFileCallback,
  recoverDlpFileCallback,
  replaceDlpLinkFileCallback,
  resumeFuseLinkCallback,
  stopFuseLinkCallback
} from '../MainAbility/data/IIdlDlpRpcServiceTs/i_id_dlpRpc_service';
import IdDlpRpcServiceProxy from '../MainAbility/data/IIdlDlpRpcServiceTs/id_dlpRpc_service_proxy';
import Constants from '../common/constant';
import { HiLog } from '../common/HiLog';

const SERVICE_EXTENSION_ABILITY_NAME = 'ServiceExtAbility';
const ERROR_CODE = -1;
const SUCCESS_CODE = 0;
const TAG = 'HomeFeature';

export default class HomeFeature {
  private connection = -1;
  private remoteCallback!: () => void;
  private context: ESObject;
  private options: common.ConnectOptions;
  private dlpRpcProxy?: IdDlpRpcServiceProxy = undefined;

  constructor(context: ESObject) {
    this.context = context;
    this.options = {
      onConnect: (elementName, proxy) => {
        HiLog.info(TAG, `onConnect success`);
        this.dlpRpcProxy = new IdDlpRpcServiceProxy(proxy);
        GlobalContext.store('dlpRpcProxy', this.dlpRpcProxy);
        HiLog.info(TAG, `DLPManager IDL onConnect success: ${JSON.stringify(this.dlpRpcProxy)}`);
        this.remoteCallback();
      },
      onDisconnect: () => {
        HiLog.info(TAG, `onDisconnect`);
      },
      onFailed: () => {
        HiLog.info(TAG, `onFailed`);
      }
    }
  }

  connectServiceExtAbility(callback: () => void) {
    HiLog.info(TAG, `connectServiceExtAbility`);
    if (this.dlpRpcProxy !== undefined) {
      return;
    }
    this.remoteCallback = callback;
    let want = {
      'bundleName': Constants.DLP_MANAGER_BUNDLE_NAME,
      'abilityName': SERVICE_EXTENSION_ABILITY_NAME
    } as Record<string, string>;
    if (this.dlpRpcProxy === undefined) {
      HiLog.info(TAG, `try connect`);
      try {
        this.connection = this.context.connectServiceExtensionAbility(want, this.options);
      } catch (err) {
        HiLog.error(TAG, `connectServiceExtensionAbility failed: ${JSON.stringify(err)}`);
      }
    }
    HiLog.info(TAG, `connectServiceExtAbility result: ${this.connection}`);
  }

  async genDlpFileHome(src:string, des:string, pop:IDLDLPProperty, callback: genDlpFileCallback) {
    HiLog.info(TAG, `genDlpFileHome start`);
    await GlobalContext.load('dlpRpcProxy').genDlpFile(src, des, pop, callback);
  }

  async openDlpFileHome(src:string, callerAppId: string, callback: openDlpFileCallback) {
    HiLog.info(TAG, `openDlpFileHome start`);
    await GlobalContext.load('dlpRpcProxy').openDlpFile(src, callerAppId, callback);
  }

  replaceDLPLinkFileHome(src:string, linkFileName:string, callback: replaceDlpLinkFileCallback) {
    HiLog.info(TAG, `replaceDLPLinkFileHome start`);
    GlobalContext.load('dlpRpcProxy').replaceDlpLinkFile(src, linkFileName, callback);
  }

  async closeDLPFileHome(src:string, callback: closeDlpFileCallback) {
    HiLog.info(TAG, `closeDLPFileHome start`);
    await GlobalContext.load('dlpRpcProxy').closeDlpFile(src, callback);
  }

  resumeFuseLinkHome(src:string, callback: resumeFuseLinkCallback) {
    HiLog.info(TAG, `resumeFuseLinkHome start`);
    GlobalContext.load('dlpRpcProxy').resumeFuseLink(src, callback);
  }

  stopFuseLinkHome(src:string, callback: stopFuseLinkCallback) {
    HiLog.info(TAG, `stopFuseLinkHome start`);
    GlobalContext.load('dlpRpcProxy').stopFuseLink(src, callback);
  }

  async recoverDLPFileHome(src:string, pathUri: string, callback: recoverDlpFileCallback) {
    HiLog.info(TAG, `recoverDLPFileHome start`);
    await GlobalContext.load('dlpRpcProxy').recoverDlpFile(src, pathUri, callback);
  }

  async sandBoxLinkFileHome(linkFileName: string, callerToken: number, callback: sandBoxLinkFileCallback) {
    HiLog.info(TAG, `sandBoxLinkFileHome start`);
    await GlobalContext.load('dlpRpcProxy').sandBoxLinkFile(linkFileName, callerToken, callback);
  }

  async fileOpenHistoryHome(src:string, callback: fileOpenHistoryCallback) {
    HiLog.info(TAG, `fileOpenHistoryHome start: ${src}`);
    await GlobalContext.load('dlpRpcProxy').fileOpenHistory(src, callback);
  }

  async linkSetHome(src:string, callback: linkSetCallback) {
    HiLog.info(TAG, `linkSetHome start: ${src}`);
    await GlobalContext.load('dlpRpcProxy').linkSet(src, callback);
  }

  disconnectServiceExtAbility(callback: (ans:number) => void) {
    HiLog.info(TAG, `disconnectServiceExtAbility`);
    if (this.context) {
      this.context.disconnectServiceExtensionAbility(this.connection).then(() => {
        callback(SUCCESS_CODE);
      }).catch((error: BusinessError) => {
        HiLog.error(TAG, `disconnectAbility failed: ${JSON.stringify(error)}`);
        callback(ERROR_CODE);
      })
    }
  };
}