BUG-7052: introduce YangStatementStreamSource
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / YangStatementSourceImpl.java
index 2b6c97f8a8c8ed8b40e36ea40b9f464a3c201b30..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.antlr.v4.runtime.ANTLRInputStream;
-import org.antlr.v4.runtime.CommonTokenStream;
+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.tree.ParseTreeWalker;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementLexer;
-import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
-import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
+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.StatementSourceReference;
 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.FileNotFoundException;
-import java.io.IOException;
-import java.io.InputStream;
-import java.net.URISyntaxException;
+import org.opendaylight.yangtools.yang.parser.util.NamedFileInputStream;
+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 static final Logger LOG = LoggerFactory.getLogger(YangStatementSourceImpl.class);
-
-    private static final StatementSourceReference REF = new StatementSourceReference() {
-
-        @Override
-        public StatementSource getStatementSource() {
-            return StatementSource.DECLARATION;
+    public YangStatementSourceImpl(final String fileName, final boolean isAbsolute) {
+        try {
+            final NamedFileInputStream is = loadFile(fileName, isAbsolute);
+            sourceName = is.toString();
+            statementContext = YangStatementStreamSource.parseYangSource(is);
+            yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
+        } catch (IOException | URISyntaxException | YangSyntaxErrorException e) {
+            throw Throwables.propagate(e);
         }
-    };
+    }
 
-    public YangStatementSourceImpl(String fileName) {
+    public YangStatementSourceImpl(final InputStream inputStream) {
         try {
-            statementContext = parseYangSource(loadFile(fileName));
-            walker = new ParseTreeWalker();
-            yangStatementModelParser = new YangStatementParserListenerImpl(REF);
-        } catch (Exception e) {
-            LOG.warn(e.getMessage(), e);
+            sourceName = inputStream instanceof NamedInputStream ? inputStream.toString() : null;
+            statementContext = YangStatementStreamSource.parseYangSource(inputStream);
+            yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
+        } catch (IOException | YangSyntaxErrorException e) {
+            throw Throwables.propagate(e);
         }
     }
 
+    public YangStatementSourceImpl(final SourceIdentifier identifier, final StatementContext statementContext) {
+        this.statementContext = statementContext;
+        this.sourceName = identifier.getName();
+        yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
+    }
+
     @Override
-    public void writeLinkage(StatementWriter writer, 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(StatementWriter writer, QNameToStatementDefinition stmtDef, PrefixToModule prefixes) throws SourceException {
+    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 writeLinkageAndStatementDefinitions(final StatementWriter writer,
+            final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) {
         yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
-        walker.walk(yangStatementModelParser, statementContext);
+        ParseTreeWalker.DEFAULT.walk(yangStatementModelParser, statementContext);
     }
 
     @Override
-    public void writeFull(StatementWriter writer, QNameToStatementDefinition stmtDef, PrefixToModule prefixes) throws SourceException {
+    public void writeFull(final StatementWriter writer, final QNameToStatementDefinition stmtDef,
+            final PrefixToModule prefixes) {
         yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
-        walker.walk(yangStatementModelParser, statementContext);
+        ParseTreeWalker.DEFAULT.walk(yangStatementModelParser, statementContext);
     }
 
-    private FileInputStream loadFile(String fileName) throws URISyntaxException, FileNotFoundException {
-        return new FileInputStream(new File(getClass().getResource(fileName).toURI()));
+    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);
     }
 
-    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);
-        return parser.statement();
+    public StatementContext getYangAST() {
+        return statementContext;
+    }
+
+    @Override
+    public String toString() {
+        return sourceName;
     }
 }