Fix repo request

BigBang1112 · Dec 18, 2024 · 326c4f4 · 326c4f4
1 parent 7ff3ff1
commit 326c4f4
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/Src/GBX.NET.Tool.CLI/ToolConsole.cs b/Src/GBX.NET.Tool.CLI/ToolConsole.cs
@@ -140,7 +140,7 @@ public async Task RunAsync(ExecuteLogic executeLogic, CancellationToken cancella
         // Request update info and additional stuff
         var updateChecker = toolSettings.ConsoleSettings.DisableUpdateCheck
             ? null
-            : ToolUpdateChecker.Check(http, cancellationToken);
+            : ToolUpdateChecker.Check(http, options.GitHubRepo, cancellationToken);
 
         if (introWriterTask is not null)
         {

diff --git a/Src/GBX.NET.Tool.CLI/ToolConsoleOptions.cs b/Src/GBX.NET.Tool.CLI/ToolConsoleOptions.cs
@@ -14,4 +14,9 @@ public sealed record ToolConsoleOptions
     public YamlDotNet.Serialization.DeserializerBuilder? YmlDeserializer { get; init; }
     public YamlDotNet.Serialization.SerializerBuilder? YmlSerializer { get; init; }
     public YamlDotNet.Serialization.StaticContext? YmlContext { get; init; }
+
+    /// <summary>
+    /// GitHub repository identifier where the releases are located. Format: <c>user/repo</c>
+    /// </summary>
+    public string? GitHubRepo { get; set; }
 }
diff --git a/Src/GBX.NET.Tool.CLI/ToolUpdateChecker.cs b/Src/GBX.NET.Tool.CLI/ToolUpdateChecker.cs
@@ -11,9 +11,14 @@ public ToolUpdateChecker(Task<HttpResponseMessage> updateInfoResponseTask)
         this.updateInfoResponseTask = updateInfoResponseTask;
     }
 
-    public static ToolUpdateChecker Check(HttpClient client, CancellationToken cancellationToken)
+    public static ToolUpdateChecker? Check(HttpClient client, string? githubRepo, CancellationToken cancellationToken)
     {
-        var responseTask = client.GetAsync("https://api.github.com/repos/GBX.NET/GBX.NET.Tool/releases/latest", cancellationToken);
+        if (githubRepo is null)
+        {
+            return null;
+        }
+
+        var responseTask = client.GetAsync($"https://api.github.com/repos/{githubRepo}/releases/latest", cancellationToken);
         return new ToolUpdateChecker(responseTask);
     }