/*
 * 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.netconf.topology.impl;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import io.netty.util.concurrent.EventExecutor;
import java.util.Collection;
import javax.annotation.PreDestroy;
import javax.inject.Inject;
import javax.inject.Singleton;
import org.opendaylight.aaa.encrypt.AAAEncryptionService;
import org.opendaylight.controller.config.threadpool.ScheduledThreadPool;
import org.opendaylight.controller.config.threadpool.ThreadPool;
import org.opendaylight.mdsal.binding.api.DataBroker;
import org.opendaylight.mdsal.binding.api.DataTreeChangeListener;
import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
import org.opendaylight.mdsal.binding.api.DataTreeModification;
import org.opendaylight.mdsal.binding.api.RpcProviderService;
import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
import org.opendaylight.mdsal.dom.api.DOMMountPointService;
import org.opendaylight.netconf.client.NetconfClientFactory;
import org.opendaylight.netconf.client.mdsal.api.BaseNetconfSchemas;
import org.opendaylight.netconf.client.mdsal.api.DeviceActionFactory;
import org.opendaylight.netconf.client.mdsal.api.SchemaResourceManager;
import org.opendaylight.netconf.topology.spi.AbstractNetconfTopology;
import org.opendaylight.netconf.topology.spi.NetconfClientConfigurationBuilderFactory;
import org.opendaylight.netconf.topology.spi.NetconfNodeUtils;
import org.opendaylight.netconf.topology.spi.NetconfTopologyRPCProvider;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.osgi.service.component.annotations.Activate;
import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.Deactivate;
import org.osgi.service.component.annotations.Reference;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

// Non-final for testing
@Singleton
@Component(service = { })
public class NetconfTopologyImpl extends AbstractNetconfTopology
        implements DataTreeChangeListener<Node>, AutoCloseable {
    private static final Logger LOG = LoggerFactory.getLogger(NetconfTopologyImpl.class);

    private Registration dtclReg;
    private NetconfTopologyRPCProvider rpcProvider;

    @Inject
    @Activate
    public NetconfTopologyImpl(
            @Reference(target = "(type=netconf-client-factory)") final NetconfClientFactory clientFactory,
            @Reference(target = "(type=global-event-executor)") final EventExecutor eventExecutor,
            @Reference(target = "(type=global-netconf-ssh-scheduled-executor)")
            final ScheduledThreadPool scheduledThreadPool,
            @Reference(target = "(type=global-netconf-processing-executor)") final ThreadPool processingThreadPool,
            @Reference final SchemaResourceManager schemaRepositoryProvider, @Reference final DataBroker dataBroker,
            @Reference final DOMMountPointService mountPointService,
            @Reference final AAAEncryptionService encryptionService,
            @Reference(target = "(type=default)") final NetconfClientConfigurationBuilderFactory builderFactory,
            @Reference final RpcProviderService rpcProviderService, @Reference final BaseNetconfSchemas baseSchemas,
            @Reference final DeviceActionFactory deviceActionFactory) {
        this(NetconfNodeUtils.DEFAULT_TOPOLOGY_NAME, clientFactory, eventExecutor, scheduledThreadPool,
            processingThreadPool, schemaRepositoryProvider, dataBroker, mountPointService, encryptionService,
            builderFactory, rpcProviderService, baseSchemas, deviceActionFactory);
    }

    public NetconfTopologyImpl(final String topologyId, final NetconfClientFactory clientclientFactory,
            final EventExecutor eventExecutor, final ScheduledThreadPool scheduledThreadPool,
            final ThreadPool processingThreadPool, final SchemaResourceManager schemaRepositoryProvider,
            final DataBroker dataBroker, final DOMMountPointService mountPointService,
            final AAAEncryptionService encryptionService, final NetconfClientConfigurationBuilderFactory builderFactory,
            final RpcProviderService rpcProviderService, final BaseNetconfSchemas baseSchemas) {
        this(topologyId, clientclientFactory, eventExecutor, scheduledThreadPool, processingThreadPool,
            schemaRepositoryProvider, dataBroker, mountPointService, encryptionService, builderFactory,
            rpcProviderService, baseSchemas, null);
    }

    @SuppressFBWarnings(value = "MC_OVERRIDABLE_METHOD_CALL_IN_CONSTRUCTOR",
        justification = "DTCL registration of 'this'")
    public NetconfTopologyImpl(final String topologyId, final NetconfClientFactory clientFactory,
            final EventExecutor eventExecutor, final ScheduledThreadPool scheduledThreadPool,
            final ThreadPool processingThreadPool, final SchemaResourceManager schemaRepositoryProvider,
            final DataBroker dataBroker, final DOMMountPointService mountPointService,
            final AAAEncryptionService encryptionService, final NetconfClientConfigurationBuilderFactory builderFactory,
            final RpcProviderService rpcProviderService, final BaseNetconfSchemas baseSchemas,
            final DeviceActionFactory deviceActionFactory) {
        super(topologyId, clientFactory, eventExecutor, scheduledThreadPool, processingThreadPool,
            schemaRepositoryProvider, dataBroker, mountPointService, builderFactory, deviceActionFactory, baseSchemas);

        LOG.debug("Registering datastore listener");
        dtclReg = dataBroker.registerDataTreeChangeListener(DataTreeIdentifier.create(
            LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.builder(NetworkTopology.class)
                .child(Topology.class, new TopologyKey(new TopologyId(topologyId)))
                .child(Node.class)
                .build()), this);
        rpcProvider = new NetconfTopologyRPCProvider(rpcProviderService, dataBroker, encryptionService, topologyId);
    }

    @PreDestroy
    @Deactivate
    @Override
    public void close() {
        if (rpcProvider != null) {
            rpcProvider.close();
            rpcProvider = null;
        }

        // close all existing connectors, delete whole topology in datastore?
        deleteAllNodes();

        if (dtclReg != null) {
            dtclReg.close();
            dtclReg = null;
        }
    }

    @Override
    public void onDataTreeChanged(final Collection<DataTreeModification<Node>> changes) {
        for (var change : changes) {
            final var rootNode = change.getRootNode();
            final var modType = rootNode.getModificationType();
            switch (modType) {
                case SUBTREE_MODIFIED -> ensureNode("updated", rootNode.getDataAfter());
                case WRITE -> ensureNode("created", rootNode.getDataAfter());
                case DELETE -> {
                    final var nodeId = InstanceIdentifier.keyOf(change.getRootPath().getRootIdentifier()).getNodeId();
                    LOG.debug("Config for node {} deleted", nodeId);
                    deleteNode(nodeId);
                }
                default -> LOG.debug("Unsupported modification type: {}.", modType);
            }
        }
    }

    private void ensureNode(final String operation, final Node node) {
        final var nodeId = node.getNodeId();
        LOG.debug("Config for node {} {}", nodeId, operation);
        ensureNode(node);
    }
}