Module:Sandbox/Aurelain/Cargo

Documentation for this module may be created at Module:Sandbox/Aurelain/Cargo/doc

local p = {}
local FACTION_ORDER = {
    human = 1,
    undead = 2,
    nature = 3,
    demon = 4,
    unfrozen = 5,
    dungeon = 6,
    neutral = 7,
}
local FACTION_IDS_IN_TRANSLATION = {
    'human',
    'undead',
    'nature',
    'demon',
    'unfrozen',
    'dungeon',
    'neutral', -- missing
}
local OTHER_IDS_IN_TRANSLATION = {
    'unit', -- missing
    'faction',
    'tier', -- missing
    'cost', -- missing
    'hp',
    'offence',
    'defence',
    'damage',
    'initiative',
    'speed',
    'ranged',
}
local RESOURCE_ICONS = {
    wood_cost     = 'Wood',
    ore_cost      = 'Ore',
    mercury_cost  = 'Mercury',
    dust_cost     = 'Dust',
    crystal_cost  = 'Crystal',
    gemstone_cost = 'Gemstones'
}

------------------------------------------------------------------------------------------------------------------------
-- Detects the current language from the URL
------------------------------------------------------------------------------------------------------------------------
local function getCurrentLang()
    local title = mw.title.getCurrentTitle()
    if title.isSubpage then
        if mw.language.isSupportedLanguage(title.subpageText) then
            return title.subpageText
        end
    end
    return 'en'
end

------------------------------------------------------------------------------------------------------------------------
-- 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 0, 1 or 2 depending on the ai_archetype value
------------------------------------------------------------------------------------------------------------------------
local function computeRange(row)
    local v = row['ai_archetype']
    if v == 'reach_type' then
        return 1
    elseif v == 'range_type' or v == 'range_type_melee_shooters' then
        return 2
    else
        return 0
    end
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)
    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.initiative=initiative, ' ..
        'Unit.speed=speed, ' ..
        'Unit.ai_archetype=ai_archetype, ' .. -- 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'

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

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

------------------------------------------------------------------------------------------------------------------------
-- Scans the Cargo results and folds the EN and lang rows into a single row
------------------------------------------------------------------------------------------------------------------------
local function consolidateResults(results, 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].ranged = computeRange(row)
            units[id].cost = computeCost(row)
        end

        -- 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, where)
    -- Initialize
    local dictionary = {}
    for _, id in ipairs(ids) do
        dictionary[id] = id:gsub("^%l", string.upper)
    end

    -- Cargo
    local idListString = '"' .. table.concat(ids, '", "') .. '"'
    local tables = 'Translation'
    local fields = 'target_id, name'
    where = where or ''
    local cargoArgs = {
        where = 'target_id IN (' .. idListString .. ') AND language = "' .. lang .. '"' .. where,
        limit = 100
    }
    local results = mw.ext.cargo.query(tables, fields, cargoArgs)

    -- Update dictionary
    if results then
        for _, row in ipairs(results) do
            dictionary[row['target_id']] = row['name']
        end
    end
    return dictionary
end

------------------------------------------------------------------------------------------------------------------------
-- Displays a variable
------------------------------------------------------------------------------------------------------------------------
local function debug(dict)
    return "<pre>" .. mw.text.jsonEncode(dict, mw.text.JSON_PRETTY) .. "</pre>"
end

------------------------------------------------------------------------------------------------------------------------
-- Builds an icon+text combo for the unit's name
------------------------------------------------------------------------------------------------------------------------
local function renderUnitName(u, suffix)
    local page = u.nameEn .. suffix
    local img = '[[File:' .. u.nameEn .. ' icon.png|40px|link=' .. page .. ']]'
    local text = '[[' .. page .. '|' .. u.nameX .. ']]'
    return '<span style="white-space: nowrap;">' .. img .. ' ' .. text .. '</span>'
end

------------------------------------------------------------------------------------------------------------------------
-- Main public function
------------------------------------------------------------------------------------------------------------------------
function p.displayOverview()
    local lang = getCurrentLang()
    local suffix = lang ~= 'en' and '/' .. lang or ''
    local wordsEn = translateIds(OTHER_IDS_IN_TRANSLATION, 'en')
    local wordsX = lang == 'en' and wordsEn or translateIds(OTHER_IDS_IN_TRANSLATION, lang)
    local factionEn = translateIds(FACTION_IDS_IN_TRANSLATION, 'en', ' AND type="faction"')
    local factionX = lang == 'en' and factionEn or translateIds(FACTION_IDS_IN_TRANSLATION, 'en', ' AND type="faction"')

    local results = query(lang)
    local units = consolidateResults(results, lang)
    table.sort(units, sortUnits)

    -- Table
    local htmlTable = mw.html.create('table')
    htmlTable:addClass('wikitable sortable')

    -- Header
    local header = htmlTable:tag('tr')
    header:tag('th'):wikitext(wordsX['unit']):done()
    header:tag('th'):wikitext(wordsX['faction']):done()
    header:tag('th'):wikitext(wordsX['tier']):done()
    header:tag('th'):wikitext(wordsX['cost']):done()
    -- main stats
    header:tag('th'):wikitext('[[File:Health Icon.png|24px|' .. wordsX['hp'] .. ']]'):done()
    header:tag('th'):wikitext('[[File:Icon_Stats_Attack.png|24px|' .. wordsX['offence'] .. ']]'):done()
    header:tag('th'):wikitext('[[File:Icon_Stats_Defence.png|24px|' .. wordsX['defence'] .. ']]'):done()
    header:tag('th'):wikitext('[[File:Icon_Stats_Damage.png|24px|' .. wordsX['damage'] .. ']]'):done()
    header:tag('th'):wikitext('[[File:Icon_Stats_Initiative.png|24px|' .. wordsX['initiative'] .. ']]'):done()
    header:tag('th'):wikitext('[[File:Icon_Stats_Speed.png|24px|' .. wordsX['speed'] .. ']]'):done()
    -- others
    header:tag('th'):wikitext('[[File:Base_passive_ranged_attack.png|24px|' .. wordsX['ranged'] .. ']]'):done()

    -- Body
    for _, u in ipairs(units) do
        local factionPage = factionEn[u.faction] .. '_Units' .. suffix
        local tr = htmlTable:tag('tr')
        tr:tag('td'):wikitext(renderUnitName(u, suffix)):done()
        tr:tag('td'):wikitext('[[' .. factionPage .. '|' .. factionX[u.faction] .. ']]'):done()
        tr:tag('td'):wikitext(u.tier):done()
        -- cost
        tr:tag('td')
            :attr('data-sort-value', string.match(u.cost, "^%d+"))
            :wikitext(u.cost)
            :done()
        -- stats
        tr:tag('td'):wikitext(u.hp):done()
        tr:tag('td'):wikitext(u.offence):done()
        tr:tag('td'):wikitext(u.defence):done()
        tr:tag('td'):wikitext(u.damage_min .. '-' .. u.damage_max):done()
        tr:tag('td'):wikitext(u.initiative):done()
        tr:tag('td'):wikitext(u.speed):done()
        tr:tag('td'):wikitext(u.ranged):done()
    end

    return debug(wordsX) .. tostring(htmlTable)
end

return p