/*
 * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.bgpcep.pcep.topology.provider.config;

import com.google.common.net.InetAddresses;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.nio.charset.StandardCharsets;
import java.util.Objects;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.Nullable;
import org.opendaylight.protocol.concepts.KeyMapping;
import org.opendaylight.protocol.pcep.SpeakerIdMapping;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IetfInetUtil;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.rfc2385.cfg.rev160324.Rfc2385Key;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.config.rev181109.PcepNodeConfig;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.TopologyTypes1;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.sync.optimizations.config.rev181109.PcepNodeSyncConfig;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.TopologyTypes;

final class PCEPTopologyProviderUtil {
    private PCEPTopologyProviderUtil() {
        // Hidden on purpose
    }

    static KeyMapping contructKeys(final @NonNull Topology topology) {
        final KeyMapping ret = KeyMapping.getKeyMapping();
        if (topology.getNode() == null) {
            return ret;
        }
        topology.getNode().stream()
                .filter(Objects::nonNull)
                .filter(node -> node.augmentation(PcepNodeConfig.class) != null)
                .filter(node -> node.augmentation(PcepNodeConfig.class).getSessionConfig() != null)
                .filter(node -> node.augmentation(PcepNodeConfig.class)
                        .getSessionConfig().getPassword() != null)
                .filter(node -> !node.augmentation(PcepNodeConfig.class)
                        .getSessionConfig().getPassword().getValue().isEmpty())
                .forEach(node -> {
                    final PcepNodeConfig config = node.augmentation(PcepNodeConfig.class);
                    final Rfc2385Key rfc2385KeyPassword = config.getSessionConfig().getPassword();
                    final InetAddress address = InetAddresses.forString(node.getNodeId().getValue());
                    ret.put(address, rfc2385KeyPassword.getValue().getBytes(StandardCharsets.US_ASCII));
                });

        return ret;
    }

    static InetSocketAddress getInetSocketAddress(final @NonNull IpAddressNoZone address,
            final @NonNull PortNumber port) {
        return new InetSocketAddress(IetfInetUtil.INSTANCE.inetAddressForNoZone(address), port.getValue().toJava());
    }

    static boolean filterPcepTopologies(final @Nullable TopologyTypes topologyTypes) {
        if (topologyTypes == null) {
            return false;
        }
        final TopologyTypes1 aug = topologyTypes.augmentation(TopologyTypes1.class);

        return aug != null && aug.getTopologyPcep() != null;
    }


    static SpeakerIdMapping contructSpeakersId(final Topology topology) {
        final SpeakerIdMapping ret = SpeakerIdMapping.getSpeakerIdMap();
        if (topology.getNode() == null) {
            return ret;
        }
        topology.getNode().stream()
                .filter(Objects::nonNull)
                .filter(node -> node.augmentation(PcepNodeConfig.class) != null)
                .filter(node -> node.augmentation(PcepNodeConfig.class).getSessionConfig() != null)
                .filter(node -> node.augmentation(PcepNodeConfig.class).getSessionConfig()
                        .augmentation(PcepNodeSyncConfig.class) != null)
                .forEach(node -> {
                    final PcepNodeConfig config = node.augmentation(PcepNodeConfig.class);
                    final PcepNodeSyncConfig nodeSyncConfig = config.getSessionConfig()
                            .augmentation(PcepNodeSyncConfig.class);
                    final InetAddress address = InetAddresses.forString(node.getNodeId().getValue());
                    ret.put(address, nodeSyncConfig.getSpeakerEntityIdValue());
                });

        return ret;
    }
}