Killed history

NadekoBot.Core/Modules/Gambling/Services/GamblingService.cs

@@ -0,0 +1,162 @@
+using Discord.WebSocket;
+using NadekoBot.Core.Modules.Gambling.Common;
+using NadekoBot.Core.Services;
+using NadekoBot.Modules.Gambling.Common.Connect4;
+using NadekoBot.Modules.Gambling.Common.WheelOfFortune;
+using Newtonsoft.Json;
+using System;
+using System.Collections.Concurrent;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+using NadekoBot.Core.Modules.Gambling.Services;
+using Serilog;
+
+namespace NadekoBot.Modules.Gambling.Services
+{
+    public class GamblingService : INService
+    {
+        private readonly DbService _db;
+        private readonly ICurrencyService _cs;
+        private readonly NadekoBot _bot;
+        private readonly DiscordSocketClient _client;
+        private readonly IDataCache _cache;
+        private readonly GamblingConfigService _gss;
+
+        public ConcurrentDictionary<(ulong, ulong), RollDuelGame> Duels { get; } = new ConcurrentDictionary<(ulong, ulong), RollDuelGame>();
+        public ConcurrentDictionary<ulong, Connect4Game> Connect4Games { get; } = new ConcurrentDictionary<ulong, Connect4Game>();
+
+        private readonly Timer _decayTimer;
+
+        public GamblingService(DbService db, NadekoBot bot, ICurrencyService cs,
+            DiscordSocketClient client, IDataCache cache, GamblingConfigService gss)
+        {
+            _db = db;
+            _cs = cs;
+            _bot = bot;
+            _client = client;
+            _cache = cache;
+            _gss = gss;
+            
+            if (_bot.Client.ShardId == 0)
+            {
+                _decayTimer = new Timer(_ =>
+                {
+                    var config = _gss.Data;
+                    var maxDecay = config.Decay.MaxDecay;
+                    if (config.Decay.Percent <= 0 || config.Decay.Percent > 1 || maxDecay < 0)
+                        return;
+
+                    using (var uow = _db.GetDbContext())
+                    {
+                        var lastCurrencyDecay = _cache.GetLastCurrencyDecay();
+                        
+                        if (DateTime.UtcNow - lastCurrencyDecay < TimeSpan.FromHours(config.Decay.HourInterval))
+                           return;
+                        
+                         Log.Information($"Decaying users' currency - decay: {config.Decay.Percent * 100}% " +
+                                   $"| max: {maxDecay} " +
+                                   $"| threshold: {config.Decay.MinThreshold}");
+                         
+                         if (maxDecay == 0)
+                             maxDecay = int.MaxValue;
+                         
+                        uow._context.Database.ExecuteSqlInterpolated($@"
+UPDATE DiscordUser
+SET CurrencyAmount=
+    CASE WHEN
+    {maxDecay} > ROUND(CurrencyAmount * {config.Decay.Percent} - 0.5)
+    THEN
+    CurrencyAmount - ROUND(CurrencyAmount * {config.Decay.Percent} - 0.5)
+    ELSE
+    CurrencyAmount - {maxDecay}
+    END
+WHERE CurrencyAmount > {config.Decay.MinThreshold} AND UserId!={_client.CurrentUser.Id};");
+
+                        _cache.SetLastCurrencyDecay();
+                        uow.SaveChanges();
+                    }
+                }, null, TimeSpan.FromMinutes(5), TimeSpan.FromMinutes(5));
+            }
+
+            //using (var uow = _db.UnitOfWork)
+            //{
+            //    //refund all of the currency users had at stake in gambling games
+            //    //at the time bot was restarted
+
+            //    var stakes = uow._context.Set<Stake>()
+            //        .ToArray();
+
+            //    var userIds = stakes.Select(x => x.UserId).ToArray();
+            //    var reasons = stakes.Select(x => "Stake-" + x.Source).ToArray();
+            //    var amounts = stakes.Select(x => x.Amount).ToArray();
+
+            //    _cs.AddBulkAsync(userIds, reasons, amounts, gamble: true).ConfigureAwait(false);
+
+            //    foreach (var s in stakes)
+            //    {
+            //        _cs.AddAsync(s.UserId, "Stake-" + s.Source, s.Amount, gamble: true)
+            //            .GetAwaiter()
+            //            .GetResult();
+            //    }
+
+            //    uow._context.Set<Stake>().RemoveRange(stakes);
+            //    uow.Complete();
+            //    Log.Information("Refunded {0} users' stakes.", stakes.Length);
+            //}
+        }
+
+        public struct EconomyResult
+        {
+            public decimal Cash { get; set; }
+            public decimal Planted { get; set; }
+            public decimal Waifus { get; set; }
+            public decimal OnePercent { get; set; }
+            public long Bot { get; set; }
+        }
+
+        public EconomyResult GetEconomy()
+        {
+            if (_cache.TryGetEconomy(out var data))
+            {
+                try
+                {
+                    return JsonConvert.DeserializeObject<EconomyResult>(data);
+                }
+                catch { }
+            }
+
+            decimal cash;
+            decimal onePercent;
+            decimal planted;
+            decimal waifus;
+            long bot;
+
+            using (var uow = _db.GetDbContext())
+            {
+                cash = uow.DiscordUsers.GetTotalCurrency();
+                onePercent = uow.DiscordUsers.GetTopOnePercentCurrency(_client.CurrentUser.Id);
+                planted = uow.PlantedCurrency.GetTotalPlanted();
+                waifus = uow.Waifus.GetTotalValue();
+                bot = uow.DiscordUsers.GetUserCurrency(_client.CurrentUser.Id);
+            }
+
+            var result = new EconomyResult
+            {
+                Cash = cash,
+                Planted = planted,
+                Bot = bot,
+                Waifus = waifus,
+                OnePercent = onePercent,
+            };
+
+            _cache.SetEconomy(JsonConvert.SerializeObject(result));
+            return result;
+        }
+
+        public Task<WheelOfFortuneGame.Result> WheelOfFortuneSpinAsync(ulong userId, long bet)
+        {
+            return new WheelOfFortuneGame(userId, bet, _gss.Data, _cs).SpinAsync();
+        }
+    }
+}