Модуль:Entity Sprite/all

Материал из Space Station 14 Вики

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

local p = {}
local JsonPaths = require('Module:JsonPaths')

local fileExistsCache = {}

local function getSpritePath(entry)
	return entry.sprite
end

local function getSpriteStates(entry)
	local result = {}

	if entry.layers and entry.sprite then
		for _, layer in ipairs(entry.layers) do
			if layer.visible ~= false then
				table.insert(result, {
					state = tostring(layer.state or entry.state or ""),
					sprite = entry.sprite
				})
			end
		end
	elseif entry.state and entry.sprite then
		table.insert(result, {
			state = tostring(entry.state or ""),
			sprite = entry.sprite
		})
	end

	return (#result > 0) and result or nil
end

local function getPrefix(id, project)
	if project ~= "" and JsonPaths.has(id, project) then
		return project .. ":"
	end
	return ""
end

local function fileExists(name)
	if fileExistsCache[name] ~= nil then
		return fileExistsCache[name]
	end

	local title = mw.title.new(name, "File")
	local exists = title and title.exists or false
	fileExistsCache[name] = exists
	return exists
end

local function generateRepeatTemplate(data, project)
	local spriteGroups = {}

	for id, entry in pairs(data) do
		local found = false

		for _, group in pairs(spriteGroups) do
			local g = group[1]

			if entry.sprite == g.entry.sprite and entry.state == g.entry.state then
				table.insert(group, { id = id, entry = entry })
				found = true
				break
			end
		end

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

	local result = {}
	local frame = mw.getCurrentFrame()

	for _, group in pairs(spriteGroups) do
		if #group > 1 then
			local idLinks = {}

			for _, obj in pairs(group) do
				local id = obj.id
				local prefix = getPrefix(id, project)
				table.insert(idLinks, "[[:Файл:" .. prefix .. id .. ".png]]")
			end

			table.insert(result, frame:preprocess(
				"{{Entity Sprite/Repeat|" ..
				table.concat(idLinks, " ") ..
				"|" .. group[1].id ..
				"}}"
			))
		end
	end

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

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

	local prefix = getPrefix(id, project)

	local states = getSpriteStates(entry)
	local stateStr = ""

	if states then
		local links = {}
		for _, item in ipairs(states) do
			local url = baseUrl .. item.sprite .. "/" .. item.state .. ".png"
			table.insert(links, "[" .. url .. " " .. item.state .. "]")
		end
		stateStr = table.concat(links, ", ")
	end

	return mw.getCurrentFrame():preprocess(
		"{{Entity Sprite/Image|файл=" .. prefix .. id ..
		"|id=" .. id ..
		"|путь=" .. baseUrl .. spritePath ..
		"|state=" .. stateStr ..
		"}}"
	)
end

function p.main(frame)
	local action = frame.args[1]
	local json = frame.args.json or "sprite_entity.json"
	local checkFile = frame.args.checkFile or ""
	if checkFile == "" then
		checkFile = false
	end

	local project = JsonPaths.project()
	local baseUrl = JsonPaths.git() .. "/Resources/Textures/"
	local dataPage = JsonPaths.get(json)
	local spriteData = mw.loadData(dataPage)

	if not spriteData or type(spriteData) ~= "table" then
		return "Ошибка загрузки JSON: " .. dataPage
	end

	if action == "repeat" then
		return generateRepeatTemplate(spriteData, project)

	elseif action == "image" then
		local result = {}

		for id, entry in pairs(spriteData) do
			local prefix = getPrefix(id, project)

			local skip = false
			if checkFile then
				local fileName = prefix .. id .. ".png"
				if fileExists(fileName) then
					skip = true
				end
			end

			if not skip then
				local t = generateTemplate(id, entry, baseUrl, project)
				if t then
					table.insert(result, t)
				end
			end
		end

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

	return nil
end

return p