#include <TinyWireM.h>
#include <Tiny4kOLED.h>
#include "font16x32digits.h" 

uint8_t width = 128;
uint8_t height = 64;

void setup() {
  pinMode(4, INPUT_PULLUP);
  pinMode(1, INPUT_PULLUP);
  pinMode(3, OUTPUT);

  oled.begin(width, height, sizeof(tiny4koled_init_128x64br), tiny4koled_init_128x64br);
  oled.clear();
  oled.on();
  delay(100);
}

int minutes = 2;
int seconds = 0;
int fase = 0;
int chosen = 0;
int times[] = {1, 2, 5, 15};
int ani = 9;
int debounce4 = 0;
int debounce1 = 0;

void loop() {
  if (fase == 0 || fase == 3) {
    if (digitalRead(4) == 0) {
      if (debounce4 == 0) {
        debounce4 = 1;
        chosen++;
        if (chosen > 3)
          chosen = 0;
        if (fase == 3)
          fase = 0;
      }
    } else {
      debounce4 = 0;
    }

    if (digitalRead(1) == 0) {
      if (debounce1 == 0) {
        debounce1 = 1;
        minutes = times[chosen];
        fase = 1;
        if (fase == 3)
          fase = 0;
      }
    } else {
      debounce1 = 0;
    }
  }

  if (fase == 1 || fase == 2) {
    delay(1000);
    seconds--;
    if (seconds <= 0) {
      seconds = 59;
      minutes--;
    }
  }

  if (fase == 3) {
    digitalWrite(3, 1);
    delay(500);
    digitalWrite(3, 0);
    delay(500);
  }

  if (minutes == 0)
    fase = 2;
  if (minutes == -1)
    fase = 3;

  updateDisplay();
}

void updateDisplay() {
  oled.clear();

  if (fase == 0) {
    oled.setFontX2(FONT6X8P);
    oled.setCursor(0, 0);
    oled.print(" SET TIME:");
    oled.setCursor(0, 1);
    oled.print("----------");
    oled.setCursor(0, 2);
    for (int i = 0; i < 4; i++) {
      oled.print(" ");
      oled.print(times[i]);
    }
    if (chosen > 1)
      oled.setCursor(30 + chosen * 25, 4);
    else
      oled.setCursor(10 + chosen * 15, 4);
    oled.print("^");
  }

  if (fase == 1) {
    oled.setFontX2(FONT6X8P);
    oled.setCursor(8, 0);
    oled.print("TIME:");
    oled.setFontX2(FONT8X16P);

    oled.setCursor(8, 2);
    if (minutes < 10)
      oled.print(0);
    oled.print(minutes);
    oled.print(" : ");
    if (seconds < 10)
      oled.print(0);
    oled.print(seconds);
    oled.setFontX2(FONT6X8P);
    oled.setCursor(8, 6);
    for (int i = ani; i > 0; i--) {
      oled.print("o");
    }
    ani--;
    if (ani == 0)
      ani = 10;
  }

  if (fase == 2) {
    oled.setFont(FONT16X32DIGITS);
    oled.setCursor(32, 2);
    oled.print(seconds);
  }

  oled.on(); // Aktualizace displeje po vykreslení
}