/*
 * Copyright (c) 2014 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.bgp.rib.impl;

import com.google.common.base.Optional;
import com.google.common.base.Preconditions;
import com.google.common.base.Stopwatch;
import io.netty.channel.Channel;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
import org.opendaylight.controller.config.yang.bgp.rib.impl.AdvertisedAddPathTableTypes;
import org.opendaylight.controller.config.yang.bgp.rib.impl.AdvertizedTableTypes;
import org.opendaylight.controller.config.yang.bgp.rib.impl.BgpSessionState;
import org.opendaylight.controller.config.yang.bgp.rib.impl.ErrorMsgs;
import org.opendaylight.controller.config.yang.bgp.rib.impl.ErrorReceived;
import org.opendaylight.controller.config.yang.bgp.rib.impl.ErrorSent;
import org.opendaylight.controller.config.yang.bgp.rib.impl.KeepAliveMsgs;
import org.opendaylight.controller.config.yang.bgp.rib.impl.MessagesStats;
import org.opendaylight.controller.config.yang.bgp.rib.impl.PeerPreferences;
import org.opendaylight.controller.config.yang.bgp.rib.impl.Received;
import org.opendaylight.controller.config.yang.bgp.rib.impl.RouteRefreshMsgs;
import org.opendaylight.controller.config.yang.bgp.rib.impl.Sent;
import org.opendaylight.controller.config.yang.bgp.rib.impl.SpeakerPreferences;
import org.opendaylight.controller.config.yang.bgp.rib.impl.TotalMsgs;
import org.opendaylight.controller.config.yang.bgp.rib.impl.UpdateMsgs;
import org.opendaylight.protocol.bgp.rib.impl.BGPSessionImpl.State;
import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
import org.opendaylight.protocol.util.StatisticsUtil;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Notify;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Open;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilities;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParameters;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.MultiprotocolCapability;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.add.path.capability.AddressFamilies;

final class BGPSessionStats {
    private final Stopwatch sessionStopwatch;
    private final BgpSessionState stats;
    private final TotalMsgs totalMsgs = new TotalMsgs();
    private final KeepAliveMsgs kaMsgs = new KeepAliveMsgs();
    private final UpdateMsgs updMsgs = new UpdateMsgs();
    private final RouteRefreshMsgs rrMsgs = new RouteRefreshMsgs();
    private final ErrorMsgs errMsgs = new ErrorMsgs();

    public BGPSessionStats(final Open remoteOpen, final int holdTimerValue, final int keepAlive, final Channel channel,
        final Optional<BGPSessionPreferences> localPreferences, final Collection<BgpTableType> tableTypes, final List<AddressFamilies> addPathTypes) {
        this.sessionStopwatch = Stopwatch.createUnstarted();
        this.stats = new BgpSessionState();
        this.stats.setHoldtimeCurrent(holdTimerValue);
        this.stats.setKeepaliveCurrent(keepAlive);
        this.stats.setPeerPreferences(setPeerPref(remoteOpen, channel, tableTypes, addPathTypes));
        this.stats.setSpeakerPreferences(setSpeakerPref(channel, localPreferences));
        initMsgs();
    }

    private void initMsgs() {
        this.totalMsgs.setReceived(new Received());
        this.totalMsgs.setSent(new Sent());
        this.kaMsgs.setReceived(new Received());
        this.kaMsgs.setSent(new Sent());
        this.updMsgs.setReceived(new Received());
        this.updMsgs.setSent(new Sent());
        this.rrMsgs.setReceived(new Received());
        this.rrMsgs.setSent(new Sent());
        this.errMsgs.setErrorReceived(new ErrorReceived());
        this.errMsgs.setErrorSent(new ErrorSent());
    }

    public void startSessionStopwatch() {
        this.sessionStopwatch.start();
    }

    public void updateSentMsgTotal() {
        updateSentMsg(this.totalMsgs.getSent());
    }

    public void updateReceivedMsgTotal() {
        updateReceivedMsg(this.totalMsgs.getReceived());
    }

    public void updateReceivedMsgKA() {
        updateReceivedMsg(this.kaMsgs.getReceived());
    }

    public void updateSentMsgKA() {
        updateSentMsg(this.kaMsgs.getSent());
    }

    public void updateSentMsgUpd() {
        updateSentMsg(this.updMsgs.getSent());
    }

    public void updateReceivedMsgUpd() {
        updateReceivedMsg(this.updMsgs.getReceived());
    }

    public void updateSentMsgRR() {
        updateSentMsg(this.rrMsgs.getSent());
    }

    public void updateReceivedMsgRR() {
        updateReceivedMsg(this.rrMsgs.getReceived());
    }

    public void updateReceivedMsgErr(final Notify error) {
        Preconditions.checkNotNull(error);
        final ErrorReceived received = this.errMsgs.getErrorReceived();
        received.setCount(received.getCount() + 1);
        received.setTimestamp(StatisticsUtil.getCurrentTimestampInSeconds());
        received.setCode(error.getErrorCode());
        received.setSubCode(error.getErrorSubcode());
    }

    public void updateSentMsgErr(final Notify error) {
        Preconditions.checkNotNull(error);
        final ErrorSent sent = this.errMsgs.getErrorSent();
        sent.setCount(sent.getCount() + 1);
        sent.setTimestamp(StatisticsUtil.getCurrentTimestampInSeconds());
        sent.setCode(error.getErrorCode());
        sent.setSubCode(error.getErrorSubcode());
    }

    public BgpSessionState getBgpSessionState(final State state) {
        Preconditions.checkNotNull(state);
        final MessagesStats msgs = new MessagesStats();
        msgs.setTotalMsgs(this.totalMsgs);
        msgs.setErrorMsgs(this.errMsgs);
        msgs.setKeepAliveMsgs(this.kaMsgs);
        msgs.setUpdateMsgs(this.updMsgs);
        msgs.setRouteRefreshMsgs(this.rrMsgs);
        this.stats.setSessionDuration(StatisticsUtil.formatElapsedTime(this.sessionStopwatch.elapsed(TimeUnit.SECONDS)));
        this.stats.setSessionState(state.toString());
        this.stats.setMessagesStats(msgs);
        return this.stats;
    }

    public void resetStats() {
        initMsgs();
    }

    private static void updateReceivedMsg(final Received received) {
        Preconditions.checkNotNull(received);
        received.setCount(received.getCount() + 1);
        received.setTimestamp(StatisticsUtil.getCurrentTimestampInSeconds());
    }

    private static void updateSentMsg(final Sent sent) {
        Preconditions.checkNotNull(sent);
        sent.setCount(sent.getCount() + 1);
        sent.setTimestamp(StatisticsUtil.getCurrentTimestampInSeconds());
    }

    private static AdvertizedTableTypes addTableType(final BgpTableType type) {
        Preconditions.checkNotNull(type);
        final AdvertizedTableTypes att = new AdvertizedTableTypes();
        att.setAfi(type.getAfi().getSimpleName());
        att.setSafi(type.getSafi().getSimpleName());
        return att;
    }

    private static AdvertisedAddPathTableTypes addAddPathTableType(final AddressFamilies addressFamilies) {
        Preconditions.checkNotNull(addressFamilies);
        final AdvertisedAddPathTableTypes att = new AdvertisedAddPathTableTypes();
        att.setAfi(addressFamilies.getAfi().getSimpleName());
        att.setSafi(addressFamilies.getSafi().getSimpleName());
        att.setSendReceive(addressFamilies.getSendReceive().toString());
        return att;
    }

    private static SpeakerPreferences setSpeakerPref(final Channel channel, final Optional<BGPSessionPreferences> localPreferences) {
        Preconditions.checkNotNull(channel);
        final SpeakerPreferences pref = new SpeakerPreferences();
        final InetSocketAddress isa = (InetSocketAddress) channel.localAddress();
        pref.setAddress(isa.getAddress().getHostAddress());
        pref.setPort(isa.getPort());
        final List<AdvertizedTableTypes> tt = new ArrayList<>();
        if (localPreferences.isPresent()) {
            final BGPSessionPreferences localPref = localPreferences.get();
            pref.setBgpId(localPref.getBgpId().getValue());
            pref.setAs(localPref.getMyAs().getValue());
            pref.setHoldtime(localPref.getHoldTime());
            if (localPref.getParams() != null) {
                for (final BgpParameters param : localPref.getParams()) {
                    for (final OptionalCapabilities capa : param.getOptionalCapabilities()) {
                        final CParameters cParam = capa.getCParameters();
                        if(cParam.getAugmentation(CParameters1.class) != null) {
                            final MultiprotocolCapability mc = cParam.getAugmentation(CParameters1.class).getMultiprotocolCapability();
                            if (mc != null) {
                                final AdvertizedTableTypes att = new AdvertizedTableTypes();
                                att.setAfi(mc.getAfi().getSimpleName());
                                att.setSafi(mc.getSafi().getSimpleName());
                                tt.add(att);
                            }
                            if (cParam.getAs4BytesCapability() != null) {
                                pref.setFourOctetAsCapability(true);
                            }
                            if (cParam.getAugmentation(CParameters1.class) != null &&
                                    cParam.getAugmentation(CParameters1.class).getGracefulRestartCapability() != null) {
                                pref.setGrCapability(true);
                            }
                            if (cParam.getAugmentation(CParameters1.class) != null &&
                                    cParam.getAugmentation(CParameters1.class).getAddPathCapability() != null) {
                                pref.setAddPathCapability(true);
                            }
                            if (cParam.getBgpExtendedMessageCapability() != null) {
                                pref.setBgpExtendedMessageCapability(true);
                            }
                            if (cParam.getAugmentation(CParameters1.class) != null &&
                                cParam.getAugmentation(CParameters1.class).getRouteRefreshCapability() != null) {
                                pref.setRouteRefreshCapability(true);
                            }
                        }
                    }
                }
            }
        }
        pref.setAdvertizedTableTypes(tt);
        return pref;
    }

    private static PeerPreferences setPeerPref(final Open remoteOpen, final Channel channel, final Collection<BgpTableType> tableTypes,
        final List<AddressFamilies> addPathTypes) {
        Preconditions.checkNotNull(remoteOpen);
        Preconditions.checkNotNull(channel);
        final PeerPreferences pref = new PeerPreferences();
        final InetSocketAddress isa = (InetSocketAddress) channel.remoteAddress();
        pref.setAddress(isa.getAddress().getHostAddress());
        pref.setPort(isa.getPort());
        pref.setBgpId(remoteOpen.getBgpIdentifier().getValue());
        pref.setAs(remoteOpen.getMyAsNumber().longValue());
        pref.setHoldtime(remoteOpen.getHoldTimer());

        final List<AdvertizedTableTypes> tt = tableTypes.stream().map(BGPSessionStats::addTableType).collect(Collectors.toList());
        final List<AdvertisedAddPathTableTypes> addPathTableTypeList = addPathTypes.stream().map(BGPSessionStats::addAddPathTableType)
            .collect(Collectors.toList());

        if (remoteOpen.getBgpParameters() != null) {
            for (final BgpParameters param : remoteOpen.getBgpParameters()) {
                for (final OptionalCapabilities capa : param.getOptionalCapabilities()) {
                    final CParameters cParam = capa.getCParameters();
                    if (cParam.getAs4BytesCapability() != null) {
                        pref.setFourOctetAsCapability(true);
                    }
                    if (cParam.getAugmentation(CParameters1.class) != null &&
                            cParam.getAugmentation(CParameters1.class).getGracefulRestartCapability() != null) {
                        pref.setGrCapability(true);
                    }
                    if (cParam.getAugmentation(CParameters1.class) != null &&
                            cParam.getAugmentation(CParameters1.class).getAddPathCapability() != null) {
                        pref.setAddPathCapability(true);
                    }
                    if (cParam.getAugmentation(CParameters1.class) != null &&
                        cParam.getAugmentation(CParameters1.class).getRouteRefreshCapability() != null) {
                        pref.setRouteRefreshCapability(true);
                    }
                    if (cParam.getBgpExtendedMessageCapability() != null) {
                        pref.setBgpExtendedMessageCapability(true);
                    }
                }

            }
        }
        pref.setAdvertizedTableTypes(tt);
        pref.setAdvertisedAddPathTableTypes(addPathTableTypeList);
        return pref;
    }
}