View source for Module:HeroSubClassData

local p = {}

-- Table de correspondance pour traduire le nom de la classe en "faction_type"
local class_to_faction_type = {
    ["herald"]      = "demon_magic",
    ["warlock"]     = "dungeon_magic",
    ["cleric"]      = "human_magic",
    ["druid"]       = "nature_magic",
    ["necromancer"] = "undead_magic",
    ["riftspeaker"] = "unfrozen_magic",
    ["enforcer"]    = "demon_might",
    ["overlord"]    = "dungeon_might",
    ["knight"]      = "human_might",
    ["warden"]      = "nature_might",
    ["death knight"]= "undead_might",
    ["oathkeeper"]  = "unfrozen_might"
}

-- Fonction interne pour convertir l'argument en ID Cargo de sous-classe valide
local function get_cargo_subclass_id(input_id)
    if not input_id then return "" end
    
    -- On nettoie les espaces et on passe tout en minuscules
    local clean_id = mw.text.trim(string.lower(input_id))
    
    -- Si l'ID commence déjà par "sub_class_", on n'y touche pas
    if string.sub(clean_id, 1, 10) == "sub_class_" then
        return clean_id
    end
    
    -- On cherche à capturer le nom de la classe et le numéro (ex: "knight_2" ou "death knight_1")
    -- Le motif cherche tout jusqu'au dernier underscore suivi d'un chiffre
    local class_name, number = string.match(clean_id, "^(.+)_(%d+)$")
    
    if class_name and number then
        -- On nettoie le nom de la classe extrait au cas où
        class_name = mw.text.trim(class_name)
        
        -- Si la classe existe dans notre dictionnaire, on assemble le nouvel ID
        if class_to_faction_type[class_name] then
            return "sub_class_" .. class_to_faction_type[class_name] .. "_" .. number
        end
    end
    
    -- Si le format ne correspondait pas ou que la classe est inconnue, on renvoie l'original
    return clean_id
end

-- Fonction principale pour récupérer les données d'une sous-classe
-- Syntaxe : {{#invoke:HeroSubClassData|get|ID_SOUS_CLASSE|CHAMP|LANGUE}}
function p.get(frame)
    local args = frame.args[1] and frame.args or frame:getParent().args
    
    -- Conversion automatique de "classe_numero" en "sub_class_faction_type_numero"
    local subclass_id = get_cargo_subclass_id(args[1])
    local field = args[2]
    local lang = args[3] or 'fr' -- Par défaut en français

    if subclass_id == "" or not field then
        return "Erreur : ID de sous-classe ou champ manquant."
    end

    -- 1. GESTION DES DONNÉES TRADUITES (Ajusté sur "hero_sub_class")
    if field == 'name' or field == 'description' then
        local queryOpts = {
            where = string.format('target_id="%s" AND language="%s" AND type="hero_sub_class"', subclass_id, lang)
        }
        local res = mw.ext.cargo.query('Translation', field, queryOpts)

        if res and res[1] and res[1][field] then
            return res[1][field]
        else
            -- Système de fallback (secours) vers le français ou l'anglais
            if lang ~= 'fr' then
                local fallbackFr = mw.ext.cargo.query('Translation', field, {where = string.format('target_id="%s" AND language="fr" AND type="hero_sub_class"', subclass_id)})
                if fallbackFr and fallbackFr[1] and fallbackFr[1][field] then return fallbackFr[1][field] end
            end
            if lang ~= 'en' then
                local fallbackEn = mw.ext.cargo.query('Translation', field, {where = string.format('where target_id="%s" AND language="en" AND type="hero_sub_class"', subclass_id)})
                if fallbackEn and fallbackEn[1] and fallbackEn[1][field] then return fallbackEn[1][field] end
            end
            return ""
        end
    end

    -- 2. GESTION DES PROPRIÉTÉS STRUCTURELLES
    local queryOpts = {
        where = string.format('id="%s"', subclass_id)
    }
    local res = mw.ext.cargo.query('HeroSubClass', field, queryOpts)

    if res and res[1] and res[1][field] then
        return res[1][field]
    end

    return ""
end

return p