dd8c76b3a00ab59b7c3f569c583097f83b8cb622
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / BelongsToStatementImpl.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 com.google.common.base.Optional;
11 import java.net.URI;
12 import java.util.Collection;
13 import org.opendaylight.yangtools.yang.common.SimpleDateFormatUtil;
14 import org.opendaylight.yangtools.yang.model.api.ModuleIdentifier;
15 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
16 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.BelongsToStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.PrefixStatement;
19 import org.opendaylight.yangtools.yang.parser.builder.impl.ModuleIdentifierImpl;
20 import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
21 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
22 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
23 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceAction;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
29 import org.opendaylight.yangtools.yang.parser.spi.source.BelongsToModuleContext;
30 import org.opendaylight.yangtools.yang.parser.spi.source.BelongsToPrefixToModuleIdentifier;
31 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleNamespaceForBelongsTo;
32 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
33 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.BelongsEffectiveToStatementImpl;
34
35 public class BelongsToStatementImpl extends AbstractDeclaredStatement<String>
36         implements BelongsToStatement {
37     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(Rfc6020Mapping
38             .BELONGS_TO)
39             .add(Rfc6020Mapping.PREFIX, 1, 1)
40             .build();
41
42     protected BelongsToStatementImpl(
43             final StmtContext<String, BelongsToStatement, ?> context) {
44         super(context);
45     }
46
47     public static class Definition
48             extends
49             AbstractStatementSupport<String, BelongsToStatement, EffectiveStatement<String, BelongsToStatement>> {
50
51         public Definition() {
52             super(Rfc6020Mapping.BELONGS_TO);
53         }
54
55         @Override
56         public String parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
57             return value;
58         }
59
60         @Override
61         public BelongsToStatement createDeclared(
62                 final StmtContext<String, BelongsToStatement, ?> ctx) {
63             return new BelongsToStatementImpl(ctx);
64         }
65
66         @Override
67         public EffectiveStatement<String, BelongsToStatement> createEffective(
68                 final StmtContext<String, BelongsToStatement, EffectiveStatement<String, BelongsToStatement>> ctx) {
69             return new BelongsEffectiveToStatementImpl(ctx);
70         }
71
72         @Override
73         public void onLinkageDeclared(final StmtContext.Mutable<String, BelongsToStatement, EffectiveStatement<String, BelongsToStatement>> belongsToCtx) throws SourceException {
74             ModelActionBuilder belongsToAction = belongsToCtx.newInferenceAction(ModelProcessingPhase.SOURCE_LINKAGE);
75
76             final ModuleIdentifier belongsToModuleIdentifier = getModuleIdentifier(belongsToCtx);
77             final ModelActionBuilder.Prerequisite<StmtContext<?, ?, ?>> belongsToPrereq = belongsToAction.requiresCtx(belongsToCtx, ModuleNamespaceForBelongsTo.class, belongsToModuleIdentifier.getName(), ModelProcessingPhase.SOURCE_LINKAGE);
78
79             belongsToAction.apply(new InferenceAction() {
80
81                 @Override
82                 public void apply() throws InferenceException {
83                     StmtContext<?, ?, ?> belongsToModuleCtx = belongsToPrereq.get();
84
85                     belongsToCtx.addToNs(BelongsToModuleContext.class, belongsToModuleIdentifier,
86                             belongsToModuleCtx);
87                     belongsToCtx.addToNs(BelongsToPrefixToModuleIdentifier.class, StmtContextUtils
88                             .findFirstDeclaredSubstatement(belongsToCtx, PrefixStatement.class).getStatementArgument
89                                     (), belongsToModuleIdentifier);
90                 }
91
92                 @Override
93                 public void prerequisiteFailed(final Collection<? extends ModelActionBuilder.Prerequisite<?>> failed) throws InferenceException {
94                     if (failed.contains(belongsToPrereq)) {
95                         throw new InferenceException("Module from belongs-to was not found: " + belongsToCtx.getStatementArgument(), belongsToCtx
96                                 .getStatementSourceReference());
97                     }
98                 }
99             });
100         }
101
102         private static ModuleIdentifier getModuleIdentifier(final StmtContext.Mutable<String, BelongsToStatement, EffectiveStatement<String, BelongsToStatement>> belongsToCtx) {
103             String moduleName = belongsToCtx.getStatementArgument();
104             return new ModuleIdentifierImpl(moduleName, Optional.<URI> absent(), Optional.of(SimpleDateFormatUtil.DEFAULT_BELONGS_TO_DATE));
105         }
106
107         @Override
108         public void onFullDefinitionDeclared(StmtContext.Mutable<String, BelongsToStatement,
109                 EffectiveStatement<String, BelongsToStatement>> stmt) throws SourceException {
110             super.onFullDefinitionDeclared(stmt);
111             SUBSTATEMENT_VALIDATOR.validate(stmt);
112         }
113     }
114
115     @Override
116     public String getModule() {
117         return argument();
118     }
119
120     @Override
121     public PrefixStatement getPrefix() {
122         return firstDeclared(PrefixStatement.class);
123     }
124
125 }