Module:Citation/BCB

Revision as of 15:29, 12 November 2024 by NeonWabbit (talk | contribs)

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
        mw.log('Template:ChapterList not found')
        return {}
    end

    local content = listPage:getContent()
    if not content then
        mw.log('Failed to get content from Template:ChapterList')
        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

    mw.log('Chapters list parsed: ' .. mw.dumpObject(chapters))
    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
        mw.log('Input is a table, extracting URL from args')
        url = input.args and input.args[1] or nil
    end

    mw.log('Received URL: ' .. (url or 'nil'))

    if not url or type(url) ~= 'string' then
        mw.log('Invalid or missing URL')
        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
        mw.log('URL did not match expected pattern')
        return nil
    end

    mw.log('Matched Chapter: ' .. chapter .. ', Page: ' .. page)

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

    if chapter_title then
        local result = string.format("[https://www.bittersweetcandybowl.com/c%s/p%s Chapter %s: %s, page %s]", chapter, page, chapter, chapter_title, page)
        mw.log('Generated Result: ' .. result)
        return result
    end

    mw.log('No matching chapter found for Chapter: ' .. chapter)
    return string.format("[https://www.bittersweetcandybowl.com/c%s/p%s Chapter %s: [Unknown Title], page %s]", chapter, page, chapter, page)
end

return p