Restructured the project structure back to the way it was, there's no reasonable way to split the modules

2025-09-12 02:08:27 -04:00 · 2024-04-26 22:26:24 +00:00
parent 6c9c8bf63e
commit e0819f760c
768 changed files with 192 additions and 1047 deletions
--- a/src/NadekoBot/_common/Services/Currency/DefaultWallet.cs
+++ b/src/NadekoBot/_common/Services/Currency/DefaultWallet.cs
@@ -0,0 +1,115 @@
+using LinqToDB;
+using LinqToDB.EntityFrameworkCore;
+using NadekoBot.Db.Models;
+using Nadeko.Bot.Db.Models;
+
+namespace NadekoBot.Services.Currency;
+
+public class DefaultWallet : IWallet
+{
+    private readonly DbService _db;
+    public ulong UserId { get; }
+
+    public DefaultWallet(ulong userId, DbService db)
+    {
+        UserId = userId;
+        _db = db;
+    }
+
+    public async Task<long> GetBalance()
+    {
+        await using var ctx = _db.GetDbContext();
+        var userId = UserId;
+        return await ctx
+            .GetTable<DiscordUser>()
+            .Where(x => x.UserId == userId)
+            .Select(x => x.CurrencyAmount)
+            .FirstOrDefaultAsync();
+    }
+
+    public async Task<bool> Take(long amount, TxData? txData)
+    {
+        if (amount < 0)
+            throw new ArgumentOutOfRangeException(nameof(amount), "Amount to take must be non negative.");
+
+        await using var ctx = _db.GetDbContext();
+
+        var userId = UserId;
+        var changed = await ctx
+            .GetTable<DiscordUser>()
+            .Where(x => x.UserId == userId && x.CurrencyAmount >= amount)
+            .UpdateAsync(x => new()
+            {
+                CurrencyAmount = x.CurrencyAmount - amount
+            });
+
+        if (changed == 0)
+            return false;
+
+        if (txData is not null)
+        {
+            await ctx
+                .GetTable<CurrencyTransaction>()
+                .InsertAsync(() => new()
+                {
+                    Amount = -amount,
+                    Note = txData.Note,
+                    UserId = userId,
+                    Type = txData.Type,
+                    Extra = txData.Extra,
+                    OtherId = txData.OtherId,
+                    DateAdded = DateTime.UtcNow
+                });
+        }
+
+        return true;
+    }
+
+    public async Task Add(long amount, TxData? txData)
+    {
+        if (amount <= 0)
+            throw new ArgumentOutOfRangeException(nameof(amount), "Amount must be greater than 0.");
+
+        await using var ctx = _db.GetDbContext();
+        var userId = UserId;
+
+        await using (var tran = await ctx.Database.BeginTransactionAsync())
+        {
+            var changed = await ctx
+                .GetTable<DiscordUser>()
+                .Where(x => x.UserId == userId)
+                .UpdateAsync(x => new()
+                {
+                    CurrencyAmount = x.CurrencyAmount + amount
+                });
+
+            if (changed == 0)
+            {
+                await ctx
+                    .GetTable<DiscordUser>()
+                    .Value(x => x.UserId, userId)
+                    .Value(x => x.Username, "Unknown")
+                    .Value(x => x.Discriminator, "????")
+                    .Value(x => x.CurrencyAmount, amount)
+                    .InsertAsync();
+            }
+
+            await tran.CommitAsync();
+        }
+
+        if (txData is not null)
+        {
+            await ctx.GetTable<CurrencyTransaction>()
+                .InsertAsync(() => new()
+                {
+                    Amount = amount,
+                    UserId = userId,
+                    Note = txData.Note,
+                    Type = txData.Type,
+                    Extra = txData.Extra,
+                    OtherId = txData.OtherId,
+                    DateAdded = DateTime.UtcNow
+                });
+        }
+    }
+}