FileSignatureDetector.cs 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. using System;
  2. using System.Collections.Generic;
  3. using System.IO;
  4. using System.Linq;
  5. using System.Reflection;
  6. namespace Masuit.Tools.Files.FileDetector;
  7. public static class FileSignatureDetector
  8. {
  9. private static List<IDetector> Detectors { get; set; } = new();
  10. public static IReadOnlyList<IDetector> Registered => Detectors;
  11. public static void AddDetector<T>() where T : IDetector
  12. {
  13. var instance = Activator.CreateInstance<T>();
  14. AddDetector(instance);
  15. }
  16. public static void AddDetector(IDetector instance)
  17. {
  18. if (!Detectors.Contains(instance))
  19. {
  20. Detectors.Add(instance);
  21. }
  22. }
  23. static FileSignatureDetector()
  24. {
  25. var detectorTypeInfo = typeof(IDetector).GetTypeInfo();
  26. foreach (var type in AppDomain.CurrentDomain.GetAssemblies().SelectMany(a => a.DefinedTypes))
  27. {
  28. if (detectorTypeInfo.IsAssignableFrom(type) && !type.IsAbstract && type.DeclaredConstructors.First().GetParameters().Length == 0)
  29. {
  30. AddDetector(Activator.CreateInstance(type.AsType()) as IDetector);
  31. }
  32. }
  33. }
  34. public static IDetector DetectFiletype(string filepath)
  35. {
  36. using var stream = File.OpenRead(filepath);
  37. return DetectFiletype(stream);
  38. }
  39. public static IDetector DetectFiletype(this FileInfo file)
  40. {
  41. using var stream = file.OpenRead();
  42. return DetectFiletype(stream);
  43. }
  44. public static IDetector DetectFiletype(this Stream stream)
  45. {
  46. string pre = null;
  47. IDetector foundDetector = new NoneDetector();
  48. while (true)
  49. {
  50. bool found = false;
  51. foreach (var detector in Detectors.Where(d => d.Precondition == pre))
  52. {
  53. stream.Position = 0;
  54. if (detector.Detect(stream))
  55. {
  56. found = true;
  57. foundDetector = detector;
  58. pre = detector.Extension;
  59. break;
  60. }
  61. }
  62. if (!found)
  63. {
  64. break;
  65. }
  66. }
  67. return foundDetector;
  68. }
  69. }