#include "video_player.h"
#include "video_player_hal.h"
#include "init_deinit.h"

#include <video_player_icons.h>
#include <furi.h>
#include <furi/core/thread.h>
#include <furi_hal.h>
#include <cli/cli.h>
#include <gui/gui.h>
#include "furi_hal_rtc.h"

void draw_callback(Canvas* canvas, void* ctx) {
    PlayerViewModel* model = (PlayerViewModel*)ctx;
    VideoPlayerApp* player = (VideoPlayerApp*)(model->player);

    canvas_draw_xbm(canvas, 0, 0, player->width, player->height, player->image_buffer);
}

bool input_callback(InputEvent* input_event, void* ctx) {
    // Проверяем, что контекст не нулевой
    furi_assert(ctx);
    PlayerView* player_view = (PlayerView*)ctx;
    VideoPlayerApp* player = (VideoPlayerApp*)(player_view->context);

    bool consumed = false;

    VideoPlayerEvent event = {.type = EventTypeInput, .input = *input_event};

    furi_message_queue_put(player->event_queue, &event, FuriWaitForever);

    consumed = true;
    return consumed;
}

void direct_input_callback(const void* value, void* ctx) {
    // Проверяем, что контекст не нулевой
    furi_assert(ctx);
    const InputEvent* input_event = value;
    VideoPlayerApp* player = (VideoPlayerApp*)(ctx);

    VideoPlayerEvent event = {.type = EventTypeInput, .input = *input_event};

    furi_message_queue_put(player->event_queue, &event, FuriWaitForever);
}

void player_view_free(PlayerView* player_view) {
    furi_assert(player_view);
    view_free(player_view->view);
    free(player_view);
}

bool open_file_stream(Stream* stream) {
    DialogsApp* dialogs = furi_record_open(RECORD_DIALOGS);
    bool result = false;
    FuriString* path;
    path = furi_string_alloc();
    furi_string_set(path, VIDEO_PLAYER_FOLDER);

    DialogsFileBrowserOptions browser_options;
    dialog_file_browser_set_basic_options(&browser_options, ".bnd", &I_vid_logo);
    browser_options.base_path = VIDEO_PLAYER_FOLDER;
    browser_options.hide_ext = false;

    bool ret = dialog_file_browser_show(dialogs, path, path, &browser_options);

    furi_record_close(RECORD_DIALOGS);
    if(ret) {
        if(!file_stream_open(stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
            FURI_LOG_E("FUCK! ", "Cannot open file \"%s\"", furi_string_get_cstr(path));
        } else {
            result = true;
        }
    }
    furi_string_free(path);
    return result;
}

void draw_progress_bar(VideoPlayerApp* player) {
    canvas_set_color(player->canvas, ColorWhite);
    canvas_draw_box(player->canvas, 0, 57, 128, 7);
    canvas_set_color(player->canvas, ColorBlack);
    canvas_draw_frame(player->canvas, 0, 58, 128, 6);
    canvas_draw_box(player->canvas, 1, 59, player->progress, 4);
}

void draw_all(VideoPlayerApp* player) {
    canvas_reset(player->canvas);

    canvas_draw_xbm(
        player->canvas,
        player->width == 128 ? 0 : (128 - player->width) / 2,
        0,
        player->width,
        player->height,
        player->image_buffer);

    if(player->seeking) {
        draw_progress_bar(player);
    }

    canvas_commit(player->canvas);
}

int32_t video_player_app(void* p) {
    UNUSED(p);

    Storage* storage = furi_record_open(RECORD_STORAGE);
    bool st = storage_simply_mkdir(storage, APPSDATA_FOLDER);
    st = storage_simply_mkdir(storage, VIDEO_PLAYER_FOLDER);
    UNUSED(st);
    furi_record_close(RECORD_STORAGE);

    bool exit = false;

    while(!exit) {
        VideoPlayerApp* player = init_player();
        if(open_file_stream(player->stream)) {
            player->quit = false;
        } else {
            player->quit = true;
            exit = true;
        }

        if(!(player->quit)) {
            char header[8];
            header[7] = '\0';
            stream_read(player->stream, (uint8_t*)header, 7);

            if(strcmp(header, "BND!VID") != 0) {
                player->quit = true;
            }

            stream_read(player->stream, (uint8_t*)&player->version, sizeof(player->version));
            stream_read(player->stream, (uint8_t*)&player->num_frames, sizeof(player->num_frames));
            stream_read(
                player->stream,
                (uint8_t*)&player->audio_chunk_size,
                sizeof(player->audio_chunk_size));
            stream_read(
                player->stream, (uint8_t*)&player->sample_rate, sizeof(player->sample_rate));
            stream_read(player->stream, &player->height, sizeof(player->height));
            stream_read(player->stream, &player->width, sizeof(player->width));

            player->header_size = stream_tell(player->stream);

            player->buffer = (uint8_t*)malloc(
                player->audio_chunk_size * 2 +
                (uint32_t)player->height * (uint32_t)player->width / 8);
            memset(
                player->buffer,
                0,
                player->audio_chunk_size * 2 +
                    (uint32_t)player->height * (uint32_t)player->width / 8);

            player->image_buffer_length = (uint32_t)player->height * (uint32_t)player->width / 8;
            player->audio_buffer = (uint8_t*)&player->buffer[player->image_buffer_length];
            player->image_buffer = player->buffer;

            player->fake_audio_buffer = (uint8_t*)malloc(player->audio_chunk_size * 2);

            player->frame_size =
                player->audio_chunk_size + player->image_buffer_length; //for seeking
            player->frames_per_turn = player->num_frames / 126;

            player->silent = furi_hal_rtc_is_flag_set(FuriHalRtcFlagStealthMode);
        }

        if(furi_hal_speaker_acquire(1000)) {
            if(!(player->quit)) {
                player_init_hardware_and_play(player);
            }

            // Текущее событие типа кастомного типа VideoPlayerEvent
            VideoPlayerEvent event;

            //view_dispatcher_switch_to_view(player->view_dispatcher, VIEW_PLAYER);

            //switch from view dispatcher to direct draw
            view_dispatcher_remove_view(player->view_dispatcher, VIEW_PLAYER);

            view_dispatcher_free(player->view_dispatcher);

            player_view_free(player->player_view);
            furi_record_close(RECORD_GUI);

            player->input = furi_record_open(RECORD_INPUT_EVENTS);
            player->gui = furi_record_open(RECORD_GUI);
            player->canvas = gui_direct_draw_acquire(player->gui);

            player->input_subscription =
                furi_pubsub_subscribe(player->input, direct_input_callback, player);

            if(player->quit) {
                deinit_player(player);
                player_deinit_hardware();
                return 0;
            }

            player->playing = true;

            //vTaskPrioritySet(furi_thread_get_current_id(), FuriThreadPriorityIdle);
            furi_thread_set_current_priority(FuriThreadPriorityIdle);

            while(!(player->quit)) {
                furi_check(
                    furi_message_queue_get(player->event_queue, &event, FuriWaitForever) ==
                    FuriStatusOk);

                if(event.type == EventTypeInput) {
                    if(event.input.key == InputKeyBack) {
                        player->quit = true;
                    }

                    if(event.input.key == InputKeyOk) {
                        player->playing = !player->playing;
                    }

                    if(event.input.key == InputKeyLeft) {
                        player->seeking = true;
                        int32_t seek = CLAMP(
                            (int32_t)stream_tell(player->stream) -
                                player->frames_per_turn * player->frame_size,
                            (int32_t)player->num_frames * player->frame_size + player->header_size,
                            player->header_size);
                        stream_seek(player->stream, seek, StreamOffsetFromStart);

                        player->progress = (uint8_t)((int64_t)stream_tell(player->stream) * (int64_t)126 / ((int64_t)player->num_frames * (int64_t)player->frame_size + (int64_t)player->header_size));

                        if(event.input.type == InputTypeRelease) {
                            player->seeking = false;
                        }

                        static VideoPlayerEvent event = {.type = EventTypeJustRedraw};
                        furi_message_queue_put(player->event_queue, &event, 0);
                    }

                    if(event.input.key == InputKeyRight) {
                        player->seeking = true;
                        int32_t seek = CLAMP(
                            (int32_t)stream_tell(player->stream) +
                                player->frames_per_turn * player->frame_size,
                            (int32_t)player->num_frames * player->frame_size + player->header_size,
                            player->header_size);
                        stream_seek(player->stream, seek, StreamOffsetFromStart);

                        player->progress = (uint8_t)((int64_t)stream_tell(player->stream) * (int64_t)126 / ((int64_t)player->num_frames * (int64_t)player->frame_size + (int64_t)player->header_size));

                        if(event.input.type == InputTypeRelease) {
                            player->seeking = false;
                        }

                        static VideoPlayerEvent event = {.type = EventTypeJustRedraw};
                        furi_message_queue_put(player->event_queue, &event, 0);
                    }

                    if(player->playing) {
                        player_start();
                    }

                    else {
                        player_stop();
                    }
                }

                if(event.type == EventType1stHalf) {
                    uint8_t* audio_buffer = player->audio_buffer;

                    stream_read(player->stream, player->image_buffer, player->image_buffer_length);

                    if(player->silent) {
                        stream_read(
                            player->stream, player->fake_audio_buffer, player->audio_chunk_size);
                    }

                    else {
                        stream_read(player->stream, audio_buffer, player->audio_chunk_size);
                    }

                    player->frames_played++;

                    draw_all(player);
                }

                if(event.type == EventType2ndHalf) {
                    uint8_t* audio_buffer = &player->audio_buffer[player->audio_chunk_size];

                    stream_read(player->stream, player->image_buffer, player->image_buffer_length);

                    if(player->silent) {
                        stream_read(
                            player->stream, player->fake_audio_buffer, player->audio_chunk_size);
                    }

                    else {
                        stream_read(player->stream, audio_buffer, player->audio_chunk_size);
                    }

                    player->frames_played++;

                    draw_all(player);
                }

                if(event.type == EventTypeJustRedraw) {
                    draw_all(player);
                }

                if(player->frames_played == player->num_frames) {
                    player->quit = true;
                }

                furi_thread_yield();
            }
        }
        deinit_player(player);
        player_deinit_hardware();
    }

    return 0;
}