blob: 7e31b62fb60d19fbe3f8e533d2527ee339dbcead [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 saraswalc0e04842016-05-12 13:16:57 +053075import static org.onosproject.yangutils.utils.UtilConstants.MANAGER;
Bharat saraswale2d51d62016-03-23 19:40:35 +053076import static org.onosproject.yangutils.utils.UtilConstants.NEW_LINE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053077import static org.onosproject.yangutils.utils.UtilConstants.OPEN_PARENTHESIS;
Bharat saraswale2d51d62016-03-23 19:40:35 +053078import static org.onosproject.yangutils.utils.UtilConstants.PACKAGE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053079import static org.onosproject.yangutils.utils.UtilConstants.PERIOD;
Bharat saraswald72411a2016-04-19 01:00:16 +053080import static org.onosproject.yangutils.utils.UtilConstants.PRIVATE;
Bharat saraswale2d51d62016-03-23 19:40:35 +053081import static org.onosproject.yangutils.utils.UtilConstants.SEMI_COLAN;
Bharat saraswalc0e04842016-05-12 13:16:57 +053082import static org.onosproject.yangutils.utils.UtilConstants.SERVICE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053083import static org.onosproject.yangutils.utils.UtilConstants.SERVICE_ANNOTATION;
Bharat saraswale2d51d62016-03-23 19:40:35 +053084import static org.onosproject.yangutils.utils.UtilConstants.SLASH;
85import static org.onosproject.yangutils.utils.UtilConstants.SPACE;
Bharat saraswal33dfa012016-05-17 19:59:16 +053086import static org.onosproject.yangutils.utils.UtilConstants.TRUE;
Bharat saraswalc0e04842016-05-12 13:16:57 +053087import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.getJavaDoc;
Bharat saraswale2d51d62016-03-23 19:40:35 +053088import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.BUILDER_CLASS;
89import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.BUILDER_INTERFACE;
Bharat saraswald72411a2016-04-19 01:00:16 +053090import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.ENUM_CLASS;
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +053091import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.EVENT;
92import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.EVENT_LISTENER;
Bharat saraswal33dfa012016-05-17 19:59:16 +053093import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.EVENT_SUBJECT_CLASS;
Bharat saraswale2d51d62016-03-23 19:40:35 +053094import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.IMPL_CLASS;
95import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.INTERFACE;
Gaurav Agrawal56527662016-04-20 15:49:17 +053096import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.RPC_INTERFACE;
Bharat saraswalc0e04842016-05-12 13:16:57 +053097import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.RPC_MANAGER;
Bharat saraswale2d51d62016-03-23 19:40:35 +053098import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.insertDataIntoJavaFile;
Bharat saraswal33dfa012016-05-17 19:59:16 +053099import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.parsePkg;
Bharat saraswale2d51d62016-03-23 19:40:35 +0530100
101/**
Bharat saraswald9822e92016-04-05 15:13:44 +0530102 * Represents utilities for java file generator.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530103 */
104public final class JavaFileGeneratorUtils {
105
106 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530107 * Creates an instance of java file generator util.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530108 */
109 private JavaFileGeneratorUtils() {
110 }
111
112 /**
113 * Returns a file object for generated file.
114 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530115 * @param fileName file name
116 * @param filePath file package path
Bharat saraswale2d51d62016-03-23 19:40:35 +0530117 * @param extension file extension
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530118 * @param handle cached file handle
Bharat saraswale2d51d62016-03-23 19:40:35 +0530119 * @return file object
120 */
121 public static File getFileObject(String filePath, String fileName, String extension, JavaFileInfo handle) {
122
123 return new File(handle.getBaseCodeGenPath() + filePath + SLASH + fileName + extension);
124 }
125
126 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530127 * Returns data stored in temporary files.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530128 *
129 * @param generatedTempFiles temporary file types
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530130 * @param tempJavaFragmentFiles temp java fragment files
Bharat saraswale2d51d62016-03-23 19:40:35 +0530131 * @return data stored in temporary files
132 * @throws IOException when failed to get the data from temporary file handle
133 */
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530134 public static String getDataFromTempFileHandle(int generatedTempFiles,
135 TempJavaFragmentFiles tempJavaFragmentFiles)
136 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530137
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530138 TempJavaTypeFragmentFiles typeFragmentFiles = null;
Bharat saraswale2d51d62016-03-23 19:40:35 +0530139
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530140 if (tempJavaFragmentFiles instanceof TempJavaTypeFragmentFiles) {
141 typeFragmentFiles = (TempJavaTypeFragmentFiles) tempJavaFragmentFiles;
142 }
143
144 TempJavaBeanFragmentFiles beanFragmentFiles = null;
145
146 if (tempJavaFragmentFiles instanceof TempJavaBeanFragmentFiles) {
147 beanFragmentFiles = (TempJavaBeanFragmentFiles) tempJavaFragmentFiles;
148 }
149
150 TempJavaServiceFragmentFiles serviceFragmentFiles = null;
151 if (tempJavaFragmentFiles instanceof TempJavaServiceFragmentFiles) {
152 serviceFragmentFiles = (TempJavaServiceFragmentFiles) tempJavaFragmentFiles;
153 }
Bharat saraswal33dfa012016-05-17 19:59:16 +0530154
Bharat saraswalc0e04842016-05-12 13:16:57 +0530155 if ((generatedTempFiles & ATTRIBUTES_MASK) != 0) {
156 return tempJavaFragmentFiles
157 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getAttributesTempFileHandle());
158 } else if ((generatedTempFiles & GETTER_FOR_INTERFACE_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530159 return tempJavaFragmentFiles
160 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getGetterInterfaceTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530161 } else if ((generatedTempFiles & SETTER_FOR_INTERFACE_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530162 return tempJavaFragmentFiles
163 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getSetterInterfaceTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530164 } else if ((generatedTempFiles & GETTER_FOR_CLASS_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530165 return tempJavaFragmentFiles
166 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getGetterImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530167 } else if ((generatedTempFiles & SETTER_FOR_CLASS_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530168 return tempJavaFragmentFiles
169 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getSetterImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530170 } else if ((generatedTempFiles & CONSTRUCTOR_IMPL_MASK) != 0) {
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530171 if (beanFragmentFiles == null) {
172 throw new TranslatorException("Required constructor info is missing.");
173 }
174 return beanFragmentFiles
175 .getTemporaryDataFromFileHandle(beanFragmentFiles.getConstructorImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530176 } else if ((generatedTempFiles & HASH_CODE_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530177 return tempJavaFragmentFiles
178 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getHashCodeImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530179 } else if ((generatedTempFiles & EQUALS_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530180 return tempJavaFragmentFiles
181 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getEqualsImplTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530182 } else if ((generatedTempFiles & TO_STRING_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530183 return tempJavaFragmentFiles
184 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getToStringImplTempFileHandle());
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530185 } else if ((generatedTempFiles & OF_STRING_IMPL_MASK) != 0) {
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530186 if (typeFragmentFiles == null) {
187 throw new TranslatorException("Required of string implementation info is missing.");
188 }
189 return typeFragmentFiles
190 .getTemporaryDataFromFileHandle(typeFragmentFiles.getOfStringImplTempFileHandle());
191 } else if ((generatedTempFiles & CONSTRUCTOR_FOR_TYPE_MASK) != 0) {
192 if (typeFragmentFiles == null) {
193 throw new TranslatorException("Required constructor implementation info is missing.");
194 }
195 return typeFragmentFiles
196 .getTemporaryDataFromFileHandle(typeFragmentFiles.getConstructorForTypeTempFileHandle());
Gaurav Agrawal56527662016-04-20 15:49:17 +0530197 } else if ((generatedTempFiles & FROM_STRING_IMPL_MASK) != 0) {
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530198 return tempJavaFragmentFiles
Bharat saraswalc0e04842016-05-12 13:16:57 +0530199 .getTemporaryDataFromFileHandle(tempJavaFragmentFiles.getFromStringImplTempFileHandle());
200 } else if ((generatedTempFiles & ENUM_IMPL_MASK) != 0) {
201 if (!(tempJavaFragmentFiles instanceof TempJavaEnumerationFragmentFiles)) {
202 throw new TranslatorException("Required enum info is missing.");
203 }
204 TempJavaEnumerationFragmentFiles enumFragmentFiles =
205 (TempJavaEnumerationFragmentFiles) tempJavaFragmentFiles;
206 return enumFragmentFiles
207 .getTemporaryDataFromFileHandle(enumFragmentFiles.getEnumClassTempFileHandle());
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530208 } else if ((generatedTempFiles & RPC_INTERFACE_MASK) != 0) {
209 if (serviceFragmentFiles == null) {
210 throw new TranslatorException("Required rpc interface info is missing.");
211 }
212 return serviceFragmentFiles
213 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getRpcInterfaceTempFileHandle());
Gaurav Agrawal56527662016-04-20 15:49:17 +0530214 } else if ((generatedTempFiles & RPC_IMPL_MASK) != 0) {
VinodKumarS-Huaweicb3a1f52016-05-10 17:58:57 +0530215 if (serviceFragmentFiles == null) {
216 throw new TranslatorException("Required rpc implementation info is missing.");
217 }
218 return serviceFragmentFiles
219 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getRpcImplTempFileHandle());
Bharat saraswal33dfa012016-05-17 19:59:16 +0530220 } else if ((generatedTempFiles & EVENT_ENUM_MASK) != 0) {
221 if (serviceFragmentFiles == null) {
222 throw new TranslatorException("Required rpc implementation info is missing.");
223 }
224 return serviceFragmentFiles
225 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventEnumTempFileHandle());
226 } else if ((generatedTempFiles & EVENT_METHOD_MASK) != 0) {
227 if (serviceFragmentFiles == null) {
228 throw new TranslatorException("Required rpc implementation info is missing.");
229 }
230 return serviceFragmentFiles
231 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventMethodTempFileHandle());
232 } else if ((generatedTempFiles & EVENT_SUBJECT_GETTER_MASK) != 0) {
233 if (serviceFragmentFiles == null) {
234 throw new TranslatorException("Required rpc implementation info is missing.");
235 }
236 return serviceFragmentFiles
237 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventSubjectGetterTempFileHandle());
238 } else if ((generatedTempFiles & EVENT_SUBJECT_SETTER_MASK) != 0) {
239 if (serviceFragmentFiles == null) {
240 throw new TranslatorException("Required rpc implementation info is missing.");
241 }
242 return serviceFragmentFiles
243 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventSubjectSetterTempFileHandle());
244 } else if ((generatedTempFiles & EVENT_SUBJECT_ATTRIBUTE_MASK) != 0) {
245 if (serviceFragmentFiles == null) {
246 throw new TranslatorException("Required rpc implementation info is missing.");
247 }
248 return serviceFragmentFiles
249 .getTemporaryDataFromFileHandle(serviceFragmentFiles.getEventSubjectAttributeTempFileHandle());
Bharat saraswale2d51d62016-03-23 19:40:35 +0530250 }
251 return null;
252 }
253
254 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530255 * Initiates generation of file based on generated file type.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530256 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530257 * @param file generated file
Bharat saraswale2d51d62016-03-23 19:40:35 +0530258 * @param className generated file class name
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530259 * @param type generated file type
260 * @param imports imports for the file
261 * @param pkg generated file package
Bharat saraswal33dfa012016-05-17 19:59:16 +0530262 * @param pluginConfig plugin configurations
Bharat saraswale2d51d62016-03-23 19:40:35 +0530263 * @throws IOException when fails to generate a file
264 */
265 public static void initiateJavaFileGeneration(File file, String className, int type, List<String> imports,
Bharat saraswal33dfa012016-05-17 19:59:16 +0530266 String pkg, YangPluginConfig pluginConfig)
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530267 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530268
269 try {
270 file.createNewFile();
Bharat saraswal33dfa012016-05-17 19:59:16 +0530271 appendContents(file, className, type, imports, pkg, pluginConfig);
Bharat saraswale2d51d62016-03-23 19:40:35 +0530272 } catch (IOException e) {
273 throw new IOException("Failed to create " + file.getName() + " class file.");
274 }
275 }
276
277 /**
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530278 * Initiates generation of file based on generated file type.
279 *
280 * @param file generated file
281 * @param type generated file type
282 * @param imports imports for the file
283 * @param curNode current YANG node
284 * @param className class name
285 * @throws IOException when fails to generate a file
286 */
287 public static void initiateJavaFileGeneration(File file, int type, List<String> imports,
288 YangNode curNode, String className) throws IOException {
289
290 try {
291 file.createNewFile();
292 appendContents(file, type, imports, curNode, className);
293 } catch (IOException e) {
294 throw new IOException("Failed to create " + file.getName() + " class file.");
295 }
296 }
297
298 /**
Bharat saraswale2d51d62016-03-23 19:40:35 +0530299 * Appends all the contents into a generated java file.
300 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530301 * @param file generated file
302 * @param fileName generated file name
303 * @param type generated file type
304 * @param pkg generated file package
Bharat saraswale2d51d62016-03-23 19:40:35 +0530305 * @param importsList list of java imports.
306 * @throws IOException when fails to append contents
307 */
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530308 private static void appendContents(File file, int type, List<String> importsList, YangNode curNode,
309 String className) throws IOException {
310
311 JavaFileInfo javaFileInfo = ((JavaFileInfoContainer) curNode).getJavaFileInfo();
312
Bharat saraswal33dfa012016-05-17 19:59:16 +0530313 String name = javaFileInfo.getJavaName();
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530314 String path = javaFileInfo.getBaseCodeGenPath() + javaFileInfo.getPackageFilePath();
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530315
Bharat saraswal33dfa012016-05-17 19:59:16 +0530316 String pkgString = null;
317 if (type == GENERATE_EVENT_CLASS
318 || type == GENERATE_EVENT_LISTENER_INTERFACE
319 || type == GENERATE_EVENT_SUBJECT_CLASS) {
320 pkgString = parsePackageString(path + PERIOD + name, importsList);
321 } else {
322 pkgString = parsePackageString(path, importsList);
323 }
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530324 switch (type) {
325 case INTERFACE_MASK:
326 appendHeaderContents(file, pkgString, importsList);
327 write(file, type, INTERFACE, curNode, className);
328 break;
329 case GENERATE_SERVICE_AND_MANAGER:
330 appendHeaderContents(file, pkgString, importsList);
331 write(file, type, RPC_INTERFACE, curNode, className);
332 break;
333 case GENERATE_EVENT_CLASS:
334 appendHeaderContents(file, pkgString, importsList);
335 write(file, type, EVENT, curNode, className);
336 break;
Bharat saraswal33dfa012016-05-17 19:59:16 +0530337 case GENERATE_EVENT_LISTENER_INTERFACE:
338 appendHeaderContents(file, pkgString, importsList);
339 write(file, type, EVENT_LISTENER, curNode, className);
340 break;
341 case GENERATE_EVENT_SUBJECT_CLASS:
342 appendHeaderContents(file, pkgString, importsList);
343 write(file, type, EVENT_SUBJECT_CLASS, curNode, className);
344 break;
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530345 default:
346 break;
347 }
348 }
349
350 /**
351 * Appends all the contents into a generated java file.
352 *
353 * @param file generated file
354 * @param fileName generated file name
355 * @param type generated file type
356 * @param pkg generated file package
357 * @param importsList list of java imports.
358 * @throws IOException when fails to append contents
359 */
Bharat saraswal33dfa012016-05-17 19:59:16 +0530360 private static void appendContents(File file, String fileName, int type, List<String> importsList, String pkg,
361 YangPluginConfig pluginConfig)
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530362 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530363
364 String pkgString = parsePackageString(pkg, importsList);
365
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530366 switch (type) {
367 case IMPL_CLASS_MASK:
Bharat saraswal33dfa012016-05-17 19:59:16 +0530368 write(file, fileName, type, IMPL_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530369 break;
370 case BUILDER_INTERFACE_MASK:
Bharat saraswal33dfa012016-05-17 19:59:16 +0530371 write(file, fileName, type, BUILDER_INTERFACE, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530372 break;
373 case GENERATE_TYPEDEF_CLASS:
Bharat saraswale2d51d62016-03-23 19:40:35 +0530374 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530375 write(file, fileName, type, IMPL_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530376 break;
377 case BUILDER_CLASS_MASK:
Bharat saraswale2d51d62016-03-23 19:40:35 +0530378 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530379 write(file, fileName, type, BUILDER_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530380 break;
381 case GENERATE_UNION_CLASS:
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530382 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530383 write(file, fileName, type, IMPL_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530384 break;
385 case GENERATE_ENUM_CLASS:
Bharat saraswald72411a2016-04-19 01:00:16 +0530386 appendHeaderContents(file, pkgString, importsList);
Bharat saraswal33dfa012016-05-17 19:59:16 +0530387 write(file, fileName, type, ENUM_CLASS, pluginConfig);
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530388 break;
389 default:
390 break;
Bharat saraswale2d51d62016-03-23 19:40:35 +0530391 }
392 }
393
394 /**
395 * Removes base directory path from package and generates package string for file.
396 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530397 * @param javaPkg generated java package
Bharat saraswale2d51d62016-03-23 19:40:35 +0530398 * @param importsList list of imports
399 * @return package string
400 */
401 private static String parsePackageString(String javaPkg, List<String> importsList) {
402
Bharat saraswal33dfa012016-05-17 19:59:16 +0530403 javaPkg = parsePkg(getJavaPackageFromPackagePath(javaPkg));
Bharat saraswale2d51d62016-03-23 19:40:35 +0530404 if (importsList != null) {
405 if (!importsList.isEmpty()) {
406 return PACKAGE + SPACE + javaPkg + SEMI_COLAN + NEW_LINE;
407 } else {
408 return PACKAGE + SPACE + javaPkg + SEMI_COLAN;
409 }
410 } else {
411 return PACKAGE + SPACE + javaPkg + SEMI_COLAN;
412 }
413 }
414
415 /**
416 * Appends other contents to interface, builder and typedef classes.
417 * for example : ONOS copyright, imports and package.
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530418 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530419 * @param file generated file
420 * @param pkg generated package
Bharat saraswale2d51d62016-03-23 19:40:35 +0530421 * @param importsList list of imports
Gaurav Agrawal338735b2016-04-18 18:53:11 +0530422 * @throws IOException when fails to append contents
Bharat saraswale2d51d62016-03-23 19:40:35 +0530423 */
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530424 private static void appendHeaderContents(File file, String pkg, List<String> importsList)
425 throws IOException {
Bharat saraswale2d51d62016-03-23 19:40:35 +0530426
427 insertDataIntoJavaFile(file, CopyrightHeader.getCopyrightHeader());
428 insertDataIntoJavaFile(file, pkg);
429
430 /*
431 * TODO: add the file header using
432 * JavaCodeSnippetGen.getFileHeaderComment
433 */
434
435 if (importsList != null) {
436 insertDataIntoJavaFile(file, NEW_LINE);
437 for (String imports : importsList) {
438 insertDataIntoJavaFile(file, imports);
439 }
440 }
441 }
442
443 /**
Bharat saraswald9822e92016-04-05 15:13:44 +0530444 * Writes data to the specific generated file.
Bharat saraswale2d51d62016-03-23 19:40:35 +0530445 *
Vinod Kumar Se4b9b0c2016-04-30 21:09:15 +0530446 * @param file generated file
447 * @param fileName file name
448 * @param genType generated file type
Bharat saraswale2d51d62016-03-23 19:40:35 +0530449 * @param javaDocType java doc type
450 * @throws IOException when fails to write into a file
451 */
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530452 private static void write(File file, int genType, JavaDocType javaDocType, YangNode curNode, String fileName)
Bharat saraswale2d51d62016-03-23 19:40:35 +0530453 throws IOException {
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530454
Bharat saraswal33dfa012016-05-17 19:59:16 +0530455 YangPluginConfig pluginConfig = ((JavaFileInfoContainer) curNode).getJavaFileInfo().getPluginConfig();
Bharat saraswalc0e04842016-05-12 13:16:57 +0530456 if ((genType & GENERATE_SERVICE_AND_MANAGER) != 0) {
457 if (!fileName.contains(SERVICE)) {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530458 insertDataIntoJavaFile(file, getJavaDoc(RPC_MANAGER, fileName + MANAGER, false, pluginConfig));
459 insertDataIntoJavaFile(file, addComponentString());
Bharat saraswalc0e04842016-05-12 13:16:57 +0530460 } else {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530461 insertDataIntoJavaFile(file, getJavaDoc(javaDocType, fileName, false, pluginConfig));
Bharat saraswalc0e04842016-05-12 13:16:57 +0530462 }
463 } else {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530464 insertDataIntoJavaFile(file, getJavaDoc(javaDocType, fileName, false, pluginConfig));
Bharat saraswalc0e04842016-05-12 13:16:57 +0530465 }
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530466 insertDataIntoJavaFile(file, generateClassDefinition(genType, fileName, curNode));
467 }
468
469 /**
470 * Writes data to the specific generated file.
471 *
472 * @param file generated file
473 * @param fileName file name
474 * @param genType generated file type
475 * @param javaDocType java doc type
476 * @throws IOException when fails to write into a file
477 */
Bharat saraswal33dfa012016-05-17 19:59:16 +0530478 private static void write(File file, String fileName, int genType, JavaDocType javaDocType,
479 YangPluginConfig pluginConfig)
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530480 throws IOException {
Bharat saraswal33dfa012016-05-17 19:59:16 +0530481 insertDataIntoJavaFile(file, getJavaDoc(javaDocType, fileName, false, pluginConfig));
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530482 insertDataIntoJavaFile(file, generateClassDefinition(genType, fileName));
Bharat saraswale2d51d62016-03-23 19:40:35 +0530483 }
484
Bharat saraswald72411a2016-04-19 01:00:16 +0530485 /**
486 * Returns integer attribute for enum's class to get the values.
487 *
488 * @param className enum's class name
489 * @return enum's attribute
490 */
491 public static String getEnumsValueAttribute(String className) {
Bharat saraswalab4c6ba2016-05-17 14:19:38 +0530492 return NEW_LINE + FOUR_SPACE_INDENTATION + PRIVATE + SPACE + INT + SPACE + getSmallCase(className)
493 + SEMI_COLAN + NEW_LINE;
Bharat saraswald72411a2016-04-19 01:00:16 +0530494 }
495
Bharat saraswal33dfa012016-05-17 19:59:16 +0530496 /**
497 * Returns component string.
498 *
499 * @return component string
500 */
501 public static String addComponentString() {
502 return NEW_LINE + COMPONENT_ANNOTATION + SPACE + OPEN_PARENTHESIS + IMMEDIATE + SPACE
503 + EQUAL + SPACE + TRUE + CLOSE_PARENTHESIS + NEW_LINE + SERVICE_ANNOTATION;
504 }
Bharat saraswale2d51d62016-03-23 19:40:35 +0530505}