From 90d1e735ff6a8f046b504d81702ebe8192d443f1 Mon Sep 17 00:00:00 2001
From: Angelo Breuer <angelobr@outlook.de>
Date: Wed, 6 Mar 2024 21:51:30 +0100
Subject: [PATCH] feat: Add UseLavalink/Configure helper method

---
 .../HostBuilderExtensions.cs                     | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/Lavalink4NET.NetCord/HostBuilderExtensions.cs b/src/Lavalink4NET.NetCord/HostBuilderExtensions.cs
index e5632d80..e03ee38a 100644
--- 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);
+            }
+        });
+    }
 }