local ui_get, ui_set, ui_ref = ui.get, ui.set, ui.reference
local ui_reference = ui.reference
local globals_curtime = globals.curtime
local ui_newslider = ui.new_slider
local ui_desynctype = ui.new_combobox
local ui_set_visible = ui.set_visible
local client_log = client.log
local renderer_text = renderer.text
local client_screen_size = client.screen_size
local client_set_event_callback = client.set_event_callback
local globals_realtime = globals.realtime
local math_floor = math.floor

local reference_pitch, reference_yaw_slider = ui_reference("AA", "Anti-aimbot angles", "Pitch")
local reference_yaw, reference_yaw_slider = ui_reference("AA", "Anti-aimbot angles", "Yaw")
local reference_body, reference_body_slider = ui_reference("AA", "Anti-aimbot angles", "Body yaw")
local reference_yaw_jitter_type, reference_yaw_jit_slider = ui_reference("AA", "Anti-aimbot angles", "Yaw jitter")

local ui_desync = ui.new_checkbox("AA", "Other", "Enable Desync")
local ui_indicator_color_picker = ui.new_color_picker("AA", "Other", "Indicator color", 0, 255, 255, 255)
local ui_desynctype = ui.new_combobox("AA", "Other", "Desync Type", {"Jitter 90/-90", "Customizable Jitter", "Static", "Offset Jitter"})
local ui_indicatortype = ui.new_combobox("AA", "Other", "Indicator Type", {"Default", "Zeus", "Fatality"})
local ui_rainbow = ui.new_checkbox("AA", "Other", "Rainbow")
local ui_rainbowspeed = ui_newslider("AA", "Other", "Rainbow Speed", 1, 10, 3, true)
local ui_left_hotkey = ui.new_hotkey("AA", "Other", "Left")
local ui_right_hotkey = ui.new_hotkey("AA", "Other", "Right")
local ui_back_hotkey = ui.new_hotkey("AA", "Other", "Back")

local isLeft, isRight, isBack = false, false, false

local function hsv_to_rgb(h, s, v, a)
    local r, g, b

    local i = math_floor(h * 6);
    local f = h * 6 - i;
    local p = v * (1 - s);
    local q = v * (1 - f * s);
    local t = v * (1 - (1 - f) * s);

    i = i % 6

    if i == 0 then r, g, b = v, t, p
    elseif i == 1 then r, g, b = q, v, p
    elseif i == 2 then r, g, b = p, v, t
    elseif i == 3 then r, g, b = p, q, v
    elseif i == 4 then r, g, b = t, p, v
    elseif i == 5 then r, g, b = v, p, q
    end

    return r * 255, g * 255, b * 255, a * 255
end

local function func_rgb_rainbowize(frequency, rgb_split_ratio)
    if ui_get(ui_rainbow) then
        local r, g, b, a = hsv_to_rgb(globals_realtime() * frequency, 1, 1, 1)

        r = r * rgb_split_ratio
        g = g * rgb_split_ratio
        b = b * rgb_split_ratio

        return r, g, b
    end
end

local function get_antiaim_dir()
    if ui_get(ui_left_hotkey) then
        isLeft = true
        isRight = false
        isBack = false
    elseif ui_get(ui_right_hotkey) then
        isRight = true
        isLeft = false
        isBack = false
    elseif ui_get(ui_back_hotkey) then
        isRight = false
        isLeft = false
        isBack = true
    end
end

local function setLeft()
    if ui_get(ui_desynctype) == "Jitter 90/-90" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, -90)
        ui_set(reference_yaw_jitter_type, "Center")
        ui_set(reference_yaw_jit_slider, 90)
        ui_set(reference_body, "Opposite")
        ui_set(reference_body_slider, 0)
    elseif ui_get(ui_desynctype) == "Customizable Jitter" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, -90)
        ui_set(reference_body, "Opposite")
        ui_set(reference_body_slider, 0)
    elseif ui_get(ui_desynctype) == "Static" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 0)
        ui_set(reference_yaw_jitter_type, "Offset")
        ui_set(reference_yaw_jit_slider, 0)
        ui_set(reference_body, "Static")
        ui_set(reference_body_slider, 90)
    elseif ui_get(ui_desynctype) == "Offset Jitter" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 0)
        ui_set(reference_yaw_jitter_type, "Offset")
        ui_set(reference_yaw_jit_slider, 158)
        ui_set(reference_body, "Opposite")
    end
end

local function setRight()
    if ui_get(ui_desynctype) == "Jitter 90/-90" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 90)
        ui_set(reference_yaw_jitter_type, "Center")
        ui_set(reference_yaw_jit_slider, 90)
        ui_set(reference_body, "Opposite")
        ui_set(reference_body_slider, 0)
    elseif ui_get(ui_desynctype) == "Customizable Jitter" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 90)
        ui_set(reference_body, "Opposite")
        ui_set(reference_body_slider, 0)
    elseif ui_get(ui_desynctype) == "Static" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 0)
        ui_set(reference_yaw_jitter_type, "Offset")
        ui_set(reference_yaw_jit_slider, 0)
        ui_set(reference_body, "Static")
        ui_set(reference_body_slider, -90)
    elseif ui_get(ui_desynctype) == "Customizable Jitter" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 90)
        ui_set(reference_body, "Opposite")
        ui_set(reference_body_slider, 0)
    elseif ui_get(ui_desynctype) == "Offset Jitter" then
        ui_set(reference_pitch, "Down")
        ui_set(reference_yaw, "180")
        ui_set(reference_yaw_slider, 0)
        ui_set(reference_yaw_jitter_type, "Offset")
        ui_set(reference_yaw_jit_slider, -158)
        ui_set(reference_body, "Opposite")
        ui_set(reference_body_slider, -8)
    end
end

local function setBack()
    ui_set(reference_yaw, "180")
    ui_set(reference_yaw_slider, 0)
    ui_set(reference_body, "Off")
end

local function on_paint()
    if not ui_get(ui_desync) then
        return
    end

    get_antiaim_dir()

    local screen_width, screen_height = client_screen_size()
    local center_x, center_y = screen_width / 2, screen_height / 2
    if ui_get(ui_indicatortype) == "Default" then
        renderer_text(center_x + 50, center_y, 255, 255, 255, 155, "c+", 0, ">")
        renderer_text(center_x - 50, center_y, 255, 255, 255, 155, "c+", 0, "<")
        renderer_text(center_x, center_y + 40, 255, 255, 255, 155, "c+", 0, "v")
    elseif ui_get(ui_indicatortype) == "Zeus" then
        renderer_text(center_x + 50, center_y, 255, 255, 255, 155, "c+", 0, "►")
        renderer_text(center_x - 50, center_y, 255, 255, 255, 155, "c+", 0, "◄")
        renderer_text(center_x, center_y + 40, 255, 255, 255, 155, "c+", 0, "▼")
    elseif ui_get(ui_indicatortype) == "Fatality" then
        renderer_text(center_x + 50, center_y, 255, 255, 255, 155, "c+", 0, "⮞")
        renderer_text(center_x - 50, center_y, 255, 255, 255, 155, "c+", 0, "⮜")
        renderer_text(center_x, center_y + 40, 255, 255, 255, 155, "c+", 0, "⮟")
    end
    local indicator_r, indicator_g, indicator_b, indicator_a = ui_get(ui_indicator_color_picker)

    if ui_get(ui_rainbow) then
        indicator_r, indicator_g, indicator_b = func_rgb_rainbowize(ui_get(ui_rainbowspeed) / 8, 1)
    end

    if isLeft then
        setLeft()
        if ui_get(ui_indicatortype) == "Default" then
            renderer_text(center_x - 50, center_y, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "<")
        elseif ui_get(ui_indicatortype) == "Zeus" then
            renderer_text(center_x - 50, center_y, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "◄")
        elseif ui_get(ui_indicatortype) == "Fatality" then
            renderer_text(center_x - 50, center_y, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "⮜")
        end
    elseif isRight then
        setRight()
        if ui_get(ui_indicatortype) == "Default" then
            renderer_text(center_x + 50, center_y, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, ">")
        elseif ui_get(ui_indicatortype) == "Zeus" then
            renderer_text(center_x + 50, center_y, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "►")
        elseif ui_get(ui_indicatortype) == "Fatality" then
            renderer_text(center_x + 50, center_y, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "⮞")
        end
    elseif isBack then
        setBack()
        if ui_get(ui_indicatortype) == "Default" then
            renderer_text(center_x, center_y + 40, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "v")
        elseif ui_get(ui_indicatortype) == "Zeus" then
            renderer_text(center_x, center_y + 40, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "▼")
        elseif ui_get(ui_indicatortype) == "Fatality" then
            renderer_text(center_x, center_y + 40, indicator_r, indicator_g, indicator_b, indicator_a, "c+", 0, "⮟")
        end
    end
end
client_set_event_callback("paint", on_paint)

local function on_rainbow_changed()
    local desync = ui_get(ui_desync)
    local rainbow = ui_get(ui_rainbow)
    ui_set_visible(ui_rainbowspeed, desync and rainbow)
end
ui.set_callback(ui_rainbow, on_rainbow_changed)

local function on_desync_changed()
    local desync = ui_get(ui_desync)
    ui_set_visible(ui_desynctype, desync)
    ui_set_visible(ui_indicatortype, desync)
    ui_set_visible(ui_rainbow, desync)
    ui_set_visible(ui_left_hotkey, desync)
    ui_set_visible(ui_right_hotkey, desync)
    ui_set_visible(ui_back_hotkey, desync)

    on_rainbow_changed()
end
ui.set_callback(ui_desync, on_desync_changed)
on_desync_changed()