control.lua

local CityPlanning = require("city-planner")
local RecipeCalculator = require("recipe-calculator")
local GuiEventHandler = require("gui-event-handler")
local ShortcutHandler = require("shortcut-handler")
local ResearchHandler = require("research-event-handler")
local OnPlayerEventHandler = require("player-event-handler")
local OnChunkChartedHandler = require("chunk-charted-handler")
local OnConstructionHandler = require("construction-event-handler")
local SurfaceEventHandler = require("surface-event-handler")
local Passengers = require("passengers")
local ChatMessages = require("chat-messages")
local Consumption = require("consumption")
local City = require("city")
local Queue = require("queue")
local PrimaryIndustries = require("primary-industries")
local UsedBottlesStore = require("used-bottles-store")
local FloorUpgradesQueue = require("floor-upgrades-queue")
local Constants = require("constants")

--- TICK HANDLERS
local ONE_SECOND = 60;
local FIVE_SECONDS = 5 * ONE_SECOND;
local THIRTY_SECONDS = 30 * ONE_SECOND;
local ONE_MINUTE = 60 * ONE_SECOND;

script.on_nth_tick(ONE_SECOND, function()
    CityPlanning.build_initial_city()
    PrimaryIndustries.spawn_initial_industry()
    City.start_house_construction()
end)

local function handle_passengers()
    for _, city in ipairs(global.tycoon_cities or {}) do
        Passengers.clearPassengers(city)
        Passengers.spawnPassengers(city)
    end
end

local function expand_roads()
    for _, city in ipairs(global.tycoon_cities or {}) do
        if Queue.count(city.buildingLocationQueue) < #city.grid then
            local coordinates = City.growAtRandomRoadEnd(city)
            if coordinates ~= nil then
                City.updatepossibleBuildingLocations(city, coordinates)
            end
        end
    end
end

script.on_nth_tick(FIVE_SECONDS, function()
    expand_roads()
    City.complete_house_construction()
    handle_passengers()
    FloorUpgradesQueue.process()
end)

local function add_more_cities()
    if #(global.tycoon_cities or {}) > 0 then
        local global_citizen_count = 0
        for _, city in pairs(global.tycoon_cities) do
            for _, n in pairs(city.citizens) do
                global_citizen_count = global_citizen_count + n
            end
        end

        local threshold = #(global.tycoon_cities or {}) * Constants.NEW_CITY_THRESHOLD
        if global_citizen_count > threshold then
            CityPlanning.addMoreCities(false)
        end
    end
end

script.on_nth_tick(THIRTY_SECONDS, function()
    City.construct_priority_buildings()
    -- todo: implement me later
    -- rediscover_unused_fields()
    add_more_cities()
    CityPlanning.tag_cities()
end)

local function display_intro_messages()
    ChatMessages.show_info_messages()
end

local function consume_resources()
    for _, city in ipairs(global.tycoon_cities or {}) do
        Consumption.consumeBasicNeeds(city)
        Consumption.consumeAdditionalNeeds(city)
        Consumption.update_construction_timers_all(city)
    end
end

script.on_nth_tick(ONE_MINUTE, function()
    for _, city in pairs(global.tycoon_cities or {}) do
        UsedBottlesStore.return_used_bottles(city)
    end
    consume_resources()
    City.construct_gardens()
    display_intro_messages()
end)

--- EVENT HANDLERS
script.on_event({
    defines.events.on_built_entity,
    defines.events.on_robot_built_entity
}, function(event)
    OnConstructionHandler.on_built(event)
end)

script.on_event({
    defines.events.on_player_mined_entity,
    defines.events.on_robot_mined_entity,
    -- Register entities with script.register_on_entity_destroyed(entity) so that this event fires.
    defines.events.on_entity_destroyed,
}, function(event)
    OnConstructionHandler.on_removed(event)
end)

script.on_event(defines.events.on_chunk_generated, function (event)
    OnChunkChartedHandler.on_chunk_generated(event)
end)

script.on_event(defines.events.on_chunk_charted, function (event)
    OnChunkChartedHandler.on_chunk_charted(event)
end)

script.on_event(defines.events.on_chunk_deleted, function (event)
    OnChunkChartedHandler.on_chunk_deleted(event)
end)

script.on_event(defines.events.on_player_cursor_stack_changed, function(event)
    OnPlayerEventHandler.on_player_cursor_stack_changed(event)
end)

script.on_event(defines.events.on_research_finished, function(event)
    ResearchHandler.on_research_finished(event)
end)

script.on_event({defines.events.on_lua_shortcut, "tycoon-cities-overview"}, function(event)
    ShortcutHandler.on_shortcut(event)
end)

script.on_init(function()
    global.tycoon_global_generator = game.create_random_generator()
    global.tycoon_city_buildings = {}
end)

script.on_event(defines.events.on_gui_opened, function (event)
    GuiEventHandler.on_gui_opened(event)
end)

script.on_event(defines.events.on_gui_text_changed, function(event)
    GuiEventHandler.on_gui_text_changed(event)
end)

script.on_event(defines.events.on_gui_click, function(event)
    GuiEventHandler.on_gui_click(event)
end)

script.on_event(defines.events.on_gui_checked_state_changed, function(event)
    GuiEventHandler.on_gui_checked_state_changed(event)
end)

-- surface events
script.on_event(defines.events.on_surface_cleared, function (event)
    SurfaceEventHandler.on_surface_cleared(event)
end)
script.on_event(defines.events.on_surface_created, function (event)
    SurfaceEventHandler.on_surface_created(event)
end)
script.on_event(defines.events.on_surface_deleted, function (event)
    SurfaceEventHandler.on_surface_deleted(event)
end)
script.on_event(defines.events.on_surface_imported, function (event)
    SurfaceEventHandler.on_surface_imported(event)
end)
script.on_event(defines.events.on_surface_renamed, function (event)
    SurfaceEventHandler.on_surface_renamed(event)
end)

--- REMOTE INTERFACES
remote.add_interface("tycoon", {
    spawn_city = CityPlanning.addCity,
    calculate_recipes = RecipeCalculator.calculateAllRecipes
})