#include "loadfiles.h"

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


void magi_loadfiles_add(struct magi_loadfiles *table,
                        const char            *name,
                        const char            *path,
                        int                    max)
{
    static const int size = sizeof(*table->files);
    if (!table) {
        return;
    }
    if (table->count) {
        table->files = realloc(table->files, size * table->count + size);
    } else {
        table->files = malloc(size);
    }
    table->files[table->count].name = name;
    table->files[table->count].path = path;
    table->files[table->count].max  = max;
    table->count++;
}

void magi_loadfiles_free(struct magi_loadfiles *table)
{
    if (!table) {
        return;
    }
    free(table->files);
    table->count = 0;
}

static void loadfiles_callback(void                   *userdata,
                               int                     newfile,
                               const struct magi_file *file,
                               const char             *addon,
                               int                     addon_len)
{
    int pos;
    struct magi_loadfiles *table = userdata;
    if (!file->filename || !*file->filename) {
        return;
    }
    for (pos = 0; pos != table->count; ++pos) {
        if (!strcmp(table->files[pos].name, file->field)) {
            static FILE *f = 0;
            static int   unlimited;
            static int   left;
            if (newfile) {
                const char *path = table->files[pos].path;
                f                = fopen(path, "wb");
                left             = table->files[pos].max;
                unlimited        = !left;
            }
            if (unlimited) {
                fwrite(addon, 1, addon_len, f);
            } else {
                int min = left < addon_len ? left : addon_len;
                fwrite(addon, 1, min, f);
                left -= min;
            }
            if (!addon) {
                fclose(f);
            }
            return;
        }
    }
}

void magi_loadfiles_set(struct magi_request   *request,
                        struct magi_loadfiles *table)
{
    request->callback.act      = loadfiles_callback;
    request->callback.userdata = table;
}