#include "webfuse_provider/impl/client.h"

#include <stdlib.h>
#include <string.h>
#include <stdbool.h>

#include <libwebsockets.h>

#include "webfuse_provider/impl/provider.h"
#include "webfuse_provider/impl/client_protocol.h"
#include "webfuse_provider/impl/client_config.h"
#include "webfuse_provider/impl/util/lws_log.h"

#define WFP_CLIENT_PROTOCOL_COUNT 2

struct wfp_client
{
    struct wfp_client_protocol protocol;
    struct lws_context_creation_info info;
    struct lws_protocols protocols[WFP_CLIENT_PROTOCOL_COUNT];
    struct lws_context * context;
    char * key_path;
    char * cert_path;
};


struct wfp_client * wfp_impl_client_create(
    struct wfp_client_config * config)
{
	wfp_impl_lwslog_disable();

    struct wfp_client * client = malloc(sizeof(struct wfp_client));
    wfp_impl_client_protocol_init(&client->protocol, &config->provider, config->fs_name, config->user_data);

    memset(client->protocols, 0, sizeof(struct lws_protocols) * WFP_CLIENT_PROTOCOL_COUNT);
    wfp_impl_client_protocol_init_lws(&client->protocol, &client->protocols[0]);

    memset(&client->info, 0, sizeof(struct lws_context_creation_info));
    client->info.port = CONTEXT_PORT_NO_LISTEN;
    client->info.protocols = client->protocols;
    client->info.uid = -1;
    client->info.gid = -1;

    if ((NULL != config->cert_path) && (NULL != config->key_path))
    {
		client->info.options |= LWS_SERVER_OPTION_EXPLICIT_VHOSTS;
    }

    client->context = lws_create_context(&client->info);

    if ((NULL != config->cert_path) && (NULL != config->key_path))
    {
        struct lws_vhost * vhost = lws_create_vhost(client->context, &client->info);
		client->info.options |= LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
        client->info.client_ssl_cert_filepath = config->cert_path;
        client->info.client_ssl_private_key_filepath = config->key_path;
        client->info.client_ssl_ca_filepath = config->ca_filepath;
        lws_init_vhost_client_ssl(&client->info, vhost);
    }

    return client;
}

void wfp_impl_client_dispose(
    struct wfp_client * client)
{
    lws_context_destroy(client->context);
    wfp_impl_client_protocol_cleanup(&client->protocol);
    free(client);
}

void wfp_impl_client_connect(
    struct wfp_client * client,
    char const * url)
{
    wfp_impl_client_protocol_connect(&client->protocol, client->context, url);
}

void wfp_impl_client_disconnect(
    struct wfp_client * client)
{
    wfp_impl_client_protocol_disconnect(&client->protocol);
}

void wfp_impl_client_service(
    struct wfp_client * client)
{
    lws_service(client->context, 0);
}

void wfp_impl_client_interrupt(
    struct wfp_client * client)
{
    lws_cancel_service(client->context);
}