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

#include "i18n_hilog.h"
#include "locale_config.h"
#include "os_account_manager.h"
#include "parameter.h"
#include "utils.h"

#include "multi_users.h"

namespace OHOS {
namespace Global {
namespace I18n {
const std::string MultiUsers::MULTI_USERS_LANGUAGE_KEY = "languageData";
const std::string MultiUsers::MULTI_USERS_LOCALE_KEY = "localeData";
const std::string MultiUsers::MULTI_USERS_HOUR_KEY = "is24HourData";
const std::string MultiUsers::INIT_KEY = "init";
const std::string MultiUsers::PREFERENCE_PATH = "/data/service/el1/public/i18n/global/GlobalParamData";
const int32_t MultiUsers::DEFAULT_LOCAL_ID = 100;
const int MultiUsers::CONFIG_LEN = 128;
std::shared_ptr<NativePreferences::Preferences> MultiUsers::preferences = nullptr;

void MultiUsers::InitMultiUser()
{
    InitPreferences();
    if (preferences == nullptr) {
        HILOG_ERROR_I18N("InitMultiUser: InitPreferences failed");
        return;
    }
    bool init = preferences->GetBool(INIT_KEY, false);
    std::string localId;
    I18nErrorCode errCode = GetForegroundLocalId(localId);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("InitMultiUser: get foreground local id failed");
        return;
    }
    if (!init) {
        AddUser(localId);
        preferences->PutBool(INIT_KEY, true);
        preferences->Flush();
        HILOG_INFO_I18N("InitMultiUser: init multi user data success");
    }
}

void MultiUsers::SwitchUser(const std::string& curLocalId)
{
    if (!IsValidLocalId(curLocalId)) {
        HILOG_ERROR_I18N("SwitchUser: curLocalId is an invalid LocalId");
        return;
    }
    I18nErrorCode errCode = LoadGlobalParam(curLocalId);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SwitchUser: load global params failed");
    }
}

void MultiUsers::AddUser(const std::string& localId)
{
    if (!IsValidLocalId(localId)) {
        HILOG_ERROR_I18N("AddUser: localId is invalid");
        return;
    }
    I18nErrorCode errCode = SaveGlobalParam(localId);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("AddUser: add global param failed");
    }
}

void MultiUsers::RemoveUser(const std::string& localId)
{
    if (!IsValidLocalId(localId)) {
        HILOG_ERROR_I18N("RemoveUser: localId is invalid");
        return;
    }
    I18nErrorCode errCode = RemoveGlobalParam(localId);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("RemoveUser: remove global param failed");
    }
}

I18nErrorCode MultiUsers::GetForegroundLocalId(std::string& localId)
{
    int id = 0;
    int errCode = OHOS::AccountSA::OsAccountManager::GetForegroundOsAccountLocalId(id);
    if (errCode != 0) {
        HILOG_ERROR_I18N("GetForegroundLocalId: get foreground locale Id failed, errCode is %{public}d", errCode);
        return I18nErrorCode::FAILED;
    }
    localId = std::to_string(id);
    return I18nErrorCode::SUCCESS;
}

I18nErrorCode MultiUsers::SaveLanguage(const std::string& localId, const std::string& language)
{
    std::string foregroundLocalId = localId;
    I18nErrorCode errCode = I18nErrorCode::SUCCESS;
    if (localId.empty()) {
        errCode = MultiUsers::GetForegroundLocalId(foregroundLocalId);
    }
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveLanguage: get foreground locale Id failed");
        return I18nErrorCode::FAILED;
    }

    errCode =
        WriteMultiUsersParameter(MULTI_USERS_LANGUAGE_KEY, language, foregroundLocalId, false);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveLanguage: save language failed");
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

I18nErrorCode MultiUsers::SaveLocale(const std::string& localId, const std::string& locale)
{
    std::string foregroundLocalId = localId;
    I18nErrorCode errCode = I18nErrorCode::SUCCESS;
    if (localId.empty()) {
        errCode = MultiUsers::GetForegroundLocalId(foregroundLocalId);
    }
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveLocale: get foreground locale Id failed");
        return I18nErrorCode::FAILED;
    }

    errCode =
        WriteMultiUsersParameter(MULTI_USERS_LOCALE_KEY, locale, foregroundLocalId, false);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveLocale: save locale failed");
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

I18nErrorCode MultiUsers::SaveIs24Hour(const std::string& localId, const std::string& is24Hour)
{
    std::string foregroundLocalId = localId;
    I18nErrorCode errCode = I18nErrorCode::SUCCESS;
    if (localId.empty()) {
        errCode = MultiUsers::GetForegroundLocalId(foregroundLocalId);
    }
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveLanguage: get foreground locale Id failed");
        return I18nErrorCode::FAILED;
    }
    errCode =
        WriteMultiUsersParameter(MULTI_USERS_HOUR_KEY, is24Hour, foregroundLocalId, false);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveIs24Hour: save is24Hour failed");
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

I18nErrorCode MultiUsers::SaveGlobalParam(const std::string& localId)
{
    std::string language = ReadSystemParameter(LocaleConfig::GetLanguageKey().data(), CONFIG_LEN);
    std::string locale = ReadSystemParameter(LocaleConfig::GetLocaleKey().data(), CONFIG_LEN);
    std::string is24Hour = ReadSystemParameter(LocaleConfig::GetHourKey().data(), CONFIG_LEN);
    I18nErrorCode errCode = SaveLanguage(localId, language);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveGlobalParam: save language failed");
        return I18nErrorCode::FAILED;
    }
    errCode = SaveLocale(localId, locale);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveGlobalParam: save locale failed");
        return I18nErrorCode::FAILED;
    }
    errCode = SaveIs24Hour(localId, is24Hour);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("SaveGlobalParam: save is24Hour failed");
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

I18nErrorCode MultiUsers::LoadGlobalParam(const std::string& localId)
{
    std::string newLocale = ReadMultiUsersParameter(MULTI_USERS_LOCALE_KEY, localId);
    if (!newLocale.empty() && SetParameter(LocaleConfig::GetLocaleKey().data(), newLocale.data()) != 0) {
        HILOG_ERROR_I18N("LoadGlobalParam: set locale failed");
        return I18nErrorCode::FAILED;
    }

    std::string newLanguage = ReadMultiUsersParameter(MULTI_USERS_LANGUAGE_KEY, localId);
    if (!newLanguage.empty() && LocaleConfig::SetSystemLanguage(newLanguage) != 0) {
        HILOG_ERROR_I18N("LoadGlobalParam: set language failed");
        return I18nErrorCode::FAILED;
    }

    std::string newIs24Hour = ReadMultiUsersParameter(MULTI_USERS_HOUR_KEY, localId);
    if (!newIs24Hour.empty() && LocaleConfig::Set24HourClock(newIs24Hour) != 0) {
        HILOG_ERROR_I18N("LoadGlobalParam: set is24Hour failed");
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

I18nErrorCode MultiUsers::RemoveGlobalParam(const std::string& localId)
{
    I18nErrorCode errCode = WriteMultiUsersParameter(MULTI_USERS_LANGUAGE_KEY, "", localId, true);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("RemoveGlobalParam: remove language failed");
        return I18nErrorCode::FAILED;
    }

    errCode = WriteMultiUsersParameter(MULTI_USERS_LOCALE_KEY, "", localId, true);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("RemoveGlobalParam: remove locale failed");
        return I18nErrorCode::FAILED;
    }

    errCode = WriteMultiUsersParameter(MULTI_USERS_HOUR_KEY, "", localId, true);
    if (errCode != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("RemoveGlobalParam: remove is24Hour failed");
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

std::string MultiUsers::ReadMultiUsersParameter(const std::string& paramKey, const std::string& localId)
{
    std::string param = GetParamFromPreferences(paramKey);
    if (param.empty()) {
        return "";
    }
    std::vector<std::string> multiUsersParam;
    Split(param, ";", multiUsersParam);
    for (auto& userParam : multiUsersParam) {
        std::vector<std::string> content;
        Split(userParam, ":", content);
        // 2 is number of param
        if (content.size() != 2) {
            continue;
        }
        if (content[0] == localId) {
            return content[1];
        }
    }
    return "";
}

I18nErrorCode MultiUsers::WriteMultiUsersParameter(const std::string& paramKey, const std::string& paramValue,
    const std::string& localId, bool isDel)
{
    std::string param = GetParamFromPreferences(paramKey);
    std::vector<std::string> multiUsersParam;
    Split(param, ";", multiUsersParam);
    std::vector<std::string> newMultiUsersParam;
    bool userIsExist = false;
    for (auto& userParam : multiUsersParam) {
        std::vector<std::string> content;
        Split(userParam, ":", content);
        // 2 is number of param
        if (content.size() != 2) {
            continue;
        }
        std::string userLocalId = content[0];
        if (!isDel && userLocalId == localId) {
            content[1] = paramValue;
            Merge(content, ":", userParam);
            userIsExist = true;
        }
        newMultiUsersParam.emplace_back(userParam);
        if (isDel && userLocalId == localId) {
            newMultiUsersParam.pop_back();
        }
    }
    if (!isDel && !userIsExist) {
        newMultiUsersParam.push_back(localId + ":" + paramValue);
    }
    std::string newParam;
    Merge(newMultiUsersParam, ";", newParam);
    if (SetParamFromPreferences(paramKey, newParam) != I18nErrorCode::SUCCESS) {
        HILOG_ERROR_I18N("WriteMultiUsersParameter: set param %{public}s failed", paramKey.c_str());
        return I18nErrorCode::FAILED;
    }
    return I18nErrorCode::SUCCESS;
}

bool MultiUsers::IsValidLocalId(const std::string& localId)
{
    if (std::atoi(localId.c_str()) < DEFAULT_LOCAL_ID) {
        HILOG_ERROR_I18N("IsValidLocalId: invalid local ID");
        return false;
    }
    return true;
}

void MultiUsers::InitPreferences()
{
    if (preferences == nullptr) {
        HILOG_INFO_I18N("InitPreferences: preferences Init");
        OHOS::NativePreferences::Options opt(PREFERENCE_PATH);
        int status;
        preferences = NativePreferences::PreferencesHelper::GetPreferences(opt, status);
        if (status != 0) {
            HILOG_ERROR_I18N("InitPreferences: get preferences failed");
            preferences = nullptr;
        }
    }
}

std::string MultiUsers::GetParamFromPreferences(const std::string& paramKey)
{
    InitPreferences();
    if (preferences == nullptr) {
        HILOG_ERROR_I18N("GetParamFromPreferences: preferences is nullptr");
        return "";
    }
    return preferences->GetString(paramKey, "");
}

I18nErrorCode MultiUsers::SetParamFromPreferences(const std::string& paramKey, const std::string& paramValue)
{
    InitPreferences();
    if (preferences == nullptr) {
        HILOG_ERROR_I18N("SetParamFromPreferences: preferences is nullptr");
        return I18nErrorCode::FAILED;
    }
    int status = preferences->PutString(paramKey, paramValue);
    if (status != 0) {
        HILOG_ERROR_I18N("SetParamFromPreferences: put param %{public}s failed", paramKey.c_str());
        return I18nErrorCode::FAILED;
    }
    preferences->Flush();
    return I18nErrorCode::SUCCESS;
}
} // namespace I18n
} // namespace Global
} // namespace OHOS