#include "stm32c0xx_hal.h"

GPIO_TypeDef* ledPorts[] = {GPIOB, GPIOB, GPIOB, GPIOB, GPIOA};
uint16_t ledPins[] = {
    GPIO_PIN_3,
    GPIO_PIN_10,
    GPIO_PIN_4,
    GPIO_PIN_5,
    GPIO_PIN_15
};

int totalLEDs = 5;
int winLedPinIndex = 3; // PB5 (LED3 / D4)
int baseDelayTime = 100;
int delayTime = 100;
int fastDelayTime = 20;
int level = 1;
int maxLevel = 5;

volatile uint8_t holdLEDState = 0;
volatile int lastLEDIndex = -1;

enum GameState {
    IDLE,
    RUNNING,
    WIN,
    LOSE
};

GameState currentState = IDLE;

/* ---------------- Helper Functions ---------------- */
void setRGBColor(GPIO_PinState red, GPIO_PinState blue) {
    HAL_GPIO_WritePin(GPIOA, GPIO_PIN_7, red);
    HAL_GPIO_WritePin(GPIOA, GPIO_PIN_5, blue);
}

void holdLEDs() {
    holdLEDState = 1;
}

/* ---------------- Arduino Style ---------------- */
void setup() {
    HAL_Init();
    __HAL_RCC_GPIOA_CLK_ENABLE();
    __HAL_RCC_GPIOB_CLK_ENABLE();

    GPIO_InitTypeDef GPIO_InitStruct = {0};

    // LED pins
    GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_LOW;

    for (int i = 0; i < totalLEDs; i++) {
        GPIO_InitStruct.Pin = ledPins[i];
        HAL_GPIO_Init(ledPorts[i], &GPIO_InitStruct);
    }

    // RGB pins
    GPIO_InitStruct.Pin = GPIO_PIN_7 | GPIO_PIN_5;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    // Button pakai Arduino attachInterrupt
    pinMode(D2, INPUT_PULLUP);
    attachInterrupt(digitalPinToInterrupt(D2), holdLEDs, FALLING);

    setRGBColor(GPIO_PIN_SET, GPIO_PIN_SET);
}

void loop() {
    switch (currentState) {

    case IDLE:
        holdLEDState = 0;
        lastLEDIndex = -1;
        setRGBColor(GPIO_PIN_SET, GPIO_PIN_SET);
        currentState = RUNNING;
        break;

    case RUNNING:
        for (int i = 0; i < totalLEDs; i++) {
            if (holdLEDState) {
                lastLEDIndex = i;
                currentState = (i == winLedPinIndex) ? WIN : LOSE;
                break;
            }
            HAL_GPIO_WritePin(ledPorts[i], ledPins[i], GPIO_PIN_SET);
            HAL_Delay(delayTime);
            HAL_GPIO_WritePin(ledPorts[i], ledPins[i], GPIO_PIN_RESET);
        }
        break;

    case WIN:
        HAL_GPIO_WritePin(
            ledPorts[lastLEDIndex],
            ledPins[lastLEDIndex],
            GPIO_PIN_SET
        );
        setRGBColor(GPIO_PIN_SET, GPIO_PIN_RESET); // merah
        HAL_Delay(1000);

        if (level < maxLevel) {
            level++;
            if (delayTime > fastDelayTime)
                delayTime -= 20;
        }
        currentState = IDLE;
        break;

    case LOSE:
        HAL_GPIO_WritePin(
            ledPorts[lastLEDIndex],
            ledPins[lastLEDIndex],
            GPIO_PIN_SET
        );
        setRGBColor(GPIO_PIN_RESET, GPIO_PIN_SET); // biru
        HAL_Delay(1000);
        currentState = IDLE;
        break;
    }
}