1 /*
2  * Copyright (c) 2023-2024 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #include "disposed_observer.h"
17 
18 #include "interceptor/disposed_rule_interceptor.h"
19 #include "ability_record.h"
20 #include "modal_system_ui_extension.h"
21 
22 namespace OHOS {
23 namespace AAFwk {
24 namespace {
25 constexpr const char* UIEXTENSION_MODAL_TYPE = "ability.want.params.modalType";
26 constexpr const char* INTERCEPT_MISSION_ID = "intercept_missionId";
27 }
28 
DisposedObserver(const AppExecFwk::DisposedRule & disposedRule,const std::shared_ptr<DisposedRuleInterceptor> & interceptor)29 DisposedObserver::DisposedObserver(const AppExecFwk::DisposedRule &disposedRule,
30     const std::shared_ptr<DisposedRuleInterceptor> &interceptor)
31     : disposedRule_(disposedRule), interceptor_(interceptor)
32 {}
33 
OnAbilityStateChanged(const AppExecFwk::AbilityStateData & abilityStateData)34 void DisposedObserver::OnAbilityStateChanged(const AppExecFwk::AbilityStateData &abilityStateData)
35 {
36     std::lock_guard<ffrt::mutex> guard(observerLock_);
37     if (abilityStateData.abilityState != static_cast<int32_t>(AppExecFwk::AbilityState::ABILITY_STATE_FOREGROUND)) {
38         return;
39     }
40     TAG_LOGD(AAFwkTag::ABILITYMGR, "Call");
41     token_ = abilityStateData.token;
42     auto abilityRecord = Token::GetAbilityRecordByToken(token_);
43     if (abilityRecord && !abilityRecord->GetAbilityInfo().isStageBasedModel) {
44         auto systemUIExtension = std::make_shared<OHOS::Rosen::ModalSystemUiExtension>();
45         Want want = *disposedRule_.want;
46         want.SetParam(UIEXTENSION_MODAL_TYPE, 1);
47         auto sessionInfo = abilityRecord->GetSessionInfo();
48         if (sessionInfo != nullptr) {
49             want.SetParam(INTERCEPT_MISSION_ID, sessionInfo->persistentId);
50         } else {
51             want.SetParam(INTERCEPT_MISSION_ID, abilityRecord->GetMissionId());
52         }
53         TAG_LOGD(AAFwkTag::ABILITYMGR, "FA modal system");
54         bool ret = IN_PROCESS_CALL(systemUIExtension->CreateModalUIExtension(want));
55         if (!ret) {
56             TAG_LOGE(AAFwkTag::ABILITYMGR, "failed to start system UIExtension");
57         }
58         interceptor_->UnregisterObserver(abilityStateData.bundleName);
59     }
60 }
61 
OnPageShow(const AppExecFwk::PageStateData & pageStateData)62 void DisposedObserver::OnPageShow(const AppExecFwk::PageStateData &pageStateData)
63 {
64     if (disposedRule_.componentType == AppExecFwk::ComponentType::UI_ABILITY) {
65         TAG_LOGD(AAFwkTag::ABILITYMGR, "Call");
66         int ret = IN_PROCESS_CALL(AbilityManagerClient::GetInstance()->StartAbility(*disposedRule_.want));
67         if (ret != ERR_OK) {
68             interceptor_->UnregisterObserver(pageStateData.bundleName);
69             TAG_LOGE(AAFwkTag::ABILITYMGR, "failed to start disposed ability");
70             return;
71         }
72     }
73     if (disposedRule_.componentType == AppExecFwk::ComponentType::UI_EXTENSION) {
74         auto abilityRecord = Token::GetAbilityRecordByToken(token_);
75         if (abilityRecord == nullptr || abilityRecord->GetAbilityInfo().type != AppExecFwk::AbilityType::PAGE) {
76             auto systemUIExtension = std::make_shared<OHOS::Rosen::ModalSystemUiExtension>();
77             Want want = *disposedRule_.want;
78             want.SetParam(UIEXTENSION_MODAL_TYPE, 1);
79             TAG_LOGD(AAFwkTag::ABILITYMGR, "modal system");
80             bool ret = IN_PROCESS_CALL(systemUIExtension->CreateModalUIExtension(want));
81             if (!ret) {
82                 interceptor_->UnregisterObserver(pageStateData.bundleName);
83                 TAG_LOGE(AAFwkTag::ABILITYMGR, "failed to start system UIExtension");
84                 return;
85             }
86         } else {
87             Want want = *disposedRule_.want;
88             auto sessionInfo = abilityRecord->GetSessionInfo();
89             if (sessionInfo != nullptr) {
90                 want.SetParam(INTERCEPT_MISSION_ID, sessionInfo->persistentId);
91             } else {
92                 want.SetParam(INTERCEPT_MISSION_ID, abilityRecord->GetMissionId());
93             }
94             TAG_LOGD(AAFwkTag::ABILITYMGR, "modal app");
95             int ret = abilityRecord->CreateModalUIExtension(want);
96             if (ret != ERR_OK) {
97                 interceptor_->UnregisterObserver(pageStateData.bundleName);
98                 TAG_LOGE(AAFwkTag::ABILITYMGR, "failed to start disposed UIExtension");
99                 return;
100             }
101         }
102     }
103     interceptor_->UnregisterObserver(pageStateData.bundleName);
104 }
105 } // namespace AAFwk
106 } // namespace OHOS
107