Модуль:Entity Sprite/all

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

local p = {}

local spriteData = mw.loadData("Модуль:IanComradeBot/prototypes/entity sprite.json/data")

local function deepEqual(t1, t2)
    if t1 == t2 then return true end
    if type(t1) ~= "table" or type(t2) ~= "table" then return false end

    local function isArray(t)
        local count = 0
        for k in pairs(t) do
            if type(k) ~= "number" then return false end
            count = count + 1
        end
        return count == #t
    end

    if isArray(t1) and isArray(t2) then
        if #t1 ~= #t2 then return false end
        for i = 1, #t1 do
            if not deepEqual(t1[i], t2[i]) then
                return false
            end
        end
        return true
    end

    for k, v in pairs(t1) do
        if t2[k] == nil or not deepEqual(v, t2[k]) then
            return false
        end
    end

    for k, v in pairs(t2) do
        if t1[k] == nil or not deepEqual(v, t1[k]) then
            return false
        end
    end

    return true
end

local function getInsensitiveKey(t, target)
    target = target:lower()
    for k, v in pairs(t) do
        if type(k) == "string" and k:lower() == target then
            return v
        end
    end
    return nil
end

local function getSpritePath(entry)
    local spriteField = getInsensitiveKey(entry, "Sprite")
    local iconField = getInsensitiveKey(entry, "Icon")

    if spriteField and spriteField.sprite then
        return spriteField.sprite
    elseif iconField and iconField.sprite then
        return iconField.sprite
    elseif spriteField and spriteField.layers then
        for _, layer in pairs(spriteField.layers) do
            if layer.sprite then
                return layer.sprite
            end
        end
    end
    return nil
end

local function getSpriteState(entry)
    local spriteBlock = getInsensitiveKey(entry, "Sprite")
    if spriteBlock then
        if spriteBlock.state then
            return spriteBlock.state
        elseif spriteBlock.layers then
            for _, layer in pairs(spriteBlock.layers) do
                if layer.state then
                    return layer.state
                end
            end
        end
    end

    local iconBlock = getInsensitiveKey(entry, "Icon")
    if iconBlock and iconBlock.state then
        return iconBlock.state
    end

    return nil
end

local function generateRepeatTemplate(data)
    local spriteGroups = {}

    for _, entry in pairs(data) do
        local found = false
        for _, group in pairs(spriteGroups) do
            if deepEqual(getInsensitiveKey(entry, "Sprite"), getInsensitiveKey(group[1], "Sprite")) and
               deepEqual(entry.EntityStorageVisuals, group[1].EntityStorageVisuals) and
               deepEqual(getInsensitiveKey(entry, "Icon"), getInsensitiveKey(group[1], "Icon")) then
                table.insert(group, entry)
                found = true
                break
            end
        end

        if not found then
            table.insert(spriteGroups, {entry})
        end
    end

    local result = {}
    for _, group in pairs(spriteGroups) do
        if #group > 1 then
            local idLinks = {}
            for _, entry in pairs(group) do
                table.insert(idLinks, "[[:Файл:" .. entry.id .. ".png]]")
            end
            table.insert(result, mw.getCurrentFrame():preprocess(
                "{{Entity Sprite/Repeat|" .. table.concat(idLinks, " ") .. "|" .. group[1].id .. "}}"
            ))
        end
    end

    return table.concat(result, "\n")
end

local function generateTemplate(entry, param)
    local spritePath = getSpritePath(entry)
    if not entry.id or not spritePath then
        return nil
    end

    local state = getSpriteState(entry)

    if param == "image" then
        local stateStr = ""
        if state then
            stateStr = " (state: " .. state .. ")"
        end
        return mw.getCurrentFrame():preprocess(
            "{{Entity Sprite/Image|" .. entry.id .. 
            "|https://github.com/space-syndicate/space-station-14/blob/master/Resources/Textures/" ..
            spritePath .. stateStr .. "}}"
        )
    end

    return nil
end

function p.main(frame)
    local param = frame.args[1]

    if not spriteData or type(spriteData) ~= "table" then
        return "Ошибка: Невозможно загрузить данные из JSON."
    end

    if param == "repeat" then
        return generateRepeatTemplate(spriteData)
    elseif param == "image" or param == "path" then
        local result = {}
        for _, entry in pairs(spriteData) do
            local template = generateTemplate(entry, param)
            if template then
                table.insert(result, template)
            end
        end
        return table.concat(result, "\n")
    else
        return nil
    end
end

return p