Update CDBuilder.cs

Adding Progress
LTRData · May 8, 2024 · de30a4f · de30a4f
1 parent b2900c2
commit de30a4f
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/Library/DiscUtils.Iso9660/CDBuilder.cs b/Library/DiscUtils.Iso9660/CDBuilder.cs
@@ -56,6 +56,17 @@ public sealed class CDBuilder : StreamBuilder, IFileSystemBuilder
     private readonly List<BuildFileInfo> _files;
     private readonly BuildDirectoryInfo _rootDirectory;
 
+     // Define a delegate for progress reporting
+    public delegate void ProgressHandler(int currentProgress, int totalItems);
+    public event ProgressHandler ProgressChanged;
+
+ // Method for updating progress
+ private void UpdateProgress(int currentProgress, int totalItems)
+ {
+     ProgressChanged?.Invoke(currentProgress, totalItems);
+ }
+
+
     /// <summary>
     /// Initializes a new instance of the CDBuilder class.
     /// </summary>
@@ -203,6 +214,7 @@ public BuildFileInfo AddFile(string name, byte[] content)
         var fi = new BuildFileInfo(nameElements[nameElements.Length - 1].ToString(), dir, content);
         AddFile(fi);
         dir.Add(fi);
+        UpdateProgress(_files.Count, _files.Count + _dirs.Count);
         return fi;
     }
 
@@ -227,6 +239,7 @@ public BuildFileInfo AddFile(string name, string sourcePath)
         var fi = new BuildFileInfo(nameElements[nameElements.Length - 1].ToString(), dir, sourcePath);
         AddFile(fi);
         dir.Add(fi);
+        UpdateProgress(_files.Count, _files.Count + _dirs.Count);
         return fi;
     }
 
@@ -256,6 +269,7 @@ public BuildFileInfo AddFile(string name, Stream source)
         var fi = new BuildFileInfo(nameElements[nameElements.Length - 1].ToString(), dir, source);
         AddFile(fi);
         dir.Add(fi);
+        UpdateProgress(_files.Count, _files.Count + _dirs.Count);
         return fi;
     }
 
@@ -635,4 +649,4 @@ void IFileSystemBuilder.AddFile(string name, string sourcePath, DateTime creatio
     bool IFileSystemBuilder.Exists(string path) => GetFile(path) != null;
 
     public IFileSystem GenerateFileSystem() => new CDReader(Build(), UseJoliet);
-}
+}