// ==UserScript==
// @name         AI Studio Chat Exporter (Markdown & Code Block Support)
// @namespace    http://tampermonkey.net/
// @version      5.0
// @author       Tokisaki Galaxy
// @match        https://aistudio.google.com/prompts/*
// @description  Export AI Studio chat history. 1. i18n support (CN/EN/DE/RU/JA). 2. Draggable button. 3. Auto system prompt detection. 4. Perfect Markdown formatting.
// @icon         https://www.google.com/s2/favicons?sz=64&domain=google.com
// @grant        none
// @license      AGPL-3.0
// @downloadURL https://update.greasyfork.icu/scripts/557864/AI%20Studio%20Chat%20Exporter%20%28Markdown%20%20Code%20Block%20Support%29.user.js
// @updateURL https://update.greasyfork.icu/scripts/557864/AI%20Studio%20Chat%20Exporter%20%28Markdown%20%20Code%20Block%20Support%29.meta.js
// ==/UserScript==

(function() {
    'use strict';

    // --- 🌍 i18n Configuration ---
    const MESSAGES = {
        'zh': {
            exportBtn: "导出 JSON",
            wait: "请稍候...",
            getSys: "获取系统提示词...",
            resetView: "重置视图...",
            analyzing: "分析抓取中...",
            packaging: "生成 JSON...",
            noChat: "未找到聊天区域，请确保处于对话页面。",
            sysFound: "已获取系统提示词",
            sysHidden: "展开侧边栏获取..."
        },
        'en': {
            exportBtn: "Export JSON",
            wait: "Please wait...",
            getSys: "Fetching System Prompt...",
            resetView: "Resetting View...",
            analyzing: "Analyzing...",
            packaging: "Generating JSON...",
            noChat: "Chat container not found.",
            sysFound: "System prompt captured",
            sysHidden: "Expanding sidebar..."
        },
        'de': {
            exportBtn: "JSON Exportieren",
            wait: "Bitte warten...",
            getSys: "System-Prompt abrufen...",
            resetView: "Ansicht zurücksetzen...",
            analyzing: "Analysieren...",
            packaging: "JSON erstellen...",
            noChat: "Chat-Bereich nicht gefunden.",
            sysFound: "System-Prompt erfasst",
            sysHidden: "Seitenleiste erweitern..."
        },
        'ru': {
            exportBtn: "Экспорт JSON",
            wait: "Подождите...",
            getSys: "Получение System Prompt...",
            resetView: "Сброс вида...",
            analyzing: "Анализ...",
            packaging: "Создание JSON...",
            noChat: "Область чата не найдена.",
            sysFound: "System prompt получен",
            sysHidden: "Расширение боковой панели..."
        },
        'ja': {
            exportBtn: "JSON をエクスポート",
            wait: "少々お待ちください...",
            getSys: "システムプロンプトを取得中...",
            resetView: "ビューをリセット中...",
            analyzing: "解析中...",
            packaging: "JSON を生成中...",
            noChat: "チャット領域が見つかりませんでした。",
            sysFound: "システムプロンプトを取得しました",
            sysHidden: "サイドバーを展開しています..."
        }
    };

    // Detect Browser Language
    const langCode = (navigator.language || navigator.userLanguage).slice(0, 2);
    const t = (key) => MESSAGES[langCode]?.[key] || MESSAGES['en'][key];

    // --- Config ---
    const CONFIG = {
        scrollStep: 350,
        scrollDelay: 1200,
        uiDelay: 1000,
    };

    let isExporting = false;

    // --- UI: Draggable Button ---
    function createExportButton() {
        if (document.getElementById('ai-studio-export-btn')) return;

        const btn = document.createElement('button');
        btn.id = 'ai-studio-export-btn';
        btn.innerText = t('exportBtn');
        
        Object.assign(btn.style, {
            position: 'fixed',
            top: '20px',
            left: '50%',
            transform: 'translateX(-50%)',
            zIndex: '99999',
            padding: '10px 16px',
            backgroundColor: '#1a73e8',
            color: 'white',
            border: 'none',
            borderRadius: '20px',
            cursor: 'move',
            boxShadow: '0 4px 8px rgba(0,0,0,0.3)',
            fontWeight: 'bold',
            fontSize: '14px',
            fontFamily: 'sans-serif',
            transition: 'background-color 0.2s, transform 0.1s',
            userSelect: 'none'
        });

        // Draggable Logic
        let isDragging = false;
        let hasMoved = false;
        let startX, startY;

        btn.addEventListener('mousedown', function(e) {
            isDragging = true;
            hasMoved = false;
            const rect = btn.getBoundingClientRect();
            startX = e.clientX - rect.left;
            startY = e.clientY - rect.top;

            btn.style.transform = 'none';
            btn.style.left = rect.left + 'px';
            btn.style.top = rect.top + 'px';
            btn.style.opacity = '0.9';
        });

        document.addEventListener('mousemove', function(e) {
            if (!isDragging) return;
            hasMoved = true;
            e.preventDefault();
            btn.style.left = `${e.clientX - startX}px`;
            btn.style.top = `${e.clientY - startY}px`;
        });

        document.addEventListener('mouseup', function() {
            if (isDragging) {
                isDragging = false;
                btn.style.opacity = '1';
            }
        });

        btn.addEventListener('click', function(e) {
            if (hasMoved) {
                e.preventDefault();
                e.stopPropagation();
                return;
            }
            startExportProcess();
        });

        document.body.appendChild(btn);
    }

    function updateBtn(text, disabled = false) {
        const btn = document.getElementById('ai-studio-export-btn');
        if (btn) {
            btn.innerText = text;
            btn.disabled = disabled;
            btn.style.backgroundColor = disabled ? '#7f8c8d' : '#1a73e8';
            btn.style.cursor = disabled ? 'wait' : 'move';
        }
    }

    const sleep = (ms) => new Promise(r => setTimeout(r, ms));

    // --- Logic 1: System Instruction ---
    async function getSystemInstruction() {
        updateBtn(t('getSys'));
        
        let target = document.querySelector('ms-system-instructions-panel .subtitle');
        if (target && target.innerText.trim()) return target.innerText.trim();

        const toggleBtn = document.querySelector('.runsettings-toggle-button');
        if (toggleBtn) {
            console.log(t('sysHidden'));
            toggleBtn.click();
            await sleep(CONFIG.uiDelay);
            
            target = document.querySelector('ms-system-instructions-panel .subtitle');
            let text = target ? target.innerText.trim() : "";
            if(!text) {
                const fallback = document.querySelector('ms-system-instruction-editor textarea');
                if(fallback) text = fallback.value;
            }

            toggleBtn.click();
            await sleep(500);
            return text;
        }
        return "";
    }

    // --- Logic 2: Markdown Converter ---
    function domToMarkdown(node) {
        if (!node) return "";
        
        const skipClasses = ['author-label', 'actions-container', 'turn-footer', 'thinking-progress-icon', 'thought-collapsed-text', 'mat-icon'];
        if (node.classList && skipClasses.some(c => node.classList.contains(c))) return "";
        if (node.tagName === 'MS-THOUGHT-CHUNK' || node.tagName === 'BUTTON') return "";

        // Code Blocks
        if (node.tagName === 'MS-CODE-BLOCK') {
            let lang = "text";
            const titleSpan = node.querySelector('.title span:last-child');
            if (titleSpan) lang = titleSpan.innerText.trim();
            
            const codeEl = node.querySelector('code');
            const codeText = codeEl ? codeEl.innerText : node.innerText;
            return `\n\`\`\`${lang}\n${codeText.trim()}\n\`\`\`\n`;
        }

        // List
        if (node.tagName === 'LI') return `- ${parseChildren(node).trim()}\n`;

        // Heading
        if (/^H[1-6]$/.test(node.tagName)) {
            const level = parseInt(node.tagName[1]);
            return `\n${'#'.repeat(level)} ${parseChildren(node).trim()}\n`;
        }

        // Paragraph & LineBreak
        if (node.tagName === 'P') return `\n${parseChildren(node).trim()}\n\n`;
        if (node.tagName === 'BR') return "\n";

        if (node.nodeType === Node.TEXT_NODE) return node.textContent;
        
        let result = parseChildren(node);

        // Formatting
        if (['STRONG', 'B'].includes(node.tagName)) result = `**${result}**`;
        if (['EM', 'I'].includes(node.tagName)) result = `*${result}*`;
        if (node.classList && node.classList.contains('inline-code')) result = `\`${result}\``;

        return result;
    }

    function parseChildren(node) {
        let text = "";
        node.childNodes.forEach(child => {
            text += domToMarkdown(child);
        });
        return text;
    }

    function extractCleanMarkdown(turnElement) {
        const contentDiv = turnElement.querySelector('.turn-content');
        if (!contentDiv) return null;
        let md = domToMarkdown(contentDiv);
        md = md.replace(/\n{3,}/g, '\n\n').trim();
        if (!md || md === "Model" || md === "User") return null;
        return md;
    }

    // --- Logic 3: Main Flow ---
    async function startExportProcess() {
        if (isExporting) return;
        isExporting = true;

        const container = document.querySelector('ms-autoscroll-container');
        if (!container) {
            alert(t('noChat'));
            isExporting = false;
            return;
        }

        // 1. System Prompt
        const sysInstruction = await getSystemInstruction();

        // 2. Scroll & Scrape
        const messageMap = new Map();
        const idOrder = [];

        updateBtn(t('resetView'));
        container.scrollTo({ top: 0, behavior: 'instant' });
        await sleep(1500);

        let lastScrollTop = -1;
        let stuckCounter = 0;

        while (true) {
            const visibleTurns = document.querySelectorAll('ms-chat-turn');
            visibleTurns.forEach(turn => {
                const uid = turn.id;
                if (!uid) return;

                if (!idOrder.includes(uid)) idOrder.push(uid);

                let role = 'user';
                if (turn.querySelector('.model-prompt-container') || turn.getAttribute('data-turn-role') === 'Model') {
                    role = 'assistant';
                }

                const content = extractCleanMarkdown(turn);
                if (content) {
                    messageMap.set(uid, { role, content });
                }
            });

            // Stop condition
            const isBottom = Math.abs(container.scrollHeight - container.clientHeight - container.scrollTop) < 20;
            if (Math.abs(container.scrollTop - lastScrollTop) < 2) stuckCounter++;
            else stuckCounter = 0;

            const percent = Math.min(99, Math.floor((container.scrollTop / (container.scrollHeight - container.clientHeight)) * 100));
            updateBtn(`${t('analyzing')} ${percent}%`);

            if (isBottom || stuckCounter >= 3) break;

            lastScrollTop = container.scrollTop;
            container.scrollBy({ top: CONFIG.scrollStep, behavior: 'smooth' });
            await sleep(CONFIG.scrollDelay);
        }

        // 3. Export
        updateBtn(t('packaging'));
        const validMessages = [];
        idOrder.forEach(id => {
            if (messageMap.has(id)) {
                validMessages.push(messageMap.get(id));
            }
        });

        downloadFile({
            system_instruction: sysInstruction,
            messages: validMessages
        });

        updateBtn(t('exportBtn'), false);
        isExporting = false;
    }

    function downloadFile(data) {
        let title = "aistudio_chat";
        try {
            const h1 = document.querySelector('.page-title h1');
            if (h1) title = h1.innerText.trim().replace(/[\\/:*?"<>|]/g, '_');
        } catch(e) {}

        const blob = new Blob([JSON.stringify(data, null, 2)], {type: "application/json"});
        const url = URL.createObjectURL(blob);
        const a = document.createElement('a');
        a.href = url;
        a.download = `${title}.json`;
        document.body.appendChild(a);
        a.click();
        document.body.removeChild(a);
        URL.revokeObjectURL(url);
    }

    function init() {
        const observer = new MutationObserver(() => createExportButton());
        observer.observe(document.body, { childList: true, subtree: true });
        createExportButton();
    }

    window.addEventListener('load', init);
    if (document.readyState === 'complete') init();

})();