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 "n_class.h"
17
18 #include "filemgmt_libhilog.h"
19
20 namespace OHOS {
21 namespace FileManagement {
22 namespace LibN {
23 using namespace std;
GetInstance()24 NClass &NClass::GetInstance()
25 {
26 static thread_local NClass nClass;
27 return nClass;
28 }
29
DefineClass(napi_env env,string className,napi_callback constructor,vector<napi_property_descriptor> && properties)30 tuple<bool, napi_value> NClass::DefineClass(napi_env env,
31 string className,
32 napi_callback constructor,
33 vector<napi_property_descriptor> &&properties)
34 {
35 napi_value classVal = nullptr;
36 napi_status stat = napi_define_class(env, className.c_str(), className.length(), constructor, nullptr,
37 properties.size(), properties.data(), &classVal);
38 if (stat != napi_ok) {
39 HILOGE("INNER BUG. Cannot define class %{public}s because of %{public}d", className.c_str(), stat);
40 }
41
42 return {stat == napi_ok, classVal};
43 }
44
SaveClass(napi_env env,string className,napi_value exClass)45 bool NClass::SaveClass(napi_env env, string className, napi_value exClass)
46 {
47 NClass &nClass = NClass::GetInstance();
48 lock_guard<std::mutex>(nClass.exClassMapLock);
49
50 if (nClass.exClassMap.find(className) != nClass.exClassMap.end()) {
51 return true;
52 }
53
54 napi_ref constructor;
55 napi_status res = napi_create_reference(env, exClass, 1, &constructor);
56 if (res == napi_ok) {
57 nClass.exClassMap.insert({className, constructor});
58 } else {
59 HILOGE("INNER BUG. Cannot ref class constructor %{public}s because of %{public}d", className.c_str(), res);
60 }
61
62 if (!nClass.addCleanHook) {
63 napi_status status = napi_add_env_cleanup_hook(env, CleanClass, env);
64 if (status != napi_ok) {
65 HILOGE("INNER BUG. Cleanup_hook registation has failed because of %{public}d", res);
66 } else {
67 nClass.addCleanHook = true;
68 }
69 }
70 return res == napi_ok;
71 }
72
CleanClass(void * arg)73 void NClass::CleanClass(void *arg)
74 {
75 napi_env env = reinterpret_cast<napi_env>(arg);
76 NClass &nClass = NClass::GetInstance();
77 lock_guard<std::mutex>(nClass.exClassMapLock);
78 {
79 lock_guard<std::mutex>(nClass.wrapLock);
80 nClass.wrapReleased = true;
81 }
82 napi_status res;
83 for (auto it = nClass.exClassMap.begin(); it != nClass.exClassMap.end(); ++it) {
84 res = napi_delete_reference(env, it->second);
85 if (res != napi_ok) {
86 HILOGE("Cannot del ref class constructor %{public}s because of %{public}d", it->first.c_str(), res);
87 }
88 }
89 nClass.exClassMap.clear();
90 }
91
InstantiateClass(napi_env env,const string & className,const vector<napi_value> & args)92 napi_value NClass::InstantiateClass(napi_env env, const string& className, const vector<napi_value>& args)
93 {
94 NClass &nClass = NClass::GetInstance();
95 lock_guard<std::mutex>(nClass.exClassMapLock);
96
97 auto it = nClass.exClassMap.find(className);
98 if (it == nClass.exClassMap.end()) {
99 HILOGE("Class %{public}s hasn't been saved yet", className.c_str());
100 return nullptr;
101 }
102
103 napi_value cons = nullptr;
104 napi_status status = napi_get_reference_value(env, it->second, &cons);
105 if (status != napi_ok) {
106 HILOGE("INNER BUG. Cannot deref class %{public}s because of %{public}d", className.c_str(), status);
107 return nullptr;
108 }
109
110 napi_value instance = nullptr;
111 status = napi_new_instance(env, cons, args.size(), args.data(), &instance);
112 if (status != napi_ok) {
113 HILOGE("INNER BUG. Cannot instantiate the class %{public}s because of %{public}d", className.c_str(), status);
114 return nullptr;
115 }
116
117 return instance;
118 }
119 } // namespace LibN
120 } // namespace FileManagement
121 } // namespace OHOS
122