Module:Chapter number

From A Wiki of Ice and Fire
Jump to: navigation, search

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

local formatNumber = require('Module:Format number').main

local p = {}

function p.main()
    local title = mw.title.getCurrentTitle().text

    local book, chapter = title:match("^(.*)-(.*)$")

    if chapter == "Prologue" then
        return 0
    elseif chapter == "Epilogue" then
        return 998
    elseif chapter == "Appendix" then
        return 999
    else
        local number = chapter:match("^Chapter (%d+)$")
        return tonumber(number)
    end
end

function p.padded()
    return formatNumber({p.main(), "%02d"})
end

return p