factorio-scenario-ExpCluster/modules/gui/autofill.lua

--[[-- Gui Module - Autofill
    - Adds a button to enable Autofill
    @gui Autofill
    @alias autofill
]]

local Game = require 'utils.game' --- @dep utils.game
local Gui = require 'expcore.gui' --- @dep expcore.gui
local Global = require 'utils.global' --- @dep utils.global
local config = require 'config.gui.autofill' --- @dep config.gui.autofill
local Event = require 'utils.event' --- @dep utils.event

local print_text = Game.print_floating_text -- (surface, position, text, color)

--- Table that stores if autofill is enabled or not
local autofill_player_settings = {}
Global.register(autofill_player_settings, function(tbl)
    autofill_player_settings = tbl
end)

local autofill_container

--- Draw a section header and main scroll
-- @element autofill_section_container
local section =
Gui.element(function(_, parent, section_name, table_size)
	-- Draw the header for the section
    Gui.header(
        parent,
        {'autofill.'..section_name..'-caption'},
        {'autofill.'..section_name..'-tooltip'},
		true,
		section_name..'-header'
    )
    -- Table used to display the settings
    local scroll_table = Gui.scroll_table(parent, 215, table_size, section_name..'-scroll-table')
    return scroll_table
end)

local toggle_item_button =
Gui.element(function(event_trigger, parent, item_name)
    return parent.add{
        name = event_trigger,
        type = 'sprite-button',
        sprite = 'item/'..item_name,
        tooltip = {'autofill.toggle-tooltip'},
        style = 'shortcut_bar_button_red'
    }
end)
:style(Gui.sprite_style(32, nil, { right_margin = -3 }))
:on_click(function(player, element)
    for _, setting in pairs(autofill_player_settings[player.name]) do
        if 'item/'..setting.item == element.sprite then
            if setting.enabled then
                setting.enabled = false
                element.style = 'shortcut_bar_button_red'
            else
                setting.enabled = true
                element.style = 'shortcut_bar_button_green'
            end
        end
    end
end)

local amount_textfield =
Gui.element(function(event_trigger, parent, amount)
    return parent.add{
        name = event_trigger,
        type = 'textfield',
        text = amount,
        tooltip = {'autofill.amount-tooltip'},
        clear_and_focus_on_right_click = true
    }
end)
:style{
    maximal_width = 90,
    height = 32,
    padding = -2
}
:on_confirmed(function(player, element, _)
    local parent_name = element.parent.name
    for _, setting in pairs(autofill_player_settings[player.name]) do
        if 'amount-setting-'..setting.item == parent_name then
            setting.amount = tonumber(element.text)
            player.print({'autofill.confirmed', setting.amount, '[img=item/'..setting.item..']'})
        end
    end
end)

local add_autofill_setting =
Gui.element(function(_, parent, item_name, amount)
    local toggle_flow = parent.add{ type = 'flow', name = 'toggle-setting-'..item_name }
    local amount_flow = parent.add{ type = 'flow', name = 'amount-setting-'..item_name }
    toggle_flow.style.padding = 0
    amount_flow.style.padding = 0
    toggle_item_button(toggle_flow, item_name)
    amount_textfield(amount_flow, amount)
end)

--- Main gui container for the left flow
-- @element autofill_container
autofill_container =
Gui.element(function(event_trigger, parent)
    -- Draw the internal container
    local container = Gui.container(parent, event_trigger, 100)
    -- Draw the header
    local ammo_table = section(container, 'ammo', 2)
    local fuel_table = section(container, 'fuel', 2)

    for _, setting in pairs(config.default_settings) do
        if setting.type == 'ammo' then
            add_autofill_setting(ammo_table, setting.item, setting.amount)
        elseif  setting.type == 'fuel' then
            add_autofill_setting(fuel_table, setting.item, setting.amount)
        end
    end

    -- Return the external container
    return container.parent
end)
:add_to_left_flow()

--- Button on the top flow used to toggle autofill container
-- @element autofill_toggle
Gui.left_toolbar_button(config.icon, {'autofill.main-tooltip'}, autofill_container)

--- When a player is created make sure they have the default autofill settings
Event.add(defines.events.on_player_created, function(event)
    local player = game.players[event.player_index]
    if not autofill_player_settings[player.name] then
        autofill_player_settings[player.name] = config.default_settings
    end
end)

local function inventories_contains_inventory(inventories, inventory)
    for _, inv in pairs(inventories) do
        if inv == inventory then
            return true
        end
    end
    return false
end

local function entity_build(event)
    -- Check if player exists
    local player = game.players[event.player_index]
    if not player then
        return
    end
    -- Check if the entity is in the config and enabled
    local entity = event.created_entity
    local entity_configs = config.entities[entity.name]
    if not entity_configs then
        return
    end

    -- Loop over each entity config and try to furfill the request amount
    for _,entity_config in pairs(entity_configs) do
        if not entity_config.enabled then
            break
        end
        local player_inventory = player.get_main_inventory()
        local entity_inventory = entity.get_inventory(entity_config.inventory)
        for _, setting in pairs(autofill_player_settings[player.name]) do
            if not setting.enabled or not inventories_contains_inventory(setting.inventories, entity_config.inventory) then
                goto continue
            end
            local item = setting.item
            local preferd_amount = setting.amount
            local item_amount = player_inventory.get_item_count(item)
            if item_amount ~= 0 then
                local inserted
                if item_amount >= preferd_amount then
                    if not entity_inventory.can_insert({name=item, count=preferd_amount}) then
                        goto continue
                    end
                    inserted = entity_inventory.insert({name=item, count=preferd_amount})
                    player_inventory.remove({name=item, count=inserted})
                    print_text(entity.surface, entity.position, {'autofill.filled', '[img=entity/'..entity.name..']', inserted, '[img=item/'..item..']' }, { r = 0, g = 255, b = 0, a = 1})
                else
                    inserted = entity_inventory.insert({name=item, count=item_amount})
                    player_inventory.remove({name=item, count=inserted})
                    print_text(entity.surface, entity.position, {'autofill.filled', '[img=entity/'..entity.name..']', inserted, '[img=item/'..item..']' }, { r = 255, g = 165, b = 0, a = 1})
                end
                goto continue
            end
            ::continue::
        end
    end
end

Event.add(defines.events.on_built_entity, entity_build)