Module:Cenrail

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

local _m = {}

local data = mw.loadJsonData "Data:NguhRoutes/network.json"
local countries = require "Module:Data/UŊCDSO/Countries"

local find_when = function (predicate, list)
	for i, v in pairs(list) do
		if predicate(i, v) then
			return i, v
		end
	end
end

_m.line_table = function (frame)
	local linecode = frame.args[1] or frame.args.code or "CE"
	local dim = linecode:sub(1,2) == "N-" and "the_nether" or "overworld"
	local _, line = find_when(function(_, v) return v.code == linecode end, data.lines[dim])
	local out = "{| class=\"wikitable\"\n"
	out = out .. "|+ " .. line.name .. " [" .. line.code .. "]\n"
	out = out .. "|-\n"
	out = out .. "! rowspan = " .. #line.stops .. " style=\"background-color:" .. ( line.color or "#000" ) .. "\" | \n"
	out = out .. "! Stop \n! Code \n"
	out = out .. "|-\n"
	for _, stop in ipairs(line.stops) do
		local code = ((dim == "the_nether" and "N-" or "") .. stop.code)
		local _, name = find_when(function(i, v)
			return i == code
		end, data.stations)
		local country = ""
		if stop.code:sub(1,1) ~= "X" then
			country = ", [[" .. countries.get_country_code2(stop.code:sub(1,2)) .. "]]"
		end
		if type(name) == "table" then name = name[1] end
		out = out .. "| " .. name .. country .. "\n"
		out = out .. "| " .. code .. "\n" 
		out = out .. "|-\n"
	end
	out = out .. "|}"
	return out
end

return _m