ability_ability_runtime/frameworks/native/ability/native/extension_impl.cpp

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

#include "ability_manager_client.h"
#include "ability_local_record.h"
#include "ability_transaction_callback_info.h"
#include "freeze_util.h"
#include "hitrace_meter.h"
#include "ipc_object_proxy.h"
#include "extension_context.h"
#include "hilog_tag_wrapper.h"
#include "ui_extension_utils.h"


namespace OHOS {
namespace AbilityRuntime {
const std::string JSON_KEY_ERR_MSG = "errMsg";
ExtensionImpl::~ExtensionImpl()
{
    TAG_LOGI(AAFwkTag::EXT, "~ExtensionImpl");
}

void ExtensionImpl::Init(const std::shared_ptr<AppExecFwk::OHOSApplication> &application,
    const std::shared_ptr<AppExecFwk::AbilityLocalRecord> &record,
    std::shared_ptr<Extension> &extension,
    std::shared_ptr<AppExecFwk::AbilityHandler> &handler,
    const sptr<IRemoteObject> &token)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    if ((token == nullptr) || (application == nullptr) || (handler == nullptr) || (record == nullptr) ||
        extension == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "init failed, some obj null");
        return;
    }

    token_ = record->GetToken();
    extension_ = extension;
    if (record->GetAbilityInfo() != nullptr) {
        extensionType_ = record->GetAbilityInfo()->extensionAbilityType;
        if (AAFwk::UIExtensionUtils::IsUIExtension(extensionType_)) {
            extension_->SetExtensionWindowLifeCycleListener(
                sptr<ExtensionWindowLifeCycleImpl>(new ExtensionWindowLifeCycleImpl(token_, shared_from_this())));
        }
    }
    extension_->Init(record, application, handler, token);
    lifecycleState_ = AAFwk::ABILITY_STATE_INITIAL;
    skipCommandExtensionWithIntent_ = false;
}

/**
 * @brief Handling the life cycle switching of Extension.
 *
 * @param want Indicates the structure containing information about the extension.
 * @param targetState The life cycle state to switch to.
 * @param sessionInfo  Indicates the sessionInfo.
 *
 */
void ExtensionImpl::HandleExtensionTransaction(const Want &want, const AAFwk::LifeCycleStateInfo &targetState,
    sptr<AAFwk::SessionInfo> sessionInfo)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGI(AAFwkTag::EXT, "%{public}s;sourceState:%{public}d;targetState:%{public}d;isNewWant:%{public}d",
        want.GetElement().GetAbilityName().c_str(), lifecycleState_, targetState.state, targetState.isNewWant);
    if (lifecycleState_ == targetState.state) {
        TAG_LOGE(AAFwkTag::EXT, "lifecycle state equal");
        return;
    }
    SetLaunchParam(targetState.launchParam);
    bool ret = true;
    switch (targetState.state) {
        case AAFwk::ABILITY_STATE_INITIAL: {
            bool isAsyncCallback = false;
            if (lifecycleState_ != AAFwk::ABILITY_STATE_INITIAL) {
                Stop(isAsyncCallback, want, sessionInfo);
            }
            if (isAsyncCallback) {
                ret = false;
            }
            break;
        }
        case AAFwk::ABILITY_STATE_INACTIVE: {
            if (lifecycleState_ == AAFwk::ABILITY_STATE_INITIAL) {
                Start(want, sessionInfo);
            }
            break;
        }
        case AAFwk::ABILITY_STATE_FOREGROUND_NEW: {
            if (lifecycleState_ == AAFwk::ABILITY_STATE_INITIAL) {
                Start(want, sessionInfo);
            }
            Foreground(want, sessionInfo);
            break;
        }
        case AAFwk::ABILITY_STATE_BACKGROUND_NEW: {
            Background(want, sessionInfo);
            break;
        }
        default: {
            ret = false;
            TAG_LOGE(AAFwkTag::EXT, "error state");
            break;
        }
    }
    if (ret && !UIExtensionAbilityExecuteInsightIntent(want)) {
        TAG_LOGD(AAFwkTag::EXT, "call abilityms");
        AAFwk::PacMap restoreData;
        AAFwk::AbilityManagerClient::GetInstance()->AbilityTransitionDone(token_, targetState.state, restoreData);
    }
}

bool ExtensionImpl::UIExtensionAbilityExecuteInsightIntent(const Want &want)
{
    return AAFwk::UIExtensionUtils::IsUIExtension(extensionType_) &&
        AppExecFwk::InsightIntentExecuteParam::IsInsightIntentExecute(want);
}

void ExtensionImpl::ScheduleUpdateConfiguration(const AppExecFwk::Configuration &config)
{
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    extension_->OnConfigurationUpdated(config);
}

void ExtensionImpl::NotifyMemoryLevel(int level)
{
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    if (lifecycleState_ != AAFwk::ABILITY_STATE_INITIAL) {
        extension_->OnMemoryLevel(level);
    }
}

/**
 * @brief Toggles the lifecycle status of Extension to AAFwk::ABILITY_STATE_INACTIVE. And notifies the application
 * that it belongs to of the lifecycle status.
 *
 * @param want  The Want object to switch the life cycle.
 * @param sessionInfo  Indicates the sessionInfo.
 */
void ExtensionImpl::Start(const Want &want, sptr<AAFwk::SessionInfo> sessionInfo)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    TAG_LOGD(AAFwkTag::EXT, "called");
    if (extension_->abilityInfo_->extensionAbilityType == AppExecFwk::ExtensionAbilityType::WINDOW ||
        AAFwk::UIExtensionUtils::IsUIExtension(extension_->abilityInfo_->extensionAbilityType)) {
        extension_->OnStart(want, sessionInfo);
    } else {
        extension_->OnStart(want);
    }
    lifecycleState_ = AAFwk::ABILITY_STATE_INACTIVE;
    TAG_LOGD(AAFwkTag::EXT, "ok");
}

/**
 * @brief Toggles the lifecycle status of Extension to AAFwk::ABILITY_STATE_INITIAL. And notifies the application
 * that it belongs to of the lifecycle status.
 *
 */
void ExtensionImpl::Stop()
{
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    extension_->OnStop();
    lifecycleState_ = AAFwk::ABILITY_STATE_INITIAL;
    TAG_LOGD(AAFwkTag::EXT, "ok");
}

void ExtensionImpl::Stop(bool &isAsyncCallback, const Want &want, sptr<AAFwk::SessionInfo> sessionInfo)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        isAsyncCallback = false;
        return;
    }

    if (AAFwk::UIExtensionUtils::IsUIExtension(extensionType_) && sessionInfo != nullptr) {
        CommandExtensionWindow(want, sessionInfo, AAFwk::WIN_CMD_DESTROY);
    }

    auto *callbackInfo = AppExecFwk::AbilityTransactionCallbackInfo<>::Create();
    if (callbackInfo == nullptr) {
        extension_->OnStop();
        lifecycleState_ = AAFwk::ABILITY_STATE_INITIAL;
        isAsyncCallback = false;
        return;
    }
    std::weak_ptr<ExtensionImpl> weakPtr = shared_from_this();
    auto asyncCallback = [ExtensionImplWeakPtr = weakPtr, state = AAFwk::ABILITY_STATE_INITIAL]() {
        auto extensionImpl = ExtensionImplWeakPtr.lock();
        if (extensionImpl == nullptr) {
            TAG_LOGE(AAFwkTag::EXT, "null extensionImpl");
            return;
        }
        extensionImpl->lifecycleState_ = AAFwk::ABILITY_STATE_INITIAL;
        extensionImpl->AbilityTransactionCallback(state);
    };
    callbackInfo->Push(asyncCallback);

    extension_->OnStop(callbackInfo, isAsyncCallback);
    if (!isAsyncCallback) {
        lifecycleState_ = AAFwk::ABILITY_STATE_INITIAL;
        AppExecFwk::AbilityTransactionCallbackInfo<>::Destroy(callbackInfo);
    }
    // else: callbackInfo will be destroyed after the async callback
    TAG_LOGD(AAFwkTag::EXT, "end");
}

void ExtensionImpl::AbilityTransactionCallback(const AAFwk::AbilityLifeCycleState &state)
{
    TAG_LOGD(AAFwkTag::EXT, "called");
    AAFwk::PacMap restoreData;
    AAFwk::AbilityManagerClient::GetInstance()->AbilityTransitionDone(token_, state, restoreData);
}

/**
 * @brief Connect the extension. and Calling information back to Extension.
 *
 * @param want The Want object to connect to.
 *
 */
sptr<IRemoteObject> ExtensionImpl::ConnectExtension(const Want &want)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return nullptr;
    }

    skipCommandExtensionWithIntent_ = true;
    sptr<IRemoteObject> object = extension_->OnConnect(want);
    lifecycleState_ = AAFwk::ABILITY_STATE_ACTIVE;
    TAG_LOGD(AAFwkTag::EXT, "ok");

    return object;
}

sptr<IRemoteObject> ExtensionImpl::ConnectExtension(const Want &want, bool &isAsyncCallback)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    FreezeUtil::GetInstance().AddLifecycleEvent(token_, "ExtensionImpl::ConnectExtension");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        isAsyncCallback = false;
        return nullptr;
    }

    skipCommandExtensionWithIntent_ = true;
    auto *callbackInfo = AppExecFwk::AbilityTransactionCallbackInfo<sptr<IRemoteObject>>::Create();
    if (callbackInfo == nullptr) {
        sptr<IRemoteObject> object = extension_->OnConnect(want);
        lifecycleState_ = AAFwk::ABILITY_STATE_ACTIVE;
        isAsyncCallback = false;
        TAG_LOGI(AAFwkTag::EXT, "end");
        return object;
    }

    std::weak_ptr<ExtensionImpl> weakPtr = shared_from_this();
    auto asyncCallback = [extensionImplWeakPtr = weakPtr](sptr<IRemoteObject> &service) {
        auto extensionImpl = extensionImplWeakPtr.lock();
        if (extensionImpl == nullptr) {
            TAG_LOGE(AAFwkTag::EXT, "null extensionImpl");
            return;
        }
        extensionImpl->lifecycleState_ = AAFwk::ABILITY_STATE_ACTIVE;
        extensionImpl->ConnectExtensionCallback(service);
    };
    callbackInfo->Push(asyncCallback);

    sptr<IRemoteObject> object = extension_->OnConnect(want, callbackInfo, isAsyncCallback);
    if (!isAsyncCallback) {
        lifecycleState_ = AAFwk::ABILITY_STATE_ACTIVE;
        AppExecFwk::AbilityTransactionCallbackInfo<sptr<IRemoteObject>>::Destroy(callbackInfo);
    }
    // else: callbackInfo will be destroyed after the async callback
    TAG_LOGD(AAFwkTag::EXT, "ok");
    return object;
}

void ExtensionImpl::ConnectExtensionCallback(sptr<IRemoteObject> &service)
{
    FreezeUtil::GetInstance().AddLifecycleEvent(token_, "ExtensionImpl::ConnectExtensionCallback");
    ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->ScheduleConnectAbilityDone(token_, service);
    if (err != ERR_OK) {
        TAG_LOGE(AAFwkTag::EXT, "err: %{public}d", err);
        FreezeUtil::GetInstance().AddLifecycleEvent(token_,
            "ExtensionImpl::ConnectExtensionCallback fail, err is " + std::to_string(err));
        return;
    }
    FreezeUtil::GetInstance().DeleteLifecycleEvent(token_);
    FreezeUtil::GetInstance().DeleteAppLifecycleEvent(0);
}

/**
 * @brief Disconnects the connected object.
 *
 * @param want The Want object to disconnect to.
 */
void ExtensionImpl::DisconnectExtension(const Want &want)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    extension_->OnDisconnect(want);
    TAG_LOGD(AAFwkTag::EXT, "ok");
}

void ExtensionImpl::DisconnectExtension(const Want &want, bool &isAsyncCallback)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "called");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        isAsyncCallback = false;
        return;
    }

    auto *callbackInfo = AppExecFwk::AbilityTransactionCallbackInfo<>::Create();
    if (callbackInfo == nullptr) {
        extension_->OnDisconnect(want);
        isAsyncCallback = false;
        return;
    }
    std::weak_ptr<ExtensionImpl> weakPtr = shared_from_this();
    auto asyncCallback = [extensionImplWeakPtr = weakPtr]() {
        auto extensionImpl = extensionImplWeakPtr.lock();
        if (extensionImpl == nullptr) {
            TAG_LOGE(AAFwkTag::EXT, "null extensionImpl");
            return;
        }
        extensionImpl->DisconnectExtensionCallback();
    };
    callbackInfo->Push(asyncCallback);

    extension_->OnDisconnect(want, callbackInfo, isAsyncCallback);
    if (!isAsyncCallback) {
        AppExecFwk::AbilityTransactionCallbackInfo<>::Destroy(callbackInfo);
    }
    // else: callbackInfo will be destroyed after the async callback
    TAG_LOGD(AAFwkTag::EXT, "end");
}

void ExtensionImpl::DisconnectExtensionCallback()
{
    ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->ScheduleDisconnectAbilityDone(token_);
    if (err != ERR_OK) {
        TAG_LOGE(AAFwkTag::EXT, "err: %{public}d", err);
    }
}

/**
 * @brief Command the Extension. and Calling information back to Extension.
 *
 * @param want The Want object to command to.
 *
 * * @param restart Indicates the startup mode. The value true indicates that Service is restarted after being
 * destroyed, and the value false indicates a normal startup.
 *
 * @param startId Indicates the number of times the Service Extension has been started. The startId is incremented by 1
 * every time the Extension is started. For example, if the Extension has been started for six times,
 * the value of startId is 6.
 */
void ExtensionImpl::CommandExtension(const Want &want, bool restart, int startId)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }
    if (!AppExecFwk::InsightIntentExecuteParam::IsInsightIntentExecute(want) || !skipCommandExtensionWithIntent_) {
        skipCommandExtensionWithIntent_ = true;
        extension_->OnCommand(want, restart, startId);
    }
    lifecycleState_ = AAFwk::ABILITY_STATE_ACTIVE;
    TAG_LOGD(AAFwkTag::EXT, "ok");
}

bool ExtensionImpl::HandleInsightIntent(const Want &want)
{
    TAG_LOGD(AAFwkTag::EXT, "call");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return false;
    }
    auto ret = extension_->HandleInsightIntent(want);
    if (!ret) {
        TAG_LOGE(AAFwkTag::EXT, "handle failed");
        return false;
    }
    TAG_LOGD(AAFwkTag::EXT, "ok");
    return true;
}

void ExtensionImpl::CommandExtensionWindow(const Want &want, const sptr<AAFwk::SessionInfo> &sessionInfo,
    AAFwk::WindowCommand winCmd)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    if (extension_ == nullptr || sessionInfo == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_ or sessionInfo");
        return;
    }

    TAG_LOGD(AAFwkTag::EXT, "persistentId: %{private}d, componentId: %{public}" PRId64 ", winCmd: %{public}d",
        sessionInfo->persistentId, sessionInfo->uiExtensionComponentId, winCmd);
    extension_->OnCommandWindow(want, sessionInfo, winCmd);
    TAG_LOGD(AAFwkTag::EXT, "ok");
}

void ExtensionImpl::SendResult(int requestCode, int resultCode, const Want &resultData)
{
    TAG_LOGD(AAFwkTag::EXT, "begin");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    extension_->OnAbilityResult(requestCode, resultCode, resultData);
    TAG_LOGD(AAFwkTag::EXT, "end");
}

void ExtensionImpl::SetLaunchParam(const AAFwk::LaunchParam &launchParam)
{
    TAG_LOGD(AAFwkTag::EXT, "called");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    extension_->SetLaunchParam(launchParam);
}

void ExtensionImpl::ScheduleAbilityRequestFailure(const std::string &requestId, const AppExecFwk::ElementName &element,
    const std::string &message, int32_t resultCode)
{
    TAG_LOGD(AAFwkTag::EXT, "ScheduleAbilityRequestFailure called");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }
    extension_->OnExtensionAbilityRequestFailure(requestId, element, message, resultCode);
}

void ExtensionImpl::ScheduleAbilityRequestSuccess(const std::string &requestId, const AppExecFwk::ElementName &element)
{
    TAG_LOGD(AAFwkTag::EXT, "ScheduleAbilityRequestSuccess called");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }
    nlohmann::json jsonObject = nlohmann::json {
        { JSON_KEY_ERR_MSG, "Succeeded" },
    };
    extension_->OnExtensionAbilityRequestSuccess(requestId, element, jsonObject.dump());
}

void ExtensionImpl::Foreground(const Want &want, sptr<AAFwk::SessionInfo> sessionInfo)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "begin");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension");
        return;
    }

    extension_->OnForeground(want, sessionInfo);
    lifecycleState_ = AAFwk::ABILITY_STATE_FOREGROUND_NEW;
}

void ExtensionImpl::Background(const Want &want, sptr<AAFwk::SessionInfo> sessionInfo)
{
    HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
    TAG_LOGD(AAFwkTag::EXT, "begin");
    if (extension_ == nullptr) {
        TAG_LOGE(AAFwkTag::EXT, "null extension_");
        return;
    }

    if (AAFwk::UIExtensionUtils::IsUIExtension(extensionType_) && sessionInfo != nullptr) {
        CommandExtensionWindow(want, sessionInfo, AAFwk::WIN_CMD_BACKGROUND);
    }

    extension_->OnBackground();
    lifecycleState_ = AAFwk::ABILITY_STATE_BACKGROUND_NEW;
}

void ExtensionImpl::ExtensionWindowLifeCycleImpl::AfterForeground()
{
    TAG_LOGD(AAFwkTag::EXT, "called");
}

void ExtensionImpl::ExtensionWindowLifeCycleImpl::AfterBackground()
{
    TAG_LOGD(AAFwkTag::EXT, "called");
}

void ExtensionImpl::ExtensionWindowLifeCycleImpl::AfterActive()
{
    TAG_LOGD(AAFwkTag::EXT, "called");
}

void ExtensionImpl::ExtensionWindowLifeCycleImpl::AfterInactive()
{
    TAG_LOGD(AAFwkTag::EXT, "called");
}
}
}