Module:Odot control

local p = { } --Package to be exported

local function countyNumber(county)
    local numbers = mw.loadData("Module:Odot control/counties")
    return numbers[county]
end

function p.url(frame)
    --This function builds URLs.
    local pframe = frame:getParent() --get arguments passed to the template
    local args = pframe.args
    local county = args['county'] or '' --this string holds the raw county name
    local countyProcessed --this string holds the processed county name to be added to the URL
    
    --Le Flore and Roger Mills need special treatment to handle the spaces.
    --Everything else just gets converted to lower case.
    if county == "Le Flore" then
        countyProcessed = "leflore"
    elseif county == "Roger Mills" then
        countyProcessed = "rogermills"
    else
        countyProcessed = county:lower()
    end
    
    --Shove the county name in the URL and call it good
    return "http://www.odot.org/hqdiv/p-r-div/maps/control-maps/" .. countyProcessed .. ".pdf"
end

function p.edition(frame)
    --This function fills in the edition field of cite map.
    local pframe = frame:getParent() --get arguments passed to the template
    local args = pframe.args
    local year = args['year'] or ''
    
    local editions = {["2004"] = "2004", ["2006"] = "2006", ["2008"] = "2008", ["2010"] = "2010–2011", ["2011"] = "2010–2011", ["2010–2011"] = "2010–2011", ["2010-2011"] = "2010–2011"}
    return editions[year] or ''
end

function p.dateOutput(frame)
    --This function fills in the date field of cite map.
    local edition = p.edition(frame)
    local dates = {["2004"] = "2004-01-01", ["2006"] = "2006-01-01", ["2008"] = "2008-01-01", ["2010–2011"] = "2010-01-01" }
    return dates[edition]
end

return p