--[[-- Commands - Waterfill
Adds a command that places shallow water
]]

local AABB = require("modules/exp_util/aabb")
local Commands = require("modules/exp_commands")
local Selection = require("modules.exp_legacy.modules.control.selection") --- @dep modules.control.selection
local SelectionName = "ExpCommand_Waterfill"

--- Toggle player selection mode for artillery
Commands.new("waterfill", { "exp-commands_waterfill.description" })
    :register(function(player)
        if Selection.is_selecting(player, SelectionName) then
            Selection.stop(player)
            return Commands.status.success{ "exp-commands_waterfill.exit" }
        else
            local item_count_cliff = player.get_item_count("cliff-explosives")
            local item_count_craft = math.min(math.floor(player.get_item_count("explosives") / 10), player.get_item_count("barrel"), player.get_item_count("grenade"))
            local item_count_total = item_count_cliff + item_count_craft
            if item_count_total == 0 then
                return player.print{ "exp-commands_waterfill.requires-explosives" }
            else
                Selection.start(player, SelectionName)
                return player.print{ "exp-commands_waterfill.enter" }
            end
        end
    end)

--- When an area is selected to be converted to water
Selection.on_selection(SelectionName, function(event)
    --- @cast event EventData.on_player_selected_area
    local area = AABB.expand(event.area)
    local player = game.players[event.player_index]
    local surface = event.surface

    if surface.planet and surface.planet ~= game.planets.nauvis then
        player.print({ "exp-commands_waterfill.nauvis-only" }, Commands.print_settings.error)
        return
    end

    local area_size = (area.right_bottom.x - area.left_top.x) * (area.right_bottom.y - area.left_top.y)
    if area_size > 1000 then
        player.print({ "exp-commands_waterfill.area-too-large", 1000, area_size }, Commands.print_settings.error)
        return
    end

    local item_count_cliff = player.get_item_count("cliff-explosives")
    local item_count_craft = math.min(math.floor(player.get_item_count("explosives") / 10), player.get_item_count("barrel"), player.get_item_count("grenade"))
    local item_count_total = item_count_cliff + item_count_craft

    if item_count_total < area_size then
        player.print({ "exp-commands_waterfill.too-few-explosives", area_size, item_count_total }, Commands.print_settings.error)
        return
    end

    local tile_count = 0
    local tiles_to_make = {}
    for x = area.left_top.x, area.right_bottom.x do
        for y = area.left_top.y, area.right_bottom.y do
            tile_count = tile_count + 1
            tiles_to_make[tile_count] = {
                name = "water-mud",
                position = { x, y },
            }
        end
    end

    surface.set_tiles(tiles_to_make, true, "abort_on_collision", true, false, player, 0)
    local remaining_tiles = surface.count_tiles_filtered{ area = area, name = "water-mud" }
    local t_diff = tile_count - remaining_tiles

    if item_count_cliff >= t_diff then
        player.remove_item{ name = "cliff-explosives", count = t_diff }
    else
        if item_count_cliff > 0 then
            player.remove_item{ name = "cliff-explosives", count = item_count_cliff }
        end
        local item_count_needed = t_diff - item_count_cliff
        if item_count_needed > 0 then
            player.remove_item{ name = "explosives", count = 10 * item_count_needed }
            player.remove_item{ name = "barrel", count = item_count_needed }
            player.remove_item{ name = "grenade", count = item_count_needed }
        end
    end

    if remaining_tiles > 0 then
        player.print({ "exp-commands_waterfill.part-complete", tile_count, remaining_tiles }, Commands.print_settings.default)
    else
        player.print({ "exp-commands_waterfill.complete", tile_count }, Commands.print_settings.default)
    end
end)
