모듈:JSON 문서 원본 보기

--[[
	ko:
	JSON 모듈
	JSON 코드에 있는 값의 내용을 반환하거나, 값을 수정한 JSON 코드를 다시 반환합니다.
	이 코드는 일단 CC BY-NC-SA로 배포되나, 추후 CC BY-SA로 변경될 수 있습니다.
	한국어 백괴게임의 Gustmd7410이 만들었습니다.
]]--
--[[
	en:
	JSON Module
	You can get and change JSON code's value with this.
	This code is distributed with CC BY-NC-SA license, but it could be changed into CC BY-SA license later.
	Made by Gustmd7410 in Korean Ungame.
]]--
local a = {}
local b = setmetatable({
	rtyp = a.typ,
	typ = false
}, {
	__index = a
})
local p = setmetatable({}, {
	__index = function(t, key)
		return b[key] and function(frame)
			local args = getRawArgs(frame.args)
			local data = mw.text.jsonDecode(table.remove(args, 1))
			local value = b[key](data, args)
			
			if type(value) == "table"
				then return mw.text.jsonEncode(value)
				else return value
			end
		end or nil
	end
})
function getRawArgs(args)
	local raw = {}
	
	for key, value in pairs(args) do
		raw[key] = value
	end
	
	return raw
end

function settype(value)
	if value:find('^("+)(.*)%1$') and pcall(mw.text.jsonDecode, value:gsub('^("+)(.*)%1$', "%2")) then
		return value:gsub('^"(("+)(.*)%2)"$', "%1")
	elseif pcall(mw.text.jsonDecode, value) then return mw.text.jsonDecode(value)
	elseif value == "" then return nil
	else return value
	end
end

function p.typ(frame)
	local key = frame.args[2]
	key = tonumber(key) or key
	local value = mw.text.jsonDecode(frame.args[1])[key]
	
	if value == null then return "null" end
	
	return type(value)
end

function a.typ(data)
	return type(data)
end

function a.get(data, args)
	for depth, key in pairs(args) do
		key = tonumber(key) or key
		data = data[key]
	end
	
	return data
end

function a.set(data, args)
	local key = args[1]
	local value = mw.text.trim(args[2])
	key = tonumber(key) or key
	
	data[key] = settype(value)
	
	return data
end

function p.new(frame)
	local data = getRawArgs(frame.args)
	
	for key, value in pairs(data) do
		data[key] = settype(mw.text.trim(value))
	end
	
	return mw.text.jsonEncode(data)
end

function a.mer(base, args)
	for seq in pairs(args) do
		local data = mw.text.jsonDecode(args[seq])
		
		for key in pairs(data) do base[key] = data[key] end
	end
	
	return base
end

function a.del(data, args)
	for seq, key in pairs(args) do
		data[tonumber(key) or key] = nil
	end
	
	return data
end

function a.len(data)
	return #data
end

function a.con(data, args)
	return table.concat(data, args[2] or ',', tonumber(args[3]) or 1, tonumber(args[4]) or #data)
end

function a.ins(data, args)
	local index = tonumber(args[3]) or #data + 1
	
	if index < 0 then
		index = #data + index
	end
	
	table.insert(data, index, args[2])
	return data
end

function a.rem(data, args)
	for seq, key in pairs(args) do
		key = tonumber(key)
		
		if key < 0 then
			key = #data + key
		end
		
		table.remove(data, key)
	end
	
	return data
end

function a.srt(data, args)
	table.sort(data)
	return data
end

function p.chk(frame)
	if pcall(mw.text.jsonDecode, frame.args[1])
		then return frame.args[2]
		else return frame.args[3]
	end
end

function p.str(frame)
	return mw.text.jsonDecode('"' .. frame.args[1] .. '"')
end

function p.pty(frame)
	local data = mw.text.jsonDecode(frame.args[1])
	return mw.text.jsonEncode(data, mw.text.JSON_PRETTY)
end

function p.cmd(frame)
	local data = mw.text.jsonDecode(frame.args[1])
	local commands = mw.text.jsonDecode('[' .. mw.text.trim(frame.args[2]):sub(1, -2) .. ']')
	local pretty = false
	
	for index, command in pairs(commands) do
		for action, args in pairs(command) do
			if action == 'pty' then
				pretty = true
			else
				data = a[action](data, args)
			end
			break
		end
	end
	
	if type(data) == "table" then
		return pretty and mw.text.jsonEncode(data, mw.text.JSON_PRETTY) or mw.text.jsonEncode(data)
	else
		return data
	end
end

function a.has(data, args)
	local valt = {}

	for key, value in pairs(data) do
		valt[value] = true
	end
	
	for index, value in pairs(args) do
		if valt[value] ~= true then
			return false
		end
	end
	
	return true
end

a['for'] = function(data, args)
	local text = ''
	local keyv = args.key or 'key'
	local valv = args.value or 'value'
	local func = require('모듈:TemplateFunction').new(args[1])
	
	for key, value in pairs(data) do
		if type(value) == "table" then
			value = mw.text.jsonEncode(value)
		end
		
		text = text .. func{
			[keyv] = key,
			[valv] = value
		}
	end
	
	return text
end

function a.map(data, args)
	local keyv = args.key or 'key'
	local valv = args.value or 'value'
	local func = require('모듈:TemplateFunction').new(args[1])
	
	for key, value in pairs(data) do
		if type(value) == "table" then
			value = mw.text.jsonEncode(value)
		end
		
		data[key] = settype(func{
			[keyv] = key,
			[valv] = value
		})
	end
	
	return data
end

function a.red(data, args)
	local val = ''
	local keyv = args.key or 'key'
	local valv = args.value or 'value'
	local prev = args.returned or '1'
	local func = require('모듈:TemplateFunction').new(args[1])
	
	for key, value in pairs(data) do
		if type(value) == "table" then
			value = mw.text.jsonEncode(value)
		end
		
		val = func{
			[keyv] = key,
			[valv] = value,
			[prev] = val
		}
	end
	
	return val
end

function p.spl(frame)
	local text = frame.args[1]
	local sep = frame.args[2]
	
	return mw.text.jsonEncode(mw.text.split(text, sep))
end

function a.eq(data, args)
	local obj = settype(args[1])
	local dlen = 0
	local olen = 0
	local thenReturn = args[2]
	local elseReturn = args[3]
	
	for key, value in pairs(data) do
		dlen = dlen + 1
		if obj[key] ~= value then
			return elseReturn
		end
	end
	
	for key, value in pairs(obj) do
		olen = olen + 1
	end
	
	return dlen == olen and thenReturn or elseReturn
end

function a.seteq(data, args)
	local arr = settype(args[1])
	local thenReturn = args[2]
	local elseReturn = args[3]
	
	if #data ~= #arr then
		return elseReturn
	end
	
	return a.has(data, arr) and thenReturn or elseReturn
end

function a.use(data, args)
	return require('모듈:TemplateFunction').create(args[1]):parse(data)
end

function a.pass(data)
	return require('모듈:Pipeline').from(data)
end

return p