Factorio Forums

Code: Select all

wallNames = {"hybrid-wall-level1",
			 "hybrid-wall-level2",
			 "hybrid-wall-level3"}
			 
gateNames = {"hybrid-gate-level1",
			 "hybrid-gate-level2",
			 "hybrid-gate-level3"}
			 
game.oninit(defines.events.ontick, function(event) OnInit() end)
game.onload(defines.events.ontick, function(event) OnInit() end)

game.onevent(defines.events.onrobotbuiltentity , function(event) OnBuilt(event.createdentity) end)
game.onevent(defines.events.onbuiltentity, function(event) OnBuilt(event.createdentity) end)


function OnInit()
	if not glob.wallLevel then
		glob.wallLevel = 1
	end
end

function OnBuilt(entity)
	-- We have to consider different names now, with the levels. So use string.find to check if it's a wall/gate.
    if string.find(entity.name, "hybrid%-wall") or string.find(entity.name, "hybrid%-gate") then
    	-- If we're at level 2 we'll need to upgade.
		local newWall = UpgradeWallSection(entity)
		-- Store it in the global list for later healing and possible levelling.
    	table.insert(glob.alienwall, newWall)
    end
end

function UpgradeWallSection(wall)
	-- Get the current health percentage. We don't want all the new wall sections to be at max health if the old ones weren't.
	local healthPercent = wall.health / game.entityprototypes[wall.name].maxhealth
	local newWall = false
	local newHealth = 0
	-- Is the current thing a gate?
	if string.find(wall.name, "gate") ~= nil then
		-- Create a gate.
		newWall = game.createentity{name = gateNames[glob.wallLevel], position = wall.position}
		-- Set the health of the new level gate.
		newWall.health = game.entityprototypes[gateNames[glob.wallLevel]].maxhealth * healthPercent
	else
		-- Create a wall section.
		newWall = game.createentity{name = wallNames[glob.wallLevel], position = wall.position}
		-- Set the health of the new level wall.
		newWall.health = game.entityprototypes[wallNames[glob.wallLevel]].maxhealth * healthPercent
	end
	
	-- Destroy the old wall.
	wall.destroy()
	
	return newWall
end

function UpgradeMaxHealth()
	local newWalls = {}
	-- Increment the level.
	glob.wallLevel = glob.wallLevel + 1
	-- Replace each wall section (or gate) with one of the newer wall/gate level.
	for _, wall in pairs(glob.alienwall) do
		-- If it's dead we don't need to upgrade.
		-- No need to remove it either, as the current list will be obsolete soon.
		if wall.valid then
			local newWall = UpgradeWallSection(wall)
			-- Insert into the list of new wall sections.
			table.insert(newWalls, newWall)
		end
	end
	-- The old list is obsolete now, simply overwrite.
	glob.alienwall = newWalls
end