BUG-7161: Do not tolerate source-level exceptions
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / impl / YangStatementParserListenerImpl.java
index 95176b1955edb44211e0e7ab8995e2aac654bb65..c9bd2c61a499ad76d73b6fb23967ea197785be46 100644 (file)
@@ -7,19 +7,16 @@
  */
 package org.opendaylight.yangtools.yang.parser.impl;
 
-import com.google.common.base.Preconditions;
-import com.google.common.base.Throwables;
 import com.google.common.base.Verify;
 import java.util.ArrayList;
 import java.util.List;
 import javax.annotation.concurrent.Immutable;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
 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.antlrv4.code.gen.YangStatementParserBaseListener;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.YangConstants;
-import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
 import org.opendaylight.yangtools.yang.parser.spi.source.DeclarationInTextSource;
 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
@@ -27,21 +24,15 @@ import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinit
 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.stmt.rfc6020.TypeUtils;
 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.Utils;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
 
 @Immutable
 public class YangStatementParserListenerImpl extends YangStatementParserBaseListener {
-
-    private StatementWriter writer;
+    private final List<String> toBeSkipped = new ArrayList<>();
     private final String sourceName;
     private QNameToStatementDefinition stmtDef;
     private PrefixToModule prefixes;
-    private final List<String> toBeSkipped = new ArrayList<>();
-    private final boolean isType = false;
-    private static final Logger LOG = LoggerFactory.getLogger(YangStatementParserListenerImpl.class);
+    private StatementWriter writer;
 
     public YangStatementParserListenerImpl(final String sourceName) {
         this.sourceName = sourceName;
@@ -59,63 +50,39 @@ public class YangStatementParserListenerImpl extends YangStatementParserBaseList
     }
 
     @Override
-    public void enterStatement(final YangStatementParser.StatementContext ctx) {
+    public void enterStatement(final StatementContext ctx) {
         final StatementSourceReference ref = DeclarationInTextSource.atPosition(sourceName, ctx
-                .getStart().getLine(), ctx.getStart().getCharPositionInLine());
-        final KeywordContext keywordCtx = Verify.verifyNotNull(ctx.getChild(KeywordContext.class, 0));
-        final ArgumentContext argumentCtx = ctx.getChild(ArgumentContext.class, 0);
+            .getStart().getLine(), ctx.getStart().getCharPositionInLine());
+        final String keywordTxt = Verify.verifyNotNull(ctx.getChild(KeywordContext.class, 0)).getText();
+        final QName identifier = QName.create(YangConstants.RFC6020_YIN_MODULE, keywordTxt);
+        final QName validStatementDefinition = Utils.getValidStatementDefinition(prefixes, stmtDef, identifier);
 
-        String keywordTxt = keywordCtx.getText();
-        try {
-            // FIXME: This is broken for extensions - not all statements are in RFC6020 ns.
-            final QName identifier = QName.create(YangConstants.RFC6020_YIN_MODULE, keywordTxt);
-            if (stmtDef != null && Utils.isValidStatementDefinition(prefixes, stmtDef, identifier)
-                    && toBeSkipped.isEmpty()) {
-                final String argument = argumentCtx != null ? Utils.stringFromStringContext(argumentCtx) : null;
-                // See,s to be fishy specialcase
-                if (identifier.equals(Rfc6020Mapping.TYPE.getStatementName())) {
-                    Preconditions.checkArgument(argument != null);
-                    if (TypeUtils.isYangTypeBodyStmtString(argument)) {
-                        writer.startStatement(QName.create(YangConstants.RFC6020_YIN_MODULE, argument), ref);
-                    } else {
-                        writer.startStatement(QName.create(YangConstants.RFC6020_YIN_MODULE, Rfc6020Mapping
-                                .TYPE.getStatementName().getLocalName()), ref);
-                    }
-                    writer.argumentValue(argument, ref);
-                } else {
-                    writer.startStatement(identifier, ref);
-                    if(argument != null) {
-                        writer.argumentValue(argument, ref);
-                    }
-                }
-            } else if (writer.getPhase().equals(ModelProcessingPhase.FULL_DECLARATION)) {
-                    throw new IllegalArgumentException(identifier.getLocalName() + " is not a YANG statement "
-                            + "or use of extension. Source: " + ref);
-            } else {
-                toBeSkipped.add(keywordTxt);
-            }
-        } catch (SourceException e) {
-            throw Throwables.propagate(e);
+        if (stmtDef == null || validStatementDefinition == null || !toBeSkipped.isEmpty()) {
+            SourceException.throwIf(writer.getPhase() == ModelProcessingPhase.FULL_DECLARATION, ref,
+                    "%s is not a YANG statement or use of extension.", keywordTxt);
+            toBeSkipped.add(keywordTxt);
+            return;
         }
+
+        final ArgumentContext argumentCtx = ctx.getChild(ArgumentContext.class, 0);
+        final String argument = argumentCtx != null ? Utils.stringFromStringContext(argumentCtx) : null;
+        writer.startStatement(validStatementDefinition, argument, ref);
     }
 
     @Override
-    public void exitStatement(final YangStatementParser.StatementContext ctx) {
-        final StatementSourceReference ref = DeclarationInTextSource.atPosition(sourceName, ctx.getStart().getLine(), ctx
-                .getStart().getCharPositionInLine());
-        try {
-            KeywordContext keyword = ctx.getChild(KeywordContext.class, 0);
-            String statementName = keyword.getText();
-            QName identifier = QName.create(YangConstants.RFC6020_YIN_MODULE, statementName);
-            if (stmtDef != null && Utils.isValidStatementDefinition(prefixes, stmtDef, identifier)
-                    && toBeSkipped.isEmpty()) {
-                writer.endStatement(ref);
-            }
+    public void exitStatement(final StatementContext ctx) {
+        final StatementSourceReference ref = DeclarationInTextSource.atPosition(
+            sourceName, ctx.getStart().getLine(), ctx.getStart().getCharPositionInLine());
 
-            // No-op if the statement is not on the list
-            toBeSkipped.remove(statementName);
-        } catch (SourceException e) {
-            LOG.warn(e.getMessage(), e);
+        KeywordContext keyword = ctx.getChild(KeywordContext.class, 0);
+        String statementName = keyword.getText();
+        QName identifier = QName.create(YangConstants.RFC6020_YIN_MODULE, statementName);
+        if (stmtDef != null && Utils.getValidStatementDefinition(prefixes, stmtDef, identifier) != null
+                && toBeSkipped.isEmpty()) {
+            writer.endStatement(ref);
         }
+
+        // No-op if the statement is not on the list
+        toBeSkipped.remove(statementName);
     }
 }