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

#include <map>

#include "hilog_tag_wrapper.h"
#include "mock_ability_mgr_host.h"
#include "system_ability_definition.h"

namespace OHOS {
namespace AppExecFwk {
static std::map<int32_t, sptr<IRemoteObject>> g_abilities;

sptr<IRemoteObject> SystemAbilityHelper::GetSystemAbility(const int32_t systemAbilityId)
{
    TAG_LOGD(AAFwkTag::TEST, "mock system ability helper get %{public}d system ability", systemAbilityId);
    auto iter = g_abilities.find(systemAbilityId);
    if (iter != g_abilities.end()) {
        return iter->second;
    }
    if (systemAbilityId == ABILITY_MGR_SERVICE_ID) {
        return new MockAbilityMgrStub();
    }
    return nullptr;
}

bool SystemAbilityHelper::AddSystemAbility(const int32_t systemAbilityId, const sptr<IRemoteObject>& systemAbility)
{
    if (g_abilities.erase(systemAbilityId) > 0) {
        TAG_LOGD(AAFwkTag::TEST, "mock system ability helper add system ability erase exist key");
    }
    TAG_LOGD(AAFwkTag::TEST, "mock system ability helper emplace %{public}d system ability", systemAbilityId);
    g_abilities.emplace(systemAbilityId, systemAbility);
    // mock helper always return true.
    return true;
}

bool SystemAbilityHelper::RemoveSystemAbility(const int32_t systemAbilityId)
{
    TAG_LOGD(AAFwkTag::TEST, "mock system ability helper remove system ability");
    if (g_abilities.erase(systemAbilityId) > 0) {
        TAG_LOGD(AAFwkTag::TEST, "mock system ability helper remove %{public}d system ability erase exist key",
                 systemAbilityId);
    }
    // mock helper always return true.
    return true;
}
}  // namespace AppExecFwk
}  // namespace OHOS