Module:UnitLinker

Documentation for this module may be created at Module:UnitLinker/doc

local p = {}

-- Charge la table de correspondance en mémoire
local function getUnitRegistry(lang)
    local registry = {}
    -- On récupère tout pour la langue en cours + l'anglais comme sécurité
    local query = "type = 'unit' AND (language = 'en' OR language = '" .. lang .. "')"
    local results = mw.ext.cargo.query("Translation", "target_id, name", { where = query })
    
    for _, row in ipairs(results) do
        -- On stocke le nom trouvé (qu'il soit anglais ou traduit) 
        -- associé à son identifiant unique (target_id)
        registry[row.name] = row.target_id
    end
    return registry
end

function p.link(frame)
    local text = frame.args[1] or ""
    local title = mw.title.getCurrentTitle().text
    -- Détecte le suffixe de langue (ex: /fr, /de, /es), sinon par défaut 'en'
    local lang = mw.ustring.match(title, "/([a-z_]+)$") or "en"
    
    local registry = getUnitRegistry(lang)
    
    -- Trier par longueur pour remplacer les noms composés avant les noms simples
    local keys = {}
    for k in pairs(registry) do table.insert(keys, k) end
    table.sort(keys, function(a, b) return #a > #b end)
    
    for _, unitName in ipairs(keys) do
        local engName = registry[unitName]
        
        -- Définir la cible du lien
        local target = (lang == "en") and engName or (engName .. "/" .. lang)
        local link = "[[" .. target .. "|" .. unitName .. "]]"
        
        -- Pattern complexe : 
        -- %f[%a] : frontière de mot
        -- [%[%]|\n] : on vérifie que le mot n'est pas déjà à l'intérieur d'un lien [[ ]]
        local escapedName = mw.ustring.gsub(unitName, "([%(%)%.%%%+%-%*%?%[%^%$])", "%%%1")
        
        -- On remplace uniquement si le mot n'est pas déjà dans un lien
        -- Cette logique cherche à éviter de casser les liens déjà présents
        local pattern = "(%f[%a])" .. escapedName .. "(%f[%A])"
        
        text = mw.ustring.gsub(text, pattern, function(prefix, suffix)
            -- On vérifie si on est déjà dans un bloc de lien (simplifié)
            if mw.ustring.find(text, "%[%[" .. escapedName) then return prefix .. unitName .. suffix end
            return prefix .. link .. suffix
        end)
    end
    
    return text
end

return p