/*
 * Copyright (c) 2024 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 common from '@ohos.app.ability.common';
import Want from '@ohos.app.ability.Want';
import { BusinessError } from '@ohos.base';
import rpc from '@ohos.rpc';
import CredCallbackStub from './CredCallback';
import Constants from '../constant';
import { HiLog } from '../HiLog';

const TAG = 'ConnectService';

export default class ConnectService {
  private context: ESObject;
  private connection = -1;
  private optionsSearchUser: common.ConnectOptions;
  private optionsGetAccount: common.ConnectOptions;

  constructor(context: ESObject) {
    let that = this;
    this.context = context;
    this.optionsSearchUser = {
      onConnect: (elementName, remote) => {
        HiLog.info(TAG, `onConnect success search account ${JSON.stringify(elementName)}`);
        that.searchUserInfo(remote);
      },
      onDisconnect: () => {
        HiLog.info(TAG, `onDisconnect`);
      },
      onFailed: () => {
        HiLog.info(TAG, `onFailed`);
      }
    }
    this.optionsGetAccount = {
      onConnect: (elementName, remote) => {
        HiLog.info(TAG, `onConnect success get account ${JSON.stringify(elementName)}`);
        that.getAccountInfo(remote);
      },
      onDisconnect: () => {
        HiLog.info(TAG, `onDisconnect`);
      },
      onFailed: () => {
        HiLog.info(TAG, `onFailed`);
      }
    }
  }

  connectServiceShareAbility(code: number) {
    HiLog.info(TAG, `connectServiceShareAbility start`);
    let want: Want = {
      bundleName: 'com.huawei.hmos.dlpcredmgr',
      abilityName: 'DlpCredDataExtAbility',
    };
    try {
      switch (code) {
        case Constants.COMMAND_SEARCH_USER_INFO: {
          this.connection = this.context.connectServiceExtensionAbility(want, this.optionsSearchUser);
          AppStorage.setOrCreate('connection', this.connection);
          break;
        }
        case Constants.COMMAND_GET_ACCOUNT_INFO: {
          this.connection = this.context.connectServiceExtensionAbility(want, this.optionsGetAccount);
          AppStorage.setOrCreate('connection', this.connection);
          break;
        }
        default: {
          HiLog.error(TAG, `code is not exist ${code}`);
        }
      }
    } catch (err) {
      HiLog.error(TAG, `connectServiceExtAbility failed: ${JSON.stringify(err)}`);
    }
    HiLog.info(TAG, `connectServiceExtAbility result: ${this.connection}`);
  }

  searchUserInfo(remote: rpc.IRemoteObject) {
    let cloudPhone: string | undefined = AppStorage.get('cloudPhone');
    HiLog.info(TAG, `searchUserInfo start`);
    let option = new rpc.MessageOption(Constants.TF_ASYNC);
    let data = new rpc.MessageSequence();
    let reply = new rpc.MessageSequence();
    data.writeInterfaceToken('OHOS.HapDlpCredAbilityServiceStub');
    let callback: CredCallbackStub = new CredCallbackStub('CredCallbackStub');
    data.writeRemoteObject(callback.asObject());
    data.writeString(JSON.stringify({'phone': cloudPhone}));
    if (remote === null) {
      HiLog.info(TAG, `onConnect remote is null.`);
      return;
    }
    remote.sendMessageRequest(Constants.COMMAND_SEARCH_USER_INFO, data, reply, option).then((result) => {
      HiLog.info(TAG, `searchUserInfo success.`);
    }).catch((e: string) => {
      HiLog.info(TAG, `searchUserInfo error: ${e}`);
    });
  }

  getAccountInfo(remote: rpc.IRemoteObject) {
    HiLog.info(TAG, `getAccountInfo start`);
    let option = new rpc.MessageOption(Constants.TF_ASYNC);
    let data = new rpc.MessageSequence();
    let reply = new rpc.MessageSequence();
    data.writeInterfaceToken('OHOS.HapDlpCredAbilityServiceStub');
    let callback: CredCallbackStub = new CredCallbackStub('CredCallbackStub');
    data.writeRemoteObject(callback.asObject());
    if (remote === null) {
      HiLog.info(TAG, `onConnect remote is null.`);
      return;
    }
    remote.sendMessageRequest(Constants.COMMAND_GET_ACCOUNT_INFO, data, reply, option).then((result) => {
      HiLog.info(TAG, `getAccountInfo success.`);
    }).catch((e: string) => {
      HiLog.info(TAG, `getAccountInfo error: ${e}`);
    });
  }

  disconnectServiceShareAbility() {
    HiLog.info(TAG, `disconnectServiceShareAbility: ${AppStorage.get('connection')}`);
    let connectionNum: number | undefined = AppStorage.get('connection');
    this.context.disconnectAbility(connectionNum).then(() => {
      HiLog.info(TAG, `disconnectAbility success.`);
    }).catch((error: BusinessError) => {
      HiLog.error(TAG, `disconnectAbility failed. Error: ${JSON.stringify(error)}`);
    })
  };
}