/*
 * 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.mdsal.binding.dom.adapter;

import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.mdsal.binding.api.ClusteredDataTreeChangeListener;
import org.opendaylight.mdsal.binding.api.DataChangeListener;
import org.opendaylight.mdsal.binding.api.DataListener;
import org.opendaylight.mdsal.binding.api.DataTreeChangeListener;
import org.opendaylight.mdsal.binding.api.DataTreeChangeService;
import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
import org.opendaylight.mdsal.dom.api.DOMDataTreeChangeService;
import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.yang.binding.Augmentation;
import org.opendaylight.yangtools.yang.binding.DataObject;

/**
 * Adapter exposing Binding {@link DataTreeChangeService} and wrapping a {@link DOMDataTreeChangeService} and is
 * responsible for translation and instantiation of {@link BindingDOMDataTreeChangeListenerAdapter} adapters.
 *
 * <p>
 * Each registered {@link DataTreeChangeListener} is wrapped using adapter and registered directly to DOM service.
 */
final class BindingDOMDataTreeChangeServiceAdapter extends AbstractBindingAdapter<DOMDataTreeChangeService>
        implements DataTreeChangeService {
    BindingDOMDataTreeChangeServiceAdapter(final AdapterContext adapterContext,
            final DOMDataTreeChangeService dataTreeChangeService) {
        super(adapterContext, dataTreeChangeService);
    }

    @Override
    public <T extends DataObject> Registration registerDataTreeChangeListener(final DataTreeIdentifier<T> treeId,
            final DataTreeChangeListener<T> listener) {
        final var domIdentifier = toDomTreeIdentifier(treeId);
        final var storeType = treeId.datastore();
        final var target = treeId.path().getTargetType();
        final var augment = Augmentation.class.isAssignableFrom(target) ? target : null;

        final var domListener = listener instanceof ClusteredDataTreeChangeListener
            ? new BindingClusteredDOMDataTreeChangeListenerAdapter<>(adapterContext(),
                (ClusteredDataTreeChangeListener<T>) listener, storeType, augment)
                : new BindingDOMDataTreeChangeListenerAdapter<>(adapterContext(), listener, storeType, augment);

        return getDelegate().registerDataTreeChangeListener(domIdentifier, domListener);
    }

    @Override
    public <T extends DataObject> Registration registerDataListener(final DataTreeIdentifier<T> treeId,
            final DataListener<T> listener) {
        return getDelegate().registerDataTreeChangeListener(toDomTreeInstance(treeId),
            new BindingDOMDataListenerAdapter<>(adapterContext(), listener));
    }

    @Override
    public <T extends DataObject> Registration registerDataChangeListener(final DataTreeIdentifier<T> treeId,
            final DataChangeListener<T> listener) {
        return getDelegate().registerDataTreeChangeListener(toDomTreeInstance(treeId),
            new BindingDOMDataChangeListenerAdapter<>(adapterContext(), listener));
    }

    private @NonNull DOMDataTreeIdentifier toDomTreeIdentifier(final DataTreeIdentifier<?> treeId) {
        return DOMDataTreeIdentifier.of(treeId.datastore(),
            currentSerializer().toYangInstanceIdentifier(treeId.path()));
    }

    private @NonNull DOMDataTreeIdentifier toDomTreeInstance(final DataTreeIdentifier<?> treeId) {
        final var instanceIdentifier = treeId.path();
        if (instanceIdentifier.isWildcarded()) {
            throw new IllegalArgumentException("Cannot register listener for wildcard " + instanceIdentifier);
        }
        return toDomTreeIdentifier(treeId);
    }
}