package org.opendaylight.openflowplugin.openflow.md.core.session;

import java.math.BigInteger;
import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.Future;

 import org.opendaylight.controller.sal.common.util.Rpcs;
import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
import org.opendaylight.openflowplugin.openflow.md.core.ConnectionConductor;
import org.opendaylight.openflowplugin.openflow.md.core.SwitchConnectionDistinguisher;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.UpdateFlowOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.UpdateFlowOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev131103.TransactionId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FlowModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FlowModInputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetAsyncInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetAsyncOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetConfigInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetConfigOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetQueueConfigInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetQueueConfigOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GroupModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GroupModInputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MeterModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MeterModInputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketOutInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.SetAsyncInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.SetConfigInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.TableModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.port.service.rev131107.UpdatePortOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.port.service.rev131107.UpdatePortOutputBuilder;
import org.opendaylight.yangtools.yang.common.RpcError;
import org.opendaylight.yangtools.yang.common.RpcResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.util.concurrent.Futures;

/**
 * message dispatch service to send the message to switch.
 *
 * @author AnilGujele
 *
 */
public class MessageDispatchServiceImpl implements IMessageDispatchService {

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

    private SessionContext session;

    /**
     * constructor
     *
     * @param session
     *            - MessageDispatchService for this session
     */
    public MessageDispatchServiceImpl(SessionContext session) {
        this.session = session;
    }

    /**
     * get proper connection adapter to send the message to switch.
     *
     * @param cookie to identify the right connection, it can be null also.
     * @return connectionAdapter associated with cookie, otherwise return best
     *         suitable connection.
     *
     */

    private ConnectionAdapter getConnectionAdapter(SwitchConnectionDistinguisher cookie) {

        if (!session.isValid()) {
            LOG.warn("Session for the cookie {} is invalid.", cookie);
            throw new IllegalArgumentException("Session for the cookie is invalid.");
        }
        LOG.debug("finding connecton for cookie value {}. ", cookie);
        // set main connection as default
        ConnectionAdapter connectionAdapter = session.getPrimaryConductor().getConnectionAdapter();
        if (null != cookie) {
            ConnectionConductor conductor = session.getAuxiliaryConductor(cookie);
            // check if auxiliary connection exist
            if (null != conductor) {
                LOG.debug("found auxiliary connection for the cookie.");
                connectionAdapter = conductor.getConnectionAdapter();
            }
        } else {
            // TODO: pick connection to utilize all the available connection.
        }
        return connectionAdapter;
    }

    @Override
    public Future<RpcResult<BarrierOutput>> barrier(BarrierInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).barrier(input);
    }

    @Override
    public Future<RpcResult<Void>> experimenter(ExperimenterInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).experimenter(input);
    }

    @Override
    public Future<RpcResult<UpdateFlowOutput>> flowMod(FlowModInput input, SwitchConnectionDistinguisher cookie) {

        // Set Xid before invoking RPC on OFLibrary
        // TODO : let caller set xid, in that case it will not be required to create FlowModInput again here to set xid 
        Long Xid = session.getNextXid();
        FlowModInputBuilder mdInput = new FlowModInputBuilder(input);
        mdInput.setXid(Xid);
        LOG.debug("Calling OFLibrary flowMod");
        Future<RpcResult<Void>> response = getConnectionAdapter(cookie).flowMod(mdInput.build());

        // Send the same Xid back to caller - MessageDrivenSwitch
        UpdateFlowOutputBuilder flowModOutput = new UpdateFlowOutputBuilder();
        String stringXid =Xid.toString();
        BigInteger bigIntXid = new BigInteger( stringXid );
        flowModOutput.setTransactionId(new TransactionId(bigIntXid));

        UpdateFlowOutput result = flowModOutput.build();
        Collection<RpcError> errors = Collections.emptyList();
        RpcResult<UpdateFlowOutput> rpcResult = Rpcs.getRpcResult(true, result, errors);

        // solution 1: sending directly and hooking listener to get error
        // hookup listener to catch the possible error with no reference to returned future-object
        LOG.debug("Returning to ModelDrivenSwitch for flowMod RPC");
        return Futures.immediateFuture(rpcResult);

    }

    @Override
    public Future<RpcResult<GetAsyncOutput>> getAsync(GetAsyncInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).getAsync(input);
    }

    @Override
    public Future<RpcResult<GetConfigOutput>> getConfig(GetConfigInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).getConfig(input);
    }

    @Override
    public Future<RpcResult<GetFeaturesOutput>> getFeatures(GetFeaturesInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).getFeatures(input);
    }

    @Override
    public Future<RpcResult<GetQueueConfigOutput>> getQueueConfig(GetQueueConfigInput input,
            SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).getQueueConfig(input);
    }

    @Override
    public Future<RpcResult<UpdateGroupOutput>> groupMod(GroupModInput input, SwitchConnectionDistinguisher cookie) {

        // Set Xid before invoking RPC on OFLibrary
        // TODO : let caller set xid, in that case it will not be required to create object again here to set xid 
        Long Xid = session.getNextXid();
        GroupModInputBuilder mdInput = new GroupModInputBuilder(input);
        mdInput.setXid(Xid);
        LOG.debug("Calling OFLibrary groupMod");
        Future<RpcResult<Void>> response = getConnectionAdapter(cookie).groupMod(mdInput.build());

        // Send the same Xid back to caller - MessageDrivenSwitch
        UpdateGroupOutputBuilder groupModOutput = new UpdateGroupOutputBuilder();
        String stringXid =Xid.toString();
        BigInteger bigIntXid = new BigInteger( stringXid );
        groupModOutput.setTransactionId(new TransactionId(bigIntXid));
       
        UpdateGroupOutput result = groupModOutput.build();
        Collection<RpcError> errors = Collections.emptyList();
        RpcResult<UpdateGroupOutput> rpcResult = Rpcs.getRpcResult(true, result, errors);

        // solution 1: sending directly and hooking listener to get error
        // hookup listener to catch the possible error with no reference to returned future-object
        LOG.debug("Returning to ModelDrivenSwitch for groupMod RPC");
        return Futures.immediateFuture(rpcResult);

    }

    @Override
    public Future<RpcResult<UpdateMeterOutput>> meterMod(MeterModInput input, SwitchConnectionDistinguisher cookie) {

        // Set Xid before invoking RPC on OFLibrary
     // TODO : let caller set xid, in that case it will not be required to create MeterModInput again here to set xid 
        Long Xid = session.getNextXid();
        MeterModInputBuilder mdInput = new MeterModInputBuilder(input);
        mdInput.setXid(Xid);
        LOG.debug("Calling OFLibrary meterMod");
        Future<RpcResult<Void>> response = getConnectionAdapter(cookie).meterMod(mdInput.build());

        // Send the same Xid back to caller - MessageDrivenSwitch
        UpdateMeterOutputBuilder meterModOutput = new UpdateMeterOutputBuilder();
        String stringXid =Xid.toString();
        BigInteger bigIntXid = new BigInteger( stringXid );
        meterModOutput.setTransactionId(new TransactionId(bigIntXid));
        
        UpdateMeterOutput result = meterModOutput.build();
        Collection<RpcError> errors = Collections.emptyList();
        RpcResult<UpdateMeterOutput> rpcResult = Rpcs.getRpcResult(true, result, errors);

        // solution 1: sending directly and hooking listener to get error
        // hookup listener to catch the possible error with no reference to returned future-object
        LOG.debug("Returning to ModelDrivenSwitch for meterMod RPC");
        return Futures.immediateFuture(rpcResult);

    }

    @Override
    public Future<RpcResult<java.lang.Void>> multipartRequest(MultipartRequestInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).multipartRequest(input);
    }

    @Override
    public Future<RpcResult<Void>> packetOut(PacketOutInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).packetOut(input);
    }

    @Override
    public Future<RpcResult<UpdatePortOutput>> portMod(PortModInput input, SwitchConnectionDistinguisher cookie) {

        LOG.debug("Calling OFLibrary portMod");
        Future<RpcResult<Void>> response = getConnectionAdapter(cookie).portMod(input);

        // Send the same Xid back to caller - ModelDrivenSwitch
        UpdatePortOutputBuilder portModOutput = new UpdatePortOutputBuilder();
        String stringXid =input.getXid().toString();
        BigInteger bigIntXid = new BigInteger( stringXid );
        portModOutput.setTransactionId(new TransactionId(bigIntXid));
        UpdatePortOutput result = portModOutput.build();
        Collection<RpcError> errors = Collections.emptyList();
        RpcResult<UpdatePortOutput> rpcResult = Rpcs.getRpcResult(true, result, errors);

        LOG.debug("Returning to ModelDrivenSwitch for portMod RPC");
        return Futures.immediateFuture(rpcResult);
    }

    @Override
    public Future<RpcResult<RoleRequestOutput>> roleRequest(RoleRequestInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).roleRequest(input);
    }

    @Override
    public Future<RpcResult<Void>> setAsync(SetAsyncInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).setAsync(input);
    }

    @Override
    public Future<RpcResult<Void>> setConfig(SetConfigInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).setConfig(input);
    }

    @Override
    public Future<RpcResult<Void>> tableMod(TableModInput input, SwitchConnectionDistinguisher cookie) {
        return getConnectionAdapter(cookie).tableMod(input);
    }
}