119 lines
4.5 KiB
C#
119 lines
4.5 KiB
C#
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());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |