Refactor argument adaptation
[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.StmtContextUtils;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
26 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.ExtensionEffectiveStatementImpl;
27
28 public class ExtensionStatementImpl extends AbstractDeclaredStatement<QName> implements ExtensionStatement {
29     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
30             .EXTENSION)
31             .addOptional(YangStmtMapping.ARGUMENT)
32             .addOptional(YangStmtMapping.DESCRIPTION)
33             .addOptional(YangStmtMapping.REFERENCE)
34             .addOptional(YangStmtMapping.STATUS)
35             .build();
36
37     protected ExtensionStatementImpl(final StmtContext<QName, ExtensionStatement,?> context) {
38         super(context);
39     }
40
41     public static class Definition
42             extends AbstractQNameStatementSupport<ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> {
43         public Definition() {
44             super(YangStmtMapping.EXTENSION);
45         }
46
47         @Override
48         public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
49             return StmtContextUtils.qnameFromArgument(ctx, value);
50         }
51
52         @Override
53         public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
54             return new ExtensionStatementImpl(ctx);
55         }
56
57         @Override
58         public EffectiveStatement<QName,ExtensionStatement> createEffective(
59                 final StmtContext<QName,ExtensionStatement ,EffectiveStatement<QName,ExtensionStatement>> ctx) {
60             return ExtensionEffectiveStatementImpl.create(ctx);
61         }
62
63         @Override
64         public void onStatementDefinitionDeclared(final StmtContext.Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) {
65             super.onStatementDefinitionDeclared(stmt);
66
67             QName stmtName = stmt.getStatementArgument();
68             if (SupportedExtensionsMapping.OPENCONFIG_VERSION.getStatementName().isEqualWithoutRevision(stmtName)) {
69                 stmtName = stmtName.withoutRevision();
70             }
71
72             stmt.addContext(ExtensionNamespace.class, stmtName, stmt);
73
74             final StmtContext<QName, ?, ?> argument = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
75                 ArgumentStatement.class);
76             final StmtContext<Boolean, ?, ?> yinElement = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
77                 YinElementStatement.class);
78
79             stmt.addToNs(StatementDefinitionNamespace.class, stmt.getStatementArgument(),
80                 new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getStatementArgument(),
81                     argument != null ? argument.getStatementArgument() : null,
82                             yinElement != null ? yinElement.getStatementArgument() : false)));
83         }
84
85         @Override
86         protected SubstatementValidator getSubstatementValidator() {
87             return SUBSTATEMENT_VALIDATOR;
88         }
89     }
90
91     @Override
92     public StatusStatement getStatus() {
93         return firstDeclared(StatusStatement.class);
94     }
95
96     @Override
97     public DescriptionStatement getDescription() {
98         return firstDeclared(DescriptionStatement.class);
99     }
100
101     @Override
102     public ReferenceStatement getReference() {
103         return firstDeclared(ReferenceStatement.class);
104     }
105
106     @Override
107     public ArgumentStatement getArgument() {
108         return firstDeclared(ArgumentStatement.class);
109     }
110
111 }