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
23namespace OHOS {
24namespace Media {
25namespace {
26static constexpr int32_t CONFIRM_CODE_SUCCESS = 0;
27static constexpr int32_t CONFIRM_CODE_USER_DENY = -1;
28static const std::string CONFIRM_BOX_DES_FILE_URIS = "desFileUris";
29static const std::string RESULT_PARAM = "result";
30static const std::string DATA_PARAM = "data";
31}
32#ifdef HAS_ACE_ENGINE_PART
33ConfirmCallback::ConfirmCallback(napi_env env, Ace::UIContent *uiContent)
34{
35    this->env_ = env;
36    this->uiContent = uiContent;
37}
38#else
39ConfirmCallback::ConfirmCallback(napi_env env)
40{
41    this->env_ = env;
42}
43#endif
44
45void ConfirmCallback::OnRelease(int32_t releaseCode)
46{
47    NAPI_INFO_LOG("ReleaseCode is %{public}d.", releaseCode);
48
49    CloseModalUIExtension();
50}
51
52void ConfirmCallback::OnResult(int32_t resultCode, const OHOS::AAFwk::Want &want)
53{
54    NAPI_INFO_LOG("ResultCode is %{public}d.", resultCode);
55
56    this->resultCode_ = resultCode;
57
58    std::vector<std::string> desFileUris;
59    if (resultCode == CONFIRM_CODE_SUCCESS) {
60        // check if the desFileUris exsit
61        if (!want.HasParameter(CONFIRM_BOX_DES_FILE_URIS)) {
62            NAPI_ERR_LOG("Can't get string array from want.");
63            CHECK_ARGS_RET_VOID(this->env_, true, JS_INNER_FAIL);
64            return;
65        }
66
67        // get desFileUris from want
68        desFileUris = want.GetStringArrayParam(CONFIRM_BOX_DES_FILE_URIS);
69    } else if (resultCode == CONFIRM_CODE_USER_DENY) {
70        this->resultCode_ = CONFIRM_CODE_SUCCESS; // user deny return success with empty uris
71    }
72
73    SendMessageBack(desFileUris);
74}
75
76void ConfirmCallback::OnError(int32_t code, const std::string &name, 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
86void ConfirmCallback::OnReceive(const OHOS::AAFwk::WantParams &request)
87{
88    NAPI_INFO_LOG("Called.");
89}
90
91void ConfirmCallback::SetSessionId(int32_t sessionId)
92{
93    this->sessionId_ = sessionId;
94}
95
96void ConfirmCallback::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
105static 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
127void ConfirmCallback::SendMessageBack(const std::vector<std::string> &desFileUris)
128{
129    CloseModalUIExtension();
130
131    napi_value undefined = nullptr;
132    CHECK_ARGS_RET_VOID(this->env_, napi_get_undefined(this->env_, &undefined), JS_INNER_FAIL);
133
134    napi_value results[ARGS_ONE] = {nullptr};
135    CHECK_ARGS_RET_VOID(this->env_, napi_create_object(this->env_, &results[PARAM0]), JS_INNER_FAIL);
136
137    // create int32_t value bind result code as first napi value
138    napi_value result = nullptr;
139    CHECK_ARGS_RET_VOID(this->env_, napi_create_int32(this->env_, this->resultCode_, &result),
140        JS_INNER_FAIL);
141    CHECK_ARGS_RET_VOID(this->env_, napi_set_named_property(this->env_, results[PARAM0], RESULT_PARAM.c_str(), result),
142        JS_INNER_FAIL);
143
144    size_t len = desFileUris.size();
145    if (len > 0) {
146        // transfer desFileUris to a napi value as second param
147        napi_value data = nullptr;
148        GenerateStringArrayValue(this->env_, desFileUris, len, data);
149
150        CHECK_ARGS_RET_VOID(this->env_, napi_set_named_property(this->env_, results[PARAM0], DATA_PARAM.c_str(), data),
151            JS_INNER_FAIL);
152    }
153
154    napi_value callback = nullptr;
155    CHECK_ARGS_RET_VOID(this->env_, napi_get_reference_value(this->env_, this->callbackRef, &callback), JS_INNER_FAIL);
156
157    napi_value returnVal;
158    CHECK_ARGS_RET_VOID(this->env_, napi_call_function(this->env_, undefined, callback, ARGS_ONE, results, &returnVal),
159        JS_INNER_FAIL);
160}
161
162void ConfirmCallback::CloseModalUIExtension()
163{
164    NAPI_INFO_LOG("Called.");
165
166#ifdef HAS_ACE_ENGINE_PART
167    if (this->uiContent != nullptr) {
168        uiContent->CloseModalUIExtension(this->sessionId_);
169    }
170#endif
171}
172}
173}