Retain DeclarationReference in DeclaredStatements

[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / extension / UnrecognizedStatementSupport.java

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/extension/UnrecognizedStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/extension/UnrecognizedStatementSupport.java
index ad17d614d3e38720a5933495d146351d44842386..57e48075129da1677edfef5c23afa87ec4df1011 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/extension/UnrecognizedStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/extension/UnrecognizedStatementSupport.java
@@ -7,21 +7,36 @@
  */
 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.extension;
 
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.collect.ImmutableList;
 import java.util.Optional;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.model.api.meta.ArgumentDefinition;
+import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
+import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
 import org.opendaylight.yangtools.yang.model.api.stmt.UnrecognizedStatement;
+import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
-import org.opendaylight.yangtools.yang.parser.spi.meta.StatementSupport;
+import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
+import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
+import org.opendaylight.yangtools.yang.parser.spi.meta.OverrideChildStatementSupport;
 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
+import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
 
-final class UnrecognizedStatementSupport extends AbstractStatementSupport<String, UnrecognizedStatement,
-        EffectiveStatement<String, UnrecognizedStatement>> {
-    UnrecognizedStatementSupport(final StatementDefinition publicDefinition) {
-        super(publicDefinition);
+final class UnrecognizedStatementSupport
+        extends AbstractStatementSupport<Object, UnrecognizedStatement, UnrecognizedEffectiveStatement>
+        implements OverrideChildStatementSupport {
+    private final YangParserConfiguration config;
+
+    UnrecognizedStatementSupport(final StatementDefinition publicDefinition, final YangParserConfiguration config) {
+        // We have no idea about the statement's semantics, hence there should be noone interested in its semantics.
+        // Nevertheless it may be of interest for various hacks to understand there was an extension involved.
+        super(publicDefinition, StatementPolicy.exactReplica(), config);
+        this.config = requireNonNull(config);
     }
 
     @Override
@@ -30,36 +45,54 @@ final class UnrecognizedStatementSupport extends AbstractStatementSupport<String
     }
 
     @Override
-    public UnrecognizedStatement createDeclared(final StmtContext<String, UnrecognizedStatement, ?> ctx) {
-        return new UnrecognizedStatementImpl(ctx);
-    }
+    public UnrecognizedStatementSupport statementDefinitionOverrideOf(final StatementDefinition childDef) {
+        /*
+         * This code wraps statements encountered inside an extension so they do not get confused with regular
+         * statements.
+         */
+        final QName baseQName = getStatementName();
+        final QName statementName = QName.create(baseQName, childDef.getStatementName().getLocalName());
 
-    @Override
-    public EffectiveStatement<String, UnrecognizedStatement> createEffective(
-            final StmtContext<String, UnrecognizedStatement,
-            EffectiveStatement<String, UnrecognizedStatement>> ctx) {
-        return new UnrecognizedEffectiveStatementImpl(ctx);
+        final ModelDefinedStatementDefinition def;
+        final Optional<ArgumentDefinition> optArgDef = childDef.getArgumentDefinition();
+        if (optArgDef.isPresent()) {
+            final ArgumentDefinition argDef = optArgDef.get();
+            def = new ModelDefinedStatementDefinition(statementName, argDef.getArgumentName(), argDef.isYinElement());
+        } else {
+            def = new ModelDefinedStatementDefinition(statementName);
+        }
+        return new UnrecognizedStatementSupport(def, config);
     }
 
     @Override
     protected SubstatementValidator getSubstatementValidator() {
+        // We know nothing about this statement
         return null;
     }
 
     @Override
-    public Optional<StatementSupport<?, ?, ?>> getUnknownStatementDefinitionOf(
-            final StatementDefinition yangStmtDef) {
-        final QName baseQName = getStatementName();
-        final QName statementName = QName.create(baseQName, yangStmtDef.getStatementName().getLocalName());
+    protected UnrecognizedStatement createDeclared(final StmtContext<Object, UnrecognizedStatement, ?> ctx,
+            final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+        return new UnrecognizedStatementImpl(ctx.rawArgument(), ctx.publicDefinition(), substatements);
+    }
 
-        final ModelDefinedStatementDefinition def;
-        final Optional<ArgumentDefinition> optArgDef = yangStmtDef.getArgumentDefinition();
-        if (optArgDef.isPresent()) {
-            final ArgumentDefinition argDef = optArgDef.get();
-            def = new ModelDefinedStatementDefinition(statementName, argDef.getArgumentName(), argDef.isYinElement());
-        } else {
-            def = new ModelDefinedStatementDefinition(statementName);
-        }
-        return Optional.of(new ModelDefinedStatementSupport(def));
+    @Override
+    protected UnrecognizedStatement attachDeclarationReference(final UnrecognizedStatement stmt,
+            final DeclarationReference reference) {
+        return new RefUnrecognizedStatement(stmt, reference);
+    }
+
+    // createEffective() should never be called, ensure that for each declared statement
+
+    @Override
+    public void onStatementAdded(final Mutable<Object, UnrecognizedStatement, UnrecognizedEffectiveStatement> stmt) {
+        stmt.setIsSupportedToBuildEffective(false);
+    }
+
+    @Override
+    protected UnrecognizedEffectiveStatement createEffective(final Current<Object, UnrecognizedStatement> stmt,
+            final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+        throw new InferenceException(stmt, "Attempted to instantiate unrecognized effective statement %s",
+            stmt.publicDefinition());
     }
 }
\ No newline at end of file