#include "ipf.h"
#include "ipmon.h"

static void *file_parse(char **);
static void file_destroy(void *);
static int file_send(void *, ipmon_msg_t *);
static void file_print(void *);
static int file_match(void *, void *);
static void *file_dup(void *);

typedef struct file_opts_s {
        FILE    *fp;
        int     raw;
        char    *path;
        int     ref;
} file_opts_t;

ipmon_saver_t filesaver = {
        "file",
        file_destroy,
        file_dup,
        file_match,
        file_parse,
        file_print,
        file_send
};


static void *
file_parse(char **strings)
{
        file_opts_t *ctx;

        ctx = calloc(1, sizeof(*ctx));
        if (ctx == NULL)
                return (NULL);

        if (strings[0] != NULL && strings[0][0] != '\0') {
                ctx->ref = 1;
                if (!strncmp(strings[0], "raw://", 6)) {
                        ctx->raw = 1;
                        ctx->path = strdup(strings[0] + 6);
                        ctx->fp = fopen(ctx->path, "ab");
                } else if (!strncmp(strings[0], "file://", 7)) {
                        ctx->path = strdup(strings[0] + 7);
                        ctx->fp = fopen(ctx->path, "a");
                } else {
                        free(ctx);
                        ctx = NULL;
                }
        } else {
                free(ctx);
                ctx = NULL;
        }

        return (ctx);
}


static int
file_match(void *ctx1, void *ctx2)
{
        file_opts_t *f1 = ctx1, *f2 = ctx2;

        if (f1->raw != f2->raw)
                return (1);
        if (strcmp(f1->path, f2->path))
                return (1);
        return (0);
}


static void *
file_dup(void *ctx)
{
        file_opts_t *f = ctx;

        f->ref++;
        return (f);
}


static void
file_print(void *ctx)
{
        file_opts_t *file = ctx;

        if (file->raw)
                printf("raw://");
        else
                printf("file://");
        printf("%s", file->path);
}


static void
file_destroy(void *ctx)
{
        file_opts_t *file = ctx;

        file->ref--;
        if (file->ref > 0)
                return;

        if (file->path != NULL)
                free(file->path);
        free(file);
}


static int
file_send(void *ctx, ipmon_msg_t *msg)
{
        file_opts_t *file = ctx;

        if (file->raw) {
                fwrite(msg->imm_data, msg->imm_dsize, 1, file->fp);
        } else {
                fprintf(file->fp, "%s", msg->imm_msg);
        }
        return (0);
}