407bfa361b5c59f0d4014eab25390e969176f948
[yangtools.git] / model / yang-model-ri / src / main / java / org / opendaylight / yangtools / yang / model / ri / stmt / impl / eff / AbstractLeafEffectiveStatement.java
1 /*
2  * Copyright (c) 2020 PANTHEON.tech, 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.model.ri.stmt.impl.eff;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.collect.ImmutableList;
13 import org.eclipse.jdt.annotation.NonNull;
14 import org.opendaylight.yangtools.yang.common.QName;
15 import org.opendaylight.yangtools.yang.model.api.DerivableSchemaNode;
16 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
17 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
18 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.DefaultEffectiveStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionEffectiveStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.LeafEffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.LeafStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.UnitsEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.ri.type.ConcreteTypeBuilder;
28 import org.opendaylight.yangtools.yang.model.ri.type.ConcreteTypes;
29 import org.opendaylight.yangtools.yang.model.spi.meta.AbstractDeclaredEffectiveStatement;
30 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.DataSchemaNodeMixin;
31 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MandatoryMixin;
32 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MustConstraintMixin;
33
34 public abstract class AbstractLeafEffectiveStatement
35         extends AbstractDeclaredEffectiveStatement.Default<QName, LeafStatement>
36         implements LeafEffectiveStatement, LeafSchemaNode, DerivableSchemaNode,
37             DataSchemaNodeMixin<LeafStatement>, MandatoryMixin<QName, LeafStatement>,
38             MustConstraintMixin<QName, LeafStatement> {
39     private final @NonNull Object substatements;
40     private final @NonNull QName argument;
41     private final @NonNull TypeDefinition<?> type;
42     private final int flags;
43
44     AbstractLeafEffectiveStatement(final LeafStatement declared, final QName argument, final int flags,
45             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
46         super(declared);
47         this.argument = requireNonNull(argument);
48         this.substatements = maskList(substatements);
49         this.flags = flags;
50         // TODO: lazy instantiation?
51         this.type = buildType();
52     }
53
54     AbstractLeafEffectiveStatement(final AbstractLeafEffectiveStatement original, final QName argument,
55             final int flags) {
56         super(original);
57         this.argument = requireNonNull(argument);
58         this.substatements = original.substatements;
59         this.flags = flags;
60         // FIXME: share with original?
61         this.type = buildType();
62     }
63
64     @Override
65     public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
66         return unmaskList(substatements);
67     }
68
69     @Override
70     public final int flags() {
71         return flags;
72     }
73
74     @Override
75     public final QName argument() {
76         return argument;
77     }
78
79     @Override
80     public final TypeDefinition<?> getType() {
81         return type;
82     }
83
84     @Override
85     public final LeafEffectiveStatement asEffectiveStatement() {
86         return this;
87     }
88
89     private TypeDefinition<?> buildType() {
90         final TypeEffectiveStatement<?> typeStmt = findFirstEffectiveSubstatement(TypeEffectiveStatement.class).get();
91         final ConcreteTypeBuilder<?> builder = ConcreteTypes.concreteTypeBuilder(typeStmt.getTypeDefinition(),
92             getQName());
93         for (final EffectiveStatement<?, ?> stmt : effectiveSubstatements()) {
94             if (stmt instanceof DefaultEffectiveStatement) {
95                 builder.setDefaultValue(((DefaultEffectiveStatement)stmt).argument());
96             } else if (stmt instanceof DescriptionEffectiveStatement) {
97                 builder.setDescription(((DescriptionEffectiveStatement)stmt).argument());
98             } else if (stmt instanceof ReferenceEffectiveStatement) {
99                 builder.setReference(((ReferenceEffectiveStatement)stmt).argument());
100             } else if (stmt instanceof StatusEffectiveStatement) {
101                 builder.setStatus(((StatusEffectiveStatement)stmt).argument());
102             } else if (stmt instanceof UnitsEffectiveStatement) {
103                 builder.setUnits(((UnitsEffectiveStatement)stmt).argument());
104             }
105         }
106         return builder.build();
107     }
108 }