#include <avr/interrupt.h>
#include <avr/sleep.h>
#include <dio.h>
#include <rfm.h>
#include <stdbool.h>
#include <string.h>

dio_t buttons[4] = {
    { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 5 },
    { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 6 },
    { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 7 },
    { .ddr = &DDRD, .port = &PORTD, .pin = &PIND, .addr = 6 },
};

typedef struct {
    bool button_press[4];
} status_flags_t;

status_flags_t status_flags = {
    .button_press = { false, false, false, false }
};

ISR(INT2_vect) {
    for (int i = 0; i < 4; i++) {
        status_flags.button_press[i] = dio_read(&buttons[i]);
    }
}

int main(void) {
    EIMSK = 1 << INT2;
    EICRA |= 1 << ISC21 | 1 << ISC20;

    dio_t int2 = { .ddr = &DDRD, .port = &PORTD, .pin = &PIND, .addr = 2 };
    dio_set_direction(&int2, LINE_IN);

    rfm_t radio = (rfm_t){
        .spi    = {
            .clock          = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 1 },
            .data_out       = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 2 },
            .data_in        = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 3 },
            .chip_select    = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 4 },
        },
        .irq    = { .ddr = &DDRE, .port = &PORTE, .pin = &PINE, .addr = 6 },
        .reset  = { .ddr = &DDRD, .port = &PORTD, .pin = &PIND, .addr = 4 },
    };

    rfm_error_e error;
    rfm_init(&radio, (uint8_t [4]){ 0xde, 0xca, 0xfb, 0xad }, 4, &error);

    set_sleep_mode(SLEEP_MODE_IDLE);
    while(1) {
        sei();
        sleep_mode();
        cli();

        char msg[60];
        if (status_flags.button_press[0]) {
            status_flags.button_press[0] = false;
            strcpy(msg, "normal");
        } else if (status_flags.button_press[1]) {
            status_flags.button_press[1] = false;
            strcpy(msg, "spooky");
        } else if (status_flags.button_press[2]) {
            status_flags.button_press[2] = false;
            strcpy(msg, "eerie");
        } else if (status_flags.button_press[3]) {
            status_flags.button_press[3] = false;
            strcpy(msg, "flash");
        }

        rfm_transmit(&radio, (uint8_t *)msg, strlen(msg) + 1);

        interrupt_flags_t flags = rfm_interrupts(&radio);
        while(!flags.packet_sent) {
            _delay_ms(1);
            flags = rfm_interrupts(&radio);
        }
        rfm_standby(&radio);
    }
}