#include <stdlib.h>
#include <getopt.h>
#include <string.h>
#include <sys/stat.h>
#include <time.h>

#include "../../include/tanabata.h"

// TFM configuration directory
#define TFM_CONFIG_DIR "/etc/tanabata/"

// Stylization macros
#define TABLE_HEADER(s) ""s""
#define HIGHLIGHT(s)    ""s""
#define SUCCESS(s)      ""s""
#define ERROR(s)        ""s""

#define DT_FORMAT       "%F %T"

static Tanabata tanabata;

// Print the list of all sasa
void print_sasa_all() {
    printf(TABLE_HEADER("         Sasa ID\tFile path")"\n");
    for (uint64_t i = 0; i < tanabata.sasahyou.size; i++) {
        if (tanabata.sasahyou.database[i].id != HOLE_ID) {
            printf("%16lx\t%s\n", tanabata.sasahyou.database[i].id, tanabata.sasahyou.database[i].path);
        }
    }
}

// Print the list of all tanzaku
void print_tanzaku_all() {
    printf(TABLE_HEADER("      Tanzaku ID\tName")"\n");
    for (uint64_t i = 0; i < tanabata.sappyou.size; i++) {
        if (tanabata.sappyou.database[i].id != HOLE_ID) {
            printf("%16lx\t%s\n", tanabata.sappyou.database[i].id, tanabata.sappyou.database[i].name);
        }
    }
}

// Sasa view menu handler
int menu_view_sasa(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    if (strcmp(arg, ".") == 0) {
        print_sasa_all();
        return 0;
    }
    char *endptr;
    uint64_t sasa_id = strtoull(arg, &endptr, 16);
    if (*endptr == 0) {
        Sasa current_sasa = tanabata_sasa_get(&tanabata, sasa_id);
        if (current_sasa.id != HOLE_ID) {
            char datetime[20];
            strftime(datetime, 20, DT_FORMAT,
                     localtime((const time_t *) &current_sasa.created_ts));
            printf(HIGHLIGHT("Sasa ID")"            %lx\n"
                   HIGHLIGHT("File path")"          %s\n"
                   HIGHLIGHT("Added datetime")"     %s\n\n",
                   sasa_id, current_sasa.path, datetime);
            Tanzaku *related_tanzaku = tanabata_tanzaku_get_by_sasa(&tanabata, current_sasa.id);
            if (related_tanzaku != NULL) {
                printf(HIGHLIGHT("↓ Related tanzaku ↓")"\n"
                       HIGHLIGHT("      Tanzaku ID\tName")"\n");
                for (Tanzaku *current_tanzaku = related_tanzaku;
                     current_tanzaku->id != HOLE_ID; current_tanzaku++) {
                    printf("%16lx\t%s\n", current_tanzaku->id, current_tanzaku->name);
                }
                printf(HIGHLIGHT("↑ Related tanzaku ↑")"\n");
            } else {
                printf(HIGHLIGHT("No related tanzaku")"\n");
            }
            return 0;
        }
        fprintf(stderr, ERROR("No sasa with this ID")"\n");
        return 1;
    }
    fprintf(stderr, ERROR("Invalid ID")"\n");
    return 1;
}

// Tanzaku view menu handler
int menu_view_tanzaku(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    if (strcmp(arg, ".") == 0) {
        print_tanzaku_all();
        return 0;
    }
    char *endptr;
    uint64_t tanzaku_id = strtoull(arg, &endptr, 16);
    if (*endptr == 0) {
        Tanzaku current_tanzaku = tanabata_tanzaku_get(&tanabata, tanzaku_id);
        if (current_tanzaku.id != HOLE_ID) {
            char datetime[20];
            strftime(datetime, 20, DT_FORMAT,
                     localtime((const time_t *) &current_tanzaku.created_ts));
            printf(HIGHLIGHT("Tanzaku ID")"         %lx\n"
                   HIGHLIGHT("Name")"               %s\n"
                   HIGHLIGHT("Created datetime")"   %s\n",
                   tanzaku_id, current_tanzaku.name, datetime);
            strftime(datetime, 20, DT_FORMAT,
                     localtime((const time_t *) &current_tanzaku.modified_ts));
            printf(HIGHLIGHT("Modified datetime")"  %s\n\n", datetime);
            if (*current_tanzaku.description != 0) {
                printf(HIGHLIGHT("↓ Description ↓")"\n"
                       "%s\n"
                       HIGHLIGHT("↑ Description ↑")"\n\n", current_tanzaku.description);
            } else {
                printf(HIGHLIGHT("No description")"\n\n");
            }
            Sasa *related_sasa = tanabata_sasa_get_by_tanzaku(&tanabata, tanzaku_id);
            if (related_sasa != NULL) {
                printf(HIGHLIGHT("↓ Related sasa ↓")"\n"
                       HIGHLIGHT("         Sasa ID\tFile path")"\n");
                for (Sasa *current_sasa = related_sasa;
                     current_sasa->id != HOLE_ID; current_sasa++) {
                    printf("%16lx\t%s\n", current_sasa->id, current_sasa->path);
                }
                printf(HIGHLIGHT("↑ Related sasa ↑")"\n");
            } else {
                printf(HIGHLIGHT("No related sasa")"\n");
            }
            return 0;
        }
        fprintf(stderr, ERROR("No tanzaku with this ID")"\n");
        return 1;
    }
    fprintf(stderr, ERROR("Invalid ID")"\n");
    return 1;
}

// Sasa add menu handler
int menu_add_sasa(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    if (tanabata.sasahyou.size == -1 && tanabata.sasahyou.hole_cnt == 0) {
        fprintf(stderr, ERROR("Failed to add file to database: sasahyou is full")"\n");
        return 1;
    }
    char *path = realpath(arg, NULL);
    if (path == NULL) {
        fprintf(stderr, ERROR("Invalid file path")"\n");
        free(path);
        return 1;
    }
    if (tanabata_sasa_add(&tanabata, path).id != HOLE_ID &&
        tanabata_save(&tanabata) == 0) {
        printf(SUCCESS("Successfully added file to database")"\n");
        free(path);
        return 0;
    }
    fprintf(stderr, ERROR("Failed to add file to database")"\n");
    free(path);
    return 1;
}

// Tanzaku add menu handler
int menu_add_tanzaku(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    if (tanabata.sappyou.size == -1 && tanabata.sappyou.hole_cnt == 0) {
        fprintf(stderr, ERROR("Failed to add tanzaku: sappyou is full")"\n");
        return 1;
    }
    if (*arg != 0) {
        char description[4096];
        printf(HIGHLIGHT("Enter tanzaku description:")"\n");
        fgets(description, 4096, stdin);
        description[strlen(description) - 1] = 0;
        if (tanabata_tanzaku_add(&tanabata, arg, description).id != HOLE_ID &&
            tanabata_save(&tanabata) == 0) {
            printf(SUCCESS("Successfully added tanzaku to database")"\n");
            return 0;
        }
    }
    fprintf(stderr, ERROR("Failed to add tanzaku to database")"\n");
    return 1;
}

// Kazari add menu handler
int menu_add_kazari(char *arg) {
    if (arg == NULL) {
        return 1;
    }
    if (tanabata.shoppyou.size == -1 && tanabata.shoppyou.hole_cnt == 0) {
        fprintf(stderr, ERROR("Failed to add kazari: shoppyou is full")"\n");
        return 1;
    }
    char *left = arg, *right = "\0", *endptr;
    for (size_t i = 0; i < strlen(arg); i++) {
        if (arg[i] == '-') {
            arg[i] = 0;
            right = arg + i + 1;
            break;
        }
    }
    if (*left == 0 || *right == 0) {
        fprintf(stderr, ERROR("Failed to add kazari: invalid argument")"\n");
        return 1;
    }
    uint64_t sasa_id = strtoull(left, &endptr, 16);
    if (*endptr != 0) {
        fprintf(stderr, ERROR("Failed to add kazari: invalid sasa ID")"\n");
        return 1;
    }
    uint64_t tanzaku_id = strtoull(right, &endptr, 16);
    if (*endptr != 0) {
        fprintf(stderr, ERROR("Failed to add kazari: invalid tanzaku ID")"\n");
        return 1;
    }
    if (tanabata_kazari_add(&tanabata, sasa_id, tanzaku_id) == 0 &&
        tanabata_save(&tanabata) == 0) {
        printf(SUCCESS("Successfully added kazari")"\n");
        return 0;
    }
    fprintf(stderr, ERROR("Failed to add kazari")"\n");
    return 1;
}

// Sasa remove menu handler
int menu_rem_sasa(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    char *endptr;
    uint64_t sasa_id = strtoull(arg, &endptr, 16);
    if (*endptr == 0) {
        if (tanabata_sasa_rem(&tanabata, sasa_id) == 0 &&
            tanabata_save(&tanabata) == 0) {
            printf(SUCCESS("Successfully removed sasa")"\n");
            return 0;
        }
        fprintf(stderr, ERROR("Failed to remove sasa")"\n");
        return 1;
    }
    fprintf(stderr, ERROR("Invalid ID")"\n");
    return 1;
}

// Tanzaku remove menu handler
int menu_rem_tanzaku(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    char *endptr;
    uint64_t tanzaku_id = strtoull(arg, &endptr, 16);
    if (*endptr == 0) {
        if (tanabata_tanzaku_rem(&tanabata, tanzaku_id) == 0 &&
            tanabata_save(&tanabata) == 0) {
            printf(SUCCESS("Successfully removed tanzaku")"\n");
            return 0;
        }
        fprintf(stderr, ERROR("Failed to remove tanzaku")"\n");
        return 1;
    }
    fprintf(stderr, ERROR("Invalid ID")"\n");
    return 1;
}

// Kazari remove menu handler
int menu_rem_kazari(char *arg) {
    if (arg == NULL) {
        return 1;
    }
    char *left = arg, *right = "\0", *endptr;
    for (size_t i = 0; i < strlen(arg); i++) {
        if (arg[i] == '-') {
            arg[i] = 0;
            right = arg + i + 1;
            break;
        }
    }
    if (*left == 0 || *right == 0) {
        fprintf(stderr, ERROR("Failed to remove kazari: invalid argument")"\n");
        return 1;
    }
    uint64_t sasa_id = strtoull(left, &endptr, 16);
    if (*endptr != 0) {
        fprintf(stderr, ERROR("Failed to remove kazari: invalid sasa ID")"\n");
        return 1;
    }
    uint64_t tanzaku_id = strtoull(right, &endptr, 16);
    if (*endptr != 0) {
        fprintf(stderr, ERROR("Failed to remove kazari: invalid tanzaku ID")"\n");
        return 1;
    }
    if (tanabata_kazari_rem(&tanabata, sasa_id, tanzaku_id) == 0 &&
        tanabata_save(&tanabata) == 0) {
        printf(SUCCESS("Successfully removed kazari")"\n");
        return 0;
    }
    fprintf(stderr, ERROR("Failed to remove kazari")"\n");
    return 1;
}

// Sasa update menu handler
int menu_upd_sasa(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    char *endptr;
    uint64_t sasa_id = strtoull(arg, &endptr, 16);
    if (*endptr == 0) {
        char *path = malloc(4096);
        printf(HIGHLIGHT("Enter the new file path (leave blank to keep current):")"\n");
        fgets(path, 4096, stdin);
        if (*path == '\n') {
            free(path);
            path = NULL;
        } else {
            path[strlen(path) - 1] = 0;
        }
        if (tanabata_sasa_upd(&tanabata, sasa_id, path) == 0 &&
            tanabata_save(&tanabata) == 0) {
            printf(SUCCESS("Successfully updated sasa")"\n");
            return 0;
        }
        fprintf(stderr, ERROR("Failed to update sasa")"\n");
        return 1;
    }
    fprintf(stderr, ERROR("Invalid ID")"\n");
    return 1;
}

// Tanzaku update menu handler
int menu_upd_tanzaku(const char *arg) {
    if (arg == NULL) {
        return 1;
    }
    char *endptr;
    uint64_t tanzaku_id = strtoull(arg, &endptr, 16);
    if (*endptr == 0) {
        char *name = malloc(4096), *description = malloc(4096);
        printf(HIGHLIGHT("Enter the new name of tanzaku (leave blank to keep current):")"\n");
        fgets(name, 4096, stdin);
        if (*name == '\n') {
            free(name);
            name = NULL;
        } else {
            name[strlen(name) - 1] = 0;
        }
        printf(HIGHLIGHT("Enter the new description of tanzaku (leave blank to keep current):")"\n");
        fgets(description, 4096, stdin);
        if (*description == '\n') {
            free(description);
            description = NULL;
        } else {
            description[strlen(description) - 1] = 0;
        }
        if (tanabata_tanzaku_upd(&tanabata, tanzaku_id, name, description) == 0 &&
            tanabata_save(&tanabata) == 0) {
            printf(SUCCESS("Successfully updated tanzaku")"\n");
            return 0;
        }
        fprintf(stderr, ERROR("Failed to update tanzaku")"\n");
        return 1;
    }
    fprintf(stderr, ERROR("Invalid ID")"\n");
    return 1;
}

int main(int argc, char **argv) {
    if (argc == 1) {
        fprintf(stderr, ERROR("No options provided")"\n");
        return 1;
    }
    char *tanabata_path;
    FILE *config = fopen(TFM_CONFIG_DIR"tfm-cli.conf", "r");
    if (config == NULL) {
        tanabata_path = NULL;
        struct stat st;
        if (stat(TFM_CONFIG_DIR, &st) == -1) {
            if (mkdir(TFM_CONFIG_DIR, 0755) != 0) {
                fprintf(stderr, ERROR("Failed to create %s directory. "
                                      "Try again with 'sudo' or check your permissions")"\n", TFM_CONFIG_DIR);
                return 1;
            }
        }
        config = fopen(TFM_CONFIG_DIR"tfm-cli.conf", "w");
        if (config == NULL) {
            fprintf(stderr, ERROR("Failed to create config file. "
                                  "Try again with 'sudo' or check your permissions")"\n");
            return 1;
        }
    } else {
        fseek(config, 0L, SEEK_END);
        long fsize = ftell(config);
        rewind(config);
        if (fsize == 0) {
            tanabata_path = NULL;
        } else {
            tanabata_path = malloc(fsize + 1);
            if (fgets(tanabata_path, INT32_MAX, config) == NULL) {
                fprintf(stderr, ERROR("Failed to read config file")"\n");
                return 1;
            }
        }
    }
    const char *shortopts = "hI:O:isuef:t:c:w";
    char *abspath = NULL;
    int opt;
    _Bool opt_i = 0;
    _Bool opt_s = 0;
    _Bool opt_u = 0;
    _Bool opt_e = 0;
    _Bool opt_f = 0;
    _Bool opt_t = 0;
    _Bool opt_c = 0;
    _Bool opt_w = 0;
    char *opt_f_arg;
    char *opt_t_arg;
    char *opt_c_arg;
    while ((opt = getopt(argc, argv, shortopts)) != -1) {
        switch (opt) {
            case 'h':
                printf(
                        HIGHLIGHT("(C) Masahiko AMANO aka H1K0, 2022—present")"\n"
                        HIGHLIGHT("(https://github.com/H1K0/tanabata)")"\n\n"
                        HIGHLIGHT("Usage:")"\n"
                        "tfm <options>\n\n"
                        HIGHLIGHT("Options:")"\n"
                        HIGHLIGHT("-h")"                         Print this help and exit\n"
                        HIGHLIGHT("-I <dir>")"                   Initialize new Tanabata database in directory <dir>\n"
                        HIGHLIGHT("-O <dir>")"                   Open existing Tanabata database from directory <dir>\n"
                        HIGHLIGHT("-i")"                         View database info\n"
                        HIGHLIGHT("-s")"                         Set or add\n"
                        HIGHLIGHT("-u")"                         Unset or remove\n"
                        HIGHLIGHT("-e")"                         Edit or update\n"
                        HIGHLIGHT("-f <sasa_id or path>")"       File-sasa menu\n"
                        HIGHLIGHT("-t <tanzaku_id or name>")"    Tanzaku menu\n"
                        HIGHLIGHT("-c <sasa_id>-<tanzaku_id>")"  Kazari menu "
                        "(can only be used with the '-s' or '-u' option)\n"
                        HIGHLIGHT("-w")"                         Weed (defragment) database\n"
                );
                if (tanabata_path != NULL) {
                    printf(HIGHLIGHT("Current database location: %s")"\n", tanabata_path);
                } else {
                    printf(HIGHLIGHT("No database connected")"\n");
                }
                return 0;
            case 'I':
                abspath = realpath(optarg, abspath);
                if (abspath == NULL) {
                    fprintf(stderr, ERROR("Invalid path")"\n");
                    return 1;
                }
                if (tanabata_init(&tanabata) == 0 &&
                    tanabata_dump(&tanabata, abspath) == 0) {
                    config = freopen(NULL, "w", config);
                    if (config == NULL) {
                        fprintf(stderr, ERROR("Failed to update config file. "
                                              "Try again with 'sudo' or check your permissions")"\n");
                        return 1;
                    }
                    fputs(abspath, config);
                    fclose(config);
                    printf(SUCCESS("Successfully initialized Tanabata database")"\n");
                    return 0;
                }
                fprintf(stderr, ERROR("Failed to initialize Tanabata database")"\n");
                return 1;
            case 'O':
                abspath = realpath(optarg, abspath);
                if (abspath == NULL) {
                    fprintf(stderr, ERROR("Invalid path")"\n");
                    return 1;
                }
                if (tanabata_open(&tanabata, abspath) == 0) {
                    config = freopen(NULL, "w", config);
                    if (config == NULL) {
                        fprintf(stderr, ERROR("Failed to update config file. "
                                              "Try again with 'sudo' or check your permissions")"\n");
                        return 1;
                    }
                    fputs(abspath, config);
                    fclose(config);
                    printf(SUCCESS("Successfully opened Tanabata database")"\n");
                    return 0;
                }
                fprintf(stderr, ERROR("Failed to open Tanabata database")"\n");
                return 1;
            case 'i':
                opt_i = 1;
                break;
            case 's':
                opt_s = 1;
                break;
            case 'u':
                opt_u = 1;
                break;
            case 'e':
                opt_e = 1;
                break;
            case 'f':
                opt_f = 1;
                opt_f_arg = optarg;
                break;
            case 't':
                opt_t = 1;
                opt_t_arg = optarg;
                break;
            case 'c':
                opt_c = 1;
                opt_c_arg = optarg;
                break;
            case 'w':
                opt_w = 1;
                break;
            case '?':
                return 1;
            default:
                break;
        }
    }
    if (tanabata_path == NULL) {
        fprintf(stderr, ERROR("No connected database")"\n");
        return 1;
    }
    if (tanabata_open(&tanabata, tanabata_path) != 0) {
        fprintf(stderr, ERROR("Failed to load database")"\n");
        return 1;
    }
    fclose(config);
    if (opt_i) {
        char datetime[20];
        printf(HIGHLIGHT("Current database location: %s")"\n\n"
               HIGHLIGHT("SASAHYOU")"\n", tanabata_path);
        strftime(datetime, 20, DT_FORMAT,
                 localtime((const time_t *) &tanabata.sasahyou.created_ts));
        printf("  "HIGHLIGHT("Created")"             %s\n", datetime);
        strftime(datetime, 20, DT_FORMAT,
                 localtime((const time_t *) &tanabata.sasahyou.modified_ts));
        printf("  "HIGHLIGHT("Last modified")"       %s\n"
               "  "HIGHLIGHT("Number of sasa")"      %lu\n"
               "  "HIGHLIGHT("Number of holes")"     %lu\n\n"
               HIGHLIGHT("SAPPYOU")"\n", datetime, tanabata.sasahyou.size, tanabata.sasahyou.hole_cnt);
        strftime(datetime, 20, DT_FORMAT,
                 localtime((const time_t *) &tanabata.sappyou.created_ts));
        printf("  "HIGHLIGHT("Created")"             %s\n", datetime);
        strftime(datetime, 20, DT_FORMAT,
                 localtime((const time_t *) &tanabata.sappyou.modified_ts));
        printf("  "HIGHLIGHT("Last modified")"       %s\n"
               "  "HIGHLIGHT("Number of tanzaku")"   %lu\n"
               "  "HIGHLIGHT("Number of holes")"     %lu\n\n"
               HIGHLIGHT("SHOPPYOU")"\n", datetime, tanabata.sappyou.size, tanabata.sappyou.hole_cnt);
        strftime(datetime, 20, DT_FORMAT,
                 localtime((const time_t *) &tanabata.shoppyou.created_ts));
        printf("  "HIGHLIGHT("Created")"             %s\n", datetime);
        strftime(datetime, 20, DT_FORMAT,
                 localtime((const time_t *) &tanabata.shoppyou.modified_ts));
        printf("  "HIGHLIGHT("Last modified")"       %s\n"
               "  "HIGHLIGHT("Number of kazari")"    %lu\n"
               "  "HIGHLIGHT("Number of holes")"     %lu\n",
               datetime, tanabata.shoppyou.size, tanabata.shoppyou.hole_cnt);
        return 0;
    }
    free(tanabata_path);
    if (opt_w) {
        if (tanabata_weed(&tanabata) == 0 &&
            tanabata_save(&tanabata) == 0) {
            printf(SUCCESS("Successfully weeded database")"\n");
            return 0;
        }
        fprintf(stderr, ERROR("Failed to weed database")"\n");
        return 1;
    }
    if (opt_s && opt_u) {
        opt_s = 0;
        opt_u = 0;
    }
    if (opt_s) {
        if (opt_f) {
            return menu_add_sasa(opt_f_arg);
        }
        if (opt_t) {
            return menu_add_tanzaku(opt_t_arg);
        }
        if (opt_c) {
            return menu_add_kazari(opt_c_arg);
        }
    } else if (opt_u) {
        if (opt_f) {
            return menu_rem_sasa(opt_f_arg);
        }
        if (opt_t) {
            return menu_rem_tanzaku(opt_t_arg);
        }
        if (opt_c) {
            return menu_rem_kazari(opt_c_arg);
        }
    } else if (opt_e) {
        if (opt_f) {
            return menu_upd_sasa(opt_f_arg);
        }
        if (opt_t) {
            return menu_upd_tanzaku(opt_t_arg);
        }
    } else {
        if (opt_f) {
            return menu_view_sasa(opt_f_arg);
        }
        if (opt_t) {
            return menu_view_tanzaku(opt_t_arg);
        }
    }
    return 0;
}