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

#define SCREEN_WIDTH 128 // OLED display width, in pixels
#define SCREEN_HEIGHT 32 // OLED display height, in pixels

#define RESET_BUTTON_PIN 2
#define TOUCH_MODULE_PIN 3

uint64_t prevPressed = 0;
uint64_t prevBtnPressed = 0;
uint16_t number = 0;
bool displayReq = false;

// Declaration for an SSD1306 display connected to I2C (SDA, SCL pins)
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

void displayNum(uint16_t number){
  String numStr = (String) number;
  String expandNumStr = "";
  for(int i=0;i<4-numStr.length();i++){
    expandNumStr += "0";
  }
  expandNumStr += numStr;
  display.clearDisplay();
  display.setTextColor(WHITE);
  display.setCursor(30,0);
  display.setTextSize(3);
  display.println(expandNumStr);
  display.display();
}



void buttonISR(){
  if(millis() - prevBtnPressed > 500){
    Serial.println("Button pressed");
    number = 0;
    displayReq = true;
    prevBtnPressed = millis();
  }
}

void touchISR(){
  if(millis() - prevPressed > 500){
    Serial.println("Touch pressed");
    number++;
    displayReq = true;
    prevPressed = millis();
  }
}

void setup() {
  Serial.begin(115200);

  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { // Address 0x3D for 128x64
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }

  attachInterrupt(digitalPinToInterrupt(RESET_BUTTON_PIN), buttonISR, RISING); // Attach the interrupt for reset button
  attachInterrupt(digitalPinToInterrupt(TOUCH_MODULE_PIN), touchISR, RISING);  // Attach the interrupt for touch module

  displayNum(0);
  delay(1000);
}

void loop() {
  if(displayReq){
    displayNum(number);
    displayReq = false;
  }

  delay(10);
}