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

local p = {}

-- Define the volume ranges and styles
local volumes = {
    {
        name = 'Volume One',
        start = 1,
        end_ = 46,
        style = 'background:#707ebd;color:white;',
        group = 'group1',
        list = 'list1',
    },
    {
        name = 'Volume Two',
        start = 47,
        end_ = 58,
        style = 'background:#912f28;color:white;',
        group = 'group2',
        list = 'list2',
    },
    {
        name = 'Volume Three',
        start = 59,
        end_ = 69.1,
        style = 'background:#2b8842;color:white;',
        group = 'group3',
        list = 'list3',
    },
    {
        name = 'Volume Four',
        start = 70,
        end_ = 80,
        style = 'background:#4589c8;color:white;',
        group = 'group4',
        list = 'list4',
    },
    {
        name = 'Volume Five',
        start = 81,
        end_ = 88,
        style = 'background:#aa715e;color:white;',
        group = 'group5',
        list = 'list5',
    },
    {
        name = 'Volume Six',
        start = 89,
        end_ = 96.1,
        style = 'background:#f27191;color:white;',
        group = 'group6',
        list = 'list6',
    },
    {
        name = 'Volume Seven',
        start = 97,
        end_ = 110,
        style = 'background:#efb63b;color:white;',
        group = 'group7',
        list = 'list7',
    },
}

-- Manually defined book-exclusive chapters
local bookExclusiveChapters = {
    {title = "Curtain Rising", link = "Curtain_Rising"},
    {title = "No Service", link = "No_Service"},
    {title = "Critical Eye", link = "Critical_Eye"},
    {title = "Payback", link = "Payback"},
    {title = "Fashion Disaster", link = "Fashion_Disaster"},
    {title = "There’s a Chance", link = "There’s_a_Chance"},
    {title = "Afterglow", link = "Afterglow"},
    {title = "Unpacking Day", link = "Unpacking_Day"},
    {title = "Good Egg", link = "Good_Egg"},
}

-- Function to check if a chapter is within a volume range
local function isChapterInVolume(chapter_no, start, end_)
    local number = tonumber(chapter_no)
    return number and number >= start and number <= end_
end

function p.generateNavbox(frame)
    local chaptersModule = require('Module:Chapters')
    local chapters = chaptersModule.getChaptersList()

    local navboxParams = {
        bodyclass = 'hlist',
        name = 'Bittersweet Candy Bowl chapters',
        title = "[[Bittersweet Candy Bowl]] chapters",
        titlestyle = '',
        image = '',
        above = '',
        below = ''
    }

    -- Add volume groups and lists
    for _, volume in ipairs(volumes) do
        navboxParams[volume.group .. 'style'] = volume.style
        navboxParams[volume.group] = volume.name

        local list = {}
        for _, chapter in ipairs(chapters) do
            if isChapterInVolume(chapter.number, volume.start, volume.end_) then
                table.insert(list, '* [[' .. chapter.title .. ']]')
            end
        end
        navboxParams[volume.list] = table.concat(list, '\n')
    end

    -- Add group for chapters not in print without a style
    navboxParams['group8'] = 'Not yet in print'

    local list_not_in_print = {}
    for _, chapter in ipairs(chapters) do
        local number = tonumber(chapter.number)
        local in_volume = false
        for _, volume in ipairs(volumes) do
            if isChapterInVolume(chapter.number, volume.start, volume.end_) then
                in_volume = true
                break
            end
        end

        if not in_volume then
            table.insert(list_not_in_print, '* [[' .. chapter.title .. ']]')
        end
    end
    navboxParams['list8'] = table.concat(list_not_in_print, '\n')

    -- Add manual group for book-exclusive chapters
    navboxParams['group9'] = 'Book-exclusive'
    local book_exclusive_list = {}
    for _, chapter in ipairs(bookExclusiveChapters) do
        table.insert(book_exclusive_list, '* [[' .. chapter.link .. '|' .. chapter.title .. ']]')
    end
    navboxParams['list9'] = table.concat(book_exclusive_list, '\n')

    -- Add "Obsolete" section
    navboxParams['group10'] = 'Obsolete'
    navboxParams['list10'] = '* [[List of obsolete chapters]]'

    return frame:expandTemplate{ title = 'Navbox', args = navboxParams }
end

return p