Module:Sandbox/Pillow/UnitsOverview

Creature	Faction	Tier	Cost	Growth💬	Health	Attack	Defense	Damage	Morale	Luck	Initiative	Speed	Attack type
Swordsman	Temple	I	85		12	4	4	2-3	0	0	5	4	Melee Attack
Guard Captain	Temple	I	110		12	6	5	2-3	0	0	6	4	Melee Attack
Sun's Aegis	Temple	I	110		15	5	8	2-3	0	0	5	5	Melee Attack
Crossbowman	Temple	II	95		10	5	3	3-4	0	0	4	3	Ranged Attack
Austringer	Temple	II	130		10	5	3	3-4	0	0	4	4	Ranged Attack
Marksman	Temple	II	130		10	6	3	3-4	0	0	6	3	Ranged, no range penalty
Griffin	Temple	III	270		30	7	6	5-9	0	0	9	5	Melee Attack
Temple Griffin	Temple	III	360		30	8	6	5-9	0	0	12	7	Melee Attack
Guardian Griffin	Temple	III	360		30	9	10	7-7	0	0	10	6	Melee Attack
Lightweaver	Temple	IV	310		35	11	11	11-11	2	0	4	2	Ranged Attack
Hierophant	Temple	IV	440		35	13	16	11-11	2	0	6	4	Ranged Attack
Sun Herald	Temple	IV	440		35	16	13	11-18	2	0	5	3	Ranged Attack
Cavalry	Temple	V	850		85	12	17	10-14	1	0	10	7	Melee Attack
Noble Cavalry	Temple	V	1220		85	12	17	12-16	1	0	13	7	Melee Attack
Sunspear Cavalry	Temple	V	1220		85	14	22	12-16	2	0	10	8	Melee Attack
Inquisitor	Temple	VI	1000		90	20	24	19-23	0	0	7	4	Melee Attack
Mother Superior	Temple	VI	1500		125	20	28	19-23	0	0	9	5	Melee Attack
Excommunicator	Temple	VI	1500		90	26	24	23-33	0	0	8	6	Melee Attack
Angel	Temple	VII	2400 1		225	30	30	50-75	0	0	8	4	Ranged Attack
Archangel	Temple	VII	4000 2		225	35	30	50-75	0	0	12	5	Ranged, no penalties
Apotheosis	Temple	VII	4000 2		300	30	35	50-75	0	0	10	10	Ranged Attack
Necropolis
Skeleton	Necropolis	I	35		6	4	1	1-3	0	0	4	3	Melee Attack
Skeleton Warrior	Necropolis	I	50		6	4	3	2-3	0	0	5	4	Melee Attack
Skeleton Archer	Necropolis	I	50		6	4	1	1-3	0	0	4	3	Ranged Attack
Wight	Necropolis	II	100		8	3	0	2-4	0	0	5	6	Melee Attack
Wraith	Necropolis	II	130		8	6	0	2-4	0	0	7	7	Melee Attack
Phantasm	Necropolis	II	130		8	6	4	2-4	0	0	6	7	Melee Attack
Undead Pet	Necropolis	III	120		14	4	6	3-5	0	0	5	5	Melee Attack
Barghest	Necropolis	III	160		14	4	6	3-5	0	0	8	6	Melee Attack
Armored Hound	Necropolis	III	160		18	4	8	3-5	0	0	5	5	Melee Attack
Graverobber	Necropolis	IV	240		30	11	10	6-9	0	0	4	3	Long Reach
Merchant of Death	Necropolis	IV	340		30	11	10	7-10	0	0	5	4	Long Reach
Kennelmaster	Necropolis	IV	340		30	14	10	6-9	0	0	5	6	Long Reach
Lich	Necropolis	V	320		45	12	12	13-16	0	0	4	2	Ranged Attack
Pestilent Lich	Necropolis	V	460		45	12	12	16-16	0	0	5	3	Ranged Attack
Sanguine Lich	Necropolis	V	460		45	12	16	13-16	0	0	4	3	Ranged Attack
Dread Knight	Necropolis	VI	700		80	19	18	12-14	0	0	9	7	Melee Attack
Avatar of War	Necropolis	VI	1100		80	19	20	18-20	0	0	10	8	Melee Attack
Hollow Reaper	Necropolis	VI	1100		80	25	18	24-26	0	0	11	7	Melee Attack
Vampire	Necropolis	VII	1200 1		150	22	20	25-25	0	0	11	6	Melee Attack
Vampire Lord	Necropolis	VII	2000 2		175	22	26	30-30	0	0	13	8	Melee, no counter
Vampire Scholar	Necropolis	VII	2000 2		150	27	20	30-30	0	0	12	5	Long Reach
Grove
Faun	Grove	I	115		11	4	3	3-5	0	0	4	2	Ranged, no melee penalty
Faun Archer	Grove	I	150		11	4	3	3-5	0	0	5	3	Ranged, no penalties
Faun Warrior	Grove	I	150		11	6	6	4-6	0	0	6	4	Melee Attack
Hoplet	Grove	II	185		18	6	5	4-8	1	1	6	5	Melee Attack
Dawn Hoplet	Grove	II	250		18	9	5	4-8	1	2	6	6	Melee, no counter
Dusk Hoplet	Grove	II	250		18	7	7	4-8	2	1	9	5	Melee Attack
Vine Iriyad	Grove	III	220		45	6	11	5-7	0	0	2	3	Melee Attack
Fungal Iriyad	Grove	III	300		45	6	11	4-8	0	0	3	4	Melee Attack
Crystal Iriyad	Grove	III	300		45	6	11	5-7	0	0	3	5	Melee Attack
Naiad	Grove	IV	360		40	11	10	10-14	0	0	5	5	Ranged Attack
Vernal Naiad	Grove	IV	510		40	11	14	14-18	0	0	7	6	Ranged Attack
Brumal Naiad	Grove	IV	510		40	15	10	14-18	0	0	6	5	Ranged Attack
Herbomancer	Grove	V	640		60	12	14	15-18	0	0	6	4	Long Reach
Sporemancer	Grove	V	910		60	16	18	15-18	0	0	8	6	Long Reach
Murmurmancer	Grove	V	910		60	12	14	18-18	0	0	8	5	Long Reach
Qilin	Grove	VI	1200		130	22	20	24-30	0	2	9	6	Melee Attack
Thunder Qilin	Grove	VI	1900		130	22	20	30-36	0	2	11	8	Melee Attack
Mist Qilin	Grove	VI	1900		130	24	22	27-36	0	2	13	7	Melee Attack
Phoenix	Grove	VII	2900 1		200	28	26	45-65	0	0	14	8	Melee Attack
Flaming Phoenix	Grove	VII	4800 2		300	36	27	45-65	0	0	16	10	Melee Attack
Energy Phoenix	Grove	VII	4800 2		250	28	39	45-65	0	0	15	10	Melee Attack
Hive
Parasite	Hive	I	80		7	3	3	2-4	0	0	4	3	Long Reach
Warden Parasite	Hive	I	100		9	3	6	2-4	0	0	5	4	Long Reach
Ravager Parasite	Hive	I	100		7	6	3	2-4	0	0	6	5	Long Reach
Locust	Hive	II	130		12	5	4	3-5	0	0	6	5	Melee Attack
Overgrown Locust	Hive	II	170		16	5	4	3-5	0	0	7	6	Melee Attack
Harvester Locust	Hive	II	170		12	7	4	4-5	0	0	7	7	Melee Attack
Hornet	Hive	III	255		20	7	10	4-9	0	0	10	7	Melee Attack
Chanter	Hive	III	350		25	7	10	4-9	0	0	4	11	Melee Attack
Stinger	Hive	III	350		20	7	10	4-9	0	0	12	9	Melee Attack
Fire Larva	Hive	IV	100		16	4	4	4-4	0	0	4	4	Melee Attack
Scorpion	Hive	IV	300		40	12	12	10-12	1	1	5	4	Melee Attack
Volcanic Scorpion	Hive	IV	420		55	12	14	10-12	1	2	6	5	Melee Attack
Spelaean Scorpion	Hive	IV	420		40	14	12	12-14	2	1	6	4	Melee Attack
Reaver	Hive	V	750		45	18	12	10-40	1	0	11	6	Melee Attack
Menacing Reaver	Hive	V	1080		45	18	18	20-30	3	0	12	7	Melee Attack
Maniacal Reaver	Hive	V	1080		45	18	12	10-40	2	0	12	6	Melee Attack
Waurms	Hive	VI	1200		120	21	21	21-23	0	1	7	8	Melee Attack
Devourer	Hive	VI	1900		120	21	21	27-29	0	3	7	10	Melee Attack
Pyroboros	Hive	VI	1900		120	21	21	21-23	0	2	8	8	Ranged, no melee penalty
Hive Queen	Hive	VII	2100 1		200	25	25	60-60	0	0	8	6	Melee Attack
Hive Mother	Hive	VII	3400 2		275	25	25	60-60	0	0	13	6	Melee Attack
Hive Huntress	Hive	VII	3400 2		250	25	25	70-70	0	0	8	9	Melee, no counter
Schism
Ra'Shoth	Schism	I	80		10	3	3	3-3	0	0	4	4	Melee Attack
Stinging Ra'Shoth	Schism	I	100		10	5	5	3-3	0	0	4	4	Ranged Attack
Ferocious Ra'Shoth	Schism	I	100		10	3	3	3-3	0	0	6	6	Melee Attack
Cultist	Schism	II	165		15	5	4	4-6	0	0	4	3	Long Reach
Binder	Schism	II	220		15	5	7	4-6	0	0	5	5	Long Reach
Votary	Schism	II	220		15	9	4	4-8	0	0	4	4	Long Reach
Aga'Shoth Rider	Schism	III	195		20	7	7	5-7	0	0	6	4	Melee Attack
Aga'Shoth Tamer	Schism	III	270		20	8	8	5-7	0	0	9	7	Melee Attack
Aga'Shoth Mat'ha	Schism	III	270		25	7	7	5-7	0	0	7	6	Melee Attack
Grand Shoth	Schism	IV	450		50	11	11	10-13	0	0	8	5	Melee Attack
Unspeakable Shoth	Schism	IV	630		50	11	11	10-13	0	0	8	8	Melee Attack
Unthinkable Shoth	Schism	IV	630		50	11	11	10-13	0	0	13	6	Melee Attack
Concubus	Schism	V	580		60	16	13	14-20	0	0	5	2	Ranged Attack
Mistress of Chains	Schism	V	840		60	20	13	14-20	0	0	7	4	Ranged Attack
Bewitcher	Schism	V	840		60	16	13	20-20	0	0	6	6	Ranged, no melee penalty
Arbitrator	Schism	VI	1200		75	20	20	18-22	0	0	5	4	Ranged Attack
Rift Arbitrator	Schism	VI	1900		75	26	28	18-22	0	0	10	5	Ranged Attack
Bloated Arbitrator	Schism	VI	1900		75	28	22	22-26	0	0	8	5	Ranged Attack
Abyssal Envoy	Schism	VII	2500 1		250	30	30	40-60	0	0	8	5	Melee Attack
Abyssal Overseer	Schism	VII	4200 2		250	40	40	40-60	0	0	9	6	Melee Attack
Abyssal Executor	Schism	VII	4200 2		250	35	35	60-60	0	0	10	7	Melee Attack
Dungeon
Troglodyte	Dungeon	I	50		8	2	3	1-3	0	0	4	4	Melee Attack
Infernal Troglodyte	Dungeon	I	60		8	3	6	1-3	0	0	6	5	Melee Attack
Toxic Troglodyte	Dungeon	I	60		8	5	3	1-4	0	0	5	4	Melee Attack
Infiltrator	Dungeon	II	120		13	6	4	3-6	0	1	6	5	Melee Attack
Guile Infiltrator	Dungeon	II	160		13	6	4	3-6	0	2	6	7	Melee Attack
Bleak Infiltrator	Dungeon	II	160		13	6	4	3-6	0	2	9	6	Melee Attack
Onyx Dancer	Dungeon	III	165		20	8	8	5-7	0	0	4	2	Ranged Attack
Jasper Dancer	Dungeon	III	230		20	8	8	5-7	0	0	5	4	Ranged, no melee penalty
Aureate Dancer	Dungeon	III	230		20	8	8	5-7	0	0	7	3	Ranged Attack
Minotaur	Dungeon	IV	380		40	11	11	12-14	1	0	5	5	Melee Attack
Minotaur Lord	Dungeon	IV	530		40	11	15	12-14	1	0	6	6	Melee Attack
Minotaur Vanguard	Dungeon	IV	530		40	15	11	14-16	2	0	5	6	Melee Attack
Medusa	Dungeon	V	490		55	15	14	11-15	0	0	4	2	Ranged, no range penalty
Medusa Sculptor	Dungeon	V	700		55	20	18	15-15	0	0	5	4	Ranged, no range penalty
Medusa Queen	Dungeon	V	700		55	17	14	11-15	0	0	8	3	Ranged, no range penalty
Hydra	Dungeon	VI	1200		130	24	22	25-28	0	0	6	4	Melee, no counter
Chthonic Hydra	Dungeon	VI	1800		160	24	30	25-28	0	0	6	5	Melee, no counter
Infernal Hydra	Dungeon	VI	1800		130	30	26	30-38	0	0	8	6	Melee, no counter
Cave Dragon	Dungeon	VII	3000 1		275	29	25	50-60	0	0	11	8	Melee Attack
Black Dragon	Dungeon	VII	5000 2		350	29	32	65-75	0	0	14	9	Melee Attack
Ashen Dragon	Dungeon	VII	5000 2		300	37	25	65-75	0	0	12	11	Melee Attack
Neutral
Pixie	Neutral	I	100		6	2	2	2-2	0	0	8	8	Melee Attack
Gnat	Neutral	I	66		6	6	6	6-6	0	0	6	6	Melee Attack
Ghoul	Neutral	II	200		20	9	6	4-7	0	0	5	5	Melee Attack
Peasant	Neutral	II	150		5	1	1	1-1	0	0	1	1	Melee Attack
Halfling	Neutral	II	150		6	6	6	3-3	3	0	6	6	Ranged Attack
Animated Armor	Neutral	III	300		25	4	8	4-12	0	0	1	4	Melee Attack
Groll	Neutral	III	300		40	8	6	5-7	0	0	8	5	Melee Attack
Starchild	Neutral	IV	600		35	13	9	12-18	0	0	5	3	Ranged Attack
Unicorn	Neutral	IV	600		40	16	13	10-10	0	0	10	6	Melee Attack
Kittenhorn	Neutral	IV	600		40	10	10	10-15	3	3	6	6	Melee Attack
Primal Remnant	Neutral	IV	600		50	9	18	13-14	0	0	6	5	Melee Attack
Faerie Dragon	Neutral	V	1000		75	15	15	15-15	0	0	10	5	Melee Attack
Giant Toad	Neutral	V	1000		95	19	22	18-18	0	0	8	4	Melee Attack
Worldwatcher	Neutral	VI	1600		75	25	25	25-25	0	0	8	3	Ranged, no range penalty
Dragonslayer	Neutral	VI	1600		85	24	28	27-36	0	0	9	6	Melee Attack
Couatl	Neutral	VI	1600		95	26	19	21-28	0	0	12	8	Melee Attack
Sentinel of Glory	Neutral	VII	5000		400	40	40	50-50	0	0	7	7	Melee Attack
Dragon	Neutral	VII	3000		250	29	32	50-60	0	0	10	7	Melee Attack
Red Dragon	Neutral	VII	5000		325	33	32	60-70	0	0	12	9	Melee Attack
Avatar	Neutral	VIII	300		3	0	0	10-10	0	0	5	5	Melee Attack
Lich Dragon	Neutral	VIII	30000		999	66	66	99-99	0	0	9	9	Melee Attack

---

local p = {}

local FACTION_ORDER = {
    human = 1,
    undead = 2,
    nature = 3,
    demon = 4,
    unfrozen = 5,
    dungeon = 6,
    neutral = 7,
}

local FACTION_MAPPING = {
    Temple = 'human',
    Necropolis = 'undead',
    Grove = 'nature',
    Hive = 'demon',
    Schism = 'unfrozen',
    Dungeon = 'dungeon',
    Neutral = 'neutral'
}

-- Note: The right side is just a fallback, it will seldom be used.
local TRANSLATION_IDS = {
    creature =              'Creature',
    faction =               'Faction',
    tier =                  'Tier',
    wiki_cost =             'Cost',
    units_weekly =          'Growth',
    hp =                    'Hp',
    offence =               'Offence',
    defence =               'Defence',
    damage =                'Damage',
    initiative =            'Initiative',
    morale =                'Morale',
    luck =                  'Luck',
    speed =                 'Speed',
    wiki_units_neutral =    'Neutral',
    wiki_units_attack_type = 'Attack type',
    melee_attack =          'Melee Attack', -- melee_attack = 1
    wiki_units_no_counter = 'Melee,<br>no counter', -- melee_attack_no_counter = 2
    remote_attack         = 'Long Reach', -- remote_attack = 3
    ranged_attack         = 'Ranged Attack', -- ranged_attack = 4
    wiki_units_no_close = 'Ranged,<br>no melee penalty', -- ranged_attack_no_close = 5
    wiki_units_no_range = 'Ranged,<br>no range penalty', -- ranged_attack_no_range = 6
    wiki_units_no_range_close = 'Ranged,<br>no penalties', -- ranged_attack_no_range_close = 7
}

local RESOURCE_ICONS = {
    wood_cost     = 'Wood',
    ore_cost      = 'Ore',
    mercury_cost  = 'Mercury',
    dust_cost     = 'Dust',
    crystal_cost  = 'Crystal',
    gemstone_cost = 'Gemstones'
}

local STAT_ICONS = {
    hp          = 'Icon_Stats_Health',
    offence     = 'Icon_Stats_Attack',
    defence     = 'Icon_Stats_Defence',
    damage      = 'Icon_Stats_Damage',
    morale      = 'Icon_Stats_Morale',
    luck        = 'Icon_Stats_Luck',
    initiative  = 'Icon_Stats_Initiative',
    speed       = 'Icon_Stats_Speed',
}

local ATTACK_RANKS = {
    base_passive_melee_attack_name = 1,                 -- Swordsman     100,   0,   0,   0,   0,   0,   0,   0,   0
    base_passive_melee_attack_no_counter_name = 2,      -- Vampire Lord  101,   0,   0,   0,   0,   0,   0,   0,   0
    base_passive_remote_attack_name = 3,                -- Graverobber     0, 100,   0,   0,   0,   0,   0,   0,   0
    base_passive_ranged_attack_name = 4,                -- Onyx Dancer    50, 100, 100,  90,  80,  70,  60,  50,  50
    base_passive_ranged_attack_no_close_name = 5,       -- Apotheosis    100, 100, 100,  90,  80,  70,  60,  50,  50
    base_passive_ranged_attack_no_range_name = 6,       -- Marksman       50, 100, 100, 100, 100, 100, 100, 100, 100
    base_passive_ranged_attack_no_range_close_name = 7, -- Halfling      100, 100, 100, 100, 100, 100, 100, 100, 100
}

local ATTACK_ICONS = {
    base_passive_melee_attack_name = 'Base_passive_melee_attack',
    base_passive_melee_attack_no_counter_name = 'Base_passive_noncounter',
    base_passive_remote_attack_name = 'Base_passive_remote_attack',
    base_passive_ranged_attack_name = 'Base_passive_ranged_attack', -- TODO: clarify icon
    base_passive_ranged_attack_no_close_name = 'Base passive duelist', -- TODO: clarify icon
    base_passive_ranged_attack_no_range_name = 'Base_passive_sharpshooter', -- TODO: clarify icon
    base_passive_ranged_attack_no_range_close_name = 'Arbitrator_passive_1', -- TODO: clarify icon
}
local ATTACK_WORDS = {
    base_passive_melee_attack_name = 'melee_attack',
    base_passive_melee_attack_no_counter_name = 'wiki_units_no_counter',
    base_passive_remote_attack_name = 'remote_attack',
    base_passive_ranged_attack_name = 'ranged_attack',
    base_passive_ranged_attack_no_close_name = 'wiki_units_no_close',
    base_passive_ranged_attack_no_range_name = 'wiki_units_no_range',
    base_passive_ranged_attack_no_range_close_name = 'wiki_units_no_range_close',
}

local ROMAN = { 'I', 'II', 'III', 'IV', 'V', 'VI', 'VII', 'VIII' }

local PAGES = {
    Stinger = 'Stinger_(unit)',
}

local MORALE_AND_LUCK = {
    '<span class="elevated1">1</span>',
    '<span class="elevated2">2</span>',
    '<span class="elevated3">3</span>',
}

-- Awards a number from 0 to 3 for a unit, based on how upgraded it is
------------------------------------------------------------------------------------------------------------------------
local function computeUpgradeRank(row)
    local base = (row['base'] and row['base'] ~= "") and 1 or 0
    local upgrade = (row['upgrade'] and row['upgrade'] ~= "") and 10 or 0
    local sum = base + upgrade
    if sum == 1 then
        return 3
    elseif sum == 10 then
        return 1
    elseif sum == 0 then
        return 0
    else
        return 2
    end
end

-- Returns one of the ATTACK_RANKS keys (e.g. 'base_passive_melee_attack_name') or an empty string
------------------------------------------------------------------------------------------------------------------------
local function computeAttackType(row)
    local abilities = row['shared_abilities']
    if type(abilities) ~= "string" then
        return ""
    end
    for key, _ in pairs(ATTACK_RANKS) do
        local pattern = "%f[%w]" .. key .. "%f[%W]"
        if string.match(abilities, pattern) then
            return key
        end
    end
    return ""
end

-- Returns a rich wikitext with numbers and icons
------------------------------------------------------------------------------------------------------------------------
local function computeCost(row)
    local gold = row['gold_cost']
    local cost = gold .. '[[File:Icon_Resource_Gold.png|24px|link=]]'
    for key, icon in pairs(RESOURCE_ICONS) do
        local val = row[key]
        if val and val ~= '' and val ~= 0 then
            cost = cost .. ' ' .. val .. '[[File:Icon_Resource_' .. icon .. '.png|24px|link=]]'
        end
    end
    return cost
end

-- Retrieves the data from Cargo
------------------------------------------------------------------------------------------------------------------------
local function query(lang, forcedFaction)
    local tables = 'Unit, Translation'
    local fields = '' ..
        'Unit.id=id, ' ..
        'Unit.faction=faction, ' ..
        'Unit.tier=tier, ' ..
        'Unit.hp=hp, ' ..
        'Unit.offence=offence, ' ..
        'Unit.defence=defence, ' ..
        'Unit.damage_min=damage_min, ' ..
        'Unit.damage_max=damage_max, ' ..
        'Unit.morale=morale, ' ..
        'Unit.luck=luck, ' ..
        'Unit.initiative=initiative, ' ..
        'Unit.speed=speed, ' ..
        'Unit.shared_abilities=shared_abilities, ' .. -- we use it for range detection
        -- costs
        'Unit.gold_cost=gold_cost, ' ..
        'Unit.wood_cost=wood_cost, ' ..
        'Unit.ore_cost=ore_cost, ' ..
        'Unit.mercury_cost=mercury_cost, ' ..
        'Unit.dust_cost=dust_cost, ' ..
        'Unit.crystal_cost=crystal_cost, ' ..
        'Unit.gemstone_cost=gemstone_cost, ' ..
        -- upgrade
        'Unit.base_sid=base, ' ..
        'Unit.upgrade_sid=upgrade, ' ..
        -- Translation
        'Translation.language=language, ' ..
        'Translation.name=transName'

    -- where:
    local where = {}
    if forcedFaction then
        table.insert(where, 'Unit.faction="' .. forcedFaction .. '"')
    end
    table.insert(where, '(Translation.language = "en" OR Translation.language = "' .. lang .. '")')

    -- Get BOTH the English row AND the target language row
    local cargoArgs = {
        join = 'Unit.id = Translation.target_id',
        where = table.concat(where, ' AND '),
        orderBy = 'Unit.id ASC',
        limit = 1000 -- Increased limit since each unit will return 2 rows
    }

    return mw.ext.cargo.query(tables, fields, cargoArgs)
end

local function queryGrowths()
    local results = mw.ext.cargo.query(
		'MapObject',
		'units_hire_sid, units_weekly',
		{
			where = 'units_weekly > 0',
			limit = 200
		}
    )
   
   local growths = {}

   for _, row in ipairs(results) do
       growths[row.units_hire_sid] = row.units_weekly
   end

   return growths
end

-- Scans the Cargo results and folds the EN and lang rows into a single row
------------------------------------------------------------------------------------------------------------------------
local function consolidateResults(results, growths, lang)
    local units = {}
    for _, row in ipairs(results) do
        local id = row['id']

        -- If this is the first time seeing this id, set up its basic stats
        if not units[id] then
            units[id] = mw.clone(row)
            units[id].upgradeRank = computeUpgradeRank(row)
            units[id].attackType = computeAttackType(row)
            units[id].cost = computeCost(row)
            local baseid = id:gsub('_alt$', ''):gsub('_upg$', '')
        end

		units[id].units_weekly = growths[baseid]

        -- Save the names based on which language row we are currently looping over
        if row['language'] == 'en' then
            units[id].nameEn = row['transName']
        end
        if row['language'] == lang then
            units[id].nameX = row['transName']
        end
    end

    -- Convert the Dictionary into an Array
    local unitsArray = {}
    for _, unitData in pairs(units) do
        table.insert(unitsArray, unitData)
    end

    return unitsArray
end

-- Decides the order between two units, for sorting purposes.
------------------------------------------------------------------------------------------------------------------------
local function sortUnits(unitA, unitB)
    local rankA = FACTION_ORDER[unitA.faction] or 99
    local rankB = FACTION_ORDER[unitB.faction] or 99

    if rankA ~= rankB then
        return rankA < rankB
    end

    if unitA.tier ~= unitB.tier then
        return unitA.tier < unitB.tier
    end

    if unitA.upgradeRank ~= unitB.upgradeRank then
        return unitA.upgradeRank < unitB.upgradeRank
    end

    return unitA.hp < unitB.hp
end

-- Retrieves the text for some specific ids from Cargo Translations.
------------------------------------------------------------------------------------------------------------------------
local function translateIds(ids, lang, extra)
    -- Key list
    local list = {}
    for key, _ in pairs(ids) do
        list[#list + 1] = key
    end
    local idListString = '"' .. table.concat(list, '", "') .. '"'

    -- Cargo
    local where = {}
    table.insert(where, 'target_id IN (' .. idListString .. ')')
    table.insert(where, 'language = "' .. lang .. '"')
    if extra then
        table.insert(where, extra)
    end
    local results = mw.ext.cargo.query('Translation', 'target_id, name', {
        where = table.concat(where, ' AND '),
        limit = 100
    })

    -- Dictionary
    local dictionary = mw.clone(ids)
    for key, value in pairs(dictionary) do
        --dictionary[key] = value .. '[[Data:WikiTranslations/' .. lang .. '#' .. key .. '|💬]]'
        dictionary[key] = value .. '💬'
    end
    for _, row in ipairs(results) do
        local name = row['name']
        if name and name ~= '' and not name:find('💬') then
            dictionary[row['target_id']] = name
        end
    end
    return dictionary
end

-- Displays a variable
------------------------------------------------------------------------------------------------------------------------
local function dump(target)
    return '<pre>' .. mw.dumpObject(target) .. '</pre>'
end

-- Builds an icon+text combo for the unit's name
------------------------------------------------------------------------------------------------------------------------
local function renderUnitName(u, suffix)
    local stem = PAGES[u.nameEn] or u.nameEn
    local page = stem .. suffix
    local img = '[[File:' .. u.nameEn .. ' icon.png|64px|link=' .. page .. ']]'
    local text = '[[' .. page .. '|' .. u.nameX .. ']]'
    return img .. ' ' .. text
end

-- Boilerplate for the header cells
------------------------------------------------------------------------------------------------------------------------
local function addTh(tr, text)
    tr:tag('th'):wikitext(text):done()
end

-- Boilerplate for stat tds
------------------------------------------------------------------------------------------------------------------------
local function createStat(tr, value, pngName)
    tr:tag('td'):wikitext(value .. '[[File:' .. pngName .. '.png|link=]]'):done()
end

-- Boilerplate for the separator
------------------------------------------------------------------------------------------------------------------------
local function addSeparator(htmlTable, className, content)
    htmlTable:tag('tr')
        :addClass('separator')
        :addClass(className)
        :tag('td'):attr('data-sort-value', ''):attr('colspan', 100):wikitext(content):done()
end

------------------------------------------------------------------------------------------------------------------------
local function addFactionWords(words, lang, suffix, frame)
    for key, _ in pairs(FACTION_ORDER) do
        if key == 'neutral' then
            local file = '[[File:Primordial_Chaos.png|24px]]'
            words[key] = file .. ' [[Neutral_Units' .. suffix .. '|'.. words.wiki_units_neutral .. ']]'
        else
            words[key] = frame:preprocess('{{F|' .. key .. '|' .. lang .. '}}')
        end
    end
end

------------------------------------------------------------------------------------------------------------------------
local function renderIconWithHint(fileName, text, frame)
    local file = '[[File:' .. fileName .. '.png|32px|link=]]';
    return frame:preprocess('{{Hint|' .. file .. '|' .. text .. '}}')
end

------------------------------------------------------------------------------------------------------------------------
local function createHeader(htmlTable, words, frame)
    local header = htmlTable:tag('tr')
    -- 1 to 5 (general)
    addTh(header, words.creature)
    addTh(header, words.faction)
    addTh(header, words.tier)
    addTh(header, words.wiki_cost)
    addTh(header, words.units_weekly)
    -- 6 to 13 (main stats)
    addTh(header, renderIconWithHint(STAT_ICONS.hp, words.hp, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.offence, words.offence, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.defence, words.defence, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.damage, words.damage, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.morale, words.morale, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.luck, words.luck, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.initiative, words.initiative, frame))
    addTh(header, renderIconWithHint(STAT_ICONS.speed, words.speed, frame))
    -- 14 (others)
    addTh(header, renderIconWithHint('Sunlight_cavalry_passive_3', words.wiki_units_attack_type, frame))
end

------------------------------------------------------------------------------------------------------------------------
local function createBody(htmlTable, units, words, suffix, frame)
    local currentTier = units[1].tier
    local currentFaction = units[1].faction
    for _, u in ipairs(units) do

        -- separators
        if u.faction ~= currentFaction then
            currentFaction = u.faction
            currentTier = u.tier
            addSeparator(htmlTable, 'separator-large', words[currentFaction])
        elseif u.tier ~= currentTier then
            currentTier = u.tier
            addSeparator(htmlTable, 'separator-tiny', '')
        end

        local tr = htmlTable:tag('tr'):addClass('unit')
        if u.upgradeRank == 2 or u.upgradeRank == 3 then
            tr:addClass('upgraded')
        end
        -- 1 to 5 (general)
        tr:tag('td'):wikitext(renderUnitName(u, suffix)):done()
        tr:tag('td'):wikitext(words[currentFaction]):done()
        tr:tag('td'):wikitext(ROMAN[tonumber(u.tier)]):done()
        tr:tag('td'):attr('data-sort-value', string.match(u.cost, "^%d+")):wikitext(u.cost):done()
        tr:tag('td'):wikitext(tonumber(u.units_weekly)):done()
        -- 6 to 13 (main stats)
        createStat(tr, u.hp, STAT_ICONS.hp)
        createStat(tr, u.offence, STAT_ICONS.offence)
        createStat(tr, u.defence, STAT_ICONS.defence)
        createStat(tr, u.damage_min .. '-' .. u.damage_max, STAT_ICONS.damage)
        createStat(tr, MORALE_AND_LUCK[tonumber(u.morale)] or u.morale, STAT_ICONS.morale)
        createStat(tr, MORALE_AND_LUCK[tonumber(u.luck)] or u.luck, STAT_ICONS.luck)
        createStat(tr, u.initiative, STAT_ICONS.initiative)
        createStat(tr, u.speed, STAT_ICONS.speed)
        -- 14 (others)
        local wording = ATTACK_WORDS[u.attackType]
        local icon =  renderIconWithHint(ATTACK_ICONS[u.attackType], words[wording], frame)
        tr:tag('td'):attr('data-sort-value', ATTACK_RANKS[u.attackType]):wikitext(icon):done()
    end
end

-- Main public function
------------------------------------------------------------------------------------------------------------------------
function p.display(frame)
    --do return dump(frame) end

    -- Args
    local args = frame.args
    local lang = args.lang ~= '' and args.lang or 'en'
    --lang = 'ru'
    local forcedFaction = FACTION_MAPPING[args.faction] or nil
    --forcedFaction = 'human'

    -- Language
    local suffix = lang ~= 'en' and '/' .. lang or ''
    local words = translateIds(TRANSLATION_IDS, lang)
    addFactionWords(words, lang, suffix, frame)

    -- Cargo
    local results = query(lang, forcedFaction)
    local growth = queryGrowths()
    local units = consolidateResults(results, growth, lang)
    table.sort(units, sortUnits)

    -- Table
    local htmlTable = mw.html.create('table')
    htmlTable:addClass('wikitable sortable')
    if forcedFaction then
        htmlTable:addClass('forced-faction')
    end
    createHeader(htmlTable, words, frame)
    createBody(htmlTable, units, words, suffix, frame)

    -- Output
    local styleTag = frame:extensionTag('templatestyles', '', { src = frame:getTitle() .. '/styles.css' })
    return '<div class="units-overview">' .. styleTag .. tostring(htmlTable) .. '</div>'
end

return p