xref: /ohos5.0/base/account/os_account/services/accountmgr/src/appaccount/app_account_data_storage.cpp

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

#include <sys/stat.h>
#include "app_account_constants.h"
#include "account_file_operator.h"
#include "account_log_wrapper.h"

namespace OHOS {
namespace AccountSA {
namespace {
const std::string AUTHORIZED_ACCOUNTS = "authorizedAccounts";
}

AppAccountDataStorage::AppAccountDataStorage(const std::string &storeId, const AccountDataStorageOptions &options)
    : AccountDataStorage(Constants::APP_ACCOUNT_APP_ID, storeId, options)
{
    if (!options.baseDir.empty()) {
        AccountFileOperator fileOperator;
        fileOperator.CreateDir(options.baseDir, S_IRWXU | S_IRWXG);
    }
}

Json AppAccountDataStorage::GetAccessibleAccountsFromAuthorizedAccounts(const std::string &authorizedAccounts,
    const std::string &authorizedApp, std::vector<std::string> &accessibleAccounts)
{
    accessibleAccounts.clear();

    auto jsonObject = Json::parse(authorizedAccounts, nullptr, false);
    if (jsonObject.is_discarded()) {
        jsonObject = Json::object();
    } else {
        auto value = jsonObject.find(authorizedApp);
        if (value == jsonObject.end()) {
            jsonObject.emplace(authorizedApp, Json::array());
        } else if (value->is_array()) {
            accessibleAccounts = jsonObject[authorizedApp].get<std::vector<std::string>>();
        }
    }

    return jsonObject;
}

ErrCode AppAccountDataStorage::GetAccessibleAccountsFromDataStorage(
    const std::string &authorizedApp, std::vector<std::string> &accessibleAccounts)
{
    std::string authorizedAccounts;
    ErrCode result = GetValueFromKvStore(AUTHORIZED_ACCOUNTS, authorizedAccounts);
    if (result != ERR_OK) {
        ACCOUNT_LOGE("failed to get config by id from data storage");
    }

    GetAccessibleAccountsFromAuthorizedAccounts(authorizedAccounts, authorizedApp, accessibleAccounts);

    return ERR_OK;
}

ErrCode AppAccountDataStorage::GetAccountInfoFromDataStorage(AppAccountInfo &appAccountInfo)
{
    ErrCode result = GetAccountInfoById(appAccountInfo.GetPrimeKey(), appAccountInfo);
    if (result != ERR_OK) {
        ACCOUNT_LOGE("failed to get account info by id, result %{public}d.", result);
        return ERR_APPACCOUNT_SERVICE_GET_ACCOUNT_INFO_BY_ID;
    }

    return ERR_OK;
}

ErrCode AppAccountDataStorage::AddAccountInfoIntoDataStorage(AppAccountInfo &appAccountInfo)
{
    ErrCode result = AddAccountInfo(appAccountInfo);
    if (result != ERR_OK) {
        ACCOUNT_LOGE("failed to add account info, result = %{public}d", result);
        return result;
    }

    return ERR_OK;
}

ErrCode AppAccountDataStorage::SaveAccountInfoIntoDataStorage(AppAccountInfo &appAccountInfo)
{
    ErrCode result = SaveAccountInfo(appAccountInfo);
    if (result != ERR_OK) {
        ACCOUNT_LOGE("failed to save account info, result = %{public}d", result);
        return result;
    }

    return ERR_OK;
}

ErrCode AppAccountDataStorage::DeleteAccountInfoFromDataStorage(AppAccountInfo &appAccountInfo)
{
    ErrCode ret = RemoveValueFromKvStore(appAccountInfo.GetPrimeKey());
    if (ret != ERR_OK) {
        ACCOUNT_LOGE("RemoveValueFromKvStore failed! ret = %{public}d.", ret);
    }
    return ret;
}

void AppAccountDataStorage::SaveEntries(
    std::vector<OHOS::DistributedKv::Entry> allEntries, std::map<std::string, std::shared_ptr<IAccountInfo>> &infos)
{
    for (auto const &item : allEntries) {
        Json jsonObject = Json::parse(item.value.ToString(), nullptr, false);
        if (jsonObject.is_discarded()) {
            ACCOUNT_LOGE("error key: %{private}s", item.key.ToString().c_str());
            // it's a bad json, delete it
            {
                std::lock_guard<std::mutex> lock(kvStorePtrMutex_);
                kvStorePtr_->Delete(item.key);
            }
            continue;
        }

        AppAccountInfo appAccountInfo;
        appAccountInfo.FromJson(jsonObject);
        infos.emplace(item.key.ToString(), std::make_shared<AppAccountInfo>(appAccountInfo));
    }
}
}  // namespace AccountSA
}  // namespace OHOS