// ==UserScript==
// @name         SN's GLB Trains & BTs Remaining
// @namespace    SeattleNiner
// @description  Track career BT goals with a warning before it's too late.
// @version      6.3
// @match        https://glb.warriorgeneral.com/game/training.pl?*
// @license MIT
// @grant        none
// @downloadURL https://update.greasyfork.icu/scripts/559959/SN%27s%20GLB%20Trains%20%20BTs%20Remaining.user.js
// @updateURL https://update.greasyfork.icu/scripts/559959/SN%27s%20GLB%20Trains%20%20BTs%20Remaining.meta.js
// ==/UserScript==

(function() {
    'use strict';

    const PLATEAU_AGE = 160;
    const DAYS_PER_SEASON = 40;
    const DAILY_TP = 4;
    const ROLLOVER_TP = 36;
    const EQ_COLOR = "#0000ff";

    const CAPS = [
        48.06, 60.51, 67.97, 73.24, 77.28, 80.53, 83.25, 85.58, 87.60,
        89.40, 91.01, 92.46, 93.79, 95.00, 96.13, 97.18, 98.15, 99.06, 99.99
    ];

    async function fetchData(pId, force = false) {
        const sessionKey = `glb_session_data_${pId}`;
        const cached = sessionStorage.getItem(sessionKey);
        if (cached && !force) return JSON.parse(cached);

        const response = await fetch('https://glb.warriorgeneral.com/game/home.pl');
        const html = await response.text();
        const doc = new DOMParser().parseFromString(html, "text/html");

        const sMatch = (doc.getElementById('season')?.innerText || "").match(/Season\s*(\d+),\s*Day\s*(-?\d+)/i);
        const playerLink = doc.querySelector(`a[href*="player_id=${pId}"]`);
        const row = playerLink ? playerLink.closest('tr') : null;
        const ageCells = row ? row.querySelectorAll('.list_lv') : [];

        const results = {
            day: sMatch ? parseInt(sMatch[2]) : 0,
            season: sMatch ? parseInt(sMatch[1]) : 0,
            age: ageCells.length > 1 ? parseInt(ageCells[1].innerText) : 0
        };
        sessionStorage.setItem(sessionKey, JSON.stringify(results));
        return results;
    }

    async function init(forceRefresh = false) {
        const pId = new URLSearchParams(window.location.search).get('player_id');
        const container = document.getElementById('training_settings');
        if (!pId || !container) return;

        const oldBar = document.getElementById('sn_bt_bar');
        if (oldBar) oldBar.remove();

        handleAttributeDisplay();

        const bar = document.createElement('div');
        bar.id = 'sn_bt_bar';
        bar.style = `background:#222; color:#fff; padding:4px 10px; margin-bottom:8px; border-radius:3px; font-weight:bold; font-size:11px; display:flex; justify-content:space-between; align-items:center; border: 1px solid #444; clear:both;`;
        bar.innerHTML = `<span>Syncing...</span>`;
        container.prepend(bar);

        try {
            const data = await fetchData(pId, forceRefresh);
            const bankedTP = parseInt(document.querySelector('#training_points_box .points')?.innerText || 0);
            const currentBT = parseInt(document.querySelector('a[href*="bonus_tokens.pl"]')?.innerText || 0);
            const goal = parseInt(localStorage.getItem(`glb_bt_goal_${pId}`) || 0);

            let futureTP = 0;
            let simAge = data.age;
            let currentDay = data.day;

            // --- IMPROVED CAREER LOGIC ---
            // 1. Handle current season remaining days
            if (currentDay >= 1 && currentDay <= 40) {
                let daysLeft = 40 - currentDay;
                futureTP += (daysLeft * DAILY_TP);
                simAge += daysLeft;
                // If we finished the season, we move to the next season's rollover
                if (simAge < PLATEAU_AGE) futureTP += ROLLOVER_TP;
            }
            else if (currentDay > 40) {
                // Postseason: Age is capped at 40/80/120, but Rollover hasn't happened yet
                if (simAge < PLATEAU_AGE) futureTP += ROLLOVER_TP;
            }
            else if (currentDay <= 0) {
                // Preseason: D-4 has passed, Rollover TP is likely already in the bank
                // No immediate TP or Aging added; loop starts from Day 1 of next season
            }

            // 2. Loop for all subsequent full seasons
            while (simAge < (PLATEAU_AGE - 39)) { // Move in 40-day blocks
                futureTP += (DAYS_PER_SEASON * DAILY_TP);
                simAge += DAYS_PER_SEASON;
                if (simAge < PLATEAU_AGE) futureTP += ROLLOVER_TP;
            }

            const totalTrains = Math.floor((bankedTP + futureTP) / 2);
            const minBT = currentBT + (totalTrains * 2);
            const maxBT = currentBT + (totalTrains * 6);

            let bg = (maxBT < goal) ? "#000" : (maxBT < (goal + (goal * 0.1)) ? "#dc3545" : (minBT >= goal ? "#28a745" : "#ffc107"));
            if (goal === 0) bg = "#007bff";

            bar.style.background = bg;
            bar.style.color = (bg === "#ffc107") ? "#000" : "#fff";
            bar.innerHTML = `
                <span>S${data.season} D${data.day} | Age: ${data.age}d | ${totalTrains} trains left (${minBT}-${maxBT} BT)</span>
                <div style="display:flex; align-items:center; gap:8px;">
                    <button id="refresh_sn" type="button" style="height:18px; font-size:9px; cursor:pointer; background:#444; color:#fff; border:1px solid #666; border-radius:2px;">Refresh</button>
                    Goal: <input type="number" id="set_goal" value="${goal||''}" style="width:40px; height:16px; font-size:10px; border:none; padding:0 2px;">
                    <button id="save_bt" type="button" style="height:18px; font-size:9px; cursor:pointer; background:#eee; color:#000; border:none; border-radius:2px;">Set Goal</button>
                </div>
            `;

            document.getElementById('save_bt').onclick = () => {
                localStorage.setItem(`glb_bt_goal_${pId}`, document.getElementById('set_goal').value);
                window.location.reload();
            };
            document.getElementById('refresh_sn').onclick = () => init(true);

        } catch (e) { bar.innerHTML = "Sync Error."; }
    }

    function handleAttributeDisplay() {
        const statValueCells = document.querySelectorAll('.stat_value_tall, .stat_value_tall_boosted');
        statValueCells.forEach(cell => {
            const rawText = cell.innerText;
            const match = rawText.match(/([\d.]+)\s*\(\+(\d+)\)/);
            let baseNum, bonusText = "";
            if (match) {
                const total = parseFloat(match[1]);
                const bonus = parseInt(match[2]);
                baseNum = parseFloat((total - bonus).toFixed(2));
                bonusText = ` <span style="color:${EQ_COLOR}; font-weight:normal; font-size:10px; vertical-align:middle;">+${bonus}</span>`;
            } else { baseNum = parseFloat(rawText); }
            if (!isNaN(baseNum)) {
                let color = "inherit";
                for (let cap of CAPS) {
                    if (baseNum >= (cap - 2.0) && baseNum <= cap) { color = "#dc3545"; break; }
                }
                cell.style.color = color;
                cell.style.fontWeight = "bold";
                cell.innerHTML = `${baseNum.toFixed(2)}${bonusText}`;
            }
        });
    }

    init();
})();