View source for Module:Auphen/sandbox/frame

-- Module:Auphen/sandbox/frame -- test rig: [[Module:Auphen/frame]] pointed at
-- [[Module:Auphen/sandbox]], so a candidate engine can be run side by side with
-- the live one before it is promoted. Keep in step with the real frame.
-- Backs {{auphen}}: with no ruleset it estimates pronunciation; with a named
-- ruleset it runs that derivation. p.auphen formats the result (/.../ for an
-- estimate); p.raw returns the bare engine output (used by the testcases page).
local Auphen = require('Module:Auphen/sandbox')

local p = {}

-- Copy an mw.loadData category proxy into plain Lua tables for the engine.
local function plain_cats(src)
	if type(src) ~= 'table' then return nil end
	local out = {}
	for name, members in pairs(src) do
		local m = {}
		for i = 1, #members do m[i] = members[i] end
		out[name] = m
	end
	return out
end

-- Core: returns (output, isEstimation, errMessage). errMessage ~= nil on error.
local function run_core(a)
	local word = mw.text.trim(a[1] or '')
	local code = mw.text.trim(a[2] or '')
	local rs   = mw.text.trim(a[3] or '')
	if word == '' then return nil, nil, 'no word given' end
	if code == '' then return nil, nil, 'no language code given' end

	-- require (not mw.loadData): loadData's read-only proxy does not enumerate
	-- nested tables under pairs() in Lua 5.1, which would hide the categories.
	local ok, data = pcall(require, 'Module:Auphen/' .. code:upper())
	if not ok or type(data) ~= 'table' then
		return nil, nil, 'no sound data for code "' .. code .. '"'
	end

	if rs == '' then
		local eng = Auphen.new(plain_cats(data.ipacats or data.cats), data.ipa_rules)
		return eng:run(data.pronounce or '', word, true), true, nil
	else
		local ruleset = (type(data.sets) == 'table') and data.sets[rs] or nil
		if ruleset == nil then
			return nil, nil, 'unknown ruleset "' .. rs .. '" for code "' .. code .. '"'
		end
		local eng = Auphen.new(plain_cats(data.cats or data.ipacats))
		return eng:run(ruleset, word, false), false, nil
	end
end

local function err(msg)
	return '<strong class="error">Auphen: ' .. mw.text.nowiki(msg) ..
		'</strong>[[Category:Auphen errors]]'
end

-- {{#invoke:Auphen/frame|auphen | word | code | ruleset }}  (via {{auphen}})
function p.auphen(frame)
	local out, est, e = run_core(frame.args)
	if e then return err(e) end
	if est then return '<span class="IPA">/' .. out .. '/</span>' end
	return out
end

-- {{#invoke:Auphen/frame|raw | word | code | ruleset }} -- bare output, no
-- formatting (for the testcases page and callers that want the phonemes only).
function p.raw(frame)
	local out, _, e = run_core(frame.args)
	if e then return 'ERROR: ' .. e end
	return out
end

return p