-- launch fish into space to win the game -- by mewmew

local Event = require 'utils.event'

local function goals()
    if not global.catplanet_goals then
        global.catplanet_goals = {
            {goal = 0, rank = false, achieved = true},
            {
                goal = 100,
                rank = 'Copper',
                color = {r = 201, g = 133, b = 6},
                msg = 'You have saved the first container of fish!',
                msg2 = 'However, this is only the beginning.',
                achieved = false
            },
            {
                goal = 1000,
                rank = 'Bronze',
                color = {r = 186, g = 115, b = 39},
                msg = 'Thankful for the fish, they sent back a toy mouse made of solid bronze!',
                msg2 = 'They are demanding more.',
                achieved = false
            },
            {
                goal = 10000,
                rank = 'Silver',
                color = {r = 186, g = 178, b = 171},
                msg = 'In gratitude for the fish, they left you a silver furball!',
                msg2 = 'They are still longing for more.',
                achieved = false
            },
            {
                goal = 25000,
                rank = 'Gold',
                color = {r = 255, g = 214, b = 33},
                msg = 'Pleased about the delivery, they sent back a golden audiotape with cat purrs.',
                msg2 = 'They still demand more.',
                achieved = false
            },
            {
                goal = 50000,
                rank = 'Platinum',
                color = {r = 224, g = 223, b = 215},
                msg = 'To express their infinite love, they sent back a yarnball made of shiny material.',
                msg2 = 'Defying all logic, they still demand more fish.',
                achieved = false
            },
            {
                goal = 100000,
                rank = 'Diamond',
                color = {r = 237, g = 236, b = 232},
                msg = 'A box arrives with a mewing kitten, it a has a diamond collar.',
                msg2 = 'More fish? Why? What..',
                achieved = false
            },
            {
                goal = 250000,
                rank = 'Anti-matter',
                color = {r = 100, g = 100, b = 245},
                msg = 'The obese cat colapses and forms a black hole!',
                msg2 = ':obese:',
                achieved = false
            },
            {
                goal = 500000,
                rank = 'Black Hole',
                color = {r = 100, g = 100, b = 245},
                msg = 'A letter arrives, it reads: Go to bed hooman!',
                msg2 = 'Not yet...',
                achieved = false
            },
            {
                goal = 1000000,
                rank = 'Blue Screen',
                color = {r = 100, g = 100, b = 245},
                msg = 'Cat error #4721',
                msg2 = '....',
                achieved = false
            },
            {goal = 10000000, rank = 'Blue Screen', color = {r = 100, g = 100, b = 245}, msg = '....', msg2 = '....', achieved = false}
        }
    end
end

local function get_rank()
    if not global.catplanet_goals then
        goals()
    end
    for i = #global.catplanet_goals, 1, -1 do
        if global.fish_in_space >= global.catplanet_goals[i].goal then
            return i
        end
    end
end

local function fish_in_space_toggle_button(player)
    if player.gui.top['fish_in_space_toggle'] then
        return
    end
    local button = player.gui.top.add {name = 'fish_in_space_toggle', type = 'sprite-button', sprite = 'item/raw-fish', tooltip = 'Fish in Space'}
    button.style.font = 'default-bold'
    button.style.minimal_height = 38
    button.style.minimal_width = 38
    button.style.top_padding = 2
    button.style.left_padding = 4
    button.style.right_padding = 4
    button.style.bottom_padding = 2
end

local function level_up_popup(player)
    local reward = global.catplanet_goals[get_rank()]
    if player.gui.center['level_up_popup'] then
        player.gui.center['level_up_popup'].destroy()
    end
    local frame = player.gui.center.add({type = 'frame', name = 'level_up_popup', direction = 'vertical'})
    local label = frame.add({type = 'label', caption = reward.msg})
    label.style.font = 'default-listbox'
    label.style.font_color = reward.color
    local button = frame.add({type = 'button', caption = reward.msg2, name = 'level_up_popup_close'})
    button.style.minimal_width = string.len(reward.msg) * 7
    button.style.font = 'default-listbox'
    button.style.font_color = {r = 0.77, g = 0.77, b = 0.77}
end

local function fish_in_space_gui(player)
    if global.fish_in_space == 0 then
        return
    end
    local i = get_rank()

    fish_in_space_toggle_button(player)

    if player.gui.left['fish_in_space'] then
        player.gui.left['fish_in_space'].destroy()
    end

    local frame = player.gui.left.add({type = 'frame', name = 'fish_in_space'})
    local label = frame.add({type = 'label', caption = 'Fish rescued: '})
    label.style.font_color = {r = 0.11, g = 0.8, b = 0.44}

    local progress = global.fish_in_space / global.catplanet_goals[i + 1].goal
    if progress > 1 then
        progress = 1
    end
    local progressbar = frame.add({type = 'progressbar', value = progress})
    progressbar.style.minimal_width = 100
    progressbar.style.maximal_width = 100
    progressbar.style.top_padding = 10

    label = frame.add({type = 'label', caption = global.fish_in_space .. '/' .. tostring(global.catplanet_goals[i + 1].goal)})
    label.style.font_color = {r = 0.33, g = 0.66, b = 0.9}

    if global.catplanet_goals[i].rank then
        label = frame.add({type = 'label', caption = '  ~Rank~'})
        label.style.font_color = {r = 0.75, g = 0.75, b = 0.75}
        label = frame.add({type = 'label', caption = global.catplanet_goals[i].rank})
        label.style.font = 'default-bold'
        label.style.font_color = global.catplanet_goals[i].color
    end
end

local function fireworks(entity)
    for x = entity.position.x - 32, entity.position.x + 32, 1 do
        for y = entity.position.y - 32, entity.position.y + 32, 1 do
            if math.random(1, 150) == 1 then
                entity.surface.create_entity({name = 'big-explosion', position = {x = x, y = y}})
            end
            if math.random(1, 150) == 1 then
                entity.surface.create_entity({name = 'uranium-cannon-shell-explosion', position = {x = x, y = y}})
            end
            if math.random(1, 150) == 1 then
                entity.surface.create_entity({name = 'blood-explosion-huge', position = {x = x, y = y}})
            end
            if math.random(1, 150) == 1 then
                entity.surface.create_entity({name = 'big-artillery-explosion', position = {x = x, y = y}})
            end
        end
    end
end

local function on_rocket_launched(event)
    local rocket_inventory = event.rocket.get_inventory(defines.inventory.rocket)
    local launched_fish_count = rocket_inventory.get_item_count('raw-fish')
    if launched_fish_count == 0 then
        return
    end
    global.fish_in_space = global.fish_in_space + launched_fish_count

    local i = get_rank()

    for _, player in pairs(game.connected_players) do
        fish_in_space_gui(player)
    end

    if not global.catplanet_goals[i].achieved then
        for _, player in pairs(game.connected_players) do
            player.play_sound {path = 'utility/game_won', volume_modifier = 0.9}
            level_up_popup(player)
        end
        global.catplanet_goals[i].achieved = true
        fireworks(event.rocket_silo)
    end
end

local function init()
    global.fish_in_space = 0
end

local function on_player_joined_game(event)
    if not global.fish_in_space then
        init()
    end
    local player = game.players[event.player_index]
    fish_in_space_gui(player)
end

local function on_gui_click(event)
    if not event then
        return
    end
    if not event.element then
        return
    end
    if not event.element.valid then
        return
    end
    local player = game.players[event.element.player_index]
    local name = event.element.name

    if name == 'fish_in_space_toggle' then
        local frame = player.gui.left['fish_in_space']
        if frame then
            frame.destroy()
        else
            fish_in_space_gui(player)
        end
    end

    if name == 'level_up_popup_close' then
        player.gui.center['level_up_popup'].destroy()
    end
end

local function on_init()
    global.fish_autolaunch = true
    goals()
    global.rocket_silos = {}
end

local function tick()
    if not global.fish_autolaunch then
        return
    end
    if game.tick % 6000 == 0 then
        local found_silos = {}
        for _, surface in pairs(game.surfaces) do
            local objects = surface.find_entities_filtered {name = 'rocket-silo'}
            for _, object in pairs(objects) do
                table.insert(found_silos, object)
            end
        end
        global.rocket_silos = found_silos
    end
    for index, silo in pairs(global.rocket_silos) do
        if silo.valid and silo.name == 'rocket-silo' then
            local rocket_inventory = silo.get_inventory(defines.inventory.rocket_silo_rocket)
            local fish
            if rocket_inventory and rocket_inventory.valid then
                fish = rocket_inventory[1]
            end
            if fish and fish.valid_for_read and fish.count == 100 and fish.name == 'raw-fish' then
                silo.launch_rocket()
            end
        else
            global.rocket_silos[index] = nil
        end
    end
end

Event.on_nth_tick(60, tick)
Event.on_init(on_init)
Event.add(defines.events.on_gui_click, on_gui_click)
Event.add(defines.events.on_player_joined_game, on_player_joined_game)
Event.add(defines.events.on_rocket_launched, on_rocket_launched)
