// ==UserScript==
// @name         DeepSeek Chat Exporter (Markdown & PDF & PNG)
// @namespace    http://tampermonkey.net/
// @version      1.7
// @description  导出 DeepSeek 聊天记录为 Markdown、PDF 和 PNG 格式
// @author       HSyuf/Blueberrycongee
// @match        https://chat.deepseek.com/*
// @grant        GM_addStyle
// @grant        GM_download
// @license      MIT
// @require      https://cdnjs.cloudflare.com/ajax/libs/html2canvas/1.4.1/html2canvas.min.js
// @downloadURL none
// ==/UserScript==

(function () {
    'use strict';

    // =====================
    // 配置
    // =====================
    const config = {
        chatContainerSelector: '.dad65929', // 聊天框容器
        userClassPrefix: 'fa81',             // 用户消息 class 前缀
        aiClassPrefix: 'f9bf7997',           // AI消息相关 class 前缀
        aiReplyContainer: 'edb250b1',        // AI回复的主要容器
        searchHintSelector: '.a6d716f5.db5991dd', // 搜索/思考时间
        thinkingChainSelector: '.e1675d8b',  // 思考链
        finalAnswerSelector: 'div.ds-markdown.ds-markdown--block', // 正式回答
        exportFileName: 'DeepSeek_Chat_Export',
    };

    let __exportPNGLock = false;  // 全局锁，防止重复点击

    // =====================
    // 工具函数
    // =====================
    function isUserMessage(node) {
        return node.classList.contains(config.userClassPrefix);
    }

    function isAIMessage(node) {
        return node.classList.contains(config.aiClassPrefix);
    }

    function extractSearchOrThinking(node) {
        const hintNode = node.querySelector(config.searchHintSelector);
        return hintNode ? `**${hintNode.textContent.trim()}**` : null;
    }

    function extractThinkingChain(node) {
        const thinkingNode = node.querySelector(config.thinkingChainSelector);
        return thinkingNode ? `**思考链**\n${thinkingNode.textContent.trim()}` : null;
    }

    function extractFinalAnswer(node) {
        const answerNode = node.querySelector(config.finalAnswerSelector);
        if (!answerNode) return null;

        let answerContent = '';
        const elements = answerNode.querySelectorAll('.ds-markdown--block p, .ds-markdown--block h3, .katex-display.ds-markdown-math, hr');

        elements.forEach((element) => {
            if (element.tagName.toLowerCase() === 'p') {
                element.childNodes.forEach((childNode) => {
                    if (childNode.nodeType === Node.TEXT_NODE) {
                        answerContent += childNode.textContent.trim();
                    } else if (childNode.classList && childNode.classList.contains('katex')) {
                        const tex = childNode.querySelector('annotation[encoding="application/x-tex"]');
                        if (tex) {
                            answerContent += `$$$${tex.textContent.trim()}$$$`;
                        }
                    } else if (childNode.tagName === 'STRONG') {
                        answerContent += `**${childNode.textContent.trim()}**`;
                    } else if (childNode.tagName === 'EM') {
                        answerContent += `*${childNode.textContent.trim()}*`;
                    } else if (childNode.tagName === 'A') {
                        const href = childNode.getAttribute('href');
                        answerContent += `[${childNode.textContent.trim()}](${href})`;
                    } else if (childNode.nodeType === Node.ELEMENT_NODE) {
                        answerContent += childNode.textContent.trim();
                    }
                });
                answerContent += '\n\n';
            }
            else if (element.tagName.toLowerCase() === 'h3') {
                answerContent += `### ${element.textContent.trim()}\n\n`;
            }
            else if (element.classList.contains('katex-display')) {
                const tex = element.querySelector('annotation[encoding="application/x-tex"]');
                if (tex) {
                    answerContent += `$$${tex.textContent.trim()}$$\n\n`;
                }
            }
            else if (element.tagName.toLowerCase() === 'hr') {
                answerContent += '\n---\n';
            }
        });

        return `**正式回答**\n${answerContent.trim()}`;
    }

    function getOrderedMessages() {
        const messages = [];
        const chatContainer = document.querySelector(config.chatContainerSelector);
        if (!chatContainer) {
            console.error('未找到聊天容器');
            return messages;
        }

        for (const node of chatContainer.children) {
            if (isUserMessage(node)) {
                messages.push(`**用户：**\n${node.textContent.trim()}`);
            } else if (isAIMessage(node)) {
                let output = '';
                const aiReplyContainer = node.querySelector(`.${config.aiReplyContainer}`);
                if (aiReplyContainer) {
                    const searchHint = extractSearchOrThinking(aiReplyContainer);
                    if (searchHint) output += `${searchHint}\n\n`;
                    const thinkingChain = extractThinkingChain(aiReplyContainer);
                    if (thinkingChain) output += `${thinkingChain}\n\n`;
                } else {
                    const searchHint = extractSearchOrThinking(node);
                    if (searchHint) output += `${searchHint}\n\n`;
                }
                const finalAnswer = extractFinalAnswer(node);
                if (finalAnswer) output += `${finalAnswer}\n\n`;
                if (output.trim()) {
                    messages.push(output.trim());
                }
            }
        }
        return messages;
    }

    function generateMdContent() {
        const messages = getOrderedMessages();
        return messages.length ? messages.join('\n\n---\n\n') : '';
    }

    // =====================
    // 导出功能
    // =====================
    function exportMarkdown() {
        const mdContent = generateMdContent();
        if (!mdContent) {
            alert("未找到聊天记录！");
            return;
        }

        const fixedMdContent = mdContent.replace(/(\*\*.*?\*\*)/g, '<strong>$1</strong>')
            .replace(/\(\s*([^)]*)\s*\)/g, '\\($1\\)')
            .replace(/\$\$\s*([^$]*)\s*\$\$/g, '$$$1$$');

        const blob = new Blob([fixedMdContent], { type: 'text/markdown' });
        const url = URL.createObjectURL(blob);
        const a = document.createElement('a');
        a.href = url;
        a.download = `${config.exportFileName}_${Date.now()}.md`;
        a.click();
        setTimeout(() => URL.revokeObjectURL(url), 5000);
    }

    function exportPDF() {
        const mdContent = generateMdContent();
        if (!mdContent) return;

        const fixedMdContent = mdContent.replace(/(\*\*.*?\*\*)/g, '<strong>$1</strong>')
            .replace(/\(\s*([^)]*)\s*\)/g, '\\($1\\)')
            .replace(/\$\$\s*([^$]*)\s*\$\$/g, '$$$1$$');

        const printContent = `
            <html>
                <head>
                    <title>DeepSeek Chat Export</title>
                    <style>
                        body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif; line-height: 1.6; padding: 20px; max-width: 800px; margin: 0 auto; }
                        h2 { color: #2c3e50; border-bottom: 1px solid #eee; padding-bottom: 0.3em; }
                        .ai-answer { color: #1a7f37; margin: 15px 0; }
                        .ai-chain { color: #666; font-style: italic; margin: 10px 0; }
                        hr { border: 0; border-top: 1px solid #eee; margin: 25px 0; }
                    </style>
                </head>
                <body>
                    ${fixedMdContent.replace(/\*\*用户：\*\*\n/g, '<h2>用户提问</h2><div class="user-question">')
                        .replace(/\*\*正式回答\*\*\n/g, '</div><h2>AI 回答</h2><div class="ai-answer">')
                        .replace(/\*\*思考链\*\*\n/g, '</div><h2>思维链</h2><div class="ai-chain">')
                        .replace(/\n/g, '<br>')
                        .replace(/---/g, '</div><hr>')}
                </body>
            </html>
        `;

        const printWindow = window.open("", "_blank");
        printWindow.document.write(printContent);
        printWindow.document.close();
        setTimeout(() => { printWindow.print(); printWindow.close(); }, 500);
    }

    function exportPNG() {
        if (__exportPNGLock) return;  // 如果当前正在导出，跳过
        __exportPNGLock = true;

        const chatContainer = document.querySelector(config.chatContainerSelector);
        if (!chatContainer) {
            alert("未找到聊天容器！");
            __exportPNGLock = false;
            return;
        }

        // 创建沙盒容器
        const sandbox = document.createElement('iframe');
        sandbox.style.cssText = `
            position: fixed;
            left: -9999px;
            top: 0;
            width: 800px;
            height: ${window.innerHeight}px;
            border: 0;
            visibility: hidden;
        `;
        document.body.appendChild(sandbox);

        // 深度克隆与样式处理
        const cloneNode = chatContainer.cloneNode(true);
        cloneNode.style.cssText = `
            width: 800px !important;
            transform: none !important;
            overflow: visible !important;
            position: static !important;
            background: white !important;
            max-height: none !important;
            padding: 20px !important;
            margin: 0 !important;
            box-sizing: border-box !important;
        `;

        // 清理干扰元素，排除图标
        ['button', 'input', '.ds-message-feedback-container', '.eb23581b.dfa60d66'].forEach(selector => {
            cloneNode.querySelectorAll(selector).forEach(el => el.remove());
        });

        // 数学公式修复
        cloneNode.querySelectorAll('.katex-display').forEach(mathEl => {
            mathEl.style.transform = 'none !important';
            mathEl.style.position = 'relative !important';
        });

        // 注入沙盒
        sandbox.contentDocument.body.appendChild(cloneNode);
        sandbox.contentDocument.body.style.background = 'white';

        // 等待资源加载
        const waitReady = () => Promise.all([
            document.fonts.ready,
            new Promise(resolve => setTimeout(resolve, 300))  // 增加300ms缓冲
        ]);

        waitReady().then(() => {
            return html2canvas(cloneNode, {
                scale: 2,
                useCORS: true,
                logging: true,
                backgroundColor: "#FFFFFF"
            });
        }).then(canvas => {
            canvas.toBlob(blob => {
                const url = URL.createObjectURL(blob);
                const a = document.createElement('a');
                a.href = url;
                a.download = `${config.exportFileName}_${Date.now()}.png`;
                a.click();
                setTimeout(() => {
                    URL.revokeObjectURL(url);
                    sandbox.remove();
                }, 1000);
            }, 'image/png');
        }).catch(err => {
            console.error('截图失败:', err);
            alert(`导出失败：${err.message}`);
        }).finally(() => {
            __exportPNGLock = false;
        });
    }

    // =====================
    // 创建导出菜单
    // =====================
    function createExportMenu() {
        const menu = document.createElement("div");
        menu.className = "ds-exporter-menu";
        menu.innerHTML = `
            <button class="export-btn" id="md-btn">导出为 Markdown</button>
            <button class="export-btn" id="pdf-btn">导出为 PDF</button>
            <button class="export-btn" id="png-btn">导出图片</button>
        `;

        menu.querySelector("#md-btn").addEventListener("click", exportMarkdown);
        menu.querySelector("#pdf-btn").addEventListener("click", exportPDF);
        menu.querySelector("#png-btn").addEventListener("click", exportPNG);
        document.body.appendChild(menu);
    }

    // =====================
    // 样式
    // =====================
    GM_addStyle(`
        .ds-exporter-menu {
            position: fixed;
            top: 20px;
            right: 20px;
            z-index: 999999;
            background: rgba(255, 255, 255, 0.95);
            padding: 12px;
            border-radius: 8px;
            box-shadow: 0 2px 12px rgba(0, 0, 0, 0.15);
            display: flex;
            flex-direction: column;
            gap: 8px;
            backdrop-filter: blur(4px);
        }
        .export-btn {
            background: #2196F3;
            color: white;
            border: none;
            padding: 8px 16px;
            border-radius: 6px;
            cursor: pointer;
            font-size: 14px;
            transition: all 0.2s;
        }
        .export-btn:hover {
            background: #1976D2;
            transform: translateY(-1px);
            box-shadow: 0 2px 6px rgba(0, 0, 0, 0.2);
        }
    `);

    // =====================
    // 初始化
    // =====================
    function init() {
        const checkInterval = setInterval(() => {
            if (document.querySelector(config.chatContainerSelector)) {
                clearInterval(checkInterval);
                createExportMenu();
            }
        }, 500);
    }

    init();
})();