Update MovieHub.cs

taylorbobaylor · May 1, 2023 · 8195581 · 8195581
1 parent 8b33de5
commit 8195581
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/TrailerDownloader/SignalRHubs/MovieHub.cs b/TrailerDownloader/SignalRHubs/MovieHub.cs
@@ -100,13 +100,21 @@ private void GetMovieDirectories(string directoryPath)
             var subDirectories = Directory.EnumerateDirectories(directoryPath, "*", SearchOption.AllDirectories);
 
             // Add the movie directories to the collection
+            bool hasSubdirectories = false;
             foreach (var subDirectory in subDirectories)
             {
-                if (Directory.GetFiles(subDirectory).Length > 0)
+                if (Directory.GetDirectories(subDirectory).Length > 0)
                 {
                     _movieDirectories.Add(subDirectory);
+                    hasSubdirectories = true;
                 }
             }
+
+            // If no subdirectories were added, add the main directoryPath
+            if (!hasSubdirectories)
+            {
+                _movieDirectories.Add(directoryPath);
+            }
         }
         catch (Exception ex)
         {