binding/mdsal-binding-generator/src/main/java/org/opendaylight/mdsal/binding/generator/impl/reactor/AugmentRequirement.java

   1 /*
   2  * Copyright (c) 2022 PANTHEON.tech, 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.generator.impl.reactor;
   9
  10 import static com.google.common.base.Verify.verify;
  11 import static com.google.common.base.Verify.verifyNotNull;
  12 import static java.util.Objects.requireNonNull;
  13
  14 import java.util.HashSet;
  15 import java.util.Iterator;
  16 import java.util.Set;
  17 import org.eclipse.jdt.annotation.NonNull;
  18 import org.eclipse.jdt.annotation.Nullable;
  19 import org.opendaylight.yangtools.concepts.Mutable;
  20 import org.opendaylight.yangtools.yang.common.QName;
  21 import org.opendaylight.yangtools.yang.common.QNameModule;
  22
  23 final class AugmentRequirement implements Mutable {
  24     private final @NonNull Set<QNameModule> squashNamespaces;
  25     private final @NonNull AbstractAugmentGenerator augment;
  26     private final @NonNull Iterator<QName> remaining;
  27
  28     private QNameModule localNamespace;
  29     private QName qname;
  30
  31     private AugmentRequirement(final AbstractAugmentGenerator augment, final Iterator<QName> remaining,
  32             final QName qname, final QNameModule localNamespace, final Set<QNameModule> squashNamespaces) {
  33         this.augment = requireNonNull(augment);
  34         this.remaining = requireNonNull(remaining);
  35         this.qname = requireNonNull(qname);
  36         this.squashNamespaces = requireNonNull(squashNamespaces);
  37         this.localNamespace = localNamespace;
  38     }
  39
  40     AugmentRequirement(final AbstractAugmentGenerator augment, final Iterator<QName> path) {
  41         this(augment, path, path.next(), null, new HashSet<>(4));
  42     }
  43
  44     AugmentRequirement(final AbstractAugmentGenerator augment, final Iterator<QName> path,
  45             final GroupingGenerator grouping) {
  46         this(augment, path, path.next(), grouping.getQName().getModule(), new HashSet<>(4));
  47         squashNamespaces.add(qname.getModule());
  48     }
  49
  50     @NonNull AbstractAugmentGenerator augment() {
  51         return augment;
  52     }
  53
  54     @Nullable QName qname() {
  55         return qname;
  56     }
  57
  58     @NonNull QName adjustedQName() {
  59         final var qn = verifyNotNull(qname);
  60         return squashNamespaces.contains(qn.getModule()) ? qn.bindTo(verifyNotNull(localNamespace)) : qn;
  61     }
  62
  63     void inGrouping(final GroupingGenerator grouping, final AbstractExplicitGenerator<?> gen) {
  64         squashNamespaces.add(qname.getModule());
  65         localNamespace = grouping.statement().argument().getModule();
  66         grouping.addRequirement(this);
  67     }
  68
  69     void inLocal(final AbstractExplicitGenerator<?> gen) {
  70         addRequirement(gen);
  71     }
  72
  73     private void addRequirement(final AbstractExplicitGenerator<?> gen) {
  74         final var composite = verifyComposite(gen);
  75         qname = remaining.hasNext() ? remaining.next() : null;
  76         composite.addRequirement(this);
  77     }
  78
  79     private static AbstractCompositeGenerator<?> verifyComposite(final AbstractExplicitGenerator<?> gen) {
  80         verify(gen instanceof AbstractCompositeGenerator, "Unexpected generator %s", gen);
  81         return (AbstractCompositeGenerator<?>) gen;
  82     }
  83 }