/*
 * Copyright (c) 2013 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.controller.netconf.persist.impl.osgi;

import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import org.opendaylight.controller.netconf.client.NetconfClient;
import org.opendaylight.controller.netconf.persist.impl.ConfigPersisterNotificationHandler;
import org.opendaylight.controller.netconf.persist.impl.ConfigPusher;
import org.opendaylight.controller.netconf.persist.impl.PersisterAggregator;
import org.opendaylight.controller.netconf.persist.impl.Util;
import org.opendaylight.controller.netconf.util.osgi.NetconfConfigUtil;
import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.management.MBeanServer;
import java.lang.management.ManagementFactory;
import java.net.InetSocketAddress;
import java.util.regex.Pattern;

public class ConfigPersisterActivator implements BundleActivator {

    private static final Logger logger = LoggerFactory.getLogger(ConfigPersisterActivator.class);

    private final static MBeanServer platformMBeanServer = ManagementFactory.getPlatformMBeanServer();
    private static final String IGNORED_MISSING_CAPABILITY_REGEX_SUFFIX = "ignoredMissingCapabilityRegex";

    public static final String NETCONF_CONFIG_PERSISTER = "netconf.config.persister";

    public static final String STORAGE_ADAPTER_CLASS_PROP_SUFFIX = "storageAdapterClass";

    public static final String DEFAULT_IGNORED_REGEX = "^urn:ietf:params:xml:ns:netconf:base:1.0";


    private volatile ConfigPersisterNotificationHandler jmxNotificationHandler;
    private volatile NetconfClient netconfClient;
    private Thread initializationThread;
    private EventLoopGroup nettyThreadgroup;
    private PersisterAggregator persisterAggregator;

    @Override
    public void start(final BundleContext context) throws Exception {
        logger.debug("ConfigPersister starting");

        PropertiesProviderBaseImpl propertiesProvider = new PropertiesProviderBaseImpl(context);

        String regexProperty = propertiesProvider.getProperty(IGNORED_MISSING_CAPABILITY_REGEX_SUFFIX);
        String regex;
        if (regexProperty != null) {
            regex = regexProperty;
        } else {
            regex = DEFAULT_IGNORED_REGEX;
        }
        final Pattern ignoredMissingCapabilityRegex = Pattern.compile(regex);
        nettyThreadgroup = new NioEventLoopGroup();

        persisterAggregator = PersisterAggregator.createFromProperties(propertiesProvider);
        final InetSocketAddress address = NetconfConfigUtil.extractTCPNetconfAddress(context, "Netconf is not configured, persister is not operational", true);
        final ConfigPusher configPusher = new ConfigPusher(address, nettyThreadgroup);


        // offload initialization to another thread in order to stop blocking activator
        Runnable initializationRunnable = new Runnable() {
            @Override
            public void run() {
                try {
                    netconfClient = configPusher.init(persisterAggregator.loadLastConfigs());
                    jmxNotificationHandler = new ConfigPersisterNotificationHandler(
                            platformMBeanServer, netconfClient, persisterAggregator,
                            ignoredMissingCapabilityRegex);
                    jmxNotificationHandler.init();
                } catch (InterruptedException e) {
                    logger.info("Interrupted while waiting for netconf connection");
                }
            }
        };
        initializationThread = new Thread(initializationRunnable, "ConfigPersister-registrator");
        initializationThread.start();
    }

    @Override
    public void stop(BundleContext context) throws Exception {
        initializationThread.interrupt();
        if (jmxNotificationHandler != null) {
            jmxNotificationHandler.close();
        }
        if (netconfClient != null) {
            netconfClient = jmxNotificationHandler.getNetconfClient();
            try {
                Util.closeClientAndDispatcher(netconfClient);
            } catch (Exception e) {
                logger.warn("Unable to close connection to netconf {}", netconfClient, e);
            }
        }
        nettyThreadgroup.shutdownGracefully();
        persisterAggregator.close();
    }
}