/*
 * Copyright (c) 2016 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.protocol.bmp.mock;

import com.google.common.net.InetAddresses;
import io.netty.channel.Channel;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
import org.opendaylight.protocol.bmp.api.BmpSession;
import org.opendaylight.protocol.bmp.api.BmpSessionFactory;
import org.opendaylight.protocol.bmp.api.BmpSessionListenerFactory;
import org.opendaylight.protocol.bmp.parser.BmpActivator;
import org.opendaylight.protocol.bmp.spi.registry.BmpExtensionProviderActivator;
import org.opendaylight.protocol.bmp.spi.registry.BmpExtensionProviderContext;
import org.opendaylight.protocol.bmp.spi.registry.SimpleBmpExtensionProviderContext;
import org.opendaylight.protocol.util.LoggerUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public final class BmpMock {

    private static final Logger LOG = LoggerFactory.getLogger(BmpMock.class);

    private BmpMock() {
        throw new UnsupportedOperationException();
    }

    public static void main(final String[] args) {
        LOG.info("Starting BMP test tool.");
        final BmpMockArguments arguments = BmpMockArguments.parseArguments(args);
        LoggerUtil.initiateLogger(arguments);

        final BmpMockDispatcher dispatcher = initiateMock(arguments);
        // now start the server / client
        if (arguments.isOnPassiveMode()) {
            deployServers(dispatcher, arguments);
        } else {
            deployClients(dispatcher, arguments);
        }
    }


    private static BmpMockDispatcher initiateMock(final BmpMockArguments arguments) {
        final BmpExtensionProviderContext ctx = new SimpleBmpExtensionProviderContext();
        final BmpExtensionProviderActivator bmpActivator = new BmpActivator(
            ServiceLoaderBGPExtensionProviderContext.getSingletonInstance());
        bmpActivator.start(ctx);

        return new BmpMockDispatcher(ctx.getBmpMessageRegistry(),
            new BmpSessionFactory() {
                @Override
                public BmpSession getSession(final Channel channel,
                    final BmpSessionListenerFactory sessionListenerFactory) {
                    return new BmpMockSession(arguments.getPeersCount(), arguments.getPrePolicyRoutesCount(), arguments.getPostPolicyRoutesCount());
                }
            });
    }

    private static void deployClients(final BmpMockDispatcher dispatcher, final BmpMockArguments arguments) {
        final InetSocketAddress localAddress = arguments.getLocalAddress();
        final InetSocketAddress remoteAddress = arguments.getRemoteAddress();
        InetAddress currentLocal = localAddress.getAddress();
        final int port = localAddress.getPort();
        for (int i = 0; i < arguments.getRoutersCount(); i++) {
            dispatcher.createClient(new InetSocketAddress(currentLocal, port), remoteAddress);
            currentLocal = InetAddresses.increment(currentLocal);
        }
    }

    private static void deployServers(final BmpMockDispatcher dispatcher, final BmpMockArguments arguments) {
        final InetSocketAddress localAddress = arguments.getLocalAddress();
        InetAddress currentLocal = localAddress.getAddress();
        final int port = localAddress.getPort();
        for (int i = 0; i < arguments.getRoutersCount(); i++) {
            dispatcher.createServer(new InetSocketAddress(currentLocal, port));
            currentLocal = InetAddresses.increment(currentLocal);
        }
    }
}