1 /*
2  * Copyright (C) 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 #include "confirm_callback.h"
16 
17 #include "media_library_napi.h"
18 #include "medialibrary_napi_utils.h"
19 #include "medialibrary_napi_log.h"
20 #include "media_file_utils.h"
21 #include "userfile_client.h"
22 
23 namespace OHOS {
24 namespace Media {
25 namespace {
26 static constexpr int32_t CONFIRM_CODE_SUCCESS = 0;
27 static constexpr int32_t CONFIRM_CODE_USER_DENY = -1;
28 static const std::string CONFIRM_BOX_DES_FILE_URIS = "desFileUris";
29 static const std::string RESULT_PARAM = "result";
30 static const std::string DATA_PARAM = "data";
31 }
32 
ConfirmCallback(napi_env env,Ace::UIContent * uiContent)33 ConfirmCallback::ConfirmCallback(napi_env env, Ace::UIContent *uiContent)
34 {
35     this->env_ = env;
36     this->uiContent = uiContent;
37 }
38 
OnRelease(int32_t releaseCode)39 void ConfirmCallback::OnRelease(int32_t releaseCode)
40 {
41     NAPI_INFO_LOG("ReleaseCode is %{public}d.", releaseCode);
42 
43     CloseModalUIExtension();
44 }
45 
OnResult(int32_t resultCode,const OHOS::AAFwk::Want & want)46 void ConfirmCallback::OnResult(int32_t resultCode, const OHOS::AAFwk::Want &want)
47 {
48     NAPI_INFO_LOG("ResultCode is %{public}d.", resultCode);
49 
50     this->resultCode_ = resultCode;
51 
52     std::vector<std::string> desFileUris;
53     if (resultCode == CONFIRM_CODE_SUCCESS) {
54         // check if the desFileUris exsit
55         if (!want.HasParameter(CONFIRM_BOX_DES_FILE_URIS)) {
56             NAPI_ERR_LOG("Can't get string array from want.");
57             CHECK_ARGS_RET_VOID(this->env_, true, JS_INNER_FAIL);
58             return;
59         }
60 
61         // get desFileUris from want
62         desFileUris = want.GetStringArrayParam(CONFIRM_BOX_DES_FILE_URIS);
63     } else if (resultCode == CONFIRM_CODE_USER_DENY) {
64         this->resultCode_ = CONFIRM_CODE_SUCCESS; // user deny return success with empty uris
65     }
66 
67     SendMessageBack(desFileUris);
68 }
69 
OnError(int32_t code,const std::string & name,const std::string & message)70 void ConfirmCallback::OnError(int32_t code, const std::string &name, const std::string &message)
71 {
72     NAPI_INFO_LOG("Code is %{public}d, name is %{public}s, message is %{public}s.", code, name.c_str(),
73         message.c_str());
74 
75     this->resultCode_ = JS_INNER_FAIL;
76     std::vector<std::string> desFileUris;
77     SendMessageBack(desFileUris);
78 }
79 
OnReceive(const OHOS::AAFwk::WantParams & request)80 void ConfirmCallback::OnReceive(const OHOS::AAFwk::WantParams &request)
81 {
82     NAPI_INFO_LOG("Called.");
83 }
84 
SetSessionId(int32_t sessionId)85 void ConfirmCallback::SetSessionId(int32_t sessionId)
86 {
87     this->sessionId_ = sessionId;
88 }
89 
SetFunc(napi_value func)90 void ConfirmCallback::SetFunc(napi_value func)
91 {
92     napi_valuetype valueType = napi_undefined;
93     napi_typeof(this->env_, func, &valueType);
94     if (valueType == napi_function) {
95         napi_create_reference(this->env_, func, ARGS_ONE, &this->callbackRef);
96     }
97 }
98 
GenerateStringArrayValue(napi_env & env,const std::vector<std::string> & desFileUris,const size_t len,napi_value & value)99 static void GenerateStringArrayValue(napi_env &env, const std::vector<std::string> &desFileUris, const size_t len,
100     napi_value &value)
101 {
102     if (len == 0) {
103         return;
104     }
105 
106     CHECK_ARGS_RET_VOID(env, napi_create_array_with_length(env, len, &value), JS_INNER_FAIL); // set array length
107 
108     for (size_t i = 0; i < len; ++i) {
109         // create string value and set it as array element
110         napi_value element = nullptr;
111         CHECK_ARGS_RET_VOID(env, napi_create_string_utf8(env, desFileUris[i].c_str(), NAPI_AUTO_LENGTH, &element),
112             JS_INNER_FAIL);
113 
114         if ((element == nullptr) || (napi_set_element(env, value, i, element) != napi_ok)) {
115             NAPI_ERR_LOG("Failed to set element %{public}s to array.", desFileUris[i].c_str());
116             break;
117         }
118     }
119 }
120 
SendMessageBack(const std::vector<std::string> & desFileUris)121 void ConfirmCallback::SendMessageBack(const std::vector<std::string> &desFileUris)
122 {
123     CloseModalUIExtension();
124 
125     napi_value undefined = nullptr;
126     CHECK_ARGS_RET_VOID(this->env_, napi_get_undefined(this->env_, &undefined), JS_INNER_FAIL);
127 
128     napi_value results[ARGS_ONE] = {nullptr};
129     CHECK_ARGS_RET_VOID(this->env_, napi_create_object(this->env_, &results[PARAM0]), JS_INNER_FAIL);
130 
131     // create int32_t value bind result code as first napi value
132     napi_value result = nullptr;
133     CHECK_ARGS_RET_VOID(this->env_, napi_create_int32(this->env_, this->resultCode_, &result),
134         JS_INNER_FAIL);
135     CHECK_ARGS_RET_VOID(this->env_, napi_set_named_property(this->env_, results[PARAM0], RESULT_PARAM.c_str(), result),
136         JS_INNER_FAIL);
137 
138     size_t len = desFileUris.size();
139     if (len > 0) {
140         // transfer desFileUris to a napi value as second param
141         napi_value data = nullptr;
142         GenerateStringArrayValue(this->env_, desFileUris, len, data);
143 
144         CHECK_ARGS_RET_VOID(this->env_, napi_set_named_property(this->env_, results[PARAM0], DATA_PARAM.c_str(), data),
145             JS_INNER_FAIL);
146     }
147 
148     napi_value callback = nullptr;
149     CHECK_ARGS_RET_VOID(this->env_, napi_get_reference_value(this->env_, this->callbackRef, &callback), JS_INNER_FAIL);
150 
151     napi_value returnVal;
152     CHECK_ARGS_RET_VOID(this->env_, napi_call_function(this->env_, undefined, callback, ARGS_ONE, results, &returnVal),
153         JS_INNER_FAIL);
154 }
155 
CloseModalUIExtension()156 void ConfirmCallback::CloseModalUIExtension()
157 {
158     NAPI_INFO_LOG("Called.");
159 
160     if (this->uiContent != nullptr) {
161         uiContent->CloseModalUIExtension(this->sessionId_);
162     }
163 }
164 }
165 }