/*
 * Copyright (c) 2013 Ericsson , 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.openflowplugin.droptest;

import org.eclipse.osgi.framework.console.CommandInterpreter;
import org.eclipse.osgi.framework.console.CommandProvider;
import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
import org.opendaylight.controller.sal.binding.api.data.DataBrokerService;
import org.osgi.framework.BundleContext;

public class DropTestCommandProvider implements CommandProvider {

    private DataBrokerService dataBrokerService;
    private ProviderContext pc;
    private BundleContext ctx;
    private DropTestProvider provider;
    private DropTestRpcProvider rpcProvider;
    private boolean on = false;
    private boolean sessionInitiated = false;


    public DropTestCommandProvider(BundleContext ctx,DropTestProvider provider,DropTestRpcProvider rpcProvider) {
        this.ctx = ctx;
        this.provider = provider;
        this.rpcProvider = rpcProvider;
    }

    public void onSessionInitiated(ProviderContext session) {
        pc = session;
        dataBrokerService = session.getSALService(DataBrokerService.class);
        ctx.registerService(CommandProvider.class.getName(), this, null);
        this.sessionInitiated = true;

    }

    public void _dropAllPackets(CommandInterpreter ci) {
        if(sessionInitiated) {
            String onoff = ci.nextArgument();
            if(onoff.equals("on")) {
                if(on == false) {
                    provider.start();
                    ci.println("DropAllFlows transitions to on");
                } else {
                    ci.println("DropAllFlows is already on");
                }
                on = true;
            } else if (onoff.equals("off")) {
                if(on == true) {
                    provider.close();
                    ci.println("DropAllFlows transitions to off");
                } else {
                    ci.println("DropAllFlows is already off");
                }
                on = false;
            }
        } else {
            ci.println("Session not initiated, try again in a few seconds");
        }
    }

    public void _dropAllPacketsRpc(CommandInterpreter ci) {
        if(sessionInitiated) {
            String onoff = ci.nextArgument();
            if(onoff.equals("on")) {
                if(on == false) {
                    rpcProvider.start();
                    ci.println("DropAllFlows transitions to on");
                } else {
                    ci.println("DropAllFlows is already on");
                }
                on = true;
            } else if (onoff.equals("off")) {
                if(on == true) {
                    rpcProvider.close();
                    ci.println("DropAllFlows transitions to off");
                } else {
                    ci.println("DropAllFlows is already off");
                }
                on = false;
            }
        } else {
            ci.println("Session not initiated, try again in a few seconds");
        }
    }
    
    public void _showDropStats(CommandInterpreter ci) {
        if(sessionInitiated) {    	
        	ci.println("RPC Test Statistics: " + this.rpcProvider.getStats().toString());
        	ci.println("FRM Test Statistics: " + this.provider.getStats().toString());
       } else {
        	ci.println("Session not initiated, try again in a few seconds");
        }
   }
    
    public void _clearDropStats(CommandInterpreter ci) {
        if(sessionInitiated) {    	
        	ci.print("Clearing drop statistics... ");
        	this.rpcProvider.clearStats();
        	this.provider.clearStats();
        	ci.println("Done.");

        } else {
        	ci.println("Session not initiated, try again in a few seconds");
        }
  }
    
    @Override
    public String getHelp() {
        String helpString = "----------------- dropAllPackets--------------\n"
                + " dropAllPackets on - begin dropping all packets \n"
                + " dropAllPackets on - stop dropping all packets \n"
                + " dropAllPacketsRpc on - begin dropping all packets but bypassing dataStore \n"
                + "                      - add flow goes directly to rpc provided OFPlugin \n"
                + " dropAllPacketsRpc on - stop dropping all packets but bypassing dataStore \n";
        return helpString;
    }
}