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.impl;
10 import java.util.ArrayList;
11 import java.util.List;
12 import org.slf4j.Logger;
13 import org.slf4j.LoggerFactory;
14 import org.antlr.v4.runtime.tree.ParseTree;
15 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParser;
16 import org.opendaylight.yangtools.antlrv4.code.gen.YangStatementParserBaseListener;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.common.YangConstants;
19 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
20 import org.opendaylight.yangtools.yang.parser.spi.source.QNameToStatementDefinition;
21 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
22 import org.opendaylight.yangtools.yang.parser.spi.source.StatementSourceReference;
23 import org.opendaylight.yangtools.yang.parser.spi.source.StatementWriter;
24 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.Utils;
25 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
27 import javax.annotation.concurrent.Immutable;
30 public class YangStatementParserListenerImpl extends YangStatementParserBaseListener {
32 private StatementWriter writer;
33 private StatementSourceReference ref;
34 private QNameToStatementDefinition stmtDef;
35 private PrefixToModule prefixes;
36 private List<String> toBeSkipped = new ArrayList<>();
37 private boolean isType = false;
38 private static final Logger LOG = LoggerFactory.getLogger(YangStatementParserListenerImpl.class);
40 public YangStatementParserListenerImpl(StatementSourceReference ref) {
44 public void setAttributes(StatementWriter writer, QNameToStatementDefinition stmtDef) {
46 this.stmtDef = stmtDef;
49 public void setAttributes(StatementWriter writer, QNameToStatementDefinition stmtDef, PrefixToModule prefixes) {
51 this.stmtDef = stmtDef;
52 this.prefixes = prefixes;
55 private static boolean hasNotEmptyBody(List<ParseTree> children) {
56 for (ParseTree child : children) {
57 if (child instanceof YangStatementParser.StatementContext) {
65 public void enterStatement(YangStatementParser.StatementContext ctx) {
66 boolean action = true;
68 for (int i = 0; i < ctx.getChildCount(); i++) {
69 ParseTree child = ctx.getChild(i);
70 if (child instanceof YangStatementParser.KeywordContext) {
72 identifier = new QName(YangConstants.RFC6020_YIN_NAMESPACE,
73 ((YangStatementParser.KeywordContext) child).children.get(0).getText());
74 if (stmtDef != null && Utils.isValidStatementDefinition(prefixes, stmtDef, identifier) && toBeSkipped.isEmpty()) {
75 if (identifier.equals(Rfc6020Mapping.TYPE.getStatementName()) && hasNotEmptyBody(((YangStatementParser.KeywordContext) child).getParent().children)) {
78 writer.startStatement(identifier, ref);
82 toBeSkipped.add(((YangStatementParser.KeywordContext) child).children.get(0).getText());
84 } catch (SourceException e) {
85 LOG.warn(e.getMessage(), e);
87 } else if (child instanceof YangStatementParser.ArgumentContext) {
89 final String argument = Utils.stringFromStringContext((YangStatementParser.ArgumentContext) child);
91 writer.startStatement(new QName(YangConstants.RFC6020_YIN_NAMESPACE, argument), ref);
92 writer.argumentValue(argument, ref);
95 writer.argumentValue(Utils.stringFromStringContext((YangStatementParser.ArgumentContext) child), ref);
99 } catch (SourceException e) {
100 LOG.warn(e.getMessage(), e);
108 public void exitStatement(YangStatementParser.StatementContext ctx) {
109 for (int i = 0; i < ctx.getChildCount(); i++) {
110 ParseTree child = ctx.getChild(i);
111 if (child instanceof YangStatementParser.KeywordContext) {
113 String statementName = ((YangStatementParser.KeywordContext) child).children.get(0).getText();
114 QName identifier = new QName(YangConstants.RFC6020_YIN_NAMESPACE, statementName);
115 if (stmtDef != null && Utils.isValidStatementDefinition(prefixes, stmtDef, identifier) && toBeSkipped.isEmpty()) {
116 writer.endStatement(ref);
119 if (toBeSkipped.contains(statementName)) {
120 toBeSkipped.remove(statementName);
122 } catch (SourceException e) {
123 LOG.warn(e.getMessage(), e);