support custom maven repositories

mysticmind · Aug 29, 2023 · c666f20 · c666f20
1 parent 64abad4
commit c666f20
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/src/MysticMind.PostgresEmbed/DefaultPostgresBinaryDownloader.cs b/src/MysticMind.PostgresEmbed/DefaultPostgresBinaryDownloader.cs
@@ -13,6 +13,7 @@ internal class DefaultPostgresBinaryDownloader
     private readonly Platform _platform;
     private readonly Architecture? _architecture;
     private readonly string _destDir;
+    private readonly string _mavenRepo;
 
     /// <summary>
     /// The default Postgres binary downloader uses https://github.com/zonkyio/embedded-postgres-binaries
@@ -22,12 +23,14 @@ internal class DefaultPostgresBinaryDownloader
     /// <param name="pgVersion"></param>
     /// <param name="platform"></param>
     /// <param name="architecture"></param>
-    public DefaultPostgresBinaryDownloader(string pgVersion, string destDir, Platform platform, Architecture? architecture)
+    /// <param name="mavenRepo"></param>
+    public DefaultPostgresBinaryDownloader(string pgVersion, string destDir, Platform platform, Architecture? architecture, string mavenRepo)
     {
         _destDir = destDir;
         _pgVersion = pgVersion;
         _platform = platform;
         _architecture = architecture;
+        _mavenRepo = mavenRepo;
     }
 
     public string Download()
@@ -45,7 +48,7 @@ public string Download()
             architecture = "alpine-lite";
         }
 
-        var downloadUrl = $"https://repo1.maven.org/maven2/io/zonky/test/postgres/embedded-postgres-binaries-{platform}-{architecture}/{_pgVersion}/embedded-postgres-binaries-{platform}-{architecture}-{_pgVersion}.jar";
+        var downloadUrl = $"{_mavenRepo}/io/zonky/test/postgres/embedded-postgres-binaries-{platform}-{architecture}/{_pgVersion}/embedded-postgres-binaries-{platform}-{architecture}-{_pgVersion}.jar";
         var fileName = Path.GetFileName(downloadUrl);
         var destFile = Path.Join(_destDir, fileName);
         var zipFile = Path.Join(_destDir, Path.GetFileNameWithoutExtension(fileName) + ".txz");

diff --git a/src/MysticMind.PostgresEmbed/PgServer.cs b/src/MysticMind.PostgresEmbed/PgServer.cs
@@ -42,6 +42,7 @@ public class PgServer : IDisposable
 
         private readonly Platform _platform;
         private readonly Architecture _architecture;
+        private readonly string _mavenRepo;
 
         public PgServer(
             string pgVersion,
@@ -58,7 +59,8 @@ public PgServer(
             int deleteFolderInitialTimeout =16, 
             int deleteFolderTimeoutFactor =2,
             string locale = "",
-            Platform? platform = null)
+            Platform? platform = null,
+            string mavenRepo = "https://repo1.maven.org/maven2")
         {
 
             _pgCtlBin = "pg_ctl";
@@ -84,6 +86,8 @@ public PgServer(
 
             _architecture = Utils.GetArchitecture(_platform);
 
+            _mavenRepo = mavenRepo;
+
             PgUser = String.IsNullOrEmpty(pgUser) ? PgSuperuser : pgUser;
 
             DbDir = Path.Combine(string.IsNullOrEmpty(dbDir) ? "." : dbDir, "pg_embed");
@@ -157,7 +161,7 @@ public PgServer(
 
         private void DownloadPgBinary()
         {
-            var downloader = new DefaultPostgresBinaryDownloader(PgVersion, BinariesDir, _platform, _architecture);
+            var downloader = new DefaultPostgresBinaryDownloader(PgVersion, BinariesDir, _platform, _architecture, _mavenRepo);
 
             try
             {