Split mod

lignumis-base/scripts/wooden-rocket-silo.lua

@@ -0,0 +1,159 @@
+local RocketSilo = {
+    events = {}
+}
+
+
+-- Logic for provisional rocket silo with fake silo with inventory and custom launch GUI
+
+local function build_gui(player, rocket_silo)
+    local anchor = {
+        gui = defines.relative_gui_type.container_gui,
+        names = { "provisional-rocket-silo-ready" },
+        position = defines.relative_gui_position.right
+    }
+
+    local target_planet_name = { "space-location-name." .. storage.target_planet }
+    local launch_button = { type = "button", name = "provisional-rocket-silo-launch-button", caption = { "lignumis.provisional-rocket-silo-button", target_planet_name } }
+
+    local inventory = player.get_main_inventory()
+    local is_inventory_empty = inventory.is_empty()
+    local is_nauvis_researched = player.force.technologies["planet-discovery-" .. storage.target_planet].researched
+    launch_button.enabled = is_inventory_empty and is_nauvis_researched
+    if not is_inventory_empty then
+        launch_button.tooltip = { "lignumis.provisional-rocket-silo-button-tooltip-inventory" }
+    elseif not is_nauvis_researched then
+        launch_button.tooltip = { "lignumis.provisional-rocket-silo-button-tooltip-target" }
+    end
+
+
+    storage.rocket_silo_guis = storage.rocket_silo_guis or {}
+    storage.rocket_silo_guis[player.index] = {
+        fake_silo = rocket_silo,
+        gui = gui.add(player.gui.relative, {
+            {
+                type = "frame",
+                name = "provisional-rocket-silo-relative-frame",
+                direction = "vertical",
+                anchor = anchor,
+                children = {
+                    { type = "label", style = "frame_title", caption = { "lignumis.provisional-rocket-silo-button", target_planet_name }, ignored_by_interaction = true },
+                    {
+                        type = "frame",
+                        direction = "vertical",
+                        style = "inside_shallow_frame_with_padding",
+                        children = {
+                            {
+                                type = "flow",
+                                direction = "vertical",
+                                style = "inset_frame_container_vertical_flow",
+                                children = {
+                                    { type = "label", caption = { "lignumis.provisional-rocket-silo-description", target_planet_name } },
+                                    launch_button
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        })
+    }
+end
+
+
+local function on_gui_opened(event)
+    local entity = event.entity
+    if not entity or not entity.valid then return end
+    if entity.name ~= "provisional-rocket-silo-ready" then return end
+    local player = game.get_player(event.player_index)
+
+    if player.gui.relative["provisional-rocket-silo-relative-frame"] then
+        player.gui.relative["provisional-rocket-silo-relative-frame"].destroy()
+    end
+
+    if player.controller_type ~= defines.controllers.character then return end
+
+    build_gui(player, entity)
+end
+
+
+local function on_gui_closed(event)
+    local entity = event.entity
+    if not entity or not entity.valid then return end
+    if entity.name ~= "provisional-rocket-silo-ready" then return end
+end
+
+
+local function launch(event)
+    if event.element.name ~= "provisional-rocket-silo-launch-button" then return end
+    local player = game.get_player(event.player_index)
+    local fake_silo = storage.rocket_silo_guis[player.index].fake_silo
+    rocket_content = fake_silo.get_inventory(defines.inventory.chest).get_contents()
+    for _, entry in pairs(storage.rocket_silos) do
+        if entry.fake_silo == fake_silo then
+            entry.player = event.player_index
+            entry.rocket_content = rocket_content
+            entry.cargo_pod = entry.real_silo.rocket.attached_cargo_pod
+            fake_silo.destroy()
+            entry.real_silo.launch_rocket()
+            entry.cargo_pod.set_passenger(player)
+            break
+        end
+    end
+end
+
+
+local function on_silo_mined(event)
+    if event.entity.name ~= "provisional-rocket-silo-ready" then return end
+    local fake_silo = event.entity
+    local player = game.get_player(event.player_index)
+    for i, entry in pairs(storage.rocket_silos) do
+        if entry.fake_silo == fake_silo then
+            if entry.real_silo.valid then
+                player.mine_entity(entry.real_silo, true)
+            end
+            table.remove(storage.rocket_silos, i)
+            break
+        end
+    end
+end
+
+
+RocketSilo.on_init = function()
+    storage.rocket_silos = {}
+    storage.rocket_silo_guis = {}
+end
+
+
+RocketSilo.events[defines.events.on_script_trigger_effect] = function(event)
+    if event.effect_id ~= "provisional-rocket-ready" then return end
+
+    local lignumis = game.planets["lignumis"].surface
+
+    if event.surface_index ~= lignumis.index then return end
+
+    local real_silo = event.cause_entity
+    local fake_silo = real_silo.surface.create_entity({
+        name = "provisional-rocket-silo-ready",
+        position = real_silo.position,
+        force = real_silo.force,
+        create_build_effect_smoke = false,
+    })
+    storage.rocket_silos = storage.rocket_silos or {}
+    table.insert(storage.rocket_silos, {
+        real_silo = real_silo,
+        fake_silo = fake_silo
+    })
+    for _, player in pairs(real_silo.force.players) do
+        if player.selected == real_silo then
+            player.selected = fake_silo
+        end
+    end
+end
+
+RocketSilo.events[defines.events.on_gui_opened] = on_gui_opened
+RocketSilo.events[defines.events.on_gui_closed] = on_gui_closed
+RocketSilo.events[defines.events.on_gui_click] = launch
+RocketSilo.events[defines.events.on_player_mined_entity] = on_silo_mined
+
+
+return RocketSilo