enemy_themed.lua

local _ = wesnoth.textdomain 'wesnoth-wc'
local on_event = wesnoth.require("on_event")

local strings = {
	enemy_pet = _ "$name|'s pet" 
}
-- in the later scenarios there is a small chance that a scenario will be themed for an enemy
-- which means in paticular changing the castle of the enemy accorign to the unit type of that
-- enemy, and giving him an extra unit.
local function wct_map_enemy_themed(race, pet, castle, village, chance)
	if wesnoth.random(100) > chance then
		return
	end
	local boss = wesnoth.units.find_on_map {
		side="4,5,6,7,8,9",
		canrecruit=true,
		race=race,
		-- human means only outlaw
		wml.tag["not"] {
			race="human",
			wml.tag["not"] {
				wml.tag.filter_wml {
					alignment="chaotic",
				}
			}
		}
		
	}
	boss = boss[1]
	if boss == nil then
		return
	end
	--give themed castle
	wesnoth.set_terrain(boss.loc, "K" .. castle, "base")
	wesnoth.wml_actions.terrain {
		terrain="C" .. castle,
		wml.tag["and"] {
			terrain = "C*,*^C*",
			wml.tag["and"] {
				wml.tag.filter {
					x=boss.x,
					y=boss.y,
				},
				radius=999,
				wml.tag.filter_radius {
					terrain="K*^*,C*^*,*^K*,*^C*"
				},
			},
		},
	}
	local elvish_castle = wesnoth.get_locations {
		terrain="Cv",
		wml.tag.filter_adjacent_location {
			terrain="Kv^*"
		}
	}
	-- extra tweak with trees to elvish castle
	for i, tile in ipairs(elvish_castle) do
		if wesnoth.random(10) <= 4 then
			wesnoth.set_terrain(tile, "Cv^Fet")
		end
	end
	-- adjacent themed villages
	wesnoth.wml_actions.terrain {
		terrain=village,
		wml.tag["and"] {
			terrain="*^V*",
			wml.tag.filter_adjacent_location {
				terrain="C" .. castle .. ",K" .. castle .. "^*",
			},
		},
	}
	-- give pet
	wesnoth.wml_actions.unit {
		x = boss.x,
		y = boss.y,
		type=pet,
		side = boss.side,
		name= stringx.vformat(enemy_pet, { name = boss.name }),
		role = "hero",
		overlays = "misc/hero-icon.png",
		wml.tag.modifications {
			wc2_heroes.trait_heroic,
			wc2_heroes.trait_expert,
			wml.tag.object {
				id = "wc2_hero_overlay",
				wml.tag.effect {
					apply_to="overlay",
					add = "misc/hero-icon.png",
				} 
			}
		}
	}
end

function wesnoth.wml_actions.wc2_enemy_themed(cfg)
	wct_map_enemy_themed(cfg.race, cfg.pet, cfg.castle, cfg.village, cfg.chance)
end