shkong/clash-verge-rev
1
0
Fork 0
mirror of https://github.com/clash-verge-rev/clash-verge-rev synced 2025-05-05 04:53:44 +08:00
Code Packages Projects Releases Activity

feat: add icon file content check to prevent saving failed downloads

Browse Source
This commit is contained in:
wonfen 2025-03-09 02:34:57 +08:00
parent 18f86874ee
commit c1a9de4d66
1 changed files with 47 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src-tauri/src/cmd/app.rs
View File

@ -90,10 +90,27 @@ pub async fn download_icon_cache(url: String, name: String) -> CmdResult<String>
let temp_path = icon_cache_dir.join(format!("{}.downloading", &name));
// 下载文件到临时位置
let response = wrap_err!(reqwest::get(url).await)?;
let response = wrap_err!(reqwest::get(&url).await)?;
// 检查内容类型是否为图片
let content_type = response.headers()
.get(reqwest::header::CONTENT_TYPE)
.and_then(|v| v.to_str().ok())
.unwrap_or("");
let is_image = content_type.starts_with("image/");
// 获取响应内容
let content = wrap_err!(response.bytes().await)?;
// 写入临时文件
// 检查内容是否为HTML (针对CDN错误页面)
let is_html = content.len() > 15 &&
(content.starts_with(b"<!DOCTYPE html") ||
content.starts_with(b"<html") ||
content.starts_with(b"<?xml"));
// 只有当内容确实是图片时才保存
if is_image && !is_html {
{
let mut file = match std::fs::File::create(&temp_path) {
Ok(file) => file,
@ -111,7 +128,6 @@ pub async fn download_icon_cache(url: String, name: String) -> CmdResult<String>
// 再次检查目标文件是否已存在，避免重命名覆盖其他线程已完成的文件
if !icon_path.exists() {
// 使用原子重命名操作将临时文件移动到最终位置
match std::fs::rename(&temp_path, &icon_path) {
Ok(_) => {},
Err(_) => {
@ -126,6 +142,10 @@ pub async fn download_icon_cache(url: String, name: String) -> CmdResult<String>
}
Ok(icon_path.to_string_lossy().to_string())
} else {
let _ = std::fs::remove_file(&temp_path);
Err(format!("下载的内容不是有效图片: {}", url).into())
}
}
#[derive(Debug, serde::Serialize, serde::Deserialize)]