#include "ate0003.h"

static i2c_master_bus_handle_t _i2c_bus_handle = NULL;
static httpd_handle_t _webserver_handle = NULL;

static zh_pcf8574_handle_t _button_handle = {0};
static zh_pcf8574_handle_t _led_1_handle = {0};
static zh_pcf8574_handle_t _led_2_handle = {0};
static zh_pcf8574_handle_t _relay_handle = {0};
static zh_pcf8574_handle_t lcd_handle = {0};

static zh_encoder_handle_t _power_encoder_handle = {0};
static zh_encoder_handle_t _component_encoder_handle = {0};

volatile static bool _is_work = false;
volatile static bool _is_dmm_enabled = false;
volatile static bool _is_fix_enabled = false;
volatile static uint8_t _permitted_channels = 0;
volatile static uint8_t _used_channels = 0;
volatile static bool _is_initialized = false;

volatile static bool _is_num_1_fixed = false;
volatile static bool _is_num_2_fixed = false;
volatile static bool _is_num_3_fixed = false;
volatile static bool _is_num_4_fixed = false;

static void _zh_wifi_softap_init(void);
static void _zh_webserver_init(void);
static void _zh_encoder_init(void);
static void _zh_ac_dimmer_init(void);
static void _zh_pcf8574_init(void);
static void _zh_pcf8574_event_handler(void *arg, esp_event_base_t event_base, int32_t event_id, void *event_data);
static void _zh_encoder_event_handler(void *arg, esp_event_base_t event_base, int32_t event_id, void *event_data);

static void _zh_component_select(uint8_t component);
static void _zh_relay_on(uint8_t relay_number);
static void _zh_relay_off(uint8_t relay_number);

void app_main(void)
{
    nvs_flash_init();
    esp_event_loop_create_default();
    _zh_wifi_softap_init();
    _zh_webserver_init();
    _zh_encoder_init();
    _zh_ac_dimmer_init();
    _zh_pcf8574_init();
    // zh_pcf8574_write_gpio(&led_handle, TS_LED_GREEN, LED_ON);
    // zh_pcf8574_write_gpio(&led_handle, RET_LED_BLUE, LED_ON);
    // zh_pcf8574_write_gpio(&led_handle, EXT_LED_BLUE, LED_ON);
}

static void _zh_wifi_softap_init(void)
{
    esp_netif_init();
    esp_netif_create_default_wifi_ap();
    wifi_init_config_t wifi_config = WIFI_INIT_CONFIG_DEFAULT();
    esp_wifi_init(&wifi_config);
    wifi_config_t ap_config = {
        .ap = {
            .ssid = WIFI_SSID,
            .password = WIFI_PASS,
            .max_connection = 4,
            .authmode = WIFI_AUTH_WPA2_PSK,
        },
    };
    esp_wifi_set_mode(WIFI_MODE_AP);
    esp_wifi_set_config(WIFI_IF_AP, &ap_config);
    esp_wifi_start();
}

static void _zh_webserver_init(void)
{
    httpd_config_t config = HTTPD_DEFAULT_CONFIG();
    httpd_start(&_webserver_handle, &config);
    zh_ota_server_init(_webserver_handle);
}

static void _zh_encoder_init(void)
{
    esp_event_handler_instance_register(ZH_ENCODER, ESP_EVENT_ANY_ID, &_zh_encoder_event_handler, NULL, NULL);
    zh_encoder_init_config_t config = ZH_ENCODER_INIT_CONFIG_DEFAULT();
    // config.a_gpio_number = PORTC0;
    // config.b_gpio_number = PORTC1;
    config.encoder_min_value = 0;
    config.encoder_max_value = 100;
    config.encoder_step = 5;
    config.encoder_number = POWER_ENCODER;
    zh_encoder_init(&config, &_power_encoder_handle);
    zh_encoder_set(&_power_encoder_handle, 100);
    // config.a_gpio_number = PORTC2;
    // config.b_gpio_number = PORTC3;
    config.encoder_min_value = 0;
    // config.encoder_max_value = (sizeof(component_cmm) / sizeof(component_cmm[0])) - 1;
    config.encoder_step = 1;
    config.encoder_number = COMPONENT_ENCODER;
    zh_encoder_init(&config, &_component_encoder_handle);
    zh_encoder_set(&_component_encoder_handle, 0);
}

static void _zh_ac_dimmer_init(void)
{
    zh_ac_dimmer_init_config_t config = ZH_AC_DIMMER_INIT_CONFIG_DEFAULT();
    config.zero_cross_gpio = GPIO_NUM_16;
    config.triac_gpio = GPIO_NUM_4;
    zh_ac_dimmer_init(&config);
}

static void _zh_pcf8574_init(void)
{
    i2c_master_bus_config_t i2c_bus_config = {
        .clk_source = I2C_CLK_SRC_DEFAULT,
        .scl_io_num = GPIO_NUM_22,
        .sda_io_num = GPIO_NUM_21,
        .glitch_ignore_cnt = 7,
    };
    i2c_new_master_bus(&i2c_bus_config, &_i2c_bus_handle);
    esp_event_handler_instance_register(ZH_PCF8574, ESP_EVENT_ANY_ID, &_zh_pcf8574_event_handler, NULL, NULL);
    zh_pcf8574_init_config_t config = ZH_PCF8574_INIT_CONFIG_DEFAULT();
    config.i2c_handle = _i2c_bus_handle;
    config.i2c_address = LED_1_I2C_ADDRESS;
    zh_pcf8574_init(&config, &_led_1_handle);
    config.i2c_address = LED_2_I2C_ADDRESS;
    zh_pcf8574_init(&config, &_led_2_handle);
    config.i2c_address = RELAY_I2C_ADDRESS;
    zh_pcf8574_init(&config, &_relay_handle);
    config.i2c_address = LCD_I2C_ADDRESS;
    zh_pcf8574_init(&config, &lcd_handle);
    zh_160x_init(&lcd_handle, ZH_LCD_16X4);
    config.p0_gpio_work_mode = true;
    config.p1_gpio_work_mode = true;
    config.p2_gpio_work_mode = true;
    config.p3_gpio_work_mode = true;
    config.p4_gpio_work_mode = true;
    config.p5_gpio_work_mode = true;
    config.interrupt_gpio = GPIO_NUM_17;
    config.i2c_address = BUTTON_I2C_ADDRESS;
    zh_pcf8574_init(&config, &_button_handle);
}

static void _zh_pcf8574_event_handler(void *arg, esp_event_base_t event_base, int32_t event_id, void *event_data)
{
    zh_pcf8574_event_on_isr_t *event = event_data;
    switch (event->gpio_number)
    {
    case DMM_BUTTON:
        if (event->gpio_level == LOW && _is_work == false)
        {
            if (_is_dmm_enabled == true)
            {
                zh_pcf8574_write_gpio(&_led_1_handle, DMM_LED_RED, LED_ON);
                zh_pcf8574_write_gpio(&_led_1_handle, DMM_LED_GREEN, LED_OFF);
                zh_pcf8574_write_gpio(&_relay_handle, DMM_RELAY, RELAY_ON);
                _is_dmm_enabled = false;
            }
            else
            {
                zh_pcf8574_write_gpio(&_led_1_handle, DMM_LED_RED, LED_OFF);
                zh_pcf8574_write_gpio(&_led_1_handle, DMM_LED_GREEN, LED_ON);
                zh_pcf8574_write_gpio(&_relay_handle, DMM_RELAY, RELAY_OFF);
                _is_dmm_enabled = true;
            }
        }
        break;
    case FIX_BUTTON:
        if (event->gpio_level == LOW)
        {
            zh_pcf8574_write_gpio(&_led_1_handle, FIX_LED_BLUE, LED_OFF);
            zh_pcf8574_write_gpio(&_led_1_handle, FIX_LED_GREEN, LED_ON);
            _is_fix_enabled = true;
        }
        else
        {
            zh_pcf8574_write_gpio(&_led_1_handle, FIX_LED_BLUE, LED_ON);
            zh_pcf8574_write_gpio(&_led_1_handle, FIX_LED_GREEN, LED_OFF);
            _is_fix_enabled = false;
        }
        break;
    case NUM_1_BUTTON:
        if ((_permitted_channels & CHANNEL_1) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            if (_is_num_1_fixed == true)
            {
                break;
            }
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_1_LED_BLUE, LED_OFF);
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_1_LED_GREEN, LED_ON);
            _zh_relay_on(L1_RELAY);
        }
        else
        {
            if (_is_fix_enabled == true)
            {
                _is_num_1_fixed = true;
                break;
            }
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_1_LED_BLUE, LED_ON);
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_1_LED_GREEN, LED_OFF);
            _zh_relay_off(L1_RELAY);
            _is_num_1_fixed = false;
        }
        break;
    case NUM_2_BUTTON:
        if ((_permitted_channels & CHANNEL_2) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            if (_is_num_2_fixed == true)
            {
                break;
            }
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_2_LED_BLUE, LED_OFF);
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_2_LED_GREEN, LED_ON);
            _zh_relay_on(L2_RELAY);
        }
        else
        {
            if (_is_fix_enabled == true)
            {
                _is_num_2_fixed = true;
                break;
            }
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_2_LED_BLUE, LED_ON);
            zh_pcf8574_write_gpio(&_led_1_handle, NUM_2_LED_GREEN, LED_OFF);
            _zh_relay_off(L2_RELAY);
            _is_num_2_fixed = false;
        }
        break;
    case NUM_3_BUTTON:
        if ((_permitted_channels & CHANNEL_3) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            if (_is_num_3_fixed == true)
            {
                break;
            }
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_3_LED_BLUE, LED_OFF);
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_3_LED_GREEN, LED_ON);
            _zh_relay_on(L3_RELAY);
        }
        else
        {
            if (_is_fix_enabled == true)
            {
                _is_num_3_fixed = true;
                break;
            }
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_3_LED_BLUE, LED_ON);
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_3_LED_GREEN, LED_OFF);
            _zh_relay_off(L3_RELAY);
            _is_num_3_fixed = false;
        }
        break;
    case NUM_4_BUTTON:
        if ((_permitted_channels & CHANNEL_4) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            if (_is_num_4_fixed == true)
            {
                break;
            }
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_4_LED_BLUE, LED_OFF);
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_4_LED_GREEN, LED_ON);
            _zh_relay_on(L4_RELAY);
        }
        else
        {
            if (_is_fix_enabled == true)
            {
                _is_num_4_fixed = true;
                break;
            }
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_4_LED_BLUE, LED_ON);
            zh_pcf8574_write_gpio(&_led_2_handle, NUM_4_LED_GREEN, LED_OFF);
            _zh_relay_off(L4_RELAY);
            _is_num_4_fixed = false;
        }
        break;
    default:
        break;
    }
}

static void _zh_encoder_event_handler(void *arg, esp_event_base_t event_base, int32_t event_id, void *event_data)
{
    zh_encoder_event_on_isr_t *event = event_data;
    switch (event->encoder_number)
    {
    case POWER_ENCODER:
        zh_160x_set_cursor(&lcd_handle, 0, 11);
        zh_160x_print_int(&lcd_handle, (uint8_t)event->encoder_position);
        zh_160x_print_char(&lcd_handle, "% ");
        zh_ac_dimmer_set(event->encoder_position);
        break;
    case COMPONENT_ENCODER:
        _zh_component_select((uint8_t)event->encoder_position);
        break;
    default:
        break;
    }
}

static void _zh_component_select(uint8_t component)
{
    // char text_buffer[LCD_TEXT_BUFFER];
    // permitted_channels = component_channel[component];
    // zh_avr_160x_set_cursor(&lcd_handle, 1, 10);
    // strcpy_P(text_buffer, component_cmm[component]);
    // zh_avr_160x_print_char(&lcd_handle, text_buffer);
    // zh_avr_160x_set_cursor(&lcd_handle, 2, 0);
    // strcpy_P(text_buffer, component_line_1[component]);
    // zh_avr_160x_print_char(&lcd_handle, text_buffer);
    // zh_avr_160x_set_cursor(&lcd_handle, 3, 0);
    // strcpy_P(text_buffer, component_line_2[component]);
    // zh_avr_160x_print_char(&lcd_handle, text_buffer);
    // zh_avr_pcf8574_write(&led1_handle, is_dmm == true ? 0xF9 : 0xFA);
    // zh_avr_pcf8574_write(&led2_handle, 0x0F);
    if ((_permitted_channels & CHANNEL_1) == CHANNEL_1)
    {
        zh_pcf8574_write_gpio(&_led_1_handle, NUM_1_LED_BLUE, LED_ON);
    }
    if ((_permitted_channels & CHANNEL_2) == CHANNEL_2)
    {
        zh_pcf8574_write_gpio(&_led_1_handle, NUM_2_LED_BLUE, LED_ON);
    }
    if ((_permitted_channels & CHANNEL_3) == CHANNEL_3)
    {
        zh_pcf8574_write_gpio(&_led_2_handle, NUM_3_LED_BLUE, LED_ON);
    }
    if ((_permitted_channels & CHANNEL_4) == CHANNEL_4)
    {
        zh_pcf8574_write_gpio(&_led_2_handle, NUM_4_LED_BLUE, LED_ON);
    }
}

static void _zh_relay_on(uint8_t relay_number)
{
    if (++_used_channels > 1)
    {
        zh_ac_dimmer_stop();
        vTaskDelay(5);
        zh_pcf8574_write_gpio(&_relay_handle, relay_number, RELAY_ON);
        vTaskDelay(20);
        zh_ac_dimmer_start();
    }
    else
    {
        _is_work = true;
        zh_pcf8574_write_gpio(&_relay_handle, relay_number, RELAY_ON);
        zh_pcf8574_write_gpio(&_relay_handle, GROUND_RELAY, RELAY_ON);
        vTaskDelay(20);
        zh_ac_dimmer_start();
    }
}

static void _zh_relay_off(uint8_t relay_number)
{
    if (--_used_channels > 0)
    {
        zh_ac_dimmer_stop();
        vTaskDelay(5);
        zh_pcf8574_write_gpio(&_relay_handle, relay_number, RELAY_OFF);
        vTaskDelay(20);
        zh_ac_dimmer_start();
    }
    else
    {
        zh_ac_dimmer_stop();
        vTaskDelay(5);
        zh_pcf8574_write_gpio(&_relay_handle, relay_number, RELAY_OFF);
        zh_pcf8574_write_gpio(&_relay_handle, GROUND_RELAY, RELAY_OFF);
        _is_work = false;
    }
}