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

typedef enum AssetInfoType
{
    TEXTURE_INFO,
    SPRITE_INFO,
    SOUND_INFO,
    EMITTER_INFO,
    LEVELPACK_INFO,
    INVALID_INFO
}AssetInfoType_t;

typedef struct SpriteInfo
{
    char* tex;
    Vector2 origin;
    Vector2 frame_size;
    int speed;
    int frame_count;
}SpriteInfo_t;

static bool parse_sprite_info(char* sprite_info_str, SpriteInfo_t* spr_info)
{
    char* tex_name = strtok(sprite_info_str, ",");
    if (tex_name == NULL) return false;
    spr_info->tex = tex_name;
    char* spr_data = tex_name + strlen(tex_name) + 1;
    int data_count = sscanf(
        spr_data, "%f,%f,%f,%f,%d,%d",
        &spr_info->origin.x, &spr_info->origin.y,
        &spr_info->frame_size.x, &spr_info->frame_size.y,
        &spr_info->frame_count, &spr_info->speed
    );
    return data_count == 6;
}

static bool parse_emitter_info(char* emitter_info_str, EmitterConfig_t* conf)
{
    char emitter_type;
    uint8_t one_shot;
    int data_count = sscanf(
        emitter_info_str, "%c,%f-%f,%f-%f,%u-%u,%c",
        &emitter_type,
        conf->launch_range, conf->launch_range + 1,
        conf->speed_range, conf->speed_range + 1,
        conf->particle_lifetime, conf->particle_lifetime + 1,
        &one_shot
    );

    if (data_count == 8)
    {
        conf->type = (emitter_type == 'b') ? EMITTER_BURST : EMITTER_UNKNOWN;
        conf->one_shot = (one_shot == 1);
    }
    return data_count == 8;
}

static inline AssetInfoType_t get_asset_type(const char* str)
{
    if (strcmp(str, "Texture") == 0) return TEXTURE_INFO;

    if (strcmp(str, "Sprite") == 0) return SPRITE_INFO;

    if (strcmp(str, "Sound") == 0) return SOUND_INFO;

    if (strcmp(str, "Emitter") == 0) return EMITTER_INFO;

    if (strcmp(str, "LevelPack") == 0) return LEVELPACK_INFO;

    return INVALID_INFO;
}

bool load_from_rres(const char* file, Assets_t* assets)
{
    RresFileInfo_t rres_file;
    rres_file.dir = rresLoadCentralDirectory(file);
    rres_file.fname = file;

    if (rres_file.dir.count == 0)
    {
        puts("Empty central directory");
        return false;
    }

    int res_id = rresGetResourceId(rres_file.dir, "assets.info");
    rresResourceChunk chunk = rresLoadResourceChunk(file, res_id); // Hardcoded
    bool okay = false;

    if (chunk.info.baseSize > 0)
    {
        FILE* in_file = fmemopen(chunk.data.raw, chunk.info.baseSize, "rb");

        char buffer[256];
        char* tmp;
        size_t line_num = 0;
        AssetInfoType_t info_type = INVALID_INFO;

        while (true)
        {
            tmp = fgets(buffer, 256, in_file);
            if (tmp == NULL) break;
            tmp[strcspn(tmp, "\r\n")] = '\0';

            if (tmp[0] == '-')
            {
                tmp++;
                info_type = get_asset_type(tmp);
            }
            else
            {
                char* name = strtok(buffer, ":");
                char* info_str = strtok(NULL, ":");
                if (name == NULL || info_str == NULL) continue;

                while(*name == ' ' || *name == '\t') name++;
                while(*info_str == ' ' || *info_str == '\t') info_str++;
                switch(info_type)
                {
                    case TEXTURE_INFO:
                    {
                        //if (add_texture(assets, name, info_str) == NULL)
                        if (add_texture_rres(assets, name, info_str, &rres_file) == NULL)
                        {
                            printf("Unable to add texture at line %lu\n", line_num);
                            break;
                        }
                        printf("Added texture %s as %s\n", info_str, name);
                        //strcpy(tmp2, name);
                    }
                    break;
                    case LEVELPACK_INFO:
                    {
                        //if (add_level_pack(assets, name, info_str) == NULL)
                        if (add_level_pack_rres(assets, name, info_str, &rres_file) == NULL)
                        {
                            printf("Unable to add level pack at line %lu\n", line_num);
                            break;
                        }
                        printf("Added level pack %s as %s\n", info_str, name);
                    }
                    break;
                    case SOUND_INFO:
                    {
                        if (add_sound_rres(assets, name, info_str, &rres_file) == NULL)
                        {
                            printf("Unable to add sound at line %lu\n", line_num);
                            break;
                        }
                        printf("Added sound %s as %s\n", info_str, name);
                    }
                    break;
                    case SPRITE_INFO:
                    {
                        SpriteInfo_t spr_info = {0};
                        if (!parse_sprite_info(info_str, &spr_info))
                        {
                            printf("Unable to parse info for sprite at line %lu\n", line_num);
                            break;
                        }
                        //printf("Compare %s,%s = %d\n", tmp2, spr_info.tex, strcmp(tmp2, spr_info.tex));
                        Texture2D* tex = get_texture(assets, spr_info.tex);
                        if (tex == NULL)
                        {
                            printf("Unable to get texture info %s for sprite %s\n", spr_info.tex, name);
                            break;
                        }
                        printf("Added Sprite %s from texture %s\n", name, spr_info.tex);
                        Sprite_t* spr = add_sprite(assets, name, tex);
                        spr->origin = spr_info.origin;
                        spr->frame_size = spr_info.frame_size;
                        spr->frame_count = spr_info.frame_count;
                        spr->speed = spr_info.speed;
                    }
                    break;
                    default:
                    break;
                }
            }
            line_num++; 
        }
        fclose(in_file);
        okay = true;
    }
    rresUnloadResourceChunk(chunk);
    rresUnloadCentralDirectory(rres_file.dir);
    return okay;
}

bool load_from_infofile(const char* file, Assets_t* assets)
{
    FILE* in_file = fopen(file, "r");
    if (in_file == NULL)
    {
        printf("Unable to open file %s\n", file);
        return false;
    }
    
    char buffer[256];
    char* tmp;
    size_t line_num = 0;
    AssetInfoType_t info_type = INVALID_INFO;

    while (true)
    {
        tmp = fgets(buffer, 256, in_file);
        if (tmp == NULL) break;
        tmp[strcspn(tmp, "\r\n")] = '\0';

        if (tmp[0] == '-')
        {
            tmp++;
            info_type = get_asset_type(tmp);
        }
        else
        {
            char* name = strtok(buffer, ":");
            char* info_str = strtok(NULL, ":");
            if (name == NULL || info_str == NULL) continue;

            while(*name == ' ' || *name == '\t') name++;
            while(*info_str == ' ' || *info_str == '\t') info_str++;
            switch(info_type)
            {
                case TEXTURE_INFO:
                {
                    if (add_texture(assets, name, info_str) == NULL)
                    {
                        printf("Unable to add texture at line %lu\n", line_num);
                        break;
                    }
                    printf("Added texture %s as %s\n", info_str, name);
                }
                break;
                case SOUND_INFO:
                {
                    if (add_sound(assets, name, info_str) == NULL)
                    {
                        printf("Unable to add texture at line %lu\n", line_num);
                        break;
                    }
                    printf("Added sound %s as %s\n", info_str, name);
                }
                break;
                case LEVELPACK_INFO:
                {
                    //if (add_level_pack(assets, name, info_str) == NULL)
                    if (uncompress_level_pack(assets, name, info_str) == NULL)
                    {
                        printf("Unable to add level pack at line %lu\n", line_num);
                        break;
                    }
                    printf("Added level pack %s as %s\n", info_str, name);
                }
                break;
                case SPRITE_INFO:
                {
                    SpriteInfo_t spr_info = {0};
                    if (!parse_sprite_info(info_str, &spr_info))
                    {
                        printf("Unable to parse info for sprite at line %lu\n", line_num);
                        break;
                    }
                    //printf("Compare %s,%s = %d\n", tmp2, spr_info.tex, strcmp(tmp2, spr_info.tex));
                    Texture2D* tex = get_texture(assets, spr_info.tex);
                    if (tex == NULL)
                    {
                        printf("Unable to get texture info %s for sprite %s\n", spr_info.tex, name);
                        break;
                    }
                    printf("Added Sprite %s from texture %s\n", name, spr_info.tex);
                    Sprite_t* spr = add_sprite(assets, name, tex);
                    spr->origin = spr_info.origin;
                    spr->frame_size = spr_info.frame_size;
                    spr->frame_count = spr_info.frame_count;
                    spr->speed = spr_info.speed;
                }
                break;
                case EMITTER_INFO:
                {
                    EmitterConfig_t parsed_conf;
                    if (!parse_emitter_info(info_str, &parsed_conf))
                    {
                        printf("Parse error for emitter %s", name);
                        break;
                    }
                    EmitterConfig_t* conf = add_emitter_conf(assets, name);
                    *conf = parsed_conf;
                    printf("Added Emitter %s\n", name);
                }
                break;
                default:
                break;
            }
        }
        line_num++; 
    }
    fclose(in_file);
    return true;
}