local nickname = dofile_once("mods/quant.ew/files/system/nickname.lua")

local spectate = {}

local gui = GuiCreate()

local camera_player

local re_cam = false

local camera_player_id

local camera_target

local cam_target

local inventory_target

local was_notplayer = false

local has_switched = false

local attached = false

local redo = false

local function cant_spectate(ent)
    return ((GameHasFlagRun("ending_game_completed") or ctx.proxy_opt.perma_death)
            and EntityHasTag(ent, "ew_notplayer"))
            or (EntityHasTag(ent, "polymorphed_cessation")
                and cam_target ~= nil and cam_target.entity ~= ent)
end

local function perks_ui(enable)
    for _, child in ipairs(EntityGetAllChildren(ctx.my_player.entity) or {}) do
        if EntityHasTag(child, "perk_entity") then
            local ui = EntityGetFirstComponentIncludingDisabled(child, "UIIconComponent")
            if ui ~= nil then
                local sprite = ComponentGetValue2(ui, "icon_sprite_file")
                if string.sub(sprite, -1, -1) == "g" then
                    if not enable then
                        ComponentSetValue2(ui, "icon_sprite_file", string.sub(sprite, 0, -2))
                    end
                else
                    if enable then
                        ComponentSetValue2(ui, "icon_sprite_file", sprite .. "g")
                    end
                end
            end
        end
    end
end

spectate.nofun = false

function spectate.disable_throwing(enable, entity)
    if entity == nil then
        entity = cam_target.entity
    end
    if spectate.nofun and not enable and entity == ctx.my_player.entity and EntityHasTag(ctx.my_player.entity, "ew_notplayer") then
        return
    end
    local inv
    for _, child in ipairs(EntityGetAllChildren(entity) or {}) do
        if EntityGetName(child) == "inventory_quick" then
            inv = child
            break
        end
    end
    for _, item in ipairs(EntityGetAllChildren(inv) or {}) do
        local com = EntityGetFirstComponentIncludingDisabled(item, "ItemComponent")
        if com ~= nil then
            ComponentSetValue2(com, "permanently_attached", enable)
        end
    end
end

local function get_me()
    local i = 0
    local alive = -1, -1
    for peer_id, potential_target in pairs(ctx.players) do
        if cant_spectate(potential_target.entity) then
            goto continue
        end
        i = i + 1
        alive = peer_id, i
        if peer_id == ctx.my_id then
            return peer_id, i
        end
        ::continue::
    end
    return alive
end

local function set_camera_free(enable)
    local cam = EntityGetFirstComponentIncludingDisabled(ctx.my_player.entity, "PlatformShooterPlayerComponent")
    if EntityHasTag(ctx.my_player.entity, "polymorphed_cessation") then
        return
    end
    if cam ~= nil then
        ComponentSetValue2(cam, "center_camera_on_this_entity", not enable)
        ComponentSetValue2(cam, "move_camera_with_aim", not enable)
    end
end

local function set_camera_position(x, y)
    local cam = EntityGetFirstComponentIncludingDisabled(ctx.my_player.entity, "PlatformShooterPlayerComponent")
    if cam == nil then
        cam = EntityAddComponent2(ctx.my_player.entity, "PlatformShooterPlayerComponent", {center_camera_on_this_entity=false, move_camera_with_aim=false})
    end
    if cam ~= nil then
        ComponentSetValue2(cam, "mDesiredCameraPos", x, y)
    end
end

local function target()
    if cam_target.entity == ctx.my_player.entity and not EntityHasTag(ctx.my_player.entity, "ew_notplayer") then
        perks_ui(true)
        if attached then
            local inv_me = EntityGetFirstComponent(ctx.my_player.entity, "InventoryGuiComponent")
            if inv_me ~= nil then
                ComponentSetValue2(inv_me, "mActive", attached)
            end
            attached = false
        end
        local sprite = EntityGetFirstComponent(ctx.my_player.entity, "SpriteComponent", "ew_nickname")
        if sprite ~= nil and not EntityHasTag(ctx.my_player.entity, "ew_notplayer") then
            EntityRemoveComponent(ctx.my_player.entity, sprite)
        end
        set_camera_free(false)
        if camera_target == nil then
            camera_target = ctx.my_player
        elseif camera_target.entity ~= ctx.my_player.entity then
            local audio = EntityGetFirstComponent(camera_target.entity, "AudioListenerComponent")
            local audio_n = EntityAddComponent2(cam_target.entity, "AudioListenerComponent")
            if audio == nil then
                ComponentSetValue2(audio_n, "z", -60)
            else
                ComponentSetValue2(audio_n, "z", ComponentGetValue2(audio, "z"))
            end
            --local keep_alive = EntityGetFirstComponent(camera_target.entity, "StreamingKeepAliveComponent")
            EntityRemoveComponent(camera_target.entity, audio)
            --if keep_alive ~= nil then
            --    EntityRemoveComponent(camera_target.entity, keep_alive)
            --end
            EntityRemoveComponent(camera_target.entity, inventory_target)
            camera_target = cam_target
        end
        return
    end
    set_camera_free(true)
    if cam_target == nil or not EntityGetIsAlive(cam_target.entity) then
        redo = true
        return
    end
    local my_x, my_y = GameGetCameraPos()
    local mx, my
    local to_x, to_y = EntityGetTransform(cam_target.entity)
    if to_x == nil then
        return
    end
    if GameGetIsGamepadConnected() then
        mx, my = InputGetJoystickAnalogStick(0, 1)
        mx, my = mx * 256 + to_x, my * 256 + to_y
    else
        mx, my = DEBUG_GetMouseWorld()
    end
    local t_x, t_y = to_x + (mx - to_x) / 3 , to_y + (my - to_y) / 3
    local dx, dy = t_x - my_x, t_y - my_y
    local di = dx * dx + dy * dy
    if di > 512 * 512 then
        set_camera_position(to_x, to_y)
    else
        local cos, sin = dx / 512, dy / 512
        local d = math.sqrt(di)
        dx, dy = d * cos, d * sin
        set_camera_position(my_x + dx, my_y + dy)
    end
    if camera_target == nil then
        camera_target = ctx.my_player
    end
    if camera_target.entity ~= cam_target.entity or redo then
        redo = false
        if EntityHasTag(ctx.my_player.entity, "ew_notplayer") then
            if cam_target.entity ~= ctx.my_player.entity then
                spectate.disable_throwing(false, ctx.my_player.entity)
            elseif attached then
                spectate.disable_throwing(true, ctx.my_player.entity)
            end
        end
        if attached then
            local inv_me = EntityGetFirstComponent(ctx.my_player.entity, "InventoryGuiComponent")
            if inv_me ~= nil then
                ComponentSetValue2(inv_me, "mActive", attached)
            end
            attached = false
        end
        if cam_target == ctx.my_player then
            perks_ui(true)
        else
            if not EntityHasTag(ctx.my_player.entity, "ew_notplayer") then
                nickname.add_label(ctx.my_player.entity, ctx.my_player.name, "data/fonts/font_pixel_white.xml", 0.75)
            end
        end
        if ctx.my_player.entity ~= camera_target.entity and inventory_target ~= nil then
            EntityRemoveComponent(camera_target.entity, inventory_target)
        end

        inventory_target = nil
        if ctx.my_player.entity ~= cam_target.entity then
            inventory_target = EntityAddComponent2(cam_target.entity, "InventoryGuiComponent")
        end
        local audio = EntityGetFirstComponent(camera_target.entity, "AudioListenerComponent")
        --local keep_alive = EntityGetFirstComponent(camera_target.entity, "StreamingKeepAliveComponent")
        if audio ~= nil then
            local audio_n = EntityAddComponent2(cam_target.entity, "AudioListenerComponent")
            ComponentSetValue2(audio_n, "z", ComponentGetValue2(audio, "z"))
            EntityRemoveComponent(camera_target.entity, audio)
            --if camera_target.entity ~= ctx.my_player.entity and keep_alive ~= nil and not EntityHasTag(camera_target.entity, "ew_notplayer") then
            --    EntityRemoveComponent(camera_target.entity, keep_alive)
            --end
            --if cam_target.entity ~= ctx.my_player.entity and not EntityHasTag(cam_target, "ew_notplayer") then
            --    EntityAddComponent2(cam_target.entity, "StreamingKeepAliveComponent")
            --end
        end
    end
    camera_target = cam_target
end

local function set_camera_pos()
    if cam_target == nil or re_cam then
        local i = 0
        for peer_id, potential_target in pairs(ctx.players) do
            if cant_spectate(potential_target.entity) then
                goto continue
            end
            i = i + 1
            if i == camera_player then
                cam_target = potential_target
                camera_player = i
                re_cam = false
                camera_player_id = peer_id
                break
            end
            ::continue::
        end
        if cam_target == nil then
            camera_player_id, camera_player = get_me()
            re_cam = true
            set_camera_pos()
        else
            target()
        end
    else
        target()
    end
end

local function update_i()
    local i = 0
    for peer_id, potential_target in pairs(ctx.players) do
        if cant_spectate(potential_target.entity) then
            goto continue
        end
        i = i + 1
        if peer_id == camera_player_id then
            camera_player = i
            re_cam = true
            return
        end
        ::continue::
    end
    camera_player_id, camera_player = get_me()
end

local function number_of_players()
    local i = 0
    for _, potential_target in pairs(ctx.players) do
        if cant_spectate(potential_target.entity) then
            goto continue
        end
        i = i + 1
        ::continue::
    end
    return i
end

local last_len

function spectate.on_world_update()
    if ctx.run_ended then
        return
    end
    if EntityHasTag(ctx.my_player.entity, "ew_notplayer") then
        was_notplayer = true
    elseif was_notplayer then
        was_notplayer = false
        camera_player_id, camera_player = get_me()
        re_cam = true
    end
    if camera_player == nil then
        camera_player_id, camera_player = get_me()
        re_cam = true
    end
    if last_len == nil then
        last_len = number_of_players()
    end
    if last_len ~= number_of_players() or (cam_target ~= nil and not EntityGetIsAlive(cam_target.entity)) then
        update_i()
        last_len = number_of_players()
    end
    if cam_target ~= nil
            and cant_spectate(cam_target.entity) then
        update_i()
        last_len = number_of_players()
    end
    if camera_player == -1 then
        return
    end

    if InputIsKeyJustDown(tonumber(ModSettingGet("quant.ew.rebind_lspectate"))) or InputIsJoystickButtonJustDown(0, 13) then
        camera_player = camera_player - 1
        if camera_player < 1 then
            camera_player = last_len
        end

        has_switched = true
        re_cam = true
    elseif InputIsKeyJustDown(tonumber(ModSettingGet("quant.ew.rebind_rspectate"))) or InputIsJoystickButtonJustDown(0, 14) then
        camera_player = camera_player + 1
        if camera_player > last_len then
            camera_player = 1
        end

        has_switched = true
        re_cam = true
    elseif InputIsKeyJustDown(tonumber(ModSettingGet("quant.ew.rebind_sspectate"))) then
        camera_player_id, camera_player = get_me()
        re_cam = true
    end
    set_camera_pos()
    ctx.spectating_over_peer_id = camera_player_id

    if GameHasFlagRun("ew_flag_notplayer_active") and not has_switched then
        GuiStartFrame(gui)
        GuiZSet(gui, 11)
        local w, h = GuiGetScreenDimensions(gui)
        local text
        if GameGetIsGamepadConnected() then
            text = "Use d-pad-left and d-pad-right keys to spectate over other players."
        else
            text = "Use ',' and '.' keys to spectate over other players. '/' for self"
        end
        local tw, th = GuiGetTextDimensions(gui, text)
        GuiText(gui, w-2-tw, h-1-th, text)
    end
    if cam_target.entity ~= ctx.my_player.entity then
        local inv_spec = EntityGetFirstComponent(cam_target.entity, "InventoryGuiComponent")
        local inv_me = EntityGetFirstComponent(ctx.my_player.entity, "InventoryGuiComponent")
        if inv_spec ~= nil then
            if inv_me == nil then
                if InputIsKeyJustDown(43) or InputIsJoystickButtonJustDown(0, 16) then
                    attached = not ComponentGetValue2(inv_spec, "mActive")
                    ComponentSetValue2(inv_spec, "mActive", attached)
                end
            else
                if ComponentGetValue2(inv_me, "mActive") then
                    attached = not ComponentGetValue2(inv_spec, "mActive")
                    ComponentSetValue2(inv_spec, "mActive", attached)
                end
            end
            spectate.disable_throwing(attached)
        end
        if inv_me ~= nil then
            ComponentSetValue2(inv_me, "mActive", false)
        end
        perks_ui(false)
    elseif GameGetFrameNum() % 10 == 0 then
        for peer_id, data in pairs(ctx.players) do
            if peer_id ~= ctx.my_id then
                local audio = EntityGetFirstComponent(data.entity, "AudioListenerComponent")
                local inv_target = EntityGetFirstComponent(data.entity, "InventoryGuiComponent")
                --local keep_alive = EntityGetFirstComponent(data.entity, "StreamingKeepAliveComponent")
                if audio ~= nil then
                    EntityRemoveComponent(data.entity, audio)
                end
                --if keep_alive ~= nil and not EntityHasTag(data.entity, "ew_notplayer") then
                --    EntityRemoveComponent(data.entity, keep_alive)
                --end
                if inv_target ~= nil then
                    EntityRemoveComponent(data.entity, inv_target)
                end
            end
        end
    end
end

return spectate