#include "../mass_storage_app_i.h"

#define WRITE_BUF_LEN 4096

static void mass_storage_file_name_text_callback(void* context) {
    furi_assert(context);

    MassStorageApp* app = context;
    view_dispatcher_send_custom_event(app->view_dispatcher, MassStorageCustomEventNameInput);
}

static void mass_storage_create_image(Storage* storage, const char* file_path, uint32_t size) {
    FURI_LOG_I("TAG", "Creating image %s, len:%lu", file_path, size);
    File* file = storage_file_alloc(storage);

    do {
        if(!storage_file_open(file, file_path, FSAM_WRITE, FSOM_CREATE_ALWAYS)) break;
        uint32_t size_left = size;
        uint8_t* buf = malloc(WRITE_BUF_LEN);
        memset(buf, 0, WRITE_BUF_LEN);
        while(size_left > 0) {
            uint32_t wr_len = size_left;
            if(wr_len > WRITE_BUF_LEN) {
                wr_len = WRITE_BUF_LEN;
            }
            if(storage_file_write(file, buf, wr_len) != wr_len) break;
            size_left -= wr_len;
        }
        free(buf);

    } while(false);

    storage_file_close(file);
    storage_file_free(file);
}

void mass_storage_scene_file_name_on_enter(void* context) {
    MassStorageApp* app = context;

    text_input_set_header_text(app->text_input, "Enter image name");
    ValidatorIsFile* validator_is_file =
        validator_is_file_alloc_init(MASS_STORAGE_APP_PATH_FOLDER, MASS_STORAGE_APP_EXTENSION, "");
    text_input_set_validator(app->text_input, validator_is_file_callback, validator_is_file);

    text_input_set_result_callback(
        app->text_input,
        mass_storage_file_name_text_callback,
        app,
        app->new_file_name,
        MASS_STORAGE_FILE_NAME_LEN,
        true);
    view_dispatcher_switch_to_view(app->view_dispatcher, MassStorageAppViewTextInput);
}

bool mass_storage_scene_file_name_on_event(void* context, SceneManagerEvent event) {
    UNUSED(event);
    MassStorageApp* app = context;

    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == MassStorageCustomEventNameInput) {
            mass_storage_app_show_loading_popup(app, true);
            furi_string_printf(
                app->file_path,
                "%s/%s%s",
                MASS_STORAGE_APP_PATH_FOLDER,
                app->new_file_name,
                MASS_STORAGE_APP_EXTENSION);
            mass_storage_create_image(
                app->fs_api, furi_string_get_cstr(app->file_path), app->new_file_size);
            scene_manager_next_scene(app->scene_manager, MassStorageSceneWork);
        }
    }
    return consumed;
}

void mass_storage_scene_file_name_on_exit(void* context) {
    UNUSED(context);
    MassStorageApp* app = context;
    void* validator_context = text_input_get_validator_callback_context(app->text_input);
    text_input_set_validator(app->text_input, NULL, NULL);
    validator_is_file_free(validator_context);
    text_input_reset(app->text_input);
}