Split out RFC7950 ANTLR grammars

[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / repo / StatementContextVisitor.java

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/repo/StatementContextVisitor.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/repo/StatementContextVisitor.java
index aaed74b1c5c1e206225566c9d1eddb947360ab53..faefa05cc12a071621c3648a2159218fe7f4786c 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/repo/StatementContextVisitor.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/repo/StatementContextVisitor.java
@@ -10,21 +10,23 @@ package org.opendaylight.yangtools.yang.parser.rfc7950.repo;
 import static com.google.common.base.Verify.verifyNotNull;
 import static java.util.Objects.requireNonNull;
 
+import java.util.Optional;
 import org.antlr.v4.runtime.tree.ParseTree;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser.ArgumentContext;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser.KeywordContext;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser.StatementContext;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.QNameModule;
 import org.opendaylight.yangtools.yang.common.YangConstants;
 import org.opendaylight.yangtools.yang.common.YangVersion;
 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
+import org.opendaylight.yangtools.yang.parser.antlr.YangStatementParser.ArgumentContext;
+import org.opendaylight.yangtools.yang.parser.antlr.YangStatementParser.KeywordContext;
+import org.opendaylight.yangtools.yang.parser.antlr.YangStatementParser.StatementContext;
 import org.opendaylight.yangtools.yang.parser.spi.source.DeclarationInTextSource;
 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
 import org.opendaylight.yangtools.yang.parser.spi.source.StatementSourceReference;
 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
+import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter.ResumedStatement;
 
 class StatementContextVisitor {
     private final QNameToStatementDefinition stmtDef;
@@ -60,7 +62,7 @@ class StatementContextVisitor {
     QName getValidStatementDefinition(final String keywordText, final StatementSourceReference ref) {
         final int firstColon = keywordText.indexOf(':');
         if (firstColon == -1) {
-            final StatementDefinition def = stmtDef.get(QName.create(YangConstants.RFC6020_YIN_NAMESPACE, keywordText));
+            final StatementDefinition def = stmtDef.get(QName.create(YangConstants.RFC6020_YIN_MODULE, keywordText));
             return def != null ? def.getStatementName() : null;
         }
 
@@ -88,29 +90,45 @@ class StatementContextVisitor {
         return stmtDef.get(QName.create(module, localName));
     }
 
-    private void processStatement(final int myOffset, final StatementContext ctx) {
-        final String keywordTxt = verifyNotNull(ctx.getChild(KeywordContext.class, 0)).getText();
-        final StatementSourceReference ref = DeclarationInTextSource.atPosition(sourceName, ctx.getStart().getLine(),
-            ctx.getStart().getCharPositionInLine());
-        final QName def = getValidStatementDefinition(keywordTxt, ref);
-        if (def == null) {
-            return;
-        }
+    private boolean processStatement(final int myOffset, final StatementContext ctx) {
+        final Optional<? extends ResumedStatement> optResumed = writer.resumeStatement(myOffset);
+        final StatementSourceReference ref;
+        if (optResumed.isPresent()) {
+            final ResumedStatement resumed = optResumed.get();
+            if (resumed.isFullyDefined()) {
+                return true;
+            }
 
-        final ArgumentContext argumentCtx = ctx.getChild(ArgumentContext.class, 0);
-        final String argument = argumentCtx == null ? null
-                : ArgumentContextUtils.stringFromStringContext(argumentCtx, yangVersion, ref);
-        writer.startStatement(myOffset, def, argument, ref);
+            ref = resumed.getSourceReference();
+        } else {
+            final String keywordTxt = verifyNotNull(ctx.getChild(KeywordContext.class, 0)).getText();
+            ref = DeclarationInTextSource.atPosition(sourceName, ctx.getStart().getLine(),
+                ctx.getStart().getCharPositionInLine());
+            final QName def = getValidStatementDefinition(keywordTxt, ref);
+            if (def == null) {
+                return false;
+            }
+
+            final ArgumentContext argumentCtx = ctx.getChild(ArgumentContext.class, 0);
+            final String argument = argumentCtx == null ? null
+                    : ArgumentContextUtils.stringFromStringContext(argumentCtx, yangVersion, ref);
+            writer.startStatement(myOffset, def, argument, ref);
+        }
 
+        int childOffset = 0;
+        boolean fullyDefined = true;
         if (ctx.children != null) {
-            int childOffset = 0;
             for (ParseTree s : ctx.children) {
                 if (s instanceof StatementContext) {
-                    processStatement(childOffset++, (StatementContext) s);
+                    if (!processStatement(childOffset++, (StatementContext) s)) {
+                        fullyDefined = false;
+                    }
                 }
             }
         }
 
+        writer.storeStatement(childOffset, fullyDefined);
         writer.endStatement(ref);
+        return fullyDefined;
     }
 }