/*
* 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.
*/

#define LOG_TAG "ObjectSnapshot"
#include "object_snapshot.h"

#include "eventcenter/event_center.h"
#include "log_print.h"
#include "snapshot/bind_event.h"
#include "store/general_store.h"
#include "utils/anonymous.h"
namespace OHOS::DistributedObject {
using namespace OHOS::DistributedData;
int32_t ObjectSnapshot::Upload(Asset& asset)
{
    if (!IsBoundAsset(asset)) {
        return 0;
    }
    return assetMachine_->DFAPostEvent(UPLOAD, changedAssets_[asset.uri], asset);
}

bool ObjectSnapshot::IsBoundAsset(const Asset& asset)
{
    auto it = changedAssets_.find(asset.uri);
    if (it != changedAssets_.end()) {
        return true;
    }
    return false;
}

int32_t ObjectSnapshot::Download(Asset& asset)
{
    if (!IsBoundAsset(asset)) {
        return 0;
    }
    return assetMachine_->DFAPostEvent(DOWNLOAD, changedAssets_[asset.uri], asset);
}

TransferStatus ObjectSnapshot::GetAssetStatus(Asset& asset)
{
    if (!IsBoundAsset(asset)) {
        return STATUS_BUTT;
    }
    return changedAssets_[asset.uri].status;
}

int32_t ObjectSnapshot::Uploaded(Asset& asset)
{
    if (!IsBoundAsset(asset)) {
        return E_OK;
    }
    return assetMachine_->DFAPostEvent(UPLOAD_FINISHED, changedAssets_[asset.uri], asset);
}

int32_t ObjectSnapshot::Downloaded(Asset& asset)
{
    if (!IsBoundAsset(asset)) {
        return E_OK;
    }
    return assetMachine_->DFAPostEvent(DOWNLOAD_FINISHED, changedAssets_[asset.uri], asset);
}

int32_t ObjectSnapshot::OnDataChanged(Asset& asset, const std::string& deviceId)
{
    if (!IsBoundAsset(asset)) {
        return E_OK;
    }
    std::pair<std::string, Asset> newAsset{ deviceId, asset };
    return assetMachine_->DFAPostEvent(REMOTE_CHANGED, changedAssets_[asset.uri], asset, newAsset);
}

int32_t ObjectSnapshot::BindAsset(const Asset& asset, const DistributedData::AssetBindInfo& bindInfo,
    const StoreInfo& storeInfo)
{
    if (IsBoundAsset(asset)) {
        ZLOGD("Asset is bound. asset.uri:%{public}s :", Anonymous::Change(asset.uri, true).c_str());
        return E_OK;
    }
    changedAssets_[asset.uri] = ChangedAssetInfo(asset, bindInfo, storeInfo);
    return E_OK;
}

int32_t ObjectSnapshot::Transferred(Asset& asset)
{
    if (!IsBoundAsset(asset)) {
        return E_OK;
    }
    return assetMachine_->DFAPostEvent(TRANSFER_FINISHED, changedAssets_[asset.uri], asset);
}
ObjectSnapshot::~ObjectSnapshot() {}

ObjectSnapshot::ObjectSnapshot()
{
    assetMachine_ = std::make_shared<ObjectAssetMachine>();
}
} // namespace OHOS::DistributedObject