/*
 * Copyright (c) 2021-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.
 */

#ifndef OS_ACCOUNT_SERVICES_ACCOUNTMGR_INCLUDE_OSACCOUNT_IINNER_OS_ACCOUNT_MANAGER_H
#define OS_ACCOUNT_SERVICES_ACCOUNTMGR_INCLUDE_OSACCOUNT_IINNER_OS_ACCOUNT_MANAGER_H

#include <memory>
#include "iinner_os_account.h"
#include "inner_domain_account_manager.h"
#include "ios_account_control.h"
#include "ios_account_subscribe.h"
#include "ohos_account_manager.h"
#include "os_account_interface.h"
#include "os_account_plugin_manager.h"
#include "safe_map.h"
#include "singleton.h"

namespace OHOS {
namespace AccountSA {
class IInnerOsAccountManager : public IInnerOsAccount {
public:
    static IInnerOsAccountManager &GetInstance();
    void Init() override;
    ErrCode CreateOsAccount(
        const std::string &name, const OsAccountType &type, OsAccountInfo &osAccountInfo) override;
    ErrCode CreateOsAccount(const std::string &localName, const std::string &shortName,
        const OsAccountType &type, OsAccountInfo &osAccountInfo, const CreateOsAccountOptions &options = {}) override;
    ErrCode CreateOsAccountWithFullInfo(OsAccountInfo &osAccountInfo,
        const CreateOsAccountOptions &options = {}) override;
    ErrCode UpdateOsAccountWithFullInfo(OsAccountInfo &osAccountInfo) override;
    ErrCode CreateOsAccountForDomain(const OsAccountType &type, const DomainAccountInfo &domainInfo,
        const sptr<IDomainAccountCallback> &callback, const CreateOsAccountForDomainOptions &options = {}) override;
    ErrCode RemoveOsAccount(const int id) override;
    ErrCode IsOsAccountExists(const int id, bool &isOsAccountExits) override;
    ErrCode IsOsAccountActived(const int id, bool &isOsAccountActived) override;
    ErrCode IsOsAccountConstraintEnable(
        const int id, const std::string &constraint, bool &isOsAccountConstraintEnable) override;
    ErrCode IsOsAccountVerified(const int id, bool &isVerified) override;
    ErrCode GetCreatedOsAccountsCount(unsigned int &createdOsAccountCount) override;
    ErrCode QueryMaxOsAccountNumber(uint32_t &maxOsAccountNumber) override;
    ErrCode QueryMaxLoggedInOsAccountNumber(uint32_t &maxNum) override;
    ErrCode GetOsAccountAllConstraints(const int id, std::vector<std::string> &constraints) override;
    ErrCode QueryAllCreatedOsAccounts(std::vector<OsAccountInfo> &osAccountInfos) override;
    ErrCode QueryOsAccountById(const int id, OsAccountInfo &osAccountInfo) override;
    ErrCode GetOsAccountShortName(const int id, std::string &shortName) override;
    ErrCode GetOsAccountName(const int id, std::string &name) override;
    ErrCode GetOsAccountType(const int id, OsAccountType &type) override;
    ErrCode GetOsAccountProfilePhoto(const int id, std::string &photo) override;
    ErrCode IsMultiOsAccountEnable(bool &isMultiOsAccountEnable) override;
    ErrCode SetOsAccountName(const int id, const std::string &name) override;
    ErrCode SetOsAccountConstraints(
        const int id, const std::vector<std::string> &constraints, const bool enable) override;
    ErrCode SetOsAccountProfilePhoto(const int id, const std::string &photo) override;
    ErrCode ActivateOsAccount(const int id, const bool startStorage = true,
        const uint64_t displayId = 0, bool isAppRecovery = false) override;
    ErrCode DeactivateOsAccount(const int id, bool isStopStorage = true) override;
    ErrCode StartOsAccount(const int id) override;
    ErrCode GetOsAccountLocalIdBySerialNumber(const int64_t serialNumber, int &id) override;
    ErrCode GetSerialNumberByOsAccountLocalId(const int &id, int64_t &serialNumber) override;
    ErrCode SubscribeOsAccount(
        const OsAccountSubscribeInfo &subscribeInfo, const sptr<IRemoteObject> &eventListener) override;
    ErrCode UnsubscribeOsAccount(const sptr<IRemoteObject> &eventListener) override;
    const std::shared_ptr<OsAccountSubscribeInfo> GetSubscribeRecordInfo(
        const sptr<IRemoteObject> &eventListener) override;
    OS_ACCOUNT_SWITCH_MOD GetOsAccountSwitchMod() override;
    ErrCode IsOsAccountCompleted(const int id, bool &isOsAccountCompleted) override;
    ErrCode SetOsAccountIsVerified(const int id, const bool isVerified) override;
    ErrCode SetOsAccountIsLoggedIn(const int32_t id, const bool isLoggedIn) override;
    ErrCode GetOsAccountCredentialId(const int id, uint64_t &credentialId) override;
    ErrCode SetOsAccountCredentialId(const int id, uint64_t credentialId) override;
    ErrCode IsAllowedCreateAdmin(bool &isAllowedCreateAdmin) override;
    ErrCode GetOsAccountLocalIdFromDomain(const DomainAccountInfo &domainInfo, int &id) override;
    ErrCode GetCreatedOsAccountNumFromDatabase(const std::string& storeID,
        int &createdOsAccountNum) override;
    ErrCode GetSerialNumberFromDatabase(const std::string& storeID, int64_t &serialNumber) override;
    ErrCode GetMaxAllowCreateIdFromDatabase(const std::string& storeID, int &id) override;
    ErrCode GetOsAccountFromDatabase(const std::string& storeID, const int id,
        OsAccountInfo &osAccountInfo) override;
    ErrCode GetOsAccountListFromDatabase(const std::string& storeID,
        std::vector<OsAccountInfo> &osAccountList) override;
    ErrCode QueryActiveOsAccountIds(std::vector<int32_t>& ids) override;
    ErrCode QueryOsAccountConstraintSourceTypes(const int32_t id,
        const std::string &constraint, std::vector<ConstraintSourceTypeInfo> &constraintSourceTypeInfos) override;
    ErrCode SetBaseOsAccountConstraints(const int32_t id,
        const std::vector<std::string> &constraints, const bool enable) override;
    ErrCode SetGlobalOsAccountConstraints(const std::vector<std::string> &constraints,
        const bool enable, const int32_t enforcerId, const bool isDeviceOwner) override;
    ErrCode SetSpecificOsAccountConstraints(const std::vector<std::string> &constraints,
        const bool enable, const int32_t targetId, const int32_t enforcerId, const bool isDeviceOwner) override;

    ErrCode SetDefaultActivatedOsAccount(const int32_t id) override;
    ErrCode GetDefaultActivatedOsAccount(int32_t &id) override;
    ErrCode ValidateShortName(const std::string &shortName) override;
    ErrCode IsOsAccountForeground(const int32_t localId, const uint64_t displayId, bool &isForeground) override;
    ErrCode GetForegroundOsAccountLocalId(const uint64_t displayId, int32_t &localId) override;
    ErrCode GetForegroundOsAccounts(std::vector<ForegroundOsAccount> &accounts) override;
    ErrCode GetBackgroundOsAccountLocalIds(std::vector<int32_t> &localIds) override;
    ErrCode SetOsAccountToBeRemoved(int32_t localId, bool toBeRemoved) override;
    ErrCode BindDomainAccount(const OsAccountType &type, const DomainAccountInfo &domainAccountInfo,
        const sptr<IDomainAccountCallback> &callback, const CreateOsAccountForDomainOptions &options = {});
    ErrCode SendMsgForAccountCreate(OsAccountInfo &osAccountInfo, const CreateOsAccountOptions &options = {});
    ErrCode GetOsAccountInfoById(const int id, OsAccountInfo &osAccountInfo);
    ErrCode UpdateAccountStatusForDomain(const int id, DomainAccountStatus status);
    ErrCode GetTypeNumber(const OsAccountType& type, int32_t& typeNumber) override;
    ErrCode CheckTypeNumber(const OsAccountType& type) override;
    ErrCode UpdateAccountInfoByDomainAccountInfo(int32_t userId, const DomainAccountInfo &newDomainAccountInfo);
    bool IsSameAccount(const DomainAccountInfo &domainInfoSrc, const DomainAccountInfo &domainInfoTar);
    ErrCode ActivateDefaultOsAccount() override;
    int32_t CleanGarbageOsAccounts(int32_t excludeId = -1) override;
    void ResetAccountStatus() override;
    bool CheckAndCleanOsAccounts();

private:
    IInnerOsAccountManager();
    ~IInnerOsAccountManager() = default;
    DISALLOW_COPY_AND_MOVE(IInnerOsAccountManager);
    void SetOsAccountControl(std::shared_ptr<IOsAccountControl> ptr);
    void RestartActiveAccount();
    void CreateBaseAdminAccount();
    void CreateBaseStandardAccount();
    ErrCode PrepareRemoveOsAccount(OsAccountInfo &osAccountInfo, bool isCleanGarbage = false);
    ErrCode RemoveOsAccountOperate(const int id, OsAccountInfo &osAccountInfo, bool isCleanGarbage = false);
    ErrCode DeactivateOsAccountById(const int id);
    ErrCode DeactivateOsAccountByInfo(OsAccountInfo &osAccountInfo);
    ErrCode PrepareOsAccountInfo(const std::string &name, const OsAccountType &type,
        const DomainAccountInfo &domainAccount, OsAccountInfo &osAccountInfo);
    ErrCode PrepareOsAccountInfo(const std::string &localName, const std::string &shortName, const OsAccountType &type,
        const DomainAccountInfo &domainAccount, OsAccountInfo &osAccountInfo);
    ErrCode FillOsAccountInfo(const std::string &localName, const std::string &shortName, const OsAccountType &type,
        const DomainAccountInfo &domainAccount, OsAccountInfo &osAccountInfo);
    ErrCode PrepareOsAccountInfoWithFullInfo(OsAccountInfo &osAccountInfo);
    ErrCode SendMsgForAccountActivate(OsAccountInfo &osAccountInfo, const bool startStorage = true,
                                      const uint64_t dispalyId = 0, const bool isAppRecovery = false);
    ErrCode SendMsgForAccountDeactivate(OsAccountInfo &osAccountInfo, bool isStopStorage = true);
    ErrCode SendMsgForAccountStop(OsAccountInfo &osAccountInfo);
    ErrCode SendMsgForAccountRemove(OsAccountInfo &osAccountInfo);
    ErrCode ValidateOsAccount(const OsAccountInfo &osAccountInfo);
    void RemoveLocalIdToOperating(int32_t localId);
    bool CheckAndAddLocalIdOperating(int32_t localId);
    ErrCode DealWithDeviceOwnerId(const bool isDeviceOwner, const int32_t localId);
    void CheckAndRefreshLocalIdRecord(const int id);
    bool IsToBeRemoved(int32_t localId);

    // operations for active list
    void PushIdIntoActiveList(int32_t id);
    void EraseIdFromActiveList(int32_t id);
    bool IsOsAccountIDInActiveList(int32_t id);
    void CopyFromActiveList(std::vector<int32_t>& idList);
    bool CheckDomainAccountBound(const std::vector<OsAccountInfo> &osAccountInfos, const DomainAccountInfo &info);
    void RetryToGetAccount(OsAccountInfo &osAccountInfo);
    bool JudgeOsAccountUpdate(Json &accountIndexJson);
    std::shared_ptr<std::mutex> GetOrInsertUpdateLock(int32_t id);
    ErrCode UpdateAccountToForeground(const uint64_t displayId, OsAccountInfo &osAccountInfo);
    ErrCode UpdateAccountToBackground(int32_t oldId);
    ErrCode IsValidOsAccount(const OsAccountInfo &osAccountInfo);
    ErrCode GetRealOsAccountInfoById(const int id, OsAccountInfo &osAccountInfo);
    ErrCode GetNonSACreatedOACount(unsigned int &nonSACreatedOACount) const;

private:
    std::shared_ptr<IOsAccountControl> osAccountControl_;
    std::vector<int32_t> activeAccountId_;
    std::vector<int32_t> operatingId_;
    IOsAccountSubscribe &subscribeManager_;
    std::int32_t deviceOwnerId_ = -1;
    std::int32_t defaultActivatedId_ = -1;
    OsAccountConfig config_;
    mutable std::mutex ativeMutex_;
    mutable std::mutex operatingMutex_;
    mutable std::mutex updateLockMutex_;
    SafeMap<uint64_t, int32_t> foregroundAccountMap_;
    OsAccountPluginManager &pluginManager_;
    SafeMap<int32_t, bool> loggedInAccounts_;
    SafeMap<int32_t, bool> verifiedAccounts_;
    std::map<int32_t, std::shared_ptr<std::mutex>> updateLocks_;
};
}  // namespace AccountSA
}  // namespace OHOS

#endif  // OS_ACCOUNT_SERVICES_ACCOUNTMGR_INCLUDE_OSACCOUNT_IINNER_OS_ACCOUNT_MANAGER_H