#include <avr/io.h>
#include <avr/interrupt.h>
#include <avr/sleep.h>
#include <util/delay.h>

#include "lights.h"

static uint16_t button_start[N_BUTTONS];
static unsigned char button_pressed[N_BUTTONS];
static unsigned char just_waked_up;

static unsigned char user_params[MAX_USER_PARAMS] = { 0, 0, 0 };
static unsigned char user_params_max[MAX_USER_PARAMS] = { 3, 2, 2 };

static unsigned char user_params_state = 0;
	/*
	 * Here 0 means "no setup currently in progress",
	 * 1 .. MAX_USER_PARAMS means "now short presses increase or decrease
	 *           the value of user_params[user_params_state-1]"
	 */

static uint16_t user_params_starttime;

static void do_sleep();

static void toggle_bright_mode()
{
	// TODO
}

static void set_panic_mode()
{
	// TODO
}

unsigned char get_user_param(unsigned char param)
{
	if (param < MAX_USER_PARAMS)
		return user_params[param];
	return 0; // FIXME: internal error?
}

static inline void short_press(unsigned char button)
{
	unsigned char param;

	if (user_params_state == 0) {
		if (button == 0)
			toggle_bright_mode();
		else
			set_panic_mode();
		return;
	}

	param = user_params_state-1;

	if (button == 0) {
		if (user_params[param])
			user_params[param]--;
		else
			user_params[param] = user_params_max[param];
	} else {
		user_params[param]++;

		if (user_params[param] >= user_params_max[param])
			user_params[param] = 0;
	}
	// FIXME: notify somebody about user_params change?

	// set_status_led(2, user_params[param]);

	user_params_starttime = jiffies;
}

static inline void long_press(unsigned char button)
{
	if (button == 0) {
		do_sleep();
		return;
	}

	// button 1 - cycle through states
	user_params_state++;

	if (user_params_state >= MAX_USER_PARAMS)
		user_params_state = 1;

	// set_status_led(1, state);
	user_params_starttime = jiffies;
}

void init_buttons()
{
	unsigned char i;

	DDRA &= ~(_BV(PA3) | _BV(PA4));
	PORTA |=  _BV(PA3) | _BV(PA4);

	for (i=0; i < N_BUTTONS; i++) {
		button_start[i] = 0;
		button_pressed[i] = 0;
	}

	just_waked_up = 1;
}

static void do_sleep()
{
	log_byte(0xb0);
	log_flush();

	hw_suspend();
	gpio_before_poweroff(); // Set the status LED on again

	while((PINA & _BV(PA3)) == 0)
		; // wait for button release

	_delay_ms(100);

	susp_gpio(); // disable the status LED

	power_down();
}

void timer_check_buttons()
{
	unsigned char pin = PINA;
	unsigned char i;
	unsigned char port_states[N_BUTTONS] = {
		pin & _BV(PA3),
		pin & _BV(PA4),
	};

	for (i = 0; i < N_BUTTONS; i++) {
		if (!port_states[i]) { // is pressed
			if (i == 0 && just_waked_up)
				continue;
			if (button_pressed[i] == 0) {
				// begin of button press
				button_pressed[i] = 1;
				button_start[i] = jiffies;
#if 0
				log_byte(0xC0);
				log_word(jiffies);
#endif
			} else if (button_pressed[i] == 1) {
				// been already pressed
				uint16_t duration = jiffies - button_start[i];
				if (duration > 80) {
					// long button press
					button_pressed[i] = 2;
					long_press(i);
				}
			}
		} else { // is not pressed
			if (i == 0)
				just_waked_up = 0;
			if (button_pressed[i]) { // just depressed
				uint16_t duration = jiffies - button_start[i];
#if 1
				log_byte(0xC1);
				log_word(duration);
				log_flush();
#endif
				button_pressed[i] = 0;
				if (duration > 6 && duration < 30)
					short_press(i);
			}
		}
	}
}