Integrate {Inference,Source}Exception with CommonStmtCtx

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

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/identity/AbstractIdentityStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/identity/AbstractIdentityStatementSupport.java
index b1037dbd2170b4354a9645b112176c4560008f8f..a042dabb833b1cb64a0d42f8eb1a233cff4cf6a2 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/identity/AbstractIdentityStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/identity/AbstractIdentityStatementSupport.java
@@ -7,18 +7,35 @@
  */
 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.identity;
 
+import static com.google.common.base.Verify.verify;
+import static com.google.common.base.Verify.verifyNotNull;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
+import java.util.ArrayList;
+import java.util.List;
+import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.model.api.IdentitySchemaNode;
+import org.opendaylight.yangtools.yang.model.api.Status;
 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
+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.stmt.BaseEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseQNameStatementSupport;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.EffectiveStatementWithFlags.FlagsBuilder;
 import org.opendaylight.yangtools.yang.parser.spi.IdentityNamespace;
-import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractQNameStatementSupport;
+import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
 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.StmtContextUtils;
+import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
 
 abstract class AbstractIdentityStatementSupport
-        extends AbstractQNameStatementSupport<IdentityStatement, IdentityEffectiveStatement> {
+        extends BaseQNameStatementSupport<IdentityStatement, IdentityEffectiveStatement> {
 
     AbstractIdentityStatementSupport() {
         super(YangStmtMapping.IDENTITY);
@@ -30,19 +47,47 @@ abstract class AbstractIdentityStatementSupport
     }
 
     @Override
-    public final IdentityStatement createDeclared(final StmtContext<QName, IdentityStatement, ?> ctx) {
-        return new IdentityStatementImpl(ctx);
+    public final void onStatementDefinitionDeclared(
+            final Mutable<QName, IdentityStatement, IdentityEffectiveStatement> stmt) {
+        final QName qname = stmt.getArgument();
+        final StmtContext<?, ?, ?> prev = stmt.getFromNamespace(IdentityNamespace.class, qname);
+        SourceException.throwIf(prev != null, stmt, "Duplicate identity definition %s", qname);
+        stmt.addToNs(IdentityNamespace.class, qname, stmt);
     }
 
     @Override
-    public final IdentityEffectiveStatement createEffective(
-            final StmtContext<QName, IdentityStatement, IdentityEffectiveStatement> ctx) {
-        return new IdentityEffectiveStatementImpl(ctx);
+    protected final IdentityStatement createDeclared(final StmtContext<QName, IdentityStatement, ?> ctx,
+            final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+        return new RegularIdentityStatement(ctx.getArgument(), substatements);
     }
 
     @Override
-    public final void onStatementDefinitionDeclared(
-            final Mutable<QName, IdentityStatement, IdentityEffectiveStatement> stmt) {
-        stmt.addToNs(IdentityNamespace.class, stmt.coerceStatementArgument(), stmt);
+    protected final IdentityStatement createEmptyDeclared(@NonNull final StmtContext<QName, IdentityStatement, ?> ctx) {
+        return new EmptyIdentityStatement(ctx.getArgument());
+    }
+
+    @Override
+    protected final IdentityEffectiveStatement createEffective(final Current<QName, IdentityStatement> stmt,
+            final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+        if (substatements.isEmpty()) {
+            return new EmptyIdentityEffectiveStatement(stmt.declared(), stmt.getSchemaPath());
+        }
+
+        final List<IdentitySchemaNode> identities = new ArrayList<>();
+        for (EffectiveStatement<?, ?> substatement : substatements) {
+            if (substatement instanceof BaseEffectiveStatement) {
+                final QName qname = ((BaseEffectiveStatement) substatement).argument();
+                final IdentityEffectiveStatement identity =
+                        verifyNotNull(stmt.getFromNamespace(IdentityNamespace.class, qname),
+                            "Failed to find identity %s", qname)
+                        .buildEffective();
+                verify(identity instanceof IdentitySchemaNode, "%s is not a IdentitySchemaNode", identity);
+                identities.add((IdentitySchemaNode) identity);
+            }
+        }
+
+        return new RegularIdentityEffectiveStatement(stmt.declared(), stmt.getSchemaPath(), new FlagsBuilder()
+            .setStatus(findFirstArgument(substatements, StatusEffectiveStatement.class, Status.CURRENT))
+            .toFlags(), substatements, ImmutableSet.copyOf(identities));
     }
 }