binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/TopLevelJavaGeneratedType.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 com.google.common.collect.BiMap;
  11 import com.google.common.collect.HashBiMap;
  12 import java.util.Comparator;
  13 import java.util.Iterator;
  14 import java.util.Map.Entry;
  15 import java.util.stream.Stream;
  16 import org.eclipse.jdt.annotation.NonNull;
  17 import org.eclipse.jdt.annotation.NonNullByDefault;
  18 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
  19 import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
  20
  21 /**
  22  * This class tracks types generated into a single Java compilation unit (file) and manages imports of that compilation
  23  * unit. Since we are generating only public classes, this is synonymous to a top-level Java type.
  24  */
  25 @NonNullByDefault
  26 final class TopLevelJavaGeneratedType extends AbstractJavaGeneratedType {
  27     private final BiMap<JavaTypeName, String> importedTypes = HashBiMap.create();
  28
  29     TopLevelJavaGeneratedType(final GeneratedType genType) {
  30         super(genType);
  31     }
  32
  33     @Override
  34     String localTypeName(@NonNull final JavaTypeName type) {
  35         // Locally-anchored type, this is simple: just strip the first local name component and concat the others
  36         final Iterator<String> it = type.localNameComponents().iterator();
  37         it.next();
  38
  39         final StringBuilder sb = new StringBuilder().append(it.next());
  40         while (it.hasNext()) {
  41             sb.append('.').append(it.next());
  42         }
  43         return sb.toString();
  44     }
  45
  46     @Override
  47     boolean importCheckedType(final JavaTypeName type) {
  48         if (importedTypes.containsKey(type)) {
  49             return true;
  50         }
  51         final String simpleName = type.simpleName();
  52         if (importedTypes.containsValue(simpleName)) {
  53             return false;
  54         }
  55         importedTypes.put(type, simpleName);
  56         return true;
  57     }
  58
  59     Stream<JavaTypeName> imports() {
  60         return importedTypes.entrySet().stream().filter(this::needsExplicitImport).map(Entry::getKey)
  61                 .sorted(Comparator.comparing(JavaTypeName::toString));
  62     }
  63
  64     private boolean needsExplicitImport(final Entry<JavaTypeName, String> entry) {
  65         final JavaTypeName name = entry.getKey();
  66
  67         if (!getName().packageName().equals(name.packageName())) {
  68             // Different package: need to import it
  69             return true;
  70         }
  71
  72         if (!name.immediatelyEnclosingClass().isPresent()) {
  73             // This a top-level class import, we can skip it
  74             return false;
  75         }
  76
  77         // This is a nested class, we need to spell it out if the import entry points to the simple name
  78         return entry.getValue().equals(name.simpleName());
  79     }
  80 }