Supporting folders in AwaitFileCopy

TorrentExtractor/Extensions.cs

@@ -0,0 +1,10 @@
+using System.IO;
+using System.Linq;
+
+namespace TorrentExtractor;
+
+public static class Extensions
+{
+    public static long Length(this DirectoryInfo dir) =>
+        dir.GetFiles().Sum(fi => fi.Length) + dir.GetDirectories().Sum(di => di.Length());
+}

TorrentExtractor/Worker.cs

@@ -130,7 +130,7 @@ await ExtractAndMoveAsync(
 
     private async Task AwaitFileCopy(
         string sourcePath,
-        long previousFileLength,
+        long previousLength,
         Core coreSettings,
         CancellationToken cancellationToken
     )
@@ -140,10 +140,20 @@ CancellationToken cancellationToken
             throw new TaskCanceledException();
         }
 
+        if (
+            string.IsNullOrWhiteSpace(sourcePath)
+            || !File.Exists(sourcePath) && !Directory.Exists(sourcePath)
+        )
+        {
+            throw new FileNotFoundException("Source path is not found!", sourcePath);
+        }
+
         var interval = coreSettings.FileCompareInterval;
-        var fileLength = new FileInfo(sourcePath).Length;
+        var length = Directory.Exists(sourcePath)
+            ? new DirectoryInfo(sourcePath).Length()
+            : new FileInfo(sourcePath).Length;
 
-        if (fileLength != previousFileLength)
+        if (length != previousLength)
         {
             _logger.LogInformation(
                 "File '{SourcePath}' is still being copied. Waiting for {Interval} seconds...",
@@ -152,7 +162,7 @@ CancellationToken cancellationToken
             );
 
             await Task.Delay(TimeSpan.FromSeconds(interval), cancellationToken);
-            await AwaitFileCopy(sourcePath, fileLength, coreSettings, cancellationToken);
+            await AwaitFileCopy(sourcePath, length, coreSettings, cancellationToken);
         }
     }