/*
 * Copyright (c) 2015, 2016 China Telecom Beijing Research Institute 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.ovsdb.hwvtepsouthbound.transact;

import static org.opendaylight.ovsdb.lib.operations.Operations.op;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Objects;
import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepDeviceInfo;
import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepSouthboundUtil;
import org.opendaylight.ovsdb.lib.notation.UUID;
import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
import org.opendaylight.ovsdb.schema.hardwarevtep.McastMacsRemote;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacs;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class McastMacsRemoteRemoveCommand extends AbstractTransactCommand<RemoteMcastMacs, HwvtepGlobalAugmentation> {
    private static final Logger LOG = LoggerFactory.getLogger(McastMacsRemoteRemoveCommand.class);

    public McastMacsRemoteRemoveCommand(HwvtepOperationalState state,
            Collection<DataTreeModification<Node>> changes) {
        super(state, changes);
    }

    @Override
    public void execute(TransactionBuilder transaction) {
        Map<InstanceIdentifier<Node>, List<RemoteMcastMacs>> removeds =
                extractRemoved(getChanges(),RemoteMcastMacs.class);
        if (!removeds.isEmpty()) {
            for (Entry<InstanceIdentifier<Node>, List<RemoteMcastMacs>> removed:
                removeds.entrySet()) {
                removeMcastMacRemote(transaction,  removed.getKey(), removed.getValue());
            }
        }
        //Remove the ones whose locator set got emptied
        Map<InstanceIdentifier<Node>, List<RemoteMcastMacs>> updated =
                extractUpdated(getChanges(),RemoteMcastMacs.class);
        if (!HwvtepSouthboundUtil.isEmptyMap(updated)) {
            for (Entry<InstanceIdentifier<Node>, List<RemoteMcastMacs>> entry:
                    updated.entrySet()) {
                List<RemoteMcastMacs> updatedList = entry.getValue();
                List<RemoteMcastMacs> tobeRemovedList = new ArrayList<>();
                if (!HwvtepSouthboundUtil.isEmpty(updatedList)) {
                    for (RemoteMcastMacs mac: updatedList) {
                        if (HwvtepSouthboundUtil.isEmpty(mac.getLocatorSet())) {
                            tobeRemovedList.add(mac);
                        }
                    }
                    removeMcastMacRemote(transaction, entry.getKey(), tobeRemovedList);
                }
            }
        }
    }

    private void removeMcastMacRemote(final TransactionBuilder transaction,
                                      final InstanceIdentifier<Node> nodeIid, final List<RemoteMcastMacs> macList) {
        for (RemoteMcastMacs mac : macList) {
            InstanceIdentifier<RemoteMcastMacs> macKey = nodeIid.augmentation(HwvtepGlobalAugmentation.class)
                    .child(RemoteMcastMacs.class, mac.key());
            onConfigUpdate(transaction, nodeIid, mac, macKey);
        }
    }

    @Override
    public void onConfigUpdate(final TransactionBuilder transaction,
                               final InstanceIdentifier<Node> nodeIid,
                               final RemoteMcastMacs remoteMcastMac,
                               final InstanceIdentifier macKey,
                               final Object... extraData) {
        processDependencies(null, transaction, nodeIid, macKey, remoteMcastMac);
    }

    @Override
    public void doDeviceTransaction(final TransactionBuilder transaction,
                                    final InstanceIdentifier<Node> instanceIdentifier,
                                    final RemoteMcastMacs mac,
                                    final InstanceIdentifier macIid,
                                    final Object... extraData) {
        LOG.debug("Removing remoteMcastMacs, mac address: {}", mac.getMacEntryKey().getValue());
        HwvtepDeviceInfo.DeviceData operationalMacOptional =
                getDeviceInfo().getDeviceOperData(RemoteMcastMacs.class, macIid);
        McastMacsRemote mcastMacsRemote = TyperUtils.getTypedRowWrapper(transaction.getDatabaseSchema(),
                McastMacsRemote.class, null);
        if (operationalMacOptional != null && operationalMacOptional.getUuid() != null) {
            //when mac entry is deleted, its referenced locator set and locators are deleted automatically.
            //TODO: locator in config DS is not deleted
            UUID macEntryUUID = operationalMacOptional.getUuid();
            mcastMacsRemote.getUuidColumn().setData(macEntryUUID);
            transaction.add(op.delete(mcastMacsRemote.getSchema())
                    .where(mcastMacsRemote.getUuidColumn().getSchema().opEqual(macEntryUUID)).build());
            transaction.add(op.comment("McastMacRemote: Deleting " + mac.getMacEntryKey().getValue()));
            updateCurrentTxDeleteData(RemoteMcastMacs.class, macIid, mac);
        } else {
            LOG.warn("Unable to delete remoteMcastMacs {} because it was not found in the operational store",
                    mac.getMacEntryKey().getValue());
        }
    }

    @Override
    protected List<RemoteMcastMacs> getData(HwvtepGlobalAugmentation augmentation) {
        return augmentation.getRemoteMcastMacs();
    }

    @Override
    protected boolean areEqual(RemoteMcastMacs macs1, RemoteMcastMacs macs2) {
        return macs1.key().equals(macs2.key()) && Objects.equals(macs1.getLocatorSet(), macs2.getLocatorSet());
    }

    @Override
    protected boolean isRemoveCommand() {
        return true;
    }


    @Override
    public void onCommandSucceeded() {
        //remove the refcounts of the deleted macs
        for (MdsalUpdate mdsalUpdate : updates.get(getDeviceTransaction())) {
            RemoteMcastMacs deletedMac = (RemoteMcastMacs) mdsalUpdate.getNewData();
            InstanceIdentifier<RemoteMcastMacs> macIid = mdsalUpdate.getKey();
            getDeviceInfo().removeRemoteMcast(
                    (InstanceIdentifier<LogicalSwitches>) deletedMac.getLogicalSwitchRef().getValue(), macIid);
        }
    }
}