blob: 707d94768c0e99b23105a2fd62729e5e1bb7af44 [file] [log] [blame]
Bharat saraswale2d51d62016-03-23 19:40:35 +05301/*
Brian O'Connor5ab426f2016-04-09 01:19:45 -07002 * Copyright 2016-present Open Networking Laboratory
Bharat saraswale2d51d62016-03-23 19:40:35 +05303 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package org.onosproject.yangutils.translator.tojava.utils;
18
19import java.io.File;
20import java.io.IOException;
21import java.util.List;
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +053022
Bharat saraswalab4c6ba2016-05-17 14:19:38 +053023import org.onosproject.yangutils.datamodel.YangNode;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053024import org.onosproject.yangutils.translator.exception.TranslatorException;
Bharat saraswale2d51d62016-03-23 19:40:35 +053025import org.onosproject.yangutils.translator.tojava.JavaFileInfo;
Bharat saraswalab4c6ba2016-05-17 14:19:38 +053026import org.onosproject.yangutils.translator.tojava.JavaFileInfoContainer;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053027import org.onosproject.yangutils.translator.tojava.TempJavaBeanFragmentFiles;
Bharat saraswalc0e04842016-05-12 13:16:57 +053028import org.onosproject.yangutils.translator.tojava.TempJavaEnumerationFragmentFiles;
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +053029import org.onosproject.yangutils.translator.tojava.TempJavaFragmentFiles;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053030import org.onosproject.yangutils.translator.tojava.TempJavaServiceFragmentFiles;
31import org.onosproject.yangutils.translator.tojava.TempJavaTypeFragmentFiles;
Bharat saraswale2d51d62016-03-23 19:40:35 +053032import org.onosproject.yangutils.utils.io.impl.CopyrightHeader;
33import org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType;
34
35import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.BUILDER_CLASS_MASK;
36import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.BUILDER_INTERFACE_MASK;
Bharat saraswald72411a2016-04-19 01:00:16 +053037import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_ENUM_CLASS;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053038import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_EVENT_CLASS;
39import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_EVENT_LISTENER_INTERFACE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053040import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_EVENT_SUBJECT_CLASS;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053041import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_SERVICE_AND_MANAGER;
Bharat saraswale2d51d62016-03-23 19:40:35 +053042import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_TYPEDEF_CLASS;
Gaurav Agrawal338735b2016-04-18 18:53:11 +053043import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_UNION_CLASS;
Bharat saraswale2d51d62016-03-23 19:40:35 +053044import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.IMPL_CLASS_MASK;
45import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.INTERFACE_MASK;
Bharat saraswalc0e04842016-05-12 13:16:57 +053046import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.ATTRIBUTES_MASK;
Gaurav Agrawal338735b2016-04-18 18:53:11 +053047import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.CONSTRUCTOR_FOR_TYPE_MASK;
Bharat saraswale2d51d62016-03-23 19:40:35 +053048import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.CONSTRUCTOR_IMPL_MASK;
Bharat saraswald72411a2016-04-19 01:00:16 +053049import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.ENUM_IMPL_MASK;
Bharat saraswale2d51d62016-03-23 19:40:35 +053050import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.EQUALS_IMPL_MASK;
Bharat saraswal33dfa012016-05-17 19:59:16 +053051import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.EVENT_ENUM_MASK;
52import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.EVENT_METHOD_MASK;
53import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.EVENT_SUBJECT_ATTRIBUTE_MASK;
54import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.EVENT_SUBJECT_GETTER_MASK;
55import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.EVENT_SUBJECT_SETTER_MASK;
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +053056import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.FROM_STRING_IMPL_MASK;
Bharat saraswale2d51d62016-03-23 19:40:35 +053057import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.GETTER_FOR_CLASS_MASK;
58import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.GETTER_FOR_INTERFACE_MASK;
59import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.HASH_CODE_IMPL_MASK;
Gaurav Agrawal338735b2016-04-18 18:53:11 +053060import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.OF_STRING_IMPL_MASK;
Gaurav Agrawal56527662016-04-20 15:49:17 +053061import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.RPC_IMPL_MASK;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053062import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.RPC_INTERFACE_MASK;
Bharat saraswale2d51d62016-03-23 19:40:35 +053063import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.SETTER_FOR_CLASS_MASK;
64import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.SETTER_FOR_INTERFACE_MASK;
65import static org.onosproject.yangutils.translator.tojava.GeneratedTempFileType.TO_STRING_IMPL_MASK;
Bharat saraswalab4c6ba2016-05-17 14:19:38 +053066import static org.onosproject.yangutils.translator.tojava.utils.ClassDefinitionGenerator.generateClassDefinition;
Bharat saraswal2f11f652016-03-25 18:19:46 +053067import static org.onosproject.yangutils.translator.tojava.utils.JavaIdentifierSyntax.getJavaPackageFromPackagePath;
Bharat saraswald72411a2016-04-19 01:00:16 +053068import static org.onosproject.yangutils.translator.tojava.utils.JavaIdentifierSyntax.getSmallCase;
Bharat saraswal33dfa012016-05-17 19:59:16 +053069import static org.onosproject.yangutils.utils.UtilConstants.CLOSE_PARENTHESIS;
70import static org.onosproject.yangutils.utils.UtilConstants.COMPONENT_ANNOTATION;
71import static org.onosproject.yangutils.utils.UtilConstants.EQUAL;
Bharat saraswald72411a2016-04-19 01:00:16 +053072import static org.onosproject.yangutils.utils.UtilConstants.FOUR_SPACE_INDENTATION;
Bharat saraswal33dfa012016-05-17 19:59:16 +053073import static org.onosproject.yangutils.utils.UtilConstants.IMMEDIATE;
Bharat saraswald72411a2016-04-19 01:00:16 +053074import static org.onosproject.yangutils.utils.UtilConstants.INT;
Bharat saraswale2d51d62016-03-23 19:40:35 +053075import static org.onosproject.yangutils.utils.UtilConstants.NEW_LINE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053076import static org.onosproject.yangutils.utils.UtilConstants.OPEN_PARENTHESIS;
Bharat saraswale2d51d62016-03-23 19:40:35 +053077import static org.onosproject.yangutils.utils.UtilConstants.PACKAGE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053078import static org.onosproject.yangutils.utils.UtilConstants.PERIOD;
Bharat saraswald72411a2016-04-19 01:00:16 +053079import static org.onosproject.yangutils.utils.UtilConstants.PRIVATE;
janani bdd1314f2016-05-19 17:39:50 +053080import static org.onosproject.yangutils.utils.UtilConstants.REGEX_FOR_ANY_STRING_ENDING_WITH_SERVICE;
Bharat saraswale2d51d62016-03-23 19:40:35 +053081import static org.onosproject.yangutils.utils.UtilConstants.SEMI_COLAN;
Bharat saraswal33dfa012016-05-17 19:59:16 +053082import static org.onosproject.yangutils.utils.UtilConstants.SERVICE_ANNOTATION;
Bharat saraswale2d51d62016-03-23 19:40:35 +053083import static org.onosproject.yangutils.utils.UtilConstants.SLASH;
84import static org.onosproject.yangutils.utils.UtilConstants.SPACE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053085import static org.onosproject.yangutils.utils.UtilConstants.TRUE;
Bharat saraswalc0e04842016-05-12 13:16:57 +053086import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.getJavaDoc;
Bharat saraswale2d51d62016-03-23 19:40:35 +053087import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.BUILDER_CLASS;
88import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.BUILDER_INTERFACE;
Bharat saraswald72411a2016-04-19 01:00:16 +053089import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.ENUM_CLASS;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053090import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.EVENT;
91import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.EVENT_LISTENER;
Bharat saraswal33dfa012016-05-17 19:59:16 +053092import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.EVENT_SUBJECT_CLASS;
Bharat saraswale2d51d62016-03-23 19:40:35 +053093import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.IMPL_CLASS;
94import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.INTERFACE;
Gaurav Agrawal56527662016-04-20 15:49:17 +053095import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.RPC_INTERFACE;
Bharat saraswalc0e04842016-05-12 13:16:57 +053096import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.RPC_MANAGER;
Bharat saraswale2d51d62016-03-23 19:40:35 +053097import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.insertDataIntoJavaFile;
Bharat saraswal33dfa012016-05-17 19:59:16 +053098import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.parsePkg;
Bharat saraswale2d51d62016-03-23 19:40:35 +053099
100/**
Bharat saraswald9822e92016-04-05 15:13:44 +0530101 * Represents utilities for java file generator.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530102 */
103public final class JavaFileGeneratorUtils {
104
105 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530106 * Creates an instance of java file generator util.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530107 */
108 private JavaFileGeneratorUtils() {
109 }
110
111 /**
112 * Returns a file object for generated file.
113 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530114 * @param fileName file name
115 * @param filePath file package path
Bharat saraswale2d51d62016-03-23 19:40:35 +0530116 * @param extension file extension
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530117 * @param handle cached file handle
Bharat saraswale2d51d62016-03-23 19:40:35 +0530118 * @return file object
119 */
120 public static File getFileObject(String filePath, String fileName, String extension, JavaFileInfo handle) {
121
122 return new File(handle.getBaseCodeGenPath() + filePath + SLASH + fileName + extension);
123 }
124
125 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530126 * Returns data stored in temporary files.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530127 *
128 * @param generatedTempFiles temporary file types
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530129 * @param tempJavaFragmentFiles temp java fragment files
Bharat saraswale2d51d62016-03-23 19:40:35 +0530130 * @return data stored in temporary files
131 * @throws IOException when failed to get the data from temporary file handle
132 */
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530133 public static String getDataFromTempFileHandle(int generatedTempFiles,
134 TempJavaFragmentFiles tempJavaFragmentFiles)
135 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530136
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530137 TempJavaTypeFragmentFiles typeFragmentFiles = null;
Bharat saraswale2d51d62016-03-23 19:40:35 +0530138
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530139 if (tempJavaFragmentFiles instanceof TempJavaTypeFragmentFiles) {
140 typeFragmentFiles = (TempJavaTypeFragmentFiles) tempJavaFragmentFiles;
141 }
142
143 TempJavaBeanFragmentFiles beanFragmentFiles = null;
144
145 if (tempJavaFragmentFiles instanceof TempJavaBeanFragmentFiles) {
146 beanFragmentFiles = (TempJavaBeanFragmentFiles) tempJavaFragmentFiles;
147 }
148
149 TempJavaServiceFragmentFiles serviceFragmentFiles = null;
150 if (tempJavaFragmentFiles instanceof TempJavaServiceFragmentFiles) {
151 serviceFragmentFiles = (TempJavaServiceFragmentFiles) tempJavaFragmentFiles;
152 }
Bharat saraswal33dfa012016-05-17 19:59:16 +0530153
Bharat saraswalc0e04842016-05-12 13:16:57 +0530154 if ((generatedTempFiles & ATTRIBUTES_MASK) != 0) {
155 return tempJavaFragmentFiles
156 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getAttributesTempFileHandle());
157 } else if ((generatedTempFiles & GETTER_FOR_INTERFACE_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530158 return tempJavaFragmentFiles
159 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getGetterInterfaceTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530160 } else if ((generatedTempFiles & SETTER_FOR_INTERFACE_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530161 return tempJavaFragmentFiles
162 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getSetterInterfaceTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530163 } else if ((generatedTempFiles & GETTER_FOR_CLASS_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530164 return tempJavaFragmentFiles
165 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getGetterImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530166 } else if ((generatedTempFiles & SETTER_FOR_CLASS_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530167 return tempJavaFragmentFiles
168 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getSetterImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530169 } else if ((generatedTempFiles & CONSTRUCTOR_IMPL_MASK) != 0) {
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530170 if (beanFragmentFiles == null) {
171 throw new TranslatorException("Required constructor info is missing.");
172 }
173 return beanFragmentFiles
174 .getTemporaryDataFromFileHandle(beanFragmentFiles.getConstructorImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530175 } else if ((generatedTempFiles & HASH_CODE_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530176 return tempJavaFragmentFiles
177 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getHashCodeImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530178 } else if ((generatedTempFiles & EQUALS_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530179 return tempJavaFragmentFiles
180 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getEqualsImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530181 } else if ((generatedTempFiles & TO_STRING_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530182 return tempJavaFragmentFiles
183 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getToStringImplTempFileHandle());
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530184 } else if ((generatedTempFiles & OF_STRING_IMPL_MASK) != 0) {
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530185 if (typeFragmentFiles == null) {
186 throw new TranslatorException("Required of string implementation info is missing.");
187 }
188 return typeFragmentFiles
189 .getTemporaryDataFromFileHandle(typeFragmentFiles.getOfStringImplTempFileHandle());
190 } else if ((generatedTempFiles & CONSTRUCTOR_FOR_TYPE_MASK) != 0) {
191 if (typeFragmentFiles == null) {
192 throw new TranslatorException("Required constructor implementation info is missing.");
193 }
194 return typeFragmentFiles
195 .getTemporaryDataFromFileHandle(typeFragmentFiles.getConstructorForTypeTempFileHandle());
Gaurav Agrawal56527662016-04-20 15:49:17 +0530196 } else if ((generatedTempFiles & FROM_STRING_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530197 return tempJavaFragmentFiles
Bharat saraswalc0e04842016-05-12 13:16:57 +0530198 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getFromStringImplTempFileHandle());
199 } else if ((generatedTempFiles & ENUM_IMPL_MASK) != 0) {
200 if (!(tempJavaFragmentFiles instanceof TempJavaEnumerationFragmentFiles)) {
201 throw new TranslatorException("Required enum info is missing.");
202 }
203 TempJavaEnumerationFragmentFiles enumFragmentFiles =
204 (TempJavaEnumerationFragmentFiles) tempJavaFragmentFiles;
205 return enumFragmentFiles
206 .getTemporaryDataFromFileHandle(enumFragmentFiles.getEnumClassTempFileHandle());
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530207 } else if ((generatedTempFiles & RPC_INTERFACE_MASK) != 0) {
208 if (serviceFragmentFiles == null) {
209 throw new TranslatorException("Required rpc interface info is missing.");
210 }
211 return serviceFragmentFiles
212 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getRpcInterfaceTempFileHandle());
Gaurav Agrawal56527662016-04-20 15:49:17 +0530213 } else if ((generatedTempFiles & RPC_IMPL_MASK) != 0) {
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530214 if (serviceFragmentFiles == null) {
215 throw new TranslatorException("Required rpc implementation info is missing.");
216 }
217 return serviceFragmentFiles
218 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getRpcImplTempFileHandle());
Bharat saraswal33dfa012016-05-17 19:59:16 +0530219 } else if ((generatedTempFiles & EVENT_ENUM_MASK) != 0) {
220 if (serviceFragmentFiles == null) {
221 throw new TranslatorException("Required rpc implementation info is missing.");
222 }
223 return serviceFragmentFiles
224 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventEnumTempFileHandle());
225 } else if ((generatedTempFiles & EVENT_METHOD_MASK) != 0) {
226 if (serviceFragmentFiles == null) {
227 throw new TranslatorException("Required rpc implementation info is missing.");
228 }
229 return serviceFragmentFiles
230 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventMethodTempFileHandle());
231 } else if ((generatedTempFiles & EVENT_SUBJECT_GETTER_MASK) != 0) {
232 if (serviceFragmentFiles == null) {
233 throw new TranslatorException("Required rpc implementation info is missing.");
234 }
235 return serviceFragmentFiles
236 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventSubjectGetterTempFileHandle());
237 } else if ((generatedTempFiles & EVENT_SUBJECT_SETTER_MASK) != 0) {
238 if (serviceFragmentFiles == null) {
239 throw new TranslatorException("Required rpc implementation info is missing.");
240 }
241 return serviceFragmentFiles
242 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventSubjectSetterTempFileHandle());
243 } else if ((generatedTempFiles & EVENT_SUBJECT_ATTRIBUTE_MASK) != 0) {
244 if (serviceFragmentFiles == null) {
245 throw new TranslatorException("Required rpc implementation info is missing.");
246 }
247 return serviceFragmentFiles
248 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventSubjectAttributeTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530249 }
250 return null;
251 }
252
253 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530254 * Initiates generation of file based on generated file type.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530255 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530256 * @param file generated file
Bharat saraswale2d51d62016-03-23 19:40:35 +0530257 * @param className generated file class name
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530258 * @param type generated file type
259 * @param imports imports for the file
260 * @param pkg generated file package
Bharat saraswal33dfa012016-05-17 19:59:16 +0530261 * @param pluginConfig plugin configurations
Bharat saraswale2d51d62016-03-23 19:40:35 +0530262 * @throws IOException when fails to generate a file
263 */
264 public static void initiateJavaFileGeneration(File file, String className, int type, List<String> imports,
Bharat saraswal33dfa012016-05-17 19:59:16 +0530265 String pkg, YangPluginConfig pluginConfig)
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530266 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530267
268 try {
269 file.createNewFile();
Bharat saraswal33dfa012016-05-17 19:59:16 +0530270 appendContents(file, className, type, imports, pkg, pluginConfig);
Bharat saraswale2d51d62016-03-23 19:40:35 +0530271 } catch (IOException e) {
272 throw new IOException("Failed to create " + file.getName() + " class file.");
273 }
274 }
275
276 /**
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530277 * Initiates generation of file based on generated file type.
278 *
279 * @param file generated file
280 * @param type generated file type
281 * @param imports imports for the file
282 * @param curNode current YANG node
283 * @param className class name
284 * @throws IOException when fails to generate a file
285 */
286 public static void initiateJavaFileGeneration(File file, int type, List<String> imports,
287 YangNode curNode, String className) throws IOException {
288
289 try {
290 file.createNewFile();
291 appendContents(file, type, imports, curNode, className);
292 } catch (IOException e) {
293 throw new IOException("Failed to create " + file.getName() + " class file.");
294 }
295 }
296
297 /**
Bharat saraswale2d51d62016-03-23 19:40:35 +0530298 * Appends all the contents into a generated java file.
299 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530300 * @param file generated file
301 * @param fileName generated file name
302 * @param type generated file type
303 * @param pkg generated file package
Bharat saraswale2d51d62016-03-23 19:40:35 +0530304 * @param importsList list of java imports.
305 * @throws IOException when fails to append contents
306 */
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530307 private static void appendContents(File file, int type, List<String> importsList, YangNode curNode,
308 String className) throws IOException {
309
310 JavaFileInfo javaFileInfo = ((JavaFileInfoContainer) curNode).getJavaFileInfo();
311
Bharat saraswal33dfa012016-05-17 19:59:16 +0530312 String name = javaFileInfo.getJavaName();
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530313 String path = javaFileInfo.getBaseCodeGenPath() + javaFileInfo.getPackageFilePath();
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530314
Bharat saraswal33dfa012016-05-17 19:59:16 +0530315 String pkgString = null;
316 if (type == GENERATE_EVENT_CLASS
317 || type == GENERATE_EVENT_LISTENER_INTERFACE
318 || type == GENERATE_EVENT_SUBJECT_CLASS) {
janani bdd1314f2016-05-19 17:39:50 +0530319 pkgString = parsePackageString((path + PERIOD + name).toLowerCase(), importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530320 } else {
321 pkgString = parsePackageString(path, importsList);
322 }
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530323 switch (type) {
324 case INTERFACE_MASK:
325 appendHeaderContents(file, pkgString, importsList);
326 write(file, type, INTERFACE, curNode, className);
327 break;
328 case GENERATE_SERVICE_AND_MANAGER:
329 appendHeaderContents(file, pkgString, importsList);
330 write(file, type, RPC_INTERFACE, curNode, className);
331 break;
332 case GENERATE_EVENT_CLASS:
333 appendHeaderContents(file, pkgString, importsList);
334 write(file, type, EVENT, curNode, className);
335 break;
Bharat saraswal33dfa012016-05-17 19:59:16 +0530336 case GENERATE_EVENT_LISTENER_INTERFACE:
337 appendHeaderContents(file, pkgString, importsList);
338 write(file, type, EVENT_LISTENER, curNode, className);
339 break;
340 case GENERATE_EVENT_SUBJECT_CLASS:
341 appendHeaderContents(file, pkgString, importsList);
342 write(file, type, EVENT_SUBJECT_CLASS, curNode, className);
343 break;
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530344 default:
345 break;
346 }
347 }
348
349 /**
350 * Appends all the contents into a generated java file.
351 *
352 * @param file generated file
353 * @param fileName generated file name
354 * @param type generated file type
355 * @param pkg generated file package
356 * @param importsList list of java imports.
357 * @throws IOException when fails to append contents
358 */
Bharat saraswal33dfa012016-05-17 19:59:16 +0530359 private static void appendContents(File file, String fileName, int type, List<String> importsList, String pkg,
360 YangPluginConfig pluginConfig)
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530361 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530362
363 String pkgString = parsePackageString(pkg, importsList);
364
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530365 switch (type) {
366 case IMPL_CLASS_MASK:
Bharat saraswal33dfa012016-05-17 19:59:16 +0530367 write(file, fileName, type, IMPL_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530368 break;
369 case BUILDER_INTERFACE_MASK:
Bharat saraswal33dfa012016-05-17 19:59:16 +0530370 write(file, fileName, type, BUILDER_INTERFACE, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530371 break;
372 case GENERATE_TYPEDEF_CLASS:
Bharat saraswale2d51d62016-03-23 19:40:35 +0530373 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530374 write(file, fileName, type, IMPL_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530375 break;
376 case BUILDER_CLASS_MASK:
Bharat saraswale2d51d62016-03-23 19:40:35 +0530377 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530378 write(file, fileName, type, BUILDER_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530379 break;
380 case GENERATE_UNION_CLASS:
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530381 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530382 write(file, fileName, type, IMPL_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530383 break;
384 case GENERATE_ENUM_CLASS:
Bharat saraswald72411a2016-04-19 01:00:16 +0530385 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530386 write(file, fileName, type, ENUM_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530387 break;
388 default:
389 break;
Bharat saraswale2d51d62016-03-23 19:40:35 +0530390 }
391 }
392
393 /**
394 * Removes base directory path from package and generates package string for file.
395 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530396 * @param javaPkg generated java package
Bharat saraswale2d51d62016-03-23 19:40:35 +0530397 * @param importsList list of imports
398 * @return package string
399 */
400 private static String parsePackageString(String javaPkg, List<String> importsList) {
401
Bharat saraswal33dfa012016-05-17 19:59:16 +0530402 javaPkg = parsePkg(getJavaPackageFromPackagePath(javaPkg));
Bharat saraswale2d51d62016-03-23 19:40:35 +0530403 if (importsList != null) {
404 if (!importsList.isEmpty()) {
405 return PACKAGE + SPACE + javaPkg + SEMI_COLAN + NEW_LINE;
406 } else {
407 return PACKAGE + SPACE + javaPkg + SEMI_COLAN;
408 }
409 } else {
410 return PACKAGE + SPACE + javaPkg + SEMI_COLAN;
411 }
412 }
413
414 /**
415 * Appends other contents to interface, builder and typedef classes.
416 * for example : ONOS copyright, imports and package.
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530417 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530418 * @param file generated file
419 * @param pkg generated package
Bharat saraswale2d51d62016-03-23 19:40:35 +0530420 * @param importsList list of imports
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530421 * @throws IOException when fails to append contents
Bharat saraswale2d51d62016-03-23 19:40:35 +0530422 */
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530423 private static void appendHeaderContents(File file, String pkg, List<String> importsList)
424 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530425
426 insertDataIntoJavaFile(file, CopyrightHeader.getCopyrightHeader());
427 insertDataIntoJavaFile(file, pkg);
428
429 /*
430 * TODO: add the file header using
431 * JavaCodeSnippetGen.getFileHeaderComment
432 */
433
434 if (importsList != null) {
435 insertDataIntoJavaFile(file, NEW_LINE);
436 for (String imports : importsList) {
437 insertDataIntoJavaFile(file, imports);
438 }
439 }
440 }
441
442 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530443 * Writes data to the specific generated file.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530444 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530445 * @param file generated file
446 * @param fileName file name
447 * @param genType generated file type
Bharat saraswale2d51d62016-03-23 19:40:35 +0530448 * @param javaDocType java doc type
449 * @throws IOException when fails to write into a file
450 */
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530451 private static void write(File file, int genType, JavaDocType javaDocType, YangNode curNode, String fileName)
Bharat saraswale2d51d62016-03-23 19:40:35 +0530452 throws IOException {
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530453
Bharat saraswal33dfa012016-05-17 19:59:16 +0530454 YangPluginConfig pluginConfig = ((JavaFileInfoContainer) curNode).getJavaFileInfo().getPluginConfig();
Bharat saraswalc0e04842016-05-12 13:16:57 +0530455 if ((genType & GENERATE_SERVICE_AND_MANAGER) != 0) {
janani bdd1314f2016-05-19 17:39:50 +0530456 if (!fileName.matches(REGEX_FOR_ANY_STRING_ENDING_WITH_SERVICE)) {
457 insertDataIntoJavaFile(file, getJavaDoc(RPC_MANAGER, fileName, false, pluginConfig));
Bharat saraswal33dfa012016-05-17 19:59:16 +0530458 insertDataIntoJavaFile(file, addComponentString());
Bharat saraswalc0e04842016-05-12 13:16:57 +0530459 } else {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530460 insertDataIntoJavaFile(file, getJavaDoc(javaDocType, fileName, false, pluginConfig));
Bharat saraswalc0e04842016-05-12 13:16:57 +0530461 }
462 } else {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530463 insertDataIntoJavaFile(file, getJavaDoc(javaDocType, fileName, false, pluginConfig));
Bharat saraswalc0e04842016-05-12 13:16:57 +0530464 }
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530465 insertDataIntoJavaFile(file, generateClassDefinition(genType, fileName, curNode));
466 }
467
468 /**
469 * Writes data to the specific generated file.
470 *
471 * @param file generated file
472 * @param fileName file name
473 * @param genType generated file type
474 * @param javaDocType java doc type
475 * @throws IOException when fails to write into a file
476 */
Bharat saraswal33dfa012016-05-17 19:59:16 +0530477 private static void write(File file, String fileName, int genType, JavaDocType javaDocType,
478 YangPluginConfig pluginConfig)
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530479 throws IOException {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530480 insertDataIntoJavaFile(file, getJavaDoc(javaDocType, fileName, false, pluginConfig));
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530481 insertDataIntoJavaFile(file, generateClassDefinition(genType, fileName));
Bharat saraswale2d51d62016-03-23 19:40:35 +0530482 }
483
Bharat saraswald72411a2016-04-19 01:00:16 +0530484 /**
485 * Returns integer attribute for enum's class to get the values.
486 *
487 * @param className enum's class name
488 * @return enum's attribute
489 */
490 public static String getEnumsValueAttribute(String className) {
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530491 return NEW_LINE + FOUR_SPACE_INDENTATION + PRIVATE + SPACE + INT + SPACE + getSmallCase(className)
492 + SEMI_COLAN + NEW_LINE;
Bharat saraswald72411a2016-04-19 01:00:16 +0530493 }
494
Bharat saraswal33dfa012016-05-17 19:59:16 +0530495 /**
496 * Returns component string.
497 *
498 * @return component string
499 */
500 public static String addComponentString() {
501 return NEW_LINE + COMPONENT_ANNOTATION + SPACE + OPEN_PARENTHESIS + IMMEDIATE + SPACE
502 + EQUAL + SPACE + TRUE + CLOSE_PARENTHESIS + NEW_LINE + SERVICE_ANNOTATION;
503 }
Bharat saraswale2d51d62016-03-23 19:40:35 +0530504}