Module:Chapters

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

-- Module:Chapters
local p = {}

function p.urlDecode(str)
    str = string.gsub(str, '%%(%x%x)', function(x) return string.char(tonumber(x, 16)) end)
    return str
end

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

    -- Replace various apostrophe encodings to ensure consistency
    content = string.gsub(content, "%%E2%%80%%99", "%%27")
    content = string.gsub(content, "’", "'")
    content = string.gsub(content, "'", "'")

    local chapters = {}
    for number, title, date, pagecount in string.gmatch(content, "{{Chapter|number=([%d%.]+)|title=([^|]+)|date=([^|]+)|pagecount=(%d+)}}") do
        table.insert(chapters, {number = number, title = title, date = date, pagecount = tonumber(pagecount)})
    end

    return chapters
end

function p.getMainChaptersCount()
    local chapters = p.getChaptersList()
    local count = 0
    for _, chapter in ipairs(chapters) do
        if not string.match(chapter.number, "%.") then
            count = count + 1
        end
    end
    return count
end

function p.getChapterDetails(chapterTitle, detail)
    local chapters = p.getChaptersList()
    local decodedTitle = p.urlDecode(chapterTitle)
    mw.log("Looking for chapter: " .. decodedTitle)  -- Debugging: log the decoded chapter title being searched
    for _, chapter in ipairs(chapters) do
        mw.log("Checking chapter: " .. chapter.title)  -- Debugging: log each chapter title
        if chapter.title == decodedTitle then
            mw.log("Found chapter: " .. chapter.title)  -- Debugging: log the matched chapter title
            return chapter[detail]
        end
    end
    mw.log("Chapter not found: " .. decodedTitle)  -- Debugging: log if the chapter is not found
    return nil
end

function p.getTotalChapters()
    return p.getMainChaptersCount()
end

function p.getNextChapter(chapterTitle)
    local chapters = p.getChaptersList()
    local decodedTitle = p.urlDecode(chapterTitle)
    for i, chapter in ipairs(chapters) do
        if chapter.title == decodedTitle and i < #chapters then
            return chapters[i + 1].title
        end
    end
    return nil
end

function p.getPreviousChapter(chapterTitle)
    local chapters = p.getChaptersList()
    local decodedTitle = p.urlDecode(chapterTitle)
    for i, chapter in ipairs(chapters) do
        if chapter.title == decodedTitle and i > 1 then
            return chapters[i - 1].title
        end
    end
    return nil
end

function p.formatDate(date)
    local year, month, day = string.match(date, "(%d+)%-(%d+)%-(%d+)")
    if year and month and day then
        return string.format("{{Start date|%s|%s|%s}}", year, month, day)
    else
        return date
    end
end

function p.infoboxChapter(frame)
    local chapterTitle = frame.args.chapter_title or mw.title.getCurrentTitle().text
    mw.log("Infobox for chapter: " .. chapterTitle)  -- Debugging: log the chapter title for the infobox
    local image = frame.args.image
    local caption = frame.args.caption
    local major_characters = frame.args.major_characters
    local minor_characters = frame.args.minor_characters
    local locations = frame.args.locations
    local collected_in = frame.args.collected_in

    caption = caption and caption ~= "" and caption or nil

    local chapter_no = p.getChapterDetails(chapterTitle, "number") or "N/A"
    local date = p.getChapterDetails(chapterTitle, "date") or "N/A"
    local pagecount = p.getChapterDetails(chapterTitle, "pagecount") or "N/A"
    local total_chapters = p.getTotalChapters() or "N/A"
    local next_chapter = p.getNextChapter(chapterTitle) or "N/A"
    local prev_chapter = p.getPreviousChapter(chapterTitle) or "N/A"

    mw.log("Chapter details - Number: " .. chapter_no .. ", Date: " .. date .. ", Pagecount: " .. pagecount)  -- Debugging: log chapter details
    mw.log("Navigation - Previous: " .. prev_chapter .. ", Next: " .. next_chapter)  -- Debugging: log navigation details

    local is_intermission = string.match(chapter_no, "%.") and true or false

    local infobox = mw.html.create('table')
    infobox:addClass('infobox')

    -- Add title as heading
    infobox:tag('tr')
        :tag('th')
            :attr('colspan', '2')
            :css('text-align', 'center')
            :css('font-size', '125%')
            :css('padding', '5px')
            :wikitext(chapterTitle)
            :done()
        :done()
    
    -- Add subtitle for intermission chapters
    if is_intermission then
        infobox:tag('tr')
            :tag('th')
                :attr('colspan', '2')
                :css('text-align', 'center')
                :css('font-size', '110%')
                :css('padding', '2px')
                :css('background-color', '#f0f0f0')
                :css('font-style', 'italic')
                :wikitext('(Intermission)')
                :done()
            :done()
    end

    -- Optionally add image with standard size
    if image and image ~= "" then
        infobox:tag('tr')
            :tag('td')
                :attr('colspan', '2')
                :css('text-align', 'center')
                :wikitext('[[File:' .. image .. '|250px]]')
                :done()
            :done()

        -- Add caption below image if provided
        if caption then
            infobox:tag('tr')
                :tag('td')
                    :attr('colspan', '2')
                    :css('text-align', 'center')
                    :css('font-size', '90%')
                    :css('padding', '5px')
                    :wikitext(caption)
                    :done()
                :done()
        end
    end

    -- Add chapter number and total chapters on the same line
    local chapter_info = string.format("%s of %s [[Template:ChapterList|(list of chapters)]]", chapter_no, total_chapters)
    infobox:tag('tr')
        :tag('th'):wikitext('Chapter'):done()
        :tag('td'):wikitext(chapter_info):done()
        :done()

    infobox:tag('tr')
        :tag('th'):wikitext('Date'):done()
        :tag('td'):wikitext(p.formatDate(date)):done()
        :done()

    infobox:tag('tr')
        :tag('th'):wikitext('Page count'):done()
        :tag('td'):wikitext(pagecount):done()
        :done()

    -- Add additional manual entry parameters
    if major_characters and major_characters ~= "" then
        infobox:tag('tr')
            :tag('th'):wikitext('Major characters'):done()
            :tag('td'):wikitext(major_characters):done()
            :done()
    end

    if minor_characters and minor_characters ~= "" then
        infobox:tag('tr')
            :tag('th'):wikitext('Minor characters'):done()
            :tag('td'):wikitext(minor_characters):done()
            :done()
    end

    if locations and locations ~= "" then
        infobox:tag('tr')
            :tag('th'):wikitext('Locations'):done()
            :tag('td'):wikitext(locations):done()
            :done()
    end

    if collected_in and collected_in ~= "" then
        infobox:tag('tr')
            :tag('th'):wikitext('Collected in'):done()
            :tag('td'):wikitext(collected_in):done()
            :done()
    end

    -- Add chapter navigation heading
    infobox:tag('tr')
        :tag('th')
            :attr('colspan', '2')
            :css('text-align', 'center')
            :css('font-size', '110%')
            :css('padding', '5px')
            :css('background-color', '#f0f0f0')
            :wikitext('Chapter navigation')
            :done()
        :done()

    -- Add chapter navigation links in separate columns
    local nav_row = infobox:tag('tr')
    nav_row:tag('td')
        :css('text-align', 'left')
        :wikitext(prev_chapter ~= "N/A" and string.format("[[%s:%s|← %s]]", mw.title.getCurrentTitle().nsText, prev_chapter, prev_chapter) or "← Previous")
        :done()
    nav_row:tag('td')
        :css('text-align', 'right')
        :wikitext(next_chapter ~= "N/A" and string.format("[[%s:%s|%s →]]", mw.title.getCurrentTitle().nsText, next_chapter, next_chapter) or "Next →")
        :done()
    nav_row:done()

    -- Return the complete wikitext
    return frame:preprocess(tostring(infobox))
end

return p