alter.lua

-- increase max cached surfaces
-- for surface dense uis
sdlext.surface_cache_max = 1024

local path = GetParentPath(...)

local vanillaCorporations = { "Corp_Grass", "Corp_Desert", "Corp_Snow", "Corp_Factory" }
local difficulties = { DIFF_EASY, DIFF_NORMAL, DIFF_HARD, DIFF_UNFAIR }

local islands = { "archive", "rst", "pinnacle", "detritus" }
local islandNames = { "Archive", "R.S.T.", "Pinnacle", "Detritus" }
local tileTypes = { TERRAIN_FOREST, TERRAIN_SAND, TERRAIN_ICE, TERRAIN_ACID }
local islandShifts = { Point(14,5), Point(16,15), Point(17,12), Point(18,15), Point(0,0) }

local corporations = { "archive", "rst", "pinnacle", "detritus" }
local ceos = { "dewey", "jessica", "zenith", "vikram" }
local missionLists = corporations
local bossLists = corporations

local tilesets = { "grass", "sand", "snow", "acid", "lava", "volcano", "vine", "hologram", "debris" }
local climate = { "", "", "", "", "Scorching", "Fiery", "Lush", "Holograpic", "debris" }
local rainChance = { 0, 0, 0, 0, 0, 0, 70, 30, 10 }
local crackChance = { 0, 0, 7, 0, 0, 0, 0, 0, 2 }
local environmentChance = { 0, 0, 0, 0, 0, 0, { [TERRAIN_FOREST] = 60 }, 0, 0 }

local mechList = {
	"PunchMech", "TankMech", "ArtiMech",
	"JetMech", "RocketMech", "PulseMech",
	"LaserMech", "ChargeMech", "ScienceMech",
	"ElectricMech", "WallMech", "RockartMech",
	"JudoMech", "DStrikeMech", "GravMech",
	"FlameMech", "IgniteMech", "TeleMech",
	"GuardMech", "MirrorMech", "IceMech",
	"LeapMech", "UnstableTank", "NanoMech",
}

local UNIT_EXCLUSION = {
	"Pawn",
	"PawnTable",

	"SlugBoss",
	"SlugEgg1",
	"Slug1",
	"Slug2",
	"Jelly_Necro1",
}

local WEAPON_EXCLUSION = {
}

local MISSION_EXCLUSION = {
	"Mission",
	"Mission_Test",
	"Mission_Auto",
	"Mission_Battle",
	"Mission_Infinite",
	"Mission_Tutorial",
	"Mission_Critical",
	"Mission_MineBase",
	"Mission_Final",
	"Mission_Final_Cave",
	"Mission_Boss",

	"Mission_Sandstorm",
	"Mission_SlugBoss",
	"Mission_LeaperBoss",
	"Mission_DiggerBoss",
}

local STRUCTURE_EXCLUSION = {

}

local function updateMissingNames()
	Jelly_Armor1.Name = "Shell Psion"
	Jelly_Health1.Name = "Soldier Psion"
	Jelly_Regen1.Name = "Blood Psion"
	Jelly_Explode1.Name = "Blast Psion"
	Jelly_Lava1.Name = "Psion Tyrant"
	Jelly_Spider1.Name = "Arachnid Psion"
	Jelly_Fire1.Name = "Smoldering Psion"
	Jelly_Boost1.Name = "Raging Psion"
	Shaman1.Name = "Plasmodia"
	ShamanBoss.Name = "Plasmodia Leader"
	Totem1.Name = "Spore"
	BlobberBoss.Name = "Blobber Leader"
	Burnbug1.Name = "Gastropod"
	BurnbugBoss.Name = "Gastropod Leader"
	Dung1.Name = "Tumblebug"
	DungBoss.Name = "Tumblebug Leader"

	Mission_BeetleBoss.Name = "Beetle Leader"
	Mission_BotBoss.Name = "Bot Leader"
	Mission_FireflyBoss.Name = "Firefly Leader"
	Mission_BlobBoss.Name = "Large Goo"
	Mission_HornetBoss.Name = "Hornet Leader"
	Mission_JellyBoss.Name = "Psion Abomination"
	Mission_ScorpionBoss.Name = "Scorpion Leader"
	Mission_SpiderBoss.Name = "Spider Leader"
	Mission_BlobberBoss.Name = "Blobber Leader"
	Mission_BouncerBoss.Name = "Bouncer Leader"
	Mission_BurnbugBoss.Name = "Gastropod Leader"
	Mission_CentipedeBoss.Name = "Centipede Leader"
	Mission_CrabBoss.Name = "Crab Leader"
	Mission_DungBoss.Name = "Tumblebug Leader"
	Mission_MosquitoBoss.Name = "Mosquito Leader"
	Mission_ScarabBoss.Name = "Scarab Leader"
	Mission_ShamanBoss.Name = "Plasmodia Leader"
	Mission_StarfishBoss.Name = "Starfish Leader"

	Mission_Acid.Name = "A.C.I.D."
	Mission_AcidStorm.Name = "A.C.I.D. Storm"
	Mission_AcidTank.Name = "A.C.I.D. Tank"
	Mission_Airstrike.Name = "Airstrike"
	Mission_Armored_Train.Name = "Armored Train"
	Mission_Barrels.Name = "A.C.I.D. Vats"
	Mission_Belt.Name = "Conveyor Belt"
	Mission_BeltRandom.Name = "Conveyor Belt Random"
	Mission_Bomb.Name = "Prototype Renfield Bomb"
	Mission_BoomBots.Name = "Boom Bots"
	Mission_BotDefense.Name = "Bot Defense"
	Mission_Cataclysm.Name = "Cataclysm"
	Mission_Civilians.Name = "V.I.P."
	Mission_Crack.Name = "Seismic Activity"
	Mission_Dam.Name = "Dam"
	Mission_Disposal.Name = "Disposal Unit"
	Mission_Factory.Name = "Robot Factory"
	Mission_Fence.Name = "Barrier"
	Mission_Filler.Name = "Earth Mover"
	Mission_Force.Name = "Destruction"
	Mission_ForestFire.Name = "Forest Fire"
	Mission_FreezeBldg.Name = "Freeze Buildings"
	Mission_FreezeBots.Name = "Freeze Tank"
	Mission_FreezeMines.Name = "Cryo-Mines"
	Mission_Hacking.Name = "Hacking"
	Mission_Laser.Name = "Laser"
	Mission_Lightning.Name = "Lightning"
	Mission_Mines.Name = "Mines"
	Mission_Missiles.Name = "Detritus Contraption"
	Mission_NanoStorm.Name = "Nanostorm"
	Mission_Piston.Name = "Trash Compactor"
	Mission_Power.Name = "Power"
	Mission_Reactivation.Name = "Thawing Enemies"
	Mission_Repair.Name = "Repair Platforms"
	Mission_Respawn.Name = "Respawning Enemies"
	Mission_Shields.Name = "Shield Generator"
	Mission_SnowBattle.Name = "Sentient Weapons"
	Mission_SnowStorm.Name = "Snowstorm"
	Mission_Solar.Name = "Solar Farms"
	Mission_Stasis.Name = "Freese Bots"
	Mission_Survive.Name = "Survive"
	Mission_Tanks.Name = "Archive Tanks"
	Mission_Teleporter.Name = "Teleporters"
	Mission_Terraform.Name = "Terraformer"
	Mission_Terratide.Name = "Sandstorm"
	Mission_Tides.Name = "Tidal Waves"
	Mission_Train.Name = "Train"
	Mission_Trapped.Name = "Evacuated Buildings"
	Mission_Volatile.Name = "Volatile Vek"
	Mission_Wind.Name = "Windstorm"

	Str_Bar.Name = "Old Earth Bar"
	Str_Battery.Name = "Emergency Batteries"
	Str_Clinic.Name = "Clinic"
	Str_Combat.Name = "Airfield"
	Str_Nimbus.Name = "Power Generator"
	Str_Power.Name = "Coal Plant"
	Str_Research.Name = "Defense Lab"
	Str_Robotics.Name = "Robotics Lab"
	Str_Shield.Name = "Shield Generator"
	Str_Tower.Name = "Corporate Tower"
end

local function upper_first(str)
	return str:gsub("^.", string.upper)
end

local function registerWeapon(weapon_id)
	local base = _G[weapon_id]

	if false
		or list_contains(WEAPON_EXCLUSION, weapon_id)
		or base == nil
		or base.ExcludeFromEasyEdit == true
	then
		return
	end

	local weapon = easyEdit.weapons:get(weapon_id)

	weapon = weapon or easyEdit.weapons:add(weapon_id)
	weapon:copy(base)
	weapon.__Id = weapon_id
	weapon.Name = GetText(weapon_id.."_Name") or base.Name
	weapon.Description = GetText(weapon_id.."_Description") or base.Description
end

local function registerWeapons()
	for weapon_id, _ in pairs(modApi.weaponDeck) do
		registerWeapon(weapon_id)
	end
end

local units = easyEdit.units
local unitImages = easyEdit.unitImage
local isValidUnit = units._class.isValid
local function registerUnit(unit_id)
	local base = _G[unit_id]

	if false
		or list_contains(UNIT_EXCLUSION, unit_id)
		or base == nil
		or base.ExcludeFromEasyEdit == true
	then
		return
	end

	local unit = units:get(unit_id)

	if base.Name == "Unnamed Pawn" or base.Name == "PawnTable" then
		base.Name = GetText(unit_id)
	end

	if isValidUnit(base) then
		unit = unit or units:add(unit_id, base)
		units:addSoundBase(unit)
	end

	unitImageId = base.Image
	unitImage = unitImages:get(unitImageId)

	if unitImage == nil then
		unitImage = unitImages:add(unitImageId)
		unitImage.Name = unitImageId
		unitImage.Image = base.Image
		unitImage.ImageOffset = base.ImageOffset
	end

	for _, weaponId in ipairs(base.SkillList) do
		registerWeapon(weaponId)
	end
end

local function registerUnits()
	for unit_id, unit in pairs(_G) do
		local isUnit = true
			and type(unit) == 'table'
			and type(unit.Name) == 'string'
			and type(unit.Class) == 'string'
			and type(unit.Image) == 'string'
			and type(unit.ImageOffset) == 'number'
			and type(unit.Health) == 'number'
			and type(unit.MoveSpeed) == 'number'
			and type(unit.SkillList) == 'table'

		if isUnit then
			registerUnit(unit_id)
		end
	end
end

local function appendMissionImage(mission_id, subpath)
	local appendLoc = string.format("img/strategy/%s%s.png", subpath, mission_id)

	if modApi.resource:file_exists(appendLoc) then
		return
	end

	local filename = string.format("%simg/%s%s.png", path, subpath, mission_id)

	if not modApi:fileExists(filename) then
		filename = string.format("%simg/%sIcon_Missing.png", path, subpath)
	end

	modApi:appendAsset(appendLoc, filename)
end

local function appendMissionSmallImage(mission_id)

end

local function registerMission(mission_id)
	local base = _G[mission_id]

	if false
		or list_contains(MISSION_EXCLUSION, mission_id)
		or base == nil
		or base.ExcludeFromEasyEdit == true
	then
		return
	end

	local mission = easyEdit.missions:get(mission_id)

	mission = mission or easyEdit.missions:add(mission_id)
	mission:copy(base)

	appendMissionImage(mission_id, "mission/")
	appendMissionImage(mission_id, "mission/small/")
end

local function registerMissions()
	Mission_Windmill = Mission_Critical:new{
		Name = "Wind Farms",
		Image = "str_wind1",
		FlavorBase = "Mission_Wind",
		Objectives = PowerObjective("Mission_Wind_Objective", 2)
	}
	for i, _ in ipairs(corporations) do
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		local function addMissions(missionTable)
			for _, mission_id in ipairs(missionTable) do
				registerMission(mission_id)
			end
		end

		addMissions(base.Missions_High)
		addMissions(base.Missions_Low)
		addMissions(base.Bosses)
		addMissions(base.UniqueBosses)
	end

	for mission_id, mission in pairs(_G) do
		local isMission = true
			and type(mission) == 'table'
			and type(mission.Name) == 'string'
			and type(mission.MapList) == 'table'
			and type(mission.MapTags) ~= 'nil'
			and type(mission.MapVetoes) == 'table'

		if isMission then
			registerMission(mission_id)
		end
	end
end

local function registerStructure(structure_id)
	local base = _G[structure_id]

	if false
		or list_contains(STRUCTURE_EXCLUSION, structure_id)
		or base == nil
		or base.ExcludeFromEasyEdit == true
	then
		return
	end

	local structure = easyEdit.structures:get(structure_id)

	structure = structure or easyEdit.structures:add(structure_id)
	structure.Name = GetText(structure_id.."_Name")
	structure:copy(base)
end

local function registerStructures()
	for i, _ in ipairs(corporations) do
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		local function addStructures(structureTable)
			for _, structure_id in ipairs(structureTable) do
				registerStructure(structure_id)
			end
		end

		addStructures(base.PowAssets)
		addStructures(base.TechAssets)
		addStructures(base.RepAssets)
	end

	for structure_id, structure in pairs(_G) do
		local isStructure = true
			and type(structure) == 'table'
			and type(structure.Image) == 'string'
			and type(structure.Reward) == 'number'
			and structure.Reward >= 0
			and structure.Reward <= 2
			and structure_id:find("^Mission") == nil
			and structure.nonStructure ~= true

		if isStructure then
			registerStructure(structure_id)
		end
	end
end

local function registerIslands()
	for i, id in ipairs(islands) do
		local island = easyEdit.island:add(id)
		local n = i-1

		island:copyAssets({_id = tostring(n)})
		island.name = islandNames[i]
		island.shift = islandShifts[i]
		island.magic = Island_Magic[i]
		island.regionData = {}
		island.network = {}

		if i <= 4 then
			for k = 0, 7 do
				table.insert(island.regionData, Region_Data[string.format("island_%s_%s", n, k)])
			end

			for k = 0, 7 do
				table.insert(island.network, _G["Network_Island_".. n][tostring(k)])
			end
		end
	end
end

local function registerCorporations()
	for i, id in ipairs(corporations) do
		local corp = easyEdit.corporation:add(id)
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		corp:copy(base)
		base.OldUniqueBosses = base.UniqueBosses
		base.UniqueBosses = {}
		corp.Name = GetText(corp_id.."_Name")
		corp.Description = GetText(corp_id.."_Description")
	end
end

local function registerCEOs()
	for i, id in ipairs(ceos) do
		local ceo = easyEdit.ceo:add(id)
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		ceo:copyAssets({_id = "ceo_"..corporations[i]})
		ceo:copy(base)
		ceo.CEO_Name = GetText(corp_id.."_CEO_Name")
	end
end

local function registerTilesets()
	modApi:appendAsset(
		"img/strategy/corp/lava_env.png",
		path.."img/env/lava_env.png")
	modApi:appendAsset(
		"img/strategy/corp/volcano_env.png",
		path.."img/env/volcano_env.png")
	modApi:appendAsset(
		"img/strategy/corp/vine_env.png",
		path.."img/env/vine_env.png")
	modApi:appendAsset(
		"img/strategy/corp/hologram_env.png",
		path.."img/env/hologram_env.png")
	modApi:appendAsset(
		"img/strategy/corp/debris_env.png",
		path.."img/env/debris_env.png")
	modApi:copyAsset(
		"img/combat/tiles_grass/building_sheet_vines.png",
		"img/combat/tiles_vine/building_sheet.png")

	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/building_1_tile.png",
		"img/combat/tiles_debris/building_1_tile.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/building_2_tile.png",
		"img/combat/tiles_debris/building_2_tile.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/building_3_tile.png",
		"img/combat/tiles_debris/building_3_tile.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/building_collapse.png",
		"img/combat/tiles_debris/building_collapse.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/building_sheet.png",
		"img/combat/tiles_debris/building_sheet.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/ground_0.png",
		"img/combat/tiles_debris/ground_0.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/ground_1.png",
		"img/combat/tiles_debris/ground_1.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/ground_rail.png",
		"img/combat/tiles_debris/ground_rail.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/mountain_0.png",
		"img/combat/tiles_debris/mountain_0.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/mountain_0_broken.png",
		"img/combat/tiles_debris/mountain_0_broken.png")
	modApi:copyAsset(
		"img/advanced/combat/tiles_debris/mountain_explode.png",
		"img/combat/tiles_debris/mountain_explode.png")

	-- temporarily override GetRealDifficulty
	-- while extracting environmentChance for
	-- vanilla tilesets
	local oldGetDifficulty = GetRealDifficulty
	local difficulty
	function GetRealDifficulty()
		return difficulty
	end

	for i, id in ipairs(tilesets) do
		local corp_id = vanillaCorporations[i]

		if corp_id then
			local tileset = easyEdit.tileset:add(id)
			tileset.name = upper_first(id)
			-- fill in missing conveyor assets
			tileset:copyAssets({_id = "acid"}, false)

			tileset.climate = GetText(corp_id.."_Environment")
			tileset.rainChance = getRainChance(id)
			tileset.crackChance = getCrackChance(id)
			tileset.environmentChance = {}

			for _, diff in ipairs(difficulties) do
				difficulty = diff
				tileset.environmentChance[diff] = {}

				for _, tileType in ipairs(tileTypes) do
					tileset.environmentChance[diff][tileType] = getEnvironmentChance(id, tileType)
				end
			end
		else
			local tileset = easyEdit.tileset:add(id)
			tileset.name = upper_first(id)
			-- set missing locations
			tileset:copyAssets(tileset)
			-- fill in missing assets
			tileset:copyAssets({_id = "grass"}, false)
			tileset:copyAssets({_id = "acid"}, false)

			tileset.climate = climate[i]
			tileset.rainChance = rainChance[i]
			tileset.crackChance = crackChance[i]
			tileset.environmentChance = environmentChance[i]
		end
	end

	GetRealDifficulty = oldGetDifficulty

	function getRainChance(sectorType)
		local tileset = easyEdit.tileset:get(sectorType)
		local noDataFound = not tileset or not tileset.getRainChance

		if noDataFound then
			return 0
		end

		return tileset:getRainChance()
	end

	function getCrackChance(sectorType)
		local tileset = easyEdit.tileset:get(sectorType)
		local noDataFound = not tileset or not tileset.getCrackChance

		if noDataFound then
			return 0
		end

		return tileset:getCrackChance()
	end

	function getEnvironmentChance(sectorType, tileType)
		local tileset = easyEdit.tileset:get(sectorType)
		local noDataFound = not tileset or not tileset.getEnvironmentChance

		if noDataFound then
			return 0
		end

		return tileset:getEnvironmentChance(tileType, GetDifficulty())
	end

	modApi.events.onTilesetChanged:subscribe(function(newTileset, oldTileset)
		local oldTileset = easyEdit.tileset:get(oldTileset)
		local newTileset = easyEdit.tileset:get(newTileset)

		oldTileset:onDisabled()
		newTileset:onEnabled()
	end)
end

function pickEnemies(categories, enemies, islandNumber, timesPicked)
	timesPicked = timesPicked or {}
	local result = {}
	local choices = {}
	local excluded = {}

	local function isUnlocked(unit)
		local lock = IslandLocks[unit] or 0
		return islandNumber == nil or islandNumber >= lock or Game:IsIslandUnlocked(lock-1)
	end

	local function addExclusions(unit)
		for i, list in ipairs(exclusiveElements) do
			if list_contains(list, unit) then
				for i, excludedId in ipairs(list) do
					excluded[excludedId] = true
				end
			end
		end
		--Flying Enemies
		if FlyingEnemies[unit] then --If flying exclude all flying
			for excludedId, _ in pairs(FlyingEnemies) do
				excluded[excludedId] = true
			end
		end
	end

	local function getEnemyChoices(category)
		if type(category) ~= 'string' then
			return {}
		end

		if choices[category] and #choices[category] > 0 then
			return choices[category]
		end

		local leastPicked = INT_MAX

		choices[category] = {}
		enemies[category] = enemies[category] or {}
		for _, enemy in ipairs(enemies[category]) do
			if isUnlocked(enemy) and not excluded[enemy] then
				table.insert(choices[category], enemy)
			end
		end

		shuffle_list(choices[category])
		table.sort(choices[category], function(a,b)
			return (timesPicked[a] or 0) > (timesPicked[b] or 0)
		end)

		return choices[category]
	end

	for _, category in ipairs(categories) do
		local enemyChoices = getEnemyChoices(category)
		local choice = "Scorpion"

		for i = #enemyChoices, 1, -1 do
			if not excluded[enemyChoices[i]] then
				choice = enemyChoices[i]
				table.remove(enemyChoices, i)

				break
			end
		end

		timesPicked[choice] = (timesPicked[choice] or 0) + 1
		addExclusions(choice)
		table.insert(result, choice)
	end

	Assert.Equals(6, #result, "Result")

	return result
end

local function getNoNewBosses()
	return {}
end

local function registerEnemyLists()
	for i, id in ipairs(corporations) do
		local enemyList = easyEdit.enemyList:add(id)
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		enemyList.name = base.Bark_Name
		enemyList.enemies = getFinalEnemyLists()
	end

	for i, corp_id in ipairs(vanillaCorporations) do
		local categories = easyEdit.enemyList:get("archive").categories
		local corp = _G[corp_id]
		corp.Enemies = getFinalEnemyLists()
		corp.EnemyCategories = copy_table(categories)
	end

	function Mission:NextRobot(name_only)
		local islandSlot = easyEdit:getCurrentIslandSlot()
		local bots = GAME.Bots[islandSlot]

		if bots == nil or #bots == 0 then
			bots = EnemyLists.Bots
		end

		return self:NextPawn(bots, name_only or false)
	end

	local oldStartNewGame = startNewGame
	function startNewGame()
		-- Give Easy Edit full control of which bosses are added
		local getNewBossesVanilla = getNewBosses
		getNewBosses = getNoNewBosses

		oldStartNewGame()
		GAME.Bots = {}

		getNewBosses = getNewBossesVanilla

		local timesPicked = {}
		for i, corp_id in ipairs(vanillaCorporations) do
			local corp = _G[corp_id]
			local enemies = corp.Enemies
			local categories = corp.EnemyCategories

			GAME.Enemies[i] = pickEnemies(categories, enemies, i, timesPicked)
			GAME.Bots[i] = copy_table(enemies.Bots)
		end
	end
end

local function registerBossLists()
	for i, id in ipairs(corporations) do
		local bossList = easyEdit.bossList:add(id)
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		bossList:copy(base)

		for _, boss in ipairs(getNewBosses()) do
			bossList:addBoss(boss)
		end

		bossList.name = base.Bark_Name
	end
end

local function registerMissionLists()
	for i, id in ipairs(corporations) do
		local missionList = easyEdit.missionList:add(id)
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		missionList:copy(base)
		missionList.name = base.Bark_Name
	end
end

local function registerStructureLists()
	for i, id in ipairs(corporations) do
		local corp_id = vanillaCorporations[i]
		local base = _G[corp_id]

		local structureList = easyEdit.structureList:add(id)
		structureList:copy(Corp_Default)
		structureList.name = base.Bark_Name
	end
end

local function registerIslandComposites()
	for i = 1, 4 do
		local id = islands[i]
		local islandComposite = easyEdit.islandComposite:add(id)
		islandComposite.name = islandNames[i]
		islandComposite.island = islands[i]
		islandComposite.corporation = corporations[i]
		islandComposite.ceo = ceos[i]
		islandComposite.tileset = tilesets[i]
		islandComposite.missionList = missionLists[i]
		islandComposite.bossList = bossLists[i]
		islandComposite.enemyList = corporations[i]
		islandComposite.structureList = corporations[i]
	end
end

local function registerIcons()
	local icons = {
		"delete",
		"reset",
		"delete_small",
		"reset_small",
		"warning_small",
	}

	for _, name in ipairs(icons) do
		local appendLoc = string.format("img/ui/easyEdit/%s.png", name)
		local filename = string.format("%simg/icons/%s.png", path, name)
		modApi:appendAsset(appendLoc, filename)
	end
end

local function registerFinalEnemyList()
	local enemyList = easyEdit.enemyList:add("finale")

	enemyList.name = "Finale"
	enemyList.categories = { "Enemies" }
	enemyList.enemies = { Enemies = {} }

	enemyList:addEnemy("Firefly", "Enemies")
	enemyList:addEnemy("Hornet", "Enemies")
	enemyList:addEnemy("Scarab", "Enemies")
	enemyList:addEnemy("Scorpion", "Enemies")
	enemyList:addEnemy("Crab", "Enemies")
	enemyList:addEnemy("Beetle", "Enemies")
	enemyList:addEnemy("Digger", "Enemies")
	enemyList:addEnemy("Blobber", "Enemies")
	enemyList:addEnemy("Jelly_Lava", "Enemies") -- jelly x3
	enemyList:addEnemy("Jelly_Lava", "Enemies")
	enemyList:addEnemy("Jelly_Lava", "Enemies")

	local oldGetSpawnList = GameObject.GetSpawnList
	function GameObject:GetSpawnList(island, ...)
		local result = oldGetSpawnList(self, island, ...)

		if island == 5 then -- final island!
			local enemyList = easyEdit.enemyList:get("finale")
			result = copy_table(enemyList.enemies.Enemies)
		end

		return result
	end
end

local function registerFinalBossList()
	local bossList_phase1 = easyEdit.bossList:add("finale1")
	local bossList_phase2 = easyEdit.bossList:add("finale2")
	bossList_phase1.name = "Finale I"
	bossList_phase2.name = "Finale II"

	bossList_phase1:addBoss("Mission_ScorpionBoss")
	bossList_phase1:addBoss("Mission_FireflyBoss")
	bossList_phase1:addBoss("Mission_HornetBoss")

	bossList_phase2:addBoss("Mission_BeetleBoss")
	bossList_phase2:addBoss("Mission_FireflyBoss")
	bossList_phase2:addBoss("Mission_HornetBoss")
end

local function markRegisteredAsVanilla()
	local function markAsVanilla(indexedList)
		for _, indexedEntry in pairs(indexedList._children) do
			indexedEntry._vanilla = true
		end
	end

	markAsVanilla(easyEdit.units)
	markAsVanilla(easyEdit.unitImage)
	markAsVanilla(easyEdit.weapons)
	markAsVanilla(easyEdit.missions)
	markAsVanilla(easyEdit.structures)
	markAsVanilla(easyEdit.corporation)
	markAsVanilla(easyEdit.tileset)
	markAsVanilla(easyEdit.structureList)
	markAsVanilla(easyEdit.enemyList)
	markAsVanilla(easyEdit.bossList)
	markAsVanilla(easyEdit.missionList)
	markAsVanilla(easyEdit.island)
	markAsVanilla(easyEdit.islandComposite)
end

local function markRegisteredAsMod(modId)
	local function markAsMod(indexedList)
		for _, indexedEntry in pairs(indexedList._children) do
			if not indexedEntry:isVanilla() and indexedEntry.mod == nil then
				indexedEntry.mod = modId
			end
		end
	end

	markAsMod(easyEdit.units)
	markAsMod(easyEdit.unitImage)
	markAsMod(easyEdit.weapons)
	markAsMod(easyEdit.missions)
	markAsMod(easyEdit.structures)
	markAsMod(easyEdit.corporation)
	markAsMod(easyEdit.tileset)
	markAsMod(easyEdit.structureList)
	markAsMod(easyEdit.enemyList)
	markAsMod(easyEdit.bossList)
	markAsMod(easyEdit.missionList)
	markAsMod(easyEdit.island)
	markAsMod(easyEdit.islandComposite)
end

local function lockEverything()
	local function lockChildren(indexedList)
		for _, indexedEntry in pairs(indexedList._children) do
			indexedEntry:lock()
		end
	end

	lockChildren(easyEdit.ceo)
	lockChildren(easyEdit.corporation)
	lockChildren(easyEdit.units)
	lockChildren(easyEdit.tileset)
	lockChildren(easyEdit.island)
	lockChildren(easyEdit.islandComposite)
	lockChildren(easyEdit.structureList)
	lockChildren(easyEdit.enemyList)
	lockChildren(easyEdit.bossList)
	lockChildren(easyEdit.missionList)
end

local function onModInitialized(modId)
	registerUnits()
	registerWeapons()
	registerMissions()
	registerStructures()
	markRegisteredAsMod(modId)
end

local function onModsInitialized()
	lockEverything()
	easyEdit.savedata:init()
end

updateMissingNames()
registerUnits()
registerWeapons()
registerMissions()
registerStructures()
registerIslands()
registerCorporations()
registerCEOs()
registerTilesets()
registerEnemyLists()
registerBossLists()
registerMissionLists()
registerStructureLists()
registerIslandComposites()
registerIcons()
registerFinalEnemyList()
registerFinalBossList()
markRegisteredAsVanilla()

modApi.events.onModInitialized:subscribe(onModInitialized)
modApi.events.onModsInitialized:subscribe(onModsInitialized)