Use EFCore to store preferences (#506)

* Use EFcore to store preferences * Fixed nullabilty warnings
2020-01-15 15:10:18 +01:00
parent 1856cb079c
commit c4ea6e53e8
12 changed files with 473 additions and 359 deletions
--- a/Content.Server.Database/PrefsDb.cs
+++ b/Content.Server.Database/PrefsDb.cs
@@ -0,0 +1,60 @@
+using System.Linq;
+using Microsoft.EntityFrameworkCore;
+
+namespace Content.Server.Database
+{
+    public class PrefsDb
+    {
+        private readonly PreferencesDbContext _prefsCtx;
+
+        public PrefsDb(string dbPath)
+        {
+            var optionsBuilder = new DbContextOptionsBuilder<PreferencesDbContext>();
+            optionsBuilder.UseSqlite($"Data Source={dbPath}");
+
+            _prefsCtx = new PreferencesDbContext(optionsBuilder.Options);
+            _prefsCtx.Database.Migrate();
+        }
+
+        public Prefs GetPlayerPreferences(string username)
+        {
+            return _prefsCtx.Preferences.SingleOrDefault(p => p.Username == username);
+        }
+
+        public void SaveSelectedCharacterIndex(string username, int slot)
+        {
+            var prefs = _prefsCtx.Preferences.SingleOrDefault(p => p.Username == username);
+            if (prefs is null)
+                _prefsCtx.Preferences.Add(new Prefs
+                {
+                    Username = username,
+                    SelectedCharacterSlot = slot
+                });
+            else
+                prefs.SelectedCharacterSlot = slot;
+            _prefsCtx.SaveChanges();
+        }
+
+        public void SaveCharacterSlot(string username, HumanoidProfile newProfile)
+        {
+            var prefs = _prefsCtx
+                .Preferences
+                .Single(p => p.Username == username);
+            var oldProfile = prefs
+                .HumanoidProfiles
+                .SingleOrDefault(h => h.Slot == newProfile.Slot);
+            if (!(oldProfile is null)) prefs.HumanoidProfiles.Remove(oldProfile);
+            prefs.HumanoidProfiles.Add(newProfile);
+        }
+
+        public void DeleteCharacterSlot(string username, int slot)
+        {
+            var profile = _prefsCtx
+                .Preferences
+                .Single(p => p.Username == username)
+                .HumanoidProfiles
+                .RemoveAll(h => h.Slot == slot);
+            _prefsCtx.SaveChanges();
+        }
+    }
+}