let currentCategory = '';
let currentDoc = '';
let docList = [];
let currentMarkdownContent = ''; // 保存当前文档的markdown源码
let isEditMode = false; // 是否处于编辑模式
let savedScrollPosition = 0; // 保存的滚动位置
let currentScrollObserver = null; // 保存当前的滚动监听器

// DOM 元素缓存
const DOM = {
    loading: null,
    content: null,
    docNav: null,
    toc: null,
    leftSidebar: null,
    rightSidebar: null,
    editBtn: null,
    editorContainer: null,
    markdownEditor: null,
    editorDocName: null,
    navPrev: null,
    navNext: null,
    contentArea: null,
    searchContainer: null,
    searchToggleBtn: null,
    searchInput: null,
    searchBtn: null,
    searchResults: null,
    closeSearchBoxBtn: null,
    saveBtn: null,
    cancelBtn: null,
    backToTopBtn: null,
    docNavigation: null,
    toggleLeft: null,
    toggleRight: null,
    categoryNameSpan: null,
    currentDocResults: null,
    otherDocResults: null
};

// 配置 marked
marked.setOptions({
    highlight: function(code, lang) {
        if (lang && hljs.getLanguage(lang)) {
            try {
                const result = hljs.highlight(code, { language: lang });
                return result.value;
            } catch (err) {
                console.error('Highlight error:', err);
            }
        }

        // 自动检测语言
        const autoResult = hljs.highlightAuto(code);
        return autoResult.value;
    },
    breaks: true,
    gfm: true
});

// 从 URL 获取参数
function getQueryParams() {
    const params = new URLSearchParams(window.location.search);
    return {
        category: params.get('category'),
        doc: params.get('doc')
    };
}

// 根据文档路径获取展示名称（仅保留最后一段）
function getDocDisplayName(docName) {
    if (!docName || typeof docName !== 'string') {
        return '';
    }

    const segments = docName.split(/[\\\/]/);
    const name = segments.pop();
    return name || docName;
}

// ========== 工具函数 ==========

// URL 编码路径（支持子目录）
function encodeDocPath(docName) {
    return docName.split('/').map(part => encodeURIComponent(part)).join('/');
}

// API 错误处理
async function handleApiError(response, defaultMessage) {
    const errorData = await response.json().catch(() => ({}));
    throw new Error(errorData.error || `${defaultMessage} (${response.status})`);
}

// 确保 DOM 元素已缓存
function ensureDOM(key, id) {
    if (!DOM[key]) DOM[key] = document.getElementById(id);
    return DOM[key];
}

// 通用防抖函数
function debounce(func, wait) {
    let timeout;
    return function executedFunction(...args) {
        clearTimeout(timeout);
        timeout = setTimeout(() => func.apply(this, args), wait);
    };
}

// 高亮代码块
function highlightCodeBlock(block) {
    block.classList.add('hljs');

    // 如果已经高亮过了，跳过
    if (block.querySelector('span')) return;

    const lang = block.className.match(/language-(\w+)/)?.[1];

    try {
        const highlighted = (lang && hljs.getLanguage(lang))
            ? hljs.highlight(block.textContent, { language: lang })
            : hljs.highlightAuto(block.textContent);

        block.innerHTML = highlighted.value;
        if (highlighted.language) {
            block.classList.add(`language-${highlighted.language}`);
        }
    } catch (err) {
        console.error('Error highlighting code:', err);
    }
}

// 加载文档列表
async function loadDocList() {
    const { category } = getQueryParams();
    if (!category) {
        window.location.href = '/';
        return;
    }

    currentCategory = category;
    if (!DOM.categoryNameSpan) DOM.categoryNameSpan = document.querySelector('#category-title .category-name');
    if (DOM.categoryNameSpan) {
        DOM.categoryNameSpan.textContent = category;
    }

    try {
        const response = await fetch(`/api/category/${encodeURIComponent(category)}`);
        if (!response.ok) throw new Error('获取文档列表失败');

        const categoryData = await response.json();
        docList = categoryData.docs.map(doc => ({
            ...doc,
            displayName: getDocDisplayName(doc.name)
        }));

        renderDocNav(docList);

        // 加载第一篇文档（或 URL 指定的文档）
        const { doc } = getQueryParams();
        const firstDoc = doc || docList[0]?.name;

        if (firstDoc) {
            loadDocument(firstDoc);
        }

    } catch (err) {
        console.error('Error loading doc list:', err);
        showError('加载文档列表失败');
    }
}

// 渲染文档导航
function renderDocNav(docs) {
    const nav = DOM.docNav || document.getElementById('doc-nav');
    if (!DOM.docNav) DOM.docNav = nav;

    nav.innerHTML = '';

    docs.forEach(doc => {
        const item = document.createElement('div');
        item.className = `nav-item level-${doc.level}`;
        item.dataset.docName = doc.name;

        const link = document.createElement('a');
        link.href = '#';
        link.className = 'nav-link';
        link.textContent = doc.displayName || getDocDisplayName(doc.name);
        link.title = doc.name;

        item.appendChild(link);
        nav.appendChild(item);
    });

    // 使用事件委托 - 只添加一个监听器
    nav.removeEventListener('click', handleNavClick);
    nav.addEventListener('click', handleNavClick);
}

// 导航点击处理函数
function handleNavClick(e) {
    if (e.target.classList.contains('nav-link')) {
        e.preventDefault();
        const item = e.target.parentElement;
        const docName = item.dataset.docName;

        if (docName) {
            loadDocument(docName);
            updateURL(currentCategory, docName);

            // 更新活动状态 - 使用缓存的查询结果
            const navItems = DOM.docNav.querySelectorAll('.nav-item');
            navItems.forEach(el => el.classList.toggle('active', el === item));
        }
    }
}

// 加载文档内容
async function loadDocument(docName, scrollToText = null) {
    ensureDOM('loading', 'loading');
    ensureDOM('content', 'markdown-content');

    DOM.loading.style.display = 'flex';
    DOM.content.innerHTML = '';

    try {
        const encodedDocName = encodeDocPath(docName);
        const response = await fetch(`/api/doc/${encodeURIComponent(currentCategory)}/${encodedDocName}`);

        if (!response.ok) {
            await handleApiError(response, '加载失败');
        }

        const data = await response.json();

        // 验证数据有效性
        if (!data.content || typeof data.content !== 'string') {
            throw new Error('文档内容格式错误');
        }

        currentDoc = docName;
        currentMarkdownContent = data.content; // 保存markdown源码

        // 渲染 Markdown
        const html = marked.parse(data.content);
        DOM.content.innerHTML = html;

        // 处理图片路径（将相对路径转换为API路径）
        DOM.content.querySelectorAll('img').forEach(img => {
            const src = img.getAttribute('src');
            if (src && !src.match(/^https?:\/\//)) {
                img.setAttribute('data-original-src', src);
                img.src = `/api/image/${encodeURIComponent(currentCategory)}/${src}`;
            }
        });

        // 代码块高亮
        DOM.content.querySelectorAll('pre code').forEach(highlightCodeBlock);

        // 为所有代码块添加复制按钮
        addCopyButtonsToCodeBlocks();

        // 生成 TOC
        generateTOC();

        // 更新文档导航（上一页/下一页）
        updateDocNavigation();

        // 更新活动文档 - 使用 toggle 优化
        if (DOM.docNav) {
            const navItems = DOM.docNav.querySelectorAll('.nav-item');
            navItems.forEach(el => el.classList.toggle('active', el.dataset.docName === docName));
        }

        DOM.loading.style.display = 'none';

        // 如果指定了要滚动到的文本，等待渲染完成后滚动
        if (scrollToText) {
            setTimeout(() => {
                scrollToSearchMatch(scrollToText);
            }, 100);
        }

    } catch (err) {
        console.error('Error loading document:', err);
        DOM.loading.style.display = 'none';
        showError(`加载文档失败：${err.message || '请稍后重试'}`);
    }
}

// 生成 TOC
function generateTOC() {
    ensureDOM('toc', 'toc');
    ensureDOM('content', 'markdown-content');

    // 先清空旧的TOC内容
    DOM.toc.innerHTML = '';

    const headings = DOM.content.querySelectorAll('h1, h2, h3, h4, h5, h6');

    if (headings.length === 0) {
        DOM.toc.innerHTML = '<p class="toc-empty">本文档没有标题</p>';
        return;
    }

    // 创建标题 ID 映射
    const headingMap = new Map();

    headings.forEach((heading, index) => {
        // 给标题添加 ID
        if (!heading.id) {
            heading.id = `heading-${index}`;
        }

        const level = parseInt(heading.tagName.substring(1));
        const link = document.createElement('a');
        link.href = `#${heading.id}`;
        link.className = `toc-link toc-level-${level}`;
        link.textContent = heading.textContent;

        headingMap.set(heading.id, heading);
        DOM.toc.appendChild(link);
    });

    // 使用事件委托 - 只添加一个监听器
    DOM.toc.removeEventListener('click', handleTocClick);
    DOM.toc.addEventListener('click', (e) => handleTocClick(e, headingMap));

    // 监听滚动，高亮当前标题
    setupScrollSpy(headings);
}

// TOC 点击处理函数
function handleTocClick(e, headingMap) {
    if (e.target.classList.contains('toc-link')) {
        e.preventDefault();
        const headingId = e.target.getAttribute('href').substring(1);
        const heading = headingMap.get(headingId);

        if (heading) {
            heading.scrollIntoView({ behavior: 'smooth', block: 'start' });

            // 更新活动状态
            const tocLinks = DOM.toc.querySelectorAll('.toc-link');
            tocLinks.forEach(el => el.classList.toggle('active', el === e.target));
        }
    }
}

// 滚动监听
function setupScrollSpy(headings) {
    // 清理之前的 observer
    if (currentScrollObserver) {
        currentScrollObserver.disconnect();
        currentScrollObserver = null;
    }

    // 如果没有标题，直接返回
    if (!headings || headings.length === 0) {
        return;
    }

    let activeLink = null;

    const observer = new IntersectionObserver((entries) => {
        entries.forEach(entry => {
            if (entry.isIntersecting) {
                const id = entry.target.id;
                const link = DOM.toc.querySelector(`.toc-link[href="#${id}"]`);

                if (link && link !== activeLink) {
                    // 只在需要时更新，减少 DOM 操作
                    if (activeLink) activeLink.classList.remove('active');
                    link.classList.add('active');
                    activeLink = link;
                }
            }
        });
    }, {
        rootMargin: '-100px 0px -80% 0px',
        threshold: 0
    });

    headings.forEach(heading => observer.observe(heading));

    // 保存当前的 observer 以便后续清理
    currentScrollObserver = observer;
}

// 更新 URL
function updateURL(category, doc) {
    const url = new URL(window.location);
    url.searchParams.set('category', category);
    url.searchParams.set('doc', doc);
    window.history.pushState({}, '', url);
}

// 更新文档导航（上一页/下一页）
function updateDocNavigation() {
    // 缓存 DOM 元素
    if (!DOM.navPrev) DOM.navPrev = document.getElementById('nav-prev');
    if (!DOM.navNext) DOM.navNext = document.getElementById('nav-next');

    if (!DOM.navPrev || !DOM.navNext || docList.length === 0) return;

    if (!DOM.docNavigation) DOM.docNavigation = document.getElementById('doc-navigation');
    if (!DOM.docNavigation) return;

    // 找到当前文档在列表中的索引
    const currentIndex = docList.findIndex(doc => doc.name === currentDoc);

    if (currentIndex === -1) {
        DOM.navPrev.style.display = 'none';
        DOM.navNext.style.display = 'none';
        DOM.docNavigation.className = 'doc-navigation';
        return;
    }

    const hasPrev = currentIndex > 0;
    const hasNext = currentIndex < docList.length - 1;

    // 辅助函数：设置导航按钮
    const setupNavButton = (button, doc) => {
        if (doc) {
            button.style.display = 'flex';
            const displayName = doc.displayName || getDocDisplayName(doc.name);
            button.querySelector('.nav-doc-name').textContent = displayName;
            button.onclick = () => {
                loadDocument(doc.name);
                updateURL(currentCategory, doc.name);
                // 滚动到顶部
                if (!DOM.contentArea) DOM.contentArea = document.getElementById('content-area');
                if (DOM.contentArea) {
                    DOM.contentArea.scrollTo({ top: 0, behavior: 'smooth' });
                }
            };
        } else {
            button.style.display = 'none';
        }
    };

    // 设置上一页和下一页按钮
    setupNavButton(DOM.navPrev, hasPrev ? docList[currentIndex - 1] : null);
    setupNavButton(DOM.navNext, hasNext ? docList[currentIndex + 1] : null);

    // 根据导航按钮状态添加类，用于CSS定位
    if (hasPrev && hasNext) {
        DOM.docNavigation.className = 'doc-navigation has-both';
    } else if (hasPrev) {
        DOM.docNavigation.className = 'doc-navigation has-prev-only';
    } else if (hasNext) {
        DOM.docNavigation.className = 'doc-navigation has-next-only';
    } else {
        DOM.docNavigation.className = 'doc-navigation';
    }
}

// 显示错误
function showError(message) {
    if (!DOM.content) DOM.content = document.getElementById('markdown-content');
    if (DOM.content) {
        DOM.content.innerHTML = `<div class="error-message">${message}</div>`;
    }
}

// 切换侧边栏（移动端）
function setupSidebarToggles() {
    if (!DOM.toggleLeft) DOM.toggleLeft = document.getElementById('toggle-left');
    if (!DOM.toggleRight) DOM.toggleRight = document.getElementById('toggle-right');
    const toggleLeft = DOM.toggleLeft;
    const toggleRight = DOM.toggleRight;

    // 使用 DOM 缓存
    if (!DOM.leftSidebar) DOM.leftSidebar = document.getElementById('left-sidebar');
    if (!DOM.rightSidebar) DOM.rightSidebar = document.getElementById('right-sidebar');

    // 移动端/平板初始化：默认折叠侧边栏
    if (window.innerWidth <= 768) {
        DOM.leftSidebar.classList.add('collapsed');
        DOM.rightSidebar.classList.add('collapsed');
    } else if (window.innerWidth <= 1024) {
        // 平板：只折叠右侧栏
        DOM.rightSidebar.classList.add('collapsed');
    }

    // 监听窗口大小变化
    window.addEventListener('resize', () => {
        if (window.innerWidth <= 768) {
            // 移动端：确保左右侧边栏都折叠
            if (!DOM.leftSidebar.classList.contains('collapsed')) {
                DOM.leftSidebar.classList.add('collapsed');
            }
            if (!DOM.rightSidebar.classList.contains('collapsed')) {
                DOM.rightSidebar.classList.add('collapsed');
            }
        } else if (window.innerWidth <= 1024) {
            // 平板：只折叠右侧栏，展开左侧栏
            DOM.leftSidebar.classList.remove('collapsed');
            if (!DOM.rightSidebar.classList.contains('collapsed')) {
                DOM.rightSidebar.classList.add('collapsed');
            }
        } else {
            // 桌面端：展开所有侧边栏
            DOM.leftSidebar.classList.remove('collapsed');
            DOM.rightSidebar.classList.remove('collapsed');
        }
    });

    toggleLeft.onclick = (e) => {
        e.stopPropagation();

        // 如果右侧栏展开，先隐藏它
        if (!DOM.rightSidebar.classList.contains('collapsed')) {
            DOM.rightSidebar.classList.add('collapsed');
        }

        DOM.leftSidebar.classList.toggle('collapsed');
    };

    toggleRight.onclick = (e) => {
        e.stopPropagation();

        // 如果左侧栏展开，先隐藏它
        if (!DOM.leftSidebar.classList.contains('collapsed')) {
            DOM.leftSidebar.classList.add('collapsed');
        }

        DOM.rightSidebar.classList.toggle('collapsed');
    };

    // 点击侧边栏外部时隐藏所有已展开的侧边栏（仅在移动端和平板）
    document.addEventListener('click', (e) => {
        // 仅在移动端和平板上启用此功能
        if (window.innerWidth > 1024) return;

        const isClickInsideLeftSidebar = DOM.leftSidebar.contains(e.target);
        const isClickInsideRightSidebar = DOM.rightSidebar.contains(e.target);
        const isToggleButton = e.target.closest('#toggle-left') || e.target.closest('#toggle-right');

        // 如果点击的不是侧边栏内部，也不是切换按钮，则隐藏所有展开的侧边栏
        if (!isClickInsideLeftSidebar && !isClickInsideRightSidebar && !isToggleButton) {
            if (!DOM.leftSidebar.classList.contains('collapsed')) {
                DOM.leftSidebar.classList.add('collapsed');
            }
            if (!DOM.rightSidebar.classList.contains('collapsed')) {
                DOM.rightSidebar.classList.add('collapsed');
            }
        }
    });
}

// 搜索功能
function initSearch() {
    // 初始化搜索相关的DOM缓存
    if (!DOM.searchContainer) DOM.searchContainer = document.getElementById('search-container');
    if (!DOM.searchToggleBtn) DOM.searchToggleBtn = document.getElementById('search-toggle-btn');
    if (!DOM.searchInput) DOM.searchInput = document.getElementById('search-input');
    if (!DOM.searchBtn) DOM.searchBtn = document.getElementById('search-btn');
    if (!DOM.searchResults) DOM.searchResults = document.getElementById('search-results');
    if (!DOM.closeSearchBoxBtn) DOM.closeSearchBoxBtn = document.getElementById('close-search-box');

    // 检查元素是否存在
    if (!DOM.searchContainer || !DOM.searchToggleBtn || !DOM.searchInput || !DOM.searchBtn || !DOM.searchResults || !DOM.closeSearchBoxBtn) {
        console.error('Search elements not found:', {
            searchContainer: !!DOM.searchContainer,
            searchToggleBtn: !!DOM.searchToggleBtn,
            searchInput: !!DOM.searchInput,
            searchBtn: !!DOM.searchBtn,
            searchResults: !!DOM.searchResults,
            closeSearchBoxBtn: !!DOM.closeSearchBoxBtn
        });
        return;
    }

    const searchContainer = DOM.searchContainer;
    const searchToggleBtn = DOM.searchToggleBtn;
    const searchInput = DOM.searchInput;
    const searchBtn = DOM.searchBtn;
    const searchResults = DOM.searchResults;
    const closeSearchBoxBtn = DOM.closeSearchBoxBtn;

    let searchTimeout;
    const SEARCH_HISTORY_KEY = 'cjydocs_search_history';

    // 保存搜索历史到localStorage
    const saveSearchHistory = (query) => {
        if (!query || query.trim().length < 2) return;

        try {
            let history = JSON.parse(localStorage.getItem(SEARCH_HISTORY_KEY) || '{}');
            history[currentCategory] = query;
            localStorage.setItem(SEARCH_HISTORY_KEY, JSON.stringify(history));
        } catch (err) {
            console.error('Failed to save search history:', err);
        }
    };

    // 搜索函数
    const performSearch = async () => {
        const query = searchInput.value.trim();

        if (query.length < 2) {
            searchResults.style.display = 'none';
            return;
        }

        // 保存搜索历史
        saveSearchHistory(query);

        try {
            const response = await fetch(
                `/api/search/${encodeURIComponent(currentCategory)}?q=${encodeURIComponent(query)}&currentDoc=${encodeURIComponent(currentDoc)}`
            );

            if (!response.ok) {
                await handleApiError(response, '搜索失败');
            }

            const data = await response.json();
            displaySearchResults(data);
            searchResults.style.display = 'block';

        } catch (err) {
            console.error('Search error:', err);
            displaySearchError(err.message || '搜索失败，请稍后重试');
        }
    };

    // 加载搜索历史
    const loadSearchHistory = () => {
        try {
            const history = JSON.parse(localStorage.getItem(SEARCH_HISTORY_KEY) || '{}');
            const lastQuery = history[currentCategory];

            if (lastQuery) {
                searchInput.value = lastQuery;
                // 自动搜索上次的内容
                performSearch();
            }
        } catch (err) {
            console.error('Failed to load search history:', err);
        }
    };

    // 清除其他分类的搜索历史
    const clearOtherCategoryHistory = () => {
        try {
            let history = JSON.parse(localStorage.getItem(SEARCH_HISTORY_KEY) || '{}');
            // 只保留当前分类
            history = { [currentCategory]: history[currentCategory] || '' };
            localStorage.setItem(SEARCH_HISTORY_KEY, JSON.stringify(history));
        } catch (err) {
            console.error('Failed to clear search history:', err);
        }
    };

    // 切换搜索框显示/隐藏
    searchToggleBtn.addEventListener('click', (e) => {
        e.stopPropagation();
        const isActive = searchContainer.classList.toggle('active');

        if (isActive) {
            // 加载搜索历史
            loadSearchHistory();
            // 聚焦到搜索框
            setTimeout(() => searchInput.focus(), 100);
        } else {
            // 隐藏搜索结果
            searchResults.style.display = 'none';
        }
    });

    // 输入时实时搜索（防抖） - 500ms 减少 API 调用
    searchInput.addEventListener('input', debounce(performSearch, 500));

    // 点击搜索按钮
    searchBtn.addEventListener('click', performSearch);

    // 按回车搜索
    searchInput.addEventListener('keypress', (e) => {
        if (e.key === 'Enter') {
            performSearch();
        }
    });

    // 关闭搜索框按钮
    closeSearchBoxBtn.addEventListener('click', (e) => {
        e.stopPropagation();
        searchContainer.classList.remove('active');
        searchResults.style.display = 'none';
    });

    // 点击搜索结果外部关闭
    searchResults.addEventListener('click', (e) => {
        // 如果点击的是搜索结果容器本身（即遮罩背景），关闭搜索
        if (e.target === searchResults) {
            searchResults.style.display = 'none';
        }
    });

    // 点击外部关闭
    document.addEventListener('click', (e) => {
        if (!searchResults.contains(e.target) &&
            !searchInput.contains(e.target) &&
            !searchBtn.contains(e.target) &&
            !searchContainer.contains(e.target) &&
            !searchToggleBtn.contains(e.target)) {
            searchResults.style.display = 'none';
        }
    });

    // 初始化时清除其他分类的历史
    clearOtherCategoryHistory();
}

// 显示搜索错误
function displaySearchError(message) {
    if (!DOM.currentDocResults) DOM.currentDocResults = document.getElementById('current-doc-results');
    if (!DOM.otherDocResults) DOM.otherDocResults = document.getElementById('other-docs-results');
    const currentDocSection = DOM.currentDocResults ? DOM.currentDocResults.querySelector('.results-list') : null;
    const otherDocsSection = DOM.otherDocResults ? DOM.otherDocResults.querySelector('.results-list') : null;

    // 清空之前的结果
    currentDocSection.innerHTML = '';
    otherDocsSection.innerHTML = '';

    // 隐藏分组标题
    DOM.currentDocResults.style.display = 'none';
    DOM.otherDocResults.style.display = 'none';

    // 显示错误信息
    if (currentDocSection) {
        currentDocSection.innerHTML = `<p class="search-error" style="color: #d73a49; padding: 20px; text-align: center;">${message}</p>`;
    }
    DOM.currentDocResults.style.display = 'block';

    // 显示搜索结果面板
    if (DOM.searchResults) {
        DOM.searchResults.style.display = 'block';
    }
}

// 显示搜索结果
function displaySearchResults(data) {
    // 获取搜索结果内容容器
    const searchResultsContent = document.querySelector('.search-results-content');

    // 检查是否完全没有结果
    const hasNoResults = data.currentDoc.length === 0 && data.otherDocs.length === 0;

    if (hasNoResults) {
        // 完全没有搜索结果时，显示友好的空状态
        // 先清空内容区
        searchResultsContent.innerHTML = '';

        // 创建空状态容器
        const emptyState = document.createElement('div');
        emptyState.className = 'empty-search-state';
        emptyState.innerHTML = `
            <div class="empty-icon">
                <svg width="64" height="64" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
                    <circle cx="11" cy="11" r="8" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
                    <path d="m21 21-4.35-4.35" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
                    <path d="M8 11h6" stroke="currentColor" stroke-width="2" stroke-linecap="round"/>
                </svg>
            </div>
            <h3 class="empty-title">没有找到相关内容</h3>
            <p class="empty-text">试试其他关键词，或者检查拼写是否正确</p>
        `;

        searchResultsContent.appendChild(emptyState);
        return;
    }

    // 有搜索结果时，需要恢复正常的DOM结构
    // 先检查是否需要重建结构（从空状态恢复）
    let currentDocResults = document.getElementById('current-doc-results');
    let otherDocResults = document.getElementById('other-docs-results');

    // 如果结构不存在（被空状态替换了），重新创建
    if (!currentDocResults || !otherDocResults) {
        searchResultsContent.innerHTML = `
            <div id="current-doc-results" class="results-section">
                <h3 class="results-section-title">当前文档</h3>
                <div class="results-list"></div>
            </div>
            <div id="other-docs-results" class="results-section">
                <h3 class="results-section-title">其他文档</h3>
                <div class="results-list"></div>
            </div>
        `;

        // 重新获取引用
        currentDocResults = document.getElementById('current-doc-results');
        otherDocResults = document.getElementById('other-docs-results');

        // 更新DOM缓存
        DOM.currentDocResults = currentDocResults;
        DOM.otherDocResults = otherDocResults;
    }

    const currentDocSection = currentDocResults.querySelector('.results-list');
    const otherDocsSection = otherDocResults.querySelector('.results-list');

    // 清空之前的结果
    currentDocSection.innerHTML = '';
    otherDocsSection.innerHTML = '';

    // 隐藏/显示分组标题
    currentDocResults.style.display =
        data.currentDoc.length > 0 ? 'block' : 'none';
    otherDocResults.style.display =
        data.otherDocs.length > 0 ? 'block' : 'none';

    // 渲染当前文档结果
    if (data.currentDoc.length > 0) {
        data.currentDoc.forEach(doc => {
            const docResults = createDocResultElement(doc, true);
            currentDocSection.appendChild(docResults);
        });
    }

    // 渲染其他文档结果
    if (data.otherDocs.length > 0) {
        data.otherDocs.forEach(doc => {
            const docResults = createDocResultElement(doc, false);
            otherDocsSection.appendChild(docResults);
        });
    }
}

// 创建搜索结果元素
function createDocResultElement(doc, isCurrent) {
    const container = document.createElement('div');
    container.className = 'search-result-item';

    const header = document.createElement('div');
    header.className = 'result-header';
    header.innerHTML = `
        <span class="result-doc-name">${doc.docName}</span>
        <span class="result-count">${doc.matchCount} 个匹配</span>
    `;
    container.appendChild(header);

    // 添加匹配片段
    doc.matches.forEach(match => {
        const matchItem = document.createElement('div');
        matchItem.className = 'result-match';

        // 高亮搜索词
        const highlightedSnippet = highlightSearchTerm(match.snippet, DOM.searchInput ? DOM.searchInput.value : '');

        matchItem.innerHTML = `
            <div class="match-line-number">行 ${match.line}</div>
            <div class="match-snippet">${highlightedSnippet}</div>
        `;

        // 点击跳转
        matchItem.onclick = () => {
            // 隐藏搜索框和搜索结果
            if (DOM.searchContainer) {
                DOM.searchContainer.classList.remove('active');
            }
            if (DOM.searchResults) {
                DOM.searchResults.style.display = 'none';
            }

            // 在移动端和平板上关闭所有侧边栏，确保用户能看到跳转后的内容
            if (window.innerWidth <= 1024) {
                if (DOM.leftSidebar && !DOM.leftSidebar.classList.contains('collapsed')) {
                    DOM.leftSidebar.classList.add('collapsed');
                }
                if (DOM.rightSidebar && !DOM.rightSidebar.classList.contains('collapsed')) {
                    DOM.rightSidebar.classList.add('collapsed');
                }
            }

            if (isCurrent) {
                // 当前文档，滚动到对应位置
                scrollToSearchMatch(match.fullLine);
            } else {
                // 其他文档，加载该文档并跳转到具体位置
                loadDocument(doc.docName, match.fullLine);
            }
        };

        container.appendChild(matchItem);
    });

    return container;
}

// 高亮搜索词
function highlightSearchTerm(text, searchTerm) {
    const regex = new RegExp(`(${searchTerm})`, 'gi');
    return text.replace(regex, '<mark class="search-highlight">$1</mark>');
}

// 滚动到搜索匹配位置
function scrollToSearchMatch(fullLine) {
    const content = document.getElementById('markdown-content');
    const searchText = fullLine.trim();

    if (!searchText) return;

    // 使用TreeWalker遍历所有文本节点
    const walker = document.createTreeWalker(
        content,
        NodeFilter.SHOW_TEXT,
        null
    );

    let found = false;
    let node;

    while (node = walker.nextNode()) {
        if (node.textContent.includes(searchText)) {
            const parentElement = node.parentElement;

            // 滚动到元素
            parentElement.scrollIntoView({ behavior: 'smooth', block: 'center' });

            // 临时高亮父元素
            parentElement.classList.add('temp-highlight');
            setTimeout(() => {
                parentElement.classList.remove('temp-highlight');
            }, 2000);

            found = true;
            break;
        }
    }

    // 如果没有找到精确匹配，尝试部分匹配
    if (!found) {
        const elements = content.querySelectorAll('p, li, td, th, h1, h2, h3, h4, h5, h6, blockquote, pre, code');

        for (const element of elements) {
            const text = element.textContent.trim();
            // 尝试匹配至少50%的内容
            if (text.length > 10 && searchText.includes(text.substring(0, Math.min(text.length, 50)))) {
                element.scrollIntoView({ behavior: 'smooth', block: 'center' });

                // 临时高亮
                element.classList.add('temp-highlight');
                setTimeout(() => {
                    element.classList.remove('temp-highlight');
                }, 2000);

                break;
            }
        }
    }
}

// 设置回到顶部按钮
function setupBackToTop() {
    if (!DOM.backToTopBtn) DOM.backToTopBtn = document.getElementById('back-to-top');
    const backToTopBtn = DOM.backToTopBtn;

    if (!backToTopBtn) return;

    // 初始状态：隐藏按钮
    backToTopBtn.classList.add('hidden');

    // 点击按钮滚动到顶部
    backToTopBtn.addEventListener('click', (e) => {
        e.preventDefault();
        e.stopPropagation();

        // 滚动到顶部
        window.scrollTo({
            top: 0,
            behavior: 'smooth'
        });
    });

    // 监听窗口滚动事件，根据滚动位置控制按钮显示/隐藏
    let scrollTimeout;

    const handleScroll = () => {
        clearTimeout(scrollTimeout);
        scrollTimeout = setTimeout(() => {
            const scrollTop = window.pageYOffset || document.documentElement.scrollTop;

            // 滚动超过 100px 时显示按钮
            if (scrollTop > 100) {
                backToTopBtn.classList.remove('hidden');
            } else {
                backToTopBtn.classList.add('hidden');
            }
        }, 100);
    };

    window.addEventListener('scroll', handleScroll);
}

// 编辑功能
function setupEditFeature() {
    // 缓存DOM元素
    if (!DOM.editBtn) DOM.editBtn = document.getElementById('edit-btn');
    if (!DOM.editorContainer) DOM.editorContainer = document.getElementById('editor-container');
    if (!DOM.markdownEditor) DOM.markdownEditor = document.getElementById('markdown-editor');
    if (!DOM.editorDocName) DOM.editorDocName = document.getElementById('editor-doc-name');

    if (!DOM.saveBtn) DOM.saveBtn = document.getElementById('save-btn');
    if (!DOM.cancelBtn) DOM.cancelBtn = document.getElementById('cancel-edit-btn');
    if (!DOM.contentArea) DOM.contentArea = document.getElementById('content-area');
    const saveBtn = DOM.saveBtn;
    const cancelBtn = DOM.cancelBtn;
    const contentArea = DOM.contentArea;

    // 编辑按钮点击事件 - 切换编辑/查看模式
    DOM.editBtn.addEventListener('click', () => {
        if (isEditMode) {
            exitEditMode();
        } else {
            enterEditMode();
        }
    });

    // 取消编辑
    cancelBtn.addEventListener('click', () => {
        exitEditMode();
    });

    // 保存按钮
    saveBtn.addEventListener('click', async () => {
        await saveDocument();
    });
}

// 进入编辑模式
function enterEditMode() {
    if (!currentDoc || !currentMarkdownContent) {
        alert('请先加载一个文档');
        return;
    }

    isEditMode = true;

    // 更新编辑按钮状态
    DOM.editBtn.classList.add('active');
    DOM.editBtn.title = '退出编辑';

    // 保存当前滚动位置
    if (!DOM.contentArea) DOM.contentArea = document.getElementById('content-area');
    savedScrollPosition = DOM.contentArea.scrollTop;

    // 找到当前视口中心附近的元素
    const visibleElement = findVisibleElement();

    // 隐藏渲染的内容
    if (!DOM.content) DOM.content = document.getElementById('markdown-content');
    DOM.content.style.display = 'none';

    // 显示编辑器
    DOM.editorContainer.style.display = 'flex';
    DOM.editorDocName.textContent = `${currentDoc}.md`;
    DOM.markdownEditor.value = currentMarkdownContent;

    // 定位光标到对应位置
    setTimeout(() => {
        if (visibleElement) {
            positionCursorByElement(visibleElement);
        } else {
            // 如果找不到元素，使用滚动比例
            const scrollRatio = savedScrollPosition / DOM.contentArea.scrollHeight;
            DOM.markdownEditor.scrollTop = DOM.markdownEditor.scrollHeight * scrollRatio;
        }
        DOM.markdownEditor.focus();
    }, 50);
}

// 找到当前视口中可见的元素
function findVisibleElement() {
    if (!DOM.contentArea) DOM.contentArea = document.getElementById('content-area');
    const viewportTop = DOM.contentArea.scrollTop;
    const viewportMiddle = viewportTop + DOM.contentArea.clientHeight / 3; // 视口上方1/3处

    // 查找所有重要元素（标题、段落等）
    const elements = DOM.content.querySelectorAll('h1, h2, h3, h4, h5, h6, p, li, blockquote, pre');

    let closestElement = null;
    let minDistance = Infinity;

    elements.forEach(el => {
        const rect = el.getBoundingClientRect();
        const elementTop = el.offsetTop;
        const distance = Math.abs(elementTop - viewportMiddle);

        if (distance < minDistance && elementTop <= viewportMiddle + 200) {
            minDistance = distance;
            closestElement = el;
        }
    });

    return closestElement;
}

// 根据元素定位光标
function positionCursorByElement(element) {
    const elementText = element.textContent.trim();

    if (!elementText) {
        return;
    }

    // 获取元素标签类型
    const tagName = element.tagName.toLowerCase();
    let searchText = elementText;

    // 如果是标题，添加markdown标记进行搜索
    if (tagName.match(/^h[1-6]$/)) {
        const level = parseInt(tagName[1]);
        const hashes = '#'.repeat(level);

        // 尝试匹配标题行
        const lines = currentMarkdownContent.split('\n');
        let targetLine = -1;

        for (let i = 0; i < lines.length; i++) {
            const line = lines[i].trim();
            // 匹配 "# 标题" 格式
            if (line.startsWith(hashes + ' ') && line.substring(level + 1).trim() === elementText) {
                targetLine = i;
                break;
            }
        }

        if (targetLine !== -1) {
            setEditorCursorToLine(targetLine);
            return;
        }
    }

    // 对于其他元素，搜索文本内容
    const lines = currentMarkdownContent.split('\n');
    for (let i = 0; i < lines.length; i++) {
        const line = lines[i].trim();
        // 移除markdown标记进行比较
        const cleanLine = line.replace(/^[#\-*>]+\s*/, '').trim();

        if (cleanLine.includes(elementText.substring(0, Math.min(50, elementText.length)))) {
            setEditorCursorToLine(i);
            return;
        }
    }
}

// 设置编辑器光标到指定行
function setEditorCursorToLine(lineNumber) {
    const lines = currentMarkdownContent.split('\n');

    // 计算目标位置的字符索引
    let charPosition = 0;
    for (let i = 0; i < lineNumber && i < lines.length; i++) {
        charPosition += lines[i].length + 1; // +1 for newline
    }

    // 设置光标位置
    DOM.markdownEditor.setSelectionRange(charPosition, charPosition);

    // 滚动到光标位置
    const linesBeforeCursor = lineNumber;
    const lineHeight = 22; // 大约的行高（px）
    const targetScrollTop = linesBeforeCursor * lineHeight - DOM.markdownEditor.clientHeight / 3;

    DOM.markdownEditor.scrollTop = Math.max(0, targetScrollTop);
}

// 退出编辑模式
function exitEditMode() {
    isEditMode = false;

    // 更新编辑按钮状态
    DOM.editBtn.classList.remove('active');
    DOM.editBtn.title = '编辑文档';

    // 隐藏编辑器
    DOM.editorContainer.style.display = 'none';

    // 显示渲染的内容
    DOM.content.style.display = 'block';

    // 恢复滚动位置
    if (!DOM.contentArea) DOM.contentArea = document.getElementById('content-area');
    setTimeout(() => {
        DOM.contentArea.scrollTop = savedScrollPosition;
    }, 0);
}

// 保存文档
async function saveDocument() {
    const newContent = DOM.markdownEditor.value;

    if (!currentCategory || !currentDoc) {
        alert('无法保存：缺少文档信息');
        return;
    }

    // 禁用保存按钮，防止重复点击
    ensureDOM('saveBtn', 'save-btn');
    DOM.saveBtn.disabled = true;
    DOM.saveBtn.textContent = '保存中...';

    try {
        const encodedDocName = encodeDocPath(currentDoc);
        const response = await fetch(`/api/doc/${encodeURIComponent(currentCategory)}/${encodedDocName}`, {
            method: 'PUT',
            headers: {
                'Content-Type': 'application/json'
            },
            body: JSON.stringify({
                content: newContent
            })
        });

        if (!response.ok) {
            await handleApiError(response, '保存失败');
        }

        const result = await response.json();

        // 更新本地的markdown内容
        currentMarkdownContent = newContent;

        // 重新渲染内容
        const html = marked.parse(newContent);
        DOM.content.innerHTML = html;

        // 重新生成TOC
        generateTOC();

        // 退出编辑模式
        exitEditMode();

        // 显示成功提示
        showSuccessMessage('保存成功！');

    } catch (error) {
        console.error('Save error:', error);
        alert(`保存失败：${error.message || '请稍后重试'}`);
    } finally {
        // 恢复保存按钮
        DOM.saveBtn.disabled = false;
        DOM.saveBtn.textContent = '保存';
    }
}

// 显示成功提示
function showSuccessMessage(message) {
    const toast = document.createElement('div');
    toast.className = 'success-toast';
    toast.textContent = message;
    toast.style.cssText = `
        position: fixed;
        top: 20px;
        right: 20px;
        background: #28a745;
        color: white;
        padding: 12px 24px;
        border-radius: 6px;
        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.15);
        z-index: 10000;
        animation: slideInRight 0.3s ease-out;
    `;

    document.body.appendChild(toast);

    setTimeout(() => {
        toast.style.animation = 'slideOutRight 0.3s ease-out';
        setTimeout(() => {
            document.body.removeChild(toast);
        }, 300);
    }, 2000);
}

// 为代码块添加复制按钮
function addCopyButtonsToCodeBlocks() {
    const codeBlocks = document.querySelectorAll('pre');

    codeBlocks.forEach(pre => {
        // 如果已经有复制按钮，跳过
        if (pre.querySelector('.code-copy-btn')) {
            return;
        }

        // 创建包装容器
        const wrapper = document.createElement('div');
        wrapper.className = 'code-block-wrapper';

        // 将pre元素包装起来
        pre.parentNode.insertBefore(wrapper, pre);
        wrapper.appendChild(pre);

        // 创建复制按钮
        const copyBtn = document.createElement('button');
        copyBtn.className = 'code-copy-btn';
        copyBtn.innerHTML = `
            <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
                <rect x="9" y="9" width="13" height="13" rx="2" ry="2"></rect>
                <path d="M5 15H4a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h9a2 2 0 0 1 2 2v1"></path>
            </svg>
            <span>复制</span>
        `;

        // 添加点击事件
        copyBtn.addEventListener('click', () => {
            copyCodeToClipboard(pre, copyBtn);
        });

        wrapper.appendChild(copyBtn);
    });
}

// 复制代码到剪贴板
async function copyCodeToClipboard(preElement, button) {
    const codeElement = preElement.querySelector('code');
    if (!codeElement) return;

    // 获取纯文本内容（去除HTML标签）
    const text = codeElement.textContent || codeElement.innerText;

    try {
        // 使用现代API复制
        await navigator.clipboard.writeText(text);

        // 显示成功状态
        showCopySuccess(button);

    } catch (err) {
        // 降级方案：使用传统方法
        const textArea = document.createElement('textarea');
        textArea.value = text;
        textArea.style.position = 'fixed';
        textArea.style.left = '-9999px';
        document.body.appendChild(textArea);
        textArea.select();

        try {
            document.execCommand('copy');
            showCopySuccess(button);
        } catch (err) {
            console.error('复制失败:', err);
            showCopyError(button);
        } finally {
            document.body.removeChild(textArea);
        }
    }
}

// 显示复制成功
function showCopySuccess(button) {
    // 更新按钮状态
    const originalHTML = button.innerHTML;
    button.classList.add('copied');
    button.innerHTML = `
        <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
            <polyline points="20 6 9 17 4 12"></polyline>
        </svg>
        <span>已复制</span>
    `;

    // 2秒后恢复原状
    setTimeout(() => {
        button.classList.remove('copied');
        button.innerHTML = originalHTML;
    }, 2000);
}

// 显示复制失败
function showCopyError(button) {
    const originalHTML = button.innerHTML;
    button.innerHTML = `
        <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
            <circle cx="12" cy="12" r="10"></circle>
            <line x1="12" y1="8" x2="12" y2="12"></line>
            <line x1="12" y1="16" x2="12.01" y2="16"></line>
        </svg>
        <span>复制失败</span>
    `;

    setTimeout(() => {
        button.innerHTML = originalHTML;
    }, 2000);
}

// ==================== 语音朗读功能 ====================
let speechSynthesis = window.speechSynthesis;
let currentUtterance = null;
let isPaused = false;
let isPlaying = false;
let currentHighlightElement = null;
let speechTexts = []; // 存储所有要朗读的文本段落
let currentSpeechIndex = 0; // 当前朗读的段落索引
let lastPausedElement = null; // 记录上次暂停的元素

// 初始化语音朗读
function initSpeech() {
    const speechBtn = document.getElementById('speech-btn');
    if (!speechBtn) return;

    // 检查浏览器是否支持语音合成
    if (!('speechSynthesis' in window)) {
        speechBtn.style.display = 'none';
        return;
    }

    // 某些浏览器需要先加载语音列表
    if (speechSynthesis.getVoices().length === 0) {
        speechSynthesis.addEventListener('voiceschanged', () => {}, { once: true });
    }

    // 添加点击事件监听
    speechBtn.addEventListener('click', (e) => {
        e.preventDefault();
        e.stopPropagation();
        handleSpeechButtonClick();
    });
}

// 获取当前视口中第一个可见的文本元素
function getFirstVisibleTextElement() {
    const content = document.getElementById('markdown-content');
    if (!content) return null;

    // 获取所有可朗读的文本元素
    const elements = content.querySelectorAll('p, h1, h2, h3, h4, h5, h6, li, blockquote, td, th');
    const viewportHeight = window.innerHeight;

    for (const element of elements) {
        // 跳过代码块内的元素
        if (element.closest('pre') || element.closest('code')) {
            continue;
        }

        const rect = element.getBoundingClientRect();
        // 检查元素是否在视口内（顶部在视口内或元素跨越视口）
        if ((rect.top >= 0 && rect.top < viewportHeight) ||
            (rect.top < 0 && rect.bottom > 0)) {
            return element;
        }
    }

    return null;
}

// 判断元素是否在视口内可见
function isElementVisible(element) {
    if (!element) return false;

    const rect = element.getBoundingClientRect();
    const windowHeight = window.innerHeight;

    // 元素至少有一部分在视口内
    return rect.bottom > 0 && rect.top < windowHeight;
}

// 处理语音按钮点击
function handleSpeechButtonClick() {
    const speechBtn = document.getElementById('speech-btn');

    if (isPlaying && !isPaused) {
        // 正在播放 -> 暂停
        pauseSpeech();
        speechBtn.classList.remove('playing');
        speechBtn.classList.add('paused');
    } else if (isPaused) {
        // 已暂停 -> 判断暂停位置是否可见
        if (isElementVisible(lastPausedElement)) {
            // 暂停位置可见，继续播放
            resumeSpeech();
            speechBtn.classList.remove('paused');
            speechBtn.classList.add('playing');
        } else {
            // 暂停位置不可见，从当前可见的第一行开始
            const currentVisibleElement = getFirstVisibleTextElement();
            stopSpeech();
            isPaused = false;
            isPlaying = false;
            startSpeechFromElement(currentVisibleElement);
            speechBtn.classList.remove('paused');
            speechBtn.classList.add('playing');
        }
    } else {
        // 未开始 -> 从当前滚动位置开始朗读
        const currentVisibleElement = getFirstVisibleTextElement();
        startSpeechFromElement(currentVisibleElement);
        speechBtn.classList.add('playing');
    }
}

// 从指定元素开始朗读
function startSpeechFromElement(startElement) {
    if (!DOM.content) DOM.content = document.getElementById('markdown-content');

    if (!DOM.content || !DOM.content.textContent.trim()) {
        alert('请先加载文档内容');
        return;
    }

    // 提取要朗读的文本
    speechTexts = extractSpeechTexts();

    if (speechTexts.length === 0) {
        alert('没有找到可朗读的文本内容');
        return;
    }

    // 如果指定了起始元素，找到它在列表中的位置
    if (startElement) {
        const startIndex = speechTexts.findIndex(item => item.element === startElement);
        currentSpeechIndex = startIndex !== -1 ? startIndex :
            // 如果没找到，找最接近的元素
            speechTexts.reduce((closestIdx, item, idx) => {
                const rect = item.element.getBoundingClientRect();
                const distance = Math.abs(rect.top);
                const prevRect = speechTexts[closestIdx].element.getBoundingClientRect();
                return distance < Math.abs(prevRect.top) ? idx : closestIdx;
            }, 0);
    } else {
        currentSpeechIndex = 0;
    }

    isPlaying = true;
    isPaused = false;
    lastPausedElement = null; // 清除之前的暂停记录

    // 开始朗读
    speakNextSegment();
}

// 提取要朗读的文本（跳过图片和代码块）
function extractSpeechTexts() {
    const texts = [];
    const content = document.getElementById('markdown-content');

    if (!content) return texts;

    // 获取所有文本元素，排除代码块和图片
    const elements = content.querySelectorAll('p, h1, h2, h3, h4, h5, h6, li, blockquote, td, th');

    elements.forEach(element => {
        // 跳过代码块内的元素
        if (element.closest('pre') || element.closest('code')) {
            return;
        }

        // 获取纯文本内容
        const text = element.textContent.trim();

        if (text) {
            texts.push({
                text: text,
                element: element
            });
        }
    });

    return texts;
}

// 朗读下一段
function speakNextSegment() {
    if (currentSpeechIndex >= speechTexts.length) {
        // 朗读完成
        console.log('朗读完成');
        stopSpeech();
        return;
    }

    const segment = speechTexts[currentSpeechIndex];
    console.log('朗读第', currentSpeechIndex + 1, '段，内容:', segment.text.substring(0, 50) + '...');

    // 清除之前的高亮
    clearHighlight();

    // 高亮当前段落
    highlightElement(segment.element);

    // 取消之前的朗读（如果有）
    if (speechSynthesis.speaking) {
        speechSynthesis.cancel();
    }

    // 创建语音实例
    currentUtterance = new SpeechSynthesisUtterance(segment.text);

    // 设置语音参数 - 优先尝试中文，如果不支持则使用默认语言
    const voices = speechSynthesis.getVoices();
    const chineseVoice = voices.find(voice =>
        voice.lang === 'zh-CN' ||
        voice.lang.startsWith('zh') ||
        voice.lang.includes('Chinese')
    );

    if (chineseVoice) {
        currentUtterance.voice = chineseVoice;
        currentUtterance.lang = chineseVoice.lang;
    } else {
        currentUtterance.lang = 'zh-CN'; // 仍然设置为中文，让系统选择最佳声音
    }

    currentUtterance.rate = 1.0; // 语速
    currentUtterance.pitch = 1.0; // 音调
    currentUtterance.volume = 1.0; // 音量

    // 语音错误处理
    currentUtterance.onerror = (event) => {
        alert('语音朗读出现错误，请检查浏览器设置');
        stopSpeech();
    };

    // 语音结束事件
    currentUtterance.onend = () => {
        currentSpeechIndex++;
        if (isPlaying && !isPaused) {
            // 延迟一点再朗读下一段，让用户有停顿感
            setTimeout(() => {
                if (isPlaying && !isPaused) {
                    speakNextSegment();
                }
            }, 300);
        }
    };

    // 语音边界事件（用于更精确的高亮，但Web Speech API支持有限）
    currentUtterance.onboundary = (event) => {
        // 检查是否需要滚动
        checkAndScroll(segment.element);
    };

    // 开始朗读
    try {
        speechSynthesis.speak(currentUtterance);
    } catch (error) {
        alert('无法启动语音朗读，请检查浏览器权限');
        stopSpeech();
    }
}

// 高亮元素
function highlightElement(element) {
    if (!element) return;

    currentHighlightElement = element;
    element.classList.add('speech-highlight');

    // 确保元素在视口内
    checkAndScroll(element);
}

// 清除高亮
function clearHighlight() {
    if (currentHighlightElement) {
        currentHighlightElement.classList.remove('speech-highlight');
        currentHighlightElement = null;
    }
}

// 检查并滚动
function checkAndScroll(element) {
    if (!element) return;

    const rect = element.getBoundingClientRect();
    const viewportHeight = window.innerHeight;

    // 如果元素在视口底部附近（最后20%），向下滚动
    if (rect.bottom > viewportHeight * 0.8) {
        element.scrollIntoView({
            behavior: 'smooth',
            block: 'center'
        });
    }
    // 如果元素在视口顶部之上，也滚动到中心
    else if (rect.top < 0) {
        element.scrollIntoView({
            behavior: 'smooth',
            block: 'center'
        });
    }
}

// 暂停朗读
function pauseSpeech() {
    if (speechSynthesis && isPlaying) {
        // 记录当前朗读的元素（优先使用当前高亮的元素）
        if (currentHighlightElement) {
            lastPausedElement = currentHighlightElement;
        } else if (currentSpeechIndex < speechTexts.length) {
            lastPausedElement = speechTexts[currentSpeechIndex].element;
        }
        speechSynthesis.pause();
        isPaused = true;
    }
}

// 继续朗读
function resumeSpeech() {
    if (speechSynthesis && isPaused) {
        speechSynthesis.resume();
        isPaused = false;
    }
}

// 停止朗读
function stopSpeech() {
    if (speechSynthesis) {
        speechSynthesis.cancel();
    }
    cleanupSpeechState();
}

// 清理语音状态
function cleanupSpeechState() {
    isPlaying = false;
    isPaused = false;
    currentSpeechIndex = 0;
    currentUtterance = null;
    speechTexts = [];
    lastPausedElement = null; // 清除暂停记录

    // 清除高亮
    clearHighlight();

    // 重置按钮状态
    const speechBtn = document.getElementById('speech-btn');
    if (speechBtn) {
        speechBtn.classList.remove('playing', 'paused');
    }
}

// 当加载新文档时停止朗读
const originalLoadDocument = loadDocument;
loadDocument = function(docName, scrollToText = null) {
    // 停止当前朗读
    stopSpeech();

    // 调用原始的loadDocument
    return originalLoadDocument.call(this, docName, scrollToText);
};

// 初始化
document.addEventListener('DOMContentLoaded', () => {
    loadDocList();
    setupSidebarToggles();
    initSearch();
    setupBackToTop();
    setupEditFeature();
    // 延迟初始化语音朗读，确保DOM完全加载
    setTimeout(() => {
        initSpeech();
    }, 500);
});

// 处理浏览器后退/前进
window.addEventListener('popstate', () => {
    const { doc } = getQueryParams();
    if (doc) {
        loadDocument(doc);
    }
});