Editing Module:NpcDataGive feedback

local p = {}

-- Load data and utility modules
local npcs_data = mw.loadJsonData("Data:NpcData.json")
local util_module = require('Module:Utilities')

-- returns the entire data table for a specific NPC, used by other Lua modules
function p.get_npc_data(name)
    if not name or name == '' then
        return nil
    end
    return npcs_data[name]
end

-- Retrieves a specific stat value for a given NPC.
-- Retrieves a specific stat value for a given NPC. Handles both simple and nested calls.
-- {{#invoke:NpcData|get_npc_var|NPC_KEY|STAT_KEY_1|STAT_KEY_2|...|SIG_FIGS}}
p.get_npc_var = function(frame)
    -- Safer argument handling for both live and debug environments
    local args = {}
    if frame.getParent then
        local parent_args = frame:getParent().args
        for k, v in pairs(parent_args) do args[k] = v end
    end
    -- Direct args from #invoke or the debug table overwrite parent args
    for k, v in pairs(frame.args) do args[k] = v end

    local npc_name = args[1] and mw.text.trim(args[1]) or nil
    if not npc_name or npc_name == '' then
        return '<span class="error">Error: NPC name not provided.</span>'
    end

    local npc = npcs_data[npc_name]
    if not npc then
        return '<span class="error">Error: NPC "' .. npc_name .. '" not found.</span>'
    end

    local current_value = npc
    local last_key_index = 1
    local path = {npc_name} -- Track the path for better error messages

    -- Traverse through the arguments to find the nested value
    for i = 2, #args do
        local key = args[i] and mw.text.trim(args[i]) or nil
        if not key or key == '' then break end

        -- Check if this might be the sig_figs argument (numeric on a non-table)
        if tonumber(key) and type(current_value) ~= "table" then
            break
        end

        if type(current_value) ~= "table" then
            return '<span class="error">Error: Invalid path at "' .. table.concat(path, '.') .. '". Tried to index a ' .. type(current_value) .. ' value.</span>'
        end
        
        -- Try string key first
        local next_value = current_value[key]
        
        -- If nil, try numeric key
        if next_value == nil then
            local num_key = tonumber(key)
            if num_key then
                next_value = current_value[num_key]
            end
        end
        
        if next_value == nil then
            return '<span class="error">Error: Key "' .. key .. '" not found in path "' .. table.concat(path, '.') .. '".</span>'
        end
        
        current_value = next_value
        table.insert(path, key)
        last_key_index = i
    end

    -- Handle sig_figs if provided
    local sig_figs_arg = args[last_key_index + 1]
    local sig_figs = sig_figs_arg and tonumber(mw.text.trim(sig_figs_arg)) or nil
    
    if sig_figs then
        if type(current_value) ~= "number" then
            return '<span class="error">Error: Cannot apply sig_figs to non-numeric value at "' .. table.concat(path, '.') .. '" (type: ' .. type(current_value) .. ').</span>'
        end
        current_value = util_module.round_to_sig_fig(current_value, sig_figs)
        if current_value == nil then
            return '<span class="error">Error: Rounding failed for value at "' .. table.concat(path, '.') .. '".</span>'
        end
    end

    -- Convert tables/booleans to strings for display
    if type(current_value) == "table" then
        return '<span class="error">Error: Path "' .. table.concat(path, '.') .. '" points to a table/object. Specify a deeper path to get a value.</span>'
    elseif type(current_value) == "boolean" then
        return tostring(current_value)
    end

    return current_value
end

-- Retrieves a nested value using dot notation (e.g., "Shield.BaseAbsorptionPerSecond")
--{{#invoke:NpcData|get_nested_var|NPC_KEY|PATH|SIG_FIGS}}--
p.get_nested_var = function(frame)
	local args = frame.args
	local npc_name = args[1] and mw.text.trim(args[1]) or nil
	local path = args[2] and mw.text.trim(args[2]) or nil
	local sig_figs = args[3] and mw.text.trim(args[3]) or nil

	if not npc_name or npc_name == '' then
		return '<span class="error">Error: NPC name not provided.</span>'
	end
	if not path or path == '' then
		return '<span class="error">Error: Path not provided.</span>'
	end

	local npc = npcs_data[npc_name]
	if not npc then
		return '<span class="error">Error: NPC "' .. npc_name .. '" not found.</span>'
	end

	-- Traverse nested structure
	local value = npc
	for key in string.gmatch(path, '([^.]+)') do
	    if type(value) ~= "table" then
	        return '<span class="error">Error: Path "' .. path .. '" is invalid.</span>'
	    end
	    -- try first with the raw key, then with a number conversion
	    local next = value[key] or value[tonumber(key)] or value[tostring(key)]
	    if next == nil then
	        return '<span class="error">Error: Key "' .. key .. '" not found in path.</span>'
	    end
	    value = next
	end

	-- Apply rounding if needed
	if sig_figs and tonumber(sig_figs) then
		value = util_module.round_to_sig_fig(value, sig_figs)
		if value == nil then
			return '<span class="error">Error: Rounding failed for value.</span>'
		end
	end

	return value
end

return p