Added Birthday handling

2026-04-03 11:10:05 +00:00 · 2018-10-08 18:36:22 +02:00
parent b331bf5223
commit 599d24ab3f
3 changed files with 150 additions and 1 deletions
--- a/DeukBot4/Database/BirthdayHandler.cs
+++ b/DeukBot4/Database/BirthdayHandler.cs
@@ -0,0 +1,119 @@
+using System;
+using System.Threading.Tasks;
+using DeukBot4.MessageHandlers.CommandHandler;
+using StackExchange.Redis;
+
+namespace DeukBot4.Database
+{
+    public static class BirthdayHandler
+    {
+        private static ConnectionMultiplexer Redis => ReminderHandler.Redis;
+
+        public static async Task AddBirthday(DateTime birthday, ulong server, ulong user)
+        {
+            try
+            {
+                var db           = Redis.GetDatabase();
+                var exists = db.HashGet("birthdays", (RedisValue) $"{server}-{user}");
+                if (exists.HasValue)
+                {
+                    var bdayEntries = db.HashGetAll(exists.ToString());
+                    DateTime bday = DateTime.MinValue;
+                    foreach (var bdayEntry in bdayEntries)
+                    {
+                        if (bdayEntry.Name == "birthday") bday = DateTime.FromBinary(long.Parse(bdayEntry.Value));
+                    }
+                    db.SetRemove(bday.ToString("ddMM"), exists);
+                    db.KeyDelete(exists.ToString());
+                }
+                var id = Guid.NewGuid().ToString();
+                db.HashSet("birthdays", (RedisValue) $"{server}-{user}", id);
+                db.SetAdd(birthday.ToString("ddMM"), id);
+                db.HashSetAsync((RedisKey) id, new[]
+                {
+                    new HashEntry("server", server.ToString("D")),
+                    new HashEntry("user", user.ToString("D")),
+                    new HashEntry("birthday", birthday.ToBinary().ToString("D")),
+                });
+
+            }
+            catch (Exception e)
+            {
+                Logger.Main.LogError(e);
+            }
+        }
+
+        private static int Years(DateTime start, DateTime end)
+        {
+            return (end.Year - start.Year - 1) +
+                   (((end.Month > start.Month) ||
+                     ((end.Month == start.Month) && (end.Day >= start.Day))) ? 1 : 0);
+        }
+
+        private static DateTime _lastCheckedDate;
+        // ReSharper disable once FunctionRecursiveOnAllPaths
+        public static async Task CheckBirthdays()
+        {
+            var today = DateTime.UtcNow.Date;
+            while (!Program.IsConnected)
+            {
+                await Task.Delay(500);
+            }
+            try
+            {
+                if (today != _lastCheckedDate)
+                {
+                    _lastCheckedDate = today;
+                    await ActualBirthdayCheck(today);
+                }
+            }
+            catch (Exception e)
+            {
+                Logger.Main.LogError(e);
+            }
+            await Task.Delay(TimeSpan.FromSeconds(1));
+            await CheckBirthdays();
+        }
+
+
+        private static async Task ActualBirthdayCheck(DateTime today)
+        {
+            var db        = Redis.GetDatabase();
+            var reminders = db.SetMembers(today.ToString("ddMM"));
+            foreach (var redisValue in reminders)
+            {
+                ulong    server   = 0;
+                ulong    user     = 0;
+                DateTime birthday = today;
+                var      val      = redisValue.ToString();
+                var      data     = db.HashGetAll(val);
+                foreach (var hashEntry in data)
+                {
+                    if (hashEntry.Name      == "server") server     = ulong.Parse(hashEntry.Value);
+                    else if (hashEntry.Name == "user") user         = ulong.Parse(hashEntry.Value);
+                    else if (hashEntry.Name == "birthday") birthday = DateTime.FromBinary(long.Parse(hashEntry.Value));
+                }
+
+                var guild = Program.Client.GetGuild(server);
+                if (guild != null)
+                {
+                    var guildUser = guild.GetUser(user);
+                    if (guildUser == null)
+                    {
+                        Console.WriteLine("Can't celebrate birthday for user, not in guild anymore :(");
+                        db.KeyDelete(val);
+                        return;
+                    }
+
+                    var eb = EmbedFactory.GetStandardEmbedBuilder();
+                    eb.Title = $":confetti_ball::confetti_ball: Happy Birthday! :confetti_ball::confetti_ball: ";
+                    var age = Years(birthday, today);
+                    eb.Description  = $"It's <@!{guildUser.Id}>'s {age}th birthday!! Have a good one!";
+                    eb.ThumbnailUrl = guildUser.GetAvatarUrl();
+
+                    guild.DefaultChannel.SendMessageAsync("", embed: eb.Build());
+                }
+            }
+        }
+    }
+}