/*
 * 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<A, D extends DeclaredStatement<A>, E extends EffectiveStatement<A, D>> {
    private final @NonNull StatementSupport<A, D, E> support;
    private final Map<String, StatementDefinitionContext<?, ?, ?>> argumentSpecificSubDefinitions;
    private Map<StatementDefinitionContext<?, ?, ?>, StatementDefinitionContext<?,?,?>> unknownStmtDefsOfYangStmts;

    public StatementDefinitionContext(final StatementSupport<A, D, E> support) {
        this.support = requireNonNull(support);
        this.argumentSpecificSubDefinitions = support.hasArgumentSpecificSupports() ? new HashMap<>() : null;
    }

    public @NonNull StatementFactory<A, D, E> getFactory() {
        return support;
    }

    public A parseArgumentValue(final StmtContext<A, D, E> context, final String value) {
        return support.parseArgumentValue(context, value);
    }

    public A adaptArgumentValue(final StmtContext<A, D, E> 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<StatementSupport<?, ?, ?>> getImplicitParentFor(final StatementDefinition stmtDef) {
        return support instanceof ImplicitParentAwareStatementSupport
                ? ((ImplicitParentAwareStatementSupport) support).getImplicitParentFor(stmtDef) : Optional.empty();
    }

    public boolean onStatementAdded(final Mutable<A, D, E> stmt) {
        support.onStatementAdded(stmt);
        return false;
    }

    public void onDeclarationFinished(final Mutable<A, D, E> 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<ArgumentDefinition> 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<A, D, E> 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;
    }
}