feat: Add UseLavalink/Configure helper method

angelobreuer · Mar 6, 2024 · 90d1e73 · 90d1e73
1 parent 5c92914
commit 90d1e73
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/Lavalink4NET.NetCord/HostBuilderExtensions.cs b/src/Lavalink4NET.NetCord/HostBuilderExtensions.cs
@@ -1,6 +1,7 @@
 namespace Lavalink4NET.NetCord;
 
 using System;
+using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 
 public static class HostBuilderExtensions
@@ -11,4 +12,19 @@ public static IHostBuilder UseLavalink(this IHostBuilder hostBuilder)
 
         return hostBuilder.ConfigureServices(static (_, services) => services.AddLavalink());
     }
+
+    public static IHostBuilder UseLavalink(this IHostBuilder hostBuilder, Action<AudioServiceOptions>? configure)
+    {
+        ArgumentNullException.ThrowIfNull(hostBuilder);
+
+        return hostBuilder.ConfigureServices((_, services) =>
+        {
+            services.AddLavalink();
+
+            if (configure is not null)
+            {
+                services.Configure(configure);
+            }
+        });
+    }
 }