Module:ChapterVariables

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

local p = {}

function p.main(frame)
    local chapterVariablesText = frame.args[1]
    local chapterVariablesData = ""
    local chapterVariablesResult = ""
    if frame.args[2] == nil then
        local variablesSubPage = 'data'
    else
        local variablesSubPage = 'data/' .. frame.args[2]
    end
    chapterVariablesData = mw.loadData('Module:ChapterVariables/' .. variablesSubPage)
    if chapterVariablesData[chapterVariablesText] == nil and mw.Title.exists('Module:MagazineVariables/' .. variablesSubpage) then
        chapterVariablesData = mw.loadData('Module:MagazineVariables/' .. variablesSubPage)
    end
    if frame.args[2] == 'ja' then
        chapterVariablesResult = frame:preprocess(chapterVariablesData[chapterVariablesText])
    else
        chapterVariablesResult = chapterVariablesData[chapterVariablesText]
    end
    
    -- Special case for CDDH chapters
    if string.find(chapterVariablesText, "CDDH Chapter") then
        local chapterNumber = string.match(chapterVariablesText, "%d+")
        if chapterNumber and chapterNumber ~= "16" then
            chapterVariablesResult = "Chapter " .. chapterNumber
        end
    end
    
    return chapterVariablesResult or chapterVariablesText or ""
end

return p