/*
 * Copyright (c) 2015 Cisco Systems, 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.impl.util;

import com.google.common.collect.ImmutableMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.function.Function;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import org.opendaylight.openflowplugin.openflow.md.core.extension.ExtensionResolvers;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.set.field._case.SetField;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.set.field._case.SetFieldBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.Match;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.FlowWildcardsV10;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.v10.grouping.MatchV10Builder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlowBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlowWriteActionsSetField;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlowWriteActionsSetFieldBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNotifPacketIn;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNotifPacketInBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNotifSwitchFlowRemoved;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNotifSwitchFlowRemovedBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNotifUpdateFlowStats;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchPacketInMessage;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchPacketInMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.list.grouping.ExtensionList;

public final class MatchUtil {

    private static final MacAddress ZERO_MAC_ADDRESS = new MacAddress("00:00:00:00:00:00");
    private static final Ipv4Address ZERO_IPV4_ADDRESS = new Ipv4Address("0.0.0.0");

    private static final Map<Class<? extends Match>, Function<Match, Match>> TRANSFORMERS = ImmutableMap
            .<Class<? extends Match>, Function<Match, Match>>builder()
            .put(SetField.class, match -> {
                final SetFieldBuilder matchBuilder = new SetFieldBuilder(match);

                resolveExtensions(match).ifPresent(extensionLists -> matchBuilder
                        .addAugmentation(GeneralAugMatchNodesNodeTableFlowWriteActionsSetField.class,
                                new GeneralAugMatchNodesNodeTableFlowWriteActionsSetFieldBuilder()
                                        .setExtensionList(extensionLists)
                                        .build()));

                return matchBuilder.build();
            })
            .put(org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow
                    .Match.class, (match) -> {
                    final MatchBuilder matchBuilder = new MatchBuilder(match);

                    resolveExtensions(match).ifPresent(extensionLists -> matchBuilder
                            .addAugmentation(GeneralAugMatchNotifUpdateFlowStats.class,
                                    new GeneralAugMatchNodesNodeTableFlowBuilder()
                                            .setExtensionList(extensionLists)
                                            .build()));

                    return matchBuilder.build();
                })
            .put(org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.mod.removed
                    .Match.class, (match) -> {
                    final org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.mod.removed
                            .MatchBuilder matchBuilder = new org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types
                            .rev131026.flow.mod.removed.MatchBuilder(match);

                    resolveExtensions(match).ifPresent(extensionLists -> matchBuilder
                            .addAugmentation(GeneralAugMatchNotifSwitchFlowRemoved.class,
                                    new GeneralAugMatchNotifSwitchFlowRemovedBuilder()
                                            .setExtensionList(extensionLists)
                                            .build()));

                    return matchBuilder.build();
                })
            .put(org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.packet.received
                    .Match.class, (match) -> {
                    final org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.packet.received
                            .MatchBuilder matchBuilder =
                            new org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service
                            .rev130709.packet.received.MatchBuilder(match);

                    resolveExtensions(match).ifPresent(extensionLists -> matchBuilder
                            .addAugmentation(GeneralAugMatchNotifPacketIn.class,
                                    new GeneralAugMatchNotifPacketInBuilder()
                                            .setExtensionList(extensionLists)
                                            .build()));

                    return matchBuilder.build();
                })
            .put(org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.packet.in.message
                    .Match.class, (match) -> {
                    final org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.packet.in.message
                            .MatchBuilder matchBuilder =
                            new org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service
                            .rev130709.packet.in.message.MatchBuilder(match);

                    resolveExtensions(match).ifPresent(extensionLists -> matchBuilder
                            .addAugmentation(GeneralAugMatchPacketInMessage.class,
                                    new GeneralAugMatchPacketInMessageBuilder()
                                            .setExtensionList(extensionLists)
                                            .build()));

                    return matchBuilder.build();
                })
            .build();

    private MatchUtil() {
        throw new IllegalStateException("This class should not be instantiated.");
    }


    public static MatchV10Builder createEmptyV10Match() {
        Short zeroShort = 0;
        Integer zeroInteger = 0;
        MatchV10Builder matchV10Builder = new MatchV10Builder();
        matchV10Builder.setDlDst(ZERO_MAC_ADDRESS);
        matchV10Builder.setDlSrc(ZERO_MAC_ADDRESS);
        matchV10Builder.setDlType(zeroInteger);
        matchV10Builder.setDlVlan(zeroInteger);
        matchV10Builder.setDlVlanPcp(zeroShort);
        matchV10Builder.setInPort(zeroInteger);
        matchV10Builder.setNwDst(ZERO_IPV4_ADDRESS);
        matchV10Builder.setNwDstMask(zeroShort);
        matchV10Builder.setNwProto(zeroShort);
        matchV10Builder.setNwSrc(ZERO_IPV4_ADDRESS);
        matchV10Builder.setNwSrcMask(zeroShort);
        matchV10Builder.setNwTos(zeroShort);
        matchV10Builder.setTpDst(zeroInteger);
        matchV10Builder.setTpSrc(zeroInteger);
        FlowWildcardsV10 flowWildcardsV10 =
                new FlowWildcardsV10(true, true, true, true, true, true, true, true, true, true);
        matchV10Builder.setWildcards(flowWildcardsV10);
        return matchV10Builder;
    }

    @Nullable
    public static <T extends Match> T transformMatch(@Nullable final Match match,
                                                     @Nonnull final Class<T> implementedInterface) {
        if (match == null) {
            return null;
        }

        if (implementedInterface.equals(match.implementedInterface())) {
            return implementedInterface.cast(match);
        }

        final Function<Match, Match> matchMatchFunction = TRANSFORMERS.get(implementedInterface);
        return matchMatchFunction == null ? null : implementedInterface.cast(matchMatchFunction.apply(match));
    }

    private static Optional<List<ExtensionList>> resolveExtensions(final Match match) {
        return ExtensionResolvers
                .getMatchExtensionResolver()
                .getExtension(match)
                .flatMap(matchExtension -> Optional.ofNullable(matchExtension.getExtensionList()));
    }
}