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

import static java.util.Objects.requireNonNull;

import java.util.Set;
import org.eclipse.jdt.annotation.Nullable;
import org.opendaylight.lispflowmapping.lisp.type.MappingData;
import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping._record.container.MappingRecord;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping._record.container.MappingRecordBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.EidUri;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.XtrIdUri;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKeyBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.mapping.XtrIdMapping;
import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.mapping.XtrIdMappingBuilder;

/**
 * DataStoreBackEnd utilities.
 *
 * @author Florin Coras
 */
public final class DSBEInputUtil {
    // Utility class, should not be instantiated
    private DSBEInputUtil() {
    }

    public static Mapping toMapping(MappingOrigin origin, Eid key, SiteId siteId, MappingData mappingData) {
        MappingRecord record = mappingData != null ? mappingData.getRecord() : null;
        Set<SiteId> siteIds = siteId != null ? Set.of(siteId) : null;
        return new MappingBuilder()
                    .setEidUri(new EidUri(LispAddressStringifier.getURIString(key)))
                    .setOrigin(origin)
                    .setSiteId(siteIds)
                    .setMappingRecord(record).build();
    }

    public static Mapping toMapping(MappingOrigin origin, Eid key, @Nullable MappingData mappingData) {
        MappingRecord record = mappingData != null ? mappingData.getRecord() :
                                                        new MappingRecordBuilder().setEid(key).build();
        SiteId siteId = record != null ? record.getSiteId() : null;
        Set<SiteId> siteIds = siteId != null ? Set.of(siteId) : null;
        return new MappingBuilder()
                .setEidUri(new EidUri(LispAddressStringifier.getURIString(key)))
                .setOrigin(origin)
                .setSiteId(siteIds)
                .setMappingRecord(record).build();
    }

    public static Mapping toMapping(MappingOrigin origin, Eid key) {
        MappingBuilder mb = new MappingBuilder();
        mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(key)));
        mb.setOrigin(origin);
        mb.setMappingRecord(new MappingRecordBuilder().setEid(key).build());
        return mb.build();
    }

    public static XtrIdMapping toXtrIdMapping(MappingData mappingData) {
        requireNonNull(mappingData);
        MappingRecord record = mappingData.getRecord();
        requireNonNull(mappingData.getRecord());
        return new XtrIdMappingBuilder()
                    .setXtrIdUri(new XtrIdUri(LispAddressStringifier.getURIString(record.getXtrId())))
                    .setMappingRecord(record).build();
    }

    public static AuthenticationKey toAuthenticationKey(Eid key, MappingAuthkey authKey) {
        AuthenticationKeyBuilder akb = new AuthenticationKeyBuilder();
        akb.setEidUri(new EidUri(LispAddressStringifier.getURIString(key)));
        akb.setEid(key);
        akb.setMappingAuthkey(authKey);
        return akb.build();
    }
}