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 }
90 
InstantiateClass(napi_env env,const string & className,const vector<napi_value> & args)91 napi_value NClass::InstantiateClass(napi_env env, const string& className, const vector<napi_value>& args)
92 {
93     NClass &nClass = NClass::GetInstance();
94     lock_guard<std::mutex>(nClass.exClassMapLock);
95 
96     auto it = nClass.exClassMap.find(className);
97     if (it == nClass.exClassMap.end()) {
98         HILOGE("Class %{public}s hasn't been saved yet", className.c_str());
99         return nullptr;
100     }
101 
102     napi_value cons = nullptr;
103     napi_status status = napi_get_reference_value(env, it->second, &cons);
104     if (status != napi_ok) {
105         HILOGE("INNER BUG. Cannot deref class %{public}s because of %{public}d", className.c_str(), status);
106         return nullptr;
107     }
108 
109     napi_value instance = nullptr;
110     status = napi_new_instance(env, cons, args.size(), args.data(), &instance);
111     if (status != napi_ok) {
112         HILOGE("INNER BUG. Cannot instantiate the class %{public}s because of %{public}d", className.c_str(), status);
113         return nullptr;
114     }
115 
116     return instance;
117 }
118 } // namespace LibN
119 } // namespace FileManagement
120 } // namespace OHOS
121