1 /*
2  * Copyright (c) 2022 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 "gettype_context.h"
17 
18 #include "network_constant.h"
19 #include "netmanager_base_log.h"
20 #include "napi_utils.h"
21 
22 static constexpr const int PARAM_NO_OPTIONS = 0;
23 
24 static constexpr const int PARAM_HAS_OPTIONS = 1;
25 
26 namespace OHOS::NetManagerStandard {
GetTypeContext(napi_env env,EventManager * manager)27 GetTypeContext::GetTypeContext(napi_env env, EventManager *manager)
28     : BaseContext(env, manager), successCallback_(nullptr), failCallback_(nullptr), completeCallback_(nullptr)
29 {
30 }
31 
~GetTypeContext()32 GetTypeContext::~GetTypeContext()
33 {
34     if (successCallback_ != nullptr) {
35         (void)napi_delete_reference(GetEnv(), successCallback_);
36     }
37     if (failCallback_ != nullptr) {
38         (void)napi_delete_reference(GetEnv(), failCallback_);
39     }
40     if (completeCallback_ != nullptr) {
41         (void)napi_delete_reference(GetEnv(), completeCallback_);
42     }
43 }
44 
ParseParams(napi_value * params,size_t paramsCount)45 void GetTypeContext::ParseParams(napi_value *params, size_t paramsCount)
46 {
47     if (!CheckParamsType(params, paramsCount)) {
48         return;
49     }
50 
51     SetNeedPromise(false);
52 
53     if (paramsCount == PARAM_HAS_OPTIONS) {
54         SetParseOK(SetSuccessCallback(params[0]) && SetFailCallback(params[0]) && SetCompleteCallback(params[0]));
55         return;
56     }
57     NETMANAGER_BASE_LOGI("no callback");
58     SetParseOK(true);
59 }
60 
CheckParamsType(napi_value * params,size_t paramsCount)61 bool GetTypeContext::CheckParamsType(napi_value *params, size_t paramsCount)
62 {
63     if (paramsCount == PARAM_NO_OPTIONS) {
64         return true;
65     }
66 
67     if (paramsCount == PARAM_HAS_OPTIONS) {
68         return NapiUtils::GetValueType(GetEnv(), params[0]) == napi_object;
69     }
70     return false;
71 }
72 
SetSuccessCallback(napi_value options)73 bool GetTypeContext::SetSuccessCallback(napi_value options)
74 {
75     if (!NapiUtils::HasNamedProperty(GetEnv(), options, KEY_SUCCESS)) {
76         NETMANAGER_BASE_LOGI("do not need success fun");
77         return true;
78     }
79     napi_value callback = NapiUtils::GetNamedProperty(GetEnv(), options, KEY_SUCCESS);
80     if (NapiUtils::GetValueType(GetEnv(), callback) != napi_function) {
81         NETMANAGER_BASE_LOGE("success should be function");
82         return false;
83     }
84     if (successCallback_ != nullptr) {
85         (void)napi_delete_reference(GetEnv(), successCallback_);
86     }
87     return napi_create_reference(GetEnv(), callback, 1, &successCallback_) == napi_ok;
88 }
89 
SetFailCallback(napi_value options)90 bool GetTypeContext::SetFailCallback(napi_value options)
91 {
92     if (!NapiUtils::HasNamedProperty(GetEnv(), options, KEY_FAIL)) {
93         NETMANAGER_BASE_LOGD("do not need fail fun");
94         return true;
95     }
96     napi_value callback = NapiUtils::GetNamedProperty(GetEnv(), options, KEY_FAIL);
97     if (NapiUtils::GetValueType(GetEnv(), callback) != napi_function) {
98         NETMANAGER_BASE_LOGE("success should be function");
99         return false;
100     }
101     if (failCallback_ != nullptr) {
102         (void)napi_delete_reference(GetEnv(), failCallback_);
103     }
104     return napi_create_reference(GetEnv(), callback, 1, &failCallback_) == napi_ok;
105 }
106 
SetCompleteCallback(napi_value options)107 bool GetTypeContext::SetCompleteCallback(napi_value options)
108 {
109     if (!NapiUtils::HasNamedProperty(GetEnv(), options, KEY_COMPLETE)) {
110         NETMANAGER_BASE_LOGD("do not need complete fun");
111         return true;
112     }
113     napi_value callback = NapiUtils::GetNamedProperty(GetEnv(), options, KEY_COMPLETE);
114     if (NapiUtils::GetValueType(GetEnv(), callback) != napi_function) {
115         NETMANAGER_BASE_LOGE("success should be function");
116         return false;
117     }
118     if (completeCallback_ != nullptr) {
119         (void)napi_delete_reference(GetEnv(), completeCallback_);
120     }
121     return napi_create_reference(GetEnv(), callback, 1, &completeCallback_) == napi_ok;
122 }
123 
GetSuccessCallback() const124 napi_value GetTypeContext::GetSuccessCallback() const
125 {
126     if (successCallback_ == nullptr) {
127         return nullptr;
128     }
129     napi_value callback = nullptr;
130     NAPI_CALL(GetEnv(), napi_get_reference_value(GetEnv(), successCallback_, &callback));
131     return callback;
132 }
133 
GetFailCallback() const134 napi_value GetTypeContext::GetFailCallback() const
135 {
136     if (failCallback_ == nullptr) {
137         return nullptr;
138     }
139     napi_value callback = nullptr;
140     NAPI_CALL(GetEnv(), napi_get_reference_value(GetEnv(), failCallback_, &callback));
141     return callback;
142 }
143 
GetCompleteCallback() const144 napi_value GetTypeContext::GetCompleteCallback() const
145 {
146     if (completeCallback_ == nullptr) {
147         return nullptr;
148     }
149     napi_value callback = nullptr;
150     NAPI_CALL(GetEnv(), napi_get_reference_value(GetEnv(), completeCallback_, &callback));
151     return callback;
152 }
153 
SetCap(const NetAllCapabilities & cap)154 void GetTypeContext::SetCap(const NetAllCapabilities &cap)
155 {
156     cap_ = cap;
157 }
158 
GetCap()159 NetAllCapabilities GetTypeContext::GetCap()
160 {
161     return cap_;
162 }
163 } // namespace OHOS::NetManagerStandard