BUG-6497: Do not lose augmentation statement order
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / SubmoduleStatementImpl.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 static org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator.MAX;
11 import static org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils.findFirstDeclaredSubstatement;
12 import static org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils.firstAttributeOf;
13 import com.google.common.base.Optional;
14 import java.util.Date;
15 import org.opendaylight.yangtools.yang.common.SimpleDateFormatUtil;
16 import org.opendaylight.yangtools.yang.model.api.ModuleIdentifier;
17 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
18 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.BelongsToStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.PrefixStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.SubmoduleStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.YangVersionStatement;
23 import org.opendaylight.yangtools.yang.parser.builder.impl.ModuleIdentifierImpl;
24 import org.opendaylight.yangtools.yang.parser.spi.SubmoduleNamespace;
25 import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
29 import org.opendaylight.yangtools.yang.parser.spi.source.BelongsToPrefixToModuleName;
30 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
31 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.SubmoduleEffectiveStatementImpl;
32
33 public class SubmoduleStatementImpl extends AbstractRootStatement<SubmoduleStatement> implements SubmoduleStatement {
34     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(Rfc6020Mapping
35             .SUBMODULE)
36             .add(Rfc6020Mapping.ANYXML, 0, MAX)
37             .add(Rfc6020Mapping.AUGMENT, 0, MAX)
38             .add(Rfc6020Mapping.BELONGS_TO, 1, 1)
39             .add(Rfc6020Mapping.CHOICE, 0, MAX)
40             .add(Rfc6020Mapping.CONTACT, 0, 1)
41             .add(Rfc6020Mapping.CONTAINER, 0, MAX)
42             .add(Rfc6020Mapping.DESCRIPTION, 0, 1)
43             .add(Rfc6020Mapping.DEVIATION, 0, MAX)
44             .add(Rfc6020Mapping.EXTENSION, 0, MAX)
45             .add(Rfc6020Mapping.FEATURE, 0, MAX)
46             .add(Rfc6020Mapping.GROUPING, 0, MAX)
47             .add(Rfc6020Mapping.IDENTITY, 0, MAX)
48             .add(Rfc6020Mapping.IMPORT, 0, MAX)
49             .add(Rfc6020Mapping.INCLUDE, 0, MAX)
50             .add(Rfc6020Mapping.LEAF, 0, MAX)
51             .add(Rfc6020Mapping.LEAF_LIST, 0, MAX)
52             .add(Rfc6020Mapping.LIST, 0, MAX)
53             .add(Rfc6020Mapping.NOTIFICATION, 0, MAX)
54             .add(Rfc6020Mapping.ORGANIZATION, 0, 1)
55             .add(Rfc6020Mapping.REFERENCE, 0, 1)
56             .add(Rfc6020Mapping.REVISION, 0, MAX)
57             .add(Rfc6020Mapping.RPC, 0, MAX)
58             .add(Rfc6020Mapping.TYPEDEF, 0, MAX)
59             .add(Rfc6020Mapping.USES, 0, MAX)
60             .add(Rfc6020Mapping.YANG_VERSION, 0, 1)
61             .build();
62     private static final Optional<Date> DEFAULT_REVISION = Optional.of(SimpleDateFormatUtil.DEFAULT_DATE_REV);
63
64     protected SubmoduleStatementImpl(final StmtContext<String, SubmoduleStatement, ?> context) {
65         super(context);
66     }
67
68     public static class Definition extends AbstractStatementSupport<String, SubmoduleStatement,
69             EffectiveStatement<String, SubmoduleStatement>> {
70
71         public Definition() {
72             super(Rfc6020Mapping.SUBMODULE);
73         }
74
75         @Override
76         public String parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
77             return value;
78         }
79
80         @Override
81         public SubmoduleStatement createDeclared(
82                 final StmtContext<String, SubmoduleStatement, ?> ctx) {
83             return new SubmoduleStatementImpl(ctx);
84         }
85
86         @Override
87         public EffectiveStatement<String, SubmoduleStatement> createEffective(
88                 final StmtContext<String, SubmoduleStatement, EffectiveStatement<String, SubmoduleStatement>> ctx) {
89             return new SubmoduleEffectiveStatementImpl(ctx);
90         }
91
92         @Override
93         public void onLinkageDeclared(
94                 final Mutable<String, SubmoduleStatement, EffectiveStatement<String, SubmoduleStatement>> stmt) {
95
96             final Optional<Date> revisionDate = Optional.fromNullable(
97                 Utils.getLatestRevision(stmt.declaredSubstatements())).or(DEFAULT_REVISION);
98
99             ModuleIdentifier submoduleIdentifier = new ModuleIdentifierImpl(stmt.getStatementArgument(),
100                 Optional.absent(), revisionDate);
101
102             stmt.addContext(SubmoduleNamespace.class, submoduleIdentifier, stmt);
103
104             String belongsToModuleName = firstAttributeOf(
105                     stmt.declaredSubstatements(), BelongsToStatement.class);
106             StmtContext<?, ?, ?> prefixSubStmtCtx = findFirstDeclaredSubstatement(
107                     stmt, 0, BelongsToStatement.class, PrefixStatement.class);
108             SourceException.throwIfNull(prefixSubStmtCtx, stmt.getStatementSourceReference(),
109                 "Prefix of belongsTo statement is missing in submodule [%s]", stmt.getStatementArgument());
110
111             String prefix = (String) prefixSubStmtCtx.getStatementArgument();
112
113             stmt.addToNs(BelongsToPrefixToModuleName.class, prefix, belongsToModuleName);
114         }
115
116         @Override
117         public void onFullDefinitionDeclared(final Mutable<String, SubmoduleStatement,
118                 EffectiveStatement<String, SubmoduleStatement>> stmt) {
119             super.onFullDefinitionDeclared(stmt);
120             SUBSTATEMENT_VALIDATOR.validate(stmt);
121         }
122     }
123
124     @Override
125     public String getName() {
126         return rawArgument();
127     }
128
129     @Override
130     public YangVersionStatement getYangVersion() {
131         return firstDeclared(YangVersionStatement.class);
132     }
133
134     @Override
135     public BelongsToStatement getBelongsTo() {
136         return firstDeclared(BelongsToStatement.class);
137     }
138
139 }