/*
 * Copyright (c) 2016 Cisco and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.fd.hc2vpp.lisp.translate.write;


import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
import static io.fd.hc2vpp.lisp.translate.write.RemoteMappingCustomizer.LocatorListType.NEGATIVE;
import static io.fd.hc2vpp.lisp.translate.write.RemoteMappingCustomizer.LocatorListType.POSITIVE;

import com.google.common.base.Preconditions;
import io.fd.hc2vpp.common.translate.util.AddressTranslator;
import io.fd.hc2vpp.common.translate.util.FutureJVppCustomizer;
import io.fd.hc2vpp.common.translate.util.JvppReplyConsumer;
import io.fd.hc2vpp.lisp.context.util.EidMappingContext;
import io.fd.hc2vpp.lisp.translate.read.trait.MappingProducer;
import io.fd.hc2vpp.lisp.translate.util.EidTranslator;
import io.fd.honeycomb.translate.spi.write.ListWriterCustomizer;
import io.fd.honeycomb.translate.write.WriteContext;
import io.fd.honeycomb.translate.write.WriteFailedException;
import io.fd.vpp.jvpp.VppBaseCallException;
import io.fd.vpp.jvpp.core.dto.OneAddDelRemoteMapping;
import io.fd.vpp.jvpp.core.future.FutureJVppCore;
import io.fd.vpp.jvpp.core.types.OneRemoteLocator;
import java.io.IOException;
import java.util.Optional;
import java.util.concurrent.TimeoutException;
import javax.annotation.Nonnull;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.MapReplyAction;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.MappingId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.dp.subtable.grouping.remote.mappings.RemoteMapping;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.dp.subtable.grouping.remote.mappings.RemoteMappingKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.dp.subtable.grouping.remote.mappings.remote.mapping.LocatorList;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.dp.subtable.grouping.remote.mappings.remote.mapping.locator.list.NegativeMapping;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.dp.subtable.grouping.remote.mappings.remote.mapping.locator.list.PositiveMapping;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.dp.subtable.grouping.remote.mappings.remote.mapping.locator.list.positive.mapping.Rlocs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev171013.eid.table.grouping.eid.table.VniTable;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
 * Customizer for {@link RemoteMapping}
 */
public class RemoteMappingCustomizer extends FutureJVppCustomizer
        implements ListWriterCustomizer<RemoteMapping, RemoteMappingKey>, EidTranslator,
        AddressTranslator, JvppReplyConsumer, MappingProducer {

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

    private final EidMappingContext remoteMappingContext;

    public RemoteMappingCustomizer(@Nonnull final FutureJVppCore futureJvpp,
                                   @Nonnull final EidMappingContext remoteMappingContext) {
        super(futureJvpp);
        this.remoteMappingContext = remoteMappingContext;
    }

    @Override
    public void writeCurrentAttributes(InstanceIdentifier<RemoteMapping> id, RemoteMapping dataAfter,
                                       WriteContext writeContext) throws WriteFailedException {
        checkNotNull(dataAfter, "Mapping is null");
        checkNotNull(dataAfter.getEid(), "Eid is null");
        checkState(id.firstKeyOf(VniTable.class) != null, "Parent vni table not found");
        checkAllowedCombination(id, dataAfter);

        try {
            addDelRemoteMappingAndReply(true, dataAfter,
                    id.firstKeyOf(VniTable.class).getVirtualNetworkIdentifier().intValue());
        } catch (VppBaseCallException | TimeoutException | IOException e) {
            throw new WriteFailedException.CreateFailedException(id, dataAfter, e);
        }

        // after successfull adition adds mapping
        MappingId mappingId = id.firstKeyOf(RemoteMapping.class).getId();
        remoteMappingContext.addEid(mappingId, dataAfter.getEid(), writeContext.getMappingContext());
    }

    @Override
    public void updateCurrentAttributes(InstanceIdentifier<RemoteMapping> id, RemoteMapping dataBefore,
                                        RemoteMapping dataAfter, WriteContext writeContext)
            throws WriteFailedException {
        // case that happens during initialization
        checkIgnoredSubnetUpdate(dataBefore.getEid().getAddress(), dataAfter.getEid().getAddress(), LOG);
    }

    @Override
    public void deleteCurrentAttributes(InstanceIdentifier<RemoteMapping> id, RemoteMapping dataBefore,
                                        WriteContext writeContext) throws WriteFailedException {
        checkNotNull(dataBefore, "Mapping is null");
        checkNotNull(dataBefore.getEid(), "Eid is null");

        //checks whether mapping already contains such key
        MappingId mappingId = id.firstKeyOf(RemoteMapping.class).getId();
        checkState(remoteMappingContext.containsEid(mappingId, writeContext.getMappingContext()),
                "Mapping for id %s is not existing,nothing to remove", mappingId);

        try {
            addDelRemoteMappingAndReply(false, dataBefore,
                    id.firstKeyOf(VniTable.class).getVirtualNetworkIdentifier().intValue());
        } catch (VppBaseCallException | TimeoutException | IOException e) {
            throw new WriteFailedException.DeleteFailedException(id, e);
        }

        //remote mapping after successfull remove of data
        remoteMappingContext.removeEid(mappingId, writeContext.getMappingContext());
    }

    private void addDelRemoteMappingAndReply(boolean add, RemoteMapping data, int vni)
            throws VppBaseCallException, TimeoutException, IOException {

        OneAddDelRemoteMapping request = new OneAddDelRemoteMapping();

        request.isAdd = booleanToByte(add);
        request.vni = vni;
        request.eidType = (byte) getEidType(data.getEid()).getVppTypeBinding();
        request.eid = getEidAsByteArray(data.getEid());

        //this is not length of eid array,but prefix length(bad naming by vpp)
        request.eidLen = getPrefixLength(data.getEid());

        if (LocatorListType.NEGATIVE
                .equals(resolveType(data.getLocatorList()))) {
            request.action = (byte) extractAction(data.getLocatorList()).getIntValue();
        } else {
            Rlocs rlocs = extractOneRemoteLocators(data.getLocatorList());

            checkArgument(rlocs != null, "No remote locators set for Positive mapping");

            request.rlocs = rlocs.getLocator().stream()
                    .map(locator -> {
                        OneRemoteLocator remoteLocator = new OneRemoteLocator();
                        remoteLocator.addr = ipAddressToArray(locator.getAddress());
                        remoteLocator.isIp4 = booleanToByte(!isIpv6(locator.getAddress()));
                        Optional.ofNullable(locator.getPriority())
                                .ifPresent(priority -> remoteLocator.priority = priority.byteValue());
                        Optional.ofNullable(locator.getWeight())
                                .ifPresent(weight -> remoteLocator.weight = weight.byteValue());

                        return remoteLocator;
                    }).toArray(OneRemoteLocator[]::new);
            request.rlocNum = Integer.valueOf(rlocs.getLocator().size()).byteValue();
        }

        getReply(getFutureJVpp().oneAddDelRemoteMapping(request).toCompletableFuture());
    }

    private static LocatorListType resolveType(LocatorList locatorList) {
        checkNotNull(locatorList, "Locator List cannot be null");

        if (locatorList instanceof PositiveMapping) {
            return POSITIVE;
        } else {
            return NEGATIVE;
        }
    }

    private static MapReplyAction extractAction(LocatorList locatorList) {
        checkNotNull(locatorList, "Locator List cannot be null");
        Preconditions.checkArgument(NEGATIVE.equals(resolveType(locatorList)),
                "Action can be extracted only from Negative Mapping");

        return ((NegativeMapping) locatorList).getMapReply().getMapReplyAction();
    }

    private static Rlocs extractOneRemoteLocators(LocatorList locatorList) {
        checkNotNull(locatorList, "Locator List cannot be null");
        Preconditions.checkArgument(POSITIVE.equals(resolveType(locatorList)),
                "RLocs can be extracted only from Positive Mapping");

        return ((PositiveMapping) locatorList).getRlocs();
    }

    public enum LocatorListType {

        /**
         * Represents locator list as negative mapping
         */
        NEGATIVE,

        /**
         * Represents locator list as positive mapping
         */
        POSITIVE
    }

}