/*
 * Copyright (c) 2022 PANTHEON.tech, s.r.o. 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.mdsal.binding.generator.impl.reactor;

import static com.google.common.base.Verify.verify;
import static java.util.Objects.requireNonNull;

import com.google.common.collect.Collections2;
import com.google.common.collect.ImmutableList;
import java.util.Collection;
import java.util.List;
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.model.api.ActionDefinition;
import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
import org.opendaylight.yangtools.yang.model.api.Status;
import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.UsesNode;
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.StatementOrigin;
import org.opendaylight.yangtools.yang.model.api.stmt.AugmentEffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.AugmentStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeAwareEffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
import org.opendaylight.yangtools.yang.xpath.api.YangXPathExpression.QualifiedBound;

final class TargetAugmentEffectiveStatement implements AugmentEffectiveStatement, AugmentationSchemaNode {
    private final @NonNull List<EffectiveStatement<?, ?>> substatements;
    private final @NonNull AugmentEffectiveStatement delegate;
    private final @NonNull AugmentationSchemaNode schemaDelegate;

    TargetAugmentEffectiveStatement(final AugmentEffectiveStatement augment,
            final SchemaTreeAwareEffectiveStatement<?, ?> target) {
        delegate = requireNonNull(augment);
        verify(augment instanceof AugmentationSchemaNode, "Unsupported augment implementation %s", augment);
        schemaDelegate = (AugmentationSchemaNode) augment;

        final var stmts = augment.effectiveSubstatements();
        final var builder = ImmutableList.<EffectiveStatement<?, ?>>builderWithExpectedSize(stmts.size());
        for (var stmt : stmts) {
            if (stmt instanceof SchemaTreeEffectiveStatement) {
                final var qname = ((SchemaTreeEffectiveStatement<?>) stmt).getIdentifier();
                target.get(SchemaTreeNamespace.class, qname).ifPresent(builder::add);
            } else {
                builder.add(stmt);
            }
        }

        substatements = builder.build();
    }

    @NonNull AugmentEffectiveStatement delegate() {
        return delegate;
    }

    @Override
    public AugmentStatement getDeclared() {
        return delegate.getDeclared();
    }

    @Override
    public SchemaNodeIdentifier argument() {
        return delegate.argument();
    }

    @Override
    public StatementOrigin statementOrigin() {
        return delegate.statementOrigin();
    }

    @Override
    public <K, V, N extends IdentifierNamespace<K, V>> Optional<V> get(final Class<N> namespace, final K identifier) {
        return Optional.empty();
    }

    @Override
    public <K, V, N extends IdentifierNamespace<K, V>> Map<K, V> getAll(final Class<N> namespace) {
        return Map.of();
    }

    @Override
    public List<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
        return substatements;
    }

    @Override
    public Collection<? extends TypeDefinition<?>> getTypeDefinitions() {
        return schemaDelegate.getTypeDefinitions();
    }

    @Override
    @SuppressWarnings("unchecked")
    public Collection<? extends DataSchemaNode> getChildNodes() {
        return (Collection) Collections2.filter(substatements, DataSchemaNode.class::isInstance);
    }

    @Override
    public Collection<? extends GroupingDefinition> getGroupings() {
        return schemaDelegate.getGroupings();
    }

    @Override
    public DataSchemaNode dataChildByName(final QName name) {
        return getChildNodes().stream().filter(child -> name.equals(child.getQName())).findFirst().orElse(null);
    }

    @Override
    public Collection<? extends UsesNode> getUses() {
        return schemaDelegate.getUses();
    }

    @Override
    @SuppressWarnings("unchecked")
    public Collection<? extends NotificationDefinition> getNotifications() {
        return (Collection) Collections2.filter(substatements, NotificationDefinition.class::isInstance);
    }

    @Override
    @SuppressWarnings("unchecked")
    public Collection<? extends ActionDefinition> getActions() {
        return (Collection) Collections2.filter(substatements, ActionDefinition.class::isInstance);
    }

    @Override
    public Optional<? extends QualifiedBound> getWhenCondition() {
        return schemaDelegate.getWhenCondition();
    }

    @Override
    public @NonNull Status getStatus() {
        return schemaDelegate.getStatus();
    }

    @Override
    public Optional<String> getDescription() {
        return schemaDelegate.getDescription();
    }

    @Override
    public Optional<String> getReference() {
        return schemaDelegate.getReference();
    }

    @Override
    public AugmentEffectiveStatement asEffectiveStatement() {
        return this;
    }
}