Difference between revisions of "Module:Database"

From Cassette Beasts
Line 10: Line 10:
 
end
 
end
 
if type(id) == "number" then
 
if type(id) == "number" then
return db.species.by_index[id]
+
return db.species.by_index[id] or p.species.unknown
 
else
 
else
return db.species.by_name[id]
+
return db.species.by_name[id] or p.species.unknown
 
end
 
end
 
end
 
end
Line 34: Line 34:
 
elseif value == false and frame.args.if_false ~= nil then
 
elseif value == false and frame.args.if_false ~= nil then
 
value = frame.args.if_false
 
value = frame.args.if_false
elseif type(value) == "number" and frame.args.num_format ~= nil then
 
value = frame.args.num_format:format(value)
 
 
elseif type(value) == "string" and frame.args["if_" .. value] ~= nil then
 
elseif type(value) == "string" and frame.args["if_" .. value] ~= nil then
 
value = frame.args["if_" .. value]
 
value = frame.args["if_" .. value]
Line 49: Line 47:
 
value = result
 
value = result
 
end
 
end
 +
if type(frame.args.format) == "string" then
 +
value = string.format(frame.args.format, value)
 +
end
 +
if type(value) == "table" and value.name ~= nil then
 +
value = value.name
 +
end
 +
 
if type(value) == "string" or type(value) == "number" or type(value) == "boolean" then
 
if type(value) == "string" or type(value) == "number" or type(value) == "boolean" then
 
return tostring(value)
 
return tostring(value)
Line 59: Line 64:
 
function p.get_species(frame)
 
function p.get_species(frame)
 
local species = get_species(frame.args[1])
 
local species = get_species(frame.args[1])
 +
return textify(frame, subscript(species, frame.args, 2))
 +
end
 +
 +
function p.get_prev_species(frame)
 +
local species = get_species(frame.args[1])
 +
if species.bestiary_index ~= nil and species.bestiary_index >= 0 then
 +
species = get_species(species.bestiary_index - 1)
 +
end
 +
return textify(frame, subscript(species, frame.args, 2))
 +
end
 +
 +
function p.get_next_species(frame)
 +
local species = get_species(frame.args[1])
 +
if species.bestiary_index ~= nil and species.bestiary_index >= 0 then
 +
species = get_species(species.bestiary_index + 1)
 +
end
 
return textify(frame, subscript(species, frame.args, 2))
 
return textify(frame, subscript(species, frame.args, 2))
 
end
 
end
  
 
return p
 
return p

Revision as of 20:30, 21 May 2022

Documentation for this module may be created at Module:Database/doc

local p = {}

local db = mw.loadData("Module:Database/data")

-- Private helpers

function get_species(id)
	if tonumber(id) ~= nil then
		id = tonumber(id)
	end
	if type(id) == "number" then
		return db.species.by_index[id] or p.species.unknown
	else
		return db.species.by_name[id] or p.species.unknown
	end
end

function subscript(value, path, path_start)
	while value ~= nil and path[path_start] ~= nil do
		local key = path[path_start]
		if tonumber(key) ~= nil then
			value = value[tonumber(key)]
		else
			value = value[key]
		end
		path_start = path_start + 1
	end
	return value
end

function textify(frame, value)
	if value == true and frame.args.if_true ~= null then
		value = frame.args.if_true
	elseif value == false and frame.args.if_false ~= nil then
		value = frame.args.if_false
	elseif type(value) == "string" and frame.args["if_" .. value] ~= nil then
		value = frame.args["if_" .. value]
	elseif type(value) == "table" and frame.args["foreach"] ~= nil then
		local delim = frame.args.delim or ""
		local result = ""
		for i,elem in ipairs(value) do
			if i > 1 then
				result = result .. delim
			end
			result = result .. frame:expandTemplate{title = frame.args["foreach"], args = { elem }}
		end
		value = result
	end
	if type(frame.args.format) == "string" then
		value = string.format(frame.args.format, value)
	end
	if type(value) == "table" and value.name ~= nil then
		value = value.name
	end

	if type(value) == "string" or type(value) == "number" or type(value) == "boolean" then
		return tostring(value)
	end
	return "<" .. tostring(value) .. ">"
end

-- Public functions for use with #invoke

function p.get_species(frame)
	local species = get_species(frame.args[1])
	return textify(frame, subscript(species, frame.args, 2))
end

function p.get_prev_species(frame)
	local species = get_species(frame.args[1])
	if species.bestiary_index ~= nil and species.bestiary_index >= 0 then
		species = get_species(species.bestiary_index - 1)
	end
	return textify(frame, subscript(species, frame.args, 2))
end

function p.get_next_species(frame)
	local species = get_species(frame.args[1])
	if species.bestiary_index ~= nil and species.bestiary_index >= 0 then
		species = get_species(species.bestiary_index + 1)
	end
	return textify(frame, subscript(species, frame.args, 2))
end

return p