#include <RH_ASK.h>
#include <SPI.h>

#define btn1  4
#define btn2  5
#define rLED1 8
#define rLED2 9
#define buzz  6

RH_ASK rf_driver;
const char *msg1 = "1";
const char *msg2 = "2";
int btn1_state = 0;
int btn2_state = 0;
int safty_flag = 0;

void setup() {
  Serial.begin(9600);
  rf_driver.init();
  pinMode(btn1, INPUT);
  pinMode(btn2, INPUT);
  pinMode(rLED1, OUTPUT);
  pinMode(rLED2, OUTPUT);
  pinMode(buzz, OUTPUT);
  digitalWrite(rLED1, LOW);
  digitalWrite(rLED2, LOW);
  digitalWrite(buzz, LOW);
}

void loop() {
  read_btn1_and_btn2();
  if (btn1_state) {
    send_msg_1();
    safty_flag = 1;
  }
  if (btn2_state) {
    send_msg_2();
    safty_flag = 1;
  }
  if (safty_flag) {
    safty();
  }
}

void read_btn1_and_btn2() {
  btn1_state = digitalRead(btn1);
  btn2_state = digitalRead(btn2);
  Serial.print(btn1_state);
  Serial.print("  |  ");
  Serial.println(btn2_state);
  Serial.println("---------------------");
}
void send_msg_1() {
  Serial.println("Sending \"1\"");
  rf_driver.send((uint8_t *)msg1, strlen(msg1));
  rf_driver.waitPacketSent();
  delay(1000);
}
void send_msg_2() {
  Serial.println("Sending \"2\"");
  rf_driver.send((uint8_t *)msg2, strlen(msg2));
  rf_driver.waitPacketSent();
  delay(1000);
}

void safty() {
  digitalWrite(rLED1, LOW);
  digitalWrite(rLED2, HIGH);
  digitalWrite(buzz, HIGH);
  delay(100);
  digitalWrite(rLED2, LOW);
  digitalWrite(rLED1, HIGH);
  digitalWrite(buzz, LOW);
  delay(100);
}