Module:Sandbox/Broono/Item Recipes

Documentation for this module may be created at Module:Sandbox/Broono/Item Recipes/doc

local p = {}

local addSeparator = require('Module:AddSeparator')
local findId = require('Module:FindId')
local img = require('Module:Img')

local data = {}
data.craftingAugmenting = mw.loadData('Module:CraftingAugmenting/data')
data.items = mw.loadData('Module:Items/data')


local COLLAPSIBLE = 7
local COLLAPSED = 15


local function strip(s)
    return s:match('^%s*(.-)%s*$')
end

local function split(s, sep)
   local t = {}
   if sep == nil or sep == '' then
      sep = '%s'
   end
   for m in s:gmatch('([^' .. sep .. ']+)') do
      table.insert(t, m)
   end
   return t
end

function table.length(t)
    local count = 0
    for _ in pairs(t) do count = count + 1 end
    return count
end

local function addItemIds(t, names)
    local _names = split(names, ';')
    for _, name in pairs(_names) do
        name = strip(name)
        if name ~= '' then
            local id = findId._findId({name, 'item'})
            if type(id) == 'number' then
                table.insert(t, {['id'] = id})
            else
                table.insert(t, {['id'] = id, ['name'] = name})
            end
        end
    end
end

local function addOtherItemFields(t)
    for i in pairs(t) do
        if type(t[i].id) == 'number' then
            local id = tostring(t[i].id)
            local craftingStats
            if data.items[id].craftingStats then
                craftingStats = data.items[id].craftingStats
            else
                craftingStats = data.items[id]
            end
            t[i].name = data.items[id].name
            t[i].level = craftingStats.level
            t[i].experience = craftingStats.experience
            t[i].category = craftingStats.category
        end
    end
end

local function addRecipes(t)
    for i in pairs(t) do
        if type(t[i].id) == 'number' then
            local id = tostring(t[i].id)
            t[i].recipes = {}
            if data.craftingAugmenting[id].crafting then
                for _, recipe in pairs(data.craftingAugmenting[id].crafting) do
                    table.insert(t[i].recipes, recipe.recipe)
                    t[i].skill = 'Crafting'
                end
            elseif data.craftingAugmenting[id].scrollcrafting then
                table.insert(t[i].recipes, data.craftingAugmenting[id].scrollcrafting)
                t[i].skill = 'Scrollcrafting'
            elseif data.craftingAugmenting[id].runecrafting then
                table.insert(t[i].recipes, data.craftingAugmenting[id].runecrafting)
                t[i].skill = 'Runecrafting'
            elseif data.craftingAugmenting[id].smithing then
                table.insert(t[i].recipes, data.craftingAugmenting[id].smithing)
                t[i].skill = 'Smithing'
            end
        end
    end
end

local function renderRecipes(t)
    local recipes = {}
    for _, r in pairs(t) do
        local ingredients = {}
        for id, amt in pairs(r) do
            local ingredient = addSeparator._addSeparator({amt}) .. 'x'
            ingredient = ingredient .. ' {{Img|' .. data.items[id].name .. '|30}}'
            table.insert(ingredients, ingredient)
        end
        table.insert(recipes, table.concat(ingredients, ', '))
    end
    return '\n| ' .. table.concat(recipes, '\n|-\n| ')
end

local function generateTableRecipe(t)
    local ingredients = {}
    for id, amt in pairs(t) do
        local name = data.items[id].name
        local image = img._img({name, 30})
        table.insert(ingredients, image .. ' x'.. addSeparator._addSeparator({amt}))
    end
    return table.concat(ingredients, ', ')
end

local function generateTableBody(t)

    local tbody = mw.html.create(nil)

    for i = 1, table.length(t) do

        local row = mw.html.create('tr')
        local rowspan = 0

        if type(t[i].id) ~= 'number' then
            local text = '\'\'Could not find item: \'\'\'' .. t[i].name .. '\'\'\'.'
            text = text .. ' Please check spelling or update data module.'

            row:tag('td')
                :css('text-align', 'center')
                :attr('colspan', 5)
                :wikitext(text)
        else
            rowspan = table.length(t[i].recipes)
            -- Icon
            row:node(mw.html.create('td')
                :attr('rowspan', rowspan)
                :css('text-align', 'center')
                :wikitext(img._img({t[i].name, 30, alt=''})))

            -- Name
            local name
            if t[i].name == mw.title.getCurrentTitle().fullText then
                name = '\'\'\'' .. t[i].name .. '\'\'\''
            else
                name = '\'\'\'[[' .. t[i].name .. ']]\'\'\''
            end
            row:node(mw.html.create('td')
                :attr('rowspan', rowspan)
                :attr('scope', 'row')
                :wikitext(name))

            -- Skill: Level
            local skill
            if t[i].category then
                skill = '[[' .. t[i].skill .. ']] (' .. t[i].category .. '): ' .. t[i].level
            else
                skill = '[[' .. t[i].skill .. ']]: ' .. t[i].level
            end
            row:node(mw.html.create('td')
                :attr('rowspan', rowspan)
                :wikitext(skill))

            -- Experience
            row:node(mw.html.create('td')
                :attr('rowspan', rowspan)
                :css('text-align', 'center')
                :wikitext(addSeparator._addSeparator({t[i].experience})))

            -- Recipe
            row:node(mw.html.create('td')
                :wikitext(generateTableRecipe(t[i].recipes[1])))
        end
        tbody:node(row)

        -- More recipes if item has more than one
        if rowspan > 1 then
            for j = 2, rowspan do
            tbody
            :tag('tr')
                :tag('td')
                    :wikitext(generateTableRecipe(t[i].recipes[j]))
            end
        end
    end

    return tbody
end

local function generateTableHead()

    local thead = mw.html.create(nil)
    local row = mw.html.create('tr')

    row
    :tag('th')
        :attr('scope', 'col')
        :wikitext('Icon')
    :done()
    :tag('th')
        :attr('scope', 'col')
        :wikitext('Name')
    :done()
    :tag('th')
        :attr('scope', 'col')
        :wikitext('Skill: Level')
    :done()
    :tag('th')
        :attr('scope', 'col')
        :wikitext('Experience')
    :done()
    :tag('th')
        :attr('scope', 'col')
        :wikitext('Recipe')
    :done()

    thead:node(row)

    return thead
end

local function generateTable(t)
    local tbl = mw.html.create('table')

    tbl
    :addClass('wikitable')
    :addClass('sortable')
    if table.length(t) > COLLAPSIBLE then tbl:addClass('mw-collapsible') end
    if table.length(t) > COLLAPSED then tbl:addClass('mw-collapsed') end

    tbl:node(generateTableHead())
    tbl:node(generateTableBody(t))

    return tbl
end

function p.itemRecipes(frame)
    return p._itemRecipes(frame:getParent().args)
end

function p._itemRecipes(args)

    if not args[1] or args[1] == '' then
        args[1] = mw.title.getCurrentTitle().fullText
    end

    local items = {}

    addItemIds(items, args[1])
    addOtherItemFields(items)
    addRecipes(items)

    return generateTable(items)
end

return p