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 "codegen/code_generator.h"
17 #include <cstdlib>
18 #include <unistd.h>
19 #include <sys/stat.h>
20 #include "codegen/rust_code_emitter.h"
21 #include "codegen/cpp_code_emitter.h"
22 #include "codegen/ts_code_emitter.h"
23 #include "util/logger.h"
24
25 namespace OHOS {
26 namespace Idl {
27 const char* CodeGenerator::TAG = "CodeGenerator";
CodeGenerator(MetaComponent * mc,const String & language,const String & dir)28 CodeGenerator::CodeGenerator(MetaComponent* mc, const String& language, const String& dir)
29 : targetLanguage_(language),
30 targetDirectory_(dir),
31 metaComponent_(mc)
32 {
33 if (language.Equals("rust")) {
34 emitter_ = new RustCodeEmitter(metaComponent_);
35 } else if (language.Equals("cpp")) {
36 emitter_ = new CppCodeEmitter(metaComponent_);
37 } else if (language.Equals("ts")) {
38 emitter_ = new TsCodeEmitter(metaComponent_);
39 } else {
40 Logger::E(TAG, "Unknown language: %s.", language.string());
41 }
42 }
43
ResolveDirectory()44 bool CodeGenerator::ResolveDirectory()
45 {
46 #ifdef __MINGW32__
47 if (targetDirectory_.IndexOf(":\\") == -1) {
48 char* cmd = getcwd(nullptr, 0);
49 targetDirectory_ = String::Format("%s\\%s", cmd, targetDirectory_.string());
50 free(cmd);
51 }
52 #else
53 if (!targetDirectory_.StartsWith("/")) {
54 char* cwd = getcwd(nullptr, 0);
55 targetDirectory_ = String::Format("%s/%s", cwd, targetDirectory_.string());
56 free(cwd);
57 }
58 #endif
59
60 if (!access(targetDirectory_.string(), R_OK | W_OK)) {
61 return true;
62 }
63
64 #ifdef __MINGW32__
65 if (mkdir(targetDirectory_.string()) != 0) {
66 #else
67 if (mkdir(targetDirectory_.string(), S_IRWXU | S_IRWXG | S_IRWXO) != 0) {
68 #endif
69 Logger::E(TAG, "Create \"%s\" directory failed.", targetDirectory_.string());
70 return false;
71 }
72
73 return true;
74 }
75
76 bool CodeGenerator::Generate()
77 {
78 if (!ResolveDirectory()) {
79 return false;
80 }
81
82 emitter_->SetDirectory(targetDirectory_);
83
84 emitter_->EmitInterface();
85 emitter_->EmitInterfaceProxy();
86 emitter_->EmitInterfaceStub();
87
88 return true;
89 }
90 } // namespace Idl
91 } // namespace OHOS