Add retry mechanism for doujin download

Xwilarg · Sep 22, 2024 · 73f92fd · 73f92fd
1 parent 14a4ccf
commit 73f92fd
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/Sanara/Module/Utility/EHentai.cs b/Sanara/Module/Utility/EHentai.cs
@@ -48,11 +48,25 @@ public static async Task EHentaiDownloadAsync(IContext ctx, IServiceProvider pro
                 {
                     if (img.HasClass("gdtm")) // Get all images to download them
                     {
-                        var node = img.FirstChild.FirstChild.Attributes["href"].Value;
-                        var image = web.Load(node).GetElementbyId("img").Attributes["src"].Value;
-                        File.WriteAllBytes($"Saves/Download/{dirName}/{pageIndex:000}{Path.GetExtension(image)}",
-                        await provider.GetRequiredService<HttpClient>().GetByteArrayAsync(image));
-                        pageIndex++;
+                        System.Exception err = null;
+                        for (int c = 0; c < 2; c++)
+                        {
+                            try
+                            {
+                                var node = img.FirstChild.FirstChild.Attributes["href"].Value;
+                                var image = web.Load(node).GetElementbyId("img").Attributes["src"].Value;
+                                File.WriteAllBytes($"Saves/Download/{dirName}/{pageIndex:000}{Path.GetExtension(image)}",
+                                await provider.GetRequiredService<HttpClient>().GetByteArrayAsync(image));
+                                pageIndex++;
+                                err = null;
+                                break;
+                            }
+                            catch (System.Exception e)
+                            {
+                                err = e;
+                            }
+                        }
+                        if (err != null) throw err;
                     }
                 }