/*
 * Copyright (c) 2016 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.vpnservice.itm.cli;

import java.math.BigInteger;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.net.util.SubnetUtils;
import org.opendaylight.controller.md.sal.binding.api.DataBroker;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.vpnservice.interfacemgr.exceptions.InterfaceNotFoundException;
import org.opendaylight.vpnservice.interfacemgr.interfaces.IInterfaceManager;
//import org.opendaylight.vpnservice.interfacemgr.util.OperationalIfmUtil;
import org.opendaylight.vpnservice.itm.globals.ITMConstants;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.IfL2vlan;
import org.opendaylight.vpnservice.itm.impl.ItmUtils;
import org.opendaylight.vpnservice.mdsalutil.MDSALDataStoreUtils;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpPrefix;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.AdminStatus;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.OperStatus;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.config.rev151102.TunnelMonitorEnabled;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.config.rev151102.TunnelMonitorEnabledBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.config.rev151102.TunnelMonitorInterval;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.config.rev151102.TunnelMonitorIntervalBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.op.rev150701.TunnelList ;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.op.rev150701.tunnel.list.InternalTunnel;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.op.rev150701.tunnel.list.InternalTunnelKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.TunnelTypeBase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.TunnelTypeGre;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.IfTunnel;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.TunnelTypeVxlan;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.TransportZones;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.TransportZonesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.TransportZone;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.TransportZoneBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.TransportZoneKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.transport.zone.Subnets;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.transport.zone.SubnetsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.transport.zone.SubnetsKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.transport.zone.subnets.Vteps;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.transport.zone.subnets.VtepsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.itm.rev150701.transport.zones.transport.zone.subnets.VtepsKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Optional;
import com.google.common.base.Preconditions;


public class TepCommandHelper {

    private static final Logger LOG = LoggerFactory.getLogger(TepCommandHelper.class);
    private DataBroker dataBroker;
    static int check = 0;
    static short flag = 0;
    /*
     * boolean flag add_or_delete --- can be set to true if the last called tep
     * command is Tep-add else set to false when Tep-delete is called
     * tepCommandHelper object is created only once in session initiated
     */
    final Map<String, Map<SubnetObject, List<Vteps>>> tZones = new HashMap<String, Map<SubnetObject, List<Vteps>>>();
    private List<Subnets> subnetList = new ArrayList<Subnets>();
    private List<TransportZone> tZoneList = new ArrayList<TransportZone>();
    private List<Vteps> vtepDelCommitList = new ArrayList<Vteps>();
    private IInterfaceManager interfaceManager;

    // private List<InstanceIdentifier<? extends DataObject>> vtepPaths = new
    // ArrayList<>();


    public TepCommandHelper(final DataBroker broker) {
        this.dataBroker = broker;
    }


    public void setInterfaceManager(IInterfaceManager interfaceManager) {
        this.interfaceManager = interfaceManager;
    }

    public void createLocalCache(BigInteger dpnId, String portName, Integer vlanId, String ipAddress,
                    String subnetMask, String gatewayIp, String transportZone) {

        check++;
        IpAddress ipAddressObj = null;
        IpAddress gatewayIpObj = null;
        IpPrefix subnetMaskObj = null;
        VtepsKey vtepkey = new VtepsKey(dpnId, portName);
        try {
            ipAddressObj = new IpAddress(ipAddress.toCharArray());
            gatewayIpObj = new IpAddress("0.0.0.0".toCharArray());
            if (gatewayIp != null) {
                gatewayIpObj = new IpAddress(gatewayIp.toCharArray());
            } else {
                LOG.debug("gateway is null");
            }
        } catch (Exception e) {
            System.out.println("Invalid IpAddress. Expected: 1.0.0.0 to 254.255.255.255");
            return;
        }
        try {
            subnetMaskObj = new IpPrefix(subnetMask.toCharArray());
        } catch (Exception e) {
            System.out.println("Invalid Subnet Mask. Expected: 0.0.0.0/0 to 255.255.255.255/32");
            return;
        }

        if (!validateIPs(ipAddress, subnetMask, gatewayIp)) {
            System.out.println("IpAddress and gateWayIp should belong to the subnet provided");
            return;
        }

        if (checkTepPerTzPerDpn(transportZone, dpnId)) {
            System.out.println("Only one end point per transport Zone per Dpn is allowed");
            return;
        }
        Vteps vtepCli = new VtepsBuilder().setDpnId(dpnId).setIpAddress(ipAddressObj).setKey(vtepkey)
                .setPortname(portName).build();
        validateForDuplicates(vtepCli, transportZone);

        SubnetsKey subnetsKey = new SubnetsKey(subnetMaskObj);
        SubnetObject subObCli = new SubnetObject(gatewayIpObj, subnetsKey, subnetMaskObj, vlanId);
        if (tZones.containsKey(transportZone)) {
            Map<SubnetObject, List<Vteps>> subVtepMapTemp = (Map<SubnetObject, List<Vteps>>) tZones.get(transportZone);
            if (subVtepMapTemp.containsKey(subObCli)) { // if Subnet exists
                List<Vteps> vtepListTemp = (List<Vteps>) subVtepMapTemp.get(subObCli);
                if (vtepListTemp.contains(vtepCli)) {
                    // do nothing
                } else {
                    vtepListTemp.add(vtepCli);
                }
            } else { // subnet doesnt exist
                if (checkExistingSubnet(subVtepMapTemp, subObCli)) {
                    System.out.println("subnet with subnet mask " + subObCli.get_key() + "already exists");
                    return;
                }
                List<Vteps> vtepListTemp = new ArrayList<Vteps>();
                vtepListTemp.add(vtepCli);
                subVtepMapTemp.put(subObCli, vtepListTemp);
            }
        } else {
            List<Vteps> vtepListTemp = new ArrayList<Vteps>();
            vtepListTemp.add(vtepCli);
            Map<SubnetObject, List<Vteps>> subVtepMapTemp = new HashMap<SubnetObject, List<Vteps>>();
            subVtepMapTemp.put(subObCli, vtepListTemp);
            tZones.put(transportZone, subVtepMapTemp);
        }
    }

    private boolean validateIPs(String ipAddress, String subnetMask, String gatewayIp) {
        SubnetUtils utils = new SubnetUtils(subnetMask);
        if ((utils.getInfo().isInRange(ipAddress)) && ((gatewayIp == null) || (utils.getInfo().isInRange(gatewayIp)))) {
            return true;
        } else {
            LOG.trace("InValid IP");
            return false;
        }
    }

    /**
     * Validate for duplicates.
     *
     * @param inputVtep
     *            the input vtep
     * @param transportZone
     *            the transport zone
     */
    public void validateForDuplicates(Vteps inputVtep, String transportZone) {
        Map<String, TransportZone> tZoneMap = getAllTransportZonesAsMap();

        boolean isConfiguredTepGreType = isGreTunnelType(transportZone, tZoneMap);
        // Checking for duplicates in local cache
        for (String tZ : tZones.keySet()) {
            boolean isGreType = isGreTunnelType(tZ, tZoneMap);
            Map<SubnetObject, List<Vteps>> subVtepMapTemp = (Map<SubnetObject, List<Vteps>>) tZones.get(tZ);
            for (SubnetObject subOb : subVtepMapTemp.keySet()) {
                List<Vteps> vtepList = subVtepMapTemp.get(subOb);
                validateForDuplicateAndSingleGreTep(inputVtep, isConfiguredTepGreType, isGreType, vtepList);
            }
        }
        // Checking for duplicates in config DS
        for (TransportZone tZ : tZoneMap.values()) {
            boolean isGreType = false;
            if (tZ.getTunnelType().equals(TunnelTypeGre.class)) {
                isGreType = true;
            }
            for (Subnets sub : ItmUtils.emptyIfNull(tZ.getSubnets())) {
                List<Vteps> vtepList = sub.getVteps();
                validateForDuplicateAndSingleGreTep(inputVtep, isConfiguredTepGreType, isGreType, vtepList);
            }
        }
    }

    private void validateForDuplicateAndSingleGreTep(Vteps inputVtep, boolean isConfiguredTepGreType, boolean isGreType,
            List<Vteps> vtepList) {
        if (ItmUtils.isEmpty(vtepList)) {
            return;
        }
        if (vtepList.contains(inputVtep)) {
            Preconditions.checkArgument(false, "VTEP already exists");
        }
        BigInteger dpnId = inputVtep.getDpnId();
        if (isConfiguredTepGreType && isGreType) {
            for (Vteps vtep : vtepList) {
                if (vtep.getDpnId().equals(dpnId)) {
                    String errMsg = new StringBuilder("DPN [").append(dpnId)
                            .append("] already configured with GRE TEP. Mutiple GRE TEP's on a single DPN are not allowed.")
                            .toString();
                    Preconditions.checkArgument(false, errMsg);
                }
            }
        }
    }

    /**
     * Gets all transport zones as map.
     *
     * @return all transport zones as map
     */
    private Map<String, TransportZone> getAllTransportZonesAsMap() {
        TransportZones tZones = getAllTransportZones();
        Map<String, TransportZone> tZoneMap = new HashMap<>();
        if( null != tZones) {
           for (TransportZone tzone : ItmUtils.emptyIfNull(tZones.getTransportZone())) {
             tZoneMap.put(tzone.getZoneName(), tzone);
           }
        }
        return tZoneMap;
    }

    /**
     * Checks if is gre tunnel type.
     *
     * @param tZoneName
     *            the zone name
     * @param tZoneMap
     *            the zone map
     * @return true, if is gre tunnel type
     */
    private boolean isGreTunnelType(String tZoneName, Map<String, TransportZone> tZoneMap) {
        TransportZone tzone = tZoneMap.get(tZoneName);
        /*
        if (tzone != null && StringUtils.equalsIgnoreCase(ITMConstants.TUNNEL_TYPE_GRE, tzone.getTunnelType())) {
            return true;
        }
        */
        if( (tzone != null) && (tzone.getTunnelType()).equals(TunnelTypeGre.class) ) {
           return true;
        }
        return false;
    }

    /**
     * Gets the transport zone.
     *
     * @param tzone
     *            the tzone
     * @return the transport zone
     */
    public TransportZone getTransportZone(String tzone) {
        InstanceIdentifier<TransportZone> tzonePath = InstanceIdentifier.builder(TransportZones.class)
                .child(TransportZone.class, new TransportZoneKey(tzone)).build();
        Optional<TransportZone> tZoneOptional = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, tzonePath,
                dataBroker);
        if (tZoneOptional.isPresent()) {
            return tZoneOptional.get();
        }
        return null;
    }

    /**
     * Gets the transport zone from config ds.
     *
     * @param tzone
     *            the tzone
     * @return the transport zone
     */
    public TransportZone getTransportZoneFromConfigDS(String tzone) {
        InstanceIdentifier<TransportZone> tzonePath = InstanceIdentifier.builder(TransportZones.class)
                .child(TransportZone.class, new TransportZoneKey(tzone)).build();
        Optional<TransportZone> tZoneOptional = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, tzonePath,
                dataBroker);
        if (tZoneOptional.isPresent()) {
            return tZoneOptional.get();
        }
        return null;
    }

    /**
     * Gets all transport zones.
     *
     * @return all transport zones
     */
    public TransportZones getAllTransportZones() {
        InstanceIdentifier<TransportZones> path = InstanceIdentifier.builder(TransportZones.class).build();
        Optional<TransportZones> tZonesOptional = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, path, dataBroker);
        if (tZonesOptional.isPresent()) {
            return tZonesOptional.get();
        }
        return null;
    }

    public boolean checkExistingSubnet(Map<SubnetObject, List<Vteps>> subVtepMapTemp, SubnetObject subObCli) {
        for (SubnetObject subOb : subVtepMapTemp.keySet()) {
            if (subOb.get_key().equals(subObCli.get_key())) {
                if (!(subOb.get_vlanId().equals(subObCli.get_vlanId())))
                    return true;
                if (!(subOb.get_gatewayIp().equals(subObCli.get_gatewayIp())))
                    return true;
            }
        }
        return false;
    }

    public boolean checkTepPerTzPerDpn(String tzone, BigInteger dpnId) {
        // check in local cache
        if (tZones.containsKey(tzone)) {
            Map<SubnetObject, List<Vteps>> subVtepMapTemp = (Map<SubnetObject, List<Vteps>>) tZones.get(tzone);
            for (SubnetObject subOb : subVtepMapTemp.keySet()) {
                List<Vteps> vtepList = subVtepMapTemp.get(subOb);
                for (Vteps vtep : vtepList)
                    if (vtep.getDpnId().equals(dpnId))
                        return true;
            }
        }

        // check in DS
        InstanceIdentifier<TransportZone> tzonePath =
                        InstanceIdentifier.builder(TransportZones.class)
                                        .child(TransportZone.class, new TransportZoneKey(tzone)).build();
        Optional<TransportZone> tZoneOptional =
                        ItmUtils.read(LogicalDatastoreType.CONFIGURATION, tzonePath, dataBroker);
        if (tZoneOptional.isPresent()) {
            TransportZone tz = tZoneOptional.get();
            if (tz.getSubnets() == null || tz.getSubnets().isEmpty())
                return false;
            for (Subnets sub : tz.getSubnets()) {
                if (sub.getVteps() == null || sub.getVteps().isEmpty())
                    continue;
                for (Vteps vtep : sub.getVteps()) {
                    if (vtep.getDpnId().equals(dpnId)) {
                        return true;
                    }
                }
            }
        }
        return false;
    }

    public void buildTeps() {
        TransportZones tZonesBuilt = null;
        TransportZone tZone = null;
        try {
            LOG.debug("no of teps added" + check);
            if (tZones != null || !tZones.isEmpty()) {
                tZoneList = new ArrayList<TransportZone>();
                for (String tZ : tZones.keySet()) {
                    LOG.debug("tZones" + tZ);
                    subnetList = new ArrayList<Subnets>();
                    Map<SubnetObject, List<Vteps>> subVtepMapTemp = (Map<SubnetObject, List<Vteps>>) tZones.get(tZ);
                    for (SubnetObject subOb : subVtepMapTemp.keySet()) {
                        LOG.debug("subnets" + subOb.get_prefix());
                        List<Vteps> vtepList = subVtepMapTemp.get(subOb);
                        Subnets subnet =
                                        new SubnetsBuilder().setGatewayIp(subOb.get_gatewayIp())
                                                        .setKey(subOb.get_key()).setPrefix(subOb.get_prefix())
                                                        .setVlanId(subOb.get_vlanId()).setVteps(vtepList).build();
                        subnetList.add(subnet);
                        LOG.debug("vteps" + vtepList);
                    }
                    InstanceIdentifier<TransportZone> tZonepath =
                                    InstanceIdentifier.builder(TransportZones.class)
                                                    .child(TransportZone.class, new TransportZoneKey(tZ)).build();
                    Optional<TransportZone> tZoneOptional =
                                    ItmUtils.read(LogicalDatastoreType.CONFIGURATION, tZonepath, dataBroker);
                    LOG.debug("read container from DS");
                    if (tZoneOptional.isPresent()) {
                        TransportZone tzoneFromDs = tZoneOptional.get();
                        LOG.debug("read tzone container" + tzoneFromDs.toString());
                        if (tzoneFromDs.getTunnelType() == null
                                        || (tzoneFromDs.getTunnelType()).equals(TunnelTypeVxlan.class)) {
                            tZone =
                                            new TransportZoneBuilder().setKey(new TransportZoneKey(tZ))
                                                            .setTunnelType(TunnelTypeVxlan.class).setSubnets(subnetList)
                                                            .setZoneName(tZ).build();
                        } else if ((tzoneFromDs.getTunnelType()).equals(TunnelTypeGre.class)) {
                            tZone =
                                            new TransportZoneBuilder().setKey(new TransportZoneKey(tZ))
                                                            .setTunnelType(TunnelTypeGre.class).setSubnets(subnetList)
                                                            .setZoneName(tZ).build();
                        }
                    } else {
                        tZone =
                                        new TransportZoneBuilder().setKey(new TransportZoneKey(tZ))
                                                        .setTunnelType(TunnelTypeVxlan.class).setSubnets(subnetList).setZoneName(tZ)
                                                        .build();
                    }
                    LOG.debug("tzone object" + tZone);
                    tZoneList.add(tZone);
                }
                tZonesBuilt = new TransportZonesBuilder().setTransportZone(tZoneList).build();
                InstanceIdentifier<TransportZones> path = InstanceIdentifier.builder(TransportZones.class).build();
                LOG.debug("InstanceIdentifier" + path);
                ItmUtils.asyncUpdate(LogicalDatastoreType.CONFIGURATION, path, tZonesBuilt, dataBroker,
                                ItmUtils.DEFAULT_CALLBACK);
                LOG.debug("wrote to Config DS" + tZonesBuilt);
                tZones.clear();
                tZoneList.clear();
                subnetList.clear();
                LOG.debug("Everything cleared");
            } else {
                LOG.debug("NO vteps were configured");
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void showTeps(boolean monitorEnabled, int monitorInterval) {
        boolean flag = false;
        InstanceIdentifier<TransportZones> path = InstanceIdentifier.builder(TransportZones.class).build();
        Optional<TransportZones> tZonesOptional = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, path, dataBroker);
        if (tZonesOptional.isPresent()) {
            TransportZones tZones = tZonesOptional.get();
            if(tZones.getTransportZone() == null || tZones.getTransportZone().isEmpty())
            {
                System.out.println("No teps configured");
                return;
            }
            List<String> result = new ArrayList<String>();
            result.add(String.format("Tunnel Monitoring (for VXLAN tunnels): %s", (monitorEnabled ? "On" : "Off")));
            result.add(String.format("Tunnel Monitoring Interval (for VXLAN tunnels): %d", monitorInterval));
            result.add(System.lineSeparator());
            result.add(String.format("%-16s  %-16s  %-16s  %-12s  %-12s %-12s %-16s %-12s", "TransportZone", "TunnelType", "SubnetMask",
                            "GatewayIP", "VlanID", "DpnID", "IPAddress", "PortName"));
            result.add("------------------------------------------------------------------------------------------------------------------------------");
            for (TransportZone tZ : tZones.getTransportZone()) {
                if (tZ.getSubnets() == null || tZ.getSubnets().isEmpty()) {
                    LOG.error("Transport Zone " + tZ.getZoneName() + "has no subnets");
                    continue;
                }
                for (Subnets sub : tZ.getSubnets()) {
                    if (sub.getVteps() == null || sub.getVteps().isEmpty()) {
                        LOG.error("Transport Zone " + tZ.getZoneName() + "subnet " + sub.getPrefix() + "has no vteps");
                        continue;
                    }
                    for (Vteps vtep : sub.getVteps()) {
                        flag = true;
                        String strTunnelType ;
                        if( (tZ.getTunnelType()).equals(TunnelTypeGre.class) )
                          strTunnelType = ITMConstants.TUNNEL_TYPE_GRE ;
                        else
                          strTunnelType = ITMConstants.TUNNEL_TYPE_VXLAN ;
                        result.add(String.format("%-16s  %-16s  %-16s  %-12s  %-12s %-12s %-16s %-12s", tZ.getZoneName(), strTunnelType, sub
                                        .getPrefix().getIpv4Prefix().getValue(), sub.getGatewayIp().getIpv4Address()
                                        .getValue(), sub.getVlanId().toString(), vtep.getDpnId().toString(), vtep
                                        .getIpAddress().getIpv4Address().getValue(), vtep.getPortname().toString()));
                    }
                }
            }
            if (flag == true) {
                for (String p : result) {
                    System.out.println(p);
                }
            } else
                System.out.println("No teps to display");
        } else
            System.out.println("No teps configured");
    }


    public void deleteVtep(BigInteger dpnId, String portName, Integer vlanId, String ipAddress, String subnetMask,
                    String gatewayIp, String transportZone) {

        IpAddress ipAddressObj = null;
        IpAddress gatewayIpObj = null;
        IpPrefix subnetMaskObj = null;
        VtepsKey vtepkey = new VtepsKey(dpnId, portName);
        try {
            ipAddressObj = new IpAddress(ipAddress.toCharArray());
            gatewayIpObj = new IpAddress("0.0.0.0".toCharArray());
            if (gatewayIp != null) {
                gatewayIpObj = new IpAddress(gatewayIp.toCharArray());
            } else {
                LOG.debug("gateway is null");
            }
        } catch (Exception e) {
            System.out.println("Invalid IpAddress. Expected: 1.0.0.0 to 254.255.255.255");
            return;
        }
        try {
            subnetMaskObj = new IpPrefix(subnetMask.toCharArray());
        } catch (Exception e) {
            System.out.println("Invalid Subnet Mask. Expected: 0.0.0.0/0 to 255.255.255.255/32");
            return;
        }

        if (!validateIPs(ipAddress, subnetMask, gatewayIp)) {
            System.out.println("IpAddress and gateWayIp should belong to the subnet provided");
            return;
        }
        SubnetsKey subnetsKey = new SubnetsKey(subnetMaskObj);
        Vteps vtepCli = null;
        Subnets subCli = null;

        InstanceIdentifier<Vteps> vpath =
                        InstanceIdentifier.builder(TransportZones.class)
                                        .child(TransportZone.class, new TransportZoneKey(transportZone))
                                        .child(Subnets.class, subnetsKey).child(Vteps.class, vtepkey).build();
 
        // check if present in tzones and delete from cache
        boolean existsInCache = isInCache(dpnId, portName, vlanId, ipAddress, subnetMask, gatewayIp, transportZone);
        if (!existsInCache) {
            Optional<Vteps> vtepOptional = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, vpath, dataBroker);
            if (vtepOptional.isPresent()) {
                vtepCli = vtepOptional.get();
                if(vtepCli.getIpAddress().equals(ipAddressObj)){
                    InstanceIdentifier<Subnets> spath =
                                    InstanceIdentifier
                                                    .builder(TransportZones.class)
                                                    .child(TransportZone.class, new TransportZoneKey(transportZone))
                                                    .child(Subnets.class, subnetsKey).build();
                    Optional<Subnets> subOptional = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, spath, dataBroker);
                    if (subOptional.isPresent()) {
                        subCli = subOptional.get();
                        if(subCli.getGatewayIp().equals(gatewayIpObj) && subCli.getVlanId().equals(vlanId)){
                    vtepDelCommitList.add(vtepCli);
                      }
                        else
                            System.out.println(String.format("vtep with this vlan or gateway doesnt exist"));
                        }
                }
                else 
                    System.out.println(String.format("Vtep with this ipaddress doesnt exist"));
                } else {
                System.out.println(String.format("Vtep Doesnt exist"));
            }
        }
    }

    public <T extends DataObject> void deleteOnCommit() {
        List<InstanceIdentifier<T>> vtepPaths = new ArrayList<>();
        List<InstanceIdentifier<T>> subnetPaths = new ArrayList<>();
        List<InstanceIdentifier<T>> tzPaths = new ArrayList<>();
        List<Subnets> subDelList = new ArrayList<Subnets>();
        List<TransportZone> tzDelList = new ArrayList<TransportZone>();
        List<Vteps> vtepDelList = new ArrayList<Vteps>();
        List<InstanceIdentifier<T>> allPaths = new ArrayList<>();
        try {
            if (vtepDelCommitList != null && !vtepDelCommitList.isEmpty()) {
                InstanceIdentifier<TransportZones> path = InstanceIdentifier.builder(TransportZones.class).build();
                Optional<TransportZones> tZonesOptional =
                                ItmUtils.read(LogicalDatastoreType.CONFIGURATION, path, dataBroker);
                if (tZonesOptional.isPresent()) {
                    TransportZones tZones = tZonesOptional.get();
                    for (TransportZone tZ : tZones.getTransportZone()) {
                        if (tZ.getSubnets() == null || tZ.getSubnets().isEmpty())
                            continue;
                        for (Subnets sub : tZ.getSubnets()) {
                            vtepDelList.addAll(vtepDelCommitList);
                            for (Vteps vtep : vtepDelList) {
                                InstanceIdentifier<T> vpath =
                                                (InstanceIdentifier<T>) InstanceIdentifier
                                                                .builder(TransportZones.class)
                                                                .child(TransportZone.class, tZ.getKey())
                                                                .child(Subnets.class, sub.getKey())
                                                                .child(Vteps.class, vtep.getKey()).build();
                                if (sub.getVteps().remove(vtep)) {
                                    vtepPaths.add(vpath);
                                    if (sub.getVteps().size() == 0 || sub.getVteps() == null) {
                                        subDelList.add(sub);
                                    }

                                }
                            }
                        }
                    }

                    for (TransportZone tZ : tZones.getTransportZone()) {
                        if (tZ.getSubnets() == null || tZ.getSubnets().isEmpty())
                            continue;
                        for (Subnets sub : subDelList) {
                            if (tZ.getSubnets().remove(sub)) {
                                InstanceIdentifier<T> spath =
                                                (InstanceIdentifier<T>) InstanceIdentifier
                                                                .builder(TransportZones.class)
                                                                .child(TransportZone.class, tZ.getKey())
                                                                .child(Subnets.class, sub.getKey()).build();
                                subnetPaths.add(spath);
                                if (tZ.getSubnets() == null || tZ.getSubnets().size() == 0) {
                                    tzDelList.add(tZ);
                                }
                            }
                        }
                    }

                    for (TransportZone tZ : tzDelList) {
                        if (tZones.getTransportZone().remove(tZ)) {
                            InstanceIdentifier<T> tpath =
                                            (InstanceIdentifier<T>) InstanceIdentifier.builder(TransportZones.class)
                                                            .child(TransportZone.class, tZ.getKey()).build();
                            tzPaths.add(tpath);
                            if (tZones.getTransportZone() == null || tZones.getTransportZone().size() == 0) {
                                MDSALDataStoreUtils.asyncRemove(dataBroker, LogicalDatastoreType.CONFIGURATION, path,
                                                ItmUtils.DEFAULT_CALLBACK);
                                return;
                            }
                        }
                    }
                    allPaths.addAll(vtepPaths);
                    allPaths.addAll(subnetPaths);
                    allPaths.addAll(tzPaths);
                    ItmUtils.asyncBulkRemove(dataBroker, LogicalDatastoreType.CONFIGURATION, allPaths,
                                    ItmUtils.DEFAULT_CALLBACK);
                }
                vtepPaths.clear();
                subnetPaths.clear();
                tzPaths.clear();
                allPaths.clear();
                vtepDelCommitList.clear();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void showState(TunnelList tunnels, boolean tunnelMonitorEnabled) {
        IfTunnel tunnelInterface = null;
        IfL2vlan l2Vlan = null;
        List<InternalTunnel> tunnelLists = tunnels.getInternalTunnel();
        if (tunnelLists == null || tunnelLists.isEmpty()) {
            System.out.println("No Internal Tunnels Exist");
            return;
        }
        if (!tunnelMonitorEnabled) {
            System.out.println("Tunnel Monitoring is Off");
        }
        String displayFormat = "%-16s  %-16s  %-16s  %-16s  %-16s  %-8s  %-10s  %-10s";
        System.out.println(String.format(displayFormat, "Tunnel Name", "Source-DPN",
                        "Destination-DPN", "Source-IP", "Destination-IP", "Vlan Id", "Trunk-State", "Transport Type"));
        System.out.println("-------------------------------------------------------------------------------------------------------------------------------------");

        for (InternalTunnel tunnel : tunnelLists) {
            String tunnelInterfaceName = tunnel.getTunnelInterfaceName();
            LOG.trace("tunnelInterfaceName::: {}", tunnelInterfaceName);
            
            InstanceIdentifier<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> ifStateId =
                    ItmUtils.buildStateInterfaceId(tunnelInterfaceName);
            Optional<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> ifStateOptional =
                  ItmUtils.read(LogicalDatastoreType.OPERATIONAL, ifStateId, dataBroker);
            String tunnelState = "DOWN" ;
            if (ifStateOptional.isPresent()) {
                org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface iface = ifStateOptional.get() ;
                if(iface.getAdminStatus() == AdminStatus.Up && iface.getOperStatus() == OperStatus.Up)
                tunnelState = "UP" ;
            }
                InstanceIdentifier<Interface> trunkIdentifier = ItmUtils.buildId(tunnelInterfaceName);
                Optional<Interface> ifaceObj = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, trunkIdentifier, dataBroker) ;
                if (ifaceObj.isPresent()) {
                    l2Vlan = (IfL2vlan) ifaceObj.get().getAugmentation(IfL2vlan.class);
                    tunnelInterface = (IfTunnel) ifaceObj.get().getAugmentation(IfTunnel.class);
                }

                Class<? extends TunnelTypeBase> tunType = tunnelInterface.getTunnelInterfaceType();
                String tunnelType = ITMConstants.TUNNEL_TYPE_VXLAN;
                if( tunType.equals(TunnelTypeVxlan.class))
                    tunnelType = ITMConstants.TUNNEL_TYPE_VXLAN ;
                else if( tunType.equals(TunnelTypeGre.class) )
                    tunnelType = ITMConstants.TUNNEL_TYPE_GRE ;
                int vlanId = 0;
                if( l2Vlan != null ) {
                   vlanId = l2Vlan.getVlanId().getValue() ;
                }
                System.out.println(String.format(displayFormat, tunnel.getTunnelInterfaceName(), tunnel
                                .getSourceDPN().toString(), tunnel.getDestinationDPN().toString(), tunnelInterface.getTunnelSource().getIpv4Address().getValue(), tunnelInterface.getTunnelDestination().getIpv4Address().getValue(),vlanId, tunnelState ,
                                tunnelType));
         }
    }

    // deletes from ADD-cache if it exists.
    public boolean isInCache(BigInteger dpnId, String portName, Integer vlanId, String ipAddress, String subnetMask,
                    String gatewayIp, String transportZone) {
        boolean exists = false;
        VtepsKey vtepkey = new VtepsKey(dpnId, portName);
        IpAddress ipAddressObj = new IpAddress(ipAddress.toCharArray());
        IpPrefix subnetMaskObj = new IpPrefix(subnetMask.toCharArray());
        IpAddress gatewayIpObj = new IpAddress("0.0.0.0".toCharArray());
        if (gatewayIp != null) {
            gatewayIpObj = new IpAddress(gatewayIp.toCharArray());
        } else {
            LOG.debug("gateway is null");
        }
        SubnetsKey subnetsKey = new SubnetsKey(subnetMaskObj);
        Vteps vtepCli =
                        new VtepsBuilder().setDpnId(dpnId).setIpAddress(ipAddressObj).setKey(vtepkey)
                                        .setPortname(portName).build();
        SubnetObject subObCli = new SubnetObject(gatewayIpObj, subnetsKey, subnetMaskObj, vlanId);

        if (tZones.containsKey(transportZone)) {
            Map<SubnetObject, List<Vteps>> subVtepMapTemp = (Map<SubnetObject, List<Vteps>>) tZones.get(transportZone);
            if (subVtepMapTemp.containsKey(subObCli)) { // if Subnet exists
                List<Vteps> vtepListTemp = (List<Vteps>) subVtepMapTemp.get(subObCli);
                if (vtepListTemp.contains(vtepCli)) {
                    exists = true; // return true if tzones has vtep
                    vtepListTemp.remove(vtepCli);
                    if (vtepListTemp.size() == 0) {
                        subVtepMapTemp.remove(subObCli);
                        if (subVtepMapTemp.size() == 0) {
                            tZones.remove(transportZone);
                        }
                    }
                } else {
                    System.out.println("Vtep " + "has not been configured");
                }
            }
        }
        return exists;
    }

    public void configureTunnelType(String tZoneName, String tunnelType) {
        LOG.debug("configureTunnelType {} for transportZone {}", tunnelType, tZoneName);

        TransportZone tZoneFromConfigDS = getTransportZoneFromConfigDS(tZoneName);
        validateTunnelType(tZoneName, tunnelType,tZoneFromConfigDS);

        if (tZoneFromConfigDS != null) {
            LOG.debug("Transport zone {} with tunnel type {} already exists. No action required.", tZoneName,
                    tunnelType);
            return;
        }
        TransportZones transportZones = null;
        List<TransportZone> tZoneList = null;
        InstanceIdentifier<TransportZones> path = InstanceIdentifier.builder(TransportZones.class).build();
        Optional<TransportZones> tZones = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, path, dataBroker);
        Class<? extends TunnelTypeBase> tunType = TunnelTypeVxlan.class;
        if( tunnelType.equals(ITMConstants.TUNNEL_TYPE_VXLAN))
            tunType = TunnelTypeVxlan.class ;
        else if( tunnelType.equals(ITMConstants.TUNNEL_TYPE_GRE) )
            tunType = TunnelTypeGre.class ;

        TransportZone tZone = new TransportZoneBuilder().setKey(new TransportZoneKey(tZoneName))
                .setTunnelType(tunType).build();
        if (tZones.isPresent()) {
            tZoneList = tZones.get().getTransportZone();
            if(tZoneList == null || tZoneList.isEmpty()) {
                tZoneList = new ArrayList<TransportZone>();
            }
        } else {
            tZoneList = new ArrayList<TransportZone>();
        }
        tZoneList.add(tZone);
        transportZones = new TransportZonesBuilder().setTransportZone(tZoneList).build();
        ItmUtils.syncWrite(LogicalDatastoreType.CONFIGURATION, path, transportZones, dataBroker);

    }

    /**
     * Validate tunnel type.
     *
     * @param tZoneName
     *            the t zone name
     * @param tunnelType
     *            the tunnel type
     */
    private void validateTunnelType(String tZoneName, String tunnelType,TransportZone tZoneFromConfigDS) {
        /*
        String strTunnelType = ItmUtils.validateTunnelType(tunnelType);

        TransportZone tZone = getTransportZone(tZoneName);
        if (tZone != null) {
            if (!StringUtils.equalsIgnoreCase(strTunnelType, tZone.getTunnelType())
                    && ItmUtils.isNotEmpty(tZone.getSubnets())) {
                String errorMsg = new StringBuilder("Changing the tunnel type from ").append(tZone.getTunnelType())
                        .append(" to ").append(strTunnelType)
                        .append(" is not allowed for already configured transport zone [").append(tZoneName)
                        .append("].").toString();
                Preconditions.checkArgument(false, errorMsg);
            }
        }
        */
        String strTunnelType = ItmUtils.validateTunnelType(tunnelType);
        Class<? extends TunnelTypeBase> tunType ;
        if( strTunnelType.equals(ITMConstants.TUNNEL_TYPE_VXLAN))
            tunType = TunnelTypeVxlan.class ;
        else 
            tunType = TunnelTypeGre.class ;
        //TransportZone tZone = getTransportZone(tZoneName);
       // if (tZone != null) {
        if (tZoneFromConfigDS != null) {  
        if( (!tZoneFromConfigDS.getTunnelType().equals(tunType))  && ItmUtils.isNotEmpty(tZoneFromConfigDS.getSubnets())) {
              String errorMsg = new StringBuilder("Changing the tunnel type from ").append(tZoneFromConfigDS.getTunnelType())
                       .append(" to ").append(strTunnelType)
                       .append(" is not allowed for already configured transport zone [").append(tZoneName)
                       .append("].").toString();
               Preconditions.checkArgument(false, errorMsg);
           }
        }
    }

    public void configureTunnelMonitorEnabled(boolean monitorEnabled) {
        InstanceIdentifier<TunnelMonitorEnabled> path = InstanceIdentifier.builder(TunnelMonitorEnabled.class).build();
        Optional<TunnelMonitorEnabled> storedTunnelMonitor = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, path, dataBroker);
        if (!storedTunnelMonitor.isPresent() || storedTunnelMonitor.get().isEnabled() != monitorEnabled) {
            TunnelMonitorEnabled tunnelMonitor = new TunnelMonitorEnabledBuilder().setEnabled(monitorEnabled).build();
            ItmUtils.asyncUpdate(LogicalDatastoreType.CONFIGURATION, path, tunnelMonitor, dataBroker,
                                            ItmUtils.DEFAULT_CALLBACK);
        }
    }

    public void configureTunnelMonitorInterval(int interval) {
        InstanceIdentifier<TunnelMonitorInterval> path = InstanceIdentifier.builder(TunnelMonitorInterval.class).build();
        Optional<TunnelMonitorInterval> storedTunnelMonitor = ItmUtils.read(LogicalDatastoreType.CONFIGURATION, path, dataBroker);
        if (!storedTunnelMonitor.isPresent() || storedTunnelMonitor.get().getInterval() != interval) {
            TunnelMonitorInterval tunnelMonitor = new TunnelMonitorIntervalBuilder().setInterval(interval).build();
            ItmUtils.asyncUpdate(LogicalDatastoreType.CONFIGURATION, path, tunnelMonitor, dataBroker,
                                            ItmUtils.DEFAULT_CALLBACK);
        }
    }
}