Implement IKeyManager

Akari.Prototype.Server/Services/AuthManager.cs

@@ -26,8 +26,8 @@ namespace Akari.Prototype.Server.Services
 
         private readonly ILogger<AuthManager> _logger;
         private readonly AkariPath _akariPath;
+        private readonly IDictionary<string, TimedEntry<AesGcm>> _keys;
 
-        private IDictionary<string, TimedEntry<AesGcm>> _keys;
         private IDictionary<string, string> _salts;
 
         public AuthManager(ILogger<AuthManager> logger, AkariPath akariPath)
@@ -91,9 +91,49 @@ namespace Akari.Prototype.Server.Services
             SetKey(name, new AesGcm(key.Buffer));
         }
 
+        public bool TryGetKey(string name, out AesGcm key)
+        {
+            if (_keys.TryGetValue(name, out var entry))
+            {
+                key = entry.Value;
+
+                return true;
+            }
+
+            key = null;
+
+            return false;
+        }
+
+        public bool TryGetKey(IEnumerable<string> names, out string name, out AesGcm key)
+        {
+            foreach (var n in names)
+            {
+                if (TryGetKey(n, out var k))
+                {
+                    name = n;
+                    key = k;
+
+                    return true;
+                }
+            }
+
+            name = null;
+            key = null;
+
+            return false;
+        }
+
         public bool Remove(string name)
         {
-            return _keys.Remove(name);
+            if (_keys.TryGetValue(name, out var entry))
+            {
+                entry.Value.Dispose();
+
+                return _keys.Remove(name);
+            }
+
+            return false;
         }
 
         private void SetKey(string name, AesGcm aesGcm)
@@ -114,6 +154,11 @@ namespace Akari.Prototype.Server.Services
 
         public void Dispose()
         {
+            foreach (var entry in _keys.Values)
+            {
+                entry.Value.Dispose();
+            }
+
             _keys.Clear();
         }
     }