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