1 /*
2 * Copyright (c) 2023 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 "ui_extension_context.h"
17
18 #include "ability_manager_client.h"
19 #include "connection_manager.h"
20 #include "hilog_wrapper.h"
21 #include "hitrace_meter.h"
22
23 namespace OHOS {
24 namespace AbilityRuntime {
25 const size_t UIExtensionContext::CONTEXT_TYPE_ID(std::hash<const char*> {} ("UIExtensionContext"));
26 int UIExtensionContext::ILLEGAL_REQUEST_CODE(-1);
27
StartAbility(const AAFwk::Want & want) const28 ErrCode UIExtensionContext::StartAbility(const AAFwk::Want &want) const
29 {
30 HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
31 HILOG_DEBUG("Start ability begin, ability:%{public}s.", want.GetElement().GetAbilityName().c_str());
32 ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->StartAbility(want, token_, ILLEGAL_REQUEST_CODE);
33 if (err != ERR_OK) {
34 HILOG_ERROR("StartAbility is failed %{public}d", err);
35 }
36 return err;
37 }
38
StartAbility(const AAFwk::Want & want,const AAFwk::StartOptions & startOptions) const39 ErrCode UIExtensionContext::StartAbility(const AAFwk::Want &want, const AAFwk::StartOptions &startOptions) const
40 {
41 HITRACE_METER_NAME(HITRACE_TAG_ABILITY_MANAGER, __PRETTY_FUNCTION__);
42 HILOG_DEBUG("Start ability begin, ability:%{public}s.", want.GetElement().GetAbilityName().c_str());
43 ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->StartAbility(want, startOptions, token_,
44 ILLEGAL_REQUEST_CODE);
45 if (err != ERR_OK) {
46 HILOG_ERROR("StartAbility is failed %{public}d", err);
47 }
48 return err;
49 }
50
TerminateSelf()51 ErrCode UIExtensionContext::TerminateSelf()
52 {
53 HILOG_DEBUG("TerminateSelf begin.");
54 ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->TerminateAbility(token_, -1, nullptr);
55 if (err != ERR_OK) {
56 HILOG_ERROR("TerminateSelf is failed %{public}d", err);
57 }
58 HILOG_DEBUG("TerminateSelf end.");
59 return err;
60 }
61
ConnectAbility(const AAFwk::Want & want,const sptr<AbilityConnectCallback> & connectCallback) const62 ErrCode UIExtensionContext::ConnectAbility(
63 const AAFwk::Want &want, const sptr<AbilityConnectCallback> &connectCallback) const
64 {
65 HILOG_DEBUG("Connect ability begin, ability:%{public}s.", want.GetElement().GetAbilityName().c_str());
66 ErrCode ret =
67 ConnectionManager::GetInstance().ConnectAbility(token_, want, connectCallback);
68 HILOG_DEBUG("UIExtensionContext::ConnectAbility ErrorCode = %{public}d", ret);
69 return ret;
70 }
71
DisconnectAbility(const AAFwk::Want & want,const sptr<AbilityConnectCallback> & connectCallback) const72 ErrCode UIExtensionContext::DisconnectAbility(
73 const AAFwk::Want &want, const sptr<AbilityConnectCallback> &connectCallback) const
74 {
75 HILOG_DEBUG("%{public}s begin.", __func__);
76 ErrCode ret =
77 ConnectionManager::GetInstance().DisconnectAbility(token_, want.GetElement(), connectCallback);
78 if (ret != ERR_OK) {
79 HILOG_ERROR("%{public}s end DisconnectAbility error, ret=%{public}d", __func__, ret);
80 }
81 HILOG_DEBUG("%{public}s end DisconnectAbility", __func__);
82 return ret;
83 }
84
StartAbilityForResult(const AAFwk::Want & want,int requestCode,RuntimeTask && task)85 ErrCode UIExtensionContext::StartAbilityForResult(const AAFwk::Want &want, int requestCode, RuntimeTask &&task)
86 {
87 HILOG_DEBUG("begin.");
88 resultCallbacks_.insert(make_pair(requestCode, std::move(task)));
89 ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->StartAbility(want, token_, requestCode);
90 if (err != ERR_OK) {
91 HILOG_ERROR("ret=%{public}d", err);
92 OnAbilityResultInner(requestCode, err, want);
93 }
94 HILOG_DEBUG("end.");
95 return err;
96 }
97
StartAbilityForResult(const AAFwk::Want & want,const AAFwk::StartOptions & startOptions,int requestCode,RuntimeTask && task)98 ErrCode UIExtensionContext::StartAbilityForResult(
99 const AAFwk::Want &want, const AAFwk::StartOptions &startOptions, int requestCode, RuntimeTask &&task)
100 {
101 HILOG_DEBUG("begin.");
102 resultCallbacks_.insert(make_pair(requestCode, std::move(task)));
103 ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->StartAbility(want, startOptions, token_, requestCode);
104 if (err != ERR_OK) {
105 HILOG_ERROR("ret=%{public}d", err);
106 OnAbilityResultInner(requestCode, err, want);
107 }
108 HILOG_DEBUG("end.");
109 return err;
110 }
111
OnAbilityResult(int requestCode,int resultCode,const AAFwk::Want & resultData)112 void UIExtensionContext::OnAbilityResult(int requestCode, int resultCode, const AAFwk::Want &resultData)
113 {
114 HILOG_DEBUG("begin.");
115 auto callback = resultCallbacks_.find(requestCode);
116 if (callback != resultCallbacks_.end()) {
117 if (callback->second) {
118 callback->second(resultCode, resultData, false);
119 }
120 resultCallbacks_.erase(requestCode);
121 }
122 HILOG_DEBUG("end.");
123 }
124
GetAbilityInfoType() const125 AppExecFwk::AbilityType UIExtensionContext::GetAbilityInfoType() const
126 {
127 std::shared_ptr<AppExecFwk::AbilityInfo> info = GetAbilityInfo();
128 if (info == nullptr) {
129 HILOG_WARN("GetAbilityInfoType info is nullptr");
130 return AppExecFwk::AbilityType::UNKNOWN;
131 }
132
133 return info->type;
134 }
135
OnAbilityResultInner(int requestCode,int resultCode,const AAFwk::Want & resultData)136 void UIExtensionContext::OnAbilityResultInner(int requestCode, int resultCode, const AAFwk::Want &resultData)
137 {
138 HILOG_DEBUG("begin.");
139 auto callback = resultCallbacks_.find(requestCode);
140 if (callback != resultCallbacks_.end()) {
141 if (callback->second) {
142 callback->second(resultCode, resultData, true);
143 }
144 resultCallbacks_.erase(requestCode);
145 }
146 HILOG_DEBUG("end.");
147 }
148
GenerateCurRequestCode()149 int UIExtensionContext::GenerateCurRequestCode()
150 {
151 curRequestCode_ = (curRequestCode_ == INT_MAX) ? 0 : (curRequestCode_ + 1);
152 return curRequestCode_;
153 }
154 } // namespace AbilityRuntime
155 } // namespace OHOS
156