#include "eth_view_process.h"

#include "eth_worker.h"
#include "eth_worker_i.h"
#include "eth_troubleshooter_icons.h"

#include "ui_hacks_again/u8g2.h"

#include <furi_hal.h>
#include <gui/gui.h>
#include <gui/canvas.h>
#include <string.h>

#define TAG "EthView"



EthViewProcess* ethernet_view_process_malloc(EthWorkerProcess type, EthernetSaveConfig* config) {
    EthViewProcess* evp = malloc(sizeof(EthViewProcess));
    evp->type = type;
    evp->autofill = 1;
    evp->carriage = 0;
    evp->position = 0;
    evp->x = 27;
    evp->y = 6;
    evp->strings_cnt = 10;

    if(type == EthWorkerProcessInit) {
        evp->y += 22;
        EthViewDrawInit* init = malloc(sizeof(EthViewDrawInit));
        memset(init, 0, sizeof(EthViewDrawInit));
        init->mac = (uint8_t*)config->mac;
        evp->draw_struct = init;
    } else if(type == EthWorkerProcessStatic) {
        evp->y += 22;
        EthViewDrawStatic* stat = malloc(sizeof(EthViewDrawStatic));
        memset(stat, 0, sizeof(EthViewDrawStatic));
        stat->ip = config->ip;
        stat->mask = config->mask;
        stat->gateway = config->gateway;
        stat->dns = config->dns;
        evp->draw_struct = stat;
        evp->strings_cnt = 20;
    } else if(type == EthWorkerProcessDHCP) {
        evp->strings_cnt = 20;
    } else if(type == EthWorkerProcessPing) {
        evp->y += 11;
        EthViewDrawPing* ping = malloc(sizeof(EthViewDrawPing));
        memset(ping, 0, sizeof(EthViewDrawPing));
        ping->ip = config->ping_ip;
        evp->draw_struct = ping;
        evp->strings_cnt = 20;
    }

    evp->fifo = malloc(sizeof(EthViewProcessLine) * evp->strings_cnt);
    return evp;
}

void ethernet_view_process_free(EthViewProcess* evp) {
    free(evp->fifo);
    if(evp->type == EthWorkerProcessInit || evp->type == EthWorkerProcessStatic ||
       evp->type == EthWorkerProcessPing) {
        free(evp->draw_struct);
    }
    free(evp);
}

static void draw_hex_digit(Canvas* canvas, uint8_t x, uint8_t y, uint8_t digit) {
    char digit_str[] = "0";
    if(digit < 0xA) {
        digit_str[0] += digit;
    } else if(digit < 0x10) {
        digit_str[0] = 'A';
        digit_str[0] += digit - 0xA;
    } else {
        return;
    }

    canvas_draw_str(canvas, x, y, digit_str);
}

static void draw_dec_number(Canvas* canvas, uint8_t x, uint8_t y, uint8_t num) {
    char num_str[] = "0";
    {
        num_str[0] = '0' + num % 10;
        canvas_draw_str(canvas, x + 6 + 6, y, num_str);
    }
    if(num >= 10) {
        num_str[0] = '0' + num / 10 - (num / 100) * 10;
        canvas_draw_str(canvas, x + 6, y, num_str);
    }
    if(num >= 100) {
        num_str[0] = '0' + num / 100;
        canvas_draw_str(canvas, x, y, num_str);
    }
}

static void draw_static_mode(Canvas* canvas, uint8_t mode) {
    const uint8_t s1 = 13;
    const uint8_t s2 = 31;
    const uint8_t s3 = 19;
    const uint8_t s4 = 21;
    const uint8_t s = 35;
    const uint8_t h = 7;
    const uint8_t y = 10;
    const uint8_t y1 = 15;

    if(mode == EthViewDrawStaticModeIp) {
        canvas_invert_color(canvas);
        canvas_draw_box(canvas, s, y, s1, h);
        canvas_invert_color(canvas);
        canvas_draw_str(canvas, 38, y1, "ip");
    }
    if(mode == EthViewDrawStaticModeMask) {
        canvas_invert_color(canvas);
        canvas_draw_box(canvas, s + s1, y, s2, h);
        canvas_invert_color(canvas);
        canvas_draw_str(canvas, 53, y1, "mask");
    }
    if(mode == EthViewDrawStaticModeGateway) {
        canvas_invert_color(canvas);
        canvas_draw_box(canvas, s + s1 + s2, y, s3, h);
        canvas_invert_color(canvas);
        canvas_draw_str(canvas, 82, y1, "gw");
    }
    if(mode == EthViewDrawStaticModeDNS) {
        canvas_invert_color(canvas);
        canvas_draw_box(canvas, s + s1 + s2 + s3, y, s4, h);
        canvas_invert_color(canvas);
        canvas_draw_str(canvas, 102, y1, "dns");
    }
}

static uint8_t* draw_static_get_current_address(EthViewDrawStatic* evds) {
    furi_assert(evds);
    if(evds->current_mode == EthViewDrawStaticModeIp) return evds->ip;
    if(evds->current_mode == EthViewDrawStaticModeMask) return evds->mask;
    if(evds->current_mode == EthViewDrawStaticModeGateway) return evds->gateway;
    if(evds->current_mode == EthViewDrawStaticModeDNS) return evds->dns;
    return evds->ip;
}

void ethernet_view_process_draw(EthViewProcess* process, Canvas* canvas) {
    furi_assert(canvas);
    furi_assert(process);
    canvas_set_font(canvas, FontSecondary);

    const uint8_t x = process->x;
    const uint8_t y = process->y;
    const uint8_t str_height = 11;
    const uint8_t str_count = (64 - y) / str_height;
    uint8_t carriage = process->carriage;
    uint8_t position = process->position;

    if(process->autofill) {
        position = (carriage + process->strings_cnt - str_count) % process->strings_cnt;
        process->position = position;
    }

    for(uint8_t i = 0; i < str_count; ++i) {
        uint8_t y1 = y + (i + 1) * str_height;
        canvas_draw_str(canvas, x, y1, process->fifo[(position + i) % process->strings_cnt].data);
    }

    if(process->type == EthWorkerProcessInit) {
        uint8_t editing = process->editing;
        canvas_draw_icon(canvas, 27, 10, &I_init_100x19px);
        uint8_t octet = ((EthViewDrawInit*)process->draw_struct)->current_octet;
        uint8_t* mac = ((EthViewDrawInit*)process->draw_struct)->mac;
        for(uint8_t i = 0; i < 6; ++i) {
            uint8_t x1 = 29 + i * 17;
            uint8_t x2 = x1 + 6;
            draw_hex_digit(canvas, x1, 25, (mac[i] & 0xF0) >> 4);
            draw_hex_digit(canvas, x2, 25, (mac[i] & 0x0F));
            if(editing && (octet / 2 == i)) {
                uint8_t x = octet & 1 ? x2 : x1;
                canvas_draw_line(canvas, x, 26, x + 4, 26);
                canvas_draw_line(canvas, x, 27, x + 4, 27);
            }
        }
    } else if(process->type == EthWorkerProcessStatic) {
        canvas_draw_frame(canvas, 31, 18, 21, 13);
        canvas_draw_frame(canvas, 55, 18, 21, 13);
        canvas_draw_frame(canvas, 79, 18, 21, 13);
        canvas_draw_frame(canvas, 103, 18, 21, 13);
        canvas_draw_box(canvas, 29, 10, 97, 7);
        uint8_t mode = ((EthViewDrawStatic*)process->draw_struct)->current_mode;
        uint8_t current_digit = ((EthViewDrawStatic*)process->draw_struct)->current_digit;
        uint8_t* address =
            draw_static_get_current_address((EthViewDrawStatic*)process->draw_struct);
        uint8_t editing = ((EthViewDrawStatic*)process->draw_struct)->editing;
        for(uint8_t i = 0; i < 4; ++i) {
            if(i == mode && process->editing) {
                draw_static_mode(canvas, mode);
            } else {
                canvas_invert_color(canvas);
                draw_static_mode(canvas, i);
                canvas_invert_color(canvas);
            }
        }
        for(uint8_t i = 0; i < 4; ++i) {
            uint8_t x = 33 + i * 24;
            draw_dec_number(canvas, x, 27, address[i]);
            if(editing && (current_digit / 3 == i)) {
                uint8_t x1 = x + 6 * (current_digit % 3);
                canvas_draw_line(canvas, x1, 28, x1 + 4, 28);
                canvas_draw_line(canvas, x1, 29, x1 + 4, 29);
            }
        }
    } else if(process->type == EthWorkerProcessPing) {
        canvas_draw_frame(canvas, 35, 8, 21, 13);
        canvas_draw_frame(canvas, 59, 8, 21, 13);
        canvas_draw_frame(canvas, 83, 8, 21, 13);
        canvas_draw_frame(canvas, 107, 8, 21, 13);
        uint8_t current_digit = ((EthViewDrawPing*)process->draw_struct)->current_digit;
        uint8_t* address = ((EthViewDrawPing*)process->draw_struct)->ip;
        for(uint8_t i = 0; i < 4; ++i) {
            uint8_t x = 37 + i * 24;
            draw_dec_number(canvas, x, 17, address[i]);
            if(process->editing && (current_digit / 3 == i)) {
                uint8_t x1 = x + 6 * (current_digit % 3);
                canvas_draw_line(canvas, x1, 18, x1 + 4, 18);
                canvas_draw_line(canvas, x1, 19, x1 + 4, 19);
            }
        }
    } else if(process->type == EthWorkerProcessReset) {
        process->editing = 0;
    }
}

static void mac_change_hex_digit(uint8_t* mac, uint8_t octet, int8_t diff) {
    uint8_t digit = (octet & 1) ? (mac[octet / 2] & 0x0F) : (mac[octet / 2] >> 4);
    digit = (digit + diff) & 0xF;
    if(octet & 1) {
        mac[octet / 2] = (mac[octet / 2] & 0xF0) | digit;
    } else {
        mac[octet / 2] = (digit << 4) | (mac[octet / 2] & 0x0F);
    }
}

static void address_change_dec_digit(uint8_t* ip, uint8_t digit, int8_t diff) {
    {
        uint8_t k = 0;
        k = (digit % 3 == 0) ? 100 : k;
        k = (digit % 3 == 1) ? 10 : k;
        k = (digit % 3 == 2) ? 1 : k;
        diff *= k;
    }
    {
        int16_t ip1 = ip[digit / 3];
        if(diff > 0 && ((0x100 - ip1) > diff)) ip1 += diff;
        if(diff < 0 && (ip1 + diff >= 0)) ip1 += diff;
        ip[digit / 3] = ip1;
    }
}

void ethernet_view_process_keyevent(EthViewProcess* process, InputKey key) {
    furi_assert(process);
    if(process->type == EthWorkerProcessInit) {
        uint8_t octet = ((EthViewDrawInit*)process->draw_struct)->current_octet;
        uint8_t* mac = ((EthViewDrawInit*)process->draw_struct)->mac;
        if(key == InputKeyLeft) {
            if(octet > 0) octet -= 1;
        } else if(key == InputKeyRight) {
            if(octet < 11) octet += 1;
        } else if(key == InputKeyUp) {
            mac_change_hex_digit(mac, octet, 1);
        } else if(key == InputKeyDown) {
            mac_change_hex_digit(mac, octet, -1);
        } else if(key == InputKeyOk || key == InputKeyBack) {
            process->editing = 0;
        }
        ((EthViewDrawInit*)process->draw_struct)->current_octet = octet;
    } else if(process->type == EthWorkerProcessDHCP) {
        process->editing = 0;
        if(key == InputKeyUp || key == InputKeyDown) {
            ethernet_view_process_move(process, (key == InputKeyDown) ? 1 : -1);
        }
    } else if(process->type == EthWorkerProcessStatic) {
        uint8_t digit = ((EthViewDrawStatic*)process->draw_struct)->current_digit;
        uint8_t mode = ((EthViewDrawStatic*)process->draw_struct)->current_mode;
        uint8_t* address =
            draw_static_get_current_address((EthViewDrawStatic*)process->draw_struct);
        uint8_t editing = ((EthViewDrawStatic*)process->draw_struct)->editing;
        if(editing) {
            if(key == InputKeyLeft) {
                if(digit > 0) {
                    digit -= 1;
                } else {
                    ((EthViewDrawStatic*)process->draw_struct)->editing = 0;
                }
            } else if(key == InputKeyRight) {
                if(digit < 11) digit += 1;
            } else if(key == InputKeyUp) {
                address_change_dec_digit(address, digit, 1);
            } else if(key == InputKeyDown) {
                address_change_dec_digit(address, digit, -1);
            } else if(key == InputKeyOk || key == InputKeyBack) {
                ((EthViewDrawStatic*)process->draw_struct)->editing = 0;
            }
        } else {
            if(key == InputKeyLeft) {
                if(mode > 0) {
                    mode -= 1;
                } else {
                    process->editing = 0;
                }
            } else if(key == InputKeyRight) {
                if(mode < 3) {
                    mode += 1;
                }
            } else if(key == InputKeyUp || key == InputKeyDown) {
                ethernet_view_process_move(process, (key == InputKeyDown) ? 1 : -1);
            } else if(key == InputKeyOk) {
                ((EthViewDrawStatic*)process->draw_struct)->editing = 1;
            } else if(key == InputKeyBack) {
                mode = 0;
                process->editing = 0;
            }
        }
        ((EthViewDrawStatic*)process->draw_struct)->current_mode = mode;
        ((EthViewDrawStatic*)process->draw_struct)->current_digit = digit;
    } else if(process->type == EthWorkerProcessPing) {
        uint8_t digit = ((EthViewDrawPing*)process->draw_struct)->current_digit;
        uint8_t* address = ((EthViewDrawPing*)process->draw_struct)->ip;
        if(key == InputKeyLeft) {
            if(digit > 0) {
                digit -= 1;
            } else {
                process->editing = 0;
            }
        } else if(key == InputKeyRight) {
            if(digit < 11) digit += 1;
        } else if(key == InputKeyUp) {
            address_change_dec_digit(address, digit, 1);
        } else if(key == InputKeyDown) {
            address_change_dec_digit(address, digit, -1);
        } else if(key == InputKeyOk || key == InputKeyBack) {
            process->editing = 0;
        }
        ((EthViewDrawPing*)process->draw_struct)->current_digit = digit;
    } else {
        if(key == InputKeyBack || key == InputKeyLeft) {
            process->editing = 0;
        }
    }
}

void ethernet_view_process_move(EthViewProcess* process, int8_t shift) {
    furi_assert(process);
    uint8_t position = process->position;
    if(shift <= -process->strings_cnt) {
        position = 0;
    } else if(shift >= process->strings_cnt) {
        position = process->carriage - 1;
    } else {
        position = (position + (process->strings_cnt + shift)) % process->strings_cnt;
    }
    process->position = position;
    process->autofill = !shift;
}

void ethernet_view_process_autofill(EthViewProcess* process, uint8_t state) {
    furi_assert(process);
    process->autofill = state;
}

static uint16_t get_string_with_width(const char* str, uint16_t width) {
    u8g2_t canvas_memory;
    Canvas* canvas = (Canvas*)&canvas_memory; // grazniy hack
    canvas_set_font(canvas, FontSecondary);

    uint8_t end = 0;
    char copy[SCREEN_SYMBOLS_WIDTH + 1] = {0};

    for(;;) {
        if(str[end] == '\n') {
            break;
        }
        if(str[end] == '\0') {
            break;
        }
        if(end == SCREEN_SYMBOLS_WIDTH) {
            break;
        }
        copy[end] = str[end];
        if(canvas_string_width(canvas, copy) > width) {
            end -= 1;
            break;
        }
        end += 1;
    }

    return end;
}

void evp_printf(EthViewProcess* process, const char* format, ...) {
    va_list args;
    va_start(args, format);
    FuriString* fstring = furi_string_alloc_vprintf(format, args);
    va_end(args);
    ethernet_view_process_print(process, furi_string_get_cstr(fstring));
    furi_string_free(fstring);
}

void ethernet_view_process_print(EthViewProcess* process, const char* str) {
    furi_assert(process);

    uint16_t max_width = 126 - process->x;
    uint16_t ptr = 0;
    uint16_t len = strlen(str);

    while(ptr < len) {
        if(str[ptr] == '\n') ptr += 1;
        uint16_t start = ptr;
        ptr += get_string_with_width(str + ptr, max_width);
        uint8_t carriage = process->carriage;
        uint8_t carriage1 = (carriage + 1) % process->strings_cnt;
        uint8_t carriage2 = (carriage + 2) % process->strings_cnt;
        memset(process->fifo[carriage].data, 0, SCREEN_SYMBOLS_WIDTH);
        memset(process->fifo[carriage1].data, 0, SCREEN_SYMBOLS_WIDTH);
        memset(process->fifo[carriage2].data, 0, SCREEN_SYMBOLS_WIDTH);
        memcpy(process->fifo[carriage].data, str + start, ptr - start);
        process->carriage = carriage1;
    }
}