#include <gui/gui.h>
#include <stdlib.h>
#include <dolphin/dolphin.h>
#include <dialogs/dialogs.h>
#include <math.h>
#include "util.h"
#include "defines.h"
#include "card.h"
#include "util.h"
#include "ui.h"

#include "blackjack_icons.h"

#define DEALER_MAX 17

void start_round(GameState *game_state);

void init(GameState *game_state);

static void draw_ui(Canvas *const canvas, const GameState *game_state) {

    draw_money(canvas, game_state->player_score);

    draw_score(canvas, true, handCount(game_state->player_cards, game_state->player_card_count));

    if (!game_state->animating && game_state->state == GameStatePlay) {
        draw_play_menu(canvas, game_state);
    }
}

static void render_callback(Canvas *const canvas, void *ctx) {
    const GameState *game_state = acquire_mutex((ValueMutex *) ctx, 25);

    if (game_state == NULL) {
        return;
    }

    canvas_set_color(canvas, ColorBlack);
    canvas_draw_frame(canvas, 0, 0, 128, 64);

    if (game_state->state == GameStateStart) {
        canvas_draw_icon(canvas, 0, 0, &I_blackjack);
    }
    if (game_state->state == GameStateGameOver) {
        canvas_draw_icon(canvas, 0, 0, &I_endscreen);
    }

    if (game_state->state == GameStatePlay || game_state->state == GameStateDealer) {
        if (game_state->state == GameStatePlay)
            draw_player_scene(canvas, game_state);
        else
            draw_dealer_scene(canvas, game_state);
        animateQueue(game_state, canvas);
        draw_ui(canvas, game_state);
    } else if (game_state->state == GameStateSettings) {
        settings_page(canvas, game_state);
    }

    release_mutex((ValueMutex *) ctx, game_state);
}

//region card draw
Card draw_card(GameState *game_state) {
    Card c = game_state->deck.cards[game_state->deck.index];
    game_state->deck.index++;
    return c;
}


void drawPlayerCard(GameState *game_state) {
    Card c = draw_card(game_state);
    game_state->player_cards[game_state->player_card_count] = c;
    game_state->player_card_count++;
    if (game_state->selectedMenu == 0 &&
        (game_state->player_score < game_state->settings.round_price || game_state->doubled))
        game_state->selectedMenu = 1;
}

void drawDealerCard(GameState *game_state) {
    Card c = draw_card(game_state);
    game_state->dealer_cards[game_state->dealer_card_count] = c;
    game_state->dealer_card_count++;
}
//endregion

//region queue callbacks
void to_lose_state(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    UNUSED(game_state);
    UNUSED(margin);
    if (duration == 0)
        return;
    popup_frame(canvas);
    elements_multiline_text_aligned(canvas, 64, 22, AlignCenter, AlignCenter, "You lost");
}

void to_bust_state(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    UNUSED(game_state);
    UNUSED(margin);
    if (duration == 0)
        return;
    popup_frame(canvas);
    elements_multiline_text_aligned(canvas, 64, 22, AlignCenter, AlignCenter, "Busted!");
}

void to_draw_state(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    UNUSED(game_state);
    UNUSED(margin);

    if (duration == 0)
        return;
    popup_frame(canvas);
    elements_multiline_text_aligned(canvas, 64, 22, AlignCenter, AlignCenter, "Draw");
}

void to_dealer_turn(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    UNUSED(game_state);
    UNUSED(margin);

    if (duration == 0)
        return;
    popup_frame(canvas);
    elements_multiline_text_aligned(canvas, 64, 22, AlignCenter, AlignCenter, "Dealers turn");
}

void to_win_state(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    UNUSED(game_state);
    UNUSED(margin);

    if (duration == 0)
        return;
    popup_frame(canvas);
    elements_multiline_text_aligned(canvas, 64, 22, AlignCenter, AlignCenter, "You win");
}

void to_start(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    UNUSED(game_state);
    UNUSED(margin);
    if (duration == 0)
        return;
    popup_frame(canvas);
    elements_multiline_text_aligned(canvas, 64, 22, AlignCenter, AlignCenter, "Round started");
}

void before_start(GameState *gameState) {
    gameState->dealer_card_count = 0;
    gameState->player_card_count = 0;
}


void start(GameState *game_state) {
    start_round(game_state);
}

void draw(GameState *game_state) {
    game_state->player_score += game_state->bet;
    game_state->bet = 0;
    queue(game_state, start, before_start, to_start, game_state->settings.message_duration, 0);
}

void game_over(GameState *game_state) {
    game_state->state = GameStateGameOver;
}

void lose(GameState *game_state) {
    game_state->state = GameStatePlay;
    game_state->bet = 0;
    if (game_state->player_score >= game_state->settings.round_price)
        queue(game_state, start, before_start, to_start, game_state->settings.message_duration, 0);
    else
        queue(game_state, game_over, NULL, NULL, 0, 0);
}

void win(GameState *game_state) {
    game_state->state = GameStatePlay;
    game_state->player_score += game_state->bet * 2;
    game_state->bet = 0;
    queue(game_state, start, before_start, to_start, game_state->settings.message_duration, 0);
}


void dealerTurn(GameState *game_state) {
    game_state->state = GameStateDealer;
}

void dealer_card_animation(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    float t = (float) (furi_get_tick() - game_state->animationStart) / (duration - margin);

    Card animatingCard = game_state->deck.cards[game_state->deck.index];
    if (game_state->dealer_card_count > 1) {
        Vector end = card_pos_at_index(game_state->dealer_card_count);
        if (!is_at_edge(game_state->dealer_card_count))
            end.x -= CARD_HALF_WIDTH;
        draw_card_animation(animatingCard,
                            (Vector) {0, 64},
                            (Vector) {0, 32},
                            end,
                            t,
                            true,
                            canvas);
    } else {
        draw_card_animation(animatingCard,
                            (Vector) {32, -CARD_HEIGHT},
                            (Vector) {64, 32},
                            (Vector) {2, 2},
                            t,
                            false,
                            canvas);
//        drawPlayerDeck(game_state->dealer_cards, game_state->dealer_card_count, canvas);
    }
}

void dealer_back_card_animation(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    float t = (float) (furi_get_tick() - game_state->animationStart) / (duration - margin);
    Vector currentPos = quadratic_2d((Vector) {32, -CARD_HEIGHT}, (Vector) {64, 32}, (Vector) {13, 5}, t);
    drawCardBackAt(currentPos.x, currentPos.y, canvas);
}

void player_card_animation(const GameState *game_state, Canvas *const canvas, uint32_t duration, uint32_t margin) {
    float t = (float) (furi_get_tick() - game_state->animationStart) / (duration - margin);
    Card animatingCard = game_state->deck.cards[game_state->deck.index];
    Vector end = card_pos_at_index(game_state->player_card_count);
    if (!is_at_edge(game_state->player_card_count))
        end.x -= CARD_HALF_WIDTH;
    draw_card_animation(animatingCard,
                        (Vector) {32, -CARD_HEIGHT},
                        (Vector) {0, 32},
                        end,
                        t,
                        true,
                        canvas);
//    drawPlayerDeck(game_state->dealer_cards, game_state->player_card_count, canvas);
}
//endregion

void player_tick(GameState *game_state) {
    uint8_t score = handCount(game_state->player_cards, game_state->player_card_count);
    if ((game_state->doubled && score <= 21) || score == 21) {
        queue(game_state, dealerTurn, NULL, NULL, game_state->settings.message_duration, 0);
    } else if (score > 21) {
        queue(game_state, lose, NULL, to_bust_state, game_state->settings.message_duration, 0);
    } else {
        if (game_state->selectDirection == DirectionUp && game_state->selectedMenu > 0 &&
            game_state->player_score >= game_state->settings.round_price && !game_state->doubled) {
            game_state->selectedMenu--;
        }
        if (game_state->selectDirection == DirectionDown && game_state->selectedMenu < 2) {
            game_state->selectedMenu++;
        }
        if (game_state->selectDirection == Select) {
            //double
            if (!game_state->doubled && game_state->selectedMenu == 0 &&
                game_state->player_score >= game_state->settings.round_price) {

                game_state->player_score -= game_state->settings.round_price;
                game_state->bet += game_state->settings.round_price;
                game_state->doubled = true;
                game_state->selectedMenu = 1;
                queue(game_state, drawPlayerCard, NULL, player_card_animation, game_state->settings.animation_duration,
                      game_state->settings.animation_margin);
                game_state->player_cards[game_state->player_card_count] = game_state->deck.cards[game_state->deck.index];
                score = handCount(game_state->player_cards, game_state->player_card_count + 1);
                if (score > 21)
                    queue(game_state, lose, NULL, to_bust_state, game_state->settings.message_duration, 0);
                else
                    queue(game_state, dealerTurn, NULL, to_dealer_turn, game_state->settings.message_duration, 0);

            } //hit
            else if (game_state->selectedMenu == 1) {
                queue(game_state, drawPlayerCard, NULL, player_card_animation, game_state->settings.animation_duration,
                      game_state->settings.animation_margin);
            } //stay
            else if (game_state->selectedMenu == 2) {
                queue(game_state, dealerTurn, NULL, to_dealer_turn, game_state->settings.message_duration, 0);
            }
        }
    }
}

void dealer_tick(GameState *game_state) {
    uint8_t dealer_score = handCount(game_state->dealer_cards, game_state->dealer_card_count);
    uint8_t player_score = handCount(game_state->player_cards, game_state->player_card_count);

    if (dealer_score >= DEALER_MAX) {
        if (dealer_score > 21 || dealer_score < player_score)
            queue(game_state, win, NULL, to_win_state, game_state->settings.message_duration, 0);
        else if (dealer_score > player_score)
            queue(game_state, lose, NULL, to_lose_state, game_state->settings.message_duration, 0);
        else if (dealer_score == player_score)
            queue(game_state, draw, NULL, to_draw_state, game_state->settings.message_duration, 0);
    } else {
        queue(game_state, drawDealerCard, NULL, dealer_card_animation, game_state->settings.animation_duration,
              game_state->settings.animation_margin);
    }
}

void settings_tick(GameState *game_state) {
    if (game_state->selectDirection == DirectionDown && game_state->selectedMenu < 4) {
        game_state->selectedMenu++;
    }
    if (game_state->selectDirection == DirectionUp && game_state->selectedMenu > 0) {
        game_state->selectedMenu--;
    }

    if (game_state->selectDirection == DirectionLeft || game_state->selectDirection == DirectionRight) {
        int nextScore = 0;
        switch (game_state->selectedMenu) {
            case 0:
                nextScore = game_state->settings.starting_money;
                if (game_state->selectDirection == DirectionLeft)
                    nextScore -= 10;
                else
                    nextScore += 10;
                if (nextScore >= (int) game_state->settings.round_price && nextScore < 400)
                    game_state->settings.starting_money = nextScore;
                break;
            case 1:
                nextScore = game_state->settings.round_price;
                if (game_state->selectDirection == DirectionLeft)
                    nextScore -= 10;
                else
                    nextScore += 10;
                if (nextScore >= 5 && nextScore <= (int) game_state->settings.starting_money)
                    game_state->settings.round_price = nextScore;
                break;
            case 2:
                nextScore = game_state->settings.animation_margin;
                if (game_state->selectDirection == DirectionLeft)
                    nextScore -= 100;
                else
                    nextScore += 100;
                if (nextScore >= 0 && nextScore <= (int) game_state->settings.animation_duration)
                    game_state->settings.animation_margin = nextScore;
                break;
            case 3:
                nextScore = game_state->settings.animation_duration;
                if (game_state->selectDirection == DirectionLeft)
                    nextScore -= 100;
                else
                    nextScore += 100;
                if (nextScore >= (int) game_state->settings.animation_margin && nextScore < 2000)
                    game_state->settings.animation_duration = nextScore;
                break;
            case 4:
                nextScore = game_state->settings.message_duration;
                if (game_state->selectDirection == DirectionLeft)
                    nextScore -= 100;
                else
                    nextScore += 100;
                if (nextScore >= 0 && nextScore < 2000)
                    game_state->settings.message_duration = nextScore;
                break;
            case 5:
                game_state->settings.sound_effects = !game_state->settings.sound_effects;
            default:
                break;
        }
    }

}

void tick(GameState *game_state) {
    game_state->last_tick = furi_get_tick();
    bool queue_ran = run_queue(game_state);

    switch (game_state->state) {
        case GameStateGameOver:
        case GameStateStart:
            if (game_state->selectDirection == Select)
                init(game_state);
            else if (game_state->selectDirection == DirectionRight) {
                game_state->selectedMenu = 0;
                game_state->state = GameStateSettings;
            }
            break;
        case GameStatePlay:
            if (!game_state->started) {
                game_state->selectedMenu = 0;
                game_state->started = true;
                queue(game_state, drawDealerCard, NULL, dealer_back_card_animation,
                      game_state->settings.animation_duration, game_state->settings.animation_margin);
                queue(game_state, drawPlayerCard, NULL, player_card_animation, game_state->settings.animation_duration,
                      game_state->settings.animation_margin);
                queue(game_state, drawDealerCard, NULL, dealer_card_animation, game_state->settings.animation_duration,
                      game_state->settings.animation_margin);
                queue(game_state, drawPlayerCard, NULL, player_card_animation, game_state->settings.animation_duration,
                      game_state->settings.animation_margin);
            }
            if (!queue_ran)
                player_tick(game_state);
            break;
        case GameStateDealer:
            if (!queue_ran)
                dealer_tick(game_state);
            break;
        case GameStateSettings:
            settings_tick(game_state);
            break;
        default:
            break;
    }

    game_state->selectDirection = None;

}

void start_round(GameState *game_state) {
    game_state->player_card_count = 0;
    game_state->dealer_card_count = 0;
    game_state->selectedMenu = 0;
    game_state->started = false;
    game_state->doubled = false;
    game_state->animating = true;
    game_state->animationStart = 0;
    shuffleDeck(&(game_state->deck));
    game_state->doubled = false;
    game_state->bet = game_state->settings.round_price;
    if (game_state->player_score < game_state->settings.round_price) {
        game_state->state = GameStateGameOver;
    } else {
        game_state->player_score -= game_state->settings.round_price;
    }
    game_state->state = GameStatePlay;
}

void init(GameState *game_state) {
    game_state->settings = load_settings();
    game_state->last_tick = 0;
    game_state->processing = true;
    game_state->player_score = game_state->settings.starting_money;
    generateDeck(&(game_state->deck));
    start_round(game_state);
}

static void input_callback(InputEvent *input_event, FuriMessageQueue *event_queue) {
    furi_assert(event_queue);
    AppEvent event = {.type = EventTypeKey, .input = *input_event};
    furi_message_queue_put(event_queue, &event, FuriWaitForever);
}

static void update_timer_callback(FuriMessageQueue *event_queue) {
    furi_assert(event_queue);
    AppEvent event = {.type = EventTypeTick};
    furi_message_queue_put(event_queue, &event, 0);
}

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

    int32_t return_code = 0;

    FuriMessageQueue *event_queue = furi_message_queue_alloc(8, sizeof(AppEvent));

    GameState *game_state = malloc(sizeof(GameState));
    init(game_state);
    game_state->state = GameStateStart;

    ValueMutex state_mutex;
    if (!init_mutex(&state_mutex, game_state, sizeof(GameState))) {
        FURI_LOG_E(APP_NAME, "cannot create mutex\r\n");
        return_code = 255;
        goto free_and_exit;
    }

    ViewPort *view_port = view_port_alloc();
    view_port_draw_callback_set(view_port, render_callback, &state_mutex);
    view_port_input_callback_set(view_port, input_callback, event_queue);

    FuriTimer *timer =
            furi_timer_alloc(update_timer_callback, FuriTimerTypePeriodic, event_queue);
    furi_timer_start(timer, furi_kernel_get_tick_frequency() / 25);

    Gui *gui = furi_record_open("gui");
    gui_add_view_port(gui, view_port, GuiLayerFullscreen);

    AppEvent event;

    for (bool processing = true; processing;) {
        FuriStatus event_status = furi_message_queue_get(event_queue, &event, 100);
        GameState *game_state = (GameState *) acquire_mutex_block(&state_mutex);
        if (event_status == FuriStatusOk) {
            if (event.type == EventTypeKey) {

                if (event.input.type == InputTypePress) {
                    switch (event.input.key) {
                        case InputKeyUp:
                            game_state->selectDirection = DirectionUp;
                            break;
                        case InputKeyDown:
                            game_state->selectDirection = DirectionDown;
                            break;
                        case InputKeyRight:
                            game_state->selectDirection = DirectionRight;
                            break;
                        case InputKeyLeft:
                            game_state->selectDirection = DirectionLeft;
                            break;
                        case InputKeyBack:
                            if (game_state->state == GameStateSettings) {
                                game_state->state = GameStateStart;
                                save_settings(game_state->settings);
                            } else
                                processing = false;
                            break;
                        case InputKeyOk:
                            game_state->selectDirection = Select;
                            break;
                    }
                }
            } else if (event.type == EventTypeTick) {
                tick(game_state);
                processing = game_state->processing;
            }
        } else {
            FURI_LOG_D(APP_NAME, "osMessageQueue: event timeout");
            // event timeout
        }
        view_port_update(view_port);
        release_mutex(&state_mutex, game_state);
    }


    furi_timer_free(timer);
    view_port_enabled_set(view_port, false);
    gui_remove_view_port(gui, view_port);
    furi_record_close(RECORD_GUI);
    view_port_free(view_port);
    delete_mutex(&state_mutex);

    free_and_exit:
    queue_clear();
    free(game_state);
    furi_message_queue_free(event_queue);

    return return_code;
}