Модуль:Loc

Для документации этого модуля может быть создана страница Модуль:Loc/doc

local data = mw.loadData("Модуль:IanComradeBot/loc/data")

local p = {}

-- индекс для регистронезависимого поиска 
local lowerIndex = nil

local function tolower(s)
    return s and mw.ustring.lower(s) or s
end

local function buildLowerIndex()
    if lowerIndex then return end
    lowerIndex = {}
    for k, v in pairs(data) do
        if type(k) == "string" then
            local lk = tolower(k)
            if not lowerIndex[lk] then
                lowerIndex[lk] = v
            end
        end
    end
end

-- Возвращает значение по ключу
local function getValueByKeyRaw(value, subkey)
    if value == nil then return nil end
    local t = type(value)
    if t == "string" then
        return value
    elseif t == "table" then
        if subkey and value[subkey] then
            return value[subkey]
        end
        return nil
    end
    return nil
end

-- Основной поиск
local function findTextByKey(key, subkey)
    if not key then return nil end

    local direct = data[key]
    local result = getValueByKeyRaw(direct, subkey)
    if result then return result end

    buildLowerIndex()
    local lk = tolower(key)
    local v = lowerIndex[lk]
    result = getValueByKeyRaw(v, subkey)
    if result then return result end

    return nil
end

function p.GetString(frame)
    local key = frame.args[1]
    local subkey = frame.args[2] or "_value"

    if not key then
        return "Ошибка: Не указаны все необходимые параметры."
    end

    local result = findTextByKey(key, subkey)

    if not result then
        return key
    end

    return frame:preprocess("{{#invoke:Loc/Marking|main|<nowiki>".. result .. "</nowiki>}}")
end

function p.GetRawString(frame)
    local key = frame.args[1]
    local subkey = frame.args[2] or "_value"

    if not key then
        return "Ошибка: Не указаны все необходимые параметры."
    end

    local result = findTextByKey(key, subkey)
    if not result then
        return key
    end

    return frame:preprocess("<nowiki>".. result .. "</nowiki>")
end

return p