/*
* Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.yangtools.yang.parser.stmt.reactor;
import static java.util.Objects.requireNonNull;
import com.google.common.base.MoreObjects;
import com.google.common.base.MoreObjects.ToStringHelper;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.QNameModule;
import org.opendaylight.yangtools.yang.model.api.meta.ArgumentDefinition;
import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.meta.IdentifierNamespace;
import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
import org.opendaylight.yangtools.yang.parser.spi.meta.ImplicitParentAwareStatementSupport;
import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
import org.opendaylight.yangtools.yang.parser.spi.meta.StatementFactory;
import org.opendaylight.yangtools.yang.parser.spi.meta.StatementSupport;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
public class StatementDefinitionContext, E extends EffectiveStatement> {
private final @NonNull StatementSupport support;
private final Map> argumentSpecificSubDefinitions;
private Map, StatementDefinitionContext,?,?>> unknownStmtDefsOfYangStmts;
public StatementDefinitionContext(final StatementSupport support) {
this.support = requireNonNull(support);
this.argumentSpecificSubDefinitions = support.hasArgumentSpecificSupports() ? new HashMap<>() : null;
}
public @NonNull StatementFactory getFactory() {
return support;
}
public A parseArgumentValue(final StmtContext context, final String value) {
return support.parseArgumentValue(context, value);
}
public A adaptArgumentValue(final StmtContext context, final QNameModule targetModule) {
return support.adaptArgumentValue(context, targetModule);
}
public void checkNamespaceAllowed(final Class extends IdentifierNamespace,?>> namespace) {
// Noop
}
public @NonNull StatementDefinition getPublicView() {
return support.getPublicView();
}
public Optional> getImplicitParentFor(final StatementDefinition stmtDef) {
return support instanceof ImplicitParentAwareStatementSupport
? ((ImplicitParentAwareStatementSupport) support).getImplicitParentFor(stmtDef) : Optional.empty();
}
public boolean onStatementAdded(final Mutable stmt) {
support.onStatementAdded(stmt);
return false;
}
public void onDeclarationFinished(final Mutable statement, final ModelProcessingPhase phase) {
switch (phase) {
case SOURCE_PRE_LINKAGE:
support.onPreLinkageDeclared(statement);
break;
case SOURCE_LINKAGE:
support.onLinkageDeclared(statement);
break;
case STATEMENT_DEFINITION:
support.onStatementDefinitionDeclared(statement);
break;
case FULL_DECLARATION:
support.onFullDefinitionDeclared(statement);
break;
default:
break;
}
}
public @NonNull Class> getRepresentingClass() {
return support.getDeclaredRepresentationClass();
}
public @NonNull Optional getArgumentDefinition() {
return support.getArgumentDefinition();
}
public @NonNull QName getStatementName() {
return support.getStatementName();
}
@Override
public final String toString() {
return addToStringAttributes(MoreObjects.toStringHelper(this).omitNullValues()).toString();
}
protected ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
return toStringHelper.add("statement", getStatementName());
}
final @NonNull StatementDefinitionContext, ?, ?> getSubDefinitionSpecificForArgument(final String argument) {
if (!hasArgumentSpecificSubDefinitions()) {
return this;
}
StatementDefinitionContext, ?, ?> potential = argumentSpecificSubDefinitions.get(argument);
if (potential == null) {
final StatementSupport, ?, ?> argumentSpecificSupport = support.getSupportSpecificForArgument(argument);
potential = argumentSpecificSupport != null ? new StatementDefinitionContext<>(argumentSpecificSupport)
: this;
argumentSpecificSubDefinitions.put(argument, potential);
}
return potential;
}
final StatementSupport support() {
return support;
}
final boolean hasArgumentSpecificSubDefinitions() {
return support.hasArgumentSpecificSupports();
}
final StatementDefinitionContext, ?, ?> getAsUnknownStatementDefinition(
final StatementDefinitionContext, ?, ?> yangStmtDef) {
if (unknownStmtDefsOfYangStmts != null) {
final StatementDefinitionContext, ?, ?> existing = unknownStmtDefsOfYangStmts.get(yangStmtDef);
if (existing != null) {
return existing;
}
} else {
unknownStmtDefsOfYangStmts = new HashMap<>();
}
@SuppressWarnings("unchecked")
final StatementDefinitionContext, ?, ?> ret = support.getUnknownStatementDefinitionOf(
yangStmtDef.getPublicView()).map(StatementDefinitionContext::new).orElse(null);
if (ret != null) {
unknownStmtDefsOfYangStmts.put(yangStmtDef, ret);
}
return ret;
}
}