2
0

FileUtils.cs 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589
  1. using System.IO.Compression;
  2. namespace MeloongCore;
  3. public static class FileUtils {
  4. #region 读取
  5. /// <summary>
  6. /// 打开指定文件的只读 <see cref="Stream"/>。
  7. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  8. /// </summary>
  9. public static Stream ReadAsStream(string filePath, Type? type = null) {
  10. if (type is null) {
  11. return Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, func: _
  12. => new FileStream(PathUtils.ForApi(filePath), FileMode.Open, FileAccess.Read, FileShare.ReadWrite));
  13. } else {
  14. return type.Assembly.GetManifestResourceStream(filePath) ?? throw new FileNotFoundException($"Embedded resource \"{filePath}\" was not found.");
  15. }
  16. }
  17. /// <summary>
  18. /// 读取文件中的所有内容。
  19. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  20. /// </summary>
  21. public static byte[] ReadAsBytes(string filePath, Type? type = null) {
  22. using Stream fs = FileUtils.ReadAsStream(filePath, type); // 不能使用 File.ReadAllBytes,它不指定 FileShare.ReadWrite,会在文件被占用时抛出异常
  23. using MemoryStream ms = new();
  24. fs.CopyTo(ms);
  25. return ms.ToArray();
  26. }
  27. /// <summary>
  28. /// 读取文件中的所有内容。
  29. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  30. /// </summary>
  31. public static string ReadAsString(string filePath, Encoding? encoding = null, Type? type = null)
  32. => FileUtils.ReadAsBytes(filePath, type).GetString(encoding);
  33. /// <summary>
  34. /// 读取文件中的所有内容。
  35. /// <para/>若文件不存在或读取失败,返回 <see langword="null"/>,而不是抛出异常。
  36. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  37. /// </summary>
  38. public static string? TryReadAsString(string filePath, Encoding? encoding = null, Type? type = null) {
  39. try {
  40. if (type == null && !FileUtils.Exists(filePath)) return null;
  41. return FileUtils.ReadAsBytes(filePath, type).GetString(encoding);
  42. } catch {
  43. return null;
  44. }
  45. }
  46. /// <summary>
  47. /// 读取文件中的所有内容,并按行分割。
  48. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  49. /// </summary>
  50. public static string[] ReadAsLines(string filePath, bool skipEmptyLines = false, Encoding? encoding = null, Type? type = null)
  51. => FileUtils.ReadAsString(filePath, encoding, type).SplitLines(skipEmptyLines);
  52. /// <summary>
  53. /// 读取文件中的所有内容,并按行分割。
  54. /// <para/>若文件不存在或读取失败,返回空数组,而不是抛出异常。
  55. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  56. /// </summary>
  57. public static string[] TryReadAsLines(string filePath, bool skipEmptyLines = false, Encoding? encoding = null, Type? type = null)
  58. => FileUtils.TryReadAsString(filePath, encoding, type)?.SplitLines(skipEmptyLines) ?? [];
  59. /// <summary>
  60. /// 读取 JSON 文件中的所有内容。
  61. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  62. /// </summary>
  63. public static JToken? ReadAsJson(string filePath, Encoding? encoding = null, Type? type = null)
  64. => FileUtils.ReadAsString(filePath, encoding, type).DeserializeJson();
  65. /// <summary>
  66. /// 读取 JSON 文件中的所有内容。
  67. /// <para/>若指定了 <paramref name="type"/>,则会改为从该类型的程序集中读取嵌入的资源。
  68. /// </summary>
  69. public static T? ReadAsJson<T>(string filePath, Encoding? encoding = null, Type? type = null) where T : JToken
  70. => FileUtils.ReadAsString(filePath, encoding, type).DeserializeJson<T>();
  71. #endregion
  72. #region 创建 / 写入
  73. /// <summary>
  74. /// 创建文件,并将 <paramref name="text" /> 写入文件。
  75. /// 若文件已存在,则会覆盖原文件。
  76. /// </summary>
  77. public static void Write(string filePath, string text, Encoding? encoding = null)
  78. => FileUtils.Write(filePath, (encoding ?? new UTF8Encoding()).GetBytes(text));
  79. /// <summary>
  80. /// 创建文件,并将 <paramref name="content" /> 写入文件。
  81. /// 若文件已存在,则会覆盖原文件。
  82. /// </summary>
  83. public static void Write(string filePath, IEnumerable<byte> content)
  84. => FileUtils.Write(filePath, [.. content]);
  85. /// <summary>
  86. /// 创建文件,并将 <paramref name="content" /> 写入文件。
  87. /// 若文件已存在,则会覆盖原文件。
  88. /// </summary>
  89. public static void Write(string filePath, byte[] content) {
  90. DirectoryUtils.Create(PathUtils.RemoveLastPart(filePath));
  91. Logger.Trace($"写入文件:{filePath}({content.Length} 字节)");
  92. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => {
  93. File.WriteAllBytes(PathUtils.ForApi(filePath), content);
  94. });
  95. }
  96. /// <summary>
  97. /// 原子写入文件。成功时目标文件完整切换为新内容,失败时已有目标文件保持不变。
  98. /// 不提供并发写入合并保证。
  99. /// </summary>
  100. public static void WriteAtomic(string filePath, string text, Encoding? encoding = null) =>
  101. FileUtils.WriteAtomic(filePath, (encoding ?? new UTF8Encoding()).GetBytes(text));
  102. /// <summary>
  103. /// 原子写入文件。成功时目标文件完整切换为新内容,失败时已有目标文件保持不变。
  104. /// 不提供并发写入合并保证。
  105. /// </summary>
  106. public static void WriteAtomic(string filePath, byte[] content) {
  107. string tempPath = $"{filePath}.{Guid.NewGuid():N}.tmp";
  108. try {
  109. FileUtils.Write(tempPath, content);
  110. if (FileUtils.Exists(filePath)) {
  111. File.Replace(PathUtils.ForApi(tempPath), PathUtils.ForApi(filePath), null, true);
  112. } else {
  113. File.Move(PathUtils.ForApi(tempPath), PathUtils.ForApi(filePath));
  114. }
  115. } finally {
  116. FileUtils.Delete(tempPath);
  117. }
  118. }
  119. /// <summary>
  120. /// 创建文件,并将 <paramref name="stream" /> 写入文件。
  121. /// 会将流的位置主动重置到开头。
  122. /// 若文件已存在,则会覆盖原文件。
  123. /// </summary>
  124. public static void Write(string filePath, Stream stream) {
  125. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => {
  126. FileUtils.Delete(filePath);
  127. using FileStream fileStream = FileUtils.CreateAsStream(filePath);
  128. if (stream.CanSeek && stream.Position != 0) stream.Seek(0, SeekOrigin.Begin);
  129. Logger.Trace($"写入文件:{filePath}({stream.GetType().Name}{(stream.CanSeek ? $" {stream.Length} 字节" : "")})");
  130. stream.CopyTo(fileStream);
  131. });
  132. }
  133. /// <summary>
  134. /// 创建文件,并打开 <see cref="FileMode.Create"/> 模式的 <see cref="FileStream"/>。
  135. /// 若文件已存在,则会覆盖原文件。
  136. /// </summary>
  137. public static FileStream CreateAsStream(string filePath) {
  138. DirectoryUtils.Create(PathUtils.RemoveLastPart(filePath));
  139. Logger.Trace($"创建文件流:{filePath}");
  140. return Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, func: _ => {
  141. FileUtils.Delete(filePath);
  142. return new FileStream(PathUtils.ForApi(filePath), FileMode.Create);
  143. });
  144. }
  145. /// <summary>
  146. /// 在临时文件夹下创建一个随机名称的文件,并返回其路径。
  147. /// </summary>
  148. public static string CreateRandom()
  149. => Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, func: _
  150. => Path.GetTempFileName());
  151. /// <summary>
  152. /// 从 <paramref name="type"/> 的程序集中,读取指定 <paramref name="logicalName"/> 的嵌入的资源,写入到 <paramref name="localFilePath"/> 中。
  153. /// <para/> 如果文件已存在且内容相同,则不会写入。
  154. /// </summary>
  155. public static void ExtractResources(string logicalName, string localFilePath, Type type) {
  156. var content = ReadAsBytes(logicalName, type);
  157. if (Exists(localFilePath) && CryptographyUtils.ComputeFileHash(localFilePath) == CryptographyUtils.ComputeHash(content)) return;
  158. Logger.Info($"将资源写入到文件:{logicalName} → {localFilePath}");
  159. Write(localFilePath, content);
  160. }
  161. #endregion
  162. #region 复制 / 剪切
  163. /// <summary>
  164. /// 复制文件。
  165. /// 会创建对应文件夹、覆盖已有的文件。
  166. /// 若复制自身到自身,则不执行操作;若仅大小写不同,则重命名此文件。
  167. /// </summary>
  168. public static void Copy(string sourceFilePath, string destFilePath) {
  169. sourceFilePath = PathUtils.ForCompare(sourceFilePath);
  170. destFilePath = PathUtils.ForCompare(destFilePath);
  171. if (string.CompareOrdinal(sourceFilePath, destFilePath) == 0) {
  172. // 复制自身到自身,则不执行操作
  173. Logger.Trace($"复制文件到自身,不执行操作:{sourceFilePath} → {destFilePath}");
  174. } else if (PathUtils.IsSame(sourceFilePath, destFilePath)) {
  175. // 路径仅大小写不同,等效于重命名
  176. Logger.Trace($"复制文件到自身,但大小写不同,等效于重命名文件:{sourceFilePath} → {destFilePath}");
  177. FileUtils.Move(sourceFilePath, destFilePath);
  178. } else {
  179. // 实际的复制
  180. DirectoryUtils.Create(PathUtils.RemoveLastPart(destFilePath));
  181. Logger.Trace($"复制文件:{sourceFilePath} → {destFilePath}");
  182. if (FileUtils.Exists(destFilePath)) FileUtils.SetReadOnly(destFilePath, false);
  183. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _
  184. => File.Copy(PathUtils.ForApi(sourceFilePath), PathUtils.ForApi(destFilePath), true));
  185. }
  186. }
  187. /// <summary>
  188. /// 剪切文件。
  189. /// 会创建对应文件夹、覆盖已有的文件。
  190. /// 若指定了 <paramref name="toRecycleBin"/>,则在覆盖前将老文件删除到回收站。
  191. /// </summary>
  192. public static void Move(string sourceFilePath, string destFilePath, bool toRecycleBin = false) {
  193. sourceFilePath = PathUtils.ForCompare(sourceFilePath);
  194. destFilePath = PathUtils.ForCompare(destFilePath);
  195. if (string.CompareOrdinal(sourceFilePath, destFilePath) == 0) {
  196. // 剪切自身到自身,则不执行操作
  197. Logger.Trace($"剪切文件到自身,不执行操作:{sourceFilePath} → {destFilePath}");
  198. } else if (PathUtils.IsSame(sourceFilePath, destFilePath)) {
  199. // 路径仅大小写不同
  200. Logger.Trace($"剪切文件到自身,但大小写不同:{sourceFilePath} → {destFilePath}");
  201. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => {
  202. var temp = Path.Combine(PathUtils.RemoveLastPart(sourceFilePath), Path.GetRandomFileName());
  203. FileUtils.SetReadOnly(sourceFilePath, false);
  204. if (FileUtils.Exists(destFilePath)) FileUtils.SetReadOnly(destFilePath, false);
  205. File.Move(PathUtils.ForApi(sourceFilePath), PathUtils.ForApi(temp));
  206. File.Move(PathUtils.ForApi(temp), PathUtils.ForApi(destFilePath));
  207. });
  208. } else {
  209. // 实际的剪切
  210. DirectoryUtils.Create(PathUtils.RemoveLastPart(destFilePath));
  211. FileUtils.Delete(destFilePath, toRecycleBin);
  212. Logger.Trace($"剪切文件:{sourceFilePath} → {destFilePath}");
  213. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => {
  214. FileUtils.SetReadOnly(sourceFilePath, false);
  215. if (FileUtils.Exists(destFilePath)) FileUtils.SetReadOnly(destFilePath, false);
  216. File.Move(PathUtils.ForApi(sourceFilePath), PathUtils.ForApi(destFilePath));
  217. });
  218. }
  219. }
  220. #endregion
  221. #region 删除
  222. /// <summary>
  223. /// 删除文件。
  224. /// <para/>若指定了 <paramref name="toRecycleBin"/>,则会尝试删除到回收站,但如果失败则会回退到永久删除。
  225. /// </summary>
  226. /// <returns>
  227. /// 如果文件不存在,返回 <see langword="null"/>。
  228. /// <para/>如果成功删除到回收站,返回 <see langword="true"/>。
  229. /// <para/>如果永久删除,返回 <see langword="false"/>。
  230. /// </returns>
  231. public static bool? Delete(string filePath, bool toRecycleBin = false) {
  232. if (!FileUtils.Exists(filePath)) return null;
  233. Logger.Trace($"{(toRecycleBin ? "将文件删除到回收站" : "删除文件")}:{filePath}");
  234. // 删除到回收站
  235. if (toRecycleBin) {
  236. try {
  237. DeleteToRecycleBin(filePath);
  238. return true;
  239. } catch (Exception ex) {
  240. Logger.Warn(ex, $"无法将文件删除到回收站,回退到永久删除:{filePath}");
  241. }
  242. }
  243. // 永久删除
  244. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => {
  245. FileUtils.SetReadOnly(filePath, false);
  246. File.Delete(PathUtils.ForApi(filePath));
  247. });
  248. return false;
  249. }
  250. /// <summary>
  251. /// 将文件或文件夹删除到回收站。
  252. /// </summary>
  253. internal static void DeleteToRecycleBin(string target) {
  254. // 实际的删除方法
  255. void Run() {
  256. IShellItem? item = null;
  257. IFileOperation? op = null;
  258. try {
  259. var iid = typeof(IShellItem).GUID;
  260. Marshal.ThrowExceptionForHR(SHCreateItemFromParsingName(PathUtils.RemoveExtendedPrefix(target), IntPtr.Zero, ref iid, out item));
  261. // ReSharper disable once SuspiciousTypeConversion.Global
  262. op = (IFileOperation) new FileOperation();
  263. op.SetOperationFlags(0x0040 | 0x0010 | 0x0004); // FOF_ALLOWUNDO | FOF_NOCONFIRMATION | FOF_SILENT
  264. op.DeleteItem(item, IntPtr.Zero);
  265. op.PerformOperations();
  266. op.GetAnyOperationsAborted(out bool aborted);
  267. if (aborted) throw new OperationCanceledException("Delete operation was aborted.");
  268. } finally {
  269. if (op != null && Marshal.IsComObject(op)) Marshal.FinalReleaseComObject(op);
  270. if (item != null && Marshal.IsComObject(item)) Marshal.FinalReleaseComObject(item);
  271. }
  272. }
  273. // 在 STA 线程中执行删除方法
  274. if (Thread.CurrentThread.GetApartmentState() == ApartmentState.STA) {
  275. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => Run());
  276. } else {
  277. System.Runtime.ExceptionServices.ExceptionDispatchInfo? internalEx = null; // 捕获内部异常
  278. var thread = new Thread(() => {
  279. try {
  280. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException or COMException, action: _ => Run());
  281. } catch (Exception ex) {
  282. internalEx = System.Runtime.ExceptionServices.ExceptionDispatchInfo.Capture(ex);
  283. }
  284. }) { IsBackground = true, Name = nameof(DeleteToRecycleBin) };
  285. thread.SetApartmentState(ApartmentState.STA);
  286. thread.Start();
  287. thread.Join();
  288. internalEx?.Throw();
  289. }
  290. }
  291. // API
  292. [DllImport("shell32.dll", CharSet = CharSet.Unicode)]
  293. static extern int SHCreateItemFromParsingName([MarshalAs(UnmanagedType.LPWStr)] string pszPath, IntPtr pbc, ref Guid riid, [MarshalAs(UnmanagedType.Interface)] out IShellItem ppv);
  294. [ComImport, Guid("43826D1E-E718-42EE-BC55-A1E261C37BFE"), InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
  295. interface IShellItem { }
  296. [ComImport, Guid("3AD05575-8857-4850-9277-11B85BDB8E09")]
  297. class FileOperation { }
  298. [ComImport, Guid("947AAB5F-0A5C-4C13-B4D6-4BF7836FC9F8"), InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
  299. interface IFileOperation {
  300. void Advise(IntPtr pfops, out uint pdwCookie);
  301. void Unadvise(uint dwCookie);
  302. void SetOperationFlags(uint dwOperationFlags);
  303. void SetProgressMessage([MarshalAs(UnmanagedType.LPWStr)] string pszMessage);
  304. void SetProgressDialog(IntPtr popd);
  305. void SetProperties(IntPtr pproparray);
  306. void SetOwnerWindow(IntPtr hwndOwner);
  307. void ApplyPropertiesToItem(IShellItem psiItem);
  308. void ApplyPropertiesToItems(IntPtr punkItems);
  309. void RenameItem(IShellItem psiItem, [MarshalAs(UnmanagedType.LPWStr)] string pszNewName, IntPtr pfopsItem);
  310. void RenameItems(IntPtr pUnkItems, [MarshalAs(UnmanagedType.LPWStr)] string pszNewName);
  311. void MoveItem(IShellItem psiItem, IShellItem psiDestinationFolder, [MarshalAs(UnmanagedType.LPWStr)] string pszNewName, IntPtr pfopsItem);
  312. void MoveItems(IntPtr punkItems, IShellItem psiDestinationFolder);
  313. void CopyItem(IShellItem psiItem, IShellItem psiDestinationFolder, [MarshalAs(UnmanagedType.LPWStr)] string pszCopyName, IntPtr pfopsItem);
  314. void CopyItems(IntPtr punkItems, IShellItem psiDestinationFolder);
  315. void DeleteItem(IShellItem psiItem, IntPtr pfopsItem);
  316. void DeleteItems(IntPtr punkItems);
  317. void NewItem(IShellItem psiDestinationFolder, uint dwFileAttributes, [MarshalAs(UnmanagedType.LPWStr)] string pszName, [MarshalAs(UnmanagedType.LPWStr)] string pszTemplateName, IntPtr pfopsItem);
  318. void PerformOperations();
  319. void GetAnyOperationsAborted([MarshalAs(UnmanagedType.Bool)] out bool pfAnyOperationsAborted);
  320. }
  321. #endregion
  322. #region 压缩 / 解压
  323. /// <summary>
  324. /// 以只读模式打开压缩文件。
  325. /// 会先尝试 UTF8 编码,失败后换用 GB18030 编码。
  326. /// </summary>
  327. public static ZipArchive OpenZip(string zipFilePath, CancellationToken c = default) {
  328. ZipArchive TryOpen(Encoding encoding) {
  329. Logger.Trace($"尝试以 {encoding.EncodingName} 编码打开压缩包:{zipFilePath}");
  330. var result = ZipFile.Open(PathUtils.ForApi(zipFilePath), ZipArchiveMode.Read, encoding);
  331. try {
  332. _ = result.Entries; // 如果编码有误,会在这里抛出 DecoderFallbackException;如果文件异常,会在这里抛出 InvalidDataException
  333. return result;
  334. } catch {
  335. result.Dispose();
  336. throw;
  337. }
  338. }
  339. return Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), maxAttempts:10, isRetryAllowed: ex => ex is IOException, cancellationToken: c, func: _ => {
  340. try {
  341. try { // 尝试两种编码
  342. return TryOpen(new UTF8Encoding(false, true));
  343. } catch (DecoderFallbackException) {
  344. return TryOpen(Encoding.GetEncoding("GB18030"));
  345. }
  346. } catch (InvalidDataException ex) {
  347. throw new InvalidDataException($"文件不是压缩包,或者文件已损坏({zipFilePath})", ex);
  348. }
  349. });
  350. }
  351. /// <summary>
  352. /// 尝试根据后缀名判断文件种类并解压文件,支持 gz 与 zip,会尝试将 jar 以 zip 方式解压。
  353. /// 会自动创建文件夹。会覆盖已有文件,但不会删除多余文件。
  354. /// </summary>
  355. /// <param name="p">参数为已完成的总比例(0~1)。</param>
  356. public static void ExtractToDirectory(string compressionFile, string outputDirectory, CancellationToken c = default, ProgressProvider? p = null) {
  357. compressionFile = PathUtils.ForApi(compressionFile);
  358. DirectoryUtils.Create(outputDirectory);
  359. // 解压 gz(gz 不需要考虑编码)
  360. if (PathUtils.GetExtension(compressionFile) == "gz") {
  361. string outFilePath = Path.Combine(outputDirectory, PathUtils.GetFileNameWithoutExtension(compressionFile));
  362. Logger.Trace($"解压 gz 文件:{compressionFile} → {outFilePath}");
  363. using var fileStream = FileUtils.ReadAsStream(compressionFile);
  364. using var gzipStream = new GZipStream(fileStream, CompressionMode.Decompress);
  365. FileUtils.Write(outFilePath, gzipStream);
  366. p?.Finish();
  367. return;
  368. }
  369. // 解压 zip
  370. using var archive = FileUtils.OpenZip(compressionFile, c);
  371. var entries = archive.Entries.Where(e => !string.IsNullOrEmpty(e.Name)).ToList(); // 跳过文件夹条目(ZipArchive 会将文件夹也作为一个 entry,但它们的 Name 为空)
  372. int totalCount = entries.Count;
  373. int doneCount = 0;
  374. Logger.Trace($"解压 zip 文件:{compressionFile} → {outputDirectory}(共 {totalCount} 项)");
  375. foreach (var entry in entries) {
  376. c.ThrowIfCancellationRequested();
  377. doneCount++;
  378. if (totalCount > 0) p?.Set((double) doneCount / totalCount);
  379. // ZipSlip 修复
  380. string outputFilePath = Path.Combine(outputDirectory, entry.FullName);
  381. if (!PathUtils.IsParentOf(outputDirectory, outputFilePath))
  382. throw new InvalidDataException($"Zip 文件项 {entry.FullName} 的路径在压缩包之外,这可能导致安全问题");
  383. // 实际的解压
  384. using var entryStream = entry.Open();
  385. FileUtils.Write(outputFilePath, entryStream);
  386. }
  387. }
  388. /// <summary>
  389. /// 将指定文件夹的内容打包为 zip 文件。
  390. /// 会自动创建文件夹。会覆盖已有文件。
  391. /// </summary>
  392. public static void CreateZipFromDirectory(string outputFullPath, string sourceDirectory) {
  393. DirectoryUtils.Create(PathUtils.RemoveLastPart(outputFullPath));
  394. FileUtils.Delete(outputFullPath);
  395. Logger.Trace($"将文件夹中的内容压缩为 zip 文件:{sourceDirectory} → {outputFullPath}");
  396. Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _
  397. => ZipFile.CreateFromDirectory(PathUtils.ForApi(sourceDirectory), PathUtils.ForApi(outputFullPath)));
  398. }
  399. /// <summary>
  400. /// 将多个文件打包为 zip 文件,所有文件都会被放在压缩文件的根目录。
  401. /// 会自动创建文件夹。会覆盖已有文件。
  402. /// </summary>
  403. public static void CreateZipFromFiles(string outputFullPath, params string[] sourceFiles) {
  404. var sources = new Dictionary<string, string>();
  405. foreach (var source in sourceFiles) {
  406. string fileName = PathUtils.GetLastPart(source);
  407. if (sources.ContainsKey(fileName)) throw new ArgumentException($"尝试将多个同文件名的文件放进压缩包中({fileName})", nameof(sourceFiles));
  408. sources.Add(fileName, source);
  409. }
  410. FileUtils.CreateZipFromFiles(outputFullPath, sources);
  411. }
  412. /// <summary>
  413. /// 将多个文件打包为 zip 文件。
  414. /// 会自动创建文件夹。会覆盖已有文件。
  415. /// </summary>
  416. /// <param name="sources">键为 zip 文件下的路径,值为文件的本地路径。</param>
  417. public static void CreateZipFromFiles(string outputFullPath, IDictionary<string, string> sources) {
  418. DirectoryUtils.Create(PathUtils.RemoveLastPart(outputFullPath));
  419. FileUtils.Delete(outputFullPath);
  420. using var archive = Retrier.Attempt(delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, func: _
  421. => ZipFile.Open(PathUtils.ForApi(outputFullPath), ZipArchiveMode.Create));
  422. Logger.Trace($"创建 zip 文件:{sources.Count} 个文件 → {outputFullPath}\n{sources.Select(p => $"- {p.Value} → {p.Key}").Join('\n')}");
  423. foreach (var pair in sources) archive.CreateEntryFromFile(PathUtils.ForApi(pair.Value), pair.Key.Replace('\\', '/'), CompressionLevel.Optimal);
  424. }
  425. #endregion
  426. /// <summary>
  427. /// 确定指定的文件是否存在。
  428. /// </summary>
  429. public static bool Exists(string filePath)
  430. => File.Exists(PathUtils.ForApi(filePath));
  431. /// <summary>
  432. /// 检查文件是否未被其他进程占用
  433. /// <para>被占用或无权读取时返回 <see langword="false"/>。</para>
  434. /// </summary>
  435. public static bool IsAvailable(string filePath) {
  436. if (!Exists(filePath)) return true;
  437. try {
  438. Retrier.Attempt(maxAttempts: 3, delay: _ => TimeSpan.FromMilliseconds(200), isRetryAllowed: ex => ex is IOException, action: _ => {
  439. using var stream = new FileStream(PathUtils.ForApi(filePath), FileMode.Open, FileAccess.Read, FileShare.None);
  440. });
  441. return true;
  442. } catch (Exception ex) when (ex is IOException or UnauthorizedAccessException) {
  443. return false;
  444. }
  445. }
  446. /// <summary>
  447. /// 获取 <see cref="FileInfo"/> 对象。
  448. /// </summary>
  449. public static FileInfo GetInfo(string path)
  450. => new(PathUtils.ForApi(path));
  451. /// <summary>
  452. /// 设置文件或文件夹的只读属性。
  453. /// 对文件夹使用时,只会设置文件夹本身的属性,不会递归设置其中的文件或子文件夹。
  454. /// </summary>
  455. public static void SetReadOnly(string path, bool readOnly) {
  456. path = PathUtils.ForApi(path);
  457. var attributes = File.GetAttributes(path);
  458. var newAttributes = readOnly
  459. ? attributes | FileAttributes.ReadOnly
  460. : attributes & ~FileAttributes.ReadOnly;
  461. if (attributes == newAttributes) return;
  462. Logger.Trace($"{(readOnly ? "添加" : "移除")}只读属性:{path}(原属性:{attributes},新属性:{newAttributes})");
  463. File.SetAttributes(path, newAttributes);
  464. }
  465. }
  466. /// <summary>
  467. /// 检查文件是否符合校验规则,包括大小、Hash 值、是否为 JSON 文件。
  468. /// </summary>
  469. public class FileChecker {
  470. /// <summary>
  471. /// 文件的准确大小。
  472. /// <para/>不检查则为 -1。
  473. /// </summary>
  474. public long ActualSize = -1;
  475. /// <summary>
  476. /// 文件的最小大小。
  477. /// <para/>不检查则为 -1。
  478. /// </summary>
  479. public long MinSize = -1;
  480. /// <summary>
  481. /// 文件的 MD5、SHA1、SHA256 或 SHA512。会根据输入字符串的长度自动判断种类。
  482. /// <para/>不检查则为 <c>null</c>。
  483. /// </summary>
  484. public string? Hash = null;
  485. /// <summary>
  486. /// 是否要求为 JSON 文件。
  487. /// </summary>
  488. public bool IsJson = false;
  489. // TODO: 重构此函数,使它变成 void,并抛出异常
  490. /// <summary>
  491. /// 检查文件。
  492. /// <para/>若成功则返回 <c>null</c>,失败则返回错误的描述文本(文本不以句号结尾)。
  493. /// <para/>不会抛出异常。
  494. /// </summary>
  495. public string? Check(string localPath) {
  496. try {
  497. var info = FileUtils.GetInfo(localPath);
  498. if (!info.Exists) return "文件不存在:" + localPath;
  499. long fileSize = info.Length;
  500. if (ActualSize >= 0 && ActualSize != fileSize) {
  501. return $"文件大小应为 {ActualSize} B,实际为 {fileSize} B" +
  502. (fileSize < 2000 ? ",内容为:" + FileUtils.ReadAsString(localPath) : "");
  503. }
  504. if (MinSize >= 0 && MinSize > fileSize) {
  505. return $"文件大小应至少为 {MinSize} B,实际为 {fileSize} B" +
  506. (fileSize < 2000 ? ",内容为:" + FileUtils.ReadAsString(localPath) : "");
  507. }
  508. string? hash = Hash;
  509. if (!string.IsNullOrEmpty(hash)) {
  510. hash = hash!.Lower();
  511. switch (hash.Length) {
  512. case 32:
  513. string md5 = CryptographyUtils.ComputeFileHash(localPath, CryptographyUtils.HashMethod.Md5);
  514. if (hash != md5) return "文件 MD5 应为 " + hash + ",实际为 " + md5;
  515. break;
  516. case 40:
  517. string sha1 = CryptographyUtils.ComputeFileHash(localPath, CryptographyUtils.HashMethod.Sha1);
  518. if (hash != sha1) return "文件 SHA1 应为 " + hash + ",实际为 " + sha1;
  519. break;
  520. case 64:
  521. string sha256 = CryptographyUtils.ComputeFileHash(localPath, CryptographyUtils.HashMethod.Sha256);
  522. if (hash != sha256) return "文件 SHA256 应为 " + hash + ",实际为 " + sha256;
  523. break;
  524. case 128:
  525. string sha512 = CryptographyUtils.ComputeFileHash(localPath, CryptographyUtils.HashMethod.Sha512);
  526. if (hash != sha512) return "文件 SHA512 应为 " + hash + ",实际为 " + sha512;
  527. break;
  528. default:
  529. return $"无法识别的 Hash 类型(长度为 {hash.Length}):{hash}";
  530. }
  531. }
  532. if (IsJson) {
  533. string content = FileUtils.ReadAsString(localPath);
  534. if (content == "") return "读取到的文件为空";
  535. content.DeserializeJson();
  536. }
  537. return null;
  538. } catch (Exception ex) {
  539. Logger.Warn(ex, "检查文件出错");
  540. return ex.GetDisplay(false);
  541. }
  542. }
  543. }