`;
document.body.appendChild(modal);
// 清理文件名，移除查询参数
function cleanFileName(fileName) {
return fileName.replace(/\?.*$/, '');
}
// 判断是否为主要文件
function isMainFile(fileName) {
const mainFilePatterns = [
/^README\.md$/i,
/^config\.json$/i,
/^model\.safetensors$/i,
/^pytorch_model\.bin$/i,
/^model\.onnx$/i,
/^tokenizer\.json$/i,
/^tokenizer_config\.json$/i,
/^vocab\.txt$/i,
/^merges\.txt$/i,
/\.py$/i,
/^requirements\.txt$/i,
/^setup\.py$/i,
/^__init__\.py$/i,
/^Dockerfile$/i,
/^\.dockerignore$/i
];
return mainFilePatterns.some(pattern => pattern.test(fileName));
}
// 格式化文件大小
function formatFileSize(bytes) {
if (!bytes || bytes === 0) return '未知';
const sizes = ['B', 'KB', 'MB', 'GB'];
const i = Math.floor(Math.log(bytes) / Math.log(1024));
return Math.round(bytes / Math.pow(1024, i) * 100) / 100 + ' ' + sizes[i];
}
// 从元素中提取文件大小 - 针对HuggingFace页面优化
function extractFileSize(element) {
let sizeText = '';
const fileName = element.getAttribute('download') || element.textContent.trim();
console.log(`🔍 开始提取文件大小: ${fileName}`);
// 方法1: 查找HuggingFace特定的文件列表结构
let currentElement = element;
for (let level = 0; level < 6; level++) {
if (!currentElement) break;
// 在当前层级查找所有文本元素
const allElements = currentElement.querySelectorAll('*');
for (const el of allElements) {
const text = el.textContent.trim();
// 精确匹配文件大小格式 (如: "459 Bytes", "1 kB", "2.68 kB")
const sizeMatch = text.match(/^(\d+(?:\.\d+)?)\s*(Bytes?|kB|KB|MB|GB)$/i);
if (sizeMatch) {
sizeText = sizeMatch[0];
console.log(`✅ 在层级 ${level} 找到精确文件大小: ${sizeText}`);
break;
}
// 也匹配包含文件大小的文本
const sizeInText = text.match(/(\d+(?:\.\d+)?)\s*(Bytes?|kB|KB|MB|GB)\b/i);
if (sizeInText && text.length < 50) { // 避免匹配到很长的文本
sizeText = sizeInText[0];
console.log(`✅ 在层级 ${level} 找到文件大小: ${sizeText}`);
break;
}
}
if (sizeText) break;
currentElement = currentElement.parentElement;
}
// 方法2: 查找同一行的其他元素
if (!sizeText) {
const parentRow = element.closest('li') || element.closest('tr') || element.closest('div');
if (parentRow) {
const rowText = parentRow.textContent;
const sizeMatch = rowText.match(/(\d+(?:\.\d+)?)\s*(Bytes?|kB|KB|MB|GB)\b/i);
if (sizeMatch) {
sizeText = sizeMatch[0];
console.log(`✅ 在同一行找到文件大小: ${sizeText}`);
}
}
}
// 方法3: 全局搜索与文件名相关的大小信息
if (!sizeText && fileName) {
console.log(`🔍 全局搜索文件: ${fileName}`);
// 查找页面中所有可能包含文件大小的元素
const allTextElements = document.querySelectorAll('span, div, td, li, p');
for (const el of allTextElements) {
const text = el.textContent.trim();
// 检查是否包含文件名和大小信息
if (text.includes(fileName) || el.closest('*').textContent.includes(fileName)) {
const sizeMatch = text.match(/(\d+(?:\.\d+)?)\s*(Bytes?|kB|KB|MB|GB)\b/i);
if (sizeMatch) {
sizeText = sizeMatch[0];
console.log(`✅ 全局搜索找到文件大小: ${sizeText}`);
break;
}
}
}
}
// 方法4: 查找相邻元素
if (!sizeText && element.parentElement) {
const siblings = Array.from(element.parentElement.children);
for (const sibling of siblings) {
const text = sibling.textContent.trim();
const sizeMatch = text.match(/^(\d+(?:\.\d+)?)\s*(Bytes?|kB|KB|MB|GB)$/i);
if (sizeMatch) {
sizeText = sizeMatch[0];
console.log(`✅ 在兄弟元素找到文件大小: ${sizeText}`);
break;
}
}
}
const result = sizeText || '未知';
console.log(`📊 文件 ${fileName} 最终大小: ${result}`);
return result;
}
// 解析文件大小为字节数（用于排序）
function parseSizeToBytes(sizeStr) {
if (!sizeStr || sizeStr === '未知') return 0;
// 支持更多格式的匹配
const match = sizeStr.match(/(\d+(?:\.\d+)?)\s*(B|KB|MB|GB|kB|bytes|Bytes)\b/i);
if (!match) return 0;
const value = parseFloat(match[1]);
const unit = match[2].toUpperCase();
const multipliers = {
'B': 1,
'BYTES': 1,
'KB': 1024,
'kB': 1024, // 小写k
'MB': 1024 * 1024,
'GB': 1024 * 1024 * 1024
};
return value * (multipliers[unit] || 1);
}
// 全局变量存储当前排序方式和搜索状态
let currentSortType = 'default';
let allLinks = [];
let filteredLinks = [];
let currentSearchTerm = '';
// 提取链接函数
function extractLinks() {
const links = [];
const elements = document.querySelectorAll('a[download][href]');
console.log(`🔍 找到 ${elements.length} 个下载链接`);
elements.forEach((element, index) => {
const href = element.getAttribute('href');
if (href) {
const originalLink = href.startsWith('http') ? href : 'https://huggingface.co' + href;
const mirrorLink = originalLink.replace('huggingface.co', 'hf-mirror.com');
const rawFileName = href.split('/').pop() || 'unknown';
const fileName = cleanFileName(rawFileName);
const isMain = isMainFile(fileName);
const fileSize = extractFileSize(element);
const fileSizeBytes = parseSizeToBytes(fileSize);
console.log(`📁 文件 ${index + 1}: ${fileName}, 大小: ${fileSize}, 字节: ${fileSizeBytes}`);
links.push({
original: originalLink,
mirror: mirrorLink,
fileName: fileName,
isMainFile: isMain,
fileSize: fileSize,
fileSizeBytes: fileSizeBytes
});
}
});
allLinks = links;
filteredLinks = links;
return sortLinks(links, 'default');
}
// 排序函数
function sortLinks(links, sortType) {
const sorted = [...links];
switch (sortType) {
case 'name':
return sorted.sort((a, b) => a.fileName.localeCompare(b.fileName));
case 'size':
return sorted.sort((a, b) => b.fileSizeBytes - a.fileSizeBytes);
case 'type':
return sorted.sort((a, b) => {
const extA = a.fileName.split('.').pop().toLowerCase();
const extB = b.fileName.split('.').pop().toLowerCase();
return extA.localeCompare(extB);
});
case 'default':
default:
return sorted.sort((a, b) => {
if (a.isMainFile && !b.isMainFile) return -1;
if (!a.isMainFile && b.isMainFile) return 1;
return a.fileName.localeCompare(b.fileName);
});
}
}
// 模糊搜索功能
function fuzzySearch(links, searchTerm) {
if (!searchTerm.trim()) {
return links;
}
const term = searchTerm.toLowerCase().trim();
return links.filter(link => {
const fileName = link.fileName.toLowerCase();
// 精确匹配
if (fileName.includes(term)) {
return true;
}
// 模糊匹配：检查搜索词的每个字符是否按顺序出现在文件名中
let termIndex = 0;
for (let i = 0; i < fileName.length && termIndex < term.length; i++) {
if (fileName[i] === term[termIndex]) {
termIndex++;
}
}
return termIndex === term.length;
});
}
// 高亮搜索结果 - 使用更温和的高亮方式
function highlightSearchTerm(text, searchTerm) {
if (!searchTerm.trim()) {
return text;
}
const term = searchTerm.trim().toLowerCase();
const lowerText = text.toLowerCase();
// 只进行精确匹配高亮，避免模糊匹配造成的间距问题
const exactIndex = lowerText.indexOf(term);
if (exactIndex !== -1) {
const before = text.substring(0, exactIndex);
const match = text.substring(exactIndex, exactIndex + term.length);
const after = text.substring(exactIndex + term.length);
return before + '