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

#include "hiappevent_clean.h"

#include <memory>
#include <vector>

#include "app_event_db_cleaner.h"
#include "app_event_log_cleaner.h"
#include "app_event_observer_mgr.h"
#include "hiappevent_base.h"
#include "hiappevent_config.h"
#include "hiappevent_userinfo.h"
#include "hilog/log.h"

#undef LOG_DOMAIN
#define LOG_DOMAIN 0xD002D07

#undef LOG_TAG
#define LOG_TAG "Clean"

namespace OHOS {
namespace HiviewDFX {
namespace HiAppEventClean {
namespace {
constexpr int EVENT_COUNT_OF_CHECK_SPACE = 1000;

static int g_eventCount = -1;

void CreateCleaners(const std::string& dir, std::vector<std::shared_ptr<AppEventCleaner>>& cleaners)
{
    cleaners.push_back(std::make_shared<AppEventDbCleaner>(dir));
    cleaners.push_back(std::make_shared<AppEventLogCleaner>(dir));
}

uint64_t GetCurStorageSize(const std::string& dir)
{
    std::vector<std::shared_ptr<AppEventCleaner>> cleaners;
    CreateCleaners(dir, cleaners);
    uint64_t curSize = 0;
    for (auto& cleaner : cleaners) {
        curSize += cleaner->GetFilesSize();
    }
    return curSize;
}
}
bool IsStorageSpaceFull(const std::string& dir, uint64_t maxSize)
{
    return GetCurStorageSize(dir) > maxSize;
}

bool ReleaseSomeStorageSpace(const std::string& dir, uint64_t maxSize)
{
    HILOG_INFO(LOG_CORE, "start to clear the storage space");
    std::vector<std::shared_ptr<AppEventCleaner>> cleaners;
    CreateCleaners(dir, cleaners);
    auto curSize = GetCurStorageSize(dir);
    for (auto it = cleaners.rbegin(); it != cleaners.rend(); ++it) { // clear the log space first
        curSize = (*it)->ClearSpace(curSize, maxSize);
        if (curSize <= maxSize) {
            return true;
        }
    }
    return curSize <= maxSize;
}

void ClearData(const std::string& dir)
{
    // reset the status of observers
    AppEventObserverMgr::GetInstance().HandleClearUp();

    // clear user ids and properties
    HiAppEvent::UserInfo::GetInstance().ClearData();

    // delete data files
    std::vector<std::shared_ptr<AppEventCleaner>> cleaners;
    CreateCleaners(dir, cleaners);
    for (auto& cleaner : cleaners) { // clear the db data first
        cleaner->ClearData();
    }
}

void CheckStorageSpace()
{
    if (g_eventCount == -1 || g_eventCount >= (EVENT_COUNT_OF_CHECK_SPACE - 1)) {
        std::string dir = HiAppEventConfig::GetInstance().GetStorageDir();
        auto maxSize = HiAppEventConfig::GetInstance().GetMaxStorageSize();
        if (!IsStorageSpaceFull(dir, maxSize)) {
            return;
        }
        HILOG_INFO(LOG_CORE, "hiappevent dir space is full, start to clean");
        ReleaseSomeStorageSpace(dir, maxSize);
    }
    g_eventCount = (++g_eventCount) % EVENT_COUNT_OF_CHECK_SPACE;
}
} // namespace HiAppEventClean
} // namespace HiviewDFX
} // namespace OHOS