1 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
3 import org.antlr.v4.runtime.ANTLRInputStream;
4 import org.antlr.v4.runtime.CommonTokenStream;
5 import org.antlr.v4.runtime.tree.ParseTreeWalker;
6 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementLexer;
7 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
8 import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
9 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
10 import org.opendaylight.yangtools.yang.parser.impl.YangStatementParserListenerImpl;
11 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
12 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
13 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
14 import org.opendaylight.yangtools.yang.parser.spi.source.StatementSourceReference;
15 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
16 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
19 import java.io.FileInputStream;
20 import java.io.IOException;
21 import java.io.InputStream;
25 * This class represents implementation of StatementStreamSource
26 * in order to emit YANG statements using supplied StatementWriter
30 public class YangStatementSourceImpl implements StatementStreamSource {
32 private YangStatementParserListenerImpl yangStatementModelParser;
33 private YangStatementParser.StatementContext statementContext;
34 private ParseTreeWalker walker;
36 public YangStatementSourceImpl(String fileName) {
38 statementContext = parseYangSource(loadFile(fileName));
39 walker = new ParseTreeWalker();
40 yangStatementModelParser = new YangStatementParserListenerImpl(REF);
41 } catch (Exception e) {
46 private StatementSourceReference REF = new StatementSourceReference() {
49 public StatementSource getStatementSource() {
50 return StatementSource.DECLARATION;
55 public void writeLinkage(StatementWriter writer, QNameToStatementDefinition stmtDef) throws SourceException {
56 yangStatementModelParser.setAttributes(writer, stmtDef);
57 walker.walk(yangStatementModelParser, statementContext);
61 public void writeLinkageAndStatementDefinitions(StatementWriter writer, QNameToStatementDefinition stmtDef, PrefixToModule prefixes) throws SourceException {
62 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
63 walker.walk(yangStatementModelParser, statementContext);
67 public void writeFull(StatementWriter writer, QNameToStatementDefinition stmtDef, PrefixToModule prefixes) throws SourceException {
68 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
69 walker.walk(yangStatementModelParser, statementContext);
72 private FileInputStream loadFile(String fileName) throws Exception {
73 return new FileInputStream(new File(getClass().getResource(fileName).toURI()));
76 private YangStatementParser.StatementContext parseYangSource(final InputStream stream) throws IOException, YangSyntaxErrorException {
77 final YangStatementLexer lexer = new YangStatementLexer(new ANTLRInputStream(stream));
78 final CommonTokenStream tokens = new CommonTokenStream(lexer);
79 final YangStatementParser parser = new YangStatementParser(tokens);
80 final YangStatementParser.StatementContext result = parser.statement();