#include "eth_worker_i.h"
#include "eth_worker.h"
#include "eth_save_process.h"

#include <furi_hal.h>
#include "dhcp.h"
#include "socket.h"
#include <wizchip_conf.h>

#define ETH_RESET_PIN (&gpio_ext_pc3)
#define ETH_CS_PIN    (&gpio_ext_pa4)

#define TAG "EthWorker"

static EthWorker* static_worker = NULL;

EthWorker* eth_worker_alloc() {
    EthWorker* worker = malloc(sizeof(EthWorker));

    worker->config = ethernet_save_process_malloc();
    furi_assert(worker->config);

    worker->init_process = ethernet_view_process_malloc(EthWorkerProcessInit, worker->config);
    worker->dhcp_process = ethernet_view_process_malloc(EthWorkerProcessDHCP, worker->config);
    worker->stat_process = ethernet_view_process_malloc(EthWorkerProcessStatic, worker->config);
    worker->ping_process = ethernet_view_process_malloc(EthWorkerProcessPing, worker->config);
    worker->traceroute_process = ethernet_view_process_malloc(EthWorkerProcessTraceroute, worker->config);
    worker->reset_process = ethernet_view_process_malloc(EthWorkerProcessReset, worker->config);
    worker->active_process = worker->init_process;

    static_worker = worker;

    worker->state = worker->next_state = EthWorkerStateNotAllocated;

    worker->timer = furi_timer_alloc(dhcp_timer_callback, FuriTimerTypePeriodic, NULL);
    furi_timer_start(worker->timer, 1000);

    eth_log(EthWorkerProcessReset, "Finik Ethernet [START]");

    return worker;
}

void eth_worker_free(EthWorker* worker) {
    eth_log(EthWorkerProcessReset, "Finik Ethernet [STOP]");

    eth_run(worker, EthWorkerProcessExit);

    static_worker = NULL;
    furi_assert(worker);

    ethernet_view_process_free(worker->init_process);
    ethernet_view_process_free(worker->dhcp_process);
    ethernet_view_process_free(worker->stat_process);
    ethernet_view_process_free(worker->ping_process);
    ethernet_view_process_free(worker->traceroute_process);
    ethernet_view_process_free(worker->reset_process);
    ethernet_save_process_free(worker->config);

    furi_timer_stop(worker->timer);
    furi_timer_free(worker->timer);

    free(worker);
}

void eth_worker_change_state(EthWorker* worker, EthWorkerState state) {
    furi_assert(worker);
    worker->state = state;
}

void eth_worker_set_active_process(EthWorker* worker, EthWorkerProcess state) {
    furi_assert(worker);
    switch(state) {
    case EthWorkerProcessInit:
        worker->active_process = worker->init_process;
        break;
    case EthWorkerProcessDHCP:
        worker->active_process = worker->dhcp_process;
        break;
    case EthWorkerProcessStatic:
        worker->active_process = worker->stat_process;
        break;
    case EthWorkerProcessPing:
        worker->active_process = worker->ping_process;
        break;
    case EthWorkerProcessReset:
        worker->active_process = worker->reset_process;
        break;
    default:
        break;
    }
}

void eth_worker_log(EthWorker* worker, const char* str) {
    furi_assert(worker);
    ethernet_save_process_print(worker->config, str);
}

static EthViewProcess* get_process(EthWorker* worker, EthWorkerProcess process) {
    furi_assert(worker);
    switch(process) {
    case EthWorkerProcessInit:
        return worker->init_process;
    case EthWorkerProcessDHCP:
        return worker->dhcp_process;
    case EthWorkerProcessStatic:
        return worker->stat_process;
    case EthWorkerProcessPing:
        return worker->ping_process;
    case EthWorkerProcessReset:
        return worker->reset_process;
    case EthWorkerProcessActive:
        return worker->active_process;
    default:
        return NULL;
    }
}

void eth_log(EthWorkerProcess process, const char* format, ...) {
    furi_assert(static_worker);
    va_list args;
    va_start(args, format);
    FuriString* fstring = furi_string_alloc_vprintf(format, args);
    const char* string = furi_string_get_cstr(fstring);
    va_end(args);

    FURI_LOG_I(TAG, "%s", string);
    ethernet_save_process_print(static_worker->config, string);
    ethernet_view_process_print(get_process(static_worker, process), string);
    if(process != EthWorkerProcessReset) {
        ethernet_view_process_print(get_process(static_worker, EthWorkerProcessReset), string);
    }
    furi_string_free(fstring);
}

void eth_printf(const char* format, ...) {
    furi_assert(static_worker);
    va_list args;
    va_start(args, format);
    FuriString* fstring = furi_string_alloc_vprintf(format, args);
    const char* string = furi_string_get_cstr(fstring);
    va_end(args);

    FURI_LOG_I(TAG, "%s", string);
    //ehternet_save_process_print(static_worker->config, string);
    furi_string_free(fstring);
}

static void eth_set_force_state(EthWorkerState state) {
    EthWorker* worker = static_worker;
    furi_assert(worker);
    worker->next_state = EthWorkerStateDefaultNext;
    if(worker->state == EthWorkerStateNotAllocated) {
        return;
    }
    worker->state = state;
}

void eth_set_state() {
}

void eth_set_next_state(EthWorkerState state) {
    EthWorker* worker = static_worker;
    furi_assert(worker);
    if(state == EthWorkerStateReset || state == EthWorkerStateStop) {
        eth_set_force_state(state);
        return;
    }
    if(worker->state == EthWorkerStateNotInited) {
        if(state == EthWorkerStateInit) {
            worker->next_state = state;
        }
    }
    if(worker->state == EthWorkerStateInited) {
        if(state == EthWorkerStateDHCP || state == EthWorkerStateStaticIp) {
            worker->next_state = state;
        }
    }
    if(worker->state == EthWorkerStateOnline) {
        if(state == EthWorkerStatePing) {
            worker->next_state = state;
        }
    }
}

void eth_run(EthWorker* worker, EthWorkerProcess process) {
    furi_assert(worker);
    switch(process) {
    case EthWorkerProcessInit:
        if(worker->state == EthWorkerStateNotAllocated) {
            worker->thread = furi_thread_alloc();
            furi_thread_set_name(worker->thread, "EthWorker");
            furi_thread_set_stack_size(worker->thread, 8192);
            furi_thread_set_callback(worker->thread, eth_worker_task);
            furi_thread_set_context(worker->thread, worker);
            worker->state = EthWorkerStateNotInited;
            furi_thread_start(worker->thread);
        }
        eth_set_next_state(EthWorkerStateInit);
        break;
    case EthWorkerProcessDHCP:
        eth_set_next_state(EthWorkerStateDHCP);
        break;
    case EthWorkerProcessStatic:
        eth_set_next_state(EthWorkerStateStaticIp);
        break;
    case EthWorkerProcessPing:
        eth_set_next_state(EthWorkerStatePing);
        break;
    case EthWorkerProcessReset:
        eth_set_next_state(EthWorkerStateReset);
        eth_log(EthWorkerProcessInit, "reset module");
        eth_log(EthWorkerProcessDHCP, "reset module");
        eth_log(EthWorkerProcessStatic, "reset module");
        eth_log(EthWorkerProcessPing, "reset module");
        eth_log(EthWorkerProcessReset, "reset module");
        break;
    case EthWorkerProcessExit:
        if(worker->state != EthWorkerStateNotAllocated) {
            eth_set_force_state(EthWorkerStateStop);
            furi_thread_join(worker->thread);
            furi_thread_free(worker->thread);
            worker->state = EthWorkerStateNotAllocated;
        }
        break;
    default:
        break;
    }
}

/************************** Ethernet Worker Thread *****************************/

static uint8_t ip_assigned = 0;

static void W5500_Select(void) {
    furi_hal_gpio_write(ETH_CS_PIN, false);
}

static void W5500_Unselect(void) {
    furi_hal_gpio_write(ETH_CS_PIN, true);
}

static void Callback_IPAssigned(void) {
    eth_log(
        EthWorkerProcessDHCP, "Callback: IP assigned! Leased time: %d sec", getDHCPLeasetime());
    ip_assigned = 1;
}

static void Callback_IPConflict(void) {
    eth_log(EthWorkerProcessDHCP, "Callback: IP conflict!");
}

static void W5500_ReadBuff(uint8_t* buff, uint16_t len) {
    furi_hal_spi_bus_rx(&furi_hal_spi_bus_handle_external, buff, len, 1000);
}

static void W5500_WriteBuff(uint8_t* buff, uint16_t len) {
    furi_hal_spi_bus_tx(&furi_hal_spi_bus_handle_external, buff, len, 1000);
}

static uint8_t W5500_ReadByte(void) {
    uint8_t byte;
    W5500_ReadBuff(&byte, sizeof(byte));
    return byte;
}

static void W5500_WriteByte(uint8_t byte) {
    W5500_WriteBuff(&byte, sizeof(byte));
}

static wiz_NetInfo gWIZNETINFO;
void update_WIZNETINFO(uint8_t is_dhcp) {
    furi_assert(static_worker);
    memcpy(gWIZNETINFO.mac, static_worker->config->mac, 6);
    if(is_dhcp) {
        memset(gWIZNETINFO.ip, 0, 4);
        memset(gWIZNETINFO.sn, 0, 4);
        memset(gWIZNETINFO.gw, 0, 4);
        memset(gWIZNETINFO.dns, 0, 4);
        gWIZNETINFO.dhcp = NETINFO_DHCP;
    } else {
        memcpy(gWIZNETINFO.ip, static_worker->config->ip, 4);
        memcpy(gWIZNETINFO.sn, static_worker->config->mask, 4);
        memcpy(gWIZNETINFO.gw, static_worker->config->gateway, 4);
        memcpy(gWIZNETINFO.dns, static_worker->config->dns, 4);
        gWIZNETINFO.dhcp = NETINFO_STATIC;
    }
}

int check_phylink(EthWorker* worker, EthWorkerState state, EthWorkerProcess proc, int timeout) {
    uint32_t start_time = furi_get_tick();
    uint32_t last_log_time = start_time;
    eth_log(proc, "phy link check 0");
    for(;;) {
        if(furi_get_tick() > start_time + timeout) {
            eth_log(proc, "phy link timeout");
            break;
        }
        if(worker->state != state) {
            eth_log(proc, "state changed");
            break;
        }
        uint8_t link = PHY_LINK_OFF;
        if(ctlwizchip(CW_GET_PHYLINK, (void*)&link) == -1) {
            eth_log(proc, "Unknown PHY link status");
            break;
        }
        if(link != PHY_LINK_OFF) {
            eth_log(proc, "phy link on");
            return 1;
        }
        furi_delay_ms(20);
        if(furi_get_tick() > last_log_time + 1000) {
            eth_log(proc, "phy link check %d", (last_log_time - start_time) / 1000);
            last_log_time = furi_get_tick();
        }
    }
    return 0;
}

#define DHCP_SOCKET 0
uint8_t ping_auto(uint8_t s, uint8_t* addr);

static void load_net_parameters(const EthernetSaveConfig* cfg) {
    gWIZNETINFO.gw[0] = cfg->gateway[0];
    gWIZNETINFO.gw[1] = cfg->gateway[1];
    gWIZNETINFO.gw[2] = cfg->gateway[2];
    gWIZNETINFO.gw[3] = cfg->gateway[3];

    gWIZNETINFO.sn[0] = cfg->mask[0];
    gWIZNETINFO.sn[1] = cfg->mask[1];
    gWIZNETINFO.sn[2] = cfg->mask[2];
    gWIZNETINFO.sn[3] = cfg->mask[3];

    gWIZNETINFO.mac[0] = cfg->mac[0];
    gWIZNETINFO.mac[1] = cfg->mac[1];
    gWIZNETINFO.mac[2] = cfg->mac[2];
    gWIZNETINFO.mac[3] = cfg->mac[3];
    gWIZNETINFO.mac[4] = cfg->mac[4];
    gWIZNETINFO.mac[5] = cfg->mac[5];

    gWIZNETINFO.ip[0] = cfg->ip[0];
    gWIZNETINFO.ip[1] = cfg->ip[1];
    gWIZNETINFO.ip[2] = cfg->ip[2];
    gWIZNETINFO.ip[3] = cfg->ip[3];

    gWIZNETINFO.dns[0] = cfg->dns[0];
    gWIZNETINFO.dns[1] = cfg->dns[1];
    gWIZNETINFO.dns[2] = cfg->dns[2];
    gWIZNETINFO.dns[3] = cfg->dns[3];

    gWIZNETINFO.dhcp = NETINFO_STATIC;
}

int32_t eth_worker_task(void* context) {
    furi_assert(context);
    EthWorker* worker = (EthWorker*)context;

    furi_hal_spi_bus_handle_init(&furi_hal_spi_bus_handle_external);
    furi_hal_spi_acquire(&furi_hal_spi_bus_handle_external);

    uint8_t W5500FifoSize[2][8] = {{2, 2, 2, 2, 2, 2, 2, 2}, {2, 2, 2, 2, 2, 2, 2, 2}};
    uint8_t dhcp_buffer[2048];

    reg_wizchip_spi_cbfunc(W5500_ReadByte, W5500_WriteByte);
    reg_wizchip_spiburst_cbfunc(W5500_ReadBuff, W5500_WriteBuff);
    reg_wizchip_cs_cbfunc(W5500_Select, W5500_Unselect);

    //furi_hal_gpio_write(ETH_CS_PIN, true);
    furi_hal_gpio_write(ETH_RESET_PIN, true);
    furi_hal_gpio_init(ETH_RESET_PIN, GpioModeOutputPushPull, GpioPullUp, GpioSpeedVeryHigh);
    //furi_hal_gpio_init(ETH_CS_PIN, GpioModeOutputPushPull, GpioPullUp, GpioSpeedVeryHigh);

    while(worker->state != EthWorkerStateStop) {
        if(worker->state == EthWorkerStateNotInited) {
            if(worker->next_state == EthWorkerStateInit) {
                worker->state = worker->next_state;
            }
        } else if(worker->state == EthWorkerStateInited) {
            if(worker->next_state == EthWorkerStateDHCP ||
               worker->next_state == EthWorkerStateStaticIp) {
                worker->state = worker->next_state;
            }
        } else if(worker->state == EthWorkerStateOnline) {
            if(worker->next_state == EthWorkerStatePing) {
                worker->state = worker->next_state;
            }
        } else if(worker->state == EthWorkerStateReset) {
            worker->state = EthWorkerStateNotInited;
        }

        if(worker->state == EthWorkerStateInit) {
            if(!furi_hal_power_is_otg_enabled()) {
                furi_hal_power_enable_otg();
            }
            furi_delay_ms(300);
            furi_hal_gpio_write(ETH_RESET_PIN, false);
            furi_delay_ms(50);
            furi_hal_gpio_write(ETH_RESET_PIN, true);
            if(ctlwizchip(CW_INIT_WIZCHIP, (void*)W5500FifoSize) == -1) {
                eth_log(EthWorkerProcessInit, "[error] W5500 init fail");
                eth_set_force_state(EthWorkerStateNotInited);
                continue;
            }
            eth_log(EthWorkerProcessInit, "W5500 inited");
            furi_delay_ms(90);
            update_WIZNETINFO(false);
            wizchip_setnetinfo(&gWIZNETINFO);
            wiz_NetInfo readed_net_info;
            wizchip_getnetinfo(&readed_net_info);
            if(memcmp(&readed_net_info, &gWIZNETINFO, sizeof(wiz_NetInfo))) {
                eth_log(EthWorkerProcessInit, "[error] module not detected");
                eth_set_force_state(EthWorkerStateNotInited);
                continue;
            }
            setSHAR(gWIZNETINFO.mac);
            wiz_PhyConf conf;
            wizphy_getphyconf(&conf);
            eth_log(
                EthWorkerProcessInit,
                "conf %d %d %d %d",
                conf.by,
                conf.mode,
                conf.speed,
                conf.duplex);
            eth_log(EthWorkerProcessInit, "net info setted");
            eth_log(
                EthWorkerProcessInit,
                "mac: %02X-%02X-%02X-%02X-%02X-%02X",
                gWIZNETINFO.mac[0],
                gWIZNETINFO.mac[1],
                gWIZNETINFO.mac[2],
                gWIZNETINFO.mac[3],
                gWIZNETINFO.mac[4],
                gWIZNETINFO.mac[5]);
            eth_set_force_state(EthWorkerStateInited);
        } else if(worker->state == EthWorkerStateDHCP) {
            if(!check_phylink(worker, EthWorkerStateDHCP, EthWorkerProcessDHCP, 5000)) {
                worker->state = EthWorkerStateInited;
                continue;
            }
            reg_dhcp_cbfunc(Callback_IPAssigned, Callback_IPAssigned, Callback_IPConflict);
            DHCP_init(DHCP_SOCKET, dhcp_buffer);
            {
                // DHCP_run();
                // eth_log(EthWorkerProcessDHCP, "DHCP Send Discover");
                // furi_delay_ms(1000);
                // DHCP_stop();
            }
            uint8_t next_cycle = 1;
            uint16_t divider = 0;
            while(next_cycle && worker->state == EthWorkerStateDHCP) {
                uint8_t dhcp_ret = DHCP_run();
                switch(dhcp_ret) {
                case DHCP_IP_ASSIGN:
                case DHCP_IP_CHANGED:
                case DHCP_IP_LEASED:
                    getIPfromDHCP(gWIZNETINFO.ip);
                    getGWfromDHCP(gWIZNETINFO.gw);
                    getSNfromDHCP(gWIZNETINFO.sn);
                    getDNSfromDHCP(gWIZNETINFO.dns);
                    gWIZNETINFO.dhcp = NETINFO_DHCP;
                    ctlnetwork(CN_SET_NETINFO, (void*)&gWIZNETINFO);
                    eth_log(
                        EthWorkerProcessDHCP, "DHCP IP Leased Time : %ld Sec", getDHCPLeasetime());
                    break;
                case DHCP_FAILED:
                    eth_log(EthWorkerProcessDHCP, "DHCP Failed");
                    break;
                }
                furi_delay_ms(10);
                if(divider++ % 100 == 0) {
                    eth_log(EthWorkerProcessDHCP, "DHCP process %d", divider / 100);
                    if(divider > 2000) {
                        DHCP_stop();
                        eth_log(EthWorkerProcessDHCP, "DHCP Stop by timer");
                        eth_set_force_state(EthWorkerStateInited);
                        break;
                    }
                }
                next_cycle = (dhcp_ret == DHCP_RUNNING);
            }
            if(worker->state != EthWorkerStateDHCP) {
                continue;
            }
            eth_log(
                EthWorkerProcessDHCP,
                "IP address:\n %d.%d.%d.%d",
                gWIZNETINFO.ip[0],
                gWIZNETINFO.ip[1],
                gWIZNETINFO.ip[2],
                gWIZNETINFO.ip[3]);
            eth_log(
                EthWorkerProcessDHCP,
                "SM Mask:\n %d.%d.%d.%d",
                gWIZNETINFO.sn[0],
                gWIZNETINFO.sn[1],
                gWIZNETINFO.sn[2],
                gWIZNETINFO.sn[3]);
            eth_log(
                EthWorkerProcessDHCP,
                "Gate way:\n %d.%d.%d.%d",
                gWIZNETINFO.gw[0],
                gWIZNETINFO.gw[1],
                gWIZNETINFO.gw[2],
                gWIZNETINFO.gw[3]);
            eth_log(
                EthWorkerProcessDHCP,
                "DNS Server:\n %d.%d.%d.%d",
                gWIZNETINFO.dns[0],
                gWIZNETINFO.dns[1],
                gWIZNETINFO.dns[2],
                gWIZNETINFO.dns[3]);
            eth_set_force_state(EthWorkerStateOnline);
        } else if(worker->state == EthWorkerStateStaticIp) {
            if(!check_phylink(worker, EthWorkerStateStaticIp, EthWorkerProcessStatic, 5000)) {
                worker->state = EthWorkerStateInited;
                continue;
            }
            eth_log(EthWorkerProcessStatic, "set static ip");
            load_net_parameters(worker->config);
            eth_log(
                EthWorkerProcessStatic,
                "IP address:\n %d.%d.%d.%d",
                gWIZNETINFO.ip[0],
                gWIZNETINFO.ip[1],
                gWIZNETINFO.ip[2],
                gWIZNETINFO.ip[3]);
            eth_log(
                EthWorkerProcessStatic,
                "SM Mask:\n %d.%d.%d.%d",
                gWIZNETINFO.sn[0],
                gWIZNETINFO.sn[1],
                gWIZNETINFO.sn[2],
                gWIZNETINFO.sn[3]);
            eth_log(
                EthWorkerProcessStatic,
                "Gate way:\n %d.%d.%d.%d",
                gWIZNETINFO.gw[0],
                gWIZNETINFO.gw[1],
                gWIZNETINFO.gw[2],
                gWIZNETINFO.gw[3]);
            eth_log(
                EthWorkerProcessStatic,
                "DNS Server:\n %d.%d.%d.%d",
                gWIZNETINFO.dns[0],
                gWIZNETINFO.dns[1],
                gWIZNETINFO.dns[2],
                gWIZNETINFO.dns[3]);
            ctlnetwork(CN_SET_NETINFO, (void*)&gWIZNETINFO);
            eth_set_force_state(EthWorkerStateOnline);
        } else if(worker->state == EthWorkerStatePing) {
            uint8_t* address = worker->config->ping_ip;
            eth_log(
                EthWorkerProcessPing,
                "ping %d.%d.%d.%d",
                address[0],
                address[1],
                address[2],
                address[3]);
            const uint8_t tries = 4;
            uint8_t try = 0;
            while(try < tries && worker->state == EthWorkerStatePing) {
                try++;
                uint32_t start_time = furi_get_tick();
                uint8_t res = ping_auto_interface(address);
                uint32_t res_time = furi_get_tick();
                if(res == 3) {
                    eth_log(EthWorkerProcessPing, "%d success %d ms", try, res_time - start_time);
                } else {
                    eth_log(
                        EthWorkerProcessPing, "%d error %d, %d", try, res, res_time - start_time);
                    break;
                }
            }
            if(worker->state != EthWorkerStatePing) {
                break;
            }
            eth_set_force_state(EthWorkerStateOnline);
        }
        furi_delay_ms(5);
    }

    furi_hal_spi_release(&furi_hal_spi_bus_handle_external);
    furi_hal_spi_bus_handle_deinit(&furi_hal_spi_bus_handle_external);

    furi_hal_gpio_init_simple(ETH_RESET_PIN, GpioModeAnalog);
    //furi_hal_gpio_init_simple(ETH_CS_PIN, GpioModeAnalog);
    if(furi_hal_power_is_otg_enabled()) {
        furi_hal_power_disable_otg();
    }

    return 0;
}