#include <stdlib.h>
/**
 * Functions for handling the local dialer
 */

#include "libp2p/conn/dialer.h"
#include "libp2p/conn/connection.h"
#include "libp2p/conn/transport_dialer.h"
#include "libp2p/crypto/key.h"
#include "libp2p/utils/linked_list.h"

/**
 * Create a Dialer with the specified local information
 */
struct Dialer* libp2p_conn_dialer_new(char* peer_id, struct PrivateKey* private_key) {
	struct Dialer* dialer = (struct Dialer*)malloc(sizeof(struct Dialer));
	if (dialer != NULL) {
		dialer->peer_id = malloc(strlen(peer_id) + 1);
		if (dialer->peer_id != NULL) {
			strcpy(dialer->peer_id, peer_id);
			dialer->private_key = (struct PrivateKey*)malloc(sizeof(struct PrivateKey));
			if (dialer->private_key != NULL) {
				libp2p_crypto_private_key_copy(private_key, dialer->private_key);
				dialer->fallback_dialer = NULL;
				dialer->transport_dialers = NULL;
				return dialer;
			}
		}
	}
	libp2p_conn_dialer_free(dialer);
	return NULL;
}

/**
 * free resources from the Dialer
 */
void libp2p_conn_dialer_free(struct Dialer* in) {
	if (in != NULL) {
		free(in->peer_id);
		libp2p_crypto_private_key_free(in->private_key);
		if (in->transport_dialers != NULL) {
			struct Libp2pLinkedList* current = in->transport_dialers;
			while(current != NULL) {
				libp2p_conn_transport_dialer_free((struct TransportDialer*)current->item);
				current = current->next;
			}
		}
		if (in->fallback_dialer != NULL)
			libp2p_conn_transport_dialer_free((struct TransportDialer*)in->fallback_dialer);
	}
	return;
}

/**
 * Retrieve a Connection struct from the dialer
 * @param dialer the dialer to use
 * @param muiltiaddress who to connect to
 * @returns a Connection, or NULL
 */
struct Connection* libp2p_conn_dialer_get_connection(struct Dialer* dialer, struct MultiAddress* multiaddress) {
	struct Connection* conn = libp2p_conn_transport_dialer_get(dialer->transport_dialers, multiaddress);
	if (conn == NULL) {
		conn = libp2p_conn_connection_new(dialer->fallback_dialer, multiaddress);
	}
	return conn;
}