#include "archive_i.h"

static bool archive_get_filenames(ArchiveApp* archive);

static void update_offset(ArchiveApp* archive) {
    furi_assert(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    size_t array_size = files_array_size(model->files);
    uint16_t bounds = array_size > 3 ? 2 : array_size;

    if(model->list_offset < model->idx - bounds) {
        model->list_offset = CLAMP(model->list_offset + 1, array_size - (bounds + 2), 0);
    } else if(model->list_offset > model->idx - bounds) {
        model->list_offset = CLAMP(model->idx - 1, array_size - (bounds), 0);
    }

    view_commit_model(archive->view_archive_main, true);
}

static void archive_update_last_idx(ArchiveApp* archive) {
    furi_assert(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);

    archive->browser.last_idx[archive->browser.depth] =
        CLAMP(model->idx, files_array_size(model->files) - 1, 0);
    model->idx = 0;
    view_commit_model(archive->view_archive_main, true);

    model = NULL;
}

static void archive_switch_dir(ArchiveApp* archive, const char* path) {
    furi_assert(archive);
    furi_assert(path);

    string_set(archive->browser.path, path);
    archive_get_filenames(archive);
}

static void archive_switch_tab(ArchiveApp* archive) {
    furi_assert(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);

    model->tab_idx = archive->browser.tab_id;
    model->idx = 0;
    view_commit_model(archive->view_archive_main, true);

    model = NULL;

    archive->browser.depth = 0;
    archive_switch_dir(archive, tab_default_paths[archive->browser.tab_id]);

    update_offset(archive);
}

static void archive_leave_dir(ArchiveApp* archive) {
    furi_assert(archive);

    char* path_ptr = stringi_get_cstr(archive->browser.path);
    char* last_char = strrchr(path_ptr, '/');
    if(last_char) path_ptr[last_char - path_ptr] = '\0';

    archive->browser.depth = CLAMP(archive->browser.depth - 1, MAX_DEPTH, 0);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    model->idx = archive->browser.last_idx[archive->browser.depth];
    view_commit_model(archive->view_archive_main, true);

    archive_switch_dir(archive, string_get_cstr(archive->browser.path));

    update_offset(archive);

    model = NULL;
    path_ptr = NULL;
    last_char = NULL;
}

static void archive_enter_dir(ArchiveApp* archive, string_t name) {
    furi_assert(archive);
    furi_assert(name);

    archive_update_last_idx(archive);
    archive->browser.depth = CLAMP(archive->browser.depth + 1, MAX_DEPTH, 0);

    string_cat(archive->browser.path, "/");
    string_cat(archive->browser.path, archive->browser.name);

    archive_switch_dir(archive, string_get_cstr(archive->browser.path));

    update_offset(archive);
}

static bool filter_by_extension(ArchiveApp* archive, FileInfo* file_info, char* name) {
    furi_assert(archive);
    furi_assert(file_info);
    furi_assert(name);

    bool result = false;
    const char* filter_ext_ptr = get_tab_ext(archive->browser.tab_id);

    if(strcmp(filter_ext_ptr, "*") == 0) {
        result = true;
    } else if(strstr(name, filter_ext_ptr) != NULL) {
        result = true;
    }

    return result;
}

static void set_file_type(ArchiveFile_t* file, FileInfo* file_info) {
    furi_assert(file);
    furi_assert(file_info);

    for(size_t i = 0; i < SIZEOF_ARRAY(known_ext); i++) {
        if(string_search_str(file->name, known_ext[i], 0) != STRING_FAILURE) {
            file->type = i;
            return;
        }
    }

    if(file_info->flags & FSF_DIRECTORY) {
        file->type = ArchiveFileTypeFolder;
    } else {
        file->type = ArchiveFileTypeUnknown;
    }
}

static bool archive_get_filenames(ArchiveApp* archive) {
    furi_assert(archive);
    FS_Dir_Api* dir_api = &archive->fs_api->dir;
    ArchiveFile_t item;
    FileInfo file_info;
    File directory;
    string_t name;
    bool result;

    string_init_printf(name, "%0*d\n", MAX_NAME_LEN, 0);
    result = dir_api->open(&directory, string_get_cstr(archive->browser.path));

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    files_array_clear(model->files);

    if(!result) {
        dir_api->close(&directory);
        string_clear(name);
        return false;
    }

    while(1) {
        char* name_ptr = stringi_get_cstr(name);
        result = dir_api->read(&directory, &file_info, name_ptr, MAX_NAME_LEN);

        if(directory.error_id == FSE_NOT_EXIST || name_ptr[0] == 0) {
            view_commit_model(archive->view_archive_main, true);
            break;
        }

        if(result) {
            if(directory.error_id == FSE_OK) {
                if(filter_by_extension(archive, &file_info, name_ptr)) {
                    ArchiveFile_t_init(&item);
                    string_init_set(item.name, name);
                    set_file_type(&item, &file_info);

                    files_array_push_back(model->files, item);
                    ArchiveFile_t_clear(&item);
                }
            } else {
                dir_api->close(&directory);
                string_clear(name);
                view_commit_model(archive->view_archive_main, true);
                return false;
            }
        }
    }

    view_commit_model(archive->view_archive_main, true);
    model = NULL;

    dir_api->close(&directory);
    string_clear(name);
    return true;
}

static void archive_exit_callback(ArchiveApp* archive) {
    furi_assert(archive);

    AppEvent event;
    event.type = EventTypeExit;
    furi_check(osMessageQueuePut(archive->event_queue, &event, 0, osWaitForever) == osOK);
}

static uint32_t archive_previous_callback(void* context) {
    return ArchiveViewMain;
}

/* file menu */
static void archive_add_to_favourites(ArchiveApp* archive) {
    furi_assert(archive);

    FS_Common_Api* common_api = &archive->fs_api->common;

    string_t buffer_src;
    string_t buffer_dst;

    string_init_set(buffer_src, archive->browser.path);
    string_cat(buffer_src, "/");
    string_cat(buffer_src, archive->browser.name);

    string_init_set_str(buffer_dst, "/favourites/");
    string_cat(buffer_dst, archive->browser.name);

    common_api->rename(string_get_cstr(buffer_src), string_get_cstr(buffer_dst));

    string_clear(buffer_src);
    string_clear(buffer_dst);
}

static void archive_text_input_callback(void* context, char* text) {
    furi_assert(context);
    furi_assert(text);

    ArchiveApp* archive = (ArchiveApp*)context;
    FS_Common_Api* common_api = &archive->fs_api->common;

    string_t buffer_src;
    string_t buffer_dst;

    string_init_set(buffer_src, archive->browser.path);
    string_init_set(buffer_dst, archive->browser.path);

    string_cat(buffer_src, "/");
    string_cat(buffer_dst, "/");

    string_cat(buffer_src, archive->browser.name);
    string_cat_str(buffer_dst, text);

    // append extension
    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    ArchiveFile_t* file =
        files_array_get(model->files, CLAMP(model->idx, files_array_size(model->files) - 1, 0));
    string_cat(buffer_src, known_ext[file->type]);
    string_cat(buffer_dst, known_ext[file->type]);
    model = NULL;
    file = NULL;

    common_api->rename(string_get_cstr(buffer_src), string_get_cstr(buffer_dst));

    view_dispatcher_switch_to_view(archive->view_dispatcher, ArchiveViewMain);

    string_clear(buffer_src);
    string_clear(buffer_dst);

    archive_get_filenames(archive);
}
static void archive_enter_text_input(ArchiveApp* archive) {
    furi_assert(archive);

    string_set(archive->browser.text_input_buffer, archive->browser.name);

    archive_trim_file_ext(archive->browser.text_input_buffer);

    char* text_input_buffer_ptr = stringi_get_cstr(archive->browser.text_input_buffer);

    text_input_set_header_text(archive->text_input, "Rename:");

    text_input_set_result_callback(
        archive->text_input,
        archive_text_input_callback,
        archive,
        text_input_buffer_ptr,
        MAX_NAME_LEN);

    view_dispatcher_switch_to_view(archive->view_dispatcher, ArchiveViewTextInput);
}

static void archive_show_file_menu(ArchiveApp* archive) {
    furi_assert(archive);

    archive->browser.menu = true;

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    model->menu = true;
    model->menu_idx = 0;
    view_commit_model(archive->view_archive_main, true);
}

static void archive_close_file_menu(ArchiveApp* archive) {
    furi_assert(archive);

    archive->browser.menu = false;

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    model->menu = false;
    model->menu_idx = 0;
    view_commit_model(archive->view_archive_main, true);
}

static void archive_open_app(ArchiveApp* archive, const char* app_name, const char* args) {
    furi_assert(archive);
    furi_assert(app_name);

    app_loader_start(app_name, args);
}

static void archive_delete_file(ArchiveApp* archive, string_t name) {
    furi_assert(archive);
    furi_assert(name);

    FS_Common_Api* common_api = &archive->fs_api->common;

    string_t path;
    string_init_set(path, archive->browser.path);
    string_cat(path, "/");
    string_cat(path, name);

    common_api->remove(string_get_cstr(path));
    string_clear(path);

    archive_get_filenames(archive);

    update_offset(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    model->idx = CLAMP(model->idx, files_array_size(model->files) - 1, 0);
    view_commit_model(archive->view_archive_main, true);
}

static void archive_file_menu_callback(ArchiveApp* archive) {
    furi_assert(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    ArchiveFile_t* selected = files_array_get(model->files, model->idx);

    switch(model->menu_idx) {
    case 0:
        if((selected->type != ArchiveFileTypeFolder && selected->type != ArchiveFileTypeUnknown)) {
            string_t full_path;
            string_init_set(full_path, archive->browser.path);
            string_cat(full_path, "/");
            string_cat(full_path, selected->name);

            archive_open_app(
                archive, flipper_app_name[selected->type], string_get_cstr(full_path));

            string_clear(full_path);
        }
        break;
    case 1:

        string_set(archive->browser.name, selected->name);
        archive_add_to_favourites(archive);
        archive_close_file_menu(archive);
        break;
    case 2:
        // open rename view
        archive_enter_text_input(archive);
        break;
    case 3:
        // confirmation?
        archive_delete_file(archive, selected->name);
        archive_close_file_menu(archive);
        break;

    default:
        archive_close_file_menu(archive);
        break;
    }

    model = NULL;
    selected = NULL;
}

static void menu_input_handler(ArchiveApp* archive, InputEvent* event) {
    furi_assert(archive);
    furi_assert(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    if(event->type == InputTypeShort) {
        if(event->key == InputKeyUp) {
            model->menu_idx = CLAMP(model->menu_idx - 1, MENU_ITEMS - 1, 0);
        } else if(event->key == InputKeyDown) {
            model->menu_idx = CLAMP(model->menu_idx + 1, MENU_ITEMS - 1, 0);
        } else if(event->key == InputKeyOk) {
            archive_file_menu_callback(archive);
        } else if(event->key == InputKeyBack) {
            archive_close_file_menu(archive);
        }
    }
    view_commit_model(archive->view_archive_main, true);
}

/* main controls */

static bool archive_view_input(InputEvent* event, void* context) {
    furi_assert(event);
    furi_assert(context);

    ArchiveApp* archive = context;
    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    bool in_menu = archive->browser.menu;

    if(in_menu) {
        menu_input_handler(archive, event);
        return true;
    }

    if(event->type == InputTypeShort) {
        if(event->key == InputKeyLeft) {
            archive->browser.tab_id = CLAMP(archive->browser.tab_id - 1, ArchiveTabTotal, 0);
            archive_switch_tab(archive);
            return true;

        } else if(event->key == InputKeyRight) {
            archive->browser.tab_id = CLAMP(archive->browser.tab_id + 1, ArchiveTabTotal - 1, 0);
            archive_switch_tab(archive);
            return true;

        } else if(event->key == InputKeyBack) {
            if(archive->browser.depth == 0) {
                archive_exit_callback(archive);
            } else {
                archive_leave_dir(archive);
            }

            return true;
        }
    }

    size_t num_elements = files_array_size(model->files) - 1;
    if((event->type == InputTypeShort || event->type == InputTypeRepeat)) {
        if(event->key == InputKeyUp) {
            model->idx = CLAMP(model->idx - 1, num_elements, 0);
            update_offset(archive);
            return true;
        } else if(event->key == InputKeyDown) {
            model->idx = CLAMP(model->idx + 1, num_elements, 0);
            update_offset(archive);
            return true;
        }
    }

    if(event->key == InputKeyOk) {
        if(files_array_size(model->files) > 0) {
            ArchiveFile_t* selected = files_array_get(model->files, model->idx);
            string_set(archive->browser.name, selected->name);
            model = NULL;

            if(selected->type == ArchiveFileTypeFolder) {
                if(event->type == InputTypeShort) {
                    archive_enter_dir(archive, archive->browser.name);
                } else if(event->type == InputTypeLong) {
                    archive_show_file_menu(archive);
                }
            } else {
                if(event->type == InputTypeShort) {
                    archive_show_file_menu(archive);
                }
            }
        }
    }

    return true;
}

void archive_free(ArchiveApp* archive) {
    furi_assert(archive);

    ArchiveViewModel* model = view_get_model(archive->view_archive_main);
    files_array_clear(model->files);
    model = NULL;

    string_clear(archive->browser.name);
    string_clear(archive->browser.path);
    string_clear(archive->browser.text_input_buffer);

    text_input_free(archive->text_input);

    furi_record_close("sdcard");
    archive->fs_api = NULL;

    view_free(archive->view_archive_main);

    view_dispatcher_remove_view(archive->view_dispatcher, ArchiveViewMain);

    view_dispatcher_remove_view(archive->view_dispatcher, ArchiveViewTextInput);

    view_dispatcher_free(archive->view_dispatcher);

    furi_record_close("gui");
    archive->gui = NULL;

    furi_thread_free(archive->app_thread);

    furi_check(osMessageQueueDelete(archive->event_queue) == osOK);

    free(archive);
}

ArchiveApp* archive_alloc() {
    ArchiveApp* archive = furi_alloc(sizeof(ArchiveApp));

    archive->event_queue = osMessageQueueNew(2, sizeof(AppEvent), NULL);
    archive->app_thread = furi_thread_alloc();
    archive->gui = furi_record_open("gui");
    archive->view_dispatcher = view_dispatcher_alloc();
    archive->fs_api = furi_record_open("sdcard");
    archive->text_input = text_input_alloc();
    archive->view_archive_main = view_alloc();

    furi_check(archive->event_queue);

    view_allocate_model(
        archive->view_archive_main, ViewModelTypeLockFree, sizeof(ArchiveViewModel));
    view_set_context(archive->view_archive_main, archive);
    view_set_draw_callback(archive->view_archive_main, archive_view_render);
    view_set_input_callback(archive->view_archive_main, archive_view_input);
    view_set_previous_callback(
        text_input_get_view(archive->text_input), archive_previous_callback);

    view_dispatcher_add_view(
        archive->view_dispatcher, ArchiveViewMain, archive->view_archive_main);
    view_dispatcher_add_view(
        archive->view_dispatcher, ArchiveViewTextInput, text_input_get_view(archive->text_input));
    view_dispatcher_attach_to_gui(
        archive->view_dispatcher, archive->gui, ViewDispatcherTypeFullscreen);

    view_dispatcher_switch_to_view(archive->view_dispatcher, ArchiveTabFavourites);

    return archive;
}

int32_t app_archive(void* p) {
    ArchiveApp* archive = archive_alloc();

    // default tab
    archive_switch_tab(archive);

    AppEvent event;
    while(1) {
        furi_check(osMessageQueueGet(archive->event_queue, &event, NULL, osWaitForever) == osOK);
        if(event.type == EventTypeExit) {
            break;
        }
    }

    archive_free(archive);
    return 0;
}