// ==UserScript==
// @name         AlphaXiv AI Prompts (Smart Workflow)
// @name:zh-CN   AlphaXiv AI 提示词 (智能工作流版)
// @namespace    http://tampermonkey.net/
// @version      6.0
// @description  A smart, workflow-oriented prompt system for AlphaXiv, featuring "Last Used" and "Pinned" shortcuts.
// @description:zh-CN 为 AlphaXiv 打造的智能工作流提示词系统，包含“最近使用”与“快捷固定”功能，极致提升效率。
// @author       YourName
// @match        https://www.alphaxiv.org/overview/*
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_addStyle
// @icon         https://www.google.com/s2/favicons?sz=64&domain=alphaxiv.org
// @downloadURL none
// ==/UserScript==

(function() {
    'use strict';

    // --- 1. 核心配置区 ---

    // ⭐ 在这里“钉”上你最常用的提示词！
    // 将下面 `prompts` 数组中的任意 `id` 填入这里即可。
    // 建议1-3个，保持界面简洁。
    const PINNED_PROMPT_IDS = [
        'quick-chatgpt',
        'deep-gemini',
        'deep-claude'
    ];

    // 提示词数据库 (为每个 prompt 添加了唯一的 `id`)
    const MODULES = [
        { title: '① 快速通读', prompts: [
            { id: 'quick-chatgpt', type: 'ChatGPT', label: '结构化总结', value: `请帮我概括这篇论文的核心结构，包括：\n- 标题与作者单位\n- 背景与研究动机\n- 研究问题或假设\n- 方法概述（含模型/算法名称）\n- 实验与结果关键数据\n- 创新点与贡献总结（3条以内）\n- 局限与未来工作\n请使用简洁清晰的分点格式。` },
            { id: 'quick-claude', type: 'Claude', label: '导师式讲解', value: `假设你是一位讲解论文的导师，请用通俗的语言向本科生解释这篇论文在做什么。不要逐段总结，而是围绕“这项研究想解决什么问题？用了什么办法？发现了什么？”三个问题来展开，让读者在三分钟内明白这篇论文的价值。` }
        ]},
        { title: '② 方法详解', prompts: [
            { id: 'method-chatgpt', type: 'ChatGPT', label: '逐步解析', value: `请对论文中的方法部分进行逐步解析，要求包括：\n1. 每个模块的输入、输出和作用；\n2. 使用了哪些关键技术或机制（如 Transformer、GNN 等）；\n3. 如涉及数学公式，请简要通俗化解释；\n4. 用结构图或流程图式样展示整体框架。` },
            { id: 'method-claude', type: 'Claude', label: '故事化解读', value: `这篇论文的方法部分信息量很大，我想通过“讲故事”的方式理解：请像讲一个项目开发故事一样，把研究者是怎么一步步设计这个方法的讲出来，包括他们遇到什么挑战、为何选这个结构、每一步的逻辑联系。` }
        ]},
        { title: '③ 实验洞察', prompts: [
            { id: 'exp-chatgpt', type: 'ChatGPT', label: '逻辑分析', value: `请分析论文的实验部分，重点包括：\n- 所使用的数据集及其代表性；\n- 对比实验设置是否合理；\n- 评估指标是否充分；\n- 结果是否显著优于现有方法；\n- 结果支持作者的哪些结论？\n请用逻辑清晰的分析语言表达。` },
            { id: 'exp-claude', type: 'Claude', label: '审稿人视角', value: `请结合实验数据，为我讲清楚：这个模型到底好在哪里？是不是所有指标都更好？有没有反常的情况？作者是否过度解读结果了？如果你是审稿人，你会如何评价这些实验？` }
        ]},
        { title: '④ 批判思考', prompts: [
            { id: 'crit-chatgpt', type: 'ChatGPT', label: '价值评估', value: `请帮我总结该论文的主要创新点，并评估其在已有文献中的相对价值。是否为增量性工作？是否填补了空白？是否存在方法泛化性、数据依赖性等潜在问题？` },
            { id: 'crit-claude', type: 'Claude', label: '优劣思辨', value: `我想更深入思考这篇论文的价值与问题：请从“做得很好”和“还有待改进”的角度各列出 2-3 点，并结合你的推理说明这些优劣的判断依据。` }
        ]},
        { title: '⑤ 论文对比', prompts: [
            { id: 'comp-chatgpt', type: 'ChatGPT', label: '表格化对比', value: `我有多篇论文，请你帮助我比较它们在以下方面的异同：\n- 研究目标与问题设定；\n- 所用方法或模型结构；\n- 数据集/评估方式；\n- 关键结果与性能差距；\n- 各自的创新点或不足之处。\n请形成一个对比表格或分类结构化总结。` },
            { id: 'comp-claude', type: 'Claude', label: '知识演化视角', value: `这些论文之间有哪些研究思路上的“流派”？谁是改进谁的？是否可以归纳出一个共同的问题脉络或技术发展链条？请从知识演化视角帮助我看清“谁启发了谁”。` }
        ]},
        { title: '⑥ 深度精读', prompts: [
            { id: 'deep-gemini', type: 'Gemini', label: '结构化精读', value: `你是一位训练有素的科研助手，擅长跨学科论文的结构化理解、批判性分析与工程/应用层面迁移。\n\n我将向你提供一篇论文，请你严格按照以下结构，提炼核心信息，并输出清晰的 Markdown 格式研究笔记：\n\n---\n\n### 1️⃣ 研究动机与问题陈述\n- 论文试图解决的关键问题是什么？\n- 它为何重要？在该领域中处于什么位置？\n\n---\n\n### 2️⃣ 核心贡献（用简洁点列总结）\n- 列出本文的主要贡献（不少于3条）\n- 哪些是创新点，哪些是对现有工作的拓展？\n\n---\n\n### 3️⃣ 方法与框架结构\n- 描述论文提出的方法或架构的整体框架\n- 可以列表或绘制简易流程图（建议使用 Markdown 列表或图解语言）\n\n---\n\n### 4️⃣ 数据与实验设计（如适用）\n- 数据来源和处理方式？\n- 实验如何设置？评估指标和结果如何？\n\n---\n\n### 5️⃣ 优势与局限\n- 本方法/研究的亮点有哪些？\n- 是否存在局限、盲点或待改进之处？\n\n---\n\n### 6️⃣ 应用场景与迁移价值\n- 成果可应用于哪些实际问题或领域？\n- 能否迁移到其他任务、学科或系统中？\n\n---\n\n### 7️⃣ 个人思考与后续探索\n- 你对这篇论文的最大收获或疑问是什么？\n- 如果继续这项研究，你可能采取哪些方向？\n\n---\n\n📝 输出要求：\n- 用 **Markdown** 格式组织输出（适合导入 Notion / Obsidian / Docs）\n- 尽量使用自己的语言重述，不照搬原文\n- 如果有图示建议，可使用文字/列表描述流程\n- 适用于任何学科，不要假设是 AI 论文` },
            { id: 'deep-claude', type: 'Claude', label: '深度阅读', value: `你是一位深度思维型的科研与系统助理，擅长阅读复杂论文、理解其结构、提炼关键思想，并将研究成果转化为结构化笔记、可复现工程方案或进一步的探索路径。\n\n我将提供一篇论文的全文、摘要或片段，请严格按以下结构进行系统性解读，并用你自己的语言表达：\n\n---\n\n## 🧩 1. 研究动机与核心问题\n- 本文试图解决的核心问题是什么？\n- 这个问题在该领域的重要性体现在哪些方面？\n- 作者如何界定了问题的边界与挑战？\n\n## 🧱 2. 主要贡献总结（简洁有力，不少于3点）\n- 本文的关键贡献包括哪些方面？\n- 哪些属于方法/系统/实验创新，哪些属于理论或分析角度？\n\n## 🧠 3. 方法/系统结构图解\n- 简述作者提出的整体方法或系统架构，分模块说明每部分的功能\n- 如果是模型方法，请说明输入、处理机制与输出过程\n- 请用文字/伪代码/流程图或 Markdown 列表方式组织结构\n\n## 🧪 4. 数据、实验设计与结果（如适用）\n- 使用了哪些数据集？处理流程？\n- 实验设置如何？采用了哪些对比方法和指标？\n- 结果体现了哪些优势或特征？\n\n## 📊 5. 优势、局限与反思\n- 本文方法的亮点在哪里？是否有普适性？\n- 作者有没有回避或未充分讨论的问题？\n- 你是否发现可能存在的假设盲点、工程难点或方法局限？\n\n## 🚀 6. 迁移应用与启发\n- 本文的思路或方法可以迁移到哪些实际任务或领域？\n- 如果要落地为产品或系统，需要补充哪些模块？\n\n## 🔭 7. 延展思考或进一步研究方向\n- 如果你要继续这项研究，你会做哪些改进或拓展？\n- 有哪些你希望模型进一步挖掘或未被回答的问题？\n\n---\n\n📌 输出要求：\n- 用 Markdown 格式清晰组织内容\n- 每一部分都尽量用自己的语言重述，不照搬论文原话\n- 如需辅助说明结构，请用伪代码或有层级的 Markdown 列表\n- 如果输入的是 PDF OCR 或文本块，请自动识别结构内容并整理成以上结构\n\n🧭 角色提示（可选）：\n如适用，请以“工程复现视角”/“博士申请笔记”/“系统研读讲义”/“会议报告摘要”等不同目标给出推荐改写方式。` },
            { id: 'deep-grok', type: 'Grok', label: '高效解读', value: `你是一个严谨的科研助手与系统分析专家，擅长快速提炼科研论文的关键思想、结构框架、方法机制与实际意义。我将提供一篇论文内容（可为全文、摘要、截图OCR或主要段落），请你按以下结构系统地完成内容解读：\n\n---\n\n## 📘 1. 研究背景与核心问题\n- 本文研究了什么问题？为何重要？现实背景或理论缺口是什么？\n- 有哪些已有方法存在不足？本文打算如何突破？\n\n## 🧠 2. 核心贡献概括（不少于三点，点列式）\n- 本文的创新点、提出的新方法或系统\n- 与已有工作相比，在哪些维度有所改进？\n\n## 🧱 3. 方法原理 / 系统架构\n- 如果是理论/建模类：解释主要模型、推理机制、公式结构  \n- 如果是系统/工程类：模块组成、处理流程、信息流动  \n- 使用结构化形式：如 Markdown 层级列表、流程图（文本描述）或伪代码\n\n## 🧪 4. 数据与实验设计（如适用）\n- 使用了哪些数据集、实验设置、比较基准（baseline）？\n- 有哪些评估指标？结果说明了什么？\n\n## ✅ 5. 方法优点与不足\n- 作者在哪些地方表现出色？方法/设计是否具备可复现性与通用性？\n- 有哪些值得关注的假设、限制、风险或模糊点？\n\n## 🚀 6. 应用潜力与迁移可能\n- 本研究可以应用于哪些实际任务或行业？\n- 能否拓展到不同场景/任务？是否具有跨学科意义？\n\n## 🔍 7. 衍生问题与延展方向\n- 若继续探索，你有哪些改进想法？\n- 有哪些该论文未触及但你认为重要的研究问题？\n\n---\n\n🎯 输出要求：\n- 用 **Markdown** 或 **分段结构** 输出\n- 不照搬原文，用自己的语言总结\n- 可插入结构图（文本形式）或伪代码，说明处理流程\n- 如我输入的是截图/OCR内容，请你自动提取核心信息并按上述格式组织` }
        ]}
    ];

    // --- 2. 样式定义区 (CSS) ---
    GM_addStyle(`
        /* 主容器 */
        #prompt-main-container { margin-bottom: 10px; }

        /* 快捷访问区 */
        #prompt-quick-access { display: flex; gap: 8px; margin-bottom: 8px; }
        .quick-prompt-btn {
            flex-grow: 1; /* 平分空间 */
            padding: 5px 10px; font-size: 12px; font-weight: 500;
            border: 1px solid #D1D9E0; border-radius: 6px; cursor: pointer;
            background-color: #fff; transition: all 0.2s ease;
            white-space: nowrap; overflow: hidden; text-overflow: ellipsis;
        }
        .quick-prompt-btn:hover { border-color: #A0AEC0; background-color: #F7FAFC; }
        .last-used-btn { color: #2B6CB0; border-color: #BEE3F8; background-color: #EBF8FF; }
        .last-used-btn:hover { background-color: #EBF8FF; border-color: #90CDF4; }
        .pinned-btn { color: #9B2C2C; border-color: #FED7D7; background-color: #FFF5F5; }
        .pinned-btn:hover { background-color: #FFF5F5; border-color: #FEB2B2; }

        /* 下拉菜单 (与V5一致) */
        #prompt-select-container { position: relative; }
        #prompt-select {
            width: 100%; padding: 8px 32px 8px 12px; font-size: 14px;
            color: #4A5568; background-color: #F7FAFC; border: 1px solid #E2E8F0;
            border-radius: 8px; appearance: none; -webkit-appearance: none;
            cursor: pointer; transition: all 0.2s;
        }
        #prompt-select-container::after {
            content: '▼'; font-size: 12px; color: #718096; position: absolute;
            right: 12px; top: 50%; transform: translateY(-50%); pointer-events: none;
        }
        #prompt-select optgroup { font-weight: bold; background-color: #EDF2F7; }
        #prompt-select option { font-weight: normal; background-color: #fff; }

        /* 暗黑模式 */
        html.dark .quick-prompt-btn {
            background-color: #2D3748; color: #E2E8F0; border-color: #4A5568;
        }
        html.dark .quick-prompt-btn:hover { border-color: #718096; background-color: #1A202C; }
        html.dark .last-used-btn { color: #90CDF4; border-color: #2C5282; background-color: #2A4365; }
        html.dark .last-used-btn:hover { border-color: #4299E1; }
        html.dark .pinned-btn { color: #FEB2B2; border-color: #9B2C2C; background-color: #742A2A; }
        html.dark .pinned-btn:hover { border-color: #FC8181; }
        html.dark #prompt-select { color: #CBD5E0; background-color: #2D3748; border-color: #4A5568; }
        html.dark #prompt-select-container::after { color: #A0AEC0; }
        html.dark #prompt-select optgroup { background-color: #1A202C; color: #E2E8F0; }
        html.dark #prompt-select option { background-color: #2D3748; color: #F7FAFC; }
    `);

    // --- 3. 核心功能实现 ---
    const STORAGE_KEY_LAST_USED = 'alphaxiv_last_used_prompt_id';
    let allPrompts = {}; // 用于快速通过 id 查找 prompt

    function findPromptById(id) {
        if (Object.keys(allPrompts).length === 0) {
            MODULES.forEach(m => m.prompts.forEach(p => allPrompts[p.id] = p));
        }
        return allPrompts[id];
    }

    function injectText(value, editor) {
        if (!editor || !value) return;
        const pElement = editor.querySelector('p');
        if (pElement) {
            pElement.textContent = value;
            pElement.classList.remove('is-empty');
            const trailingBreak = pElement.querySelector('.ProseMirror-trailingBreak');
            if (trailingBreak) trailingBreak.remove();
            editor.dispatchEvent(new Event('input', { bubbles: true, cancelable: true }));
            editor.focus();
        }
    }

    const checkInterval = setInterval(async () => {
        const editor = document.querySelector('.ProseMirror[contenteditable="true"]');
        if (editor && !document.getElementById('prompt-main-container')) {
            console.log('AlphaXiv Assistant Smart Workflow: Initializing...');

            // 1. 创建主容器
            const mainContainer = document.createElement('div');
            mainContainer.id = 'prompt-main-container';

            // 2. 创建并填充快捷访问区
            const quickAccessContainer = document.createElement('div');
            quickAccessContainer.id = 'prompt-quick-access';

            // a. "最近使用" 按钮
            const lastUsedId = await GM_getValue(STORAGE_KEY_LAST_USED, null);
            if (lastUsedId) {
                const prompt = findPromptById(lastUsedId);
                if (prompt) {
                    const btn = document.createElement('button');
                    btn.className = 'quick-prompt-btn last-used-btn';
                    btn.textContent = `🕒 最近: ${prompt.label}`;
                    btn.title = `最近使用: ${prompt.label}`;
                    btn.onclick = () => injectText(prompt.value, editor);
                    quickAccessContainer.appendChild(btn);
                }
            }

            // b. "固定" 按钮
            PINNED_PROMPT_IDS.forEach(id => {
                const prompt = findPromptById(id);
                if (prompt && id !== lastUsedId) { // 避免与"最近使用"重复显示
                    const btn = document.createElement('button');
                    btn.className = 'quick-prompt-btn pinned-btn';
                    btn.textContent = `📌 ${prompt.label}`;
                    btn.title = `固定项: ${prompt.label}`;
                    btn.onclick = () => {
                        injectText(prompt.value, editor);
                        GM_setValue(STORAGE_KEY_LAST_USED, prompt.id); // 点击固定项也视为最近使用
                    };
                    quickAccessContainer.appendChild(btn);
                }
            });

            if (quickAccessContainer.hasChildNodes()) {
                mainContainer.appendChild(quickAccessContainer);
            }

            // 3. 创建并填充下拉菜单
            const selectContainer = document.createElement('div');
            selectContainer.id = 'prompt-select-container';
            const select = document.createElement('select');
            select.id = 'prompt-select';

            select.innerHTML = `<option value="" disabled selected>--- 浏览所有模板 ---</option>` +
                MODULES.map(module =>
                    `<optgroup label="${module.title}">` +
                    module.prompts.map(p => `<option value="${p.id}">[${p.type}] ${p.label}</option>`).join('') +
                    `</optgroup>`
                ).join('');

            select.onchange = (e) => {
                const selectedId = e.target.value;
                const prompt = findPromptById(selectedId);
                if (prompt) {
                    injectText(prompt.value, editor);
                    GM_setValue(STORAGE_KEY_LAST_USED, selectedId); // 保存为最近使用
                }
                e.target.selectedIndex = 0; // 重置下拉菜单
            };

            selectContainer.appendChild(select);
            mainContainer.appendChild(selectContainer);

            // 4. 注入到页面
            editor.parentElement.before(mainContainer);

            clearInterval(checkInterval);
        }
    }, 500);

})();