yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/DeviateStatementImpl.java

   1 /*
   2  * Copyright (c) 2015 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 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
   9
  10 import com.google.common.collect.ImmutableMap;
  11 import com.google.common.collect.ImmutableMap.Builder;
  12 import com.google.common.collect.ImmutableSet;
  13 import com.google.common.collect.Iterables;
  14 import java.util.Collection;
  15 import java.util.Map;
  16 import java.util.Objects;
  17 import java.util.Set;
  18 import javax.annotation.Nonnull;
  19 import org.opendaylight.yangtools.yang.common.QNameModule;
  20 import org.opendaylight.yangtools.yang.common.YangVersion;
  21 import org.opendaylight.yangtools.yang.model.api.DeviateKind;
  22 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
  23 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  24 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
  25 import org.opendaylight.yangtools.yang.model.api.stmt.DeviateStatement;
  26 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
  27 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
  28 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
  29 import org.opendaylight.yangtools.yang.parser.spi.meta.CopyType;
  30 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
  31 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
  32 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceAction;
  33 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceContext;
  34 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.Prerequisite;
  35 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
  36 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  37 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
  38 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
  39 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
  40 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleCtxToModuleQName;
  41 import org.opendaylight.yangtools.yang.parser.spi.source.ModulesDeviatedByModules;
  42 import org.opendaylight.yangtools.yang.parser.spi.source.ModulesDeviatedByModules.SupportedModules;
  43 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
  44 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase;
  45 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.DeviateEffectiveStatementImpl;
  46 import org.slf4j.Logger;
  47 import org.slf4j.LoggerFactory;
  48
  49 public class DeviateStatementImpl extends AbstractDeclaredStatement<DeviateKind> implements DeviateStatement {
  50     private static final Logger LOG = LoggerFactory.getLogger(DeviateStatementImpl.class);
  51
  52     private static final SubstatementValidator DEVIATE_NOT_SUPPORTED_SUBSTATEMENT_VALIDATOR =
  53             SubstatementValidator.builder(YangStmtMapping.DEVIATE).build();
  54
  55     private static final SubstatementValidator DEVIATE_ADD_SUBSTATEMENT_VALIDATOR =
  56             SubstatementValidator.builder(YangStmtMapping.DEVIATE)
  57                 .addOptional(YangStmtMapping.CONFIG)
  58                 .addOptional(YangStmtMapping.DEFAULT)
  59                 .addOptional(YangStmtMapping.MANDATORY)
  60                 .addOptional(YangStmtMapping.MAX_ELEMENTS)
  61                 .addOptional(YangStmtMapping.MIN_ELEMENTS)
  62                 .addAny(YangStmtMapping.MUST)
  63                 .addAny(YangStmtMapping.UNIQUE)
  64                 .addOptional(YangStmtMapping.UNITS)
  65                 .build();
  66
  67     private static final SubstatementValidator DEVIATE_REPLACE_SUBSTATEMENT_VALIDATOR =
  68             SubstatementValidator.builder(YangStmtMapping.DEVIATE)
  69                 .addOptional(YangStmtMapping.CONFIG)
  70                 .addOptional(YangStmtMapping.DEFAULT)
  71                 .addOptional(YangStmtMapping.MANDATORY)
  72                 .addOptional(YangStmtMapping.MAX_ELEMENTS)
  73                 .addOptional(YangStmtMapping.MIN_ELEMENTS)
  74                 .addOptional(YangStmtMapping.TYPE)
  75                 .addOptional(YangStmtMapping.UNITS)
  76                 .build();
  77
  78     private static final SubstatementValidator DEVIATE_DELETE_SUBSTATEMENT_VALIDATOR =
  79             SubstatementValidator.builder(YangStmtMapping.DEVIATE)
  80                 .addOptional(YangStmtMapping.DEFAULT)
  81                 .addAny(YangStmtMapping.MUST)
  82                 .addAny(YangStmtMapping.UNIQUE)
  83                 .addOptional(YangStmtMapping.UNITS)
  84                 .build();
  85
  86     protected DeviateStatementImpl(final StmtContext<DeviateKind, DeviateStatement, ?> context) {
  87         super(context);
  88     }
  89
  90     public static class Definition extends AbstractStatementSupport<DeviateKind, DeviateStatement,
  91             EffectiveStatement<DeviateKind, DeviateStatement>> {
  92         private static final Map<String, DeviateKind> KEYWORD_TO_DEVIATE_MAP;
  93         static {
  94             final Builder<String, DeviateKind> keywordToDeviateMapBuilder = ImmutableMap.builder();
  95             for (final DeviateKind deviate : DeviateKind.values()) {
  96                 keywordToDeviateMapBuilder.put(deviate.getKeyword(), deviate);
  97             }
  98             KEYWORD_TO_DEVIATE_MAP = keywordToDeviateMapBuilder.build();
  99         }
 100
 101         private static final Set<YangStmtMapping> SINGLETON_STATEMENTS = ImmutableSet.of(
 102                 YangStmtMapping.UNITS, YangStmtMapping.CONFIG, YangStmtMapping.MANDATORY,
 103                 YangStmtMapping.MIN_ELEMENTS, YangStmtMapping.MAX_ELEMENTS);
 104
 105         public Definition() {
 106             super(YangStmtMapping.DEVIATE);
 107         }
 108
 109         @Override
 110         public DeviateKind parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
 111             return SourceException.throwIfNull(KEYWORD_TO_DEVIATE_MAP.get(value),
 112                 ctx.getStatementSourceReference(), "String '%s' is not valid deviate argument", value);
 113         }
 114
 115         @Override
 116         public DeviateStatement createDeclared(final StmtContext<DeviateKind, DeviateStatement, ?> ctx) {
 117             return new DeviateStatementImpl(ctx);
 118         }
 119
 120         @Override
 121         public EffectiveStatement<DeviateKind, DeviateStatement> createEffective(
 122                 final StmtContext<DeviateKind, DeviateStatement, EffectiveStatement<DeviateKind,
 123                         DeviateStatement>> ctx) {
 124             return new DeviateEffectiveStatementImpl(ctx);
 125         }
 126
 127         @Override
 128         public void onFullDefinitionDeclared(final Mutable<DeviateKind, DeviateStatement,
 129                 EffectiveStatement<DeviateKind, DeviateStatement>> deviateStmtCtx) {
 130             final DeviateKind deviateKind = deviateStmtCtx.getStatementArgument();
 131             getSubstatementValidatorForDeviate(deviateKind).validate(deviateStmtCtx);
 132
 133             final SchemaNodeIdentifier deviationTarget =
 134                     (SchemaNodeIdentifier) deviateStmtCtx.getParentContext().getStatementArgument();
 135
 136             if (!isDeviationSupported(deviateStmtCtx, deviationTarget)) {
 137                 return;
 138             }
 139
 140             final ModelActionBuilder deviateAction = deviateStmtCtx.newInferenceAction(
 141                     ModelProcessingPhase.EFFECTIVE_MODEL);
 142
 143             final Prerequisite<StmtContext<DeviateKind, DeviateStatement, EffectiveStatement<DeviateKind,
 144                     DeviateStatement>>> sourceCtxPrerequisite =
 145                     deviateAction.requiresCtx(deviateStmtCtx, ModelProcessingPhase.EFFECTIVE_MODEL);
 146
 147             final Prerequisite<Mutable<?, ?, EffectiveStatement<?, ?>>> targetCtxPrerequisite =
 148                     deviateAction.mutatesEffectiveCtx(deviateStmtCtx.getRoot(),
 149                         SchemaNodeIdentifierBuildNamespace.class,  deviationTarget);
 150
 151             deviateAction.apply(new InferenceAction() {
 152                 @Override
 153                 public void apply(final InferenceContext ctx) throws InferenceException {
 154                     // FIXME once BUG-7760 gets fixed, there will be no need for these dirty casts
 155                     final StatementContextBase<?, ?, ?> sourceNodeStmtCtx =
 156                             (StatementContextBase<?, ?, ?>) sourceCtxPrerequisite.resolve(ctx);
 157                     final StatementContextBase<?, ?, ?> targetNodeStmtCtx =
 158                             (StatementContextBase<?, ?, ?>) targetCtxPrerequisite.resolve(ctx);
 159
 160                     switch (deviateKind) {
 161                         case NOT_SUPPORTED:
 162                             targetNodeStmtCtx.setIsSupportedToBuildEffective(false);
 163                             break;
 164                         case ADD:
 165                             performDeviateAdd(sourceNodeStmtCtx, targetNodeStmtCtx);
 166                             break;
 167                         case REPLACE:
 168                             performDeviateReplace(sourceNodeStmtCtx, targetNodeStmtCtx);
 169                             break;
 170                         case DELETE:
 171                             performDeviateDelete(sourceNodeStmtCtx, targetNodeStmtCtx);
 172                     }
 173                 }
 174
 175                 @Override
 176                 public void prerequisiteFailed(final Collection<? extends Prerequisite<?>> failed) {
 177                     throw new InferenceException(deviateStmtCtx.getParentContext().getStatementSourceReference(),
 178                         "Deviation target '%s' not found.", deviationTarget);
 179                 }
 180             });
 181         }
 182
 183         private static boolean isDeviationSupported(final Mutable<DeviateKind, DeviateStatement,
 184                 EffectiveStatement<DeviateKind, DeviateStatement>> deviateStmtCtx,
 185                 final SchemaNodeIdentifier deviationTarget) {
 186             final Map<QNameModule, Set<QNameModule>> modulesDeviatedByModules = deviateStmtCtx.getFromNamespace(
 187                     ModulesDeviatedByModules.class, SupportedModules.SUPPORTED_MODULES);
 188             if (modulesDeviatedByModules == null) {
 189                 return true;
 190             }
 191
 192             final QNameModule currentModule = deviateStmtCtx.getFromNamespace(ModuleCtxToModuleQName.class,
 193                     deviateStmtCtx.getRoot());
 194             final QNameModule targetModule = deviationTarget.getLastComponent().getModule();
 195
 196             final Set<QNameModule> deviationModulesSupportedByTargetModule = modulesDeviatedByModules.get(targetModule);
 197             if (deviationModulesSupportedByTargetModule != null) {
 198                 return deviationModulesSupportedByTargetModule.contains(currentModule);
 199             }
 200
 201             return false;
 202         }
 203
 204         private static void performDeviateAdd(final StatementContextBase<?, ?, ?> deviateStmtCtx,
 205                 final StatementContextBase<?, ?, ?> targetCtx) {
 206             for (Mutable<?, ?, ?> originalStmtCtx : deviateStmtCtx.mutableDeclaredSubstatements()) {
 207                 validateDeviationTarget(originalStmtCtx, targetCtx);
 208                 addStatement(originalStmtCtx, targetCtx);
 209             }
 210         }
 211
 212         private static void addStatement(final Mutable<?, ?, ?> stmtCtxToBeAdded,
 213                 final StatementContextBase<?, ?, ?> targetCtx) {
 214             if (!StmtContextUtils.isUnknownStatement(stmtCtxToBeAdded)) {
 215                 final StatementDefinition stmtToBeAdded = stmtCtxToBeAdded.getPublicDefinition();
 216                 if (SINGLETON_STATEMENTS.contains(stmtToBeAdded) || YangStmtMapping.DEFAULT.equals(stmtToBeAdded)
 217                         && YangStmtMapping.LEAF.equals(targetCtx.getPublicDefinition())) {
 218                     final Iterable<StmtContext<?, ?, ?>> targetCtxSubstatements = Iterables.concat(
 219                             targetCtx.declaredSubstatements(), targetCtx.effectiveSubstatements());
 220
 221                     for (final StmtContext<?, ?, ?> targetCtxSubstatement : targetCtxSubstatements) {
 222                         InferenceException.throwIf(stmtToBeAdded.equals(targetCtxSubstatement.getPublicDefinition()),
 223                             stmtCtxToBeAdded.getStatementSourceReference(), "Deviation cannot add substatement %s "
 224                         + "to target node %s because it is already defined in target and can appear only once.",
 225                         stmtToBeAdded.getStatementName(), targetCtx.getStatementArgument());
 226                     }
 227                 }
 228             }
 229
 230             targetCtx.addEffectiveSubstatement(targetCtx.childCopyOf(stmtCtxToBeAdded, CopyType.ORIGINAL));
 231         }
 232
 233         private static void performDeviateReplace(final StatementContextBase<?, ?, ?> deviateStmtCtx,
 234                 final StatementContextBase<?, ?, ?> targetCtx) {
 235             for (Mutable<?, ?, ?> originalStmtCtx : deviateStmtCtx.mutableDeclaredSubstatements()) {
 236                 validateDeviationTarget(originalStmtCtx, targetCtx);
 237                 replaceStatement(originalStmtCtx, targetCtx);
 238             }
 239         }
 240
 241         private static void replaceStatement(final Mutable<?, ?, ?> stmtCtxToBeReplaced,
 242                 final StatementContextBase<?, ?, ?> targetCtx) {
 243             final StatementDefinition stmtToBeReplaced = stmtCtxToBeReplaced.getPublicDefinition();
 244
 245             if (YangStmtMapping.DEFAULT.equals(stmtToBeReplaced)
 246                     && YangStmtMapping.LEAF_LIST.equals(targetCtx.getPublicDefinition())) {
 247                 LOG.error("Deviation cannot replace substatement {} in target leaf-list {} because a leaf-list can " +
 248                         "have multiple default statements. At line: {}", stmtToBeReplaced.getStatementName(),
 249                         targetCtx.getStatementArgument(), stmtCtxToBeReplaced.getStatementSourceReference());
 250                 return;
 251             }
 252
 253             for (final StmtContext<?, ?, ?> targetCtxSubstatement : targetCtx.effectiveSubstatements()) {
 254                 if (stmtToBeReplaced.equals(targetCtxSubstatement.getPublicDefinition())) {
 255                     targetCtx.removeStatementFromEffectiveSubstatements(stmtToBeReplaced);
 256                     targetCtx.addEffectiveSubstatement(targetCtx.childCopyOf(stmtCtxToBeReplaced, CopyType.ORIGINAL));
 257                     return;
 258                 }
 259             }
 260
 261             for (final Mutable<?, ?, ?> targetCtxSubstatement : targetCtx.mutableDeclaredSubstatements()) {
 262                 if (stmtToBeReplaced.equals(targetCtxSubstatement.getPublicDefinition())) {
 263                     targetCtxSubstatement.setIsSupportedToBuildEffective(false);
 264                     targetCtx.addEffectiveSubstatement(targetCtx.childCopyOf(stmtCtxToBeReplaced, CopyType.ORIGINAL));
 265                     return;
 266                 }
 267             }
 268
 269             throw new InferenceException(stmtCtxToBeReplaced.getStatementSourceReference(), "Deviation cannot " +
 270                     "replace substatement %s in target node %s because it does not exist in target node.",
 271                     stmtToBeReplaced.getStatementName(), targetCtx.getStatementArgument());
 272         }
 273
 274         private static void performDeviateDelete(final StatementContextBase<?, ?, ?> deviateStmtCtx,
 275                 final StatementContextBase<?, ?, ?> targetCtx) {
 276             for (Mutable<?, ?, ?> originalStmtCtx : deviateStmtCtx.mutableDeclaredSubstatements()) {
 277                 validateDeviationTarget(originalStmtCtx, targetCtx);
 278                 deleteStatement(originalStmtCtx, targetCtx);
 279             }
 280         }
 281
 282         private static void deleteStatement(final StmtContext<?, ?, ?> stmtCtxToBeDeleted,
 283                 final StatementContextBase<?, ?, ?> targetCtx) {
 284             final StatementDefinition stmtToBeDeleted = stmtCtxToBeDeleted.getPublicDefinition();
 285             final String stmtArgument = stmtCtxToBeDeleted.rawStatementArgument();
 286
 287             for (final Mutable<?, ?, ?> targetCtxSubstatement : targetCtx.mutableEffectiveSubstatements()) {
 288                 if (statementsAreEqual(stmtToBeDeleted, stmtArgument, targetCtxSubstatement.getPublicDefinition(),
 289                         targetCtxSubstatement.rawStatementArgument())) {
 290                     targetCtx.removeStatementFromEffectiveSubstatements(stmtToBeDeleted, stmtArgument);
 291                     return;
 292                 }
 293             }
 294
 295             for (final Mutable<?, ?, ?> targetCtxSubstatement : targetCtx.mutableDeclaredSubstatements()) {
 296                 if (statementsAreEqual(stmtToBeDeleted, stmtArgument, targetCtxSubstatement.getPublicDefinition(),
 297                         targetCtxSubstatement.rawStatementArgument())) {
 298                     targetCtxSubstatement.setIsSupportedToBuildEffective(false);
 299                     return;
 300                 }
 301             }
 302
 303             LOG.error("Deviation cannot delete substatement {} with argument '{}' in target node {} because it does " +
 304                     "not exist in the target node. At line: {}", stmtToBeDeleted.getStatementName(), stmtArgument,
 305                     targetCtx.getStatementArgument(), stmtCtxToBeDeleted.getStatementSourceReference());
 306         }
 307
 308         private static boolean statementsAreEqual(final StatementDefinition firstStmtDef, final String firstStmtArg,
 309                 final StatementDefinition secondStmtDef, final String secondStmtArg) {
 310             return firstStmtDef.equals(secondStmtDef) && Objects.equals(firstStmtArg, secondStmtArg);
 311         }
 312
 313         private static void validateDeviationTarget(final StmtContext<?, ?, ?> deviateSubStmtCtx,
 314                 final StmtContext<?, ?, ?> targetCtx) {
 315             InferenceException.throwIf(!isSupportedDeviationTarget(deviateSubStmtCtx, targetCtx,
 316                     targetCtx.getRootVersion()), deviateSubStmtCtx.getStatementSourceReference(),
 317                     "%s is not a valid deviation target for substatement %s.",
 318                     targetCtx.getStatementArgument(), deviateSubStmtCtx.getPublicDefinition().getStatementName());
 319         }
 320
 321         private static boolean isSupportedDeviationTarget(final StmtContext<?, ?, ?> deviateSubstatementCtx,
 322                 final StmtContext<?, ?, ?> deviateTargetCtx, final YangVersion yangVersion) {
 323             Set<StatementDefinition> supportedDeviationTargets =
 324                     YangValidationBundles.SUPPORTED_DEVIATION_TARGETS.get(deviateTargetCtx.getRootVersion(),
 325                             deviateSubstatementCtx.getPublicDefinition());
 326
 327             if (supportedDeviationTargets == null) {
 328                 supportedDeviationTargets = YangValidationBundles.SUPPORTED_DEVIATION_TARGETS.get(YangVersion.VERSION_1,
 329                         deviateSubstatementCtx.getPublicDefinition());
 330             }
 331
 332             // if supportedDeviationTargets is null, it means that the deviate substatement is an unknown statement
 333             return supportedDeviationTargets == null || supportedDeviationTargets.contains(
 334                     deviateTargetCtx.getPublicDefinition());
 335         }
 336
 337         protected SubstatementValidator getSubstatementValidatorForDeviate(final DeviateKind deviateKind) {
 338             switch (deviateKind) {
 339                 case NOT_SUPPORTED:
 340                     return DEVIATE_NOT_SUPPORTED_SUBSTATEMENT_VALIDATOR;
 341                 case ADD:
 342                     return DEVIATE_ADD_SUBSTATEMENT_VALIDATOR;
 343                 case REPLACE:
 344                     return DEVIATE_REPLACE_SUBSTATEMENT_VALIDATOR;
 345                 case DELETE:
 346                     return DEVIATE_DELETE_SUBSTATEMENT_VALIDATOR;
 347                 default:
 348                     throw new IllegalStateException(String.format(
 349                             "Substatement validator for deviate %s has not been defined.", deviateKind));
 350             }
 351         }
 352
 353         @Override
 354         protected SubstatementValidator getSubstatementValidator() {
 355             return null;
 356         }
 357     }
 358
 359     @Nonnull @Override
 360     public DeviateKind getValue() {
 361         return argument();
 362     }
 363 }