Teach YANG parser to retain DeclarationReference
[yangtools.git] / yang / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / thirdparty / plugin / ThirdPartyExtensionSupport.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.thirdparty.plugin;
9
10 import com.google.common.collect.ImmutableList;
11 import org.opendaylight.yangtools.yang.common.Empty;
12 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
13 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
14 import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
15 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStringStatementSupport;
16 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
17 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
18 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
19 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
20
21 public final class ThirdPartyExtensionSupport
22         extends AbstractStringStatementSupport<ThirdPartyExtensionStatement, ThirdPartyExtensionEffectiveStatement> {
23     public ThirdPartyExtensionSupport(final YangParserConfiguration config) {
24         super(ThirdPartyExtensionsMapping.THIRD_PARTY_EXTENSION, StatementPolicy.contextIndependent(), config);
25     }
26
27     @Override
28     public void onFullDefinitionDeclared(
29             final Mutable<String, ThirdPartyExtensionStatement, ThirdPartyExtensionEffectiveStatement> stmt) {
30         super.onFullDefinitionDeclared(stmt);
31         stmt.addToNs(ThirdPartyNamespace.class, Empty.getInstance(), "Third-party namespace test.");
32     }
33
34     @Override
35     protected SubstatementValidator getSubstatementValidator() {
36         return null;
37     }
38
39     @Override
40     protected ThirdPartyExtensionStatement createDeclared(
41             final StmtContext<String, ThirdPartyExtensionStatement, ?> ctx,
42             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
43         return new ThirdPartyExtensionStatementImpl(ctx.getRawArgument(), substatements);
44     }
45
46     @Override
47     protected ThirdPartyExtensionEffectiveStatement createEffective(
48             final Current<String, ThirdPartyExtensionStatement> stmt,
49             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
50         return new ThirdPartyExtensionEffectiveStatementImpl(stmt, substatements);
51     }
52 }