local Event = require("modules/exp_legacy/utils/event")
local general = require("modules.exp_legacy.modules.graftorio.general")
local config = require("modules.exp_legacy.config.graftorio")

local lib = {}

lib.collect_production = function()
    for _, force in pairs(game.forces) do
        general.data.output[force.name].production = {}
        for _, surface in pairs(game.surfaces) do
            --- @class ItemStats
            --- @field count number

            --- @class ProductionStatistics
            --- @field item_input table<string, ItemStats>
            --- @field item_output table<string, ItemStats>
            --- @field fluid_input table<string, ItemStats>
            --- @field fluid_output table<string, ItemStats>
            --- @field kill_input table<string, ItemStats>
            --- @field kill_output table<string, ItemStats>
            --- @field build_input table<string, ItemStats>
            --- @field build_output table<string, ItemStats>
            local stats = {
                item_input = {},
                item_output = {},
                fluid_input = {},
                fluid_output = {},
                kill_input = {},
                kill_output = {},
                build_input = {},
                build_output = {},
            }

            for name, count in pairs(force.get_item_production_statistics(surface).input_counts) do
                local itemstats = stats.item_input[name] or {}
                itemstats.count = count
                stats.item_input[name] = itemstats
            end

            for name, count in pairs(force.get_item_production_statistics(surface).output_counts) do
                local itemstats = stats.item_output[name] or {}
                itemstats.count = count
                stats.item_output[name] = itemstats
            end

            for name, count in pairs(force.get_fluid_production_statistics(surface).input_counts) do
                local fluidstats = stats.fluid_input[name] or {}
                fluidstats.count = count
                stats.fluid_input[name] = fluidstats
            end

            for name, count in pairs(force.get_fluid_production_statistics(surface).output_counts) do
                local fluidstats = stats.fluid_output[name] or {}
                fluidstats.count = count
                stats.fluid_output[name] = fluidstats
            end

            for name, count in pairs(force.get_kill_count_statistics(surface).input_counts) do
                local killstats = stats.kill_input[name] or {}
                killstats.count = count
                stats.kill_input[name] = killstats
            end

            for name, count in pairs(force.get_kill_count_statistics(surface).output_counts) do
                local killstats = stats.kill_output[name] or {}
                killstats.count = count
                stats.kill_output[name] = killstats
            end

            for name, count in pairs(force.get_entity_build_count_statistics(surface).input_counts) do
                local buildstats = stats.build_input[name] or {}
                buildstats.count = count
                stats.build_input[name] = buildstats
            end

            for name, count in pairs(force.get_entity_build_count_statistics(surface).output_counts) do
                local buildstats = stats.build_output[name] or {}
                buildstats.count = count
                stats.build_output[name] = buildstats
            end

            general.data.output[force.name].production[surface.name] = stats
        end
    end
end

lib.collect_loginet = function()
    for _, force in pairs(game.forces) do
        --- @class RobotStatistics
        --- @field all_construction_robots uint
        --- @field available_construction_robot uint
        --- @field all_logistic_robots uint
        --- @field available_logistic_robots uint
        --- @field charging_robot_count uint
        --- @field to_charge_robot_count uint
        --- @field items table<string, uint>
        --- @field pickups table<string, uint>
        --- @field deliveries table<string, uint>
        local stats = {
            all_construction_robots = 0,
            available_construction_robots = 0,

            all_logistic_robots = 0,
            available_logistic_robots = 0,

            charging_robot_count = 0,
            to_charge_robot_count = 0,

            items = {},
            pickups = {},
            deliveries = {},
        }
        for _, networks in pairs(force.logistic_networks) do
            for _, network in pairs(networks) do
                stats.available_construction_robots = network.available_construction_robots
                stats.all_construction_robots = network.all_construction_robots

                stats.available_logistic_robots = network.available_logistic_robots
                stats.all_logistic_robots = network.all_logistic_robots

                stats.charging_robot_count = 0
                stats.to_charge_robot_count = 0
                for _, cell in pairs(network.cells) do
                    stats.charging_robot_count = (stats.charging_robot_count) + cell.charging_robot_count
                    stats.to_charge_robot_count = (stats.to_charge_robot_count) + cell.to_charge_robot_count
                end

                if config.modules.logistorage then
                    for name, v in pairs(network.get_contents()) do
                        stats.items[name] = (stats.items[name] or 0) + v
                    end

                    -- pickups and deliveries of items
                    for _, point_list in pairs{ network.provider_points, network.requester_points, network.storage_points } do
                        for _, point in pairs(point_list) do
                            for name, qty in pairs(point.targeted_items_pickup) do
                                stats.pickups[name] = (stats.pickups[name] or 0) + qty
                            end

                            for name, qty in pairs(point.targeted_items_deliver) do
                                stats.deliveries[name] = (stats.deliveries[name] or 0) + qty
                            end
                        end
                    end
                end
            end
        end

        general.data.output[force.name].robots = stats
    end
end

--- @class Research
--- @field name string
--- @field level uint
--- @field progress double

Event.add(defines.events.on_research_finished, function(evt)
    local research = evt.research
    if not general.data.output[research.force.name] then general.data.output[research.force.name] = {} end
    if not general.data.output[research.force.name].research then general.data.output[research.force.name].research = {} end

    local force_research = general.data.output[research.force.name].research or {}
    table.remove(force_research, 1)
    general.data.output[research.force.name].research = force_research
end)

Event.add(defines.events.on_research_started, function(evt)
    -- move queue up
    local research = evt.research
    if not general.data.output[research.force.name].research then general.data.output[research.force.name].research = {} end

    local force_research = general.data.output[research.force.name].research or {}
    table.remove(force_research, 1)
    general.data.output[research.force.name].research = force_research
end)

Event.on_nth_tick(60, function()
    for _, force in pairs(game.forces) do
        if not general.data.output[force.name].research then general.data.output[force.name].research = {} end

        local force_research = {}
        -- this works even if the queue is disabled, but it will always be just 1 long in that case
        for _, research in pairs(force.research_queue) do
            table.insert(force_research, {
                name = research.name,
                level = research.level,
                progress = force.technologies[research.name].saved_progress
            })
        end

        general.data.output[force.name].research = force_research
    end
end)

return lib