#include <stdio.h>

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

#include <um.h>
#include <err.h>
#include <parser.h>
#include <lib/sv.h>
#include <net.h>

/**
 *  Install package:
 *      pkgman install <package_name>
 *
 *  Remove package:
 *      pkgman remove <package_name>
 *
 *  Update:
 *      pkgman update
 *      pkgman update <package_name>
 *  
 *  Fetch:
 *      pkgman fetch <query>
 *
 *  Pack:
 *      pkgman pack <directory>
 *
 *  System:
 *      pkgman system install
 */

void install_usage()
{
    printf("-------------------------------\n");
    printf("Package Manager Install Command\n");
    printf("command:\n");
    printf("    pkgman install <pkg1> ... <pkgn>\n");
}

int cmd_install(int argc, char **argv)
{
    if(argc < 1) {
        install_usage();
        return USAGE;
    }
    
    int ret = 1;

    struct memory mem = { 0 };
    net_send_request("https://packages.0xinfinity.dev/list", &mem, WRITE_OPT_MEMORY);

    struct um_user_data userdata = { 0 };
    struct parser parser;
    struct parser_backend backend = um_backend();

    parser_init(&parser, &mem, &backend, (void*)&userdata);
    parser_parse(&parser);

    struct string_view pkg = sv_create(argv[0], strlen(argv[0])); 

    int found = 0;

    LL_FOREACH(manifest, &userdata.manifest) {
        if (!current->data.key.buf)
            continue;

        if (sv_equal(&pkg, &current->data.key))
            found = 1;
    }

    if (!found) {
        printf("Package '%s' not found!\n",argv[0]);
        ret = -PKGNOTFND;
        goto cleanup;
    }

    printf("Package '%s' found on upstream!\n", argv[0]);
    ret = SUCCESS;

cleanup:
    // clean LL items
     LL_FOREACH(manifest, &userdata.manifest) {
        if (current->data.key.buf == NULL)
            continue;
        
        free(current->data.key.buf);
        free(current->data.value.buf);
    }

    ll_manifest_free(&userdata.manifest);

    free(mem.buffer);

    return ret;
}

typedef int (*cmd_fn)(int, char**);

struct cmd_entry
{
    struct string_view key;
    cmd_fn func;
};

// First-level command table
struct cmd_entry table[] = {
    { SV("install"), cmd_install },
    { SV(NULL), NULL },
};

#define ARRAY_SIZE(x) sizeof((x)) / sizeof((x)[0])

void usage()
{
    printf("-------------------------------\n");
    printf("pkgman v0.0.1\n");
    printf("-------------------------------\n");
    printf("Available commands:\n"); 
    printf("    pkgman install <pkg>\n"); 
    printf("    pkgman update <pkg?>\n"); 
}

int main(int argc, char **argv) 
{
    if (argc < 2) { 
        usage();
        return 0;
    }

    cmd_fn cmd_func = NULL;
    
    struct string_view argv1 = SV(argv[1]);

    for (int i = 0; i < ARRAY_SIZE(table); i++) {
        if (table[i].key.buf == NULL)
            break;
        if (sv_equal(&argv1, &table[i].key))
            cmd_func = table[i].func;
    }
    
    if (cmd_func == NULL) {
        usage();
        return 1;
    }
    
    if (net_init() != 0) {
        fprintf(stderr, "curl: network initialization error!");
        return 1;
    }

    int ret = 1;

    if(argv[2] == NULL)
        ret = cmd_func(0, NULL);
    else
        ret = cmd_func(argc - 2, &argv[2]);

    if(ret != 0)
        printf("Command '%s' ended with error '%d' and/or usage screen.\n",
                argv[1], ret);

    net_shutdown();
    return 0;
}