Bug 3670 (part 3/5): Use of new statement parser in yang-maven-plugin
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / ModuleStatementSupport.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.meta.StmtContextUtils.firstAttributeOf;
11
12 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleCtxToModuleQName;
13
14 import com.google.common.base.Optional;
15 import java.net.URI;
16 import java.util.Date;
17 import org.opendaylight.yangtools.yang.common.QNameModule;
18 import org.opendaylight.yangtools.yang.common.SimpleDateFormatUtil;
19 import org.opendaylight.yangtools.yang.model.api.ModuleIdentifier;
20 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
21 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.ModuleStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.NamespaceStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.PrefixStatement;
25 import org.opendaylight.yangtools.yang.parser.builder.impl.ModuleIdentifierImpl;
26 import org.opendaylight.yangtools.yang.parser.spi.ModuleNamespace;
27 import org.opendaylight.yangtools.yang.parser.spi.NamespaceToModule;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
31 import org.opendaylight.yangtools.yang.parser.spi.source.ImpPrefixToModuleIdentifier;
32 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleIdentifierToModuleQName;
33 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleNameToModuleQName;
34 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleNamespaceForBelongsTo;
35 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleQNameToModuleName;
36 import org.opendaylight.yangtools.yang.parser.spi.source.PrefixToModule;
37 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
38 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.ModuleEffectiveStatementImpl;
39
40 public class ModuleStatementSupport extends
41         AbstractStatementSupport<String, ModuleStatement, EffectiveStatement<String, ModuleStatement>> {
42
43     private QNameModule qNameModule;
44
45     public ModuleStatementSupport() {
46         super(Rfc6020Mapping.MODULE);
47     }
48
49     @Override
50     public String parseArgumentValue(StmtContext<?, ?, ?> ctx, String value) {
51         return value;
52     }
53
54     @Override
55     public ModuleStatement createDeclared(StmtContext<String, ModuleStatement, ?> ctx) {
56         return new ModuleStatementImpl(ctx);
57     }
58
59     @Override
60     public EffectiveStatement<String, ModuleStatement> createEffective(
61             StmtContext<String, ModuleStatement, EffectiveStatement<String, ModuleStatement>> ctx) {
62         return new ModuleEffectiveStatementImpl(ctx);
63     }
64
65     @Override
66     public void onLinkageDeclared(Mutable<String, ModuleStatement, EffectiveStatement<String, ModuleStatement>> stmt)
67             throws SourceException {
68
69         Optional<URI> moduleNs = Optional.fromNullable(firstAttributeOf(stmt.declaredSubstatements(),
70                 NamespaceStatement.class));
71         if (!moduleNs.isPresent()) {
72             throw new IllegalArgumentException("Namespace of the module [" + stmt.getStatementArgument()
73                     + "] is missing.");
74         }
75
76         Optional<Date> revisionDate = Optional.fromNullable(Utils.getLatestRevision(stmt.declaredSubstatements()));
77         if (!revisionDate.isPresent()) {
78             revisionDate = Optional.of(SimpleDateFormatUtil.DEFAULT_DATE_REV);
79         }
80
81         qNameModule = QNameModule.cachedReference(QNameModule.create(moduleNs.get(), revisionDate.orNull()));
82         ModuleIdentifier moduleIdentifier = new ModuleIdentifierImpl(stmt.getStatementArgument(),
83                 Optional.<URI> absent(), revisionDate);
84
85         stmt.addContext(ModuleNamespace.class, moduleIdentifier, stmt);
86         stmt.addContext(ModuleNamespaceForBelongsTo.class, moduleIdentifier.getName(), stmt);
87         stmt.addContext(NamespaceToModule.class, qNameModule, stmt);
88
89         String modulePrefix = firstAttributeOf(stmt.declaredSubstatements(), PrefixStatement.class);
90         if (modulePrefix == null) {
91             throw new IllegalArgumentException("Prefix of the module [" + stmt.getStatementArgument() + "] is missing.");
92         }
93
94         stmt.addToNs(PrefixToModule.class, modulePrefix, qNameModule);
95         stmt.addToNs(ModuleNameToModuleQName.class, stmt.getStatementArgument(), qNameModule);
96         stmt.addToNs(ModuleCtxToModuleQName.class, stmt, qNameModule);
97         stmt.addToNs(ModuleQNameToModuleName.class, qNameModule, stmt.getStatementArgument());
98         stmt.addToNs(ModuleIdentifierToModuleQName.class, moduleIdentifier, qNameModule);
99
100         stmt.addToNs(ImpPrefixToModuleIdentifier.class, modulePrefix, moduleIdentifier);
101     }
102
103     @Override
104     public void onFullDefinitionDeclared(
105             final Mutable<String, ModuleStatement, EffectiveStatement<String, ModuleStatement>> stmt)
106             throws SourceException {
107
108         stmt.addContext(NamespaceToModule.class, qNameModule, stmt);
109     }
110 }