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;
10 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
12 import java.io.IOException;
13 import java.io.InputStream;
14 import java.net.URISyntaxException;
15 import org.antlr.v4.runtime.ANTLRInputStream;
16 import org.antlr.v4.runtime.CommonTokenStream;
17 import org.antlr.v4.runtime.tree.ParseTreeWalker;
18 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementLexer;
19 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
20 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
21 import org.opendaylight.yangtools.yang.parser.impl.YangStatementParserListenerImpl;
22 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
23 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
24 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
25 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
26 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
27 import org.opendaylight.yangtools.yang.parser.util.NamedFileInputStream;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
33 * This class represents implementation of StatementStreamSource
34 * in order to emit YANG statements using supplied StatementWriter
37 public final class YangStatementSourceImpl implements StatementStreamSource {
39 private YangStatementParserListenerImpl yangStatementModelParser;
40 private YangStatementParser.StatementContext statementContext;
41 private ParseTreeWalker walker;
42 private String sourceName;
43 // private String source;
44 // private InputStream sourceStream;
45 private static final Logger LOG = LoggerFactory.getLogger(YangStatementSourceImpl.class);
47 public YangStatementSourceImpl(final String fileName, boolean isAbsolute) {
49 statementContext = parseYangSource(loadFile(fileName, isAbsolute));
50 walker = new ParseTreeWalker();
51 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
52 } catch (Exception e) {
53 LOG.warn(e.getMessage(), e);
57 public YangStatementSourceImpl(final InputStream inputStream) {
59 statementContext = parseYangSource(inputStream);
60 walker = new ParseTreeWalker();
61 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
62 } catch (Exception e) {
63 LOG.warn(e.getMessage(), e);
67 public YangStatementSourceImpl(SourceIdentifier identifier, YangStatementParser.StatementContext statementContext) {
69 this.statementContext = statementContext;
70 this.sourceName = identifier.getName();
71 walker = new ParseTreeWalker();
72 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
73 } catch (Exception e) {
74 LOG.warn(e.getMessage(), e);
79 public void writeLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef) throws SourceException {
80 yangStatementModelParser.setAttributes(writer, stmtDef);
81 walker.walk(yangStatementModelParser, statementContext);
85 public void writeLinkageAndStatementDefinitions(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) throws SourceException {
86 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
87 walker.walk(yangStatementModelParser, statementContext);
91 public void writeFull(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) throws SourceException {
92 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
93 walker.walk(yangStatementModelParser, statementContext);
96 private NamedFileInputStream loadFile(final String fileName, boolean isAbsolute) throws URISyntaxException,
98 //TODO: we need absolute path first!
99 return isAbsolute ? new NamedFileInputStream(new File(fileName), fileName) : new NamedFileInputStream(new File
100 (getClass().getResource(fileName).toURI()), fileName);
102 // final File file = new File(fileName);
103 // final ByteSource byteSource = BuilderUtils.fileToByteSource(file);
104 // source = byteSource.asCharSource(Charsets.UTF_8).read();
105 // return isAbsolute ? new NamedFileInputStream(file, fileName) : new NamedFileInputStream(new File
106 // (getClass().getResource(fileName).toURI()), fileName);
109 private YangStatementParser.StatementContext parseYangSource(final InputStream stream) throws IOException,
110 YangSyntaxErrorException {
111 final YangStatementLexer lexer = new YangStatementLexer(new ANTLRInputStream(stream));
112 final CommonTokenStream tokens = new CommonTokenStream(lexer);
113 final YangStatementParser parser = new YangStatementParser(tokens);
115 if(stream instanceof NamedFileInputStream) {
116 sourceName = stream.toString();
121 // sourceStream = stream;
122 // sourceName = parser.getSourceName();
124 // if (sourceName == null) {
125 // sourceName = stream.toString();
128 return parser.statement();
131 public YangStatementParser.StatementContext getYangAST() {
132 return statementContext;
135 // public InputStream getSourceStream() {
136 // return sourceStream;