Module:Citation/BCB

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

local p = {}
local mw = require('mw')

-- Function to retrieve chapter list from `Template:ChapterList`
function p.getChaptersList()
    local listPage = mw.title.new('Template:ChapterList')
    if not listPage then
        return {}
    end

    local content = listPage:getContent()
    if not content then
        return {}
    end

    local chapters = {}
    -- Match raw `{{Chapter|number=...|title=...|date=...|pagecount=...}}` calls
    for number, title, date, pagecount in string.gmatch(content, "{{%s*Chapter%s*|%s*number%s*=%s*([%d%.]+)%s*|%s*title%s*=%s*([^|]+)%s*|%s*date%s*=%s*([^|]+)%s*|%s*pagecount%s*=%s*(%d+)%s*}}") do
        chapters[number] = title
    end

    return chapters
end

-- Function to format the BCB reference
function p.formatBCBReference(input)
    local url = input

    -- Handle if input is a table (e.g., frame object from a template call)
    if type(input) == 'table' then
        url = input.args and input.args[1] or nil
    end

    if not url or type(url) ~= 'string' then
        return nil
    end

    -- Match the URL pattern to extract chapter and page
    local chapter, page = url:match("^https://www%.bittersweetcandybowl%.com/c(%d+)/p(%d+)")
    if not chapter then
        chapter, page = url:match("^https://bcb%.cat/c(%d+)/p(%d+)")
    end
    if not chapter or not page then
        return nil
    end

    -- Retrieve chapter data
    local chapters = p.getChaptersList()
    local chapter_title = chapters[chapter]

    if chapter_title then
        -- Check if the chapter number contains a decimal (indicating an intermission)
        if string.match(chapter, "%d+%.%d+") then
            -- Format without "Chapter [number]:"
            return string.format("[https://www.bittersweetcandybowl.com/c%s/p%s %s, page %s]", chapter, page, chapter_title, page)
        else
            -- Format with "Chapter [number]:"
            return string.format("[https://www.bittersweetcandybowl.com/c%s/p%s Chapter %s: %s, page %s]", chapter, page, chapter, chapter_title, page)
        end
    end

    -- Fallback if no matching chapter is found
    return string.format("[https://www.bittersweetcandybowl.com/c%s/p%s Chapter %s: [Unknown Title], page %s]", chapter, page, chapter, page)
end

return p