binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/JavaFileTemplate.java

   1 /*
   2  * Copyright (c) 2018 Pantheon Technologies, s.r.o. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.binding.java.api.generator;
   9
  10 import static com.google.common.base.Verify.verify;
  11 import static java.util.Objects.requireNonNull;
  12
  13 import java.util.Optional;
  14 import java.util.stream.Collectors;
  15 import org.opendaylight.mdsal.binding.model.api.GeneratedProperty;
  16 import org.opendaylight.mdsal.binding.model.api.GeneratedTransferObject;
  17 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
  18 import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
  19 import org.opendaylight.mdsal.binding.model.api.Type;
  20 import org.opendaylight.mdsal.binding.model.util.Types;
  21
  22 /**
  23  * Base Java file template. Contains a non-null type and imports which the generated code refers to.
  24  */
  25 class JavaFileTemplate {
  26     private final AbstractJavaGeneratedType javaType;
  27     private final GeneratedType type;
  28
  29     JavaFileTemplate(final GeneratedType type) {
  30         this(new TopLevelJavaGeneratedType(type), type);
  31     }
  32
  33     JavaFileTemplate(final AbstractJavaGeneratedType javaType, final GeneratedType type) {
  34         this.javaType = requireNonNull(javaType);
  35         this.type = requireNonNull(type);
  36     }
  37
  38     final AbstractJavaGeneratedType javaType() {
  39         return javaType;
  40     }
  41
  42     final GeneratedType type() {
  43         return type;
  44     }
  45
  46     final GeneratedProperty findProperty(final GeneratedTransferObject gto, final String name) {
  47         final Optional<GeneratedProperty> optProp = gto.getProperties().stream()
  48                 .filter(prop -> prop.getName().equals(name)).findFirst();
  49         if (optProp.isPresent()) {
  50             return optProp.get();
  51         }
  52
  53         final GeneratedTransferObject parent = gto.getSuperType();
  54         return parent != null ? findProperty(parent, name) : null;
  55     }
  56
  57     final String generateImportBlock() {
  58         verify(javaType instanceof TopLevelJavaGeneratedType);
  59         return ((TopLevelJavaGeneratedType) javaType).imports().map(name -> "import " + name + ";\n")
  60                 .collect(Collectors.joining());
  61     }
  62
  63     final String importedName(final Type intype) {
  64         return javaType.getReferenceString(intype);
  65     }
  66
  67     final String importedName(final Class<?> cls) {
  68         return importedName(Types.typeForClass(cls));
  69     }
  70
  71     final String importedName(final JavaTypeName intype) {
  72         return javaType.getReferenceString(intype);
  73     }
  74
  75     final void addImport(final Class<?> cls) {
  76         javaType.getReferenceString(JavaTypeName.create(cls));
  77     }
  78
  79     // Exposed for BuilderTemplate
  80     boolean isLocalInnerClass(final JavaTypeName name) {
  81         final Optional<JavaTypeName> optEnc = name.immediatelyEnclosingClass();
  82         return optEnc.isPresent() && type.getIdentifier().equals(optEnc.get());
  83     }
  84 }