yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/rpc/AbstractRpcStatementSupport.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.parser.rfc7950.stmt.rpc;
   9
  10 import org.opendaylight.yangtools.yang.common.QName;
  11 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
  12 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  13 import org.opendaylight.yangtools.yang.model.api.stmt.InputStatement;
  14 import org.opendaylight.yangtools.yang.model.api.stmt.OutputStatement;
  15 import org.opendaylight.yangtools.yang.model.api.stmt.RpcStatement;
  16 import org.opendaylight.yangtools.yang.parser.rfc7950.namespace.ChildSchemaNodeNamespace;
  17 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractQNameStatementSupport;
  18 import org.opendaylight.yangtools.yang.parser.spi.meta.StatementSupport;
  19 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  20 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
  21 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
  22 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
  23 import org.opendaylight.yangtools.yang.parser.spi.source.ImplicitSubstatement;
  24 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase;
  25 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementDefinitionContext;
  26
  27 abstract class AbstractRpcStatementSupport
  28         extends AbstractQNameStatementSupport<RpcStatement, EffectiveStatement<QName, RpcStatement>> {
  29     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(
  30         YangStmtMapping.RPC)
  31         .addOptional(YangStmtMapping.DESCRIPTION)
  32         .addAny(YangStmtMapping.GROUPING)
  33         .addAny(YangStmtMapping.IF_FEATURE)
  34         .addOptional(YangStmtMapping.INPUT)
  35         .addOptional(YangStmtMapping.OUTPUT)
  36         .addOptional(YangStmtMapping.REFERENCE)
  37         .addOptional(YangStmtMapping.STATUS)
  38         .addAny(YangStmtMapping.TYPEDEF)
  39         .build();
  40
  41     AbstractRpcStatementSupport() {
  42         super(YangStmtMapping.RPC);
  43     }
  44
  45     @Override
  46     public final QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
  47         return StmtContextUtils.qnameFromArgument(ctx, value);
  48     }
  49
  50     @Override
  51     public final void onStatementAdded(
  52             final Mutable<QName, RpcStatement, EffectiveStatement<QName, RpcStatement>> stmt) {
  53         stmt.getParentContext().addToNs(ChildSchemaNodeNamespace.class, stmt.getStatementArgument(), stmt);
  54     }
  55
  56     @Override
  57     public final RpcStatement createDeclared(final StmtContext<QName, RpcStatement, ?> ctx) {
  58         return new RpcStatementImpl(ctx);
  59     }
  60
  61     @Override
  62     public final EffectiveStatement<QName, RpcStatement> createEffective(
  63             final StmtContext<QName, RpcStatement, EffectiveStatement<QName, RpcStatement>> ctx) {
  64         return new RpcEffectiveStatementImpl(ctx);
  65     }
  66
  67     @Override
  68     public final void onFullDefinitionDeclared(
  69             final Mutable<QName, RpcStatement, EffectiveStatement<QName, RpcStatement>> stmt) {
  70         super.onFullDefinitionDeclared(stmt);
  71
  72         if (StmtContextUtils.findFirstDeclaredSubstatement(stmt, InputStatement.class) == null) {
  73             addImplicitStatement((StatementContextBase<?, ?, ?>) stmt, implictInput());
  74         }
  75
  76         if (StmtContextUtils.findFirstDeclaredSubstatement(stmt, OutputStatement.class) == null) {
  77             addImplicitStatement((StatementContextBase<?, ?, ?>) stmt, implictOutput());
  78         }
  79     }
  80
  81
  82     @Override
  83     protected final SubstatementValidator getSubstatementValidator() {
  84         return SUBSTATEMENT_VALIDATOR;
  85     }
  86
  87     abstract StatementSupport<?, ?, ?> implictInput();
  88
  89     abstract StatementSupport<?, ?, ?> implictOutput();
  90
  91     private static void addImplicitStatement(final StatementContextBase<?, ?, ?> parent,
  92             final StatementSupport<?, ?, ?> statementToAdd) {
  93         final StatementDefinitionContext<?, ?, ?> stmtDefCtx = new StatementDefinitionContext<>(statementToAdd);
  94
  95         parent.createSubstatement(parent.declaredSubstatements().size(), stmtDefCtx,
  96             ImplicitSubstatement.of(parent.getStatementSourceReference()), null);
  97     }
  98 }