/*
 * Copyright (c) 2016 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.openflow.md.core.sal.convertor;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import org.opendaylight.openflowplugin.extension.api.path.ActionPath;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.action.data.ActionResponseConvertorData;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.common.Convertor;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.data.VersionConvertorData;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.Action;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.BucketId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupTypes;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.buckets.Bucket;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.buckets.BucketBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.buckets.BucketKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.desc.stats.reply.GroupDescStats;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.desc.stats.reply.GroupDescStatsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.desc.stats.reply.GroupDescStatsKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.buckets.grouping.BucketsList;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.desc._case.multipart.reply.group.desc.GroupDesc;
import org.opendaylight.yangtools.yang.binding.util.BindingMap;
import org.opendaylight.yangtools.yang.common.Uint32;
import org.opendaylight.yangtools.yang.common.Uint8;

/**
 * Converts GroupDesc message from library to MD SAL defined GroupDescStats.
 *
 * <p>
 * Example usage:
 * <pre>
 * {@code
 * VersionConvertorData data = new VersionConvertorData(version);
 * Optional<List<GroupDescStats>> salGroupStats = convertorManager.convert(ofGroupStats, data);
 * }
 * </pre>
 */
public class GroupDescStatsResponseConvertor extends Convertor<List<GroupDesc>, List<GroupDescStats>,
        VersionConvertorData> {

    private static final Set<Class<?>> TYPES = Set.of(GroupDesc.class);

    private org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.Buckets toSALBucketsDesc(
            final List<BucketsList> bucketDescStats, final Uint8 version) {
        final ActionResponseConvertorData data = new ActionResponseConvertorData(version);
        data.setActionPath(ActionPath.GROUP_DESC_STATS_UPDATED_BUCKET_ACTION);

        BindingMap.Builder<BucketKey, Bucket> allBuckets = BindingMap.orderedBuilder();
        int bucketKey = 0;

        for (BucketsList bucketDetails : bucketDescStats) {
            BucketBuilder bucketDesc = new BucketBuilder();
            final Optional<List<org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.Action>>
                convertedSalActions = getConvertorExecutor().convert(bucketDetails.getAction(), data);

            if (convertedSalActions.isPresent()) {
                BindingMap.Builder<ActionKey, Action> actions = BindingMap.orderedBuilder();

                int actionKey = 0;

                for (org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.Action action :
                        convertedSalActions.get()) {
                    ActionBuilder wrappedAction = new ActionBuilder();
                    wrappedAction.setAction(action);
                    wrappedAction.withKey(new ActionKey(actionKey));
                    wrappedAction.setOrder(actionKey);
                    actions.add(wrappedAction.build());
                    actionKey++;
                }

                bucketDesc.setAction(actions.build());
            } else {
                bucketDesc.setAction(Map.of());
            }

            bucketDesc.setWeight(bucketDetails.getWeight());
            bucketDesc.setWatchPort(bucketDetails.getWatchPort().getValue());
            bucketDesc.setWatchGroup(bucketDetails.getWatchGroup());
            BucketId bucketId = new BucketId(Uint32.valueOf(bucketKey));
            bucketDesc.setBucketId(bucketId);
            bucketDesc.withKey(new BucketKey(bucketId));
            bucketKey++;
            allBuckets.add(bucketDesc.build());
        }

        return new org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.BucketsBuilder()
            .setBucket(allBuckets.build())
            .build();
    }

    @Override
    public Collection<Class<?>> getTypes() {
        return TYPES;
    }

    @Override
    public List<GroupDescStats> convert(final List<GroupDesc> source, final VersionConvertorData data) {
        List<GroupDescStats> convertedSALGroupsDesc = new ArrayList<>();

        for (GroupDesc groupDesc : source) {
            GroupDescStatsBuilder salGroupDescStats = new GroupDescStatsBuilder();

            salGroupDescStats.setBuckets(toSALBucketsDesc(groupDesc.nonnullBucketsList(), data.getVersion()));
            salGroupDescStats.setGroupId(new GroupId(groupDesc.getGroupId().getValue()));
            salGroupDescStats.setGroupType(GroupTypes.forValue(groupDesc.getType().getIntValue()));
            salGroupDescStats.withKey(new GroupDescStatsKey(salGroupDescStats.getGroupId()));

            convertedSALGroupsDesc.add(salGroupDescStats.build());
        }

        return convertedSALGroupsDesc;
    }
}