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

#define tombolStart 13
#define tombolPause 12
#define tombolReset 14

#define relayLampu 15

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
Adafruit_SSD1306 oled(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

int defaultJam = 0;
int defaultMenit = 2;
int detik = 12;

int defaultJamPause = 0;
int defaultMenitPause = 1;

int jam;
int menit;

boolean running = false;
boolean paused = false;

boolean lampu = true;

void setup() {
  Serial.begin(115200);
  if (!oled.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("failed to start SSD1306 OLED"));
    while (1);
  }

  pinMode(tombolStart, INPUT_PULLUP);
  pinMode(tombolPause, INPUT_PULLUP);
  pinMode(tombolReset, INPUT_PULLUP);

  pinMode(relayLampu, OUTPUT);
  digitalWrite(relayLampu, LOW);

  delay(2000);
  oled.clearDisplay();
  oled.setTextSize(3);
  oled.setTextColor(WHITE);
  oled.setCursor(2, 10);
  oled.println("Loading");
  oled.println(".......");
  oled.display();

  delay(1000);

  oled.clearDisplay();
  oled.setTextSize(2);
  oled.setTextColor(WHITE);
  oled.setCursor(8, 24);
  oled.println("Klik Mulai");
  oled.display();
}

void loop() {

  int runningState = digitalRead(tombolStart);

  if (runningState == LOW && running == false) {
    running = true;
    jam = defaultJam;
    menit = defaultMenit;
    oled.clearDisplay();
    delay(500);
    oled.setTextSize(2);
    oled.setTextColor(WHITE);
    oled.setCursor(16, 20);
    oled.println("........");
    oled.display();
    delay(2000);

    digitalWrite(relayLampu, HIGH);
  }

  if (running == true && paused == false) {
    if (jam > 0) {
      jam = jam - 1;
    }
    if (menit > 0) {
      menit = menit - 1;
    }
    for(int i = detik-1; i >= 0; i--) {
      int resetState = digitalRead(tombolReset);
      if ( resetState == LOW ) {
        jam = defaultJam;
        menit = defaultMenit;

        oled.clearDisplay();
        delay(100);
        oled.setTextSize(2);
        oled.setTextColor(WHITE);
        oled.setCursor(16, 24);
        oled.println("Reset...");
        oled.display();

        return;
      } 

      oled.clearDisplay();
      oled.setTextColor(WHITE);

      oled.setTextSize(2);
      oled.setCursor(16, 4);
      oled.print("Lampu ON");

      oled.setTextSize(1);
      oled.setCursor(3, 30);
      oled.print("Lampu Akan Mati Pada");

      oled.setTextSize(2);
      oled.setCursor(16, 44);

      if(jam < 10) {
        // Serial.print("0");
        oled.print("0");
      }
      // Serial.print(jam);
      oled.print(jam);
      if(menit < 10) {
        // Serial.print(":0");
        oled.print(":0");
      } else {
        // Serial.print(":");
        oled.print(":");
      }
      // Serial.print(menit);
      oled.print(menit);
      if(i < 10) {
        // Serial.print(":0");
        oled.print(":0");
      } else {
        // Serial.print(":");
        oled.print(":");
      }
      // Serial.println(i);
      oled.println(i);

      oled.display();

      delay(1000);
    }

    if (jam == 0 && menit == 0) {
      if (running == true) {
        running = false;
        return;
      }
    }

  } else if(running == false) {
    digitalWrite(relayLampu, HIGH);

    jam = defaultJamPause;
    menit = defaultMenitPause;

    if (jam > 0) {
      jam = jam - 1;
    }
    if (menit > 0) {
      menit = menit - 1;
    }

    for(int j = detik-1; j >= 0; j--) {
      int resetState = digitalRead(tombolReset);
      if ( resetState == LOW ) {
        running = true;
        jam = defaultJam;
        menit = defaultMenit;

        oled.clearDisplay();
        delay(100);
        oled.setTextSize(2);
        oled.setTextColor(WHITE);
        oled.setCursor(16, 24);
        oled.println("Reset...");
        oled.display();

        return;
      } 

      oled.clearDisplay();
      oled.setTextColor(WHITE);

      oled.setTextSize(2);
      oled.setCursor(10, 4);
      oled.print("Lampu OFF");

      oled.setTextSize(1);
      oled.setCursor(0, 30);
      oled.print("Lampu Akan Nyala Pada");

      oled.setTextSize(2);
      oled.setCursor(16, 44);

      if(jam < 10) {
        // Serial.print("0");
        oled.print("0");
      }
      // Serial.print(jam);
      oled.print(jam);
      if(menit < 10) {
        // Serial.print(":0");
        oled.print(":0");
      } else {
        // Serial.print(":");
        oled.print(":");
      }
      // Serial.print(menit);
      oled.print(menit);
      if(j < 10) {
        // Serial.print(":0");
        oled.print(":0");
      } else {
        // Serial.print(":");
        oled.print(":");
      }
      // Serial.println(i);
      oled.println(j);

      oled.display();
      delay(1000);
    }

    if (jam == 0 && menit == 0) {
      if (running == false) {
        running = true;
        digitalWrite(relayLampu, HIGH);
        return;
      }
    }
  }
}