BUG-7052: introduce YangStatementStreamSource
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / YangStatementSourceImpl.java
index 9bcc6717ed01e3edad2c9388e4ded9492daab15c..3b561918bfccdeb3675035beb7ee8e00a6a73676 100644 (file)
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
+
 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
 
-import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
+import com.google.common.base.Throwables;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URISyntaxException;
-import org.antlr.v4.runtime.ANTLRInputStream;
-import org.antlr.v4.runtime.CommonTokenStream;
 import org.antlr.v4.runtime.tree.ParseTreeWalker;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementLexer;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
+import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser.StatementContext;
 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
+import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
 import org.opendaylight.yangtools.yang.parser.impl.YangStatementParserListenerImpl;
+import org.opendaylight.yangtools.yang.parser.rfc6020.repo.YangStatementStreamSource;
 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.StatementStreamSource;
 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
 import org.opendaylight.yangtools.yang.parser.util.NamedFileInputStream;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
+import org.opendaylight.yangtools.yang.parser.util.NamedInputStream;
 
 /**
+ * This class represents implementation of StatementStreamSource in order to emit YANG statements using supplied
+ * StatementWriter.
  *
- * This class represents implementation of StatementStreamSource
- * in order to emit YANG statements using supplied StatementWriter
- *
+ * @deprecated Use {@link YangStatementStreamSource} instead.
  */
+@Deprecated
 public final class YangStatementSourceImpl implements StatementStreamSource {
+    private final YangStatementParserListenerImpl yangStatementModelParser;
+    private final StatementContext statementContext;
+    private final String sourceName;
 
-    private YangStatementParserListenerImpl yangStatementModelParser;
-    private YangStatementParser.StatementContext statementContext;
-    private ParseTreeWalker walker;
-    private String sourceName;
-//    private String source;
-//    private InputStream sourceStream;
-    private static final Logger LOG = LoggerFactory.getLogger(YangStatementSourceImpl.class);
-
-    public YangStatementSourceImpl(final String fileName, boolean isAbsolute) {
+    public YangStatementSourceImpl(final String fileName, final boolean isAbsolute) {
         try {
-            statementContext = parseYangSource(loadFile(fileName, isAbsolute));
-            walker = new ParseTreeWalker();
+            final NamedFileInputStream is = loadFile(fileName, isAbsolute);
+            sourceName = is.toString();
+            statementContext = YangStatementStreamSource.parseYangSource(is);
             yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
-        } catch (Exception e) {
-            LOG.warn(e.getMessage(), e);
+        } catch (IOException | URISyntaxException | YangSyntaxErrorException e) {
+            throw Throwables.propagate(e);
         }
     }
 
     public YangStatementSourceImpl(final InputStream inputStream) {
         try {
-            statementContext = parseYangSource(inputStream);
-            walker = new ParseTreeWalker();
+            sourceName = inputStream instanceof NamedInputStream ? inputStream.toString() : null;
+            statementContext = YangStatementStreamSource.parseYangSource(inputStream);
             yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
-        } catch (Exception e) {
-            LOG.warn(e.getMessage(), e);
+        } catch (IOException | YangSyntaxErrorException e) {
+            throw Throwables.propagate(e);
         }
     }
 
-    public YangStatementSourceImpl(SourceIdentifier identifier, YangStatementParser.StatementContext statementContext) {
-        try {
-            this.statementContext = statementContext;
-            this.sourceName = identifier.getName();
-            walker = new ParseTreeWalker();
-            yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
-        } catch (Exception e) {
-            LOG.warn(e.getMessage(), e);
-        }
+    public YangStatementSourceImpl(final SourceIdentifier identifier, final StatementContext statementContext) {
+        this.statementContext = statementContext;
+        this.sourceName = identifier.getName();
+        yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
     }
 
     @Override
-    public void writeLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef) throws SourceException {
+    public void writePreLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef) {
         yangStatementModelParser.setAttributes(writer, stmtDef);
-        walker.walk(yangStatementModelParser, statementContext);
+        ParseTreeWalker.DEFAULT.walk(yangStatementModelParser, statementContext);
     }
 
     @Override
-    public void writeLinkageAndStatementDefinitions(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) throws SourceException {
-        yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
-        walker.walk(yangStatementModelParser, statementContext);
+    public void writeLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef,
+            final PrefixToModule preLinkagePrefixes) {
+        yangStatementModelParser.setAttributes(writer, stmtDef, preLinkagePrefixes);
+        ParseTreeWalker.DEFAULT.walk(yangStatementModelParser, statementContext);
     }
 
     @Override
-    public void writeFull(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) throws SourceException {
+    public void writeLinkageAndStatementDefinitions(final StatementWriter writer,
+            final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) {
         yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
-        walker.walk(yangStatementModelParser, statementContext);
+        ParseTreeWalker.DEFAULT.walk(yangStatementModelParser, statementContext);
     }
 
-    private NamedFileInputStream loadFile(final String fileName, boolean isAbsolute) throws URISyntaxException,
-            IOException {
-        //TODO: we need absolute path first!
-        return isAbsolute ? new NamedFileInputStream(new File(fileName), fileName) : new NamedFileInputStream(new File
-                (getClass().getResource(fileName).toURI()), fileName);
-
-//        final File file = new File(fileName);
-//        final ByteSource byteSource = BuilderUtils.fileToByteSource(file);
-//        source = byteSource.asCharSource(Charsets.UTF_8).read();
-//        return isAbsolute ? new NamedFileInputStream(file, fileName) : new NamedFileInputStream(new File
-//                (getClass().getResource(fileName).toURI()), fileName);
+    @Override
+    public void writeFull(final StatementWriter writer, final QNameToStatementDefinition stmtDef,
+            final PrefixToModule prefixes) {
+        yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
+        ParseTreeWalker.DEFAULT.walk(yangStatementModelParser, statementContext);
     }
 
-    private YangStatementParser.StatementContext parseYangSource(final InputStream stream) throws IOException,
-            YangSyntaxErrorException {
-        final YangStatementLexer lexer = new YangStatementLexer(new ANTLRInputStream(stream));
-        final CommonTokenStream tokens = new CommonTokenStream(lexer);
-        final YangStatementParser parser = new YangStatementParser(tokens);
-
-        if(stream instanceof NamedFileInputStream) {
-            sourceName = stream.toString();
-        } else {
-            sourceName = null;
-        }
-
-//        sourceStream = stream;
-//        sourceName = parser.getSourceName();
-//
-//        if (sourceName == null) {
-//            sourceName = stream.toString();
-//        }
-
-        return parser.statement();
+    private NamedFileInputStream loadFile(final String fileName, final boolean isAbsolute)
+            throws URISyntaxException, IOException {
+        //TODO: we need absolute path first!
+        return isAbsolute ? new NamedFileInputStream(new File(fileName), fileName)
+                : new NamedFileInputStream(new File(getClass().getResource(fileName).toURI()), fileName);
     }
 
-    public YangStatementParser.StatementContext getYangAST() {
+    public StatementContext getYangAST() {
         return statementContext;
     }
 
-//    public InputStream getSourceStream() {
-//        return sourceStream;
-//    }
+    @Override
+    public String toString() {
+        return sourceName;
+    }
 }