local p = {}
local data = mw.loadData('Module: Table of spells')
local function table_has_value(t, v)
if t ~= nil then
for _,x in pairs(t) do if x == v then return true end end
end
return false
end
local function table_keys_sorted(t)
local keys = {}
for k in pairs(t) do
table.insert(keys, k)
end
table.sort(keys)
return keys
end
local function names_by_level()
local names = table_keys_sorted(data)
table.sort(names, function(a, b)
if data[a].level == data[b].level then
return a < b
else
return data[a].level < data[b].level
end
end)
return names
end
local function spell_table_header()
return [=[{| class="prettytable"
!|Icon
!|Name
!|Schools
!|Level
!|Power<br>cap
!|Range
!|Noise
!|Flags
!|Books
|----
]=]
end
local function spell_table_section(heading)
return '! colspan=9 style="text-align:left"|\n' ..
'====' .. heading .. '====\n' ..
'|----\n'
end
local function spell_school_link(school)
local skill = nil
if school == 'Poison' or school == 'Air' or school == 'Fire' or
school == 'Ice' or school == 'Earth' then
skill = school .. ' Magic'
elseif school:sub(-1) ~= 'y' and school:sub(-1) ~= 's' then
-- "Necromancy" isn't pluralised as a skill,
-- and "Hexes" and "Charms" are already
-- pluralized as a magic school. The others
-- are singular as a school, plural as a skill.
skill = school .. 's'
end
if skill ~= nil then
return '[[' .. skill .. '|' .. school .. ']]'
else
return '[[' .. school .. ']]'
end
end
local function format_schools(schools, no_link_for)
local ret = ''
for _,school in ipairs(schools) do
if school == no_link_for then
ret = ret .. school .. '/'
else
ret = ret .. spell_school_link(school) .. '/'
end
end
return ret:sub(1, -2)
end
local function format_range(range)
if range == nil then
return ''
elseif type(range) == 'table' then
return range[1] .. '-' .. range[2]
else
return range
end
end
local function format_flags(flags)
if flags == nil then
return ''
end
local ret = ''
for _,flag in ipairs(flags) do
ret = ret .. flag .. ', '
end
return ret:sub(1, -3)
end
local function format_books(books)
local ret = ''
for _,book in ipairs(books) do
ret = ret .. '[[' .. book .. ']]<br>'
end
return ret:sub(1, -5)
end
local function spell_table_line(name, info, no_link_for)
return '|[[File:' .. name:lower() .. '.png]]\n' ..
'|style="padding-left:1em"|[[' .. name .. ']]\n' ..
'|' .. format_schools(info['schools'], no_link_for) .. '\n' ..
'|' .. info['level'] .. '\n' ..
'|' .. info['cap'] .. '\n' ..
'|' .. format_range(info['range']) .. '\n' ..
'|' .. info['noise'] .. '\n' ..
'|' .. format_flags(info['flags']) .. '\n' ..
'|' .. format_books(info['books']) .. '\n' ..
'|----\n'
end
function p.spell_table(frame)
local alphabet = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
local ret = '==Spells==\n' .. spell_table_header()
local c = nil
for _, name in ipairs(table_keys_sorted(data)) do
if name:sub(1, 1) ~= c then
c = name:sub(1, 1)
ret = ret .. spell_table_section(c)
end
ret = ret .. spell_table_line(name, data[name])
end
return ret .. '|}\n'
end
function p.spell_table_by_level(frame)
local school = frame.args[1]
if school == '' then
school = nil
end
local ret = '==Spells==\n' .. spell_table_header()
for level=1,9 do
local found = false
for _, name in ipairs(table_keys_sorted(data)) do
if data[name]['level'] == level then
if school == nil or table_has_value(data[name]['schools'], school) then
if not found then
found = true
ret = ret .. spell_table_section('Level ' .. level)
end
ret = ret .. spell_table_line(name, data[name], school)
end
end
end
end
return ret .. '|}\n'
end
function p.spell_table_by_school(frame)
local schools = { 'Air', 'Charms', 'Conjuration', 'Earth', 'Fire', 'Hexes',
'Ice', 'Necromancy', 'Poison', 'Summoning', 'Translocation',
'Transmutation' }
local ret = '==Spells==\n' .. spell_table_header()
for _,school in ipairs(schools) do
ret = ret .. spell_table_section(school)
for _, name in ipairs(names_by_level(data)) do
if table_has_value(data[name]['schools'], school) then
ret = ret .. spell_table_line(name, data[name], school)
end
end
end
return ret .. '|}\n'
end
function p.spell_table_by_book(frame)
local books = {}
for _,spell in pairs(data) do if spell['books'] ~= nil then
for _,book in pairs(spell['books']) do
books[book] = true
end
end end
local ret = '==Spells==\n' .. spell_table_header()
for _,book in ipairs(table_keys_sorted(books)) do
ret = ret .. spell_table_section('[[' .. book .. ']]')
for _, name in ipairs(names_by_level(data)) do
if table_has_value(data[name]['books'], book) then
ret = ret .. spell_table_line(name, data[name])
end
end
end
return ret .. '|}\n'
end
function p.spell_table_by_flag(frame)
local flags = {}
for _,spell in pairs(data) do if spell['flags'] ~= nil then
for _,flag in pairs(spell['flags']) do
flags[flag] = true
end
end end
local ret = '==Spells==\n' .. spell_table_header()
ret = ret .. spell_table_section('No flags')
for _, name in ipairs(names_by_level(data)) do
if data[name]['flags'] == nil then
ret = ret .. spell_table_line(name, data[name])
end
end
for _,flag in ipairs(table_keys_sorted(flags)) do
ret = ret .. spell_table_section(flag) ..
'| colspan=9 |' ..
frame:expandTemplate{title = 'SpellFlagDesc ' .. flag, args = {}} ..
'\n|----\n'
for _, name in ipairs(names_by_level(data)) do
if table_has_value(data[name]['flags'], flag) then
ret = ret .. spell_table_line(name, data[name])
end
end
end
return ret .. '|}\n'
end
return p
