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.FileInputStream;
12 import java.io.FileNotFoundException;
13 import java.io.IOException;
14 import java.io.InputStream;
15 import java.net.URISyntaxException;
16 import org.antlr.v4.runtime.ANTLRInputStream;
17 import org.antlr.v4.runtime.CommonTokenStream;
18 import org.antlr.v4.runtime.tree.ParseTreeWalker;
19 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementLexer;
20 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
21 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
22 import org.opendaylight.yangtools.yang.parser.impl.YangStatementParserListenerImpl;
23 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
24 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
25 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
26 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
27 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
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 static final Logger LOG = LoggerFactory.getLogger(YangStatementSourceImpl.class);
45 public YangStatementSourceImpl(final String fileName, boolean isAbsolute) {
47 statementContext = parseYangSource(loadFile(fileName, isAbsolute));
48 walker = new ParseTreeWalker();
49 yangStatementModelParser = new YangStatementParserListenerImpl(fileName);
50 } catch (Exception e) {
51 LOG.warn(e.getMessage(), e);
55 public YangStatementSourceImpl(final InputStream inputStream) {
57 statementContext = parseYangSource(inputStream);
58 walker = new ParseTreeWalker();
59 yangStatementModelParser = new YangStatementParserListenerImpl(sourceName);
60 } catch (Exception e) {
61 LOG.warn(e.getMessage(), e);
66 public void writeLinkage(final StatementWriter writer, final QNameToStatementDefinition stmtDef) throws SourceException {
67 yangStatementModelParser.setAttributes(writer, stmtDef);
68 walker.walk(yangStatementModelParser, statementContext);
72 public void writeLinkageAndStatementDefinitions(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) throws SourceException {
73 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
74 walker.walk(yangStatementModelParser, statementContext);
78 public void writeFull(final StatementWriter writer, final QNameToStatementDefinition stmtDef, final PrefixToModule prefixes) throws SourceException {
79 yangStatementModelParser.setAttributes(writer, stmtDef, prefixes);
80 walker.walk(yangStatementModelParser, statementContext);
83 private FileInputStream loadFile(final String fileName, boolean isAbsolute) throws URISyntaxException, FileNotFoundException {
84 return isAbsolute ? new FileInputStream(new File(fileName)) : new FileInputStream(new File(getClass().getResource(fileName).toURI()));
87 private YangStatementParser.StatementContext parseYangSource(final InputStream stream) throws IOException, YangSyntaxErrorException {
88 final YangStatementLexer lexer = new YangStatementLexer(new ANTLRInputStream(stream));
89 final CommonTokenStream tokens = new CommonTokenStream(lexer);
90 final YangStatementParser parser = new YangStatementParser(tokens);
91 sourceName = parser.getSourceName();
92 return parser.statement();