From 1834d0182e84f28ec8b1abd17920a8c488f48a20 Mon Sep 17 00:00:00 2001
From: Pat Hartl <pat@pathar.tl>
Date: Sat, 11 Feb 2023 21:39:06 -0600
Subject: [PATCH] Release keys by supplying key model

---
 LANCommander/Services/KeyService.cs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/LANCommander/Services/KeyService.cs b/LANCommander/Services/KeyService.cs
index d331bac..cda2b36 100644
--- a/LANCommander/Services/KeyService.cs
+++ b/LANCommander/Services/KeyService.cs
@@ -29,13 +29,18 @@ namespace LANCommander.Services
             return key;
         }
 
-        public async Task Release(Guid id)
+        public async Task<Key> Release(Guid id)
         {
             var key = await Get(id);
 
             if (key == null)
-                return;
+                return null;
 
+            return await Release(key);
+        }
+
+        public async Task<Key> Release(Key key)
+        {
             switch (key.AllocationMethod)
             {
                 case KeyAllocationMethod.UserAccount:
@@ -49,7 +54,7 @@ namespace LANCommander.Services
                     break;
             }
 
-            await Update(key);
+            return await Update(key);
         }
     }
 }