#include <TM1637.h>
#include <LiquidCrystal_I2C.h>
#include <Keypad.h>

enum STAGE{
  INIT,
  SETTIME,
  READY,
  COUNTDOWN,
  FAILED,
  COMPLETED,
};
STAGE stage = INIT;
int CLK = 12;
int DIO = 11;
int buzzer = 9;
int relay = 10;
int redled = A1;
int yellowled = A2;
int greenled = A3;
int Minutes = 00;
int Seconds = 00;
int trycount = 0;
int keycount = 0;
int i = 0;
uint16_t pre_timestamp;

const byte ROWS = 4;
const byte COLS = 3;
char keys[ROWS][COLS] = {
  {'1', '2', '3'},
  {'4', '5', '6'},
  {'7', '8', '9'},
  {'*', '0', '#'}
};
byte rowPins[ROWS] = {7, 2, 3, 5};  //7,2,3,5 for Black 4x3 keypad
byte colPins[COLS] = {6, 8, 4};     //6,8,4 for Black 4x3 Keypad


char password[7] = "7355608";
char entered[7] = "";
char strTime[4] = "0500";
char tpos = 0;

Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);
LiquidCrystal_I2C lcd(0x27, 16, 2);
TM1637 tm(CLK, DIO);


void setup() {
  tm.init();
  tm.set(2);
  pinMode(buzzer, OUTPUT);
  pinMode(relay, OUTPUT);
  pinMode(redled, OUTPUT);
  pinMode(yellowled, OUTPUT);
  pinMode(greenled, OUTPUT);
  lcd.init();
  lcd.backlight();
  digitalWrite(redled,HIGH);
  digitalWrite(yellowled,HIGH);
  digitalWrite(greenled,HIGH); 
  digitalWrite(buzzer,LOW);
  delay(200);
  digitalWrite(redled,LOW);
  digitalWrite(yellowled,LOW);
  digitalWrite(greenled,LOW); 
  digitalWrite(buzzer,HIGH);
  updateTM();
  lcd.clear();
  lcd.print("Enter code:");
  lcd.setCursor(keycount, 1);
}

void loop() {
	uint16_t timestamp = millis();
  
  char armcode = keypad.getKey();

  if (armcode != NO_KEY)
  {
      switch(stage){

        case INIT:
          switch(armcode){
            case '*':
            case '#':
              stage = SETTIME;
              tpos = 0;
              lcd.clear();
              lcd.print("Set timer:");
            break;
            default:
              entered[keycount] = armcode;
              keycount++;
              lcd.setCursor(keycount+4, 1);
              if(keycount==6){
                if (entered[0] == password[0] && entered[1] == password[1] && entered[2] == password[2] && entered[3] == password[3]){
                    stage = READY;
                    lcd.clear();
                    lcd.print("press # to activate.");
                }else{
                  trycount++;
                  keycount = 0;
                }
              }
            break;
          }
          break;

        case READY:
          if(armcode == '#'){
              stage = COUNTDOWN;
              lcd.clear();
              lcd.print("Activated.");
          }
        break;

        case SETTIME:
          switch(armcode){
            case '*':
            case '#':
              stage = INIT;
              keycount = 0;
              lcd.clear();
              lcd.print("Enter code:");
            break;
            default:
              strTime[tpos] = armcode;
              updateTM();
              tpos++;
              if(tpos>3)tpos = 0;
            break;
          }
          break;



        case COUNTDOWN:
          
          break;
        case FAILED:
          
          break;
        case COMPLETED:
          
          break;
      }

  }


  pre_timestamp = timestamp;

}



void updateTM(){
  tm.display(0, strTime[0]-0x30);
  tm.display(1, strTime[1]-0x30);
  tm.point(1);
  tm.display(2, strTime[2]-0x30);
  tm.display(3, strTime[3]-0x30);
}