binding2/mdsal-binding2-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/javav2/java/api/generator/BuilderGenerator.java

   1 /*
   2  * Copyright (c) 2017 Cisco Systems, Inc. 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
   9 package org.opendaylight.mdsal.binding.javav2.java.api.generator;
  10
  11 import com.google.common.annotations.Beta;
  12 import org.opendaylight.mdsal.binding.javav2.java.api.generator.renderers.BuilderRenderer;
  13 import org.opendaylight.mdsal.binding.javav2.model.api.CodeGenerator;
  14 import org.opendaylight.mdsal.binding.javav2.model.api.GeneratedTransferObject;
  15 import org.opendaylight.mdsal.binding.javav2.model.api.GeneratedType;
  16 import org.opendaylight.mdsal.binding.javav2.model.api.Type;
  17 import org.opendaylight.mdsal.binding.javav2.model.api.UnitName;
  18 import org.opendaylight.mdsal.binding.javav2.spec.structural.Augmentable;
  19 import org.opendaylight.mdsal.binding.javav2.spec.structural.Augmentation;
  20 import org.opendaylight.yangtools.concepts.Identifier;
  21
  22 /**
  23  * Transformer of the data from the virtual form to JAVA programming language.
  24  * The result source code represent java class. For generation of the source
  25  * code is used the template written in Twirl (Scala based) language.
  26  */
  27 @Beta
  28 public final class BuilderGenerator implements CodeGenerator {
  29
  30     /**
  31      * Constant used as suffix for builder name.
  32      */
  33     public static final String BUILDER = "Builder";
  34
  35     @Override
  36     public String generate(Type type) {
  37         if ((type instanceof GeneratedType) && !(type instanceof GeneratedTransferObject)) {
  38             final GeneratedType genType = (GeneratedType) type;
  39             return new BuilderRenderer(genType).generateTemplate();
  40         } else {
  41             return "";
  42         }
  43     }
  44
  45     @Override
  46     public boolean isAcceptable(Type type) {
  47         if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
  48             for (Type t : ((GeneratedType) type).getImplements()) {
  49                 // "rpc" and "grouping" elements do not implement Augmentable
  50                 if (t.getFullyQualifiedName().equals(Augmentable.class.getName())) {
  51                     return true;
  52                 } else if (t.getFullyQualifiedName().equals(Augmentation.class.getName())) {
  53                     return true;
  54                 }
  55
  56             }
  57         }
  58         return false;
  59     }
  60
  61     @Override
  62     public Identifier getUnitName(Type type) {
  63         return new UnitName(type.getName() + BUILDER);
  64     }
  65 }