AbstractZipDetailDetector.cs 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. using System.Collections.Generic;
  2. using System.IO;
  3. using System.IO.Compression;
  4. using System.Linq;
  5. using System.Reflection;
  6. using Masuit.Tools.Mime;
  7. namespace Masuit.Tools.Files.FileDetector;
  8. public abstract class AbstractZipDetailDetector : IDetector
  9. {
  10. public abstract IEnumerable<string> Files { get; }
  11. public abstract string Extension { get; }
  12. public virtual string Precondition => null;
  13. public virtual string MimeType => new MimeMapper().GetMimeFromExtension("." + Extension);
  14. public virtual List<FormatCategory> FormatCategories => GetType().GetCustomAttributes<FormatCategoryAttribute>().Select(a => a.Category).ToList();
  15. protected virtual bool IsValid(string filename, ZipArchiveEntry entry)
  16. {
  17. return true;
  18. }
  19. public bool Detect(Stream stream)
  20. {
  21. try
  22. {
  23. using var archive = new ZipArchive(stream, ZipArchiveMode.Read, true);
  24. foreach (string filename in Files)
  25. {
  26. bool succeed = archive.Entries.Any(entry => entry.FullName == filename && IsValid(filename, entry));
  27. if (!succeed)
  28. {
  29. return false;
  30. }
  31. }
  32. return true;
  33. }
  34. catch
  35. {
  36. return false;
  37. }
  38. }
  39. }