Convert trivial CopyPolicy users to StatementPolicy
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / identity / IdentityStatementSupport.java
1 /*
2  * Copyright (c) 2017 Pantheon Technologies, s.r.o. 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.rfc7950.stmt.identity;
9
10 import static com.google.common.base.Verify.verify;
11 import static com.google.common.base.Verify.verifyNotNull;
12 import static java.util.Objects.requireNonNull;
13
14 import com.google.common.annotations.Beta;
15 import com.google.common.collect.ImmutableList;
16 import com.google.common.collect.ImmutableSet;
17 import java.util.ArrayList;
18 import java.util.List;
19 import org.eclipse.jdt.annotation.NonNull;
20 import org.opendaylight.yangtools.yang.common.QName;
21 import org.opendaylight.yangtools.yang.model.api.IdentitySchemaNode;
22 import org.opendaylight.yangtools.yang.model.api.Status;
23 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
24 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
25 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.BaseEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityEffectiveStatement;
28 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityStatement;
29 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
30 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseQNameStatementSupport;
31 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.EffectiveStatementWithFlags.FlagsBuilder;
32 import org.opendaylight.yangtools.yang.parser.spi.IdentityNamespace;
33 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
35 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
36 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
37 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
38 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
39
40 @Beta
41 public final class IdentityStatementSupport
42         extends BaseQNameStatementSupport<IdentityStatement, IdentityEffectiveStatement> {
43     private static final @NonNull IdentityStatementSupport RFC6020_INSTANCE = new IdentityStatementSupport(
44         SubstatementValidator.builder(YangStmtMapping.IDENTITY)
45             .addOptional(YangStmtMapping.BASE)
46             .addOptional(YangStmtMapping.DESCRIPTION)
47             .addOptional(YangStmtMapping.REFERENCE)
48             .addOptional(YangStmtMapping.STATUS)
49             .build());
50     private static final @NonNull IdentityStatementSupport RFC7950_INSTANCE = new IdentityStatementSupport(
51         SubstatementValidator.builder(YangStmtMapping.IDENTITY)
52             .addAny(YangStmtMapping.BASE)
53             .addOptional(YangStmtMapping.DESCRIPTION)
54             .addAny(YangStmtMapping.IF_FEATURE)
55             .addOptional(YangStmtMapping.REFERENCE)
56             .addOptional(YangStmtMapping.STATUS)
57             .build());
58
59     private final SubstatementValidator validator;
60
61     private IdentityStatementSupport(final SubstatementValidator validator) {
62         super(YangStmtMapping.IDENTITY, StatementPolicy.reject());
63         this.validator = requireNonNull(validator);
64     }
65
66     public static @NonNull IdentityStatementSupport rfc6020Instance() {
67         return RFC6020_INSTANCE;
68     }
69
70     public static @NonNull IdentityStatementSupport rfc7950Instance() {
71         return RFC7950_INSTANCE;
72     }
73
74     @Override
75     public QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
76         return StmtContextUtils.parseIdentifier(ctx, value);
77     }
78
79     @Override
80     public void onStatementDefinitionDeclared(
81             final Mutable<QName, IdentityStatement, IdentityEffectiveStatement> stmt) {
82         final QName qname = stmt.getArgument();
83         final StmtContext<?, ?, ?> prev = stmt.getFromNamespace(IdentityNamespace.class, qname);
84         SourceException.throwIf(prev != null, stmt, "Duplicate identity definition %s", qname);
85         stmt.addToNs(IdentityNamespace.class, qname, stmt);
86     }
87
88     @Override
89     protected SubstatementValidator getSubstatementValidator() {
90         return validator;
91     }
92
93     @Override
94     protected IdentityStatement createDeclared(final StmtContext<QName, IdentityStatement, ?> ctx,
95             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
96         return new RegularIdentityStatement(ctx.getArgument(), substatements);
97     }
98
99     @Override
100     protected IdentityStatement createEmptyDeclared(final StmtContext<QName, IdentityStatement, ?> ctx) {
101         return new EmptyIdentityStatement(ctx.getArgument());
102     }
103
104     @Override
105     protected IdentityEffectiveStatement createEffective(final Current<QName, IdentityStatement> stmt,
106             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
107         if (substatements.isEmpty()) {
108             return new EmptyIdentityEffectiveStatement(stmt.declared(), stmt.wrapSchemaPath());
109         }
110
111         final List<IdentitySchemaNode> identities = new ArrayList<>();
112         for (EffectiveStatement<?, ?> substatement : substatements) {
113             if (substatement instanceof BaseEffectiveStatement) {
114                 final QName qname = ((BaseEffectiveStatement) substatement).argument();
115                 final IdentityEffectiveStatement identity =
116                         verifyNotNull(stmt.getFromNamespace(IdentityNamespace.class, qname),
117                             "Failed to find identity %s", qname)
118                         .buildEffective();
119                 verify(identity instanceof IdentitySchemaNode, "%s is not a IdentitySchemaNode", identity);
120                 identities.add((IdentitySchemaNode) identity);
121             }
122         }
123
124         return new RegularIdentityEffectiveStatement(stmt.declared(), stmt.wrapSchemaPath(), new FlagsBuilder()
125             .setStatus(findFirstArgument(substatements, StatusEffectiveStatement.class, Status.CURRENT))
126             .toFlags(), substatements, ImmutableSet.copyOf(identities));
127     }
128 }