Module:MonsterDrops

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

local p = {}

-- Drop rarity sort order (ascending)
local rarityOrder = {
    common = 1,
    semirare = 2,
    rare = 3,
    veryrare = 4,
    ultrarare = 5
}

-- Get sprite for an item from Cargo, fallback to item name
local function getItemSprite(itemName)
    local result = mw.ext.cargo.query('Items', 'sprite', {
        where = 'name="' .. itemName .. '"',
        limit = 1
    })
    
    if result and result[1] and result[1].sprite and result[1].sprite ~= '' then
        return result[1].sprite
    end
    return itemName
end

-- Format quantity display
local function formatQuantity(min, max)
    if not min or min == '' then
        return '?'
    end
    if not max or max == '' or min == max then
        return tostring(min)
    end
    return min .. '-' .. max
end

-- Get rarity sort value
local function getRarityValue(rarity)
    if not rarity then return 99 end
    return rarityOrder[string.lower(rarity)] or 99
end

-- Get quantity sort value (max if present, otherwise min)
local function getQuantitySortValue(min, max)
    if max and max ~= '' then
        return tonumber(max) or 0
    end
    return tonumber(min) or 0
end

-- Main function to display monster drops
function p.display(frame)
    local args = frame:getParent().args
    local source = args.source or mw.title.getCurrentTitle().text
    local sourceType = args.type or 'Monster'
    
    -- Query ItemSource for drops
    local drops = mw.ext.cargo.query('ItemSource', 'item,rarity,min,max', {
        where = 'source="' .. source .. '" AND source_type="' .. sourceType .. '"'
    })
    
    if not drops or #drops == 0 then
        return ''
    end
    
    -- Sort by rarity ascending, then quantity ascending
    table.sort(drops, function(a, b)
        local rarityA = getRarityValue(a.rarity)
        local rarityB = getRarityValue(b.rarity)
        if rarityA ~= rarityB then
            return rarityA < rarityB
        end
        return getQuantitySortValue(a.min, a.max) < getQuantitySortValue(b.min, b.max)
    end)
    
    -- Build table using mw.html
    local tbl = mw.html.create('table')
        :addClass('wikitable')
        :addClass('sortable')
    
    -- Header row
    local headerRow = tbl:tag('tr')
    headerRow:tag('th'):addClass('unsortable'):wikitext('')
    headerRow:tag('th'):wikitext('Item')
    headerRow:tag('th'):wikitext('Quantity')
    headerRow:tag('th'):wikitext('Rarity')
    
    for _, drop in ipairs(drops) do
        local sprite = getItemSprite(drop.item)
        local quantity = formatQuantity(drop.min, drop.max)
        local rarityValue = getRarityValue(drop.rarity)
        local quantityValue = getQuantitySortValue(drop.min, drop.max)
        
        local row = tbl:tag('tr')
        
        -- Sprite cell
        local spriteWikitext = frame:preprocess(string.format('{{Sprite|%s|x2}}', sprite))
        row:tag('td')
            :css('text-align', 'center')
            :wikitext(spriteWikitext)
        
        -- Item cell
        row:tag('td')
            :wikitext(string.format('[[%s]]', drop.item))
        
        -- Quantity cell with sort value
        row:tag('td')
            :attr('data-sort-value', quantityValue)
            :wikitext(quantity)
        
        -- Rarity cell with sort value
        row:tag('td')
            :attr('data-sort-value', rarityValue)
            :wikitext(drop.rarity or '')
    end
    
    return tostring(tbl)
end

return p