From 29c0def713fb668cdbd1d3bc7a74d94e0738a23c Mon Sep 17 00:00:00 2001
From: runebaas <runebaas@gmail.com>
Date: Tue, 29 Dec 2020 22:33:19 +0100
Subject: [PATCH] Start counting messages per quarter starting 1 january 2021

---
 src/Bot/Handlers/StatsHandler.cs              | 88 +++++++++++++++----
 src/Core/Database/DatabaseContext.cs          |  1 +
 .../Database/Models/MessageSeasonsModel.cs    | 21 +++++
 3 files changed, 95 insertions(+), 15 deletions(-)
 create mode 100644 src/Core/Database/Models/MessageSeasonsModel.cs

diff --git a/src/Bot/Handlers/StatsHandler.cs b/src/Bot/Handlers/StatsHandler.cs
index 197bbb8..e5962a9 100644
--- a/src/Bot/Handlers/StatsHandler.cs
+++ b/src/Bot/Handlers/StatsHandler.cs
@@ -1,5 +1,7 @@
 using System;
+using System.Globalization;
 using System.Threading.Tasks;
+using System.Timers;
 using Discord.WebSocket;
 using Geekbot.Core.Database;
 using Geekbot.Core.Database.Models;
@@ -13,11 +15,28 @@ namespace Geekbot.Bot.Handlers
     {
         private readonly IGeekbotLogger _logger;
         private readonly DatabaseContext _database;
+        private string _season;
+        // ToDo: Clean up in 2021
+        private bool _seasonsStarted;
 
         public StatsHandler(IGeekbotLogger logger, DatabaseContext database)
         {
             _logger = logger;
             _database = database;
+            _season = GetSeason();
+            _seasonsStarted = DateTime.Now.Year == 2021;
+            
+            var timer = new Timer()
+            {
+                Enabled = true,
+                AutoReset = true,
+                Interval = TimeSpan.FromHours(1).TotalMilliseconds
+            };
+            timer.Elapsed += (sender, args) =>
+            {
+                _season = GetSeason();
+                _seasonsStarted = DateTime.Now.Year == 2021;
+            };
         }
         
         public async Task UpdateStats(SocketMessage message)
@@ -32,23 +51,12 @@ namespace Geekbot.Bot.Handlers
                 }
 
                 var channel = (SocketGuildChannel) message.Channel;
-
-                var rowId = await _database.Database.ExecuteSqlRawAsync(
-                    "UPDATE \"Messages\" SET \"MessageCount\" = \"MessageCount\" + 1 WHERE \"GuildId\" = {0} AND \"UserId\" = {1}",
-                    channel.Guild.Id.AsLong(),
-                    message.Author.Id.AsLong()
-                );
-
-                if (rowId == 0)
+                await UpdateTotalTable(message, channel);  
+                if (_seasonsStarted)
                 {
-                    await _database.Messages.AddAsync(new MessagesModel
-                    {
-                        UserId = message.Author.Id.AsLong(),
-                        GuildId = channel.Guild.Id.AsLong(),
-                        MessageCount = 1
-                    });
-                    await _database.SaveChangesAsync();
+                    await UpdateSeasonsTable(message, channel);
                 }
+                
 
                 if (message.Author.IsBot) return;
                 _logger.Information(LogSource.Message, message.Content, SimpleConextConverter.ConvertSocketMessage(message));
@@ -58,5 +66,55 @@ namespace Geekbot.Bot.Handlers
                 _logger.Error(LogSource.Message, "Could not process message stats", e);
             }
         }
+
+        private async Task UpdateTotalTable(SocketMessage message, SocketGuildChannel channel)
+        {
+            var rowId = await _database.Database.ExecuteSqlRawAsync(
+                "UPDATE \"Messages\" SET \"MessageCount\" = \"MessageCount\" + 1 WHERE \"GuildId\" = {0} AND \"UserId\" = {1}",
+                channel.Guild.Id.AsLong(),
+                message.Author.Id.AsLong()
+            );
+
+            if (rowId == 0)
+            {
+                await _database.Messages.AddAsync(new MessagesModel
+                {
+                    UserId = message.Author.Id.AsLong(),
+                    GuildId = channel.Guild.Id.AsLong(),
+                    MessageCount = 1
+                });
+                await _database.SaveChangesAsync();
+            }
+        }
+
+        private async Task UpdateSeasonsTable(SocketMessage message, SocketGuildChannel channel)
+        {
+            var rowId = await _database.Database.ExecuteSqlRawAsync(
+                "UPDATE \"MessagesSeasons\" SET \"MessageCount\" = \"MessageCount\" + 1 WHERE \"GuildId\" = {0} AND \"UserId\" = {1} AND \"Season\" = {2}",
+                channel.Guild.Id.AsLong(),
+                message.Author.Id.AsLong(),
+                _season
+            );
+
+            if (rowId == 0)
+            {
+                await _database.MessagesSeasons.AddAsync(new MessageSeasonsModel()
+                {
+                    UserId = message.Author.Id.AsLong(),
+                    GuildId = channel.Guild.Id.AsLong(),
+                    Season = _season,
+                    MessageCount = 1
+                });
+                await _database.SaveChangesAsync();
+            }
+        }
+
+        private static string GetSeason()
+        {
+            var now = DateTime.Now;
+            var year = (now.Year - 2000).ToString(CultureInfo.InvariantCulture);
+            var quarter = Math.Ceiling(now.Month / 3.0).ToString(CultureInfo.InvariantCulture);
+            return $"{year}Q{quarter}";
+        }
     }
 }
\ No newline at end of file
diff --git a/src/Core/Database/DatabaseContext.cs b/src/Core/Database/DatabaseContext.cs
index 93d46d3..14a1275 100644
--- a/src/Core/Database/DatabaseContext.cs
+++ b/src/Core/Database/DatabaseContext.cs
@@ -11,6 +11,7 @@ namespace Geekbot.Core.Database
         public DbSet<KarmaModel> Karma { get; set; }
         public DbSet<ShipsModel> Ships { get; set; }
         public DbSet<RollsModel> Rolls { get; set; }
+        public DbSet<MessageSeasonsModel> MessagesSeasons { get; set; }
         public DbSet<MessagesModel> Messages { get; set; }
         public DbSet<SlapsModel> Slaps { get; set; }
         public DbSet<GlobalsModel> Globals { get; set; }
diff --git a/src/Core/Database/Models/MessageSeasonsModel.cs b/src/Core/Database/Models/MessageSeasonsModel.cs
new file mode 100644
index 0000000..5a4252c
--- /dev/null
+++ b/src/Core/Database/Models/MessageSeasonsModel.cs
@@ -0,0 +1,21 @@
+using System.ComponentModel.DataAnnotations;
+
+namespace Geekbot.Core.Database.Models
+{
+    public class MessageSeasonsModel
+    {
+        [Key]
+        public int Id { get; set; }
+        
+        [Required]
+        public long GuildId { get; set; }
+        
+        [Required]
+        public long UserId { get; set; }
+        
+        [Required]
+        public string Season { get; set; }
+        
+        public int MessageCount { get; set; }
+    }
+}
\ No newline at end of file