/*
 * Copyright (c) 2023-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 OHOS_DM_DEVICEPROFILE_CONNECTOR_H
#define OHOS_DM_DEVICEPROFILE_CONNECTOR_H
#include <string>
#include <algorithm>
#include "access_control_profile.h"
#include "dm_device_info.h"
#include "dm_single_instance.h"

constexpr uint32_t ALLOW_AUTH_ONCE = 1;
constexpr uint32_t ALLOW_AUTH_ALWAYS = 2;

constexpr uint32_t INVALIED_TYPE = 0;
constexpr uint32_t APP_PEER_TO_PEER_TYPE = 1;
constexpr uint32_t APP_ACROSS_ACCOUNT_TYPE = 2;
constexpr uint32_t DEVICE_PEER_TO_PEER_TYPE = 3;
constexpr uint32_t DEVICE_ACROSS_ACCOUNT_TYPE = 4;
constexpr uint32_t IDENTICAL_ACCOUNT_TYPE = 5;

constexpr uint32_t DM_IDENTICAL_ACCOUNT = 1;
constexpr uint32_t DM_POINT_TO_POINT = 256;
constexpr uint32_t DM_ACROSS_ACCOUNT = 1282;

constexpr uint32_t DEVICE = 1;
constexpr uint32_t SERVICE = 2;
constexpr uint32_t APP = 3;

constexpr uint32_t INACTIVE = 0;
constexpr uint32_t ACTIVE = 1;

typedef struct DmDiscoveryInfo {
    std::string pkgname;
    std::string localDeviceId;
    std::string remoteDeviceIdHash;
} DmDiscoveryInfo;

typedef struct DmAclInfo {
    std::string sessionKey;
    int32_t bindType;
    int32_t state;
    std::string trustDeviceId;
    int32_t bindLevel;
    int32_t authenticationType;
    std::string deviceIdHash;
} DmAclInfo;

typedef struct DmAccesser {
    uint64_t requestTokenId;
    std::string requestBundleName;
    int32_t requestUserId;
    std::string requestAccountId;
    std::string requestDeviceId;
    int32_t requestTargetClass;
} DmAccesser;

typedef struct DmAccessee {
    uint64_t trustTokenId;
    std::string trustBundleName;
    int32_t trustUserId;
    std::string trustAccountId;
    std::string trustDeviceId;
    int32_t trustTargetClass;
} DmAccessee;

typedef struct DmOfflineParam {
    uint32_t bindType;
    std::vector<std::string> pkgNameVec;
    int32_t leftAclNumber;
} DmOfflineParam;

namespace OHOS {
namespace DistributedHardware {
class IDeviceProfileConnector {
public:
    virtual ~IDeviceProfileConnector() {}
    virtual std::vector<DistributedDeviceProfile::AccessControlProfile> GetAccessControlProfile() = 0;
    virtual uint32_t CheckBindType(std::string trustDeviceId, std::string requestDeviceId) = 0;
    virtual int32_t PutAccessControlList(DmAclInfo aclInfo, DmAccesser dmAccesser, DmAccessee dmAccessee) = 0;
    virtual int32_t UpdateAccessControlList(int32_t userId, std::string &oldAccountId, std::string &newAccountId) = 0;
    virtual std::unordered_map<std::string, DmAuthForm> GetAppTrustDeviceList(const std::string &pkgName,
        const std::string &deviceId) = 0;
    virtual DmOfflineParam GetOfflineParamFromAcl(std::string trustDeviceId, std::string requestDeviceId) = 0;
    virtual std::vector<int32_t> GetBindTypeByPkgName(std::string pkgName, std::string requestDeviceId,
        std::string trustUdid) = 0;
    virtual std::vector<int32_t> SyncAclByBindType(std::string pkgName, std::vector<int32_t> bindTypeVec,
        std::string localDeviceId, std::string targetDeviceId) = 0;
    virtual int32_t GetDeviceAclParam(DmDiscoveryInfo discoveryInfo, bool &isOnline, int32_t &authForm) = 0;
    virtual int32_t DeleteAccessControlList(int32_t userId, std::string &accountId) = 0;
    virtual DmOfflineParam DeleteAccessControlList(std::string pkgName, std::string localDeviceId,
        std::string remoteDeviceId) = 0;
    virtual std::vector<std::string> GetPkgNameFromAcl(std::string &localDeviceId, std::string &targetDeviceId) = 0;
    virtual bool CheckIdenticalAccount(int32_t userId, const std::string &accountId) = 0;
    virtual int32_t DeleteP2PAccessControlList(int32_t userId, std::string &accountId) = 0;
    virtual bool CheckSrcDevIdInAclForDevBind(const std::string &pkgName, const std::string &deviceId) = 0;
    virtual bool CheckSinkDevIdInAclForDevBind(const std::string &pkgName, const std::string &deviceId) = 0;
    virtual uint32_t DeleteTimeOutAcl(const std::string &deviceId) = 0;
    virtual int32_t GetTrustNumber(const std::string &deviceId) = 0;
    virtual bool CheckDevIdInAclForDevBind(const std::string &pkgName, const std::string &deviceId) = 0;
    virtual bool CheckPkgnameInAcl(std::string pkgName, std::string localDeviceId, std::string remoteDeviceId) = 0;
    virtual std::vector<int32_t> CompareBindType(std::vector<DistributedDeviceProfile::AccessControlProfile> profiles,
        std::string pkgName, std::vector<int32_t> &sinkBindType, std::string localDeviceId,
        std::string targetDeviceId) = 0;
};

class DeviceProfileConnector : public IDeviceProfileConnector {
    DM_DECLARE_SINGLE_INSTANCE(DeviceProfileConnector);
public:
    std::vector<DistributedDeviceProfile::AccessControlProfile> GetAccessControlProfile();
    uint32_t CheckBindType(std::string trustDeviceId, std::string requestDeviceId);
    int32_t PutAccessControlList(DmAclInfo aclInfo, DmAccesser dmAccesser, DmAccessee dmAccessee);
    int32_t UpdateAccessControlList(int32_t userId, std::string &oldAccountId, std::string &newAccountId);
    std::unordered_map<std::string, DmAuthForm> GetAppTrustDeviceList(const std::string &pkgName,
        const std::string &deviceId);
    DmOfflineParam GetOfflineParamFromAcl(std::string trustDeviceId, std::string requestDeviceId);
    std::vector<int32_t> GetBindTypeByPkgName(std::string pkgName, std::string requestDeviceId,
        std::string trustUdid);
    std::vector<int32_t> SyncAclByBindType(std::string pkgName, std::vector<int32_t> bindTypeVec,
        std::string localDeviceId, std::string targetDeviceId);
    int32_t GetDeviceAclParam(DmDiscoveryInfo discoveryInfo, bool &isOnline, int32_t &authForm);
    int32_t DeleteAccessControlList(int32_t userId, std::string &accountId);
    DmOfflineParam DeleteAccessControlList(std::string pkgName, std::string localDeviceId,
        std::string remoteDeviceId);
    void DeleteAclForUserRemoved(int32_t userId);
    std::vector<std::string> GetPkgNameFromAcl(std::string &localDeviceId, std::string &targetDeviceId);
    bool CheckIdenticalAccount(int32_t userId, const std::string &accountId);
    int32_t DeleteP2PAccessControlList(int32_t userId, std::string &accountId);
    bool CheckSrcDevIdInAclForDevBind(const std::string &pkgName, const std::string &deviceId);
    bool CheckSinkDevIdInAclForDevBind(const std::string &pkgName, const std::string &deviceId);
    uint32_t DeleteTimeOutAcl(const std::string &deviceId);
    int32_t GetTrustNumber(const std::string &deviceId);
    bool CheckDevIdInAclForDevBind(const std::string &pkgName, const std::string &deviceId);
    bool CheckPkgnameInAcl(std::string pkgName, std::string localDeviceId, std::string remoteDeviceId);
    std::vector<int32_t> CompareBindType(std::vector<DistributedDeviceProfile::AccessControlProfile> profiles,
        std::string pkgName, std::vector<int32_t> &sinkBindType, std::string localDeviceId, std::string targetDeviceId);
    int32_t IsSameAccount(const std::string &udid);
    int32_t CheckAccessControl(const DmAccessCaller &caller, const std::string &srcUdid,
        const DmAccessCallee &callee, const std::string &sinkUdid);
    int32_t CheckIsSameAccount(const DmAccessCaller &caller, const std::string &srcUdid,
        const DmAccessCallee &callee, const std::string &sinkUdid);
    void DeleteAccessControlList(const std::string &udid);
    std::vector<DistributedDeviceProfile::AccessControlProfile> GetAccessControlProfileByUserId(int32_t userId);
    void DeleteAclForAccountLogOut(const std::string &localUdid, int32_t userId, const std::string &remoteUdid);
    std::map<std::string, int32_t> GetDeviceIdAndBindType(int32_t userId, const std::string &accountId,
        const std::string &localUdid);
private:
    int32_t HandleDmAuthForm(DistributedDeviceProfile::AccessControlProfile profiles, DmDiscoveryInfo discoveryInfo);
    void GetParamBindTypeVec(DistributedDeviceProfile::AccessControlProfile profiles, std::string pkgName,
        std::string requestDeviceId, std::vector<int32_t> &bindTypeVec);
    void ProcessBindType(DistributedDeviceProfile::AccessControlProfile profiles, DmDiscoveryInfo paramInfo,
        std::vector<int32_t> &sinkBindType, std::vector<int32_t> &bindTypeIndex, uint32_t index);
    int32_t GetAuthForm(DistributedDeviceProfile::AccessControlProfile profiles, const std::string &trustDev,
        const std::string &reqDev);
    int32_t CheckAuthForm(DmAuthForm form, DistributedDeviceProfile::AccessControlProfile profiles,
        DmDiscoveryInfo discoveryInfo);
    bool SingleUserProcess(const DistributedDeviceProfile::AccessControlProfile &profile,
        const DmAccessCaller &caller, const DmAccessCallee &callee);
    void UpdateBindType(const std::string &udid, int32_t bindType, std::map<std::string, int32_t> &deviceMap);
};

extern "C" IDeviceProfileConnector *CreateDpConnectorInstance();
using CreateDpConnectorFuncPtr = IDeviceProfileConnector *(*)(void);
} // namespace DistributedHardware
} // namespace OHOS
#endif // OHOS_DM_DEVICEPROFILE_CONNECTOR_H