/*
 * Copyright (c) 2017 Ericsson India Global Services Pvt Ltd. 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.netvirt.bgpmanager.commands;

import org.apache.karaf.shell.commands.Argument;
import org.apache.karaf.shell.commands.Command;
import org.apache.karaf.shell.commands.Option;
import org.apache.karaf.shell.console.OsgiCommandSupport;
import org.omg.CORBA.Object;
import org.opendaylight.netvirt.bgpmanager.BgpManager;
import org.opendaylight.netvirt.bgpmanager.thrift.gen.af_afi;
import org.opendaylight.netvirt.bgpmanager.thrift.gen.af_safi;

@Command(scope = "odl", name = "multipath", description = "Enable/Disable multipaths")
public class Multipath extends OsgiCommandSupport {

    private static final String AF = "--address-family";
    private static final String RD = "--rd";
    private static final String MAXPATH = "--maxpath";
    private static final int MIN_MAXPATH = 1;
    private static final int MAX_MAXPATH = 64;

    @Option(name = RD, aliases = { "-r" },
            description = "rd",
            required = false,
            multiValued = false)
    String rd;

    @Option(name = MAXPATH, aliases = { "-n" },
            description = "max number of paths",
            required = false,
            multiValued = false)
    String maxpath;

    @Option(name = AF, aliases = {"-f"},
            description = "Address family",
            required = false, multiValued = false)

    String addrFamily;


    @Argument(name = "enable|disable|setmaxpath",
            description = "The desired operation",
            required = true, multiValued = false)

    String multipathEnable;

    private final BgpManager bgpManager;

    public Multipath(BgpManager bgpManager) {
        this.bgpManager = bgpManager;
    }

    @Override
    protected Object doExecute() throws Exception {
        af_afi afi = af_afi.findByValue(1);
        af_safi safi = af_safi.findByValue(5);

        if (addrFamily != null) {
            if (!addrFamily.equals("lu") && !addrFamily.equals("vpnv6")
                 && !addrFamily.equals("evpn") && !addrFamily.equals("vpnv4")) {
                session.getConsole().println("error: " + AF + " must be lu/evpn/vpnv4/vpnv6 ");
                return null;
            }
            if (addrFamily.equals("vpnv6")) {
                afi = af_afi.findByValue(2);
                safi = af_safi.findByValue(5);
            } else if (addrFamily.equals("evpn")) {
                afi = af_afi.findByValue(3);
                safi = af_safi.findByValue(6);
            } else if (addrFamily.equals("lu")) {
                afi = af_afi.findByValue(1);
                safi = af_safi.findByValue(4);
            } else { // vpnv4
                afi = af_afi.findByValue(1);
                safi = af_safi.findByValue(5);
            }
        }

        if (maxpath != null) {
            int imaxpath = Integer.parseInt(maxpath);
            if (imaxpath < MIN_MAXPATH || imaxpath > MAX_MAXPATH) {
                session.getConsole().println("error: " + MAXPATH + " range[" + MIN_MAXPATH + " - " + MAX_MAXPATH + "]");
                return null;
            }
        }

        if (multipathEnable != null) {

            switch (multipathEnable) {
                case "enable":
                    bgpManager.enableMultipath(afi, safi);
                    break;
                case "disable":
                    bgpManager.disableMultipath(afi, safi);
                    break;
                case "setmaxpath":
                    if (rd != null && maxpath != null) {
                        bgpManager.multipaths(rd, Integer.parseInt(maxpath));
                    }
                    break;

                default:
                    return usage();
            }
        }

        return null;
    }

    private Object usage() {
        session.getConsole().println("odl:multipath  -f lu <enable|disable> \n"
                + "odl:multipath -f lu -r <rd> -n <maxpath> setmaxpath");
        return null;
    }
}