diff options
Diffstat (limited to 'back-end/src/Model/UserSettingsDbStore.cs')
-rw-r--r-- | back-end/src/Model/UserSettingsDbStore.cs | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/back-end/src/Model/UserSettingsDbStore.cs b/back-end/src/Model/UserSettingsDbStore.cs new file mode 100644 index 0000000..aa44fa2 --- /dev/null +++ b/back-end/src/Model/UserSettingsDbStore.cs @@ -0,0 +1,77 @@ +// Copyright (C) 2024 Vaughn Nugent +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as +// published by the Free Software Foundation, either version 3 of the +// License, or (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program. If not, see <https://www.gnu.org/licenses/>. + +using System; +using System.Threading; +using System.Threading.Tasks; + +using Microsoft.EntityFrameworkCore; + +using VNLib.Utils; +using VNLib.Plugins.Extensions.Loading; + +namespace SimpleBookmark.Model +{ + internal sealed class UserSettingsDbStore(IAsyncLazy<DbContextOptions> dbOptions) + { + + public async Task<UserSettingsEntry?> GetSettingsForUserAsync(string userId, CancellationToken cancellation) + { + ArgumentNullException.ThrowIfNull(userId); + + //Init new db connection + await using SimpleBookmarkContext context = new(dbOptions.Value); + await context.OpenTransactionAsync(cancellation); + + UserSettingsEntry? settings = await context.BmSettings.FirstOrDefaultAsync(p => p.UserId == userId, cancellation); + + //Close db and commit transaction + await context.SaveAndCloseAsync(true, cancellation); + + return settings; + } + + public async Task<ERRNO> SetSettingsForUser(string userId, UserSettingsEntry settings, CancellationToken cancellation) + { + ArgumentNullException.ThrowIfNull(userId); + ArgumentNullException.ThrowIfNull(settings); + + //Init new db connection + await using SimpleBookmarkContext context = new(dbOptions.Value); + await context.OpenTransactionAsync(cancellation); + + //Search for existing settings entry + UserSettingsEntry? existing = await context.BmSettings.FirstOrDefaultAsync(p => p.UserId == userId, cancellation); + + if (existing is null) + { + //Add a new entry + settings.UserId = userId; + settings.LastModified = DateTime.UtcNow; + context.Add(settings); + } + else + { + //Update existing entry + existing.SettingsData = settings.SettingsData; + existing.LastModified = DateTime.UtcNow; + context.Update(existing); + } + + //Close db and commit transaction + return await context.SaveAndCloseAsync(true, cancellation); + } + } +} |