Module:GetJSON

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

local p = {}
p.main = function(frame)
    local m = {}
    m.root = mw.loadJsonData(frame.args[1])
    
    -- Function to split the key by "."
    local function splitKey(key)
        local keys = {}
        for subkey in string.gmatch(key, "([^%.]+)") do
            table.insert(keys, subkey)
        end
        return keys
    end
    
    -- Function to get the value from the nested table
    local function getValueByKeys(root, keys)
        local value = root
        for _, key in ipairs(keys) do
            if key:sub(1, 1) == "#" then
                key = tonumber(key:sub(2))
                if not key then
                    return nil
                end
                local dictKeys = {}
                for dictKey in pairs(value) do
                    table.insert(dictKeys, dictKey)
                end
                table.sort(dictKeys)
                key = dictKeys[key]
            end
            
            if value[key] == nil then
                return nil
            end
            value = value[key]
        end

        -- If the value is a table and contains json_file and json_key
        if type(value) == "table" and value.json_file and value.json_key then
            local newRoot = mw.loadJsonData(value.json_file)
            local newKeys = splitKey(value.json_key)
            return getValueByKeys(newRoot, newKeys)
        end
        
        return value
    end

    local keys = splitKey(frame.args[2])
    local value = getValueByKeys(m.root, keys)
    
    if value ~= nil then
        if type(value) == "table" then
            if frame.args[4] == "keys" then
                -- Return the keys of the table if requested
                local str = ""
                for k, _ in pairs(value) do
                    str = str .. k .. ";"
                end
                return frame:preprocess(str:sub(1, -2))
            else
                -- Return the values of the table
                local str = ""
                for _, v in pairs(value) do
                    str = str .. v .. ";"
                end
                return frame:preprocess(str:sub(1, -2))
            end
        end
        return frame:preprocess(value)
    else
        if frame.args[3] ~= nil then
            return frame.args[3]
        else
            return ""
        end
    end
end
return p