content.js

function fixGraph() {
    const graph = document.querySelector('.js-calendar-graph');
    if (!graph) return;

    const days = Array.from(graph.querySelectorAll('[data-level]'));
    if (days.length === 0) return;

    let max = 0;
    const data = days.map(day => {
        let count = 0;

        // Strategy 1: data-count (Legacy/Certain views)
        if (day.hasAttribute('data-count')) {
            count = parseInt(day.getAttribute('data-count'), 10);
        }
        // Strategy 2: aria-labelledby (Modern view - Tooltip based)
        else if (day.hasAttribute('aria-labelledby')) {
            const tooltipId = day.getAttribute('aria-labelledby');
            const tooltipEl = document.getElementById(tooltipId);
            if (tooltipEl) {
                const text = tooltipEl.textContent;
                const match = text.match(/(\d+)\s+contribution/i);
                if (match) {
                    count = parseInt(match[1], 10);
                } else if (/No\s+contribution/i.test(text)) {
                    count = 0;
                }
            }
        }
        // Strategy 3: Parse aria-label or text content directly
        else {
            const text = day.textContent || day.getAttribute('aria-label') || "";
            const match = text.match(/(\d+)\s+contribution/i);
            if (match) {
                count = parseInt(match[1], 10);
            }
        }

        if (isNaN(count)) count = 0;
        if (count > max) max = count;
        return { element: day, count };
    });

    if (max === 0) return;

    data.forEach(({ element, count }) => {
        if (count === 0) {
            element.setAttribute('data-level', '0');
            return;
        }

        // Logarithmic scale implementation
        const logValue = Math.log(count + 1);
        const logMax = Math.log(max + 1);
        const ratio = logValue / logMax;

        // Scale to range 1-4
        let level = Math.ceil(ratio * 4);
        if (level < 1) level = 1;
        if (level > 4) level = 4;

        element.setAttribute('data-level', level.toString());
    });
}

// Initial run
fixGraph();

// Handle dynamic navigation (GitHub uses Turbo/Pjax)
let timeout = null;
const observer = new MutationObserver((mutations) => {
    if (timeout) clearTimeout(timeout);
    timeout = setTimeout(() => {
        fixGraph();
    }, 300);
});

observer.observe(document.body, { childList: true, subtree: true });