/**
 * NEURON IIoT System for Industry 4.0
 * Copyright (C) 2020-2021 EMQ Technologies Co., Ltd All rights reserved.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 **/

#ifndef _NEU_PLUGIN_REST_HANDLE_H_
#define _NEU_PLUGIN_REST_HANDLE_H_

#include <stdlib.h>

#include "neu_plugin.h"
#include "utils/neu_jwt.h"
#include "json/neu_json_error.h"

#define REST_PROCESS_HTTP_REQUEST(aio, req_type, decode_fun, func)            \
    {                                                                         \
        char *    req_data      = NULL;                                       \
        size_t    req_data_size = 0;                                          \
        req_type *req           = NULL;                                       \
                                                                              \
        if (http_get_body((aio), (void **) &req_data, &req_data_size) == 0 && \
            decode_fun(req_data, &req) == 0) {                                \
            { func };                                                         \
            decode_fun##_free(req);                                           \
        } else {                                                              \
            http_bad_request(aio, "{\"error\": 1002}");                       \
        }                                                                     \
        free(req_data);                                                       \
    }

#define REST_PROCESS_HTTP_REQUEST_VALIDATE_JWT(aio, req_type, decode_fun,    \
                                               func)                         \
    {                                                                        \
        char *jwt = (char *) http_get_header(aio, (char *) "Authorization"); \
                                                                             \
        NEU_JSON_RESPONSE_ERROR(neu_jwt_validate(jwt), {                     \
            if (error_code.error != NEU_ERR_SUCCESS) {                       \
                http_response(aio, error_code.error, result_error);          \
                return;                                                      \
            }                                                                \
        })                                                                   \
        REST_PROCESS_HTTP_REQUEST(aio, req_type, decode_fun, func);          \
    }

#define VALIDATE_JWT(aio)                                                    \
    {                                                                        \
        char *jwt = (char *) http_get_header(aio, (char *) "Authorization"); \
                                                                             \
        NEU_JSON_RESPONSE_ERROR(neu_jwt_validate(jwt), {                     \
            if (error_code.error != NEU_ERR_SUCCESS) {                       \
                http_response(aio, error_code.error, result_error);          \
                return;                                                      \
            }                                                                \
        });                                                                  \
    }

enum neu_rest_method {
    NEU_REST_METHOD_GET = 0x0,
    NEU_REST_METHOD_POST,
    NEU_REST_METHOD_PUT,
    NEU_REST_METHOD_DELETE,
    NEU_REST_METHOD_OPTIONS
};
enum neu_rest_handler_type {
    NEU_REST_HANDLER_FUNCTION = 0x0,
    NEU_REST_HANDLER_DIRECTORY,
};

struct neu_rest_handler {
    enum neu_rest_method       method;
    char *                     url;
    enum neu_rest_handler_type type;
    union neu_rest_handler_value {
        void *handler;
        char *path;
    } value;
};

typedef struct neu_rest_handle_ctx neu_rest_handle_ctx_t;

neu_rest_handle_ctx_t *neu_rest_init_ctx(void *plugin);
void                   neu_rest_free_ctx(neu_rest_handle_ctx_t *ctx);
void *                 neu_rest_get_plugin();

void neu_rest_web_handler(const struct neu_rest_handler **handlers,
                          uint32_t *                      size);
void neu_rest_api_handler(const struct neu_rest_handler **handlers,
                          uint32_t *                      size);
void neu_rest_api_cors_handler(const struct neu_rest_handler **handlers,
                               uint32_t *                      size);
#endif