2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
11 import java.io.IOException;
12 import java.io.InputStream;
13 import java.net.URISyntaxException;
14 import org.antlr.v4.runtime.ANTLRInputStream;
15 import org.antlr.v4.runtime.CommonTokenStream;
16 import org.antlr.v4.runtime.tree.ParseTreeWalker;
17 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementLexer;
18 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
19 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser.StatementContext;
20 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
21 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
22 import org.opendaylight.yangtools.yang.parser.impl.YangErrorListener;
23 import org.opendaylight.yangtools.yang.parser.impl.YangStatementParserListenerImpl;
24 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
25 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
26 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
27 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
28 import org.opendaylight.yangtools.yang.parser.util.NamedFileInputStream;
29 import org.slf4j.Logger;
30 import org.slf4j.LoggerFactory;
34 * This class represents implementation of StatementStreamSource
35 * in order to emit YANG statements using supplied StatementWriter
38 public final class YangStatementSourceImpl implements StatementStreamSource {
40 private YangStatementParserListenerImpl yangStatementModelParser;
41 private YangStatementParser.StatementContext statementContext;
42 private ParseTreeWalker walker;
43 private String sourceName;
44 private static final Logger LOG = LoggerFactory.getLogger(YangStatementSourceImpl.class);
46 public YangStatementSourceImpl(final String fileName, final boolean isAbsolute) {
48 statementContext = parseYangSource(loadFile(fileName, isAbsolute));
49 walker = new ParseTreeWalker();
50 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
51 } catch (Exception e) {
56 public YangStatementSourceImpl(final InputStream inputStream) {
58 statementContext = parseYangSource(inputStream);
59 walker = new ParseTreeWalker();
60 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
61 } catch (Exception e) {
66 public YangStatementSourceImpl(final SourceIdentifier identifier, final YangStatementParser.StatementContext statementContext) {
68 this.statementContext = statementContext;
69 this.sourceName = identifier.getName();
70 walker = new ParseTreeWalker();
71 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
72 } catch (Exception e) {
78 public void writePreLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef) {
79 yangStatementModelParser.setAttributes(writer, stmtDef);
80 walker.walk(yangStatementModelParser, statementContext);
84 public void writeLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule preLinkagePrefixes) {
85 yangStatementModelParser.setAttributes(writer, stmtDef, preLinkagePrefixes);
86 walker.walk(yangStatementModelParser, statementContext);
90 public void writeLinkageAndStatementDefinitions(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) {
91 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
92 walker.walk(yangStatementModelParser, statementContext);
96 public void writeFull(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) {
97 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
98 walker.walk(yangStatementModelParser, statementContext);
101 private NamedFileInputStream loadFile(final String fileName, final boolean isAbsolute) throws URISyntaxException,
103 //TODO: we need absolute path first!
104 return isAbsolute ? new NamedFileInputStream(new File(fileName), fileName) : new NamedFileInputStream(new File
105 (getClass().getResource(fileName).toURI()), fileName);
108 private YangStatementParser.StatementContext parseYangSource(final InputStream stream) throws IOException,
109 YangSyntaxErrorException {
110 final YangStatementLexer lexer = new YangStatementLexer(new ANTLRInputStream(stream));
111 final CommonTokenStream tokens = new CommonTokenStream(lexer);
112 final YangStatementParser parser = new YangStatementParser(tokens);
113 //disconnect from console error output
114 parser.removeErrorListeners();
116 final YangErrorListener errorListener = new YangErrorListener();
117 parser.addErrorListener(errorListener);
119 if (stream instanceof NamedFileInputStream) {
120 sourceName = stream.toString();
125 final StatementContext result = parser.statement();
126 errorListener.validate();
131 public YangStatementParser.StatementContext getYangAST() {
132 return statementContext;
136 public String toString() {
140 private static void logError(final Exception e) {
141 if (e instanceof YangSyntaxErrorException) {
142 LOG.error(((YangSyntaxErrorException) e).getFormattedMessage(), e);
144 LOG.error(e.getMessage(), e);