#include "wsfs/adapter/impl/jsonrpc/server.h"
#include <string.h>

#include "wsfs/adapter/impl/jsonrpc/method_intern.h"
#include "wsfs/adapter/impl/jsonrpc/request.h"
#include "wsfs/adapter/impl/jsonrpc/response.h"

#define WSFS_DEFAULT_TIMEOUT (10 * 1000)

static struct jsonrpc_method const * jsonrpc_server_getmethod(
    struct jsonrpc_server * server,
    char const * name)
{
    struct jsonrpc_method * method = server->methods;
    while ((NULL != method) && (0 == strcmp(name, method->name)))
    {
        method = method->next;
    }

    return method;
}

static void jsonrpc_server_timeout(
    struct timer * timer)
{
    struct jsonrpc_server * server = timer->user_data;

    if (server->request.is_pending)
    {
        jsonrpc_method_finished_fn * finished = server->request.finished;
        void * user_data = server->request.user_data;

        server->request.is_pending = false;
        server->request.id = 0;
        server->request.user_data = NULL;
        server->request.finished = NULL;
        timer_cancel(&server->request.timer);

        finished(user_data, WSFS_BAD_TIMEOUT, NULL);
    }
}

void jsonrpc_server_init(
    struct jsonrpc_server * server,
    struct timeout_manager * timeout_manager)
{
    server->methods = NULL;
    server->request.is_pending = false;
    
    timer_init(&server->request.timer, timeout_manager);
}

void jsonrpc_server_cleanup(
    struct jsonrpc_server * server)
{
    timer_cleanup(&server->request.timer);

    if (server->request.is_pending)
    {
        server->request.finished(server->request.user_data, WSFS_BAD, NULL);
        server->request.is_pending = false;
    }

    struct jsonrpc_method * method = server->methods;
    while (NULL != method)
    {
        struct jsonrpc_method * next = method->next;
        method->next = NULL;
        jsonrpc_method_dispose(method);
        method = next;
    }
    server->methods = NULL;
}

void jsonrpc_server_add(
    struct jsonrpc_server * server,
    char const * name,
    jsonrpc_method_invoke_fn * invoke,
    void * user_data)
{
    struct jsonrpc_method * method = jsonrpc_method_create(name, invoke, user_data);
    method->next = server->methods;
    server->methods = method;
}

void jsonrpc_server_invoke(
	struct jsonrpc_server * server,
	jsonrpc_method_finished_fn * finished,
	void * user_data,
	char const * method_name,
	char const * param_info,
	...
)
{
    if (!server->request.is_pending)
    {
        struct jsonrpc_method const * method = jsonrpc_server_getmethod(server, method_name);
        if (NULL != method)
        {
            server->request.is_pending = true;
            server->request.finished = finished;
            server->request.user_data = user_data;
            server->request.id = 42;
            timer_start(&server->request.timer, timepoint_in_msec(WSFS_DEFAULT_TIMEOUT), 
                    &jsonrpc_server_timeout, server);
            
            va_list args;
            va_start(args, param_info);
            json_t * request = jsonrpc_request_create(method_name, server->request.id, param_info, args);
            va_end(args);
            if (NULL != request)
            {
                if (!method->invoke(method->user_data, request))
                {
                    server->request.is_pending = false;
                    server->request.finished = NULL;
                    server->request.user_data = NULL;
                    server->request.id = 0;
                    timer_cancel(&server->request.timer);

                    finished(user_data, WSFS_BAD, NULL);
                }
                json_decref(request);
            }
        }
        else
        {
            finished(user_data, WSFS_BAD_NOTIMPLEMENTED, NULL);
        }        
    }
    else
    {
        finished(user_data, WSFS_BAD_BUSY, NULL);
    }
}

extern void jsonrpc_server_notify(
	struct jsonrpc_server * server,
	char const * method_name,
	char const * param_info,
	...
)
{
        struct jsonrpc_method const * method = jsonrpc_server_getmethod(server, method_name);
        if (NULL != method)
        {
            
            va_list args;
            va_start(args, param_info);
            json_t * request = jsonrpc_request_create(method_name, 0, param_info, args);
            va_end(args);
            if (NULL != request)
            {
                method->invoke(method->user_data, request);
                json_decref(request);
            }
        }

}


void jsonrpc_server_onresult(
    struct jsonrpc_server * server,
    char const * message,
    size_t length)
{
	struct jsonrpc_response response;
	jsonrpc_response_init(&response, message, length);

    if ((server->request.is_pending) && (response.id == server->request.id))
    {
        jsonrpc_method_finished_fn * finished = server->request.finished;
        void * user_data = server->request.user_data;

        server->request.is_pending = false;
        server->request.id = 0;
        server->request.user_data = NULL;
        server->request.finished = NULL;
        timer_cancel(&server->request.timer);

        finished(user_data, response.status, response.result);
    }

    jsonrpc_response_cleanup(&response);
}