From 64510c5fc260a92268d305cf6f680495e8397422 Mon Sep 17 00:00:00 2001
From: STAM <stam@epicm.org>
Date: Sun, 4 Aug 2024 21:08:45 +0300
Subject: [PATCH] pre-2.0.0.2

---
 src/Core/DownloaderService.cs   | 46 +++++++++++++++++++++------------
 src/Models/DownloaderOptions.cs |  5 +++-
 src/Program.cs                  |  8 +++++-
 src/atlassian-downloader.csproj | 27 +++++++++----------
 4 files changed, 54 insertions(+), 32 deletions(-)

diff --git a/src/Core/DownloaderService.cs b/src/Core/DownloaderService.cs
index 81b9bf5..9a1cbbd 100644
--- a/src/Core/DownloaderService.cs
+++ b/src/Core/DownloaderService.cs
@@ -213,31 +213,43 @@ private async Task DownloadFilesFromFeed(string feedUrl, IDictionary<string, Res
 
     private async Task DownloadFile(ResponseItem file, string outputFile, CancellationToken cancellationToken)
     {
-        if (!string.IsNullOrEmpty(file.Md5))
+        for (int attempt = 1; attempt <= options.MaxRetries; attempt++)
         {
-            File.WriteAllText(outputFile + ".md5", file.Md5);
-        }
+            if (!string.IsNullOrEmpty(file.Md5))
+            {
+                File.WriteAllText(outputFile + ".md5", file.Md5);
+            }
 
-        try
-        {
-            using var outputStream = File.OpenWrite(outputFile);
-            using var request = await this.client.GetStreamAsync(file.ZipUrl, cancellationToken).ConfigureAwait(false);
-            await request.CopyToAsync(outputStream, cancellationToken).ConfigureAwait(false);
-        }
-        catch (Exception downloadEx)
-        {
-            this.logger.LogError(downloadEx, "Failed to download file \"{uri}\" to \"{outputFile}\".", file.ZipUrl, outputFile);
             try
             {
-                File.Delete(outputFile);
+                using var outputStream = File.OpenWrite(outputFile);
+                using var request = await this.client.GetStreamAsync(file.ZipUrl, cancellationToken).ConfigureAwait(false);
+                await request.CopyToAsync(outputStream, cancellationToken).ConfigureAwait(false);
             }
-            catch (Exception removeEx)
+            catch (Exception downloadEx)
             {
-                this.logger.LogError(removeEx, "Failed to remove incomplete file \"{outputFile}\".", outputFile);
+                this.logger.LogError(downloadEx, "Attempt {attempt} failed to download file \"{uri}\" to \"{outputFile}\".", attempt, file.ZipUrl, outputFile);
+
+                if (attempt == options.MaxRetries)
+                {
+                    this.logger.LogError(downloadEx, "Failed to download file \"{uri}\" to \"{outputFile}\".", file.ZipUrl, outputFile);
+                    try
+                    {
+                        File.Delete(outputFile);
+                    }
+                    catch (Exception removeEx)
+                    {
+                        this.logger.LogError(removeEx, "Failed to remove incomplete file \"{outputFile}\".", outputFile);
+                    }
+                    throw; 
+                }
+                else
+                {
+                    await Task.Delay(options.DelayBetweenRetries, cancellationToken).ConfigureAwait(false); // Задержка перед следующей попыткой
+                }
             }
+            this.logger.LogInformation("File \"{uri}\" successfully downloaded to \"{outputFile}\".", file.ZipUrl, outputFile);
         }
-
-        this.logger.LogInformation("File \"{uri}\" successfully downloaded to \"{outputFile}\".", file.ZipUrl, outputFile);
     }
 
 #pragma warning disable CS1998 // Async method lacks 'await' operators and will run synchronously
diff --git a/src/Models/DownloaderOptions.cs b/src/Models/DownloaderOptions.cs
index 289cd54..8289da9 100644
--- a/src/Models/DownloaderOptions.cs
+++ b/src/Models/DownloaderOptions.cs
@@ -8,4 +8,7 @@ public record DownloaderOptions(
     bool Version,
     string? ProductVersion,
     bool SkipFileCheck,
-    string UserAgent) { }
\ No newline at end of file
+    string UserAgent,
+    int MaxRetries,
+    int DelayBetweenRetries
+    ) { }
\ No newline at end of file
diff --git a/src/Program.cs b/src/Program.cs
index eacb8a3..44013ac 100644
--- a/src/Program.cs
+++ b/src/Program.cs
@@ -24,6 +24,8 @@ public class Program
     /// <param name="productVersion">Override target version to download some product. Advice: Use it with "customFeed".</param>
     /// <param name="skipFileCheck">Skip compare of file sizes if a local file already exists. Existing file will be skipped to check and redownload.</param>
     /// <param name="userAgent">Set custom user agent via this feature flag.</param>
+    /// <param name="maxRetries">Set custom count of download retries.</param>
+    /// <param name="delayBetweenRetries">Set custom delay between retries (in milliseconds).</param>
     static async Task Main(
         string? outputDir = default,
         Uri[]? customFeed = null,
@@ -31,6 +33,8 @@ static async Task Main(
         bool about = false,
         string? productVersion = null,
         bool skipFileCheck = false,
+        int maxRetries = 5,
+        int delayBetweenRetries = 2500,
         string userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:101.0) Gecko/20100101 Firefox/101.0") => await
         Host
             .CreateDefaultBuilder()
@@ -60,7 +64,9 @@ static async Task Main(
                         about,
                         productVersion,
                         skipFileCheck,
-                        userAgent))
+                        userAgent,
+                        maxRetries,
+                        delayBetweenRetries))
                    .AddHttpClient())
             .RunConsoleAsync()
             .ConfigureAwait(false);
diff --git a/src/atlassian-downloader.csproj b/src/atlassian-downloader.csproj
index 493a125..1c2b4d9 100644
--- a/src/atlassian-downloader.csproj
+++ b/src/atlassian-downloader.csproj
@@ -12,6 +12,7 @@
     <!--build props-->
     <Nullable>enable</Nullable>
     <OutputType>Exe</OutputType>
+    <GenerateAssemblyInfo>true</GenerateAssemblyInfo>
     <TargetFramework>net8.0</TargetFramework>
     <IsPackable>false</IsPackable>
     <ApplicationIcon>favicon.ico</ApplicationIcon>
@@ -26,25 +27,25 @@
     <RepositoryType>git</RepositoryType>
     <RepositoryUrl>https://github.com/EpicMorg/atlassian-downloader</RepositoryUrl>
     <PackageTags>atlassian, donwloader, epicmorg</PackageTags>
-    <AssemblyVersion>2.0.0.0</AssemblyVersion>
-    <FileVersion>2.0.0.0</FileVersion>
-    <Version>2.0.0.0</Version>
-    <Copyright>EpicMorg 2023</Copyright>
+    <AssemblyVersion>2.0.0.2</AssemblyVersion>
+    <FileVersion>2.0.0.2</FileVersion>
+    <Version>2.0.0.2</Version>
+    <Copyright>EpicMorg 2024</Copyright>
     <Product>Atlassian Downloader</Product>
     <Company>EpicMorg</Company>
     <PackageReadmeFile>README.md</PackageReadmeFile>
     <RootNamespace>EpicMorg.Atlassian.Downloader</RootNamespace>
   </PropertyGroup>
   <ItemGroup>
-    <PackageReference Include="Microsoft.Extensions.DependencyInjection" Version="8.0.0-rc.2.23479.6" />
-    <PackageReference Include="Microsoft.Extensions.Hosting" Version="8.0.0-rc.2.23479.6" />
-    <PackageReference Include="Microsoft.Extensions.Http" Version="8.0.0-rc.2.23479.6" />
-    <PackageReference Include="Microsoft.Extensions.Logging" Version="8.0.0-rc.2.23479.6" />
-    <PackageReference Include="Microsoft.Extensions.Logging.Console" Version="8.0.0-rc.2.23479.6" />
-    <PackageReference Include="Serilog.Extensions.Logging" Version="7.0.1-dev-10354" />
-    <PackageReference Include="Serilog.Settings.Configuration" Version="7.0.2-dev-00546" />
-    <PackageReference Include="Serilog.Sinks.Console" Version="4.2.0-dev-00918" />
-    <PackageReference Include="Serilog" Version="3.1.0-dev-02078" />
+    <PackageReference Include="Microsoft.Extensions.DependencyInjection" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.Hosting" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.Http" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.Logging" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.Logging.Console" Version="8.0.0" />
+    <PackageReference Include="Serilog.Extensions.Logging" Version="8.0.0" />
+    <PackageReference Include="Serilog.Settings.Configuration" Version="8.0.2" />
+    <PackageReference Include="Serilog.Sinks.Console" Version="6.0.0" />
+    <PackageReference Include="Serilog" Version="4.0.1" />
     <PackageReference Include="System.CommandLine.DragonFruit" Version="0.4.0-alpha.22272.1" />
     <None Include="..\README.md">
       <Pack>True</Pack>