#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <EEPROM.h>

const int SCREEN_WIDTH = 128; 
const int SCREEN_HEIGHT = 64;

int counter = 0;
int buttonUp = 8;
int buttonDown = 9;
int buttonReset = 10;

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

void setup() {
  Serial.begin(9600);
  counter = EEPROM.read(0);

  pinMode(buttonUp, INPUT_PULLUP);
  pinMode(buttonDown, INPUT_PULLUP);
  pinMode(buttonReset, INPUT_PULLUP);

  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    while(true);
  }
}

void loop() {
  if (digitalRead(buttonUp) == LOW) {
    incrementCounter();
    delay(200);
  }

  if (digitalRead(buttonDown) == LOW) {
    decrementCounter();
    delay(200);
  }

  if (digitalRead(buttonReset) == LOW) {
    resetCounter();
    delay(200);
  }
  display.clearDisplay();

  display.setTextSize(1);
  display.setTextColor(WHITE);
  display.setCursor(40, 10);
  display.println("Sanagich");
  display.setCursor(50, 30);
  display.setTextSize(2);
  display.println(counter);
  display.display();
}

void incrementCounter() {
  counter++;
  updateLCD();
}

void decrementCounter() {
  counter--;
  updateLCD();
}

void resetCounter() {
  counter = 0;
  updateLCD();
}

void updateLCD() {
  display.setTextSize(2);
  display.setTextColor(WHITE);
  display.setCursor(50, 30);
  display.println(counter);
  EEPROM.write(0, counter);
}