// ==UserScript==
// @name         [Pokechill] 队伍编辑
// @namespace    https://play-pokechill.github.io/
// @version      1.1
// @description  队伍编辑，特性包含所有可记忆的（含道具学习），技能仅排除招牌，中文显示，支持搜索
// @author       人民当家做主
// @license      MIT
// @icon         https://play-pokechill.github.io/img/icons/icon.png
// @match        https://play-pokechill.github.io/*
// @match        https://g1tyx.github.io/play-pokechill/*
// @require      https://cdn.jsdelivr.net/npm/fuse.js@7.1.0
// @grant        none
// ==/UserScript==

(function() {
    'use strict';

    // ---------- 获取中文翻译 ----------
    function getChineseName(id) {
        if (!id) return '';
        const engName = typeof format === 'function' ? format(id) : id;
        if (window.EN_CN_DICT && window.EN_CN_DICT[engName]) {
            return window.EN_CN_DICT[engName];
        }
        return engName;
    }

    // ---------- 等待游戏核心加载 ----------
    function waitForGame() {
        if (typeof saved === 'undefined' || typeof pkmn === 'undefined' || typeof move === 'undefined' || typeof ability === 'undefined' || typeof item === 'undefined') {
            setTimeout(waitForGame, 300);
            return;
        }
        if (!window.EN_CN_DICT) {
            setTimeout(waitForGame, 300);
            return;
        }
        initTeamEditor();
    }

    // ---------- 初始化 ----------
    function initTeamEditor() {
        addMenuButton();
        createEditorPanel();
        bindEvents();
    }

    // ---------- 添加菜单按钮 ----------
    function addMenuButton() {
        const menuItems = document.getElementById('menu-items');
        if (!menuItems) return;
        if (document.getElementById('team-editor-menu-btn')) return;

        const btn = document.createElement('div');
        btn.id = 'team-editor-menu-btn';
        btn.className = 'menu-item';
        btn.innerHTML = `
            <img src="img/items/fashionCase.png" style="image-rendering:pixelated">
            <span>队伍编辑</span>
        `;
        btn.addEventListener('click', (e) => {
            e.stopPropagation();
            openTeamEditor();
        });
        menuItems.appendChild(btn);
    }

    // ---------- 创建编辑器面板 ----------
    function createEditorPanel() {
        if (document.getElementById('team-editor-panel')) return;

        const panel = document.createElement('div');
        panel.id = 'team-editor-panel';
        panel.style.cssText = `
            position: fixed;
            top: 0;
            left: 0;
            width: 100%;
            height: 100%;
            background: rgba(0,0,0,0.85);
            z-index: 10000;
            display: none;
            justify-content: center;
            align-items: center;
            backdrop-filter: blur(5px);
        `;

        panel.innerHTML = `
            <div class="team-editor-container" style="
                background: var(--dark2);
                width: 90%;
                max-width: 1200px;
                height: 90%;
                border-radius: 1rem;
                padding: 1rem;
                display: flex;
                flex-direction: column;
                gap: 1rem;
                border: 2px solid var(--light1);
                box-shadow: 0 0 30px rgba(0,0,0,0.7);
                color: white;
                overflow: hidden;
            ">
                <div style="display: flex; justify-content: space-between; align-items: center;">
                    <h2 style="color: var(--light2); margin:0;">✎ 队伍编辑 (当前队伍: <span id="editor-current-team-name">队伍1</span>)</h2>
                    <button id="team-editor-close" style="background: #c45; border:none; color:white; font-size:1.5rem; width:2.5rem; height:2.5rem; border-radius:0.5rem; cursor:pointer;">✕</button>
                </div>
                <div id="team-editor-slots" style="
                    display: grid;
                    grid-template-columns: repeat(auto-fill, minmax(320px, 1fr));
                    gap: 1rem;
                    overflow-y: auto;
                    padding: 0.5rem;
                ">
                    <!-- 动态生成6个槽位 -->
                </div>
                <div style="display: flex; gap:1rem; justify-content: flex-end; margin-top:0.5rem;">
                    <button id="team-editor-refresh" style="background:var(--light1); color:white; border:none; padding:0.5rem 2rem; border-radius:0.5rem; cursor:pointer;">刷新显示</button>
                    <button id="team-editor-save" style="background:#4caf50; color:white; border:none; padding:0.5rem 2rem; border-radius:0.5rem; cursor:pointer;">保存修改</button>
                </div>
            </div>
        `;

        document.body.appendChild(panel);
    }

    // ---------- 打开编辑器 ----------
    function openTeamEditor() {
        const panel = document.getElementById('team-editor-panel');
        if (!panel) return;
        if (typeof closeTooltip === 'function') closeTooltip();
        populateTeamSlots();
        panel.style.display = 'flex';
    }

    // ---------- 关闭编辑器 ----------
    function closeTeamEditor() {
        const panel = document.getElementById('team-editor-panel');
        if (panel) panel.style.display = 'none';
    }

    // ---------- 填充6个槽位 ----------
    function populateTeamSlots() {
        const container = document.getElementById('team-editor-slots');
        if (!container) return;

        const teamName = saved.currentPreviewTeam || 'preview1';
        const team = saved.previewTeams[teamName];
        if (!team) {
            container.innerHTML = '<div style="color:red;">无法读取队伍数据</div>';
            return;
        }

        document.getElementById('editor-current-team-name').innerText = teamName.replace('preview', '队伍');

        let html = '';
        for (let i = 1; i <= 6; i++) {
            const slot = team[`slot${i}`];
            const pokeId = slot?.pkmn;
            const pokeData = pokeId ? pkmn[pokeId] : null;

            html += `
                <div class="team-editor-slot" data-slot-index="${i}" data-poke-id="${pokeId || ''}" style="
                    background: var(--dark1);
                    border-radius: 0.8rem;
                    padding: 1rem;
                    display: flex;
                    flex-direction: column;
                    gap: 0.5rem;
                    border: 1px solid var(--light1);
                    opacity: ${pokeData ? 1 : 0.5};
                ">
                    <div style="display: flex; align-items: center; gap: 1rem;">
                        <div style="width: 3rem; height: 3rem; background: var(--dark2); border-radius:0.5rem; display:flex; justify-content:center; align-items:center;">
                            ${pokeData ? `<img src="img/pkmn/sprite/${pokeId}.png" style="max-width:100%; max-height:100%; image-rendering:pixelated;">` : '❌'}
                        </div>
                        <div style="flex:1;">
                            <div><strong>${pokeData ? getChineseName(pokeId) : '空'}</strong> <span style="font-size:0.8rem; opacity:0.7;">${pokeId || ''}</span></div>
                            ${pokeData ? `
                                <div style="display:flex; gap:1rem; margin-top:0.3rem;">
                                    <span>等级</span>
                                    <input type="number" class="editor-level" data-slot="${i}" value="${pokeData.level || 1}" min="1" max="100" style="width:5rem; background:var(--dark2); color:white; border:none; border-radius:0.3rem; padding:0.2rem;">
                                </div>
                            ` : ''}
                        </div>
                    </div>
                    ${pokeData ? `
                        <div class="editor-ability-row" style="display:flex; align-items:center; gap:0.5rem;">
                            <span style="min-width:3.5rem;">特性</span>
                            <input type="text" class="editor-ability-input" data-slot="${i}" value="${getChineseName(pokeData.ability)}" readonly placeholder="点击选择" style="flex:1; background:var(--dark2); color:white; border:none; border-radius:0.3rem; padding:0.3rem; cursor:pointer;">
                            <button class="editor-ability-btn" data-slot="${i}" style="background:var(--light1); border:none; color:white; border-radius:0.3rem; padding:0.3rem 0.8rem; cursor:pointer;">搜索</button>
                        </div>
                        <div class="editor-moves" style="display:grid; grid-template-columns:1fr 1fr; gap:0.5rem;">
                            ${[1,2,3,4].map(m => {
                                const moveId = pokeData.moves?.[`slot${m}`];
                                return `
                                    <div class="pkmn-movebox" style="
                                        height: 2rem;
                                        width: 100%;
                                        display: flex;
                                        align-items: center;
                                        border: 2px solid var(--type-dark);
                                        background: var(--dark1);
                                        color: white;
                                        border-radius: 0.3rem;
                                        position: relative;
                                        cursor: help;
                                        overflow: hidden;
                                        padding-left: 0.5rem;
                                    ">
                                        <span style="z-index:2; flex:1;">${moveId ? getChineseName(moveId) : '--'}</span>
                                        <img src="img/icons/${moveId ? move[moveId].type : 'normal'}.svg" style="
                                            height: 2rem;
                                            width: 2rem;
                                            background-color: ${moveId ? 'var(--type-'+move[moveId].type+')' : '#A0A29F'};
                                            padding: 4px;
                                            z-index: 2;
                                        ">
                                        <input type="hidden" class="editor-move-value" data-slot="${i}" data-move-slot="${m}" value="${moveId || ''}">
                                        <button class="editor-move-btn" data-slot="${i}" data-move-slot="${m}" style="
                                            position: absolute;
                                            width: 100%;
                                            height: 100%;
                                            background: transparent;
                                            border: none;
                                            cursor: pointer;
                                            z-index: 3;
                                        "></button>
                                    </div>
                                `;
                            }).join('')}
                        </div>
                    ` : '<div style="color:#888; text-align:center;">空位，无法编辑</div>'}
                </div>
            `;
        }

        container.innerHTML = html;

        // 绑定事件
        container.querySelectorAll('.editor-ability-btn').forEach(btn => {
            btn.addEventListener('click', (e) => {
                e.stopPropagation();
                const slot = btn.dataset.slot;
                showAbilitySearch(slot);
            });
        });

        container.querySelectorAll('.editor-move-btn').forEach(btn => {
            btn.addEventListener('click', (e) => {
                e.stopPropagation();
                const slot = btn.dataset.slot;
                const moveSlot = btn.dataset.moveSlot;
                showMoveSearch(slot, moveSlot);
            });
        });

        container.querySelectorAll('.editor-level').forEach(input => {
            input.addEventListener('change', function() {
                const slot = this.dataset.slot;
                const val = parseInt(this.value, 10);
                updatePokemonLevel(slot, val);
            });
        });

        container.querySelectorAll('.editor-ability-input').forEach(inp => {
            inp.addEventListener('click', (e) => {
                e.stopPropagation();
                const slot = inp.dataset.slot;
                showAbilitySearch(slot);
            });
        });
    }

    // ---------- 更新等级 ----------
    function updatePokemonLevel(slot, newLevel) {
        const teamName = saved.currentPreviewTeam || 'preview1';
        const team = saved.previewTeams[teamName];
        const slotKey = `slot${slot}`;
        const pokeId = team[slotKey]?.pkmn;
        if (!pokeId) return;
        const poke = pkmn[pokeId];
        if (!poke) return;
        poke.level = Math.min(100, Math.max(1, newLevel));
    }

    // ---------- 构建可记忆特性集合 ----------
    function getMemoryAbilities() {
        const memorySet = new Set();
        for (let itemId in item) {
            const it = item[itemId];
            if (it.type === 'memory' && it.ability) {
                memorySet.add(it.ability);
            }
        }
        return memorySet;
    }

    // ---------- 显示特性搜索（包含所有可记忆特性） ----------
    function showAbilitySearch(slot) {
        const teamName = saved.currentPreviewTeam || 'preview1';
        const team = saved.previewTeams[teamName];
        const slotKey = `slot${slot}`;
        const pokeId = team[slotKey]?.pkmn;
        if (!pokeId) return;
        const poke = pkmn[pokeId];
        if (!poke) return;

        const memoryAbilities = getMemoryAbilities();
        const abilityList = [];

        for (let abId in ability) {
            const ab = ability[abId];
            // 条件：普通特性（type存在） 或者 是可记忆特性（即使type不存在）
            if (ab.type || memoryAbilities.has(abId)) {
                abilityList.push({
                    id: abId,
                    name: getChineseName(abId),
                    data: ab
                });
            }
        }

        const fuse = new Fuse(abilityList, {
            keys: ['name', 'id'],
            threshold: 0.3,
            includeScore: true
        });

        showSearchPopup('选择特性', abilityList, fuse, (selectedId) => {
            poke.ability = selectedId;
            if (poke.hiddenAbility && poke.hiddenAbility.id === selectedId) {
                poke.hiddenAbilityUnlocked = true;
            }
            const input = document.querySelector(`.editor-ability-input[data-slot="${slot}"]`);
            if (input) input.value = getChineseName(selectedId);
        });
    }

    // ---------- 显示技能搜索（仅排除招牌技能） ----------
    function showMoveSearch(slot, moveSlot) {
        const teamName = saved.currentPreviewTeam || 'preview1';
        const team = saved.previewTeams[teamName];
        const slotKey = `slot${slot}`;
        const pokeId = team[slotKey]?.pkmn;
        if (!pokeId) return;
        const poke = pkmn[pokeId];
        if (!poke) return;

        const moveList = [];
        for (let moveId in move) {
            const mv = move[moveId];
            // 仅排除招牌技能（moveset 为 undefined 的技能）
            if (!mv.moveset) continue;
            moveList.push({
                id: moveId,
                name: getChineseName(moveId),
                data: mv
            });
        }

        const fuse = new Fuse(moveList, {
            keys: ['name', 'id'],
            threshold: 0.3,
            includeScore: true
        });

        showSearchPopup(`选择技能 (槽位${moveSlot})`, moveList, fuse, (selectedId) => {
            if (!poke.movepool.includes(selectedId)) {
                poke.movepool.push(selectedId);
            }
            poke.moves[`slot${moveSlot}`] = selectedId;
            // 更新显示
            const container = document.querySelector(`.team-editor-slot[data-slot-index="${slot}"] .editor-moves`);
            if (container) {
                const moveDivs = container.querySelectorAll('.pkmn-movebox');
                if (moveDivs[moveSlot-1]) {
                    moveDivs[moveSlot-1].querySelector('span').textContent = getChineseName(selectedId);
                    const img = moveDivs[moveSlot-1].querySelector('img');
                    img.src = `img/icons/${move[selectedId].type}.svg`;
                    img.style.backgroundColor = `var(--type-${move[selectedId].type})`;
                }
            }
        });
    }

    // ---------- 通用搜索弹窗 ----------
    function showSearchPopup(title, items, fuseInstance, onSelect) {
        const existing = document.getElementById('team-editor-search-popup');
        if (existing) existing.remove();

        const popup = document.createElement('div');
        popup.id = 'team-editor-search-popup';
        popup.style.cssText = `
            position: fixed;
            top: 50%;
            left: 50%;
            transform: translate(-50%, -50%);
            width: 400px;
            max-width: 90%;
            max-height: 80%;
            background: var(--dark1);
            border: 2px solid var(--light1);
            border-radius: 1rem;
            padding: 1rem;
            z-index: 20000;
            display: flex;
            flex-direction: column;
            gap: 0.8rem;
            color: white;
            box-shadow: 0 0 30px black;
        `;

        popup.innerHTML = `
            <div style="display: flex; justify-content: space-between;">
                <h3>${title}</h3>
                <button id="search-popup-close" style="background:#c45; border:none; color:white; width:2rem; height:2rem; border-radius:0.5rem;">✕</button>
            </div>
            <input type="text" id="search-popup-input" placeholder="输入中文或英文名搜索..." style="width:100%; padding:0.5rem; background:var(--dark2); color:white; border:1px solid var(--light1); border-radius:0.5rem;">
            <div id="search-popup-results" style="overflow-y: auto; max-height: 300px; display: flex; flex-direction: column; gap: 0.3rem;"></div>
        `;

        document.body.appendChild(popup);

        const input = document.getElementById('search-popup-input');
        const resultsDiv = document.getElementById('search-popup-results');
        const closeBtn = document.getElementById('search-popup-close');

        function renderResults(list) {
            resultsDiv.innerHTML = '';
            if (!list.length) {
                resultsDiv.innerHTML = '<div style="color:#888; text-align:center;">无结果</div>';
                return;
            }
            list.slice(0, 50).forEach(item => {
                const div = document.createElement('div');
                div.style.cssText = `
                    padding: 0.5rem;
                    background: var(--dark2);
                    border-radius: 0.3rem;
                    cursor: pointer;
                    display: flex;
                    justify-content: space-between;
                `;
                div.innerHTML = `<span>${item.name}</span><span style="opacity:0.6;">${item.id}</span>`;
                div.addEventListener('click', () => {
                    onSelect(item.id);
                    popup.remove();
                });
                resultsDiv.appendChild(div);
            });
        }

        renderResults(items);

        input.addEventListener('input', () => {
            const query = input.value.trim();
            if (!query) {
                renderResults(items);
                return;
            }
            const results = fuseInstance.search(query).map(r => r.item);
            renderResults(results);
        });

        closeBtn.addEventListener('click', () => popup.remove());

        const onKeyDown = (e) => {
            if (e.key === 'Escape') popup.remove();
        };
        window.addEventListener('keydown', onKeyDown, { once: true });
    }

    // ---------- 绑定事件 ----------
    function bindEvents() {
        document.addEventListener('click', (e) => {
            const closeBtn = e.target.closest('#team-editor-close');
            if (closeBtn) closeTeamEditor();

            const refreshBtn = e.target.closest('#team-editor-refresh');
            if (refreshBtn) populateTeamSlots();

            const saveBtn = e.target.closest('#team-editor-save');
            if (saveBtn) {
                if (typeof updatePreviewTeam === 'function') updatePreviewTeam();
                closeTeamEditor();
            }
        });
    }

    waitForGame();
})();