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

   1 /*
   2  * Copyright (c) 2021 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 com.google.common.collect.ImmutableList;
  11 import org.eclipse.jdt.annotation.NonNull;
  12 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  13 import org.opendaylight.yangtools.yang.model.api.stmt.AugmentEffectiveStatement;
  14 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeAwareEffectiveStatement;
  15 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeAwareEffectiveStatement.SchemaTreeNamespace;
  16 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
  17
  18 /**
  19  * Generator corresponding to a {@code augment} statement used as a child of a {@code module} statement.
  20  */
  21 final class ModuleAugmentGenerator extends AbstractAugmentGenerator {
  22     ModuleAugmentGenerator(final AugmentEffectiveStatement statement, final AbstractCompositeGenerator<?, ?> parent) {
  23         super(statement, parent);
  24     }
  25
  26     @Override
  27     TargetAugmentEffectiveStatement effectiveIn(final SchemaTreeAwareEffectiveStatement<?, ?> target) {
  28         final var augment = statement();
  29         final var stmts = augment.effectiveSubstatements();
  30         final var builder = ImmutableList.<EffectiveStatement<?, ?>>builderWithExpectedSize(stmts.size());
  31         for (var child : stmts) {
  32             if (child instanceof SchemaTreeEffectiveStatement) {
  33                 final var qname = ((SchemaTreeEffectiveStatement<?>) child).getIdentifier();
  34                 // FIXME: orElseThrow()?
  35                 target.get(SchemaTreeNamespace.class, qname).ifPresent(builder::add);
  36             } else {
  37                 builder.add(child);
  38             }
  39         }
  40         return new TargetAugmentEffectiveStatement(augment, target, builder.build());
  41     }
  42
  43     @NonNull AugmentRequirement startLinkage(final GeneratorContext context) {
  44         return new AugmentRequirement(this,
  45             context.resolveModule(statement().argument().firstNodeIdentifier().getModule()));
  46     }
  47 }