导出 Google AI Studio 对话

在 Google AI Studio 页面显示一个悬浮按钮，点击即可导出当前对话为 MD 文件，包含文件名、导出时间和 AI 名称。
Wrap lines
// ==UserScript==
// @name         导出 Google AI Studio 对话
// @namespace    https://aistudio.google.com/
// @version      1.3
// @description  在 Google AI Studio 页面显示一个悬浮按钮，点击即可导出当前对话为 MD 文件，包含文件名、导出时间和 AI 名称。
// @author       Gemini 2.0 Flash Thinking Experimental
// @match        https://aistudio.google.com/prompts/*
// @grant        none
// @license MIT
// ==/UserScript==
(function() {
    'use strict';

    // 创建悬浮按钮
    const exportButton = document.createElement('button');
    exportButton.textContent = '导出对话';
    exportButton.style.position = 'fixed';
    exportButton.style.bottom = '20px';
    exportButton.style.right = '20px';
    exportButton.style.padding = '10px 15px';
    exportButton.style.backgroundColor = '#007bff';
    exportButton.style.color = 'white';
    exportButton.style.border = 'none';
    exportButton.style.borderRadius = '5px';
    exportButton.style.cursor = 'pointer';
    exportButton.style.zIndex = '1000'; // 确保按钮在其他元素之上

    // 格式化日期时间
    function formatDateTime(date) {
        const year = date.getFullYear();
        const month = String(date.getMonth() + 1).padStart(2, '0');
        const day = String(date.getDate()).padStart(2, '0');
        const hour = String(date.getHours()).padStart(2, '0');
        const minute = String(date.getMinutes()).padStart(2, '0');
        return `${year}-${month}-${day} ${hour}:${minute}`;
    }

    function getMessageContent(element) {
        let content = '';

        function traverseNodes(node) {
            if (node.nodeType === Node.ELEMENT_NODE) {
                if (node.nodeName === 'P') {
                    content += node.textContent + '\n';
                } else {
                    node.childNodes.forEach(traverseNodes); // Recursively traverse child nodes
                }
            } else if (node.nodeType === Node.TEXT_NODE) {
                content += node.textContent;
            }
        }

        element.childNodes.forEach(traverseNodes);
        return content.trim();
    }

    // 添加按钮点击事件
    exportButton.addEventListener('click', function() {
        const now = new Date();
        const formattedDateTime = formatDateTime(now);

        // 获取页面标题作为文件名
        const pageTitleElement = document.querySelector('.page-title');
        const baseFilename = pageTitleElement ? pageTitleElement.textContent.trim() : 'conversation';
        const filename = `${baseFilename} (${formattedDateTime}).md`;

        // 获取 AI 名称
        const aiNameElement = document.querySelector('.model-option-content .gmat-body-medium');
        const aiName = aiNameElement ? aiNameElement.textContent.trim() : 'AI';

        // 获取所有对话元素
        const userMessages = Array.from(document.querySelectorAll('.user-prompt-container'));
        const modelMessages = Array.from(document.querySelectorAll('.model-prompt-container'));

        // 将所有消息合并到一个数组中，并标记类型
        const allMessages = [];
        userMessages.forEach(message => allMessages.push({ author: '我', element: message }));
        modelMessages.forEach(message => allMessages.push({ author: 'AI', element: message }));

        // 按照元素在 DOM 中的出现顺序排序
        allMessages.sort((a, b) => {
            const parentA = a.element.parentNode;
            const parentB = b.element.parentNode;
            if (parentA === parentB) {
                return Array.from(parentA.children).indexOf(a.element) - Array.from(parentB.children).indexOf(b.element);
            }
            const treeWalker = document.createTreeWalker(document.body, NodeFilter.SHOW_ELEMENT, null, false);
            let indexA = -1;
            let indexB = -1;
            let currentIndex = 0;
            while (treeWalker.nextNode()) {
                const currentNode = treeWalker.currentNode;
                if (currentNode === a.element) {
                    indexA = currentIndex;
                }
                if (currentNode === b.element) {
                    indexB = currentIndex;
                }
                if (indexA !== -1 && indexB !== -1) {
                    return indexA - indexB;
                }
                currentIndex++;
            }
            return 0;
        });

        // 构建 Markdown 内容
        let markdownContent = '';
        if (pageTitleElement) {
            markdownContent += `# ${pageTitleElement.textContent.trim()}\n`;
        }
        markdownContent += `\n- 导出时间：${formattedDateTime}\n`;
        markdownContent += `- AI：${aiName}\n`;
        markdownContent += `\n---\n`;

        for (let i = 0; i < allMessages.length; i += 2) {
            const myMessage = allMessages[i];
            const aiMessage = allMessages[i + 1];

            if (myMessage && myMessage.author === '我') {
                markdownContent += `\n**我**: ${getMessageContent(myMessage.element)}\n\n`;
            }

            if (aiMessage && aiMessage.author === 'AI') {
                let aiContent = getMessageContent(aiMessage.element);
                if (aiContent.endsWith(' warning')) {
                    aiContent = aiContent.slice(0, -' warning'.length);
                }
                markdownContent += `**AI**: ${aiContent}\n\n`;
            }

            if (myMessage || aiMessage) {
                markdownContent += '---\n';
            }
        }

        // 移除重复的分割线
        markdownContent = markdownContent.replace(/^---\n(?:---\n)+/gm, '');

        // 移除两个或多个连续的换行
        markdownContent = markdownContent.replace(/\n{2,}/g, '\n\n');

        // 创建 Blob 对象
        const blob = new Blob([markdownContent], { type: 'text/markdown;charset=utf-8' });

        // 创建下载链接
        const url = URL.createObjectURL(blob);
        const downloadLink = document.createElement('a');
        downloadLink.href = url;
        downloadLink.download = filename; // 使用带时间戳的文件名

        // 模拟点击下载链接
        document.body.appendChild(downloadLink);
        downloadLink.click();
        document.body.removeChild(downloadLink);

        // 释放 URL 对象
        URL.revokeObjectURL(url);
    });

    // 将按钮添加到页面
    document.body.appendChild(exportButton);
})();
欢迎访问Greasy Fork镜像
