Migrate coerceStatementArgument() callers

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

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/output/AbstractOutputStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/output/AbstractOutputStatementSupport.java
index 34170bedf88e595c32aeb65411c60335b689f77f..8827681dd35784b745340dead94fec73243bf141 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/output/AbstractOutputStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/output/AbstractOutputStatementSupport.java
@@ -7,41 +7,65 @@
  */
 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.output;
 
+import com.google.common.collect.ImmutableList;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.YangConstants;
 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.meta.StatementSource;
 import org.opendaylight.yangtools.yang.model.api.stmt.OutputEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.OutputStatement;
-import org.opendaylight.yangtools.yang.parser.rfc7950.namespace.ChildSchemaNodeNamespace;
-import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractQNameStatementSupport;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseOperationContainerStatementSupport;
+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;
 
 abstract class AbstractOutputStatementSupport
-        extends AbstractQNameStatementSupport<OutputStatement, OutputEffectiveStatement> {
+        extends BaseOperationContainerStatementSupport<OutputStatement, OutputEffectiveStatement> {
     AbstractOutputStatementSupport() {
-        super(YangStmtMapping.OUTPUT);
+        super(YangStmtMapping.OUTPUT, YangConstants::operationOutputQName);
     }
 
     @Override
-    public final QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
-        return YangConstants.operationOutputQName(StmtContextUtils.getRootModuleQName(ctx));
+    protected final OutputStatement createDeclared(final StmtContext<QName, OutputStatement, ?> ctx,
+            final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+        final StatementSource source = ctx.source();
+        switch (source) {
+            case CONTEXT:
+                return new RegularUndeclaredOutputStatement(ctx.getArgument(), substatements);
+            case DECLARATION:
+                return new RegularOutputStatement(ctx.getArgument(), substatements);
+            default:
+                throw new IllegalStateException("Unhandled statement source " + source);
+        }
     }
 
     @Override
-    public final void onStatementAdded(final Mutable<QName, OutputStatement, OutputEffectiveStatement> stmt) {
-        stmt.coerceParentContext().addToNs(ChildSchemaNodeNamespace.class, stmt.coerceStatementArgument(), stmt);
+    protected final OutputStatement createEmptyDeclared(final StmtContext<QName, OutputStatement, ?> ctx) {
+        final StatementSource source = ctx.source();
+        switch (source) {
+            case CONTEXT:
+                return new EmptyUndeclaredOutputStatement(ctx.getArgument());
+            case DECLARATION:
+                return new EmptyOutputStatement(ctx.getArgument());
+            default:
+                throw new IllegalStateException("Unhandled statement source " + source);
+        }
     }
 
     @Override
-    public final OutputStatement createDeclared(final StmtContext<QName, OutputStatement, ?> ctx) {
-        return new OutputStatementImpl(ctx);
+    protected final OutputEffectiveStatement createDeclaredEffective(final int flags,
+            final Current<QName, OutputStatement> stmt,
+            final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+        return new DeclaredOutputEffectiveStatement(flags, stmt.declared(), substatements, stmt.sourceReference(),
+            stmt.getSchemaPath());
     }
 
     @Override
-    public final OutputEffectiveStatement createEffective(
-            final StmtContext<QName, OutputStatement, OutputEffectiveStatement> ctx) {
-        return new OutputEffectiveStatementImpl(ctx);
+    protected final OutputEffectiveStatement createUndeclaredEffective(final int flags,
+            final Current<QName, OutputStatement> stmt,
+            final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+        return new UndeclaredOutputEffectiveStatement(flags, substatements, stmt.sourceReference(),
+            stmt.getSchemaPath());
     }
-}
\ No newline at end of file
+}