geekbot/src/Bot/Handlers/StatsHandler.cs

using System;
using System.Threading.Tasks;
using Discord.WebSocket;
using Geekbot.Core.Database;
using Geekbot.Core.Database.Models;
using Geekbot.Core.Extensions;
using Geekbot.Core.Highscores;
using Geekbot.Core.Logger;
using Microsoft.EntityFrameworkCore;

namespace Geekbot.Bot.Handlers
{
    public class StatsHandler
    {
        private readonly IGeekbotLogger _logger;
        private readonly DatabaseContext _database;
        private string _season;

        public StatsHandler(IGeekbotLogger logger, DatabaseContext database)
        {
            _logger = logger;
            _database = database;
            _season = SeasonsUtils.GetCurrentSeason();

            var timer = new System.Timers.Timer()
            {
                Enabled = true,
                AutoReset = true,
                Interval = TimeSpan.FromMinutes(5).TotalMilliseconds
            };
            timer.Elapsed += (sender, args) =>
            {
                var current = SeasonsUtils.GetCurrentSeason();
                if (current == _season) return;
                _season = SeasonsUtils.GetCurrentSeason();
            };
        }
        
        public async Task UpdateStats(SocketMessage message)
        {
            try
            {
                if (message == null) return;
                if (message.Channel.Name.StartsWith('@'))
                {
                    _logger.Information(LogSource.Message, $"[DM-Channel] {message.Content}", SimpleConextConverter.ConvertSocketMessage(message, true));
                    return;
                }

                var channel = (SocketGuildChannel) message.Channel;

                // ignore the discord bots server
                // ToDo: create a clean solution for this...
                if (channel.Guild.Id == 110373943822540800)
                {
                    return;
                }
                
                await UpdateTotalTable(message, channel);
                await UpdateSeasonsTable(message, channel);
                

                if (message.Author.IsBot) return;
                _logger.Information(LogSource.Message, message.Content, SimpleConextConverter.ConvertSocketMessage(message));
            }
            catch (Exception e)
            {
                _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();
            }
        }
    }
}