Module:User:Benwing2/demo

local export = {}

-- Ported from ] by Benwing2 on Sep 1st 2023 around 6am UTC.

-- creates a frame object that cannot access any of the parent's args
-- unless a table containing a list keys of not to inherit is provided
--
-- FIXME: This appears to do something like argument processing using ], and should be replaced
-- with a call to that module.
local function disinherit(frame, onlyTheseKeys)
	local parent = frame:getParent() or frame
	local orphan = parent:newChild{}
	orphan.getParent = parent.getParent --returns nil
	orphan.args = {}
	if onlyTheseKeys then
		local family = {parent, frame}
		for f = 1, 2 do
			for k, v in pairs(family and family.args or {}) do
				orphan.args = orphan.args or v
			end
		end
		parent.args = mw.clone(orphan.args)
		setmetatable(orphan.args, nil)
		for _, k in ipairs(onlyTheseKeys) do
			rawset(orphan.args, k, nil)
		end
	end
	return orphan, parent
end

local function getSeparator(args, default)
	local br = tonumber(args.br) and ("<br />"):rep(args.br) or args.br
	local sep = args.sep or br or default
	return #sep > 0 and " " .. sep .. " " or sep
end

function export.get(frame, deffmt)
	local params = {
		 = {},
		 = {set = {{"inline", "i", "1"}, {"twoline", "2"}, {"multiline", "m"}, {"compact", "c"}, "raw"},
			default = deffmt or "compact"},
		 = {},
		 = {},
		 = {type = "boolean"},
		 = {type = "boolean"},
		 = {},
	}
	local parent_args = frame:getParent().args
	local args = require("Module:parameters").process(parent_args, params)

	local code = args
	if code:match('UNIQ%-%-nowiki') then
		code = mw.text.unstripNoWiki(code)
			:gsub('&lt;', '<')
			:gsub('&gt;', '>')
			:gsub('&quot;', '"')
			-- Replace `&#125;%-` with `}-` because of some server quirk leading to
			-- =mw.text.unstripNoWiki(mw.getCurrentFrame():preprocess('<nowiki>}-</nowiki>'))
			-- outputting `&#125;-` instead of `}-`, while it's ok with `<nowiki>} -</nowiki>`
			:gsub('&#125;%-', '}-')
			-- The same with `-&#123;`
			:gsub('%-&#123;', '-{')
	end
	return {
		source = code,
		output = frame:preprocess(code):gsub(args.nocat and '%%]' or '', ''),
		args = args,
	}
end

function export.main(frame, demoTable)
	local deffmt = frame.args.fmt
	local show = demoTable or export.get(frame, deffmt)
	local args = show.args
	local defsep
	if args.fmt == "inline" or args.fmt == "twoline" then
		defsep = (args.reverse and '⇐' or '⇒') .. (args.fmt == "twoline" and "<br />" or "")
	elseif args.fmt == "raw" then
		defsep = ""
	else
		defsep = (args.reverse and "generated from" or "which produces") .. "<br />"
		if args.fmt == "compact" then
			defsep = "<br />" .. defsep
		end
	end
	args.sep = getSeparator(args, defsep)
	local source = frame:extensionTag("syntaxhighlight", show.source, {
		lang = "wikitext",
		inline = (args.fmt == "inline" or args.fmt == "twoline" or args.fmt == "compact") and true or nil,
		style = args.style
	})
	return args.reverse and
		show.output .. args.sep .. source or
		source .. args.sep .. show.output
end

--passing of args into other module without preprocessing
function export.module(frame)
	local orphan, frame = disinherit(frame, {
		'demo_template',
		'demo_module',
		'demo_module_func',
		'demo_main',
		'demo_sep',
		'demo_br',
		'demo_result_arg',
		'nocat'
	})
	local template = frame.args.demo_template and 'Template:'..frame.args.demo_template
	local demoFunc = frame.args.demo_module_func or 'main\n'
	local demoModule = require('Module:' .. frame.args.demo_module)
	frame.args.br, frame.args.result_arg = frame.args.demo_sep or frame.args.demo_br, frame.args.demo_result_arg
	local kill_categories = frame.args.nocat
	if demoModule then
		local named = {insert = function(self, ...) table.insert(self, ...) return self end}
		local source = {insert = named.insert, '{{', frame.args.demo_template or frame.args.demo_module, '\n'}
		if not template then
			source:insert(2, '#invoke:'):insert(4, '|'):insert(5, demoFunc)
		end
		local insertNamed = #source + 1
		for k, v in pairs(orphan.args) do
			local nan, insert = type(k) ~= 'number', {v}
			local target = nan and named or source
			target:insert'|'
			if nan then
				target:insert(k):insert'=':insert'\n'
				table.insert(insert, 1, #target)
			end
			target:insert(unpack(insert))
			local nowiki = v:match('nowiki')
			if nowiki or v:match('{{.-}}') then
				orphan.args = frame:preprocess(nowiki and mw.text.unstripNoWiki(v) or v)
			end
		end
		source:insert'}}'
		table.insert(source, insertNamed, table.concat(named))
		return export.main(orphan, {
			source = table.concat(source), "<>'|=~",
			output = tostring(demoModule(orphan)):gsub(kill_categories and '%%]' or '', ''),
			frame = frame
		})
	else
		return "ERROR: Invalid module function: "..demoFunc
	end
end

return export