BUG-6965: introduce BuiltinDeclaredStatements
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / BuiltinTypeStatement.java
1 /*
2  * Copyright (c) 2016 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.stmt.rfc6020;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.collect.ImmutableList;
12 import com.google.common.collect.ImmutableMap;
13 import com.google.common.collect.ImmutableMap.Builder;
14 import java.util.Collection;
15 import java.util.Map;
16 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
17 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
18 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
19 import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
20 import org.opendaylight.yangtools.yang.model.api.stmt.TypeStatement;
21
22 final class BuiltinTypeStatement implements TypeStatement {
23     private static final Map<String, BuiltinTypeStatement> BUILTINS;
24     static {
25         final Builder<String, BuiltinTypeStatement> builder = ImmutableMap.builder();
26         putBuiltin(builder, TypeUtils.BINARY);
27         putBuiltin(builder, TypeUtils.BOOLEAN);
28         putBuiltin(builder, TypeUtils.EMPTY);
29         putBuiltin(builder, TypeUtils.INSTANCE_IDENTIFIER);
30         putBuiltin(builder, TypeUtils.INT8);
31         putBuiltin(builder, TypeUtils.INT16);
32         putBuiltin(builder, TypeUtils.INT32);
33         putBuiltin(builder, TypeUtils.INT64);
34         putBuiltin(builder, TypeUtils.STRING);
35         putBuiltin(builder, TypeUtils.UINT8);
36         putBuiltin(builder, TypeUtils.UINT16);
37         putBuiltin(builder, TypeUtils.UINT32);
38         putBuiltin(builder, TypeUtils.UINT64);
39         BUILTINS = builder.build();
40     }
41
42     private static void putBuiltin(final Builder<String, BuiltinTypeStatement> builder, final String argument) {
43         builder.put(argument, new BuiltinTypeStatement(argument));
44     }
45
46     private final String argument;
47
48     private BuiltinTypeStatement(final String argument) {
49         this.argument = Preconditions.checkNotNull(argument);
50     }
51
52     static TypeStatement maybeReplace(final TypeStatementImpl orig) {
53         if (orig.declaredSubstatements().isEmpty() && orig.getStatementSource() == StatementSource.DECLARATION &&
54                 orig.statementDefinition() == Rfc6020Mapping.TYPE) {
55             final BuiltinTypeStatement builtin = BUILTINS.get(orig.argument());
56             if (builtin != null) {
57                 return builtin;
58             }
59         }
60
61         return orig;
62     }
63
64     @Override
65     public String argument() {
66         return argument;
67     }
68
69     @Override
70     public String rawArgument() {
71         return argument;
72     }
73
74     @Override
75     public Collection<? extends DeclaredStatement<?>> declaredSubstatements() {
76         return ImmutableList.of();
77     }
78
79     @Override
80     public StatementDefinition statementDefinition() {
81         return Rfc6020Mapping.TYPE;
82     }
83
84     @Override
85     public String getName() {
86         return argument;
87     }
88
89     @Override
90     public StatementSource getStatementSource() {
91         return StatementSource.DECLARATION;
92     }
93 }