#include "EasyButton.h"
#include "led.h"

#define BUTTON 4
#define LED_RED 7
#define LED_GREEN 6
#define LED_YELLOW 5

EasyButton button(BUTTON);

Led red(LED_RED);
Led green(LED_GREEN);
Led yellow(LED_YELLOW);

void onPressed() {
  if (red.getEnable()) {
    red.setEnable(false);
    green. setEnable(false);
    yellow. setEnable(false);
  } else {
    red. setEnable(true);
    green. setEnable(true);
    yellow.setEnable(true);
  }
}

void setup() {
  button.begin();
  button.onPressed(onPressed);
  red.setup();
  green.setup();
  yellow.setup();
}

void loop() {
  button.read();
  red.blink(1000);
  green.blink(500);
  yellow.blink(250);
}