YANGTOOLS-706: Split up base utility classes into rfc6020.util
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / ExtensionStatementImpl.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
9
10 import org.opendaylight.yangtools.yang.common.QName;
11 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
12 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
13 import org.opendaylight.yangtools.yang.model.api.stmt.ArgumentStatement;
14 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionStatement;
15 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionStatement;
16 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.StatusStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.YinElementStatement;
19 import org.opendaylight.yangtools.yang.parser.spi.ExtensionNamespace;
20 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
21 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractQNameStatementSupport;
22 import org.opendaylight.yangtools.yang.parser.spi.meta.StatementDefinitionNamespace;
23 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
27 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.ExtensionEffectiveStatementImpl;
28
29 public class ExtensionStatementImpl extends AbstractDeclaredStatement<QName> implements ExtensionStatement {
30     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
31             .EXTENSION)
32             .addOptional(YangStmtMapping.ARGUMENT)
33             .addOptional(YangStmtMapping.DESCRIPTION)
34             .addOptional(YangStmtMapping.REFERENCE)
35             .addOptional(YangStmtMapping.STATUS)
36             .build();
37
38     protected ExtensionStatementImpl(final StmtContext<QName, ExtensionStatement,?> context) {
39         super(context);
40     }
41
42     public static class Definition
43             extends AbstractQNameStatementSupport<ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> {
44         public Definition() {
45             super(YangStmtMapping.EXTENSION);
46         }
47
48         @Override
49         public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
50             return StmtContextUtils.qnameFromArgument(ctx, value);
51         }
52
53         @Override
54         public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
55             return new ExtensionStatementImpl(ctx);
56         }
57
58         @Override
59         public EffectiveStatement<QName, ExtensionStatement> createEffective(
60                 final StmtContext<QName, ExtensionStatement, EffectiveStatement<QName,ExtensionStatement>> ctx) {
61             return ExtensionEffectiveStatementImpl.create(ctx);
62         }
63
64         @Override
65         public void onStatementDefinitionDeclared(
66                 final Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) {
67             super.onStatementDefinitionDeclared(stmt);
68
69             QName stmtName = stmt.getStatementArgument();
70             if (SupportedExtensionsMapping.OPENCONFIG_VERSION.getStatementName().isEqualWithoutRevision(stmtName)) {
71                 stmtName = stmtName.withoutRevision();
72             }
73
74             stmt.addContext(ExtensionNamespace.class, stmtName, stmt);
75
76             final StmtContext<QName, ?, ?> argument = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
77                 ArgumentStatement.class);
78             final StmtContext<Boolean, ?, ?> yinElement = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
79                 YinElementStatement.class);
80
81             stmt.addToNs(StatementDefinitionNamespace.class, stmt.getStatementArgument(),
82                 new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getStatementArgument(),
83                     argument != null ? argument.getStatementArgument() : null,
84                             yinElement != null ? yinElement.getStatementArgument() : false)));
85         }
86
87         @Override
88         protected SubstatementValidator getSubstatementValidator() {
89             return SUBSTATEMENT_VALIDATOR;
90         }
91     }
92
93     @Override
94     public StatusStatement getStatus() {
95         return firstDeclared(StatusStatement.class);
96     }
97
98     @Override
99     public DescriptionStatement getDescription() {
100         return firstDeclared(DescriptionStatement.class);
101     }
102
103     @Override
104     public ReferenceStatement getReference() {
105         return firstDeclared(ReferenceStatement.class);
106     }
107
108     @Override
109     public ArgumentStatement getArgument() {
110         return firstDeclared(ArgumentStatement.class);
111     }
112 }