1.3.21.3.2

2026-05-26 16:32:18 +08:00
parent 546cef5ba6
commit 17673b2862
89 changed files with 24098 additions and 272 deletions
@@ -0,0 +1,156 @@
+#[cfg(unix)]
+use std::collections::HashSet;
+use crate::errors::Result;
+use crate::models::LocalFileEntry;
+use crate::utils::quick_hash;
+use std::path::Path;
+use walkdir::WalkDir;
+
+/// 需要跳过的文件/目录名前缀和名称
+pub const SKIP_NAMES: &[&str] = &[
+    ".DS_Store",
+    "Thumbs.db",
+    "desktop.ini",
+];
+
+/// 需要跳过的文件扩展名（同步临时文件）
+pub const SKIP_EXTENSIONS: &[&str] = &[
+    "sync_tmp",
+    "sync_temp",
+];
+
+pub struct FsScanner;
+
+impl Default for FsScanner {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl FsScanner {
+    pub fn new() -> Self {
+        Self
+    }
+
+    /// 递归扫描本地目录
+    /// `compute_hash`: 是否计算文件 quick_hash（MirrorWcf 模式下可跳过以加速扫描）
+    pub async fn scan(
+        &self,
+        root: &Path,
+        depth_limit: u32,
+        follow_symlinks: bool,
+        compute_hash: bool,
+    ) -> Result<Vec<LocalFileEntry>> {
+        let mut entries = Vec::new();
+        #[cfg(unix)]
+        let mut visited_inodes: HashSet<(u64, u64)> = HashSet::new();
+
+        let walker = WalkDir::new(root)
+            .max_depth(depth_limit as usize)
+            .follow_links(follow_symlinks);
+
+        for entry in walker {
+            let entry = match entry {
+                Ok(e) => e,
+                Err(e) => {
+                    tracing::warn!("扫描跳过: {}", e);
+                    continue;
+                }
+            };
+
+            // 符号链接处理
+            if entry.path_is_symlink() && !follow_symlinks {
+                continue;
+            }
+
+            // 跳过同步元数据文件和系统文件
+            let file_name = entry.file_name().to_string_lossy();
+            if SKIP_NAMES.iter().any(|s| file_name == *s) {
+                continue;
+            }
+            if file_name.starts_with(".sync_") {
+                continue;
+            }
+            // 跳过临时文件扩展名
+            if let Some(ext) = entry.path().extension() {
+                if SKIP_EXTENSIONS.iter().any(|e| ext == *e) {
+                    continue;
+                }
+            }
+            // 跳过冲突副本文件
+            if crate::utils::is_conflict_file(&file_name) {
+                continue;
+            }
+
+            let metadata = match entry.metadata() {
+                Ok(m) => m,
+                Err(e) => {
+                    tracing::warn!("无法读取元数据 {}: {}", entry.path().display(), e);
+                    continue;
+                }
+            };
+
+            // inode 去重（防止硬链接/符号链接循环）
+            #[cfg(unix)]
+            {
+                use std::os::unix::fs::MetadataExt;
+                let key = (metadata.dev(), metadata.ino());
+                if !visited_inodes.insert(key) {
+                    continue;
+                }
+            }
+
+            let relative_path = entry.path().strip_prefix(root)
+                .unwrap_or(entry.path())
+                .to_path_buf();
+
+            // 跳过根目录自身（relative_path 为空）
+            if relative_path.to_string_lossy().is_empty() {
+                continue;
+            }
+
+            if metadata.is_dir() {
+                entries.push(LocalFileEntry {
+                    relative_path,
+                    size: 0,
+                    mtime_ms: 0,
+                    quick_hash: String::new(),
+                    is_dir: true,
+                    mime_type: None,
+                });
+            } else if metadata.is_file() {
+                let size = metadata.len();
+                let mtime_ms = metadata.modified()
+                    .ok()
+                    .and_then(|t| t.duration_since(std::time::UNIX_EPOCH).ok())
+                    .map(|d| d.as_millis() as i64)
+                    .unwrap_or(0);
+
+                let hash = if compute_hash {
+                    quick_hash(entry.path(), size).await.unwrap_or_default()
+                } else {
+                    String::new()
+                };
+                let mime_type = guess_mime_type(entry.path());
+
+                entries.push(LocalFileEntry {
+                    relative_path,
+                    size,
+                    mtime_ms,
+                    quick_hash: hash,
+                    is_dir: false,
+                    mime_type,
+                });
+            }
+        }
+
+        Ok(entries)
+    }
+}
+
+/// 根据文件扩展名推断 MIME 类型
+pub fn guess_mime_type(path: &Path) -> Option<String> {
+    mime_guess::from_path(path)
+        .first()
+        .map(|m| m.to_string())
+}