daan/geekbot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Port !karma

Browse source
This commit is contained in:
runebaas 2018-05-10 02:33:10 +02:00
parent 15034d63a3
commit 5117e7609f
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
2 changed files with 58 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

82
Geekbot.net/Commands/User/Karma.cs
View file

@ -1,23 +1,26 @@
using System; using System;
using System.Linq;
using System.Threading.Tasks; using System.Threading.Tasks;
using Discord; using Discord;
using Discord.Commands; using Discord.Commands;
using Geekbot.net.Database;
using Geekbot.net.Database.Models;
using Geekbot.net.Lib; using Geekbot.net.Lib;
using Geekbot.net.Lib.ErrorHandling; using Geekbot.net.Lib.ErrorHandling;
using Geekbot.net.Lib.Extensions;
using Geekbot.net.Lib.Localization; using Geekbot.net.Lib.Localization;
using StackExchange.Redis;
namespace Geekbot.net.Commands.User namespace Geekbot.net.Commands.User
{ {
public class Karma : ModuleBase public class Karma : ModuleBase
{ {
private readonly IErrorHandler _errorHandler; private readonly IErrorHandler _errorHandler;
private readonly IDatabase _redis; private readonly DatabaseContext _database;
private readonly ITranslationHandler _translation; private readonly ITranslationHandler _translation;
public Karma(IDatabase redis, IErrorHandler errorHandler, ITranslationHandler translation) public Karma(DatabaseContext database, IErrorHandler errorHandler, ITranslationHandler translation)
{ {
_redis = redis; _database = database;
_errorHandler = errorHandler; _errorHandler = errorHandler;
_translation = translation; _translation = translation;
} }
@ -30,22 +33,26 @@ namespace Geekbot.net.Commands.User
try try
{ {
var transDict = _translation.GetDict(Context); var transDict = _translation.GetDict(Context);
var lastKarmaFromRedis = _redis.HashGet($"{Context.Guild.Id}:KarmaTimeout", Context.User.Id.ToString()); var actor = GetUser(Context.User.Id);
var lastKarma = ConvertToDateTimeOffset(lastKarmaFromRedis.ToString());
if (user.Id == Context.User.Id) if (user.Id == Context.User.Id)
{ {
await ReplyAsync(string.Format(transDict["CannotChangeOwn"], Context.User.Username)); await ReplyAsync(string.Format(transDict["CannotChangeOwn"], Context.User.Username));
} }
else if (TimeoutFinished(lastKarma)) else if (TimeoutFinished(actor.TimeOut))
{ {
await ReplyAsync(string.Format(transDict["WaitUntill"], Context.User.Username, await ReplyAsync(string.Format(transDict["WaitUntill"], Context.User.Username,
GetTimeLeft(lastKarma))); GetTimeLeft(actor.TimeOut)));
} }
else else
{ {
var newKarma = _redis.HashIncrement($"{Context.Guild.Id}:Karma", user.Id.ToString()); var target = GetUser(user.Id);
_redis.HashSet($"{Context.Guild.Id}:KarmaTimeout", target.Karma = target.Karma + 1;
new[] {new HashEntry(Context.User.Id.ToString(), DateTimeOffset.Now.ToString("u"))}); SetUser(target);
actor.TimeOut = DateTimeOffset.Now;
SetUser(actor);
_database.SaveChanges();
var eb = new EmbedBuilder(); var eb = new EmbedBuilder();
eb.WithAuthor(new EmbedAuthorBuilder() eb.WithAuthor(new EmbedAuthorBuilder()
@ -56,7 +63,7 @@ namespace Geekbot.net.Commands.User
eb.Title = transDict["Increased"]; eb.Title = transDict["Increased"];
eb.AddInlineField(transDict["By"], Context.User.Username); eb.AddInlineField(transDict["By"], Context.User.Username);
eb.AddInlineField(transDict["Amount"], "+1"); eb.AddInlineField(transDict["Amount"], "+1");
eb.AddInlineField(transDict["Current"], newKarma); eb.AddInlineField(transDict["Current"], target.Karma);
await ReplyAsync("", false, eb.Build()); await ReplyAsync("", false, eb.Build());
} }
} }
@ -74,22 +81,26 @@ namespace Geekbot.net.Commands.User
try try
{ {
var transDict = _translation.GetDict(Context); var transDict = _translation.GetDict(Context);
var lastKarmaFromRedis = _redis.HashGet($"{Context.Guild.Id}:KarmaTimeout", Context.User.Id.ToString()); var actor = GetUser(Context.User.Id);
var lastKarma = ConvertToDateTimeOffset(lastKarmaFromRedis.ToString());
if (user.Id == Context.User.Id) if (user.Id == Context.User.Id)
{ {
await ReplyAsync(string.Format(transDict["CannotChangeOwn"], Context.User.Username)); await ReplyAsync(string.Format(transDict["CannotChangeOwn"], Context.User.Username));
} }
else if (TimeoutFinished(lastKarma)) else if (TimeoutFinished(actor.TimeOut))
{ {
await ReplyAsync(string.Format(transDict["WaitUntill"], Context.User.Username, await ReplyAsync(string.Format(transDict["WaitUntill"], Context.User.Username,
GetTimeLeft(lastKarma))); GetTimeLeft(actor.TimeOut)));
} }
else else
{ {
var newKarma = _redis.HashDecrement($"{Context.Guild.Id}:Karma", user.Id.ToString()); var target = GetUser(user.Id);
_redis.HashSet($"{Context.Guild.Id}:KarmaTimeout", target.Karma = target.Karma - 1;
new[] {new HashEntry(Context.User.Id.ToString(), DateTimeOffset.Now.ToString())}); SetUser(target);
actor.TimeOut = DateTimeOffset.Now;
SetUser(actor);
_database.SaveChanges();
var eb = new EmbedBuilder(); var eb = new EmbedBuilder();
eb.WithAuthor(new EmbedAuthorBuilder() eb.WithAuthor(new EmbedAuthorBuilder()
@ -100,7 +111,7 @@ namespace Geekbot.net.Commands.User
eb.Title = transDict["Decreased"]; eb.Title = transDict["Decreased"];
eb.AddInlineField(transDict["By"], Context.User.Username); eb.AddInlineField(transDict["By"], Context.User.Username);
eb.AddInlineField(transDict["Amount"], "-1"); eb.AddInlineField(transDict["Amount"], "-1");
eb.AddInlineField(transDict["Current"], newKarma); eb.AddInlineField(transDict["Current"], target.Karma);
await ReplyAsync("", false, eb.Build()); await ReplyAsync("", false, eb.Build());
} }
} }
@ -110,11 +121,6 @@ namespace Geekbot.net.Commands.User
} }
} }
private DateTimeOffset ConvertToDateTimeOffset(string dateTimeOffsetString)
{
return string.IsNullOrEmpty(dateTimeOffsetString) ? DateTimeOffset.Now.Subtract(new TimeSpan(7, 18, 0, 0)) : DateTimeOffset.Parse(dateTimeOffsetString);
}
private bool TimeoutFinished(DateTimeOffset lastKarma) private bool TimeoutFinished(DateTimeOffset lastKarma)
{ {
return lastKarma.AddMinutes(3) > DateTimeOffset.Now; return lastKarma.AddMinutes(3) > DateTimeOffset.Now;
@ -125,5 +131,31 @@ namespace Geekbot.net.Commands.User
var dt = lastKarma.AddMinutes(3).Subtract(DateTimeOffset.Now); var dt = lastKarma.AddMinutes(3).Subtract(DateTimeOffset.Now);
return $"{dt.Minutes} Minutes and {dt.Seconds} Seconds"; return $"{dt.Minutes} Minutes and {dt.Seconds} Seconds";
} }
private KarmaModel GetUser(ulong userId)
{
var user = _database.Karma.FirstOrDefault(u =>u.GuildId.Equals(Context.Guild.Id.AsLong()) && u.UserId.Equals(userId.AsLong())) ?? CreateNewRow(userId);
return user;
}
private bool SetUser(KarmaModel user)
{
_database.Karma.Update(user);
return true;
}
private KarmaModel CreateNewRow(ulong userId)
{
var user = new KarmaModel()
{
GuildId = Context.Guild.Id.AsLong(),
UserId = userId.AsLong(),
Karma = 0,
TimeOut = DateTimeOffset.MinValue
};
var newUser = _database.Karma.Add(user).Entity;
_database.SaveChanges();
return newUser;
}
} }
} }

2
Geekbot.net/Geekbot.net.csproj
View file

@ -3,7 +3,7 @@
<OutputType>Exe</OutputType> <OutputType>Exe</OutputType>
<TargetFramework>netcoreapp2.0</TargetFramework> <TargetFramework>netcoreapp2.0</TargetFramework>
<ApplicationIcon>derp.ico</ApplicationIcon> <ApplicationIcon>derp.ico</ApplicationIcon>
<Version>3.7.0</Version> <Version>4.0.0</Version>
<VersionSuffix>$(VersionSuffix)</VersionSuffix> <VersionSuffix>$(VersionSuffix)</VersionSuffix>
<Version Condition=" '$(VersionSuffix)' != '' ">$(Version)-$(VersionSuffix)</Version> <Version Condition=" '$(VersionSuffix)' != '' ">$(Version)-$(VersionSuffix)</Version>
<Company>Pizza and Coffee Studios</Company> <Company>Pizza and Coffee Studios</Company>