Module:DataFetch

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

-- Written with ❤️ by fshark

local NOT_APPLICABLE = "N/A"

local title = mw.title.new("Module:DataFetch/data.json")
local data = title and title.exists and mw.text.jsonDecode(title:getContent())

assert(data, "Data failed to load. Please check \"Module:DataFetch/data.json\" exists.")

local biomeKeys = {
    wood = "w",
    stone = "s",
    earth = "e",
    arctic = "a"
}

-- if the biome argument is the name of the biome rather than its letter, return the corresponding letter
local function getBiomeKey(biomeName)
    return biomeKeys[string.lower(biomeName)] or biomeName
end

-- returns first argument that is not an empty string
local function getValidArg(...)
    for i = 1, select("#", ...) do
        local arg = select(i, ...)
        if arg and arg ~= "" then
            return arg
        end
    end
end

return {
    -- Universal getter for any property, for use with Template:BuildingData
    GetBuildingData = function(f)
        local args = f.args

        local targetData = getValidArg(args.data)
        local biome = getValidArg(args.biome)
        local resource = getValidArg(args.resource, args.res)
        local name = getValidArg(args.name)

        local buildingData = data[name]

        if targetData == "cost" then
            local costData = buildingData.cost
            if not costData then
                return NOT_APPLICABLE
            end
            local biomeValues = costData[getBiomeKey(biome)]
            assert(biomeValues, "Invalid biome argument") -- every building with a cost value has a biome value, so this one will throw an error when invalid instead of returning N/A.
            return biomeValues[resource] or NOT_APPLICABLE

        elseif targetData == "rate" then
            local rateData = buildingData.rate
            if not rateData then
                return NOT_APPLICABLE
            end
            return rateData[resource] or NOT_APPLICABLE

        elseif targetData == "balance" then
            local rateData = buildingData.balance
            if not rateData then
                return NOT_APPLICABLE
            end
            return rateData[resource] or NOT_APPLICABLE
        end
    end
}