Editing Module:ItemSources

-- Module:ItemSources
-- Displays item drop sources from NPCs in a table format
-- Usage: {{#invoke:ItemSources|list|source1=NPC Name,Quantity,Rarity}}

local p = {}

-- Helper to get args from either direct invoke or template
local function getArgs(frame)
    local args = {}
    
    local parent = frame:getParent()
    if parent and parent.args then
        for k, v in pairs(parent.args) do
            args[k] = v
        end
    end
    
    if frame.args then
        for k, v in pairs(frame.args) do
            args[k] = v
        end
    end
    
    return args
end

-- Display a list of item sources
function p.list(frame)
    local args = getArgs(frame)
    local title = args.title or "Item sources"
    local spriteSize = args.spriteSize or "24"
    
    -- Parse all sources
    local sources = {}
    
    local i = 1
    while args["source" .. i] or args["s" .. i] do
        local sourceStr = args["source" .. i] or args["s" .. i]
        
        -- Parse: "NPC Name,Quantity,Rarity" or "NPC Name,Quantity" or just "NPC Name"
        local parts = {}
        for part in sourceStr:gmatch("[^,]+") do
            table.insert(parts, part:match("^%s*(.-)%s*$")) -- trim
        end
        
        local npcName = parts[1] or "Unknown"
        local quantity = parts[2] or "1"
        local rarity = parts[3] or ""
        
        table.insert(sources, {
            npc = npcName,
            quantity = quantity,
            rarity = rarity
        })
        
        i = i + 1
    end
    
    -- Check if any source has rarity
    local hasRarity = false
    for _, source in ipairs(sources) do
        if source.rarity ~= "" then
            hasRarity = true
            break
        end
    end
    
    -- Build HTML
    local html = {}
    
    table.insert(html, '<div class="item-sources">')
    
    -- Table
    table.insert(html, '<table class="wikitable sortable item-sources-table">')
    
    -- Header row
    table.insert(html, '<tr>')
    table.insert(html, '<th class="item-sources-header-icon"></th>')
    table.insert(html, '<th>Source</th>')
    table.insert(html, '<th>Quantity</th>')
    if hasRarity then
        table.insert(html, '<th>Rarity</th>')
    end
    table.insert(html, '</tr>')
    
    -- Render each source
    for _, source in ipairs(sources) do
        table.insert(html, '<tr>')
        
        -- Icon cell
        table.insert(html, string.format(
            '<td class="item-sources-cell-icon">{{Sprite|%s|%s}}</td>',
            source.npc, spriteSize
        ))
        
        -- Source name cell
        table.insert(html, string.format(
            '<td class="item-sources-cell-name">[[%s]]</td>',
            source.npc
        ))
        
        -- Quantity cell
        table.insert(html, string.format(
            '<td class="item-sources-cell-quantity">%s</td>',
            source.quantity
        ))
        
        -- Rarity cell (only if any source has rarity)
        if hasRarity then
            table.insert(html, string.format(
                '<td class="item-sources-cell-rarity">%s</td>',
                source.rarity
            ))
        end
        
        table.insert(html, '</tr>')
    end
    
    table.insert(html, '</table>')
    table.insert(html, '</div>')
    
    return frame:preprocess(table.concat(html, '\n'))
end

return p