helpers.lua

local helpers = {
    fontHeading = love.graphics.newFont(42),
    fontSubheading = love.graphics.newFont(32),
    fontBody = love.graphics.newFont(24),

    bodyOffset = 130,
    codeContext = {},
    codeLeft = false,
    codeLeftAlign = 50,
}

function helpers.reset()
    love.graphics.setColor(Theme.Text)
    love.graphics.setFont(helpers.fontBody)
    helpers.bodyOffset = 130
end

function helpers.heading(text)
    helpers.reset()
    love.graphics.setColor(Theme.Text)
    love.graphics.setFont(helpers.fontHeading)
    love.graphics.printf(text, 0, 50, love.graphics.getWidth(), "center")
end

function helpers.subheading(text)
    helpers.bodyOffset = helpers.bodyOffset + 20
    love.graphics.setColor(Theme.Text)
    love.graphics.setFont(helpers.fontSubheading)
    love.graphics.printf(text, helpers.codeLeft and helpers.codeLeftAlign or 0, helpers.bodyOffset, love.graphics.getWidth(), helpers.codeLeft and "left" or "center")
    helpers.bodyOffset = helpers.bodyOffset + 50
end

local colorMap = {
    B = Theme.Blue,
    G = Theme.Green,
    R = Theme.Red,
    Y = Theme.Yellow,
    W = Theme.Text,
}

function helpers.body(text)
    local text = tostring(text)
    love.graphics.setFont(helpers.fontBody)

    local defaultColor = Theme.Text
    local segments = {}

    local i = 1
    local currentColor = defaultColor

    while i <= #text do
        local s, e, code = text:find("%$([A-Z])", i)

        if not s then
            table.insert(segments, {
                text = text:sub(i),
                color = currentColor
            })
            break
        end

        if s > i then
            table.insert(segments, {
                text = text:sub(i, s - 1),
                color = currentColor
            })
        end

        currentColor = colorMap[code] or defaultColor
        i = e + 1
    end

    local x
    if helpers.codeLeft then
        x = helpers.codeLeftAlign
    else
        local totalWidth = 0
        for _, segment in ipairs(segments) do
            totalWidth = totalWidth + helpers.fontBody:getWidth(segment.text)
        end
        x = (love.graphics.getWidth() - totalWidth) / 2
    end

    local y = helpers.bodyOffset

    for _, segment in ipairs(segments) do
        love.graphics.setColor(segment.color)
        love.graphics.print(segment.text, x, y)
        x = x + helpers.fontBody:getWidth(segment.text)
    end

    love.graphics.setColor(Theme.Text)
    helpers.bodyOffset = helpers.bodyOffset + 30
end

local function Set(list)
    local set = {}
    for _, l in ipairs(list) do set[l] = true end
    return set
end

function helpers.drawWord(text, x, y)
    local keywords = Set{"function", "end", "if", "then", "else", "elseif", "for", "in", "do", "while", "repeat", "until", "return"}
    
    local color = Theme.Variable
    if keywords[text] then
        color = Theme.Keyword
    end

    if text == "(" and helpers.codeContext[#helpers.codeContext] == "function" then
        table.remove(helpers.codeContext)
    end
    if helpers.codeContext[#helpers.codeContext] == "function" then
        color = Theme.Function
    end
    if text == "function" then
        table.insert(helpers.codeContext, "function")
    elseif text:sub(1, 2) == "--" then
        love.graphics.setColor(Theme.Comment)
        love.graphics.print(text, x, y)
        return
    elseif text == "local" then
        color = Theme.Local
    elseif text:match("^%d+$") then
        color = Theme.Number
    elseif (text:sub(1, 1) == "\"" and text:sub(-1) == "\"") or (text:sub(1, 1) == "'" and text:sub(-1) == "'") then
        color = Theme.String
    end
    helpers.drawCharacters(text, x, y, color)
end

function helpers.drawCharacters(text, x, y, color)
    local operators = Set{"+", "-", "*", "/", "=", "==", "~=", "<", ">", "<=", ">=", "%", "^", "#", "..", ".", ":", ",", ";", "(", ")", "{", "}", "[", "]"}
    for char in text:gmatch(".") do
        if operators[char] then
            love.graphics.setColor(Theme.Text)
        else
            love.graphics.setColor(color)
        end
        love.graphics.print(char, x, y)
        x = x + helpers.fontBody:getWidth(char)
    end
end

function helpers.code(text)
    love.graphics.setFont(helpers.fontBody)
    helpers.codeContext = {}
    local operators = Set{"+", "-", "*", "/", "=", "==", "~=", "<", ">", "<=", ">=", "%", "^", "#", "..", ".", ":", ",", ";", "(", ")", "{", "}", "[", "]"}
    local leftAlign = helpers.codeLeft and helpers.codeLeftAlign or love.graphics.getWidth() / 7 * 2
    -- draw line by line
    for line in text:gmatch("[^\n]+") do
        helpers.codeContext = {}
        x = leftAlign
        local currentToken = ""
        local inString = false
        local stringChar = nil
        
        for i = 1, #line do
            local char = line:sub(i, i)
            local nextChar = line:sub(i + 1, i + 1)
            
            -- Handle strings
            if (char == "\"" or char == "'") and not inString then
                inString = true
                stringChar = char
                if #currentToken > 0 then
                    helpers.drawWord(currentToken, x, helpers.bodyOffset)
                    x = x + helpers.fontBody:getWidth(currentToken)
                    currentToken = ""
                end
                currentToken = char
            elseif inString and char == stringChar then
                currentToken = currentToken .. char
                helpers.drawWord(currentToken, x, helpers.bodyOffset)
                x = x + helpers.fontBody:getWidth(currentToken)
                currentToken = ""
                inString = false
                stringChar = nil
            elseif inString then
                currentToken = currentToken .. char
            -- Check if we're starting a comment
            elseif char == "-" and nextChar == "-" then
                -- Draw accumulated token first
                if #currentToken > 0 then
                    helpers.drawWord(currentToken, x, helpers.bodyOffset)
                    x = x + helpers.fontBody:getWidth(currentToken)
                    currentToken = ""
                end
                -- Draw the rest of the line as a comment
                helpers.drawWord(line:sub(i), x, helpers.bodyOffset)
                break
            elseif char == " " then
                if #currentToken > 0 then
                    helpers.drawWord(currentToken, x, helpers.bodyOffset)
                    x = x + helpers.fontBody:getWidth(currentToken)
                    currentToken = ""
                end
                x = x + helpers.fontBody:getWidth(" ")
            elseif operators[char] then
                -- Draw accumulated token first
                if #currentToken > 0 then
                    helpers.drawWord(currentToken, x, helpers.bodyOffset)
                    x = x + helpers.fontBody:getWidth(currentToken)
                    currentToken = ""
                end
                -- Draw operator
                helpers.drawWord(char, x, helpers.bodyOffset)
                x = x + helpers.fontBody:getWidth(char)
            else
                currentToken = currentToken .. char
            end
        end

        if #currentToken > 0 then
            helpers.drawWord(currentToken, x, helpers.bodyOffset)
        end
        helpers.bodyOffset = helpers.bodyOffset + 30
    end
    helpers.bodyOffset = helpers.bodyOffset + 30
end

return helpers