Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2016-present Open Networking Laboratory |
| 3 | * |
| 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 | package org.onosproject.yangutils.translator.tojava.javamodel; |
| 17 | |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 18 | import org.onosproject.yangutils.datamodel.javadatamodel.YangJavaIdentity; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 19 | import org.onosproject.yangutils.translator.exception.TranslatorException; |
| 20 | import org.onosproject.yangutils.translator.tojava.JavaCodeGenerator; |
| 21 | import org.onosproject.yangutils.translator.tojava.JavaCodeGeneratorInfo; |
Bharat saraswal | e3175d3 | 2016-08-31 17:50:11 +0530 | [diff] [blame] | 22 | import org.onosproject.yangutils.translator.tojava.JavaFileInfoTranslator; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 23 | import org.onosproject.yangutils.translator.tojava.JavaImportData; |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 24 | import org.onosproject.yangutils.translator.tojava.JavaQualifiedTypeInfoTranslator; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 25 | import org.onosproject.yangutils.translator.tojava.TempJavaCodeFragmentFiles; |
Bharat saraswal | e3175d3 | 2016-08-31 17:50:11 +0530 | [diff] [blame] | 26 | import org.onosproject.yangutils.utils.io.YangPluginConfig; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 27 | |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 28 | import java.io.File; |
| 29 | import java.io.IOException; |
| 30 | import java.util.List; |
| 31 | |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 32 | import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_IDENTITY_CLASS; |
Bharat saraswal | 039f59c | 2016-07-14 21:57:13 +0530 | [diff] [blame] | 33 | import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_INTERFACE_WITH_BUILDER; |
Bharat saraswal | af413b8 | 2016-07-14 15:18:20 +0530 | [diff] [blame] | 34 | import static org.onosproject.yangutils.translator.tojava.YangJavaModelUtils.updatePackageInfo; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 35 | import static org.onosproject.yangutils.translator.tojava.utils.JavaFileGeneratorUtils.getFileObject; |
| 36 | import static org.onosproject.yangutils.translator.tojava.utils.JavaFileGeneratorUtils.initiateJavaFileGeneration; |
| 37 | import static org.onosproject.yangutils.translator.tojava.utils.JavaIdentifierSyntax.createPackage; |
Bharat saraswal | a5c2851 | 2016-11-10 21:09:23 +0530 | [diff] [blame^] | 38 | import static org.onosproject.yangutils.translator.tojava.utils.MethodsGenerator.getFromStringMethodForIdentity; |
| 39 | import static org.onosproject.yangutils.translator.tojava.utils.MethodsGenerator.getToStringMethodForIdentity; |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 40 | import static org.onosproject.yangutils.translator.tojava.utils.TranslatorErrorType.FAIL_AT_ENTRY; |
| 41 | import static org.onosproject.yangutils.translator.tojava.utils.TranslatorErrorType.FAIL_AT_EXIT; |
| 42 | import static org.onosproject.yangutils.translator.tojava.utils.TranslatorUtils.getErrorMsg; |
Bharat saraswal | a5c2851 | 2016-11-10 21:09:23 +0530 | [diff] [blame^] | 43 | import static org.onosproject.yangutils.utils.UtilConstants.CLOSE_CURLY_BRACKET; |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 44 | import static org.onosproject.yangutils.utils.UtilConstants.EMPTY_STRING; |
| 45 | import static org.onosproject.yangutils.utils.UtilConstants.JAVA_FILE_EXTENSION; |
Bharat saraswal | d14cbe8 | 2016-07-14 13:26:18 +0530 | [diff] [blame] | 46 | import static org.onosproject.yangutils.utils.io.impl.FileSystemUtil.closeFile; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 47 | import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.getCapitalCase; |
Bharat saraswal | a5c2851 | 2016-11-10 21:09:23 +0530 | [diff] [blame^] | 48 | import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.insertDataIntoJavaFile; |
Bharat saraswal | e3175d3 | 2016-08-31 17:50:11 +0530 | [diff] [blame] | 49 | import static org.onosproject.yangutils.utils.io.impl.YangIoUtils.validateLineLength; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 50 | |
| 51 | /** |
| 52 | * Represents input information extended to support java code generation. |
| 53 | */ |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 54 | public class YangJavaIdentityTranslator extends YangJavaIdentity |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 55 | implements JavaCodeGeneratorInfo, JavaCodeGenerator { |
| 56 | |
Bharat saraswal | 8beac34 | 2016-08-04 02:00:03 +0530 | [diff] [blame] | 57 | //Contains the information of the imported. |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 58 | private transient JavaImportData importData; |
| 59 | |
| 60 | /** |
| 61 | * File handle to maintain temporary java code fragments as per the code |
| 62 | * snippet types. |
| 63 | */ |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 64 | private transient TempJavaCodeFragmentFiles tempFileHandle; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 65 | |
| 66 | /** |
| 67 | * Creates YANG java container object. |
| 68 | */ |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 69 | public YangJavaIdentityTranslator() { |
Bharat saraswal | e50edca | 2016-08-05 01:58:25 +0530 | [diff] [blame] | 70 | setJavaFileInfo(new JavaFileInfoTranslator()); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 71 | getJavaFileInfo().setGeneratedFileTypes(GENERATE_INTERFACE_WITH_BUILDER); |
| 72 | importData = new JavaImportData(); |
| 73 | } |
| 74 | |
| 75 | /** |
| 76 | * Returns the generated java file information. |
| 77 | * |
| 78 | * @return generated java file information |
| 79 | */ |
| 80 | @Override |
Bharat saraswal | e50edca | 2016-08-05 01:58:25 +0530 | [diff] [blame] | 81 | public JavaFileInfoTranslator getJavaFileInfo() { |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 82 | if (javaFileInfo == null) { |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 83 | throw new TranslatorException("Missing java info in java identity" + |
| 84 | " node " + getName()); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 85 | } |
Bharat saraswal | e50edca | 2016-08-05 01:58:25 +0530 | [diff] [blame] | 86 | return (JavaFileInfoTranslator) javaFileInfo; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 87 | } |
| 88 | |
| 89 | /** |
| 90 | * Sets the java file info object. |
| 91 | * |
| 92 | * @param javaInfo java file info object |
| 93 | */ |
| 94 | @Override |
Bharat saraswal | e50edca | 2016-08-05 01:58:25 +0530 | [diff] [blame] | 95 | public void setJavaFileInfo(JavaFileInfoTranslator javaInfo) { |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 96 | javaFileInfo = javaInfo; |
| 97 | } |
| 98 | |
| 99 | /** |
| 100 | * Returns the temporary file handle. |
| 101 | * |
| 102 | * @return temporary file handle |
| 103 | */ |
| 104 | @Override |
| 105 | public TempJavaCodeFragmentFiles getTempJavaCodeFragmentFiles() { |
| 106 | return tempFileHandle; |
| 107 | } |
| 108 | |
| 109 | /** |
| 110 | * Sets temporary file handle. |
| 111 | * |
| 112 | * @param fileHandle temporary file handle |
| 113 | */ |
| 114 | @Override |
| 115 | public void setTempJavaCodeFragmentFiles(TempJavaCodeFragmentFiles fileHandle) { |
| 116 | tempFileHandle = fileHandle; |
| 117 | } |
| 118 | |
| 119 | /** |
| 120 | * Prepare the information for java code generation corresponding to YANG |
| 121 | * container info. |
| 122 | * |
| 123 | * @param yangPlugin YANG plugin config |
| 124 | * @throws TranslatorException translator operation fail |
| 125 | */ |
| 126 | @Override |
| 127 | public void generateCodeEntry(YangPluginConfig yangPlugin) throws TranslatorException { |
| 128 | try { |
Bharat saraswal | 8beac34 | 2016-08-04 02:00:03 +0530 | [diff] [blame] | 129 | |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 130 | updatePackageInfo(this, yangPlugin); |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 131 | JavaQualifiedTypeInfoTranslator basePkgInfo = |
| 132 | new JavaQualifiedTypeInfoTranslator(); |
| 133 | JavaFileInfoTranslator itsInfo = getJavaFileInfo(); |
Bharat saraswal | a5c2851 | 2016-11-10 21:09:23 +0530 | [diff] [blame^] | 134 | String name = itsInfo.getJavaName(); |
| 135 | String className = getCapitalCase(name); |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 136 | String path = itsInfo.getPackageFilePath(); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 137 | createPackage(this); |
| 138 | List<String> imports = null; |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 139 | boolean isQualified; |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 140 | |
| 141 | if (getBaseNode() != null && getBaseNode().getReferredIdentity() != null) { |
Shankara-Huawei | b756477 | 2016-08-02 18:13:13 +0530 | [diff] [blame] | 142 | if (!(getBaseNode().getReferredIdentity() instanceof YangJavaIdentityTranslator)) { |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 143 | throw new TranslatorException(getErrorMsg(FAIL_AT_ENTRY, this, |
| 144 | EMPTY_STRING)); |
| 145 | |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 146 | } |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 147 | YangJavaIdentityTranslator base = |
| 148 | (YangJavaIdentityTranslator) getBaseNode().getReferredIdentity(); |
| 149 | JavaFileInfoTranslator info = base.getJavaFileInfo(); |
| 150 | String baseClassName = getCapitalCase(info.getJavaName()); |
| 151 | String basePkg = info.getPackage(); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 152 | basePkgInfo.setClassInfo(baseClassName); |
| 153 | basePkgInfo.setPkgInfo(basePkg); |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 154 | isQualified = importData.addImportInfo(basePkgInfo, className, |
| 155 | javaFileInfo.getPackage()); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 156 | if (!isQualified) { |
| 157 | imports = importData.getImports(); |
| 158 | } |
| 159 | } |
| 160 | |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 161 | File file = getFileObject(path, className, JAVA_FILE_EXTENSION, itsInfo); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 162 | |
| 163 | initiateJavaFileGeneration(file, GENERATE_IDENTITY_CLASS, imports, this, className); |
Bharat saraswal | e3175d3 | 2016-08-31 17:50:11 +0530 | [diff] [blame] | 164 | file = validateLineLength(file); |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 165 | //Add to string and from string method to class |
Bharat saraswal | a5c2851 | 2016-11-10 21:09:23 +0530 | [diff] [blame^] | 166 | addStringMethodsToClass(file, name); |
| 167 | insertDataIntoJavaFile(file, CLOSE_CURLY_BRACKET); |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 168 | |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 169 | closeFile(file, false); |
| 170 | } catch (IOException e) { |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 171 | throw new TranslatorException(getErrorMsg(FAIL_AT_EXIT, this, |
| 172 | e.getLocalizedMessage())); |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 173 | } |
| 174 | } |
| 175 | |
Bharat saraswal | a5c2851 | 2016-11-10 21:09:23 +0530 | [diff] [blame^] | 176 | private void addStringMethodsToClass(File file, String className) throws IOException { |
| 177 | insertDataIntoJavaFile(file, getToStringMethodForIdentity(getName())); |
| 178 | insertDataIntoJavaFile(file, getFromStringMethodForIdentity( |
| 179 | className, getName())); |
Bharat saraswal | c2fd322 | 2016-10-19 16:10:27 +0530 | [diff] [blame] | 180 | } |
| 181 | |
Shankara-Huawei | 234cd09 | 2016-07-14 11:35:34 +0530 | [diff] [blame] | 182 | /** |
| 183 | * Create a java file using the YANG container info. |
| 184 | * |
| 185 | * @throws TranslatorException translator operation fail |
| 186 | */ |
| 187 | @Override |
| 188 | public void generateCodeExit() throws TranslatorException { |
| 189 | /* Do nothing, file is already generated in entry*/ |
| 190 | } |
| 191 | } |
| 192 | |