Module:Infobox

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

local p = {}

-- Stat definitions: key -> { text, icon, width, height, isBonus }
-- Dimensions are original size, will be rendered at 3x
local stats = {
    wgt = { text = 'Weight', icon = 'Weight.png', width = 9, height = 8 },
    rng = { text = 'Range', icon = 'Range.png', width = 8, height = 8 },
    dmg = { text = 'Damage', icon = 'Damage.png', width = 8, height = 8 },
    eq = { text = 'Equip Size', icon = 'Weight.png', width = 9, height = 8 },
    hp = { text = 'HP', icon = 'Health.png', width = 9, height = 8, isBonus = true },
    mp = { text = 'MP', icon = 'Mana.png', width = 7, height = 8, isBonus = true },
    ab = { text = 'Ability', icon = 'Ability.png', width = 9, height = 9, isBonus = true },
    mag = { text = 'Magic', icon = 'Magic.png', width = 9, height = 8, isBonus = true },
    arm = { text = 'Armor', icon = 'Armor.png', width = 9, height = 8, isBonus = true },
    def = { text = 'Defense', icon = 'Defense.png', width = 6, height = 8, isBonus = true },
    ms = { text = 'Move Speed', icon = 'Move_Speed.png', width = 7, height = 8, isBonus = true },
    as = { text = 'Attack Speed', icon = 'Attack_Speed.png', width = 8, height = 8, isBonus = true },
    hpregen = { text = 'HP Regen', icon = 'Health_Regen.png', width = 8, height = 8, isBonus = true },
    mpregen = { text = 'MP Regen', icon = 'Mana_Regen.png', width = 7, height = 8, isBonus = true },
}

-- Other icons with dimensions
local icons = {
    type = { icon = 'Question_Mark.png', width = 6, height = 8 },
}

local SCALE = 3

local function formatBonus(value)
    if not value or value == '' then
        return nil
    end
    local num = tonumber(value)
    if num and num > 0 then
        return '+' .. value
    end
    return value
end

local function renderIcon(icon, width, height, tooltip)
    local w = width * SCALE
    local h = height * SCALE
    local img = string.format('[[File:%s|%dx%dpx|link=]]', icon, w, h)
    if tooltip then
        return string.format('<span title="%s">%s</span>', tooltip, img)
    end
    return img
end

function p.new(frame)
    local builder = {
        root = mw.html.create('table'),
        frame = frame
    }
    
    builder.root
        :addClass('wikitable')
        :addClass('infobox')
        :css('float', 'right')
        :css('clear', 'right')
        :css('margin', '0 0 1em 1em')
        :css('width', '250px')
    
    setmetatable(builder, { __index = p })
    return builder
end

function p:addClass(class)
    self.root:addClass(class)
    return self
end

function p:addHeader(text)
    self.root:tag('tr')
        :tag('th')
            :attr('colspan', 2)
            :css('text-align', 'center')
            :wikitext(text)
    return self
end

function p:addImage(content)
    self.root:tag('tr')
        :tag('td')
            :attr('colspan', 2)
            :css('text-align', 'center')
            :wikitext(content)
    return self
end

function p:addRow(label, value, iconKey)
    if not value or value == '' then
        return self
    end
    
    local row = self.root:tag('tr')
    local th = row:tag('th')
        :css('text-align', 'right')
    
    local iconData = icons[iconKey]
    if iconData then
        th:wikitext(renderIcon(iconData.icon, iconData.width, iconData.height, label))
    else
        th:wikitext(label)
    end
    
    row:tag('td')
        :css('text-align', 'left')
        :wikitext(value)
    return self
end

function p:addStat(key, value)
    if not value or value == '' then
        return self
    end
    
    local stat = stats[key]
    if not stat then
        return self
    end
    
    local displayValue = stat.isBonus and formatBonus(value) or value
    
    local row = self.root:tag('tr')
    local th = row:tag('th')
        :css('text-align', 'right')
    
    if stat.icon then
        th:wikitext(renderIcon(stat.icon, stat.width, stat.height, stat.text))
    else
        th:wikitext(stat.text)
    end
    
    row:tag('td')
        :css('text-align', 'left')
        :wikitext(displayValue)
    return self
end

function p:finish()
    return tostring(self.root)
end

function p.item(frame)
    local args = frame:getParent().args
    
    local box = p.new(frame)
        :addClass('item-infobox')
    
    local sprite = args.sprite or args.name
    local spriteContent = frame:expandTemplate{
        title = 'Sprite',
        args = { sprite, '64', class = 'pageimage' }
    }
    
    box:addHeader(args.name or 'Unknown')
       :addImage(spriteContent)
       :addRow('Slot', args.slot)
       :addRow('Type', args.category and string.format('[[:Category:%s|%s]]', args.category, args.category), 'type')
       :addStat('wgt', args.wgt)
       :addStat('rng', args.rng)
       :addStat('dmg', args.dmg)
       :addStat('eq', args.eq)
       :addStat('hp', args.hp)
       :addStat('mp', args.mp)
       :addStat('ab', args.ab)
       :addStat('mag', args.mag)
       :addStat('arm', args.arm)
       :addStat('def', args.def)
       :addStat('ms', args.ms)
       :addStat('as', args.as)
       :addStat('hpregen', args.hpregen)
       :addStat('mpregen', args.mpregen)
    
    local categories = ''
    if args.category then
        categories = categories .. '[[Category:' .. args.category .. ']]'
    end
    
    return box:finish() .. categories
end

return p