#include <stdio.h>
#include <string.h>

#include "dataserver_inner.h"
#include "list.h"

#define ds_inf(_fmt_,...) printf("I %s:%d "_fmt_"\n", __func__, __LINE__, ##__VA_ARGS__)
#define ds_err(_fmt_,...) printf("E %s:%d "_fmt_"\n", __func__, __LINE__, ##__VA_ARGS__)

static int server_stat(ds_connection_t *conn,
        uint8_t *cmd, size_t size, uint8_t *resp, size_t resp_size)
{
    int ret = 0;
    int cnt = 0;

    #define stat_printf(_fmt, ...) do {\
        if ((resp_size - cnt) > 1) {\
            ret = snprintf((char *)resp + cnt, resp_size - cnt, _fmt, ##__VA_ARGS__);\
            if (ret <= 0) {\
                return cnt + 1;\
            } else {\
                cnt += ret;\
            }\
        } else {\
            ds_err("resp not enough");\
        }\
    } while (0)

    stat_printf("ok");
    cnt++;
    stat_printf("conns:\n");
    {
        ds_server_t *server = (ds_server_t *)conn->server;

        ds_connection_t *conn;
        list_for_each_entry(conn, &server->conns, head) {
            stat_printf("\t[%s]-[%s] fd[%d] active[%d] rb[%#zx/%#zx] flag[%#x]\n",
                    conn->device, conn->client, conn->fd, conn->active,
                    ds_rb_avail(&conn->rb), conn->rb.size, ds_rb_get_flag(&conn->rb));
        }
    }

    return cnt + 1;
}

static ds_control_t controls[] = {
    {.cmd = "stat2", .fn = server_stat, .desc = "stat2"},
};
static int num_controls = sizeof(controls) / sizeof(controls[0]);

int load(ds_connection_t *conn)
{
    ds_inf("plugin loaded");

    int i;
    for (i = 0; i < num_controls; i++) {
        int found = 0;
        ds_control_t *ctl;
        list_for_each_entry(ctl, &conn->controls, head) {
            if (strcmp(ctl->cmd, controls[i].cmd) == 0) {
                found = 1;
                break;
            }
        }
        if (!found) {
            list_add_tail(&controls[i].head, &conn->controls);
        } else {
            ds_err("cmd[%s] conflict, ignore the one in plugin",
                    controls[i].cmd);
        }
    }

    return 0;
}

int unload(ds_connection_t *conn)
{
    int i;
    for (i = 0; i < num_controls; i++) {
        ds_control_t *ctl, *tmp;
        list_for_each_entry_safe(ctl, tmp, &conn->controls, head) {
            if (strcmp(ctl->cmd, controls[i].cmd) == 0)
                list_del(&ctl->head);
        }
    }

    ds_inf("plugin unloaded");
    return 0;
}