모듈:TemplateFunction/연습장

이 모듈에 대한 설명문서는 모듈:TemplateFunction/연습장/설명문서에서 만들 수 있습니다

local p = {}
local frame = mw.getCurrentFrame()
local Template = {}
local ParserFunction = {}

function concat(a, b)
	return tostring(a) .. tostring(b)
end

function contains(tbl, item)
	for value in ipairs(tbl) do
		if value == item then
			return true
		end
	end
	return false
end

function Template:parse(args)
	local child = frame:newChild{args = args}
	return frame:preprocess(mw.text.decode(mw.text.unstripNoWiki(self.source)))
end

function Template:bind(args)
	local length = #args
	
	return p.create(mw.text.decode(mw.text.unstripNoWiki(self.source)):gsub("{{{%f[^{](.-)}}}", function(exp)
		local group = mw.text.split(exp, "|")
		local key = tonumber(group[1]) or group[1]
		local result = args and args[key]
		
		table.remove(group, 1)
		
		if result then
			return result
		elseif type(key) == 'number' and key > length then
			key = key - length
			return group[1]
				and '{{{' .. key .. '|' .. table.concat(group, '|') .. '}}}'
				or '{{{' .. key .. '}}}'
		end
	end))
end

function Template:forward(mappings)
	return p.create(mw.text.decode(mw.text.unstripNoWiki(self.source)):gsub("{{{%f[^{](.-)}}}", function(exp)
		local group = mw.text.split(exp, "|")
		local key = table.remove(group, 1)
		key = mappings and mappings[key] or tonumber(key) and mappings[tonumber(key)] or key

		return #group >= 1
			and '{{{' .. key .. '|' .. table.concat(group, '|') .. '}}}'
			or '{{{' .. key .. '}}}'
	end))
end

function Template:finalize(keys)
	local cache = {}
	
	if not keys then
		return p.create(self:parse())
	end
	
	return p.create(mw.text.decode(mw.text.unstripNoWiki(self.source)):gsub("{{{%f[^{](.-)}}}", function(exp)
		local group = mw.text.split(exp, "|")
		local key = table.remove(group, 1)
		
		if cache[key] == nil then
			cache[key] = contains(keys, key)
		end
		
		if cache[key] and #group >= 1 then
			return table.concat(group, '|')
		end
	end))
end

function ParserFunction:bind(args)
	local newargs = {}
	local length = #self.args
	
	for key, value in pairs(self.args) do
		newargs[key] = value
	end
	
	for key, value in pairs(args) do
		key = tonumber(key) or key
		if type(key) == 'number' and key <= #args then
			key = key + length
		end
		newargs[key] = value
	end
	
	return p.import(self.name, newargs)
end

function ParserFunction:parse(args)
	if args then
		return self:bind(args):parse()
	end
	
	return frame:callParserFunction(self.name, self.args)
end

function getParserFunctionSource(name, args)
	return require('모듈:Transclution').createFunction(name, args)
end

function p.create(source)
	return setmetatable({
		source = source
	}, {
		__index = Template,
		__call = Template.bind,
		__tostring = Template.parse,
		__concat = concat
	})
end

function p.load(pagename)
	local title = mw.title.new(pagename, 'Template')
	local matches = ''
	
	while title.isRedirect do
		title = title.redirectTarget
	end
	
	for match in title:getContent():gmatch('<onlyinclude>(.-)</onlyinclude>') do
		matches = matches .. match
	end
	
	return p.create(matches:gsub('</?includeonly>', ''):gsub('<noinclude>(.-)</noinclude>', ''))
end

function p.import(name, _args)
	return setmetatable({
		name = name,
		args = _args or {},
		source = _args and #_args and getParserFunctionSource(name, _args or {})
	}, {
		__index = ParserFunction,
		__call = ParserFunction.bind,
		__tostring = ParserFunction.parse,
		__concat = concat
	})
end

function p.importTag(name)
	return p.import('#tag'):bind{name}
end

return p