#include "FileSelectScreen.hpp"
#include "../MainProgram.hpp"
#include "../GUIHolder.hpp"
#include "../GUIStuff/ElementHelpers/TextLabelHelpers.hpp"
#include "../GUIStuff/ElementHelpers/ButtonHelpers.hpp"
#include "../GUIStuff/ElementHelpers/ScrollAreaHelpers.hpp"
#include "Helpers/ConvertVec.hpp"
#include "../GUIStuff/Elements/GridScrollArea.hpp"
#include "../GUIStuff/Elements/ImageDisplay.hpp"
#include "../GUIStuff/Elements/LayoutElement.hpp"
#include "../GUIStuff/Elements/SVGIcon.hpp"
#include "../GUIStuff/Elements/PositionAdjustingPopupMenu.hpp"
#include "../World.hpp"
#include "Helpers/StringHelpers.hpp"
#include "DrawingProgramScreen.hpp"
#include <SDL3/SDL_time.h>
#include <SDL3/SDL_timer.h>

#define MAIN_MENU_SIZE 300

using namespace GUIStuff;
using namespace ElementHelpers;

FileSelectScreen::FileSelectScreen(MainProgram& m): Screen(m) {
    savePath = main.conf.configPath / "saves";
    trashPath = main.conf.configPath / "trash";
    trashInfoPath = main.conf.configPath / "trashInfo.json";

    try {
        nlohmann::json j(nlohmann::json::parse(read_file_to_string(trashInfoPath)));
        j.get_to(trashInfo);
    }
    catch(...) {}

    update_file_list(fileList, savePath, false);

    // Update trash list once to get trash info up to date
    std::vector<FileInfo> trashListTemp;
    update_file_list(trashListTemp, trashPath, true);
}

void FileSelectScreen::gui_layout_run() {
    main_display();
}

void FileSelectScreen::update_file_list(std::vector<FileInfo>& fL, const std::filesystem::path& savePath, bool trashUpdate) {
    constexpr SDL_Time NS_IN_DAY = SDL_SECONDS_TO_NS(24 * 3600);

    SDL_Time currentTime;
    SDL_GetCurrentTime(&currentTime);

    SDL_Time afterThirtyDays = currentTime + NS_IN_DAY * 30;

    fL.clear();
    int globCount;
    char** filesInPath = SDL_GlobDirectory(savePath.string().c_str(), "*", 0, &globCount);
    if(filesInPath) {
        for(int i = 0; i < globCount; i++) {
            std::filesystem::path p = filesInPath[i];
            std::string fExt = p.extension().string();
            if(fExt == "." + World::FILE_EXTENSION) {
                std::filesystem::path fullPath = savePath / filesInPath[i];
                FileInfo fileInfoToAdd;
                fileInfoToAdd.fileName = p.stem().string();

                if(trashUpdate) {
                    auto it = trashInfo.files.find(fileInfoToAdd.fileName);
                    if(it == trashInfo.files.end()) {
                        trashInfo.files[fileInfoToAdd.fileName].trashTime = currentTime;
                        fileInfoToAdd.lastModifyTime = currentTime;
                    }
                    else
                        fileInfoToAdd.lastModifyTime = it->second.trashTime;

                    if(fileInfoToAdd.lastModifyTime >= afterThirtyDays) {
                        // Remove file and thumbnail
                        SDL_RemovePath(fullPath.string().c_str());
                        SDL_RemovePath((savePath / fileInfoToAdd.fileName / ".jpg").string().c_str());
                        trashInfo.files.erase(it);
                        continue;
                    }

                    int64_t daysToDeletion = (afterThirtyDays - fileInfoToAdd.lastModifyTime) / NS_IN_DAY;

                    if(daysToDeletion == 0)
                        fileInfoToAdd.lastModifyDate = "Today";
                    else if(daysToDeletion == 1)
                        fileInfoToAdd.lastModifyDate = "Tomorrow";
                    else
                        fileInfoToAdd.lastModifyDate = std::to_string(daysToDeletion) + " days";
                }
                else {
                    SDL_PathInfo pathInfo;
                    if(SDL_GetPathInfo(fullPath.string().c_str(), &pathInfo)) {
                        SDL_DateTime pathDt;
                        fileInfoToAdd.lastModifyTime = pathInfo.modify_time;
                        if(SDL_TimeToDateTime(pathInfo.modify_time, &pathDt, true)) {
                            fileInfoToAdd.lastModifyDate = sdl_time_to_nice_access_time(pathDt, main.conf.dateFormat, main.conf.timeFormat);
                        }
                    }
                }

                fL.emplace_back(fileInfoToAdd);
            }
        }
    }
    else
        SDL_CreateDirectory(savePath.string().c_str());
}

void FileSelectScreen::main_display() {
    auto& gui = main.g.gui;
    CLAY_AUTO_ID({
        .layout = {
            .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_GROW(0)},
            .layoutDirection = CLAY_LEFT_TO_RIGHT
        },
    }) {
        if(mainMenuOpen)
            main_menu();
        if(!(mainMenuOpen && main_menu_fills_screen())) {
            CLAY_AUTO_ID({
                .layout = {
                    .sizing = {.width = CLAY_SIZING_FIXED(gui.io.windowSize.x()), .height = CLAY_SIZING_GROW(0)},
                    .childAlignment = { .x = CLAY_ALIGN_X_CENTER, .y = CLAY_ALIGN_Y_TOP },
                    .layoutDirection = CLAY_TOP_TO_BOTTOM,
                },
            }) {
                CLAY_AUTO_ID({
                    .layout = {
                        .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_GROW(0)},
                        .padding = CLAY_PADDING_ALL(gui.io.theme->padding1),
                        .childGap = gui.io.theme->childGap1,
                        .childAlignment = { .x = CLAY_ALIGN_X_CENTER, .y = CLAY_ALIGN_Y_TOP },
                        .layoutDirection = CLAY_TOP_TO_BOTTOM,
                    },
                }) {
                    if(editMode) {
                        numberOfSelectedEntries = std::count_if(fileList.begin(), fileList.end(), [] (const FileInfo& f) { return f.selected; });
                        edit_title_bar();
                    }
                    else
                        title_bar();
                    switch(selectedMenu) {
                        case SelectedMenu::FILES:
                            file_view();
                            break;
                        case SelectedMenu::TRASH:
                            file_view();
                            break;
                        case SelectedMenu::SETTINGS:
                            break;
                    }
                    menu_black_box();
                }
                if(editMode && numberOfSelectedEntries != 0)
                    edit_action_bar();
            }
        }
    }
}

bool FileSelectScreen::main_menu_fills_screen() {
    return main.g.gui.io.windowSize.x() < MAIN_MENU_SIZE + 50;
}

void FileSelectScreen::edit_action_bar_button(const char* id, const std::string& svgPath, const char* text, const std::function<void()>& onClick) {
    auto& gui = main.g.gui;
    CLAY_AUTO_ID({
        .layout = {.sizing = {.width = CLAY_SIZING_FIXED(70), .height = CLAY_SIZING_GROW(0)}}
    }) {
        gui.element<SelectableButton>(id, SelectableButton::Data{
            .drawType = SelectableButton::DrawType::TRANSPARENT_ALL,
            .onClick = onClick,
            .innerContent = [&](const SelectableButton::InnerContentCallbackParameters&) {
                CLAY_AUTO_ID({
                    .layout = {
                        .sizing = {.width = CLAY_SIZING_FIT(0), .height = CLAY_SIZING_FIT(0)},
                        .childAlignment = { .x = CLAY_ALIGN_X_CENTER, .y = CLAY_ALIGN_Y_TOP },
                        .layoutDirection = CLAY_TOP_TO_BOTTOM,
                    }
                }) {
                    CLAY_AUTO_ID({
                        .layout = {.sizing = {.width = CLAY_SIZING_FIXED(20), .height = CLAY_SIZING_FIXED(20)}}
                    }) {
                        gui.element<SVGIcon>("icon", svgPath, false);
                    }
                    text_label(gui, text);
                }
            }
        });
    }
}

void FileSelectScreen::move_selected_files(const std::filesystem::path& fromPath, const std::filesystem::path& toPath, TrashMoveType trashMoveType) {
    SDL_Time currentTime;
    SDL_GetCurrentTime(&currentTime);

    std::vector<std::string> toFolderListNames;

    int globCount;
    char** filesInPath = SDL_GlobDirectory(toPath.c_str(), "*", 0, &globCount);
    if(filesInPath) {
        for(int i = 0; i < globCount; i++) {
            std::filesystem::path p(filesInPath[i]);
            toFolderListNames.emplace_back(p.filename().stem());
        }
    }
    else
        SDL_CreateDirectory(toPath.string().c_str());

    for(const FileInfo& f : fileList) {
        if(f.selected) {
            std::string newFileName = ensure_string_unique(toFolderListNames, f.fileName);
            std::filesystem::path filePath = fromPath / (f.fileName + "." + World::FILE_EXTENSION);
            std::filesystem::path thumbnailPath = fromPath / (f.fileName + ".jpg");
            std::filesystem::path newFilePath = toPath / (newFileName + "." + World::FILE_EXTENSION);
            std::filesystem::path newThumbnailPath = toPath / (newFileName + ".jpg");
            SDL_RenamePath(filePath.string().c_str(), newFilePath.string().c_str());
            SDL_RenamePath(thumbnailPath.string().c_str(), newThumbnailPath.string().c_str());
            switch(trashMoveType) {
                case TrashMoveType::NONE:
                    break;
                case TrashMoveType::MOVE_TO_TRASH:
                    trashInfo.files[newFileName] = TrashInfo::TrashFile{
                        .trashTime = currentTime
                    };
                    break;
                case TrashMoveType::MOVE_OUT_OF_TRASH:
                    trashInfo.files.erase(f.fileName);
                    break;
            }
        }
    }
}

void FileSelectScreen::duplicate_selected_files(const std::filesystem::path& inPath) {
    std::vector<std::string> toFolderListNames;

    int globCount;
    char** filesInPath = SDL_GlobDirectory(inPath.c_str(), "*", 0, &globCount);
    if(filesInPath) {
        for(int i = 0; i < globCount; i++) {
            std::filesystem::path p(filesInPath[i]);
            toFolderListNames.emplace_back(p.filename().stem());
        }
    }
    else
        SDL_CreateDirectory(inPath.string().c_str());

    for(const FileInfo& f : fileList) {
        if(f.selected) {
            std::string newFileName = ensure_string_unique(toFolderListNames, f.fileName);
            std::filesystem::path filePath = inPath / (f.fileName + World::DOT_FILE_EXTENSION);
            std::filesystem::path thumbnailPath = inPath / (f.fileName + ".jpg");
            std::filesystem::path newFilePath = inPath / (newFileName + "." + World::FILE_EXTENSION);
            std::filesystem::path newThumbnailPath = inPath / (newFileName + ".jpg");
            SDL_CopyFile(filePath.string().c_str(), newFilePath.string().c_str());
            SDL_CopyFile(thumbnailPath.string().c_str(), newThumbnailPath.string().c_str());
        }
    }
}

void FileSelectScreen::delete_selected_files_in_trash() {
    for(const FileInfo& f : fileList) {
        if(f.selected) {
            std::filesystem::path filePath = trashPath / (f.fileName + World::DOT_FILE_EXTENSION);
            std::filesystem::path thumbnailPath = trashPath / (f.fileName + ".jpg");
            SDL_RemovePath(filePath.string().c_str());
            SDL_RemovePath(thumbnailPath.string().c_str());
            trashInfo.files.erase(f.fileName);
        }
    }
}

void FileSelectScreen::edit_action_bar() {
    auto& gui = main.g.gui;
    gui.element<LayoutElement>("edit action bar", [&](LayoutElement*, const Clay_ElementId& lId) {
        CLAY(lId, {
            .layout = {
                .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_FIT(0)},
                .childAlignment = { .x = CLAY_ALIGN_X_CENTER, .y = CLAY_ALIGN_Y_CENTER },
                .layoutDirection = CLAY_LEFT_TO_RIGHT,
            },
            .backgroundColor = convert_vec4<Clay_Color>(gui.io.theme->backColor1),
        }) {
            if(selectedMenu == SelectedMenu::FILES) {
                edit_action_bar_button("trash", "data/icons/trash.svg", "Trash", [&] {
                    move_selected_files(savePath, trashPath, TrashMoveType::MOVE_TO_TRASH);
                    update_file_list(fileList, savePath, false);
                    editMode = false;
                });
                edit_action_bar_button("duplicate", "data/icons/RemixIcon/file-copy-line.svg", "Duplicate", [&] {
                    duplicate_selected_files(savePath);
                    update_file_list(fileList, savePath, false);
                    editMode = false;
                });
            }
            else if(selectedMenu == SelectedMenu::TRASH) {
                edit_action_bar_button("restore", "data/icons/RemixIcon/refresh-line.svg", "Restore", [&] {
                    move_selected_files(trashPath, savePath, TrashMoveType::MOVE_OUT_OF_TRASH);
                    update_file_list(fileList, trashPath, true);
                    editMode = false;
                });
                edit_action_bar_button("delete", "data/icons/trash.svg", "Delete", [&] {
                    delete_selected_files_in_trash();
                    update_file_list(fileList, trashPath, true);
                    editMode = false;
                });
            }
        }
    });
}

void FileSelectScreen::edit_title_bar() {
    auto& gui = main.g.gui;
    CLAY_AUTO_ID({
        .layout = {
            .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_FIT(0)},
            .childGap = gui.io.theme->childGap1,
            .childAlignment = { .x = CLAY_ALIGN_X_LEFT, .y = CLAY_ALIGN_Y_CENTER },
            .layoutDirection = CLAY_LEFT_TO_RIGHT,
        }
    }) {
        svg_icon_button(gui, "close edit mode button", "data/icons/close.svg", {
            .drawType = SelectableButton::DrawType::TRANSPARENT_ALL,
            .onClick = [&] {
                editMode = false;
            }
        });
        mutable_text_label(gui, "select files label", (numberOfSelectedEntries == 0) ? "Select files" : (std::to_string(numberOfSelectedEntries) + " selected"));
    }
}

void FileSelectScreen::title_bar() {
    auto& gui = main.g.gui;
    CLAY_AUTO_ID({
        .layout = {
            .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_FIT(0)},
            .childGap = gui.io.theme->childGap1,
            .childAlignment = { .x = CLAY_ALIGN_X_LEFT, .y = CLAY_ALIGN_Y_CENTER },
            .layoutDirection = CLAY_LEFT_TO_RIGHT,
        }
    }) {
        svg_icon_button(gui, "main settings button", "data/icons/menu.svg", {
            .drawType = SelectableButton::DrawType::TRANSPARENT_ALL,
            .onClick = [&] {
                mainMenuOpen = true;
            }
        });
        switch(selectedMenu) {
            case SelectedMenu::FILES:
                mutable_text_label(gui, "main screen header text", "Files");
                break;
            case SelectedMenu::TRASH:
                mutable_text_label(gui, "main screen header text", "Trash");
                break;
            case SelectedMenu::SETTINGS:
                mutable_text_label(gui, "main screen header text", "Settings");
                break;
        }
        CLAY_AUTO_ID({
            .layout = {
                .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_GROW(0)},
            }
        }) {}
        if(selectedMenu == SelectedMenu::FILES || selectedMenu == SelectedMenu::TRASH) {
            SelectableButton* moreButton = svg_icon_button(gui, "more settings button", "data/icons/RemixIcon/more-2-fill.svg", {
                .drawType = SelectableButton::DrawType::TRANSPARENT_ALL,
                .onClick = [&] {
                    moreOptionsMenu = MoreOptionsMenu::MAIN;
                }
            });
            if(moreButton->get_bb().has_value() && moreOptionsMenu != MoreOptionsMenu::CLOSED) {
                gui.set_z_index(gui.get_z_index() + 10, [&] {
                    gui.element<PositionAdjustingPopupMenu>("more options popup", moreButton->get_bb().value().top_right(), [&] {
                        CLAY_AUTO_ID({
                            .layout = {
                                .sizing = {.width = CLAY_SIZING_FIT(200), .height = CLAY_SIZING_FIT(0)},
                                .layoutDirection = CLAY_TOP_TO_BOTTOM
                            },
                            .backgroundColor = convert_vec4<Clay_Color>(gui.io.theme->backColor1),
                            .cornerRadius = CLAY_CORNER_RADIUS(4)
                        }) {
                            if(moreOptionsMenu == MoreOptionsMenu::MAIN) {
                                text_transparent_option_button("Edit", "Edit", [&] {
                                    moreOptionsMenu = MoreOptionsMenu::CLOSED;
                                    start_edit_mode();
                                });
                                text_transparent_option_button("View", "View", [&] {
                                    moreOptionsMenu = MoreOptionsMenu::VIEW;
                                });
                            }
                            else if(moreOptionsMenu == MoreOptionsMenu::VIEW) {
                                text_transparent_option_button("Large Grid", "Large Grid", [&] {
                                    fileViewType = FileViewType::LARGE_GRID;
                                    moreOptionsMenu = MoreOptionsMenu::CLOSED;
                                });
                                text_transparent_option_button("Medium Grid", "Medium Grid", [&] {
                                    fileViewType = FileViewType::MEDIUM_GRID;
                                    moreOptionsMenu = MoreOptionsMenu::CLOSED;
                                });
                                text_transparent_option_button("Small Grid", "Small Grid", [&] {
                                    fileViewType = FileViewType::SMALL_GRID;
                                    moreOptionsMenu = MoreOptionsMenu::CLOSED;
                                });
                                text_transparent_option_button("List", "List", [&] {
                                    fileViewType = FileViewType::LIST;
                                    moreOptionsMenu = MoreOptionsMenu::CLOSED;
                                });
                            }
                        }
                    }, LayoutElement::Callbacks{
                        .mouseButton = [&](LayoutElement* l, const InputManager::MouseButtonCallbackArgs& m) {
                            if(!l->mouseHovering && m.down) {
                                moreOptionsMenu = MoreOptionsMenu::CLOSED;
                                main.g.gui.set_to_layout();
                            }
                        }
                    });
                });
            }
        }
    }
}

void FileSelectScreen::text_transparent_option_button(const char* id, const char* text, const std::function<void()>& onClick) {
    auto& gui = main.g.gui;
    text_button(gui, id, text, {
        .drawType = SelectableButton::DrawType::TRANSPARENT_ALL,
        .wide = true,
        .centered = false,
        .onClick = onClick
    });
}

void FileSelectScreen::icon_text_transparent_option_selected_button(const char* id, const std::string& svgPath, const char* text, bool isSelected, const std::function<void()>& onClick) {
    auto& gui = main.g.gui;
    text_button_with_icon(gui, id, svgPath, text, {
        .drawType = SelectableButton::DrawType::TRANSPARENT_ALL,
        .isSelected = isSelected,
        .wide = true,
        .centered = false,
        .onClick = onClick
    });
}

void FileSelectScreen::main_menu() {
    auto& gui = main.g.gui;
    gui.element<LayoutElement>("main menu popup", [&] (LayoutElement*, const Clay_ElementId& lId) {
        CLAY(lId, {
            .layout = {
                .sizing = {.width = CLAY_SIZING_GROW(static_cast<float>(main_menu_fills_screen() ? 0 : 300)), .height = CLAY_SIZING_GROW(0)},
                .padding = CLAY_PADDING_ALL(gui.io.theme->padding1),
                .childGap = 0,
                .layoutDirection = CLAY_TOP_TO_BOTTOM,
            },
        }) {
            icon_text_transparent_option_selected_button("Files", "data/icons/folder.svg", "Files", selectedMenu == SelectedMenu::FILES, [&] {
                selectedMenu = SelectedMenu::FILES;
                update_file_list(fileList, savePath, false);
                mainMenuOpen = false;
            });
            icon_text_transparent_option_selected_button("Trash", "data/icons/trash.svg", "Trash", selectedMenu == SelectedMenu::TRASH, [&] {
                selectedMenu = SelectedMenu::TRASH;
                update_file_list(fileList, trashPath, true);
                mainMenuOpen = false;
            });
            icon_text_transparent_option_selected_button("Settings", "data/icons/RemixIcon/settings-3-line.svg", "Settings", selectedMenu == SelectedMenu::SETTINGS, [&] {
                selectedMenu = SelectedMenu::SETTINGS;
                mainMenuOpen = false;
            });
        }
    });
}

void FileSelectScreen::file_view() {
    auto& gui = main.g.gui;
    std::filesystem::path folderPath = (selectedMenu == SelectedMenu::TRASH) ? trashPath : savePath;

    auto fileButton = [&] (size_t i, bool isList, const Vector2f& iconSize) {
        std::filesystem::path filePath = folderPath / (fileList[i].fileName + "." + World::FILE_EXTENSION);
        gui.element<SelectableButton>("file button", SelectableButton::Data{
            .isSelected = editMode && fileList[i].selected,
            .onClick = [&, filePath, i] {
                if(editMode)
                    fileList[i].selected = !fileList[i].selected;
                else {
                    CustomEvents::emit_event<CustomEvents::OpenInfiniPaintFileEvent>({
                        .isClient = false,
                        .filePathSource = filePath
                    });
                }
            },
            .innerContent = [&](const SelectableButton::InnerContentCallbackParameters& c){
                if(isList) {
                    CLAY_AUTO_ID({
                        .layout = {
                            .sizing = { .width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_GROW(0)},
                            .padding = CLAY_PADDING_ALL(gui.io.theme->padding1),
                            .childGap = 6,
                            .childAlignment = { .x = CLAY_ALIGN_X_LEFT, .y = CLAY_ALIGN_Y_CENTER },
                            .layoutDirection = CLAY_LEFT_TO_RIGHT
                        }
                    }) {
                        CLAY_AUTO_ID({
                            .layout = { .sizing = {.width = CLAY_SIZING_FIXED(iconSize.x()), .height = CLAY_SIZING_FIXED(iconSize.y()) }},
                        }) {
                            gui.element<ImageDisplay>("ico", ImageDisplay::Data{
                                .imgPath = folderPath / (fileList[i].fileName + ".jpg"),
                                .radius = 20
                            });
                        }
                        CLAY_AUTO_ID({
                            .layout = {
                                .sizing = { .width = CLAY_SIZING_FIT(0), .height = CLAY_SIZING_FIT(0)},
                                .childGap = 6,
                                .childAlignment = { .x = CLAY_ALIGN_X_LEFT, .y = CLAY_ALIGN_Y_CENTER },
                                .layoutDirection = CLAY_TOP_TO_BOTTOM
                            }
                        }) {
                            text_label(gui, fileList[i].fileName);
                            text_label_light(gui, fileList[i].lastModifyDate);
                        }
                    }
                }
                else {
                    CLAY_AUTO_ID({
                        .layout = {
                            .sizing = { .width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_GROW(0)},
                            .padding = CLAY_PADDING_ALL(gui.io.theme->padding1),
                            .childGap = 6,
                            .childAlignment = { .x = CLAY_ALIGN_X_CENTER, .y = CLAY_ALIGN_Y_CENTER },
                            .layoutDirection = CLAY_TOP_TO_BOTTOM
                        }
                    }) {
                        CLAY_AUTO_ID({
                            .layout = { .sizing = {.width = CLAY_SIZING_FIXED(iconSize.x()), .height = CLAY_SIZING_FIXED(iconSize.y()) }},
                        }) {
                            gui.element<ImageDisplay>("ico", ImageDisplay::Data{
                                .imgPath = folderPath / (fileList[i].fileName + ".jpg"),
                                .radius = 20
                            });
                        }
                        text_label(gui, fileList[i].fileName);
                        text_label_light(gui, fileList[i].lastModifyDate);
                    }
                }
            }
        });
    };

    if(fileViewType == FileViewType::LIST) {
        scroll_area_many_entries(gui, "File selector list", ScrollBarManyEntriesOptions{
            .entryHeight = 150.0f,
            .entryCount = fileList.size(),
            .clipHorizontal = true,
            .elementContent = [&](size_t i) {
                fileButton(i, true, Vector2f{100.0f, 100.0f});
            }
        });
    }
    else {
        Vector2f entrySize{0.0f, 0.0f};
        Vector2f iconSize{0.0f, 0.0f};
        switch(fileViewType) {
            case FileViewType::LARGE_GRID:
                entrySize = {200.0f, 220.0f};
                iconSize = {150.0f, 150.0f};
                break;
            case FileViewType::MEDIUM_GRID:
                entrySize = {170.0f, 190.0f};
                iconSize = {120.0f, 120.0f};
                break;
            case FileViewType::SMALL_GRID:
                entrySize = {140.0f, 160.0f};
                iconSize = {90.0f, 90.0f};
                break;
            default:
                break;
        }
        gui.element<GridScrollArea>("File selector grid", GridScrollArea::Options{
            .entryWidth = entrySize.x(),
            .childAlignmentX = CLAY_ALIGN_X_LEFT,
            .entryHeight = entrySize.y(),
            .entryCount = fileList.size(),
            .elementContent = [&](size_t i) {
                fileButton(i, false, iconSize);
            }
        });
    }
}

void FileSelectScreen::menu_black_box() {
    auto& gui = main.g.gui;
    if(mainMenuOpen) {
        gui.set_z_index(gui.get_z_index() + 1, [&] {
            gui.element<LayoutElement>("file list disable fill", [&] (LayoutElement* l, const Clay_ElementId& lId) {
                CLAY(lId, {
                    .layout = {
                        .sizing = {.width = CLAY_SIZING_GROW(0), .height = CLAY_SIZING_GROW(0)},
                    },
                    .backgroundColor = {0.0f, 0.0f, 0.0f, 0.6f},
                    .floating = {
                        .zIndex = gui.get_z_index(),
                        .attachPoints = {
                            .element = CLAY_ATTACH_POINT_LEFT_TOP,
                            .parent = CLAY_ATTACH_POINT_LEFT_TOP
                        },
                        .attachTo = CLAY_ATTACH_TO_PARENT
                    }
                }) {
                }
            }, LayoutElement::Callbacks{
                .mouseButton = [&] (LayoutElement* l, const InputManager::MouseButtonCallbackArgs& m) {
                    if(m.down && m.button == InputManager::MouseButton::LEFT && l->mouseHovering) {
                        mainMenuOpen = false;
                        gui.set_to_layout();
                    }
                }
            });
        });
    }
}

void FileSelectScreen::start_edit_mode() {
    for(auto& f : fileList)
        f.selected = false;
    editMode = true;
}

void FileSelectScreen::input_open_infinipaint_file_callback(const CustomEvents::OpenInfiniPaintFileEvent& openFile) {
    main.create_new_tab(openFile);
    main.screen = std::make_unique<DrawingProgramScreen>(main);
}

void FileSelectScreen::draw(SkCanvas* canvas) {
    canvas->clear(main.g.gui.io.theme->backColor0);
}

FileSelectScreen::~FileSelectScreen() {
    nlohmann::json j;
    nlohmann::to_json(j, trashInfo);
    std::string saveJson = nlohmann::to_string(j);
    SDL_SaveFile(trashInfoPath.string().c_str(), saveJson.c_str(), saveJson.size());
}