/*
 * Copyright (c) 2014 Contextream, 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.lispflowmapping.lisp.util;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapRegister;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapnotifymessage.MapNotifyBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;

public final class MapNotifyBuilderHelper {
    // Utility class, should not be instantiated
    private MapNotifyBuilderHelper() {
    }

    public static void setFromMapRegister(MapNotifyBuilder builder, MapRegister mapRegister) {
        setNonRecordFields(builder, mapRegister);

        if (builder.getMappingRecordItem() == null) {
            builder.setMappingRecordItem(new ArrayList<MappingRecordItem>());
        }

        for (MappingRecordItem eidToLocator : mapRegister.getMappingRecordItem()) {
            builder.getMappingRecordItem().add(eidToLocator);
        }
    }

    public static void setFromMapRegisterAndMappingRecordItems(MapNotifyBuilder builder, MapRegister mapRegister,
            List<MappingRecordItem> mappings) {
        setNonRecordFields(builder, mapRegister);
        builder.setMappingRecordItem(mappings);
    }

    private static void setNonRecordFields(MapNotifyBuilder builder, MapRegister mapRegister) {
        builder.setNonce(mapRegister.getNonce());
        builder.setKeyId(mapRegister.getKeyId());
        builder.setMergeEnabled(mapRegister.isMergeEnabled());

        byte[] authenticationData = mapRegister.getAuthenticationData();
        if (authenticationData != null) {
            authenticationData = authenticationData.clone();
            Arrays.fill(authenticationData, (byte) 0);
        }
        builder.setAuthenticationData(authenticationData);
    }
}