search-renderer.js

// 文件类型定义
const FILE_TYPES = {
    images: ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.svg', '.webp', '.ico', '.tiff'],
    documents: ['.pdf', '.doc', '.docx', '.txt', '.rtf', '.odt', '.xls', '.xlsx', '.ppt', '.pptx', '.csv'],
    videos: ['.mp4', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.webm', '.m4v'],
    audio: ['.mp3', '.wav', '.flac', '.aac', '.ogg', '.wma', '.m4a'],
    archives: ['.zip', '.rar', '.7z', '.tar', '.gz', '.bz2', '.xz']
};

// 多语言字典
const i18n = {
    'zh-CN': {
        title: '文件搜索工具',
        appTitle: '文件搜索工具',
        headerTitle: '快速文件搜索',
        headerSubtitle: '高效搜索文件,支持智能范围缩小和多种文件类型过滤',
        folderPlaceholder: '输入要搜索的文件夹路径 (例如: C:\\Users\\Desktop)',
        browseBtn: '浏览',
        searchPlaceholder: '输入文件名或关键词 (可选)',
        searchBtn: '搜索',
        stopBtn: '停止',
        fileTypeLabel: '文件类型:',
        filterAll: '全部',
        filterImages: '图像',
        filterDocuments: '文档',
        filterVideos: '视频',
        filterAudio: '音频',
        filterArchives: '压缩包',
        foundLabel: '找到文件:',
        scannedLabel: '已扫描:',
        timeLabel: '搜索时间:',
        resultsTitle: '搜索结果',
        sortDefault: '默认排序',
        sortNameAsc: '按名称 (A-Z)',
        sortNameDesc: '按名称 (Z-A)',
        sortDateDesc: '按日期 (最新)',
        sortDateAsc: '按日期 (最旧)',
        sortSizeDesc: '按大小 (从大到小)',
        sortSizeAsc: '按大小 (从小到大)',
        exportBtn: '导出',
        clearBtn: '清空结果',
        emptyState: '开始搜索以查看结果',
        cancelBtn: '取消',
        confirmBtn: '确定',
        // 动态文本
        alertTitle: '提示',
        errorTitle: '错误',
        warningTitle: '警告',
        inputFolderReq: '请输入文件夹路径',
        searchSearching: '正在搜索...',
        searchStopped: '搜索已停止',
        searchError: '搜索出错: ',
        searchDone: '搜索完成! 找到 {0} 个文件',
        scanning: '正在扫描: ',
        size: '大小',
        type: '类型',
        createTime: '创建时间',
        modTime: '修改时间',
        accessTime: '访问时间',
        noExt: '无扩展名',
        openFolderTip: '打开文件夹',
        openFileTip: '浏览文件',
        propsTip: '查看属性',
        exportEmpty: '没有可导出的文件',
        exportWarnLarge: '您将导出超过 2GB 的文件或整个磁盘内容！\n总大小: {0}\n请输入 "Yes" 确认操作:',
        exportWarnTypeYes: '请输入 "Yes" 以继续',
        exportDestReq: '请选择导出目标文件夹',
        exportDestSame: '导出目标文件夹不能是搜索目录或其子目录！',
        exporting: '正在导出文件...',
        exportDone: '导出完成！成功: {0}, 失败: {1}',
        exportError: '导出出错: '
    },
    'en': {
        title: 'File Search Tool',
        appTitle: 'File Search Tool',
        headerTitle: 'Fast File Search',
        headerSubtitle: 'Efficiently search files with smart scoping and type filtering',
        folderPlaceholder: 'Enter folder path to search (e.g. C:\\Users\\Desktop)',
        browseBtn: 'Browse',
        searchPlaceholder: 'Enter filename or keywords (optional)',
        searchBtn: 'Search',
        stopBtn: 'Stop',
        fileTypeLabel: 'File Types:',
        filterAll: 'All',
        filterImages: 'Images',
        filterDocuments: 'Documents',
        filterVideos: 'Videos',
        filterAudio: 'Audio',
        filterArchives: 'Archives',
        foundLabel: 'Found:',
        scannedLabel: 'Scanned:',
        timeLabel: 'Search Time:',
        resultsTitle: 'Search Results',
        sortDefault: 'Default Sort',
        sortNameAsc: 'Name (A-Z)',
        sortNameDesc: 'Name (Z-A)',
        sortDateDesc: 'Date (Newest)',
        sortDateAsc: 'Date (Oldest)',
        sortSizeDesc: 'Size (Large-Small)',
        sortSizeAsc: 'Size (Small-Large)',
        exportBtn: 'Export',
        clearBtn: 'Clear',
        emptyState: 'Start searching to see results',
        cancelBtn: 'Cancel',
        confirmBtn: 'OK',
        // Dynamic texts
        alertTitle: 'Info',
        errorTitle: 'Error',
        warningTitle: 'Warning',
        inputFolderReq: 'Please enter a folder path',
        searchSearching: 'Searching...',
        searchStopped: 'Search stopped',
        searchError: 'Search error: ',
        searchDone: 'Search complete! Found {0} files',
        scanning: 'Scanning: ',
        size: 'Size',
        type: 'Type',
        createTime: 'Created',
        modTime: 'Modified',
        accessTime: 'Accessed',
        noExt: 'No Extension',
        openFolderTip: 'Open Folder',
        openFileTip: 'Open File',
        propsTip: 'Properties',
        exportEmpty: 'No files to export',
        exportWarnLarge: 'You are about to export more than 2GB or an entire disk!\nTotal size: {0}\nPlease type "Yes" to confirm:',
        exportWarnTypeYes: 'Please type "Yes" to continue',
        exportDestReq: 'Please select an export destination folder',
        exportDestSame: 'Destination folder cannot be inside the search directory!',
        exporting: 'Exporting files...',
        exportDone: 'Export complete! Success: {0}, Failed: {1}',
        exportError: 'Export error: '
    }
};

let currentLang = 'zh-CN';
let currentTheme = 'light';

function t(key, ...args) {
    let text = i18n[currentLang][key] || key;
    args.forEach((arg, i) => {
        text = text.replace(`{${i}}`, arg);
    });
    return text;
}

function updateI18n() {
    document.querySelectorAll('[data-i18n]').forEach(el => {
        el.textContent = t(el.getAttribute('data-i18n'));
    });
    document.querySelectorAll('[data-i18n-placeholder]').forEach(el => {
        el.placeholder = t(el.getAttribute('data-i18n-placeholder'));
    });
    // Update sort options
    const sortSelect = document.getElementById('sortSelect');
    if (sortSelect) {
        Array.from(sortSelect.options).forEach(opt => {
            opt.text = t(opt.getAttribute('data-i18n'));
        });
    }
}

// Custom Modal Implementation
const modalOverlay = document.getElementById('customModal');
const modalTitle = document.getElementById('modalTitle');
const modalContent = document.getElementById('modalContent');
const modalInputContainer = document.getElementById('modalInputContainer');
const modalInput = document.getElementById('modalInput');
const modalCancelBtn = document.getElementById('modalCancelBtn');
const modalOkBtn = document.getElementById('modalOkBtn');

let modalResolve = null;

function showModal({ title, content, type = 'alert' }) {
    return new Promise((resolve) => {
        modalTitle.textContent = title;
        modalContent.textContent = content;
        
        modalInputContainer.style.display = type === 'prompt' ? 'block' : 'none';
        modalCancelBtn.style.display = (type === 'confirm' || type === 'prompt') ? 'block' : 'none';
        
        if (type === 'prompt') {
            modalInput.value = '';
            setTimeout(() => modalInput.focus(), 100);
        }
        
        modalResolve = resolve;
        modalOverlay.classList.add('active');
    });
}

function closeModal(result) {
    modalOverlay.classList.remove('active');
    if (modalResolve) {
        modalResolve(result);
        modalResolve = null;
    }
}

modalOkBtn.addEventListener('click', () => {
    if (modalInputContainer.style.display === 'block') {
        closeModal(modalInput.value);
    } else {
        closeModal(true);
    }
});

modalCancelBtn.addEventListener('click', () => {
    closeModal(false);
});

async function customAlert(content, title) {
    await showModal({ title: title || t('alertTitle'), content, type: 'alert' });
}

async function customConfirm(content, title) {
    return await showModal({ title: title || t('alertTitle'), content, type: 'confirm' });
}

async function customPrompt(content, title) {
    return await showModal({ title: title || t('alertTitle'), content, type: 'prompt' });
}


// 全局状态
let searchState = {
    isSearching: false,
    shouldStop: false,
    foundFiles: [],
    selectedFiles: new Set(), // 选中的文件
    scannedPaths: new Set(),
    startTime: null,
    scannedCount: 0,
    currentSort: 'default',
    displayedCount: 0, // 当前显示的文件数量
    batchSize: 1000, // 每批显示1000个
    virtualScroll: {
        itemHeight: 88, // 每个结果项的高度
        visibleCount: 20, // 可见项数量
        startIndex: 0,
        endIndex: 20
    }
};

// DOM 元素
const elements = {
    folderPath: document.getElementById('folderPath'),
    searchQuery: document.getElementById('searchQuery'),
    searchBtn: document.getElementById('searchBtn'),
    stopBtn: document.getElementById('stopBtn'),
    browseFolderBtn: document.getElementById('browseFolderBtn'),
    progressBar: document.getElementById('progressBar'),
    statusMessage: document.getElementById('statusMessage'),
    statsBar: document.getElementById('statsBar'),
    foundCount: document.getElementById('foundCount'),
    scannedCount: document.getElementById('scannedCount'),
    searchTime: document.getElementById('searchTime'),
    resultsContainer: document.getElementById('resultsContainer'),
    resultsList: document.getElementById('resultsList'),
    clearResultsBtn: document.getElementById('clearResultsBtn'),
    exportBtn: document.getElementById('exportBtn'),
    deleteBtn: document.getElementById('deleteBtn'),
    selectAllBtn: document.getElementById('selectAllBtn'),
    sortSelect: document.getElementById('sortSelect'),
    filterAll: document.getElementById('filterAll'),
    filterImages: document.getElementById('filterImages'),
    filterDocuments: document.getElementById('filterDocuments'),
    filterVideos: document.getElementById('filterVideos'),
    filterAudio: document.getElementById('filterAudio'),
    filterArchives: document.getElementById('filterArchives'),
    langToggleBtn: document.getElementById('langToggleBtn'),
    themeToggleBtn: document.getElementById('themeToggleBtn')
};

// 语言切换
elements.langToggleBtn.addEventListener('click', () => {
    currentLang = currentLang === 'zh-CN' ? 'en' : 'zh-CN';
    updateI18n();
    if (!searchState.isSearching && searchState.foundFiles.length > 0) {
        renderResults();
    }
});

// 主题切换
elements.themeToggleBtn.addEventListener('click', () => {
    currentTheme = currentTheme === 'light' ? 'dark' : 'light';
    document.documentElement.setAttribute('data-theme', currentTheme);
});

// 文件类型过滤器管理
elements.filterAll.addEventListener('change', (e) => {
    if (e.target.checked) {
        elements.filterImages.checked = false;
        elements.filterDocuments.checked = false;
        elements.filterVideos.checked = false;
        elements.filterAudio.checked = false;
        elements.filterArchives.checked = false;
    }
});

[elements.filterImages, elements.filterDocuments, elements.filterVideos, 
 elements.filterAudio, elements.filterArchives].forEach(checkbox => {
    checkbox.addEventListener('change', (e) => {
        if (e.target.checked) {
            elements.filterAll.checked = false;
        }
    });
});

// 浏览文件夹
elements.browseFolderBtn.addEventListener('click', async () => {
    const path = await window.electronAPI.selectFolder();
    if (path) {
        elements.folderPath.value = path;
    }
});

// 搜索按钮
elements.searchBtn.addEventListener('click', () => {
    startSearch();
});

// 停止按钮
elements.stopBtn.addEventListener('click', () => {
    stopSearch();
});

// 清空结果按钮
elements.clearResultsBtn.addEventListener('click', () => {
    clearResults();
});

// 全选按钮
elements.selectAllBtn.addEventListener('click', () => {
    toggleSelectAll();
});

// 删除按钮
elements.deleteBtn.addEventListener('click', async () => {
    await deleteSelectedFiles();
});

// 回车键搜索
elements.searchQuery.addEventListener('keypress', (e) => {
    if (e.key === 'Enter') {
        startSearch();
    }
});

// 排序切换
elements.sortSelect.addEventListener('change', (e) => {
    searchState.currentSort = e.target.value;
    renderResults();
});

// 导出按钮
elements.exportBtn.addEventListener('click', async () => {
    const filesToExport = searchState.selectedFiles.size > 0 
        ? Array.from(searchState.selectedFiles).map(path => searchState.foundFiles.find(f => f.path === path))
        : searchState.foundFiles;
    
    if (filesToExport.length === 0) {
        await customAlert(t('exportEmpty'), t('alertTitle'));
        return;
    }

    let totalSize = 0;
    filesToExport.forEach(f => totalSize += f.size);
    
    const pathStr = elements.folderPath.value.trim();
    const isRoot = /^[a-zA-Z]:\\?$/.test(pathStr) || pathStr === '/' || pathStr === '\\';
    const isLarge = totalSize > 2 * 1024 * 1024 * 1024; // 2GB

    if (isRoot || isLarge) {
        const input = await customPrompt(t('exportWarnLarge', formatFileSize(totalSize)), t('warningTitle'));
        if (input !== 'Yes') {
            await customAlert(t('exportWarnTypeYes'), t('alertTitle'));
            return;
        }
    }

    const destPath = await window.electronAPI.selectFolder();
    if (!destPath) {
        return;
    }
    
    // 检查目标路径不能是搜索路径
    if (destPath.startsWith(elements.folderPath.value.trim())) {
        await customAlert(t('exportDestSame'), t('errorTitle'));
        return;
    }

    // 显示导出进度
    elements.progressBar.style.display = 'block';
    updateStatus(t('exporting'), true);
    
    try {
        const result = await window.electronAPI.exportFiles({
            files: filesToExport,
            destFolder: destPath
        });
        
        elements.progressBar.style.display = 'none';
        await customAlert(t('exportDone', result.successCount, result.failCount), t('alertTitle'));
        updateStatus('');
    } catch (error) {
        elements.progressBar.style.display = 'none';
        console.error('Export error:', error);
        await customAlert(t('exportError') + error.message, t('errorTitle'));
        updateStatus('');
    }
});

// 加载更多结果
function loadMoreResults() {
    const currentDisplayed = searchState.displayedCount;
    const totalFiles = searchState.foundFiles.length;
    
    if (currentDisplayed >= totalFiles) {
        return; // 已经全部显示
    }
    
    // 获取排序后的文件列表
    let files = [...searchState.foundFiles];
    const sort = searchState.currentSort;
    
    if (sort === 'nameAsc') {
        files.sort((a, b) => a.name.localeCompare(b.name));
    } else if (sort === 'nameDesc') {
        files.sort((a, b) => b.name.localeCompare(a.name));
    } else if (sort === 'dateDesc') {
        files.sort((a, b) => new Date(b.mtime) - new Date(a.mtime));
    } else if (sort === 'dateAsc') {
        files.sort((a, b) => new Date(a.mtime) - new Date(b.mtime));
    } else if (sort === 'sizeDesc') {
        files.sort((a, b) => b.size - a.size);
    } else if (sort === 'sizeAsc') {
        files.sort((a, b) => a.size - b.size);
    }
    
    // 计算要显示的范围
    const endIndex = Math.min(currentDisplayed + searchState.batchSize, totalFiles);
    const filesToShow = files.slice(currentDisplayed, endIndex);
    
    // 移除"继续显示"按钮
    const existingBtn = elements.resultsList.querySelector('.load-more-container');
    if (existingBtn) {
        existingBtn.remove();
    }
    
    // 批量添加到DOM
    const fragment = document.createDocumentFragment();
    filesToShow.forEach(file => {
        fragment.appendChild(createResultItem(file));
    });
    elements.resultsList.appendChild(fragment);
    
    // 更新显示计数
    searchState.displayedCount = endIndex;
    
    // 更新"继续显示"按钮
    updateLoadMoreButton();
}

// 添加"继续显示"按钮
function addLoadMoreButton(remainingCount) {
    // 先移除已存在的按钮,避免重复
    const existingBtn = elements.resultsList.querySelector('.load-more-container');
    if (existingBtn) {
        existingBtn.remove();
    }
    
    const container = document.createElement('div');
    container.className = 'load-more-container';
    container.innerHTML = `
        <button class="btn btn-primary load-more-btn">
            <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
                <polyline points="6 9 12 15 18 9"></polyline>
            </svg>
            <span>继续显示 (剩余 ${remainingCount} 个文件)</span>
        </button>
    `;
    
    const btn = container.querySelector('.load-more-btn');
    btn.addEventListener('click', loadMoreResults);
    
    elements.resultsList.appendChild(container);
}

// 更新"继续显示"按钮状态
function updateLoadMoreButton() {
    const existingBtn = elements.resultsList.querySelector('.load-more-container');
    
    // 检查是否需要显示按钮
    const shouldShow = searchState.displayedCount >= searchState.batchSize && 
                       searchState.foundFiles.length > searchState.displayedCount;
    
    if (shouldShow) {
        const remaining = searchState.foundFiles.length - searchState.displayedCount;
        
        if (existingBtn) {
            // 更新现有按钮的文本
            const span = existingBtn.querySelector('span');
            if (span) {
                span.textContent = `继续显示 (剩余 ${remaining} 个文件)`;
            }
        } else {
            // 添加新按钮
            addLoadMoreButton(remaining);
        }
    } else {
        // 移除按钮
        if (existingBtn) {
            existingBtn.remove();
        }
    }
}

// 删除选中的文件
async function deleteSelectedFiles() {
    if (searchState.selectedFiles.size === 0) {
        await customAlert('请先选择要删除的文件', t('alertTitle'));
        return;
    }

    const confirmed = await customConfirm(
        `确定要删除选中的 ${searchState.selectedFiles.size} 个文件吗?\n此操作不可恢复!`,
        t('warningTitle')
    );
    
    if (!confirmed) return;

    const filesToDelete = Array.from(searchState.selectedFiles).map(path => 
        searchState.foundFiles.find(f => f.path === path)
    );

    elements.progressBar.style.display = 'block';
    updateStatus('正在删除文件...', true);
    
    try {
        const result = await window.electronAPI.deleteFiles({
            files: filesToDelete
        });
        
        elements.progressBar.style.display = 'none';
        await customAlert(`删除完成! 成功: ${result.successCount}, 失败: ${result.failCount}`, t('alertTitle'));
        
        // 从结果中移除已删除的文件
        searchState.foundFiles = searchState.foundFiles.filter(f => !searchState.selectedFiles.has(f.path));
        searchState.selectedFiles.clear();
        
        // 重新渲染结果
        searchState.displayedCount = 0;
        renderResults();
        updateStats();
        updateStatus('');
    } catch (error) {
        elements.progressBar.style.display = 'none';
        console.error('Delete error:', error);
        await customAlert('删除出错: ' + error.message, t('errorTitle'));
        updateStatus('');
    }
}

// 全选/取消全选
function toggleSelectAll() {
    if (searchState.selectedFiles.size === searchState.foundFiles.length) {
        // 取消全选
        searchState.selectedFiles.clear();
    } else {
        // 全选
        searchState.foundFiles.forEach(f => searchState.selectedFiles.add(f.path));
    }
    updateSelectAllButton();
    renderResults();
}

// 更新全选按钮文本
function updateSelectAllButton() {
    const btn = elements.selectAllBtn;
    if (searchState.selectedFiles.size === searchState.foundFiles.length && searchState.foundFiles.length > 0) {
        btn.textContent = '取消全选';
    } else {
        btn.textContent = `全选 (${searchState.selectedFiles.size}/${searchState.foundFiles.length})`;
    }
}

// 开始搜索
async function startSearch() {
    const folderPath = elements.folderPath.value.trim();
    
    if (!folderPath) {
        await customAlert(t('inputFolderReq'), t('alertTitle'));
        return;
    }

    // 重置状态
    searchState = {
        isSearching: true,
        shouldStop: false,
        foundFiles: [],
        selectedFiles: new Set(),
        scannedPaths: new Set(),
        startTime: Date.now(),
        scannedCount: 0,
        currentSort: elements.sortSelect.value,
        displayedCount: 0,
        batchSize: 1000,
        virtualScroll: {
            itemHeight: 88,
            visibleCount: 20,
            startIndex: 0,
            endIndex: 20
        }
    };

    // 更新UI
    elements.searchBtn.style.display = 'none';
    elements.stopBtn.style.display = 'flex';
    elements.progressBar.style.display = 'block';
    elements.statsBar.style.display = 'flex';
    elements.resultsContainer.style.display = 'block';
    elements.resultsList.innerHTML = '';
    searchState.displayedCount = 0; // 重置显示计数
    updateStats();

    // 获取过滤器
    const filters = getActiveFilters();
    const searchQuery = elements.searchQuery.value.trim().toLowerCase();

    // 开始搜索
    updateStatus(t('searchSearching'), true);
    
    try {
        await window.electronAPI.searchFiles({
            folderPath,
            searchQuery,
            filters
        });
    } catch (error) {
        console.error('搜索错误:', error);
        updateStatus(t('searchError') + error.message, false);
    }
}

// 停止搜索
function stopSearch() {
    searchState.shouldStop = true;
    searchState.isSearching = false;
    elements.searchBtn.style.display = 'flex';
    elements.stopBtn.style.display = 'none';
    elements.progressBar.style.display = 'none';
    
    // 清除节流定时器并立即渲染剩余结果
    if (addResultThrottle) {
        clearTimeout(addResultThrottle);
        addResultThrottle = null;
    }
    
    // 渲染所有待处理的结果
    if (pendingResults.length > 0) {
        if (searchState.currentSort === 'default') {
            const canAddMore = searchState.displayedCount < searchState.batchSize;
            
            if (canAddMore) {
                const toAdd = pendingResults.slice(0, searchState.batchSize - searchState.displayedCount);
                const fragment = document.createDocumentFragment();
                
                toAdd.forEach(file => {
                    fragment.appendChild(createResultItem(file));
                });
                
                elements.resultsList.appendChild(fragment);
                searchState.displayedCount += toAdd.length;
            }
        }
        pendingResults = [];
    }
    
    // 更新"继续显示"按钮
    updateLoadMoreButton();
    
    updateStatus(t('searchStopped'), false);
    updateSelectAllButton();
    window.electronAPI.stopSearch();
}

// 清空结果
function clearResults() {
    searchState.foundFiles = [];
    searchState.selectedFiles.clear();
    searchState.scannedPaths.clear();
    searchState.scannedCount = 0;
    searchState.displayedCount = 0;
    
    elements.resultsList.innerHTML = `
        <div class="empty-state">
            <svg viewBox="0 0 24 24">
                <circle cx="11" cy="11" r="8"/>
                <path d="m21 21-4.35-4.35"/>
            </svg>
            <p>${t('emptyState')}</p>
        </div>
    `;
    
    elements.resultsContainer.style.display = 'none';
    elements.statsBar.style.display = 'none';
    updateStats();
    updateSelectAllButton();
    updateStatus('', false);
}

// 获取激活的过滤器
function getActiveFilters() {
    if (elements.filterAll.checked) {
        return null; // 搜索所有文件
    }

    const filters = [];
    if (elements.filterImages.checked) filters.push(...FILE_TYPES.images);
    if (elements.filterDocuments.checked) filters.push(...FILE_TYPES.documents);
    if (elements.filterVideos.checked) filters.push(...FILE_TYPES.videos);
    if (elements.filterAudio.checked) filters.push(...FILE_TYPES.audio);
    if (elements.filterArchives.checked) filters.push(...FILE_TYPES.archives);

    return filters.length > 0 ? filters : null;
}

// 更新统计信息
function updateStats() {
    elements.foundCount.textContent = searchState.foundFiles.length;
    elements.scannedCount.textContent = searchState.scannedCount;
    
    if (searchState.startTime) {
        const elapsed = ((Date.now() - searchState.startTime) / 1000).toFixed(1);
        elements.searchTime.textContent = elapsed + 's';
    }
}

// 更新状态消息
function updateStatus(message, showSpinner = false) {
    if (showSpinner) {
        elements.statusMessage.innerHTML = `
            <div class="loading-spinner"></div>
            <span>${escapeHtml(message)}</span>
        `;
    } else {
        elements.statusMessage.textContent = message;
    }
}

// 添加搜索结果并渲染 (优化版 - 使用节流)
let addResultThrottle = null;
let pendingResults = [];

function addResult(fileInfo) {
    if (searchState.scannedPaths.has(fileInfo.path)) {
        return;
    }

    searchState.scannedPaths.add(fileInfo.path);
    searchState.foundFiles.push(fileInfo);
    pendingResults.push(fileInfo);
    
    // 节流渲染 - 每100ms或累积50个结果才渲染一次
    if (!addResultThrottle) {
        addResultThrottle = setTimeout(() => {
            if (searchState.currentSort === 'default' && pendingResults.length > 0) {
                // 只渲染在显示范围内的结果
                const canAddMore = searchState.displayedCount < searchState.batchSize;
                
                if (canAddMore) {
                    const toAdd = pendingResults.slice(0, searchState.batchSize - searchState.displayedCount);
                    const fragment = document.createDocumentFragment();
                    
                    toAdd.forEach(file => {
                        fragment.appendChild(createResultItem(file));
                    });
                    
                    elements.resultsList.appendChild(fragment);
                    searchState.displayedCount += toAdd.length;
                }
                
                // 检查是否需要显示"继续显示"按钮
                updateLoadMoreButton();
                
                pendingResults = [];
            }
            updateStats();
            updateSelectAllButton();
            addResultThrottle = null;
        }, 100);
    }
}

// 批量添加结果 (新增)
function addResultsBatch(files) {
    const newFiles = [];
    
    files.forEach(fileInfo => {
        if (!searchState.scannedPaths.has(fileInfo.path)) {
            searchState.scannedPaths.add(fileInfo.path);
            searchState.foundFiles.push(fileInfo);
            newFiles.push(fileInfo);
        }
    });
    
    // 如果是默认排序,直接追加(但要考虑显示限制)
    if (searchState.currentSort === 'default' && newFiles.length > 0) {
        const canAddMore = searchState.displayedCount < searchState.batchSize;
        
        if (canAddMore) {
            const toAdd = newFiles.slice(0, searchState.batchSize - searchState.displayedCount);
            const fragment = document.createDocumentFragment();
            
            toAdd.forEach(file => {
                fragment.appendChild(createResultItem(file));
            });
            
            elements.resultsList.appendChild(fragment);
            searchState.displayedCount += toAdd.length;
        }
    }
    
    // 更新"继续显示"按钮
    updateLoadMoreButton();
    
    updateStats();
    updateSelectAllButton();
}

function renderResults() {
    elements.resultsList.innerHTML = '';
    if (searchState.foundFiles.length === 0) {
        elements.resultsList.innerHTML = `
            <div class="empty-state">
                <svg viewBox="0 0 24 24"><circle cx="11" cy="11" r="8"/><path d="m21 21-4.35-4.35"/></svg>
                <p>${t('emptyState')}</p>
            </div>`;
        searchState.displayedCount = 0;
        return;
    }

    // 排序逻辑
    let files = [...searchState.foundFiles];
    const sort = searchState.currentSort;
    
    if (sort === 'nameAsc') {
        files.sort((a, b) => a.name.localeCompare(b.name));
    } else if (sort === 'nameDesc') {
        files.sort((a, b) => b.name.localeCompare(a.name));
    } else if (sort === 'dateDesc') {
        files.sort((a, b) => new Date(b.mtime) - new Date(a.mtime));
    } else if (sort === 'dateAsc') {
        files.sort((a, b) => new Date(a.mtime) - new Date(b.mtime));
    } else if (sort === 'sizeDesc') {
        files.sort((a, b) => b.size - a.size);
    } else if (sort === 'sizeAsc') {
        files.sort((a, b) => a.size - b.size);
    }

    // 渲染，考虑到性能，使用 DocumentFragment
    const fragment = document.createDocumentFragment();
    // 首次只显示1000条
    const limit = Math.min(files.length, searchState.batchSize);
    for (let i = 0; i < limit; i++) {
        fragment.appendChild(createResultItem(files[i]));
    }
    elements.resultsList.appendChild(fragment);
    
    // 更新显示计数
    searchState.displayedCount = limit;
    
    // 更新"继续显示"按钮
    updateLoadMoreButton();
    
    updateSelectAllButton();
}


// 创建结果项
function createResultItem(fileInfo) {
    const item = document.createElement('div');
    item.className = 'result-item';
    
    const isSelected = searchState.selectedFiles.has(fileInfo.path);

    const icon = getFileIcon(fileInfo.ext);
    
    // 获取当前搜索关键词用于高亮
    const searchQuery = elements.searchQuery.value.trim().toLowerCase();
    const highlightedName = highlightText(fileInfo.name, searchQuery);
    const highlightedPath = highlightText(fileInfo.dir, searchQuery);
    
    item.innerHTML = `
        <div class="file-checkbox">
            <input type="checkbox" ${isSelected ? 'checked' : ''} data-path="${escapeHtml(fileInfo.path)}">
        </div>
        <div class="file-icon">
            ${icon}
        </div>
        <div class="file-info">
            <div class="file-name">${highlightedName}</div>
            <div class="file-path">${highlightedPath}</div>
            <div class="file-meta">
                ${t('type')}: ${fileInfo.ext || t('noExt')} | 
                ${t('size')}: ${formatFileSize(fileInfo.size)} | 
                ${t('modTime')}: ${formatDate(fileInfo.mtime)}
            </div>
        </div>
        <div class="file-actions">
            <button class="action-btn" title="${t('openFolderTip')}" data-action="openFolder" data-path="${escapeHtml(fileInfo.dir)}">
                <svg viewBox="0 0 24 24">
                    <path d="M3 7v10a2 2 0 002 2h14a2 2 0 002-2V9a2 2 0 00-2-2h-6l-2-2H5a2 2 0 00-2 2z"/>
                </svg>
            </button>
            <button class="action-btn" title="${t('openFileTip')}" data-action="openFile" data-path="${escapeHtml(fileInfo.path)}">
                <svg viewBox="0 0 24 24">
                    <path d="M14 2H6a2 2 0 00-2 2v16a2 2 0 002 2h12a2 2 0 002-2V8z"/>
                    <polyline points="14 2 14 8 20 8"/>
                </svg>
            </button>
            <button class="action-btn" title="${t('propsTip')}" data-action="showProperties" data-info='${escapeHtml(JSON.stringify(fileInfo))}'>
                <svg viewBox="0 0 24 24">
                    <circle cx="12" cy="12" r="10"/>
                    <line x1="12" y1="16" x2="12" y2="12"/>
                    <line x1="12" y1="8" x2="12.01" y2="8"/>
                </svg>
            </button>
        </div>
    `;

    // 复选框事件
    const checkbox = item.querySelector('input[type="checkbox"]');
    checkbox.addEventListener('change', (e) => {
        const path = e.target.dataset.path;
        if (e.target.checked) {
            searchState.selectedFiles.add(path);
        } else {
            searchState.selectedFiles.delete(path);
        }
        updateSelectAllButton();
    });

    // 添加事件监听
    item.querySelectorAll('.action-btn').forEach(btn => {
        btn.addEventListener('click', handleAction);
    });

    return item;
}

// 高亮文本中的关键词
function highlightText(text, query) {
    if (!query) {
        return escapeHtml(text);
    }

    const escapedText = escapeHtml(text);
    const lowerText = text.toLowerCase();
    const lowerQuery = query.toLowerCase();
    
    let result = '';
    let lastIndex = 0;
    let index = lowerText.indexOf(lowerQuery);
    
    while (index !== -1) {
        result += escapeHtml(text.substring(lastIndex, index));
        result += `<span class="highlight">${escapeHtml(text.substring(index, index + query.length))}</span>`;
        lastIndex = index + query.length;
        index = lowerText.indexOf(lowerQuery, lastIndex);
    }
    
    result += escapeHtml(text.substring(lastIndex));
    return result;
}

// 处理操作按钮
async function handleAction(e) {
    const action = e.currentTarget.dataset.action;
    const path = e.currentTarget.dataset.path;
    const info = e.currentTarget.dataset.info;

    switch (action) {
        case 'openFolder':
            await window.electronAPI.openFolder(path);
            break;
        case 'openFile':
            await window.electronAPI.openFile(path);
            break;
        case 'showProperties':
            showProperties(JSON.parse(info));
            break;
    }
}

// 显示文件属性
async function showProperties(fileInfo) {
    const props = `
${t('appTitle')}: ${fileInfo.name}
路径: ${fileInfo.path}
${t('size')}: ${formatFileSize(fileInfo.size)}
${t('type')}: ${fileInfo.ext || t('noExt')}
${t('createTime')}: ${formatDate(fileInfo.birthtime)}
${t('modTime')}: ${formatDate(fileInfo.mtime)}
${t('accessTime')}: ${formatDate(fileInfo.atime)}
    `.trim();

    await customAlert(props, t('propsTip'));
}

// 获取文件图标
function getFileIcon(ext) {
    if (!ext) {
        return '<svg viewBox="0 0 24 24"><path d="M14 2H6a2 2 0 00-2 2v16a2 2 0 002 2h12a2 2 0 002-2V8z"/><polyline points="14 2 14 8 20 8"/></svg>';
    }

    ext = ext.toLowerCase();

    if (FILE_TYPES.images.includes(ext)) {
        return '<svg viewBox="0 0 24 24"><rect x="3" y="3" width="18" height="18" rx="2" ry="2"/><circle cx="8.5" cy="8.5" r="1.5"/><polyline points="21 15 16 10 5 21"/></svg>';
    }
    if (FILE_TYPES.videos.includes(ext)) {
        return '<svg viewBox="0 0 24 24"><polygon points="23 7 16 12 23 17 23 7"/><rect x="1" y="5" width="15" height="14" rx="2" ry="2"/></svg>';
    }
    if (FILE_TYPES.audio.includes(ext)) {
        return '<svg viewBox="0 0 24 24"><path d="M9 18V5l12-2v13"/><circle cx="6" cy="18" r="3"/><circle cx="18" cy="16" r="3"/></svg>';
    }
    if (FILE_TYPES.documents.includes(ext)) {
        return '<svg viewBox="0 0 24 24"><path d="M14 2H6a2 2 0 00-2 2v16a2 2 0 002 2h12a2 2 0 002-2V8z"/><polyline points="14 2 14 8 20 8"/><line x1="16" y1="13" x2="8" y2="13"/><line x1="16" y1="17" x2="8" y2="17"/><polyline points="10 9 9 9 8 9"/></svg>';
    }
    if (FILE_TYPES.archives.includes(ext)) {
        return '<svg viewBox="0 0 24 24"><polyline points="21 8 21 21 3 21 3 8"/><rect x="1" y="3" width="22" height="5"/><line x1="10" y1="12" x2="14" y2="12"/></svg>';
    }

    return '<svg viewBox="0 0 24 24"><path d="M14 2H6a2 2 0 00-2 2v16a2 2 0 002 2h12a2 2 0 002-2V8z"/><polyline points="14 2 14 8 20 8"/></svg>';
}

// 工具函数
function formatFileSize(bytes) {
    if (bytes === 0) return '0 B';
    const k = 1024;
    const sizes = ['B', 'KB', 'MB', 'GB', 'TB'];
    const i = Math.floor(Math.log(bytes) / Math.log(k));
    return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
}

function formatDate(dateString) {
    const date = new Date(dateString);
    return date.toLocaleString(currentLang === 'zh-CN' ? 'zh-CN' : 'en-US');
}

function escapeHtml(text) {
    const div = document.createElement('div');
    div.textContent = text;
    return div.innerHTML;
}

// 监听来自主进程的消息
window.electronAPI.onFileFound((fileInfo) => {
    addResult(fileInfo);
});

window.electronAPI.onFilesBatch((files) => {
    addResultsBatch(files);
});

window.electronAPI.onSearchProgress((data) => {
    searchState.scannedCount = data.scannedCount;
    updateStats();
    updateStatus(t('scanning') + data.currentPath, true);
});

window.electronAPI.onSearchComplete((data) => {
    searchState.isSearching = false;
    elements.searchBtn.style.display = 'flex';
    elements.stopBtn.style.display = 'none';
    elements.progressBar.style.display = 'none';
    
    // 清除节流定时器并立即渲染剩余结果
    if (addResultThrottle) {
        clearTimeout(addResultThrottle);
        addResultThrottle = null;
    }
    
    // 渲染所有待处理的结果
    if (pendingResults.length > 0) {
        if (searchState.currentSort === 'default') {
            const canAddMore = searchState.displayedCount < searchState.batchSize;
            
            if (canAddMore) {
                const toAdd = pendingResults.slice(0, searchState.batchSize - searchState.displayedCount);
                const fragment = document.createDocumentFragment();
                
                toAdd.forEach(file => {
                    fragment.appendChild(createResultItem(file));
                });
                
                elements.resultsList.appendChild(fragment);
                searchState.displayedCount += toAdd.length;
            }
        }
        pendingResults = [];
    }
    
    // 更新"继续显示"按钮
    updateLoadMoreButton();
    
    updateStatus(t('searchDone', searchState.foundFiles.length), false);
    updateStats();
    updateSelectAllButton();
    
    // 如果不是默认排序，搜索完成后重新排序并渲染
    if (searchState.currentSort !== 'default') {
        renderResults();
    }
});

window.electronAPI.onSearchError(async (error) => {
    searchState.isSearching = false;
    elements.searchBtn.style.display = 'flex';
    elements.stopBtn.style.display = 'none';
    elements.progressBar.style.display = 'none';
    
    updateStatus(t('searchError') + error, false);
    await customAlert(t('searchError') + error, t('errorTitle'));
});

// 导出进度监听
window.electronAPI.onExportProgress((data) => {
    updateStatus(`${t('exporting')} ${data.current} / ${data.total}`, true);
});

// 删除进度监听
window.electronAPI.onDeleteProgress((data) => {
    updateStatus(`正在删除... ${data.current} / ${data.total}`, true);
});

// 初始化语言
updateI18n();