fix: model download - windows path issue

This commit is contained in:
Louis 2025-07-10 09:42:36 +07:00
parent 2f02a228cc
commit a8ed759a06
2 changed files with 27 additions and 24 deletions

View File

@ -4,8 +4,6 @@ use crate::core::utils::normalize_path;
use futures_util::StreamExt;
use reqwest::header::{HeaderMap, HeaderName, HeaderValue};
use std::collections::HashMap;
use std::path::Component;
use std::path::Prefix;
use std::time::Duration;
use tauri::{Emitter, State};
use tokio::fs::File;
@ -166,25 +164,6 @@ async fn _download_files_internal(
let save_path = jan_data_folder.join(&item.save_path);
let save_path = normalize_path(&save_path);
// enforce scope
// Remove \\?\ prefix on Windows for correct path comparison
#[cfg(windows)]
let save_path = {
let mut comps = save_path.components();
if let Some(Component::Prefix(prefix_comp)) = comps.next() {
if let Prefix::Verbatim(_) = prefix_comp.kind() {
// Skip the \\?\ prefix
comps.as_path().to_path_buf()
} else {
save_path.clone()
}
} else {
save_path.clone()
}
};
#[cfg(not(windows))]
let save_path = save_path.clone();
if !save_path.starts_with(&jan_data_folder) {
return Err(format!(
"Path {} is outside of Jan data folder {}",

View File

@ -6,6 +6,7 @@ use std::path::{Component, Path, PathBuf};
use tauri::Runtime;
use super::cmd::get_jan_data_folder_path;
use std::path::Prefix;
pub const THREADS_DIR: &str = "threads";
pub const THREADS_FILE: &str = "thread.json";
@ -53,9 +54,32 @@ pub fn ensure_thread_dir_exists<R: Runtime>(
// https://github.com/rust-lang/cargo/blob/rust-1.67.0/crates/cargo-util/src/paths.rs#L82-L107
pub fn normalize_path(path: &Path) -> PathBuf {
let mut components = path.components().peekable();
let mut ret = if let Some(c @ Component::Prefix(..)) = components.peek().cloned() {
components.next();
PathBuf::from(c.as_os_str())
let mut ret = if let Some(c @ Component::Prefix(prefix_component)) = components.peek().cloned()
{
#[cfg(windows)]
// Remove only the Verbatim prefix, but keep the drive letter (e.g., C:\)
match prefix_component.kind() {
Prefix::VerbatimDisk(disk) => {
components.next(); // skip this prefix
// Re-add the disk prefix (e.g., C:)
let mut pb = PathBuf::new();
pb.push(format!("{}:", disk as char));
pb
}
Prefix::Verbatim(_) | Prefix::VerbatimUNC(_, _) => {
components.next(); // skip this prefix
PathBuf::new()
}
_ => {
components.next();
PathBuf::from(c.as_os_str())
}
}
#[cfg(not(windows))]
{
components.next(); // skip this prefix
PathBuf::from(c.as_os_str())
}
} else {
PathBuf::new()
};