feat(download): 增强仓库下载功能

- 修改下载命令，支持指定分支
- 更新 README.md 中的使用说明
- 调整配置文件中的最小文件大小限制
- 优化下载函数，增加 zip 文件完整性校验

Author: NaivG

README.md

@@ -30,7 +30,7 @@
 
 - **镜像加速**  
   自动测试多个Git镜像源延迟，选择最快的源进行克隆/拉取(支持`clone`/`pull`/`push`/`fetch`)。
-- **下载文件**
+- **下载文件**  
   支持直接下载仓库压缩包(TODO: 支持下载release文件)。
 - **代理支持**  
   可通过命令行参数或配置文件设置HTTP/HTTPS代理。
@@ -85,8 +85,8 @@ fgit clone <user>/<repo> # fgit的特色方式，会自动转换为https://githu
 fgit pull
 
 # 下载仓库压缩包（自动选择镜像源）
-fgit download-zip <仓库URL>
-fgit download-zip <user>/<repo>
+fgit download <仓库URL>
+fgit --branch <分支名> download <user>/<repo>
 
 # 启用代理
 fgit --use-proxy http://127.0.0.1:7890 clone <仓库URL>

config.py

@@ -42,7 +42,7 @@ def get_downloader_config(self):
             return dict(self.config.items('downloader'))
         self.config.add_section('downloader')
         self.config.set('downloader', 'chunk_size', '1024')
-        self.config.set('downloader', 'MIN_FILE_SIZE', '1')
+        self.config.set('downloader', 'MIN_FILE_SIZE', '100')
         self._save()
         return dict(self.config.items('downloader'))
 

downloader.py

@@ -1,8 +1,10 @@
+import os
+import zipfile
 from tqdm import tqdm
 import requests
 from loguru import logger
 
-def download_file(url: str, file_path: str, chunk_size: int = 1024, MIN_FILE_SIZE: int = 1) -> bool:
+def download_file(url: str, file_path: str, chunk_size: int = 1024, MIN_FILE_SIZE: int = 100) -> bool:
     try:
         response = requests.get(url, stream=True)
         total_size = int(response.headers.get('content-length', 0))
@@ -16,6 +18,15 @@ def download_file(url: str, file_path: str, chunk_size: int = 1024, MIN_FILE_SIZ
                 for data in response.iter_content(chunk_size=int(chunk_size)):
                     f.write(data)
                     pbar.update(len(data))
+        
+        with zipfile.ZipFile(file_path, 'r') as zip_ref:
+            try:
+                zip_ref.testzip()  # 校验文件完整性
+            except zipfile.BadZipFile:
+                logger.error(f"下载失败: {file_path} 不是一个有效的zip文件")
+                os.remove(file_path)
+                return False
+            
         return True
     except Exception as e:
         logger.error(f"下载失败: {str(e)}")

fgit.py

@@ -19,6 +19,8 @@
 parser = argparse.ArgumentParser(description='Git加速工具，支持镜像源和代理')
 parser.add_argument('command', type=str, help='git命令, 或是fgit命令')
 parser.add_argument('--use-proxy', type=str, help='设置HTTP代理（格式: http://[user:pass@]host:port）')
+parser.add_argument('--branch', type=str, help='分支名(仅在download命令时有效)', default='main')
+
 parser.add_argument('--verbose', action='store_true', help='显示详细输出')
 args, unknown_args = parser.parse_known_args()
 
@@ -45,7 +47,7 @@ def main():
     logger.debug(Fore.CYAN + f"命令参数: {' '.join(sys.argv)}" + Style.RESET_ALL)
 
 
-    if args.command == 'download-zip':
+    if args.command == 'download':
         handle_download_zip(args, unknown_args, config, env, args.verbose)
         return
 
@@ -66,7 +68,7 @@ def handle_download_zip(args, unknown_args, config, env, verbose):
     if not downloader_config:
         logger.warning(Fore.YELLOW + "🧐 下载配置不存在, 使用默认配置" + Style.RESET_ALL)
     chunk_size = downloader_config.get('chunk_size', 1024)
-    MIN_FILE_SIZE = downloader_config.get('min_file_size', 1)
+    MIN_FILE_SIZE = downloader_config.get('min_file_size', 100)
 
     original_url = unknown_args[0]
     if '://' not in original_url and '/' in original_url:
@@ -89,9 +91,9 @@ def handle_download_zip(args, unknown_args, config, env, verbose):
 
     mirror_list = select_mirror(config, args.verbose)
     for mirror in mirror_list:
-        new_url = convert_url(original_url, mirror) + '/archive/refs/heads/main.zip'
+        new_url = convert_url(original_url, mirror) + f'/archive/refs/heads/{args.branch}.zip'
         logger.info(Fore.GREEN + f"🔄 尝试镜像源 {mirror} [{mirror_list.index(mirror) + 1}/{len(mirror_list)}]: {new_url}" + Style.RESET_ALL)
-        if download_file(new_url, os.path.join(os.getcwd(), repo_name + '.zip'), chunk_size=chunk_size, MIN_FILE_SIZE=MIN_FILE_SIZE):
+        if download_file(new_url, os.path.join(os.getcwd(), f'{repo_name}-{args.branch}.zip'), chunk_size=chunk_size, MIN_FILE_SIZE=MIN_FILE_SIZE):
             return
     logger.error(Fore.RED + "❌ 所有镜像源尝试失败" + Style.RESET_ALL)
     return

requirements.txt

@@ -2,4 +2,5 @@ ping3
 colorama
 prettytable
 loguru
+requests
 tqdm