OpenHarmony-v6.0-Release/s

/*
 * Copyright (c) 2025 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 "enable_upgrade_manager.h"

#include "file_operator.h"
#include "ime_info_inquirer.h"
#include "os_account_adapter.h"
#include "parameter.h"
#include "settings_data_utils.h"
namespace OHOS {
namespace MiscServices {
constexpr const char *IME_CFG_FILE_PATH = "/data/service/el1/public/imf/ime_cfg.json";
EnableUpgradeManager &EnableUpgradeManager::GetInstance()
{
    static EnableUpgradeManager instance;
    return instance;
}

int32_t EnableUpgradeManager::Upgrade(int32_t userId, const std::vector<FullImeInfo> &imeInfos)
{
    std::lock_guard<std::mutex> lock(upgradedLock_);
    if (upgradedUserId_.count(userId)) {
        return ErrorCode::NO_ERROR;
    }
    std::string userContent;
    auto ret = GetUserEnabledTable(userId, userContent);
    if (ret != ErrorCode::NO_ERROR && ret != ErrorCode::ERROR_KEYWORD_NOT_FOUND) {
        return ret;
    }
    bool needUpgrade = userContent.find("version") == std::string::npos;
    if (!needUpgrade) {
        upgradedUserId_.insert(userId);
        return ErrorCode::NO_ERROR;
    }
    ImeEnabledCfg cfg;
    cfg.version = GetDisplayVersion();
    ret = MergeTwoTable(userId, cfg.enabledInfos);
    if (ret != ErrorCode::NO_ERROR) {
        return ret;
    }
    ret = PaddedByBundleMgr(userId, imeInfos, cfg.enabledInfos);
    if (ret != ErrorCode::NO_ERROR) {
        return ret;
    }
    ret = PaddedByImePersistCfg(userId, cfg.enabledInfos);
    if (ret != ErrorCode::NO_ERROR) {
        return ret;
    }
    std::string newContent;
    cfg.Marshall(newContent);
    if (!SetUserEnabledTable(userId, newContent)) {
        IMSA_HILOGE("%{public}d set enabled table failed.", userId);
        return ErrorCode::ERROR_ENABLE_IME;
    }
    upgradedUserId_.insert(userId);
    return ErrorCode::NO_ERROR;
}

int32_t EnableUpgradeManager::GetEnabledTable(int32_t userId, std::set<std::string> &bundleNames)
{
    auto ret = GetGlobalEnabledTable(userId, bundleNames);
    if (ret == ErrorCode::ERROR_EX_PARCELABLE) {
        return GetUserEnabledTable(userId, bundleNames);
    }
    return ret;
}

int32_t EnableUpgradeManager::GetGlobalEnabledTable(int32_t userId, std::string &content)
{
    return GetEnabledTable(userId, SETTING_URI_PROXY, content);
}

int32_t EnableUpgradeManager::GetUserEnabledTable(int32_t userId, std::string &content)
{
    std::string uriProxy = SETTINGS_USER_DATA_URI + std::to_string(userId) + "?Proxy=true";
    return GetEnabledTable(userId, uriProxy, content);
}

int32_t EnableUpgradeManager::GetGlobalEnabledTable(int32_t userId, std::set<std::string> &bundleNames)
{
    return GetEnabledTable(userId, SETTING_URI_PROXY, bundleNames);
}

int32_t EnableUpgradeManager::GetUserEnabledTable(int32_t userId, std::set<std::string> &bundleNames)
{
    std::string uriProxy = SETTINGS_USER_DATA_URI + std::to_string(userId) + "?Proxy=true";
    return GetEnabledTable(userId, uriProxy, bundleNames);
}

int32_t EnableUpgradeManager::GetEnabledTable(
    int32_t userId, const std::string &uriProxy, std::set<std::string> &bundleNames)
{
    std::string content;
    auto ret = GetEnabledTable(userId, uriProxy, content);
    if (ret != ErrorCode::NO_ERROR) {
        return ret;
    }
    return ParseEnabledTable(userId, content, bundleNames);
}

int32_t EnableUpgradeManager::GetEnabledTable(int32_t userId, const std::string &uriProxy, std::string &content)
{
    return SettingsDataUtils::GetInstance().GetStringValue(uriProxy, SettingsDataUtils::ENABLE_IME, content);
}

int32_t EnableUpgradeManager::ParseEnabledTable(
    int32_t userId, std::string &content, std::set<std::string> &bundleNames)
{
    EnabledImeCfg oldCfg;
    oldCfg.userImeCfg.userId = std::to_string(userId);
    if (!oldCfg.Unmarshall(content)) {
        IMSA_HILOGE("%{public}d Unmarshall failed:%{public}s.", userId, content.c_str());
        return ErrorCode::ERROR_EX_PARCELABLE;
    }
    bundleNames = std::set<std::string>(oldCfg.userImeCfg.identities.begin(), oldCfg.userImeCfg.identities.end());
    return ErrorCode::NO_ERROR;
}

int32_t EnableUpgradeManager::GetFullExperienceTable(int32_t userId, std::set<std::string> &bundleNames)
{
    std::string content;
    int32_t ret = SettingsDataUtils::GetInstance().GetStringValue(SETTING_URI_PROXY,
        SettingsDataUtils::SECURITY_MODE, content);
    if (ret != ErrorCode::NO_ERROR) {
        IMSA_HILOGW("%{public}d get full experience table failed:%{public}d.", userId, ret);
        return ret;
    }
    SecurityModeCfg cfg;
    cfg.userImeCfg.userId = std::to_string(userId);
    if (!cfg.Unmarshall(content)) {
        IMSA_HILOGE("%{public}d Unmarshall failed:%{public}s.", userId, content.c_str());
        return ErrorCode::ERROR_EX_PARCELABLE;
    }
    bundleNames = std::set<std::string>(cfg.userImeCfg.identities.begin(), cfg.userImeCfg.identities.end());
    return ErrorCode::NO_ERROR;
}

int32_t EnableUpgradeManager::MergeTwoTable(int32_t userId, std::vector<ImeEnabledInfo> &enabledInfos)
{
    std::set<std::string> enabledBundleNames;
    auto ret = GetEnabledTable(userId, enabledBundleNames);
    if (ret != ErrorCode::NO_ERROR && ret != ErrorCode::ERROR_KEYWORD_NOT_FOUND &&
        ret != ErrorCode::ERROR_EX_PARCELABLE) {
        IMSA_HILOGE("%{public}d get enabled table failed:%{public}d.", userId, ret);
        return ret;
    }
    std::set<std::string> fullModeBundleNames;
    ret = GetFullExperienceTable(userId, fullModeBundleNames);
    if (ret != ErrorCode::NO_ERROR && ret != ErrorCode::ERROR_KEYWORD_NOT_FOUND &&
        ret != ErrorCode::ERROR_EX_PARCELABLE) {
        IMSA_HILOGE("%{public}d get full experience table failed:%{public}d.", userId, ret);
        return ret;
    }
    for (const auto &bundleName : enabledBundleNames) {
        ImeEnabledInfo tmpInfo;
        tmpInfo.bundleName = bundleName;
        tmpInfo.enabledStatus = EnabledStatus::BASIC_MODE;
        enabledInfos.push_back(tmpInfo);
    }
    for (const auto &bundleName : fullModeBundleNames) {
        auto iter = std::find_if(enabledInfos.begin(), enabledInfos.end(),
            [&bundleName](const ImeEnabledInfo &info) { return info.bundleName == bundleName; });
        if (iter != enabledInfos.end()) {
            iter->enabledStatus = EnabledStatus::FULL_EXPERIENCE_MODE;
            continue;
        }
        ImeEnabledInfo tmpInfo;
        tmpInfo.bundleName = bundleName;
        tmpInfo.enabledStatus = EnabledStatus::FULL_EXPERIENCE_MODE;
        enabledInfos.push_back(tmpInfo);
    }
    return ErrorCode::NO_ERROR;
}

int32_t EnableUpgradeManager::PaddedByBundleMgr(
    int32_t userId, const std::vector<FullImeInfo> &imeInfos, std::vector<ImeEnabledInfo> &enabledInfos)
{
    auto finalImeInfos = imeInfos;
    if (finalImeInfos.empty()) {
        auto ret = ImeInfoInquirer::GetInstance().QueryFullImeInfo(userId, finalImeInfos, true);
        if (ret != ErrorCode::NO_ERROR) {
            IMSA_HILOGE("%{public}d QueryFullImeInfo failed:%{public}d.", userId, ret);
            return ret;
        }
    }
    IMSA_HILOGI("ime size, enabled:%{public}zu, installed:%{public}zu.", enabledInfos.size(), finalImeInfos.size());
    for (const auto &info : finalImeInfos) {
        auto iter = std::find_if(enabledInfos.begin(), enabledInfos.end(),
            [&info](const ImeEnabledInfo &enabledInfoTmp) { return enabledInfoTmp.bundleName == info.prop.name; });
        if (iter != enabledInfos.end()) {
            IMSA_HILOGI("%{public}d/%{public}s:%{public}d before upgrade.", userId, info.prop.name.c_str(),
                static_cast<int32_t>(iter->enabledStatus));
            iter->extensionName = info.prop.id;
            continue;
        }
        IMSA_HILOGI("%{public}d/%{public}s is disabled before upgrade or sys ime.", userId, info.prop.name.c_str());
        enabledInfos.emplace_back(
            info.prop.name, info.prop.id, ComputeEnabledStatus(info.prop.name, EnabledStatus::DISABLED));
    }
    return ErrorCode::NO_ERROR;
}
// LCOV_EXCL_START
void EnableUpgradeManager::UpdateGlobalEnabledTable(int32_t userId, const ImeEnabledCfg &newEnabledCfg)
{
    if (!ImeInfoInquirer::GetInstance().GetSystemConfig().enableInputMethodFeature) {
        return;
    }
    auto currentUserId = OsAccountAdapter::GetForegroundOsAccountLocalId();
    if (userId != currentUserId) {
        IMSA_HILOGW("[%{public}d,%{public}d] not same.", userId, currentUserId);
        return;
    }
    std::vector<std::string> newBundleNames;
    for (const auto &info : newEnabledCfg.enabledInfos) {
        if (info.enabledStatus != EnabledStatus::DISABLED &&
            ImeInfoInquirer::GetInstance().IsImeInstalled(userId, info.bundleName, info.extensionName)) {
            newBundleNames.emplace_back(info.bundleName);
        }
    }
    std::string oldGlobalContent;
    auto ret = GetGlobalEnabledTable(userId, oldGlobalContent);
    if (ret != ErrorCode::NO_ERROR && ret != ErrorCode::ERROR_KEYWORD_NOT_FOUND) {
        IMSA_HILOGW("%{public}d get global enabled table failed:%{public}d.", userId, ret);
        return;
    }
    IMSA_HILOGD("old global content:%{public}s.", oldGlobalContent.c_str());
    auto newGlobalContent = GenerateGlobalContent(userId, newBundleNames);
    if (newGlobalContent.empty()) {
        IMSA_HILOGE("newGlobalContent is empty.");
        return;
    }
    if (newGlobalContent == oldGlobalContent) {
        IMSA_HILOGD("content same, not deal.");
        return;
    }
    IMSA_HILOGD("global content, old:%{public}s, new:%{public}s.", oldGlobalContent.c_str(), newGlobalContent.c_str());
    auto globalTableUserId = GetGlobalTableUserId(oldGlobalContent);
    if (globalTableUserId == -1 || userId == globalTableUserId) {
        SetGlobalEnabledTable(newGlobalContent);
        return;
    }
    ret = Upgrade(globalTableUserId, {});
    if (ret != ErrorCode::NO_ERROR) {
        IMSA_HILOGW("%{public}d Upgrade failed:%{public}d.", userId, ret);
        return;
    }
    SetGlobalEnabledTable(newGlobalContent);
}
// LCOV_EXCL_STOP
int32_t EnableUpgradeManager::GetGlobalTableUserId(const std::string &valueStr)
{
    auto root = cJSON_Parse(valueStr.c_str());
    if (root == nullptr) {
        IMSA_HILOGE("valueStr content parse failed!");
        return -1;
    }
    auto subNode = Serializable::GetSubNode(root, "enableImeList");
    if (subNode == nullptr || !cJSON_IsObject(subNode)) {
        IMSA_HILOGW("subNode is null or not object");
        cJSON_Delete(root);
        return -1;
    }
    if (subNode->child == nullptr) {
        IMSA_HILOGW("subNode has not child");
        cJSON_Delete(root);
        return -1;
    }
    std::string userId = subNode->child->string;
    cJSON_Delete(root);
    return atoi(userId.c_str());
}

std::string EnableUpgradeManager::GenerateGlobalContent(int32_t userId, const std::vector<std::string> &bundleNames)
{
    EnabledImeCfg cfg;
    cfg.userImeCfg.identities = bundleNames;
    std::string newGlobalContent;
    cfg.userImeCfg.userId = std::to_string(userId);
    if (!cfg.Marshall(newGlobalContent)) {
        IMSA_HILOGE("%{public}d Marshall failed.", userId);
    }
    return newGlobalContent;
}
// LCOV_EXCL_START
bool EnableUpgradeManager::SetGlobalEnabledTable(const std::string &content)
{
    return SetEnabledTable(SETTING_URI_PROXY, content);
}

bool EnableUpgradeManager::SetUserEnabledTable(int32_t userId, const std::string &content)
{
    std::string uriProxy = SETTINGS_USER_DATA_URI + std::to_string(userId) + "?Proxy=true";
    return SetEnabledTable(uriProxy, content);
}

bool EnableUpgradeManager::SetEnabledTable(const std::string &uriProxy, const std::string &content)
{
    return SettingsDataUtils::GetInstance().SetStringValue(uriProxy, SettingsDataUtils::ENABLE_IME, content);
}
// LCOV_EXCL_STOP
int32_t EnableUpgradeManager::GetImePersistCfg(int32_t userId, ImePersistInfo &persisInfo)
{
    if (!FileOperator::IsExist(IME_CFG_FILE_PATH)) {
        IMSA_HILOGD("ime cfg file not found.");
        return ErrorCode::NO_ERROR;
    }
    std::string content;
    if (!FileOperator::Read(IME_CFG_FILE_PATH, content)) {
        IMSA_HILOGE("failed to read persist info!");
        return ErrorCode::ERROR_PERSIST_CONFIG;
    }
    IMSA_HILOGI("content: %{public}s", content.c_str());
    ImePersistCfg cfg;
    if (!cfg.Unmarshall(content)) {
        IMSA_HILOGE("Unmarshall failed!");
        return ErrorCode::NO_ERROR;
    }
    auto iter = std::find_if(cfg.imePersistInfo.begin(), cfg.imePersistInfo.end(),
        [userId](const auto &infoTmp) { return infoTmp.userId == userId; });
    if (iter != cfg.imePersistInfo.end()) {
        persisInfo = *iter;
    }
    return ErrorCode::NO_ERROR;
}

int32_t EnableUpgradeManager::PaddedByImePersistCfg(int32_t userId, std::vector<ImeEnabledInfo> &enabledInfos)
{
    ImePersistInfo persistInfo;
    auto ret = GetImePersistCfg(userId, persistInfo);
    if (ret != ErrorCode::NO_ERROR) {
        return ret;
    }
    std::string currentBundleName;
    std::string currentExtName;
    auto pos = persistInfo.currentIme.find('/');
    if (pos != std::string::npos && pos + 1 < persistInfo.currentIme.size()) {
        currentBundleName = persistInfo.currentIme.substr(0, pos);
        currentExtName = persistInfo.currentIme.substr(pos + 1);
    }
    auto iter = std::find_if(
        enabledInfos.begin(), enabledInfos.end(), [&currentBundleName, &currentExtName](const auto &infoTmp) {
            return infoTmp.bundleName == currentBundleName && infoTmp.extensionName == currentExtName &&
                   infoTmp.enabledStatus != EnabledStatus::DISABLED;
        });
    if (iter != enabledInfos.end()) {
        iter->extraInfo.isDefaultIme = true;
        iter->extraInfo.isDefaultImeSet = persistInfo.isDefaultImeSet;
        iter->extraInfo.currentSubName = persistInfo.currentSubName;
    } else {
        auto sysIter = std::find_if(enabledInfos.begin(), enabledInfos.end(), [](const ImeEnabledInfo &enabledInfoTmp) {
            return enabledInfoTmp.bundleName == ImeInfoInquirer::GetInstance().GetDefaultIme().bundleName;
        });
        if (sysIter != enabledInfos.end()) {
            sysIter->extraInfo.isDefaultIme = true;
        }
    }

    std::string tmpBundleName;
    std::string tmpExtName;
    pos = persistInfo.tempScreenLockIme.find('/');
    if (pos != std::string::npos && pos + 1 < persistInfo.tempScreenLockIme.size()) {
        tmpBundleName = persistInfo.tempScreenLockIme.substr(0, pos);
        tmpExtName = persistInfo.tempScreenLockIme.substr(pos + 1);
    }
    iter = std::find_if(enabledInfos.begin(), enabledInfos.end(), [&tmpBundleName, &tmpExtName](const auto &infoTmp) {
        return infoTmp.bundleName == tmpBundleName && infoTmp.extensionName == tmpExtName;
    });
    if (iter != enabledInfos.end()) {
        iter->extraInfo.isTmpIme = true;
    }
    return ErrorCode::NO_ERROR;
}

EnabledStatus EnableUpgradeManager::ComputeEnabledStatus(const std::string &bundleName, EnabledStatus initStatus)
{
    auto sysCfg = ImeInfoInquirer::GetInstance().GetSystemConfig();
    auto hasEnableSwitch = sysCfg.enableInputMethodFeature;
    auto hasFullExperienceSwitch = sysCfg.enableFullExperienceFeature;
    IMSA_HILOGI("enable cfg:[%{public}d, %{public}d].", hasEnableSwitch, hasFullExperienceSwitch);
    if (!hasEnableSwitch && !hasFullExperienceSwitch) {
        return EnabledStatus::FULL_EXPERIENCE_MODE;
    }
    auto sysIme = ImeInfoInquirer::GetInstance().GetDefaultIme();
    if (bundleName == sysIme.bundleName && initStatus == EnabledStatus::DISABLED) {
        return EnabledStatus::BASIC_MODE;
    }
    return initStatus;
}
} // namespace MiscServices
} // namespace OHOS