Module:Spellbook

local p = {}

local book_data = mw.loadData('Module:Table of spellbooks') local spell_data = mw.loadData('Module:Table of spells')

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 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 '' .. school .. ''
 else
   return '' .. school .. ''
 end

end

local function format_schools(schools, no_link_for)

 local ret = 
 for school in pairs(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 main_school(book)

 local schools = {}
 local num_spells = 0
 for _,name in pairs(book.spells) do
   num_spells = num_spells + 1
   for school in pairs(spell_data[name]['schools']) do
     if schools[school] == nil then
       schools[school] = 1
     else
       schools[school] = schools[school] + 1
     end
   end
 end

 local found = nil
 for school,count in pairs(schools) do
   if count == num_spells then
     if found ~= nil then return nil end
     found = school
   end
 end
 if found ~= nil then
   return found
 end
 for school,count in pairs(schools) do
   if count == num_spells - 1 then
     if found ~= nil then return nil end
     found = school
   end
 end
 return found

end

function p.spellbook_table(frame)

 local book = frame.args[1]
 if not book then
   return 
 end
 book = book:gsub("^Book of", "book of")
 local result = [==[{| cellpadding="5" border="1"

|- align="center" ! Tile || Spell || Type || Level ]==]

 local letters = 'abcdefghijklmnopqrstuvwxyz'
 for i,name in pairs(book_data[book].spells) do
   result = result .. '|-\n| File:' .. name:lower() .. '.png || ' ..
      letters:sub(i, i) .. ' - ' .. name .. ' || ' ..
      format_schools(spell_data[name].schools) .. ' || ' ..
      spell_data[name].level .. '\n'
 end
 result = result .. '|}\n'
 return result

end

function p.short_spell_list(frame)

 local book = frame.args[1]
 if not book then
   return 
 end
 local school = frame.args[2]
 if school ==  then school = nil end
 local result = "" .. book:gsub("^%l", string.upper) .. ": "
 local spell_list = {}
 book = book:gsub("^Book of", "book of")
 for _,name in pairs(book_data[book].spells) do
   if school == nil or spell_data[name]['schools'][school] then
     table.insert(spell_list, ''.. name .. '')
   end
 end
 result = result .. table.concat(spell_list, ', ')
 return result

end

function p.spell_sources(frame)

 local school = frame.args[1]
 local primary_books = frame.args[2]

 local done = {}
 local ret = 
 if primary_books ~= nil and primary_books ~=  then
   ret = ret .. ';Main Texts\n'
   for book in string.gmatch(primary_books, '[^,]+') do
     ret = ret .. ':' .. frame:expandTemplate{title = 'spellbook2', args = {book, school}} .. '\n'
     done[book:gsub("^Book of", "book of")] = true
   end
 end

 local found = {}
 for _,spell in pairs(spell_data) do
   if spell['schools'][school] then
     for book in pairs(spell['books']) do
       if not done[book] and not found[book] then
         found[book] = true
       end
     end
   end
 end

 if next(found) ~= nil then
   ret = ret .. ';Other Texts\n'
   -- TODO: ignore "the" while sorting
   for _,book in ipairs(table_keys_sorted(found)) do
     ret = ret .. ':' .. frame:expandTemplate{title = 'spellbook2', args = {book, school}} .. '\n'
   end
 end

 return ret:sub(1, -2)

end

function p.spellbook_info(frame)

 local name = frame.args[1]
 if not name or name == "" then
   name = mw.title.getCurrentTitle().text
 end
 name = name:gsub("^Book of", "book of")
 local book = book_data[name]
 if not book then
   return name
 end

 local args = {}
 args.name = book.name:gsub("^%l", string.upper)
 args.rarity = book.rarity

 local school = main_school(book)
 if school ~= nil then
   args.school = school
 end

 local infobox = frame:expandTemplate{title = 'book', args = args}

 local flavour = book.description
 if book.quote then
   flavour = flavour .. '\n----\n' .. book.quote:gsub('\n', '
')
 end
 flavour = frame:expandTemplate{title = 'flavour', args = {flavour}}
 return infobox .. '\n' .. flavour

end

return p