Usings and namespace cleanup

This commit is contained in:
jetsparrow 2024-11-15 19:19:40 +03:00
parent d2e8ed4d69
commit 2cc286ad9c
29 changed files with 1415 additions and 1530 deletions

View File

@ -1,162 +1,156 @@
using System;
using System.Collections.Generic;
using System.Linq;
using Perfusion;
using JetKarmaBot.Services.Handling;
using NLog;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore;
using NLog;
using JetKarmaBot.Services.Handling;
using JetKarmaBot.Models; using JetKarmaBot.Models;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
class AwardCommand : IChatCommand
{ {
class AwardCommand : IChatCommand public IReadOnlyCollection<string> Names => new[] { "award", "revoke" };
[Inject] private Logger log;
public async Task<bool> Execute(RequestContext ctx)
{ {
public IReadOnlyCollection<string> Names => new[] { "award", "revoke" }; var db = ctx.GetFeature<KarmaContext>();
[Inject] private Logger log; var currentLocale = ctx.GetFeature<Locale>();
public async Task<bool> Execute(RequestContext ctx) var awarder = ctx.EventArgs.Message.From;
if (Timeout.TimeoutCache[awarder.Id].PreviousAwardDate.AddSeconds(Config.Timeout.AwardTimeSeconds) > DateTime.Now)
{ {
var db = ctx.GetFeature<KarmaContext>(); ctx.GetFeature<TimeoutManager.Feature>().Multiplier = 0; // Doesn't count as success or failure
var currentLocale = ctx.GetFeature<Locale>(); if (!Timeout.TimeoutCache[awarder.Id].TimeoutMessaged)
await ctx.SendMessage(currentLocale["jetkarmabot.ratelimit"]);
var awarder = ctx.EventArgs.Message.From; Timeout.TimeoutCache[awarder.Id].TimeoutMessaged = true;
return false;
if (Timeout.TimeoutCache[awarder.Id].PreviousAwardDate.AddSeconds(Config.Timeout.AwardTimeSeconds) > DateTime.Now)
{
ctx.GetFeature<TimeoutManager.Feature>().Multiplier = 0; // Doesn't count as success or failure
if (!Timeout.TimeoutCache[awarder.Id].TimeoutMessaged)
await ctx.SendMessage(currentLocale["jetkarmabot.ratelimit"]);
Timeout.TimeoutCache[awarder.Id].TimeoutMessaged = true;
return false;
}
string awardTypeText = null;
long recipientId = default;
foreach (string arg in ctx.Command.Parameters)
{
if (arg.StartsWith('@'))
{
if (recipientId != default(int))
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errdup"]);
return false;
}
recipientId = await db.Users.Where(x => x.Username == arg).Select(x => x.UserId).FirstOrDefaultAsync();
if (recipientId == default(int))
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errbadusername"]);
return false;
}
}
else
{
if (awardTypeText == null)
awardTypeText = arg;
else
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errdup"]);
return false;
}
}
}
if (ctx.EventArgs.Message.ReplyToMessage != null && recipientId == default)
{
recipientId = ctx.EventArgs.Message.ReplyToMessage.From.Id;
}
if (recipientId == default(int))
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errawardnoreply"]);
return false;
}
bool awarding = ctx.Command.Command == "award";
if (awarder.Id == recipientId)
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errawardself"]);
return false;
}
if (ctx.GetFeature<ChatCommandRouter.Feature>().Router.Me.Id == recipientId)
{
await ctx.SendMessage(awarding
? currentLocale["jetkarmabot.award.errawardbot"]
: currentLocale["jetkarmabot.award.errrevokebot"]);
return false;
}
var text = ctx.EventArgs.Message.Text;
global::JetKarmaBot.Models.AwardType awardType = awardTypeText != null
? await db.AwardTypes.FirstAsync(at => at.CommandName == awardTypeText)
: await db.AwardTypes.FindAsync((sbyte)1);
var prevCount = await db.Awards
.Where(aw => aw.ToId == recipientId && aw.AwardTypeId == awardType.AwardTypeId && aw.ChatId == ctx.EventArgs.Message.Chat.Id)
.SumAsync(aw => aw.Amount);
await db.Awards.AddAsync(new Models.Award()
{
AwardTypeId = awardType.AwardTypeId,
Amount = (sbyte)(awarding ? 1 : -1),
FromId = awarder.Id,
ToId = recipientId,
ChatId = ctx.EventArgs.Message.Chat.Id
});
var recUserName = (await db.Users.FindAsync(recipientId)).Username;
log.Debug($"Awarded {(awarding ? 1 : -1)}{awardType.Symbol} to {recUserName}");
string message = awarding
? string.Format(currentLocale["jetkarmabot.award.awardmessage"], getLocalizedName(awardType, currentLocale), recUserName)
: string.Format(currentLocale["jetkarmabot.award.revokemessage"], getLocalizedName(awardType, currentLocale), recUserName);
var response = message + "\n" + String.Format(currentLocale["jetkarmabot.award.statustext"], recUserName, prevCount + (awarding ? 1 : -1), awardType.Symbol);
await ctx.SendMessage(response);
Timeout.TimeoutCache[awarder.Id].PreviousAwardDate = DateTime.Now;
return true;
} }
private string getLocalizedName(global::JetKarmaBot.Models.AwardType awardType, Locale loc) string awardTypeText = null;
long recipientId = default;
foreach (string arg in ctx.Command.Parameters)
{ {
if (loc.ContainsKey($"jetkarmabot.awardtypes.accusative.{awardType.CommandName}")) if (arg.StartsWith('@'))
{ {
return loc[$"jetkarmabot.awardtypes.accusative.{awardType.CommandName}"]; if (recipientId != default(int))
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errdup"]);
return false;
}
recipientId = await db.Users.Where(x => x.Username == arg).Select(x => x.UserId).FirstOrDefaultAsync();
if (recipientId == default(int))
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errbadusername"]);
return false;
}
} }
else else
{ {
return awardType.Name; if (awardTypeText == null)
awardTypeText = arg;
else
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errdup"]);
return false;
}
} }
} }
[Inject] Localization Locale { get; set; } if (ctx.EventArgs.Message.ReplyToMessage != null && recipientId == default)
[Inject] TimeoutManager Timeout { get; set; } {
[Inject] Config Config { get; set; } recipientId = ctx.EventArgs.Message.ReplyToMessage.From.Id;
}
public string Description => "Awards/revokes an award to a user."; if (recipientId == default(int))
public string DescriptionID => "jetkarmabot.award.help"; {
await ctx.SendMessage(currentLocale["jetkarmabot.award.errawardnoreply"]);
return false;
}
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
new ChatCommandArgument() { bool awarding = ctx.Command.Command == "award";
Name="awardtype",
Required=false, if (awarder.Id == recipientId)
Type=ChatCommandArgumentType.String, {
Description="The award to grant to/strip of the specified user", await ctx.SendMessage(currentLocale["jetkarmabot.award.errawardself"]);
DescriptionID="jetkarmabot.award.awardtypehelp" return false;
}, }
new ChatCommandArgument() {
Name="to", if (ctx.GetFeature<ChatCommandRouter.Feature>().Router.Me.Id == recipientId)
Required=false, {
Type=ChatCommandArgumentType.String, await ctx.SendMessage(awarding
Description="The user to award it to.", ? currentLocale["jetkarmabot.award.errawardbot"]
DescriptionID="jetkarmabot.award.tohelp" : currentLocale["jetkarmabot.award.errrevokebot"]);
} return false;
}; }
var text = ctx.EventArgs.Message.Text;
global::JetKarmaBot.Models.AwardType awardType = awardTypeText != null
? await db.AwardTypes.FirstAsync(at => at.CommandName == awardTypeText)
: await db.AwardTypes.FindAsync((sbyte)1);
var prevCount = await db.Awards
.Where(aw => aw.ToId == recipientId && aw.AwardTypeId == awardType.AwardTypeId && aw.ChatId == ctx.EventArgs.Message.Chat.Id)
.SumAsync(aw => aw.Amount);
await db.Awards.AddAsync(new Models.Award()
{
AwardTypeId = awardType.AwardTypeId,
Amount = (sbyte)(awarding ? 1 : -1),
FromId = awarder.Id,
ToId = recipientId,
ChatId = ctx.EventArgs.Message.Chat.Id
});
var recUserName = (await db.Users.FindAsync(recipientId)).Username;
log.Debug($"Awarded {(awarding ? 1 : -1)}{awardType.Symbol} to {recUserName}");
string message = awarding
? string.Format(currentLocale["jetkarmabot.award.awardmessage"], getLocalizedName(awardType, currentLocale), recUserName)
: string.Format(currentLocale["jetkarmabot.award.revokemessage"], getLocalizedName(awardType, currentLocale), recUserName);
var response = message + "\n" + String.Format(currentLocale["jetkarmabot.award.statustext"], recUserName, prevCount + (awarding ? 1 : -1), awardType.Symbol);
await ctx.SendMessage(response);
Timeout.TimeoutCache[awarder.Id].PreviousAwardDate = DateTime.Now;
return true;
} }
private string getLocalizedName(global::JetKarmaBot.Models.AwardType awardType, Locale loc)
{
if (loc.ContainsKey($"jetkarmabot.awardtypes.accusative.{awardType.CommandName}"))
{
return loc[$"jetkarmabot.awardtypes.accusative.{awardType.CommandName}"];
}
else
{
return awardType.Name;
}
}
[Inject] Localization Locale { get; set; }
[Inject] TimeoutManager Timeout { get; set; }
[Inject] Config Config { get; set; }
public string Description => "Awards/revokes an award to a user.";
public string DescriptionID => "jetkarmabot.award.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
new ChatCommandArgument() {
Name="awardtype",
Required=false,
Type=ChatCommandArgumentType.String,
Description="The award to grant to/strip of the specified user",
DescriptionID="jetkarmabot.award.awardtypehelp"
},
new ChatCommandArgument() {
Name="to",
Required=false,
Type=ChatCommandArgumentType.String,
Description="The user to award it to.",
DescriptionID="jetkarmabot.award.tohelp"
}
};
} }

View File

@ -1,80 +1,75 @@
using System.Collections.Generic;
using Perfusion;
using JetKarmaBot.Services.Handling;
using NLog; using NLog;
using System.Linq; using JetKarmaBot.Services.Handling;
using System.Threading.Tasks;
using JetKarmaBot.Models; using JetKarmaBot.Models;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
class LocaleCommand : IChatCommand
{ {
class LocaleCommand : IChatCommand public IReadOnlyCollection<string> Names => new[] { "changelocale", "locale" };
[Inject] private Logger log;
public async Task<bool> Execute(RequestContext ctx)
{ {
public IReadOnlyCollection<string> Names => new[] { "changelocale", "locale" }; var db = ctx.GetFeature<KarmaContext>();
[Inject] private Logger log; var currentLocale = ctx.GetFeature<Locale>();
var cmd = ctx.Command;
var args = ctx.EventArgs;
public async Task<bool> Execute(RequestContext ctx) if (cmd.Parameters.Length < 1)
{ {
var db = ctx.GetFeature<KarmaContext>(); await ctx.SendMessage(currentLocale["jetkarmabot.changelocale.getlocale"]);
var currentLocale = ctx.GetFeature<Locale>(); return false;
var cmd = ctx.Command;
var args = ctx.EventArgs;
if (cmd.Parameters.Length < 1)
{
await ctx.SendMessage(currentLocale["jetkarmabot.changelocale.getlocale"]);
return false;
}
else if (cmd.Parameters[0] == "list")
{
await ctx.SendMessage(currentLocale["jetkarmabot.changelocale.listalltext"] + "\n"
+ string.Join("\n", Locale.Select(a => a.Key)));
return false;
}
else if (cmd.Parameters[0] == "all")
{
await ctx.SendMessage(currentLocale["jetkarmabot.changelocale.errorall"]);
return false;
}
string localeId;
if (Locale.ContainsLocale(cmd.Parameters[0]))
localeId = cmd.Parameters[0];
else
try
{
localeId = Locale.FindByCommonName(cmd.Parameters[0]).Name;
}
catch (LocalizationException e)
{
await ctx.SendMessage(
currentLocale["jetkarmabot.changelocale.toomany"] + "\n"
+ string.Join("\n", (e.Data["LocaleNames"] as Locale[]).Select(x => x.Name)));
return false;
}
(await db.Chats.FindAsync(args.Message.Chat.Id)).Locale = localeId;
log.Debug($"Changed language of chat {args.Message.Chat.Id} to {localeId}");
currentLocale = Locale[localeId];
await ctx.SendMessage(
(currentLocale.HasNote ? currentLocale["jetkarmabot.changelocale.beforenote"] + currentLocale.Note + "\n" : "")
+ currentLocale["jetkarmabot.changelocale.justchanged"]);
return true;
} }
else if (cmd.Parameters[0] == "list")
[Inject] Localization Locale { get; set; } {
await ctx.SendMessage(currentLocale["jetkarmabot.changelocale.listalltext"] + "\n"
public string Description => "Switches current chat locale to [locale]"; + string.Join("\n", Locale.Select(a => a.Key)));
public string DescriptionID => "jetkarmabot.changelocale.help"; return false;
}
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] { else if (cmd.Parameters[0] == "all")
new ChatCommandArgument() { {
Name="locale", await ctx.SendMessage(currentLocale["jetkarmabot.changelocale.errorall"]);
Required=false, return false;
Type=ChatCommandArgumentType.String, }
Description="The locale to switch to. Can be \"list\" to list all possible locales. Also can be empty to get current locale.", string localeId;
DescriptionID="jetkarmabot.changelocale.localehelp" if (Locale.ContainsLocale(cmd.Parameters[0]))
localeId = cmd.Parameters[0];
else
try
{
localeId = Locale.FindByCommonName(cmd.Parameters[0]).Name;
} }
}; catch (LocalizationException e)
{
await ctx.SendMessage(
currentLocale["jetkarmabot.changelocale.toomany"] + "\n"
+ string.Join("\n", (e.Data["LocaleNames"] as Locale[]).Select(x => x.Name)));
return false;
}
(await db.Chats.FindAsync(args.Message.Chat.Id)).Locale = localeId;
log.Debug($"Changed language of chat {args.Message.Chat.Id} to {localeId}");
currentLocale = Locale[localeId];
await ctx.SendMessage(
(currentLocale.HasNote ? currentLocale["jetkarmabot.changelocale.beforenote"] + currentLocale.Note + "\n" : "")
+ currentLocale["jetkarmabot.changelocale.justchanged"]);
return true;
} }
[Inject] Localization Locale { get; set; }
public string Description => "Switches current chat locale to [locale]";
public string DescriptionID => "jetkarmabot.changelocale.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
new ChatCommandArgument() {
Name="locale",
Required=false,
Type=ChatCommandArgumentType.String,
Description="The locale to switch to. Can be \"list\" to list all possible locales. Also can be empty to get current locale.",
DescriptionID="jetkarmabot.changelocale.localehelp"
}
};
} }

View File

@ -1,51 +1,48 @@
using System; using System.Text.RegularExpressions;
using System.Linq;
using System.Text.RegularExpressions;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
public class CommandString
{ {
public class CommandString public CommandString(string command, params string[] parameters)
{ {
public CommandString(string command, params string[] parameters) Command = command;
{ Parameters = parameters;
Command = command; }
Parameters = parameters;
}
public string Command { get; } public string Command { get; }
public string UserName { get; set; } = null; public string UserName { get; set; } = null;
public string[] Parameters { get; } public string[] Parameters { get; }
public static readonly char[] WS_CHARS = new[] { ' ', '\r', '\n', '\n' }; public static readonly char[] WS_CHARS = new[] { ' ', '\r', '\n', '\n' };
public static bool TryParse(string s, out CommandString result) public static bool TryParse(string s, out CommandString result)
{ {
result = null; result = null;
if (string.IsNullOrWhiteSpace(s) || s[0] != '/') if (string.IsNullOrWhiteSpace(s) || s[0] != '/')
return false; return false;
string[] words = s.Split(WS_CHARS, StringSplitOptions.RemoveEmptyEntries); string[] words = s.Split(WS_CHARS, StringSplitOptions.RemoveEmptyEntries);
if (!words.Any()) if (!words.Any())
return false; return false;
var cmdRegex = new Regex(@"/(?<cmd>\w+)(@(?<name>\w+))?"); var cmdRegex = new Regex(@"/(?<cmd>\w+)(@(?<name>\w+))?");
var match = cmdRegex.Match(words.First()); var match = cmdRegex.Match(words.First());
if (!match.Success) if (!match.Success)
return false; return false;
string cmd = match.Groups["cmd"].Captures[0].Value; string cmd = match.Groups["cmd"].Captures[0].Value;
string username = match.Groups["name"].Captures.Count > 0 ? match.Groups["name"].Captures[0].Value : null; string username = match.Groups["name"].Captures.Count > 0 ? match.Groups["name"].Captures[0].Value : null;
string[] parameters = words.Skip(1).ToArray(); string[] parameters = words.Skip(1).ToArray();
result = new CommandString(cmd, parameters) { UserName = username }; result = new CommandString(cmd, parameters) { UserName = username };
return true; return true;
} }
public static CommandString Parse(string s) public static CommandString Parse(string s)
{ {
if (TryParse(s, out var c)) return c; if (TryParse(s, out var c)) return c;
throw new ArgumentException($"\"{s}\" is not a command"); throw new ArgumentException($"\"{s}\" is not a command");
}
} }
} }

View File

@ -1,33 +1,28 @@
using System.Collections.Generic; using Microsoft.EntityFrameworkCore;
using Perfusion;
using JetKarmaBot.Services.Handling; using JetKarmaBot.Services.Handling;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using JetKarmaBot.Models; using JetKarmaBot.Models;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
public class CurrenciesCommand : IChatCommand
{ {
public class CurrenciesCommand : IChatCommand [Inject] Localization Locale { get; set; }
public IReadOnlyCollection<string> Names => new[] { "currencies", "awardtypes" };
public string Description => "Shows all award types";
public string DescriptionID => "jetkarmabot.currencies.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
};
public async Task<bool> Execute(RequestContext ctx)
{ {
[Inject] Localization Locale { get; set; } var db = ctx.GetFeature<KarmaContext>();
public IReadOnlyCollection<string> Names => new[] { "currencies", "awardtypes" }; var currentLocale = ctx.GetFeature<Locale>();
await ctx.SendMessage(
public string Description => "Shows all award types"; currentLocale["jetkarmabot.currencies.listtext"] + "\n" + string.Join("\n",
public string DescriptionID => "jetkarmabot.currencies.help"; (await db.AwardTypes.ToListAsync())
.Select(x => $"{x.Symbol} ({x.CommandName}) <i>{currentLocale["jetkarmabot.awardtypes.nominative." + x.CommandName]}</i>")));
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] { return true;
};
public async Task<bool> Execute(RequestContext ctx)
{
var db = ctx.GetFeature<KarmaContext>();
var currentLocale = ctx.GetFeature<Locale>();
await ctx.SendMessage(
currentLocale["jetkarmabot.currencies.listtext"] + "\n" + string.Join("\n",
(await db.AwardTypes.ToListAsync())
.Select(x => $"{x.Symbol} ({x.CommandName}) <i>{currentLocale["jetkarmabot.awardtypes.nominative." + x.CommandName]}</i>")));
return true;
}
} }
} }

View File

@ -1,44 +1,38 @@
using System.Collections.Generic; using JetKarmaBot.Services.Handling;
using Perfusion;
using JetKarmaBot.Services.Handling;
using Telegram.Bot.Types.Enums;
using System.Threading.Tasks;
using JetKarmaBot.Models;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
public class HelpCommand : IChatCommand
{ {
public class HelpCommand : IChatCommand [Inject] Localization Locale { get; set; }
public IReadOnlyCollection<string> Names => new[] { "help" };
public string Description => "Displays help text for all(one) command(s)";
public string DescriptionID => "jetkarmabot.help.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
new ChatCommandArgument() {
Name="command",
Required=false,
Type=ChatCommandArgumentType.String,
Description="The command to return help text for. If empty shows all commands.",
DescriptionID="jetkarmabot.help.commandhelp"
}
};
public async Task<bool> Execute(RequestContext ctx)
{ {
[Inject] Localization Locale { get; set; } var currentLocale = ctx.GetFeature<Locale>();
public IReadOnlyCollection<string> Names => new[] { "help" }; var router = ctx.GetFeature<ChatCommandRouter.Feature>().Router;
if (ctx.Command.Parameters.Length < 1)
public string Description => "Displays help text for all(one) command(s)";
public string DescriptionID => "jetkarmabot.help.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
new ChatCommandArgument() {
Name="command",
Required=false,
Type=ChatCommandArgumentType.String,
Description="The command to return help text for. If empty shows all commands.",
DescriptionID="jetkarmabot.help.commandhelp"
}
};
public async Task<bool> Execute(RequestContext ctx)
{ {
var currentLocale = ctx.GetFeature<Locale>(); await ctx.SendMessage(router.GetHelpText(currentLocale));
var router = ctx.GetFeature<ChatCommandRouter.Feature>().Router; return true;
if (ctx.Command.Parameters.Length < 1) }
{ else
await ctx.SendMessage(router.GetHelpText(currentLocale)); {
return true; await ctx.SendMessage(router.GetHelpTextFor(ctx.Command.Parameters[0], currentLocale));
} return true;
else
{
await ctx.SendMessage(router.GetHelpTextFor(ctx.Command.Parameters[0], currentLocale));
return true;
}
} }
} }
} }

View File

@ -1,32 +1,29 @@
using System.Collections.Generic; using JetKarmaBot.Services.Handling;
using System.Threading.Tasks;
using JetKarmaBot.Services.Handling;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
public interface IChatCommand
{ {
public interface IChatCommand IReadOnlyCollection<string> Names { get; }
{ string Description { get; }
IReadOnlyCollection<string> Names { get; } string DescriptionID { get; }
string Description { get; } IReadOnlyCollection<ChatCommandArgument> Arguments { get; }
string DescriptionID { get; }
IReadOnlyCollection<ChatCommandArgument> Arguments { get; }
Task<bool> Execute(RequestContext ctx); Task<bool> Execute(RequestContext ctx);
} }
public struct ChatCommandArgument public struct ChatCommandArgument
{ {
public string Name; public string Name;
public bool Required; public bool Required;
public ChatCommandArgumentType Type; public ChatCommandArgumentType Type;
public string Description; public string Description;
public string DescriptionID; public string DescriptionID;
} }
public enum ChatCommandArgumentType public enum ChatCommandArgumentType
{ {
Boolean, Boolean,
String, String,
Integer, Integer,
}
} }

View File

@ -1,70 +1,65 @@
using System.Linq;
using System.Collections.Generic;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore;
using Perfusion;
using JetKarmaBot.Services.Handling; using JetKarmaBot.Services.Handling;
using JetKarmaBot.Models; using JetKarmaBot.Models;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
class LeaderboardCommand : IChatCommand
{ {
class LeaderboardCommand : IChatCommand public IReadOnlyCollection<string> Names => new[] { "leaderboard" };
public async Task<bool> Execute(RequestContext ctx)
{ {
public IReadOnlyCollection<string> Names => new[] { "leaderboard" }; bool isPrivate = ctx.EventArgs.Message.Chat.Type == Telegram.Bot.Types.Enums.ChatType.Private;
var currentLocale = ctx.GetFeature<Locale>();
public async Task<bool> Execute(RequestContext ctx) if (isPrivate)
{ {
bool isPrivate = ctx.EventArgs.Message.Chat.Type == Telegram.Bot.Types.Enums.ChatType.Private; await ctx.SendMessage(currentLocale["jetkarmabot.award.errawardself"]);
var currentLocale = ctx.GetFeature<Locale>();
if (isPrivate)
{
await ctx.SendMessage(currentLocale["jetkarmabot.award.errawardself"]);
return true;
}
var db = ctx.GetFeature<KarmaContext>();
var asker = ctx.EventArgs.Message.From;
var awardTypeName = ctx.Command.Parameters.FirstOrDefault();
if (string.IsNullOrWhiteSpace(awardTypeName))
awardTypeName = "star";
var awardTypeIdQuery = from awt in db.AwardTypes
where awt.CommandName == awardTypeName
select awt.AwardTypeId;
var awardTypeId = await awardTypeIdQuery.FirstAsync();
var awardType = await db.AwardTypes.FindAsync(awardTypeId);
var topEarners = await db.Awards
.Where(x => x.ChatId == ctx.EventArgs.Message.Chat.Id && x.AwardTypeId == awardTypeId)
.GroupBy(x => x.To)
.Select(x => new { User = x.Key, Amount = x.Sum(y => y.Amount) })
.OrderByDescending(x => x.Amount)
.Take(5)
.ToListAsync();
var response = string.Format(currentLocale["jetkarmabot.leaderboard.specifictext"], awardType.Symbol) + "\n"
+ string.Join('\n', topEarners.Select((x, index)
=> $"{index + 1}. {x.User.Username} - {x.Amount}")
);
await ctx.SendMessage(response);
return true; return true;
} }
[Inject] Localization Locale { get; set; } var db = ctx.GetFeature<KarmaContext>();
var asker = ctx.EventArgs.Message.From;
var awardTypeName = ctx.Command.Parameters.FirstOrDefault();
public string Description => "Shows the people with the most of a specific award."; if (string.IsNullOrWhiteSpace(awardTypeName))
public string DescriptionID => "jetkarmabot.leaderboard.help"; awardTypeName = "star";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] { var awardTypeIdQuery = from awt in db.AwardTypes
new ChatCommandArgument() { where awt.CommandName == awardTypeName
Name="awardtype", select awt.AwardTypeId;
Required=true, var awardTypeId = await awardTypeIdQuery.FirstAsync();
Type=ChatCommandArgumentType.String, var awardType = await db.AwardTypes.FindAsync(awardTypeId);
Description="The awardtype to show a leaderboard for.",
DescriptionID= "jetkarmabot.leaderboard.awardtypehelp" var topEarners = await db.Awards
} .Where(x => x.ChatId == ctx.EventArgs.Message.Chat.Id && x.AwardTypeId == awardTypeId)
}; .GroupBy(x => x.To)
.Select(x => new { User = x.Key, Amount = x.Sum(y => y.Amount) })
.OrderByDescending(x => x.Amount)
.Take(5)
.ToListAsync();
var response = string.Format(currentLocale["jetkarmabot.leaderboard.specifictext"], awardType.Symbol) + "\n"
+ string.Join('\n', topEarners.Select((x, index)
=> $"{index + 1}. {x.User.Username} - {x.Amount}")
);
await ctx.SendMessage(response);
return true;
} }
[Inject] Localization Locale { get; set; }
public string Description => "Shows the people with the most of a specific award.";
public string DescriptionID => "jetkarmabot.leaderboard.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => new ChatCommandArgument[] {
new ChatCommandArgument() {
Name="awardtype",
Required=true,
Type=ChatCommandArgumentType.String,
Description="The awardtype to show a leaderboard for.",
DescriptionID= "jetkarmabot.leaderboard.awardtypehelp"
}
};
} }

View File

@ -1,56 +1,51 @@
using System.Linq;
using System.Collections.Generic;
using Perfusion;
using JetKarmaBot.Services.Handling;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore;
using JetKarmaBot.Services.Handling;
using JetKarmaBot.Models; using JetKarmaBot.Models;
namespace JetKarmaBot.Commands namespace JetKarmaBot.Commands;
class StatusCommand : IChatCommand
{ {
class StatusCommand : IChatCommand public IReadOnlyCollection<string> Names => ["status"];
public async Task<bool> Execute(RequestContext ctx)
{ {
public IReadOnlyCollection<string> Names => ["status"]; var db = ctx.GetFeature<KarmaContext>();
var currentLocale = ctx.GetFeature<Locale>();
var asker = ctx.EventArgs.Message.From;
bool isPrivate = ctx.EventArgs.Message.Chat.Type == Telegram.Bot.Types.Enums.ChatType.Private;
public async Task<bool> Execute(RequestContext ctx) string response;
var awards = db.Awards.Where(x => x.ToId == asker.Id);
if (!isPrivate)
awards = awards.Where(x => x.ChatId == ctx.EventArgs.Message.Chat.Id);
if (!awards.Any())
response = currentLocale["jetkarmabot.status.havenothing"];
else
{ {
var db = ctx.GetFeature<KarmaContext>(); var aq = db.AwardTypes.GroupJoin(
var currentLocale = ctx.GetFeature<Locale>(); awards,
var asker = ctx.EventArgs.Message.From; at => at.AwardTypeId,
bool isPrivate = ctx.EventArgs.Message.Chat.Type == Telegram.Bot.Types.Enums.ChatType.Private; aw => aw.AwardTypeId,
(at, aws) => new {at.Symbol, Amount = aws.Sum(aw => aw.Amount) });
string response; var awardsByType = await aq.ToListAsync();
var awards = db.Awards.Where(x => x.ToId == asker.Id); response =
if (!isPrivate) currentLocale["jetkarmabot.status.listalltext"] + "\n"
awards = awards.Where(x => x.ChatId == ctx.EventArgs.Message.Chat.Id); + string.Join("\n", awardsByType.Select(a => $" - {a.Symbol} {a.Amount}"));
if (!awards.Any())
response = currentLocale["jetkarmabot.status.havenothing"];
else
{
var aq = db.AwardTypes.GroupJoin(
awards,
at => at.AwardTypeId,
aw => aw.AwardTypeId,
(at, aws) => new {at.Symbol, Amount = aws.Sum(aw => aw.Amount) });
var awardsByType = await aq.ToListAsync();
response =
currentLocale["jetkarmabot.status.listalltext"] + "\n"
+ string.Join("\n", awardsByType.Select(a => $" - {a.Symbol} {a.Amount}"));
}
await ctx.SendMessage(response);
return true;
} }
[Inject] Localization Locale { get; set; } await ctx.SendMessage(response);
return true;
public string Description => "Shows the amount of awards that you have";
public string DescriptionID => "jetkarmabot.status.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => [];
} }
[Inject] Localization Locale { get; set; }
public string Description => "Shows the amount of awards that you have";
public string DescriptionID => "jetkarmabot.status.help";
public IReadOnlyCollection<ChatCommandArgument> Arguments => [];
} }

View File

@ -2,76 +2,74 @@
using Newtonsoft.Json; using Newtonsoft.Json;
using JsonNet.PrivateSettersContractResolvers; using JsonNet.PrivateSettersContractResolvers;
using Newtonsoft.Json.Linq; using Newtonsoft.Json.Linq;
using System.Collections.Generic;
namespace JetKarmaBot namespace JetKarmaBot;
public class Config : ConfigBase
{ {
public class Config : ConfigBase public Config(string path) : base(path) { }
public string ApiKey { get; private set; }
public string ConnectionString { get; private set; }
public class ProxySettings
{ {
public Config(string path) : base(path) { } public string Url { get; private set; }
public int Port { get; private set; }
public string ApiKey { get; private set; } public string Login { get; private set; }
public string ConnectionString { get; private set; } public string Password { get; private set; }
public class ProxySettings
{
public string Url { get; private set; }
public int Port { get; private set; }
public string Login { get; private set; }
public string Password { get; private set; }
}
public ProxySettings Proxy { get; private set; }
public class TimeoutConfig
{
public int DebtLimitSeconds { get; private set; } = 60 * 60 * 2;
public Dictionary<string, int> CommandCostsSeconds { get; private set; } = new Dictionary<string, int>()
{
{"JetKarmaBot.Commands.AwardCommand (OK)", 60*15},
{"JetKarmaBot.Commands.AwardCommand (ERR)", 60*5},
{"Default", 60*5}
};
public int SaveIntervalSeconds { get; private set; } = 60 * 5;
public double AwardTimeSeconds { get; private set; } = 60;
}
public TimeoutConfig Timeout { get; private set; } = new TimeoutConfig();
public bool SqlDebug { get; private set; } = false;
} }
public abstract class ConfigBase public ProxySettings Proxy { get; private set; }
public class TimeoutConfig
{ {
public ConfigBase(string path) public int DebtLimitSeconds { get; private set; } = 60 * 60 * 2;
public Dictionary<string, int> CommandCostsSeconds { get; private set; } = new Dictionary<string, int>()
{ {
JObject configJson; {"JetKarmaBot.Commands.AwardCommand (OK)", 60*15},
{"JetKarmaBot.Commands.AwardCommand (ERR)", 60*5},
{"Default", 60*5}
};
public int SaveIntervalSeconds { get; private set; } = 60 * 5;
public double AwardTimeSeconds { get; private set; } = 60;
}
public TimeoutConfig Timeout { get; private set; } = new TimeoutConfig();
public bool SqlDebug { get; private set; } = false;
}
if (File.Exists(path)) public abstract class ConfigBase
{
public ConfigBase(string path)
{
JObject configJson;
if (File.Exists(path))
{
configJson = JObject.Parse(File.ReadAllText(path));
using (var sr = configJson.CreateReader())
{ {
configJson = JObject.Parse(File.ReadAllText(path)); var settings = new JsonSerializerSettings
using (var sr = configJson.CreateReader())
{ {
var settings = new JsonSerializerSettings ContractResolver = new PrivateSetterContractResolver()
{ };
ContractResolver = new PrivateSetterContractResolver() JsonSerializer.CreateDefault(settings).Populate(sr, this);
};
JsonSerializer.CreateDefault(settings).Populate(sr, this);
}
} }
else configJson = new JObject(); }
else configJson = new JObject();
configJson.Merge(JToken.FromObject(this), new JsonMergeSettings configJson.Merge(JToken.FromObject(this), new JsonMergeSettings
{ {
MergeArrayHandling = MergeArrayHandling.Union MergeArrayHandling = MergeArrayHandling.Union
}); });
try // populate possible missing properties in file try // populate possible missing properties in file
{ {
File.WriteAllText(path, configJson.ToString(Formatting.Indented)); File.WriteAllText(path, configJson.ToString(Formatting.Indented));
} }
catch (IOException e) catch (IOException e)
{ {
System.Diagnostics.Debug.WriteLine(e); System.Diagnostics.Debug.WriteLine(e);
}
} }
} }
} }

View File

@ -1,15 +1,12 @@
using System.Collections.Generic; namespace JetKarmaBot;
namespace JetKarmaBot public static class IReadOnlyDictionaryExtensions
{ {
public static class IReadOnlyDictionaryExtensions public static TValue GetOrDefault<TKey, TValue>(this IReadOnlyDictionary<TKey, TValue> dict, TKey key)
{ {
public static TValue GetOrDefault<TKey, TValue>(this IReadOnlyDictionary<TKey, TValue> dict, TKey key) TValue res = default;
{ if (key != null)
TValue res = default(TValue); dict.TryGetValue(key, out res);
if (key != null) return res;
dict.TryGetValue(key, out res);
return res;
}
} }
} }

View File

@ -3,43 +3,42 @@ using Newtonsoft.Json;
using Newtonsoft.Json.Serialization; using Newtonsoft.Json.Serialization;
// ReSharper disable once CheckNamespace // ReSharper disable once CheckNamespace
namespace JsonNet.PrivateSettersContractResolvers namespace JsonNet.PrivateSettersContractResolvers;
public class PrivateSetterContractResolver : DefaultContractResolver
{ {
public class PrivateSetterContractResolver : DefaultContractResolver protected override JsonProperty CreateProperty(MemberInfo member, MemberSerialization memberSerialization)
{ {
protected override JsonProperty CreateProperty(MemberInfo member, MemberSerialization memberSerialization) var jProperty = base.CreateProperty(member, memberSerialization);
{ if (jProperty.Writable)
var jProperty = base.CreateProperty(member, memberSerialization);
if (jProperty.Writable)
return jProperty;
jProperty.Writable = member.IsPropertyWithSetter();
return jProperty; return jProperty;
}
}
public class PrivateSetterCamelCasePropertyNamesContractResolver : CamelCasePropertyNamesContractResolver jProperty.Writable = member.IsPropertyWithSetter();
{
protected override JsonProperty CreateProperty(MemberInfo member, MemberSerialization memberSerialization)
{
var jProperty = base.CreateProperty(member, memberSerialization);
if (jProperty.Writable)
return jProperty;
jProperty.Writable = member.IsPropertyWithSetter(); return jProperty;
}
return jProperty; }
}
} public class PrivateSetterCamelCasePropertyNamesContractResolver : CamelCasePropertyNamesContractResolver
{
internal static class MemberInfoExtensions protected override JsonProperty CreateProperty(MemberInfo member, MemberSerialization memberSerialization)
{ {
internal static bool IsPropertyWithSetter(this MemberInfo member) var jProperty = base.CreateProperty(member, memberSerialization);
{ if (jProperty.Writable)
var property = member as PropertyInfo; return jProperty;
return property?.GetSetMethod(true) != null; jProperty.Writable = member.IsPropertyWithSetter();
}
return jProperty;
}
}
internal static class MemberInfoExtensions
{
internal static bool IsPropertyWithSetter(this MemberInfo member)
{
var property = member as PropertyInfo;
return property?.GetSetMethod(true) != null;
} }
} }

View File

@ -0,0 +1,6 @@
global using System;
global using System.Collections.Generic;
global using System.Linq;
global using System.Threading;
global using System.Threading.Tasks;
global using Perfusion;

View File

@ -3,140 +3,135 @@ using JetKarmaBot.Models;
using JetKarmaBot.Services; using JetKarmaBot.Services;
using JetKarmaBot.Services.Handling; using JetKarmaBot.Services.Handling;
using NLog; using NLog;
using Perfusion;
using System;
using System.Threading;
using System.Threading.Tasks;
using Telegram.Bot; using Telegram.Bot;
using Telegram.Bot.Polling; using Telegram.Bot.Polling;
using Telegram.Bot.Types; using Telegram.Bot.Types;
using Telegram.Bot.Types.Enums; using Telegram.Bot.Types.Enums;
namespace JetKarmaBot namespace JetKarmaBot;
public class JetKarmaBot : IDisposable
{ {
public class JetKarmaBot : IDisposable [Inject] Config Config { get; set; }
[Inject] IContainer Container { get; set; }
[Inject] KarmaContextFactory Db { get; set; }
[Inject] TimeoutManager Timeout { get; set; }
[Inject] Localization Locale { get; set; }
[Inject] Logger Log { get; set; }
TelegramBotClient Client { get; set; }
ChatCommandRouter Commands;
RequestChain Chain;
Task timeoutWaitTask;
CancellationTokenSource timeoutWaitTaskToken;
private bool stopped = false;
public async Task Init()
{ {
[Inject] Config Config { get; set; } using (KarmaContext db = Db.GetContext())
[Inject] IContainer Container { get; set; } await db.Database.EnsureCreatedAsync();
[Inject] KarmaContextFactory Db { get; set; }
[Inject] TimeoutManager Timeout { get; set; }
[Inject] Localization Locale { get; set; }
[Inject] Logger Log { get; set; }
Client = new TelegramBotClient(Config.ApiKey);
Container.AddInstance(Client);
TelegramBotClient Client { get; set; } timeoutWaitTaskToken = new CancellationTokenSource();
ChatCommandRouter Commands; timeoutWaitTask = Timeout.SaveLoop(timeoutWaitTaskToken.Token);
RequestChain Chain;
Task timeoutWaitTask;
CancellationTokenSource timeoutWaitTaskToken;
private bool stopped = false;
public async Task Init() await InitCommands(Container);
{ InitChain(Container);
using (KarmaContext db = Db.GetContext())
await db.Database.EnsureCreatedAsync();
Client = new TelegramBotClient(Config.ApiKey); var receiverOptions = new ReceiverOptions { AllowedUpdates = new[] { UpdateType.Message } };
Container.AddInstance(Client); Client.StartReceiving(
HandleUpdateAsync,
timeoutWaitTaskToken = new CancellationTokenSource(); HandleErrorAsync,
timeoutWaitTask = Timeout.SaveLoop(timeoutWaitTaskToken.Token); receiverOptions);
await InitCommands(Container);
InitChain(Container);
var receiverOptions = new ReceiverOptions { AllowedUpdates = new[] { UpdateType.Message } };
Client.StartReceiving(
HandleUpdateAsync,
HandleErrorAsync,
receiverOptions);
}
public async Task Stop()
{
if (stopped) return;
Client?.CloseAsync();
timeoutWaitTaskToken?.Cancel();
try
{
if (timeoutWaitTask != null)
await timeoutWaitTask;
}
catch (OperationCanceledException) { }
await Timeout?.Save();
Dispose();
stopped = true;
}
#region service
Task HandleErrorAsync(ITelegramBotClient botClient, Exception exception, CancellationToken cancellationToken)
{
Log.Error(exception, "Exception while handling API message");
return Task.CompletedTask;
}
async Task HandleUpdateAsync(ITelegramBotClient sender, Update update, CancellationToken cancellationToken)
{
if (update.Type != UpdateType.Message || update?.Message?.Type != MessageType.Text)
return;
var message = update.Message!;
try
{
if (message == null || message.Type != MessageType.Text)
return;
if (!CommandString.TryParse(message.Text, out var cmd))
return;
if (cmd.UserName != null && cmd.UserName != Commands.Me.Username)
return;
RequestContext ctx = new RequestContext(Client, update, cmd);
await Chain.Handle(ctx);
}
catch (Exception e)
{
Log.Error(e, "Exception while handling message {0}", message);
}
}
async Task InitCommands(IContainer c)
{
c.Add<HelpCommand>();
c.Add<AwardCommand>();
c.Add<StatusCommand>();
c.Add<LocaleCommand>();
c.Add<CurrenciesCommand>();
c.Add<LeaderboardCommand>();
Commands = c.GetInstance<ChatCommandRouter>();
await Commands.Start();
foreach (IChatCommand cmd in c.GetInstances<IChatCommand>())
{
Commands.Add(cmd);
}
}
void InitChain(IContainer c)
{
Chain = c.ResolveObject(new RequestChain());
Chain.Add(c.GetInstance<TimeoutManager.PreDbThrowout>());
Chain.Add(c.GetInstance<DatabaseHandler>());
Chain.Add(Timeout);
Chain.Add(c.GetInstance<SaveData>());
Chain.Add(Commands);
}
#endregion
#region IDisposable
public void Dispose()
{
timeoutWaitTaskToken?.Dispose();
timeoutWaitTask?.Dispose();
}
#endregion
} }
public async Task Stop()
{
if (stopped) return;
Client?.CloseAsync();
timeoutWaitTaskToken?.Cancel();
try
{
if (timeoutWaitTask != null)
await timeoutWaitTask;
}
catch (OperationCanceledException) { }
await Timeout?.Save();
Dispose();
stopped = true;
}
#region service
Task HandleErrorAsync(ITelegramBotClient botClient, Exception exception, CancellationToken cancellationToken)
{
Log.Error(exception, "Exception while handling API message");
return Task.CompletedTask;
}
async Task HandleUpdateAsync(ITelegramBotClient sender, Update update, CancellationToken cancellationToken)
{
if (update.Type != UpdateType.Message || update?.Message?.Type != MessageType.Text)
return;
var message = update.Message!;
try
{
if (message == null || message.Type != MessageType.Text)
return;
if (!CommandString.TryParse(message.Text, out var cmd))
return;
if (cmd.UserName != null && cmd.UserName != Commands.Me.Username)
return;
RequestContext ctx = new RequestContext(Client, update, cmd);
await Chain.Handle(ctx);
}
catch (Exception e)
{
Log.Error(e, "Exception while handling message {0}", message);
}
}
async Task InitCommands(IContainer c)
{
c.Add<HelpCommand>();
c.Add<AwardCommand>();
c.Add<StatusCommand>();
c.Add<LocaleCommand>();
c.Add<CurrenciesCommand>();
c.Add<LeaderboardCommand>();
Commands = c.GetInstance<ChatCommandRouter>();
await Commands.Start();
foreach (IChatCommand cmd in c.GetInstances<IChatCommand>())
{
Commands.Add(cmd);
}
}
void InitChain(IContainer c)
{
Chain = c.ResolveObject(new RequestChain());
Chain.Add(c.GetInstance<TimeoutManager.PreDbThrowout>());
Chain.Add(c.GetInstance<DatabaseHandler>());
Chain.Add(Timeout);
Chain.Add(c.GetInstance<SaveData>());
Chain.Add(Commands);
}
#endregion
#region IDisposable
public void Dispose()
{
timeoutWaitTaskToken?.Dispose();
timeoutWaitTask?.Dispose();
}
#endregion
} }

View File

@ -1,25 +1,22 @@
using System; using System.ComponentModel.DataAnnotations.Schema;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations.Schema;
namespace JetKarmaBot.Models namespace JetKarmaBot.Models;
public partial class Award
{ {
public partial class Award public int AwardId { get; set; }
{ public long ChatId { get; set; }
public int AwardId { get; set; } public long FromId { get; set; }
public long ChatId { get; set; } public long ToId { get; set; }
public long FromId { get; set; } public sbyte AwardTypeId { get; set; }
public long ToId { get; set; } public sbyte Amount { get; set; }
public sbyte AwardTypeId { get; set; } public DateTime Date { get; set; }
public sbyte Amount { get; set; } [ForeignKey("AwardTypeId")]
public DateTime Date { get; set; } public virtual AwardType AwardType { get; set; }
[ForeignKey("AwardTypeId")] [ForeignKey("ChatId")]
public virtual AwardType AwardType { get; set; } public virtual Chat Chat { get; set; }
[ForeignKey("ChatId")] [ForeignKey("FromId")]
public virtual Chat Chat { get; set; } public virtual User From { get; set; }
[ForeignKey("FromId")] [ForeignKey("ToId")]
public virtual User From { get; set; } public virtual User To { get; set; }
[ForeignKey("ToId")]
public virtual User To { get; set; }
}
} }

View File

@ -1,21 +1,17 @@
using System; namespace JetKarmaBot.Models;
using System.Collections.Generic;
namespace JetKarmaBot.Models public partial class AwardType
{ {
public partial class AwardType public AwardType()
{ {
public AwardType() Awards = new HashSet<Award>();
{
Awards = new HashSet<Award>();
}
public sbyte AwardTypeId { get; set; }
public string CommandName { get; set; }
public string Name { get; set; }
public string Symbol { get; set; }
public string Description { get; set; }
public virtual ICollection<Award> Awards { get; set; }
} }
public sbyte AwardTypeId { get; set; }
public string CommandName { get; set; }
public string Name { get; set; }
public string Symbol { get; set; }
public string Description { get; set; }
public virtual ICollection<Award> Awards { get; set; }
} }

View File

@ -1,19 +1,15 @@
using System; namespace JetKarmaBot.Models;
using System.Collections.Generic;
namespace JetKarmaBot.Models public partial class Chat
{ {
public partial class Chat public Chat()
{ {
public Chat() Awards = new HashSet<Award>();
{
Awards = new HashSet<Award>();
}
public long ChatId { get; set; }
public string Locale { get; set; } = "ru-RU";
public bool IsAdministrator { get; set; }
public virtual ICollection<Award> Awards { get; set; }
} }
public long ChatId { get; set; }
public string Locale { get; set; } = "ru-RU";
public bool IsAdministrator { get; set; }
public virtual ICollection<Award> Awards { get; set; }
} }

View File

@ -1,183 +1,179 @@
using System; using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata;
using Perfusion;
namespace JetKarmaBot.Models namespace JetKarmaBot.Models;
[Transient]
public partial class KarmaContext : DbContext
{ {
[Transient] public KarmaContext()
public partial class KarmaContext : DbContext
{ {
public KarmaContext() }
public KarmaContext(DbContextOptions<KarmaContext> options)
: base(options)
{
}
public virtual DbSet<Award> Awards { get; set; }
public virtual DbSet<AwardType> AwardTypes { get; set; }
public virtual DbSet<Chat> Chats { get; set; }
public virtual DbSet<User> Users { get; set; }
protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
{
}
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
modelBuilder.Entity<Award>(entity =>
{ {
} entity.ToTable("award");
public KarmaContext(DbContextOptions<KarmaContext> options) entity.HasIndex(e => e.AwardId)
: base(options) .HasDatabaseName("awardid_UNIQUE")
.IsUnique();
entity.HasIndex(e => e.AwardTypeId)
.HasDatabaseName("fk_awardtype_idx");
entity.HasIndex(e => e.ChatId)
.HasDatabaseName("fk_chat_idx");
entity.HasIndex(e => e.FromId)
.HasDatabaseName("fk_from_idx");
entity.HasIndex(e => e.ToId)
.HasDatabaseName("fk_to_idx");
entity.Property(e => e.AwardId)
.HasColumnName("awardid")
.HasColumnType("int(11)");
entity.Property(e => e.Amount)
.HasColumnName("amount")
.HasColumnType("tinyint(3)")
.HasDefaultValueSql("'1'");
entity.Property(e => e.AwardTypeId)
.HasColumnName("awardtypeid")
.HasColumnType("tinyint(3)");
entity.Property(e => e.ChatId)
.HasColumnName("chatid")
.HasColumnType("bigint(20)");
entity.Property(e => e.Date)
.HasColumnName("date")
.HasColumnType("datetime")
.HasDefaultValueSql("CURRENT_TIMESTAMP");
entity.Property(e => e.FromId)
.HasColumnName("fromid")
.HasColumnType("bigint(20)");
entity.Property(e => e.ToId)
.HasColumnName("toid")
.HasColumnType("bigint(20)");
entity.HasOne(d => d.AwardType)
.WithMany(p => p.Awards)
.HasForeignKey(d => d.AwardTypeId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_awardtype");
entity.HasOne(d => d.Chat)
.WithMany(p => p.Awards)
.HasForeignKey(d => d.ChatId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_chat");
entity.HasOne(d => d.From)
.WithMany(p => p.AwardsFrom)
.HasForeignKey(d => d.FromId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_from");
entity.HasOne(d => d.To)
.WithMany(p => p.AwardsTo)
.HasForeignKey(d => d.ToId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_to");
});
modelBuilder.Entity<AwardType>(entity =>
{ {
} entity.ToTable("awardtype");
public virtual DbSet<Award> Awards { get; set; } entity.HasIndex(e => e.AwardTypeId)
public virtual DbSet<AwardType> AwardTypes { get; set; } .HasDatabaseName("awardtypeid_UNIQUE")
public virtual DbSet<Chat> Chats { get; set; } .IsUnique();
public virtual DbSet<User> Users { get; set; }
protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder) entity.HasIndex(e => e.CommandName)
.HasDatabaseName("commandname_UNIQUE")
.IsUnique();
entity.Property(e => e.AwardTypeId)
.HasColumnName("awardtypeid")
.HasColumnType("tinyint(3)");
entity.Property(e => e.CommandName)
.IsRequired()
.HasColumnName("commandname")
.HasColumnType("varchar(35)");
entity.Property(e => e.Description)
.IsRequired()
.HasColumnName("description")
.HasColumnType("text");
entity.Property(e => e.Name)
.IsRequired()
.HasColumnName("name")
.HasColumnType("varchar(32)");
entity.Property(e => e.Symbol)
.IsRequired()
.HasColumnName("symbol")
.HasColumnType("varchar(16)");
});
modelBuilder.Entity<Chat>(entity =>
{ {
entity.ToTable("chat");
} entity.Property(e => e.ChatId)
.HasColumnName("chatid")
.HasColumnType("bigint(20)");
protected override void OnModelCreating(ModelBuilder modelBuilder) entity.Property(e => e.Locale)
.IsRequired()
.HasColumnName("locale")
.HasColumnType("varchar(10)")
.HasDefaultValueSql("'ru-RU'");
entity.Property(e => e.IsAdministrator)
.HasColumnName("isadministrator")
.HasColumnType("tinyint(1)")
.HasDefaultValueSql("'0'");
});
modelBuilder.Entity<User>(entity =>
{ {
modelBuilder.Entity<Award>(entity => entity.ToTable("user");
{
entity.ToTable("award");
entity.HasIndex(e => e.AwardId) entity.Property(e => e.UserId)
.HasDatabaseName("awardid_UNIQUE") .HasColumnName("userid")
.IsUnique(); .HasColumnType("bigint(20)");
entity.HasIndex(e => e.AwardTypeId) entity.Property(e => e.Username)
.HasDatabaseName("fk_awardtype_idx"); .HasColumnName("username")
.HasColumnType("varchar(45)");
entity.HasIndex(e => e.ChatId) entity.Property(e => e.CooldownDate)
.HasDatabaseName("fk_chat_idx"); .HasColumnName("cooldowndate")
.HasColumnType("timestamp")
entity.HasIndex(e => e.FromId) .HasDefaultValueSql("CURRENT_TIMESTAMP");
.HasDatabaseName("fk_from_idx"); });
entity.HasIndex(e => e.ToId)
.HasDatabaseName("fk_to_idx");
entity.Property(e => e.AwardId)
.HasColumnName("awardid")
.HasColumnType("int(11)");
entity.Property(e => e.Amount)
.HasColumnName("amount")
.HasColumnType("tinyint(3)")
.HasDefaultValueSql("'1'");
entity.Property(e => e.AwardTypeId)
.HasColumnName("awardtypeid")
.HasColumnType("tinyint(3)");
entity.Property(e => e.ChatId)
.HasColumnName("chatid")
.HasColumnType("bigint(20)");
entity.Property(e => e.Date)
.HasColumnName("date")
.HasColumnType("datetime")
.HasDefaultValueSql("CURRENT_TIMESTAMP");
entity.Property(e => e.FromId)
.HasColumnName("fromid")
.HasColumnType("bigint(20)");
entity.Property(e => e.ToId)
.HasColumnName("toid")
.HasColumnType("bigint(20)");
entity.HasOne(d => d.AwardType)
.WithMany(p => p.Awards)
.HasForeignKey(d => d.AwardTypeId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_awardtype");
entity.HasOne(d => d.Chat)
.WithMany(p => p.Awards)
.HasForeignKey(d => d.ChatId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_chat");
entity.HasOne(d => d.From)
.WithMany(p => p.AwardsFrom)
.HasForeignKey(d => d.FromId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_from");
entity.HasOne(d => d.To)
.WithMany(p => p.AwardsTo)
.HasForeignKey(d => d.ToId)
.OnDelete(DeleteBehavior.ClientSetNull)
.HasConstraintName("fk_to");
});
modelBuilder.Entity<AwardType>(entity =>
{
entity.ToTable("awardtype");
entity.HasIndex(e => e.AwardTypeId)
.HasDatabaseName("awardtypeid_UNIQUE")
.IsUnique();
entity.HasIndex(e => e.CommandName)
.HasDatabaseName("commandname_UNIQUE")
.IsUnique();
entity.Property(e => e.AwardTypeId)
.HasColumnName("awardtypeid")
.HasColumnType("tinyint(3)");
entity.Property(e => e.CommandName)
.IsRequired()
.HasColumnName("commandname")
.HasColumnType("varchar(35)");
entity.Property(e => e.Description)
.IsRequired()
.HasColumnName("description")
.HasColumnType("text");
entity.Property(e => e.Name)
.IsRequired()
.HasColumnName("name")
.HasColumnType("varchar(32)");
entity.Property(e => e.Symbol)
.IsRequired()
.HasColumnName("symbol")
.HasColumnType("varchar(16)");
});
modelBuilder.Entity<Chat>(entity =>
{
entity.ToTable("chat");
entity.Property(e => e.ChatId)
.HasColumnName("chatid")
.HasColumnType("bigint(20)");
entity.Property(e => e.Locale)
.IsRequired()
.HasColumnName("locale")
.HasColumnType("varchar(10)")
.HasDefaultValueSql("'ru-RU'");
entity.Property(e => e.IsAdministrator)
.HasColumnName("isadministrator")
.HasColumnType("tinyint(1)")
.HasDefaultValueSql("'0'");
});
modelBuilder.Entity<User>(entity =>
{
entity.ToTable("user");
entity.Property(e => e.UserId)
.HasColumnName("userid")
.HasColumnType("bigint(20)");
entity.Property(e => e.Username)
.HasColumnName("username")
.HasColumnType("varchar(45)");
entity.Property(e => e.CooldownDate)
.HasColumnName("cooldowndate")
.HasColumnType("timestamp")
.HasDefaultValueSql("CURRENT_TIMESTAMP");
});
}
} }
} }

View File

@ -1,23 +1,20 @@
using System; using System.ComponentModel.DataAnnotations.Schema;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations.Schema;
namespace JetKarmaBot.Models namespace JetKarmaBot.Models;
public partial class User
{ {
public partial class User public User()
{ {
public User() AwardsFrom = new HashSet<Award>();
{ AwardsTo = new HashSet<Award>();
AwardsFrom = new HashSet<Award>();
AwardsTo = new HashSet<Award>();
}
public long UserId { get; set; }
public string Username { get; set; }
public DateTime CooldownDate { get; set; }
[InverseProperty("From")]
public virtual ICollection<Award> AwardsFrom { get; set; }
[InverseProperty("To")]
public virtual ICollection<Award> AwardsTo { get; set; }
} }
public long UserId { get; set; }
public string Username { get; set; }
public DateTime CooldownDate { get; set; }
[InverseProperty("From")]
public virtual ICollection<Award> AwardsFrom { get; set; }
[InverseProperty("To")]
public virtual ICollection<Award> AwardsTo { get; set; }
} }

View File

@ -1,79 +1,73 @@
using System; using Microsoft.EntityFrameworkCore;
using System.Reflection;
using System.Runtime.Loader;
using System.Threading;
using JetKarmaBot.Models;
using Microsoft.EntityFrameworkCore;
using NLog; using NLog;
using Perfusion; using JetKarmaBot.Models;
namespace JetKarmaBot namespace JetKarmaBot;
public static class Program
{ {
public static class Program private static Logger log = LogManager.GetCurrentClassLogger();
public enum ExitCode : int
{ {
private static Logger log = LogManager.GetCurrentClassLogger(); Ok = 0,
public enum ExitCode : int ErrorNotStarted = 0x80,
ErrorRunning = 0x81,
ErrorException = 0x82,
ErrorInvalidCommandLine = 0x100
};
public static int Main(string[] args)
{
log.Info("Starting JetKarmaBot.");
Container c = new Container();
var cfg = new Config("karma.cfg.json");
c.AddInstance(cfg);
var connStr = cfg.ConnectionString + (cfg.ConnectionString.EndsWith(";") ? "" : ";") + "TreatTinyAsBoolean=false";
var serverVersion = ServerVersion.AutoDetect(connStr);
var dbOptions = new DbContextOptionsBuilder<KarmaContext>()
.UseMySql(connStr, serverVersion);
c.AddInfo<Logger>(new LogInfo());
if (cfg.SqlDebug)
{ {
Ok = 0, dbOptions = dbOptions.UseLoggerFactory(c.GetInstance<NLoggerFactory>());
ErrorNotStarted = 0x80,
ErrorRunning = 0x81,
ErrorException = 0x82,
ErrorInvalidCommandLine = 0x100
};
public static int Main(string[] args)
{
log.Info("Starting JetKarmaBot.");
Container c = new Container();
var cfg = new Config("karma.cfg.json");
c.AddInstance(cfg);
var connStr = cfg.ConnectionString + (cfg.ConnectionString.EndsWith(";") ? "" : ";") + "TreatTinyAsBoolean=false";
var serverVersion = ServerVersion.AutoDetect(connStr);
var dbOptions = new DbContextOptionsBuilder<KarmaContext>()
.UseMySql(connStr, serverVersion);
c.AddInfo<Logger>(new LogInfo());
if (cfg.SqlDebug)
{
dbOptions = dbOptions.UseLoggerFactory(c.GetInstance<NLoggerFactory>());
}
c.AddTransient(() => new KarmaContext(dbOptions.Options));
c.Add<JetKarmaBot>();
var bot = c.GetInstance<JetKarmaBot>();
try
{
bot.Init().GetAwaiter().GetResult();
log.Info("JetKarmaBot started. Press Ctrl-C to exit...");
}
catch (Exception ex)
{
log.Error(ex);
return (int)ExitCode.ErrorException;
}
ManualResetEvent quitEvent = new ManualResetEvent(false);
try
{
Console.CancelKeyPress += (sender, eArgs) => // ctrl-c
{
eArgs.Cancel = true;
quitEvent.Set();
};
AppDomain.CurrentDomain.ProcessExit += (sender, args) =>
{
log.Info("Received stop request, waiting for exit...");
bot?.Stop()?.GetAwaiter().GetResult();
};
}
catch { }
quitEvent.WaitOne(Timeout.Infinite);
log.Info("Waiting for exit...");
bot?.Stop()?.GetAwaiter().GetResult();
return (int)ExitCode.Ok;
} }
c.AddTransient(() => new KarmaContext(dbOptions.Options));
c.Add<JetKarmaBot>();
var bot = c.GetInstance<JetKarmaBot>();
try
{
bot.Init().GetAwaiter().GetResult();
log.Info("JetKarmaBot started. Press Ctrl-C to exit...");
}
catch (Exception ex)
{
log.Error(ex);
return (int)ExitCode.ErrorException;
}
ManualResetEvent quitEvent = new ManualResetEvent(false);
try
{
Console.CancelKeyPress += (sender, eArgs) => // ctrl-c
{
eArgs.Cancel = true;
quitEvent.Set();
};
AppDomain.CurrentDomain.ProcessExit += (sender, args) =>
{
log.Info("Received stop request, waiting for exit...");
bot?.Stop()?.GetAwaiter().GetResult();
};
}
catch { }
quitEvent.WaitOne(Timeout.Infinite);
log.Info("Waiting for exit...");
bot?.Stop()?.GetAwaiter().GetResult();
return (int)ExitCode.Ok;
} }
} }

View File

@ -1,111 +1,105 @@
using JetKarmaBot.Commands; using NLog;
using NLog;
using Perfusion;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Telegram.Bot; using Telegram.Bot;
using JetKarmaBot.Commands;
namespace JetKarmaBot.Services.Handling namespace JetKarmaBot.Services.Handling;
public class ChatCommandRouter : IRequestHandler
{ {
public class ChatCommandRouter : IRequestHandler public class Feature
{ {
public class Feature public Type CommandType;
{ public ChatCommandRouter Router;
public Type CommandType; public bool Succeded;
public ChatCommandRouter Router; }
public bool Succeded; public Telegram.Bot.Types.User Me { get; private set; }
} [Inject] private Logger log;
public Telegram.Bot.Types.User Me { get; private set; } [Inject] private TelegramBotClient Client { get; set; }
[Inject] private Logger log;
[Inject] private TelegramBotClient Client { get; set; }
public async Task Start() public async Task Start()
{ {
Me = await Client.GetMeAsync(); Me = await Client.GetMeAsync();
} }
public Task Handle(RequestContext ctx, Func<RequestContext, Task> next) public Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{ {
log.Debug("Message received"); log.Debug("Message received");
CommandString cmd = ctx.Command; CommandString cmd = ctx.Command;
Feature feature = new Feature() { Router = this }; Feature feature = new Feature() { Router = this };
ctx.AddFeature(feature); ctx.AddFeature(feature);
try try
{
if (commands.ContainsKey(cmd.Command))
{ {
if (commands.ContainsKey(cmd.Command)) feature.CommandType = commands[cmd.Command].GetType();
{ log.Debug($"Handling message via {feature.CommandType.Name}");
feature.CommandType = commands[cmd.Command].GetType(); async Task processCommand() => feature.Succeded = await commands[cmd.Command].Execute(ctx);
log.Debug($"Handling message via {feature.CommandType.Name}"); return processCommand();
async Task processCommand() => feature.Succeded = await commands[cmd.Command].Execute(ctx);
return processCommand();
}
}
catch (Exception e)
{
log.Error($"Error while handling command {cmd.Command}!");
log.Error(e);
}
return next(ctx);
}
public void Add(IChatCommand c)
{
log.ConditionalTrace($"Adding command {c.GetType().Name}");
foreach (var name in c.Names)
{
log.ConditionalTrace($"Mounting {c.GetType().Name} to {name}");
if (commands.ContainsKey(name))
throw new Exception($"command collision for name {name}, commands {commands[name].GetType()} and {c.GetType()}");
commands[name] = c;
} }
} }
catch (Exception e)
internal string GetHelpText(Locale loc)
{ {
List<string> pieces = new List<string>(); log.Error($"Error while handling command {cmd.Command}!");
foreach (IChatCommand c in commands.Values.Distinct()) log.Error(e);
{
string build = "";
List<string> names = c.Names.ToList();
for (int i = 0; i < names.Count - 1; i++)
{
build = build + "/" + names[i] + "\n";
}
build += "/" + names[names.Count - 1] + " " + string.Join(" ", c.Arguments.Select(x => (!x.Required ? "[" : "") + x.Name + (!x.Required ? "]" : ""))) + " <i>" + getLocalizedCMDDesc(c, loc) + "</i>";
pieces.Add(build);
}
return string.Join("\n", pieces);
} }
internal string GetHelpTextFor(string commandname, Locale loc) return next(ctx);
}
public void Add(IChatCommand c)
{
log.ConditionalTrace($"Adding command {c.GetType().Name}");
foreach (var name in c.Names)
{
log.ConditionalTrace($"Mounting {c.GetType().Name} to {name}");
if (commands.ContainsKey(name))
throw new Exception($"command collision for name {name}, commands {commands[name].GetType()} and {c.GetType()}");
commands[name] = c;
}
}
internal string GetHelpText(Locale loc)
{
List<string> pieces = new List<string>();
foreach (IChatCommand c in commands.Values.Distinct())
{ {
IChatCommand c = commands[commandname];
string build = ""; string build = "";
List<string> names = c.Names.ToList(); List<string> names = c.Names.ToList();
for (int i = 0; i < names.Count - 1; i++) for (int i = 0; i < names.Count - 1; i++)
{ {
build = build + "/" + names[i] + "\n"; build = build + "/" + names[i] + "\n";
} }
build += "/" + names[names.Count - 1] + " " + string.Join(" ", c.Arguments.Select(x => (!x.Required ? "[" : "") + x.Name + (!x.Required ? "]" : ""))) + " <i>" + getLocalizedCMDDesc(c, loc) + "</i>\n"; build += "/" + names[names.Count - 1] + " " + string.Join(" ", c.Arguments.Select(x => (!x.Required ? "[" : "") + x.Name + (!x.Required ? "]" : ""))) + " <i>" + getLocalizedCMDDesc(c, loc) + "</i>";
build += string.Join("\n", c.Arguments.Select(ca => (!ca.Required ? "[" : "") + ca.Name + (!ca.Required ? "]" : "") + ": <i>" + getLocalizedCMDArgDesc(ca, loc) + "</i>")); pieces.Add(build);
return build;
} }
return string.Join("\n", pieces);
private string getLocalizedCMDDesc(IChatCommand cmd, Locale loc)
{
if (loc.ContainsKey(cmd.DescriptionID)) return loc[cmd.DescriptionID];
else return cmd.Description;
}
private string getLocalizedCMDArgDesc(ChatCommandArgument arg, Locale loc)
{
if (loc.ContainsKey(arg.DescriptionID)) return loc[arg.DescriptionID];
else return arg.Description;
}
Dictionary<string, IChatCommand> commands = new Dictionary<string, IChatCommand>();
} }
internal string GetHelpTextFor(string commandname, Locale loc)
{
IChatCommand c = commands[commandname];
string build = "";
List<string> names = c.Names.ToList();
for (int i = 0; i < names.Count - 1; i++)
{
build = build + "/" + names[i] + "\n";
}
build += "/" + names[names.Count - 1] + " " + string.Join(" ", c.Arguments.Select(x => (!x.Required ? "[" : "") + x.Name + (!x.Required ? "]" : ""))) + " <i>" + getLocalizedCMDDesc(c, loc) + "</i>\n";
build += string.Join("\n", c.Arguments.Select(ca => (!ca.Required ? "[" : "") + ca.Name + (!ca.Required ? "]" : "") + ": <i>" + getLocalizedCMDArgDesc(ca, loc) + "</i>"));
return build;
}
private string getLocalizedCMDDesc(IChatCommand cmd, Locale loc)
{
if (loc.ContainsKey(cmd.DescriptionID)) return loc[cmd.DescriptionID];
else return cmd.Description;
}
private string getLocalizedCMDArgDesc(ChatCommandArgument arg, Locale loc)
{
if (loc.ContainsKey(arg.DescriptionID)) return loc[arg.DescriptionID];
else return arg.Description;
}
Dictionary<string, IChatCommand> commands = new Dictionary<string, IChatCommand>();
} }

View File

@ -1,23 +1,17 @@
using System; namespace JetKarmaBot.Services.Handling;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using Perfusion;
namespace JetKarmaBot.Services.Handling public class DatabaseHandler : IRequestHandler
{ {
public class DatabaseHandler : IRequestHandler [Inject] private KarmaContextFactory Db;
[Inject] private Localization Locale;
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{ {
[Inject] private KarmaContextFactory Db; using (var db = Db.GetContext())
[Inject] private Localization Locale;
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{ {
using (var db = Db.GetContext()) ctx.AddFeature(db); // KarmaContext
{ ctx.AddFeature(Locale[(await db.Chats.FindAsync(ctx.EventArgs.Message.Chat.Id))?.Locale ?? "ru-ru"]); // Locale
ctx.AddFeature(db); // KarmaContext await next(ctx);
ctx.AddFeature(Locale[(await db.Chats.FindAsync(ctx.EventArgs.Message.Chat.Id))?.Locale ?? "ru-ru"]); // Locale await db.SaveChangesAsync();
await next(ctx);
await db.SaveChangesAsync();
}
} }
} }
} }

View File

@ -1,42 +1,37 @@
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using NLog; using NLog;
using Perfusion;
namespace JetKarmaBot.Services.Handling namespace JetKarmaBot.Services.Handling;
public interface IRequestHandler
{ {
public interface IRequestHandler Task Handle(RequestContext ctx, Func<RequestContext, Task> next);
}
public class RequestChain : IRequestHandler
{
[Inject] private Logger log;
List<IRequestHandler> handlerStack = new List<IRequestHandler>();
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next = null)
{ {
Task Handle(RequestContext ctx, Func<RequestContext, Task> next); int i = 0;
} Func<RequestContext, Task> chainNext = null;
public class RequestChain : IRequestHandler chainNext = (newCtx) =>
{
[Inject] private Logger log;
List<IRequestHandler> handlerStack = new List<IRequestHandler>();
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next = null)
{ {
int i = 0; if (i == handlerStack.Count)
Func<RequestContext, Task> chainNext = null;
chainNext = (newCtx) =>
{ {
if (i == handlerStack.Count) log.ConditionalTrace("(next) End of request chain");
{ return Task.CompletedTask;
log.ConditionalTrace("(next) End of request chain"); }
return Task.CompletedTask; IRequestHandler handler = handlerStack[i++];
} log.ConditionalTrace($"(next) Executing handler {handler.GetType().Name}");
IRequestHandler handler = handlerStack[i++]; return handler.Handle(newCtx, chainNext);
log.ConditionalTrace($"(next) Executing handler {handler.GetType().Name}"); };
return handler.Handle(newCtx, chainNext); await chainNext(ctx);
}; if (next != null)
await chainNext(ctx); await next(ctx);
if (next != null) }
await next(ctx); public void Add(IRequestHandler handler)
} {
public void Add(IRequestHandler handler) log.ConditionalTrace($"Adding {handler.GetType().Name} to reqchain");
{ handlerStack.Add(handler);
log.ConditionalTrace($"Adding {handler.GetType().Name} to reqchain");
handlerStack.Add(handler);
}
} }
} }

View File

@ -1,39 +1,32 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using JetKarmaBot.Commands;
using JetKarmaBot.Models;
using Telegram.Bot; using Telegram.Bot;
using Telegram.Bot.Args;
using Telegram.Bot.Types; using Telegram.Bot.Types;
using JetKarmaBot.Commands;
namespace JetKarmaBot.Services.Handling namespace JetKarmaBot.Services.Handling;
public class RequestContext : IServiceProvider
{ {
public class RequestContext : IServiceProvider public ITelegramBotClient Client { get; }
public Update EventArgs { get; }
public CommandString Command { get; }
public Dictionary<Type, object> Features { get; } = new Dictionary<Type, object>();
public RequestContext(ITelegramBotClient client, Update args, CommandString cmd)
{ {
public ITelegramBotClient Client { get; } Client = client;
public Update EventArgs { get; } EventArgs = args;
public CommandString Command { get; } Command = cmd;
public Dictionary<Type, object> Features { get; } = new Dictionary<Type, object>();
public RequestContext(ITelegramBotClient client, Update args, CommandString cmd)
{
Client = client;
EventArgs = args;
Command = cmd;
}
public object GetService(Type serviceType) => Features[serviceType];
public T GetFeature<T>() => (T)Features[typeof(T)];
public void AddFeature<T>(T feat) => Features[typeof(T)] = feat;
//Method to reduce WET in commands
public Task SendMessage(string text)
=> Client.SendMessage(
chatId: EventArgs.Message.Chat.Id,
text: text,
disableNotification: true,
parseMode: Telegram.Bot.Types.Enums.ParseMode.Html,
replyParameters: new ReplyParameters { MessageId = EventArgs.Message.MessageId }
);
} }
public object GetService(Type serviceType) => Features[serviceType];
public T GetFeature<T>() => (T)Features[typeof(T)];
public void AddFeature<T>(T feat) => Features[typeof(T)] = feat;
//Method to reduce WET in commands
public Task SendMessage(string text)
=> Client.SendMessage(
chatId: EventArgs.Message.Chat.Id,
text: text,
disableNotification: true,
parseMode: Telegram.Bot.Types.Enums.ParseMode.Html,
replyParameters: new ReplyParameters { MessageId = EventArgs.Message.MessageId }
);
} }

View File

@ -1,36 +1,33 @@
using System;
using System.Threading.Tasks;
using JetKarmaBot.Models;
using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore;
using JetKarmaBot.Models;
namespace JetKarmaBot.Services.Handling namespace JetKarmaBot.Services.Handling;
public class SaveData : IRequestHandler
{ {
public class SaveData : IRequestHandler public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{ {
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next) KarmaContext db = ctx.GetFeature<KarmaContext>();
{ await AddUserToDatabase(db, ctx.EventArgs.Message.From);
KarmaContext db = ctx.GetFeature<KarmaContext>(); if (ctx.EventArgs.Message.ReplyToMessage != null)
await AddUserToDatabase(db, ctx.EventArgs.Message.From); await AddUserToDatabase(db, ctx.EventArgs.Message.ReplyToMessage.From);
if (ctx.EventArgs.Message.ReplyToMessage != null) if (!await db.Chats.AnyAsync(x => x.ChatId == ctx.EventArgs.Message.Chat.Id))
await AddUserToDatabase(db, ctx.EventArgs.Message.ReplyToMessage.From); db.Chats.Add(new Models.Chat
if (!await db.Chats.AnyAsync(x => x.ChatId == ctx.EventArgs.Message.Chat.Id)) {
db.Chats.Add(new Models.Chat ChatId = ctx.EventArgs.Message.Chat.Id
{ });
ChatId = ctx.EventArgs.Message.Chat.Id await next(ctx);
}); }
await next(ctx); private async Task AddUserToDatabase(KarmaContext db, Telegram.Bot.Types.User u)
} {
private async Task AddUserToDatabase(KarmaContext db, Telegram.Bot.Types.User u) string un;
{ if (u.Username == null)
string un; un = u.FirstName + (u.LastName != null ? " " + u.LastName : "");
if (u.Username == null) else
un = u.FirstName + (u.LastName != null ? " " + u.LastName : ""); un = "@" + u.Username;
else if (!await db.Users.AnyAsync(x => x.UserId == u.Id))
un = "@" + u.Username; await db.Users.AddAsync(new Models.User { UserId = u.Id, Username = un });
if (!await db.Users.AnyAsync(x => x.UserId == u.Id)) else
await db.Users.AddAsync(new Models.User { UserId = u.Id, Username = un }); (await db.Users.FindAsync(u.Id)).Username = un;
else
(await db.Users.FindAsync(u.Id)).Username = un;
}
} }
} }

View File

@ -1,133 +1,126 @@
using Perfusion;
using System.Collections.Generic;
using System;
using System.Threading.Tasks;
using JetKarmaBot.Models;
using System.Threading;
using System.Linq;
using NLog; using NLog;
using JetKarmaBot.Models;
namespace JetKarmaBot.Services.Handling namespace JetKarmaBot.Services.Handling;
[Singleton]
public class TimeoutManager : IRequestHandler
{ {
[Singleton] public class Feature
public class TimeoutManager : IRequestHandler
{ {
public class Feature public double Multiplier = 1;
}
public class PreDbThrowout : IRequestHandler
{
public TimeoutManager Timeout { get; }
public PreDbThrowout(TimeoutManager timeout)
{ {
public double Multiplier = 1; Timeout = timeout;
}
public class PreDbThrowout : IRequestHandler
{
public TimeoutManager Timeout { get; }
public PreDbThrowout(TimeoutManager timeout)
{
Timeout = timeout;
}
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{
var uid = ctx.EventArgs.Message.From.Id;
if (Timeout.TimeoutCache.TryGetValue(uid, out var stats))
{
DateTime debtLimit = DateTime.Now.AddSeconds(Timeout.cfg.Timeout.DebtLimitSeconds);
if (debtLimit < stats.CooldownDate && stats.TimeoutMessaged)
return;
}
await next(ctx);
}
}
public class TimeoutStats
{
public DateTime CooldownDate;
public bool TimeoutMessaged;
public DateTime PreviousAwardDate;
}
[Inject] private KarmaContextFactory Db;
[Inject] private Config cfg;
[Inject] private Localization Locale;
[Inject] private Logger log;
public Dictionary<long, TimeoutStats> TimeoutCache = new ();
private async Task ApplyCost(string name, bool succeded, long uid, KarmaContext db, Feature feature)
{
if (feature.Multiplier == 0)
return;
if (!cfg.Timeout.CommandCostsSeconds.TryGetValue(name + (succeded ? " (OK)" : "(ERR)"), out var costSeconds))
if (!cfg.Timeout.CommandCostsSeconds.TryGetValue(name, out costSeconds))
if (!cfg.Timeout.CommandCostsSeconds.TryGetValue("Default", out costSeconds))
{
throw new LocalizationException("Default key not present");
}
await PopulateStats(uid, db);
DateTime debtLimit = DateTime.Now.AddSeconds(cfg.Timeout.DebtLimitSeconds);
if (TimeoutCache[uid].CooldownDate >= debtLimit)
//Programming error
throw new NotImplementedException();
TimeoutCache[uid].CooldownDate = (TimeoutCache[uid].CooldownDate <= DateTime.Now ? DateTime.Now : TimeoutCache[uid].CooldownDate)
.AddSeconds(feature.Multiplier * costSeconds);
TimeoutCache[uid].TimeoutMessaged = false;
}
private async Task PopulateStats(long uid, KarmaContext db)
{
if (!TimeoutCache.ContainsKey(uid))
{
log.ConditionalTrace($"User {uid} not present: saving to cache");
TimeoutCache[uid] = new TimeoutStats()
{
CooldownDate = (await db.Users.FindAsync(uid))?.CooldownDate ?? DateTime.Now
};
}
}
public async Task Save(CancellationToken ct = default(CancellationToken))
{
log.Debug("Saving timeout info to database");
using (KarmaContext db = Db.GetContext())
{
foreach (var i in TimeoutCache.Keys)
{
(await db.Users.FindAsync(new object[] { i }, ct)).CooldownDate = TimeoutCache[i].CooldownDate;
}
await db.SaveChangesAsync(ct);
}
log.Debug("Saved timeout info to database");
}
public async Task SaveLoop(CancellationToken ct = default(CancellationToken))
{
while (!ct.IsCancellationRequested)
{
await Task.Delay(cfg.Timeout.SaveIntervalSeconds * 1000, ct);
await Save(ct);
}
} }
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next) public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{ {
var uid = ctx.EventArgs.Message.From.Id; var uid = ctx.EventArgs.Message.From.Id;
KarmaContext db = ctx.GetFeature<KarmaContext>(); if (Timeout.TimeoutCache.TryGetValue(uid, out var stats))
await PopulateStats(uid, db);
DateTime debtLimit = DateTime.Now.AddSeconds(cfg.Timeout.DebtLimitSeconds);
if (debtLimit < TimeoutCache[uid].CooldownDate)
{ {
if (!TimeoutCache[uid].TimeoutMessaged) DateTime debtLimit = DateTime.Now.AddSeconds(Timeout.cfg.Timeout.DebtLimitSeconds);
{ if (debtLimit < stats.CooldownDate && stats.TimeoutMessaged)
Locale currentLocale = ctx.GetFeature<Locale>(); return;
await ctx.SendMessage(currentLocale["jetkarmabot.ratelimit"]);
TimeoutCache[uid].TimeoutMessaged = true;
}
return;
} }
Feature feature = new Feature();
ctx.AddFeature(feature);
await next(ctx); await next(ctx);
var routerFeature = ctx.GetFeature<ChatCommandRouter.Feature>();
await ApplyCost(getTypeName(routerFeature.CommandType), routerFeature.Succeded, uid, db, feature);
}
private string getTypeName(Type t)
{
return (t.DeclaringType == null ? t.Namespace + "." + t.Name : getTypeName(t.DeclaringType) + "." + t.Name)
+ (t.GenericTypeArguments.Length > 0 ? "<" + string.Join(",", t.GenericTypeArguments.Select(getTypeName)) + ">" : "");
} }
} }
public class TimeoutStats
{
public DateTime CooldownDate;
public bool TimeoutMessaged;
public DateTime PreviousAwardDate;
}
[Inject] private KarmaContextFactory Db;
[Inject] private Config cfg;
[Inject] private Localization Locale;
[Inject] private Logger log;
public Dictionary<long, TimeoutStats> TimeoutCache = new ();
private async Task ApplyCost(string name, bool succeded, long uid, KarmaContext db, Feature feature)
{
if (feature.Multiplier == 0)
return;
if (!cfg.Timeout.CommandCostsSeconds.TryGetValue(name + (succeded ? " (OK)" : "(ERR)"), out var costSeconds))
if (!cfg.Timeout.CommandCostsSeconds.TryGetValue(name, out costSeconds))
if (!cfg.Timeout.CommandCostsSeconds.TryGetValue("Default", out costSeconds))
{
throw new LocalizationException("Default key not present");
}
await PopulateStats(uid, db);
DateTime debtLimit = DateTime.Now.AddSeconds(cfg.Timeout.DebtLimitSeconds);
if (TimeoutCache[uid].CooldownDate >= debtLimit)
//Programming error
throw new NotImplementedException();
TimeoutCache[uid].CooldownDate = (TimeoutCache[uid].CooldownDate <= DateTime.Now ? DateTime.Now : TimeoutCache[uid].CooldownDate)
.AddSeconds(feature.Multiplier * costSeconds);
TimeoutCache[uid].TimeoutMessaged = false;
}
private async Task PopulateStats(long uid, KarmaContext db)
{
if (!TimeoutCache.ContainsKey(uid))
{
log.ConditionalTrace($"User {uid} not present: saving to cache");
TimeoutCache[uid] = new TimeoutStats()
{
CooldownDate = (await db.Users.FindAsync(uid))?.CooldownDate ?? DateTime.Now
};
}
}
public async Task Save(CancellationToken ct = default(CancellationToken))
{
log.Debug("Saving timeout info to database");
using (KarmaContext db = Db.GetContext())
{
foreach (var i in TimeoutCache.Keys)
{
(await db.Users.FindAsync(new object[] { i }, ct)).CooldownDate = TimeoutCache[i].CooldownDate;
}
await db.SaveChangesAsync(ct);
}
log.Debug("Saved timeout info to database");
}
public async Task SaveLoop(CancellationToken ct = default(CancellationToken))
{
while (!ct.IsCancellationRequested)
{
await Task.Delay(cfg.Timeout.SaveIntervalSeconds * 1000, ct);
await Save(ct);
}
}
public async Task Handle(RequestContext ctx, Func<RequestContext, Task> next)
{
var uid = ctx.EventArgs.Message.From.Id;
KarmaContext db = ctx.GetFeature<KarmaContext>();
await PopulateStats(uid, db);
DateTime debtLimit = DateTime.Now.AddSeconds(cfg.Timeout.DebtLimitSeconds);
if (debtLimit < TimeoutCache[uid].CooldownDate)
{
if (!TimeoutCache[uid].TimeoutMessaged)
{
Locale currentLocale = ctx.GetFeature<Locale>();
await ctx.SendMessage(currentLocale["jetkarmabot.ratelimit"]);
TimeoutCache[uid].TimeoutMessaged = true;
}
return;
}
Feature feature = new Feature();
ctx.AddFeature(feature);
await next(ctx);
var routerFeature = ctx.GetFeature<ChatCommandRouter.Feature>();
await ApplyCost(getTypeName(routerFeature.CommandType), routerFeature.Succeded, uid, db, feature);
}
private string getTypeName(Type t)
{
return (t.DeclaringType == null ? t.Namespace + "." + t.Name : getTypeName(t.DeclaringType) + "." + t.Name)
+ (t.GenericTypeArguments.Length > 0 ? "<" + string.Join(",", t.GenericTypeArguments.Select(getTypeName)) + ">" : "");
}
} }

View File

@ -1,12 +1,10 @@
using JetKarmaBot.Models; using JetKarmaBot.Models;
using Perfusion;
namespace JetKarmaBot.Services namespace JetKarmaBot.Services;
public class KarmaContextFactory
{ {
public class KarmaContextFactory [Inject] IContainer C { get; set; }
{
[Inject] IContainer C { get; set; }
public KarmaContext GetContext() => C.GetInstance<KarmaContext>(); public KarmaContext GetContext() => C.GetInstance<KarmaContext>();
}
} }

View File

@ -1,171 +1,166 @@
using System;
using System.Collections; using System.Collections;
using System.Collections.Generic;
using System.IO; using System.IO;
using System.Linq;
using System.Runtime.Serialization; using System.Runtime.Serialization;
using Newtonsoft.Json.Linq; using Newtonsoft.Json.Linq;
using NLog; using NLog;
using Perfusion;
namespace JetKarmaBot namespace JetKarmaBot;
public class Localization : IReadOnlyDictionary<string, Locale>
{ {
public class Localization : IReadOnlyDictionary<string, Locale> private Dictionary<string, Locale> locales = new Dictionary<string, Locale>();
public Localization(IContainer c)
{ {
private Dictionary<string, Locale> locales = new Dictionary<string, Locale>(); c.ResolveObject(this);
log.Info("Initializing...");
string langsFolder = "lang";
if (!Directory.Exists(langsFolder))
Directory.CreateDirectory(langsFolder);
public Localization(IContainer c) foreach (string langFilePath in Directory.EnumerateFiles(langsFolder, "*.json"))
{ {
c.ResolveObject(this); try
log.Info("Initializing...");
string langsFolder = "lang";
if (!Directory.Exists(langsFolder))
Directory.CreateDirectory(langsFolder);
foreach (string langFilePath in Directory.EnumerateFiles(langsFolder, "*.json"))
{ {
try string langName = Path.GetFileNameWithoutExtension(langFilePath);
{ string langKey = langName.ToLowerInvariant();
string langName = Path.GetFileNameWithoutExtension(langFilePath); locales[langKey] = new Locale(JObject.Parse(File.ReadAllText(langFilePath)), langKey);
string langKey = langName.ToLowerInvariant(); log.Debug("Found " + langName);
locales[langKey] = new Locale(JObject.Parse(File.ReadAllText(langFilePath)), langKey);
log.Debug("Found " + langName);
}
catch (Exception e)
{
log.Error($"Error while parsing {langFilePath}!");
log.Error(e);
}
} }
catch (Exception e)
if (locales.Any())
log.Info("Initialized!");
else
throw new FileNotFoundException($"No locales found in {langsFolder}!");
}
public Locale this[string locale]
{
get
{ {
locale = locale.ToLowerInvariant(); log.Error($"Error while parsing {langFilePath}!");
return locales[locale]; log.Error(e);
} }
} }
[Inject] if (locales.Any())
private Logger log; log.Info("Initialized!");
else
throw new FileNotFoundException($"No locales found in {langsFolder}!");
}
public Locale FindByCommonName(string name) public Locale this[string locale]
{ {
log.ConditionalTrace("Trying to find locale " + name); get
foreach (Locale l in locales.Values)
{
if (l.CommonNames.Contains(name))
{
log.ConditionalTrace("Found locale " + l.Name);
return l;
}
}
// Try to find as locale prefix
IEnumerable<Locale> matchinglocales = locales.Values.Where(x => x.Name.StartsWith(name + "-"));
if (matchinglocales.Count() > 1)
{
LocalizationException l = new LocalizationException("Too many locales");
l.Data["LocaleNames"] = matchinglocales.ToArray();
throw l;
}
else if (matchinglocales.Count() == 1)
return matchinglocales.First();
log.Warn("Failed to find locale " + name);
return null;
}
public bool ContainsLocale(string locale)
{ {
locale = locale.ToLowerInvariant(); locale = locale.ToLowerInvariant();
return locales.ContainsKey(locale); return locales[locale];
}
void Log(Exception e) => Console.WriteLine(e);
public IEnumerable<string> Keys => locales.Keys;
public IEnumerable<Locale> Values => locales.Values;
public int Count => locales.Count;
public bool ContainsKey(string key) => locales.ContainsKey(key.ToLower());
public bool TryGetValue(string key, out Locale value)
{
return locales.TryGetValue(key.ToLower(), out value);
}
public IEnumerator<KeyValuePair<string, Locale>> GetEnumerator()
{
return locales.GetEnumerator();
}
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
}
public class Locale : IReadOnlyDictionary<string, string>
{
private Dictionary<string, string> locale;
private string localeName;
private string[] commonNames;
private string note;
public Locale(JObject locale, string localeName)
{
this.locale = locale.Property("strings").Value.ToObject<Dictionary<string, string>>();
this.localeName = localeName;
this.commonNames = locale.Property("names").Value.ToObject<string[]>();
if (locale.ContainsKey("note")) this.note = locale.Property("note").Value.ToObject<string>();
}
public string[] CommonNames => commonNames;
public string Name => localeName;
public bool HasNote => note != null;
public string Note => note;
public IEnumerable<string> Keys => ((IReadOnlyDictionary<string, string>)locale).Keys;
public IEnumerable<string> Values => ((IReadOnlyDictionary<string, string>)locale).Values;
public int Count => ((IReadOnlyDictionary<string, string>)locale).Count;
public string this[string name] => locale.ContainsKey(name) ? locale[name] : "Unmapped locale key: " + name;
public bool ContainsKey(string key)
{
return ((IReadOnlyDictionary<string, string>)locale).ContainsKey(key);
}
public bool TryGetValue(string key, out string value)
{
return ((IReadOnlyDictionary<string, string>)locale).TryGetValue(key, out value);
}
public IEnumerator<KeyValuePair<string, string>> GetEnumerator()
{
return ((IReadOnlyDictionary<string, string>)locale).GetEnumerator();
}
IEnumerator IEnumerable.GetEnumerator()
{
return ((IReadOnlyDictionary<string, string>)locale).GetEnumerator();
} }
} }
[System.Serializable]
public class LocalizationException : Exception [Inject]
private Logger log;
public Locale FindByCommonName(string name)
{ {
public LocalizationException() { } log.ConditionalTrace("Trying to find locale " + name);
public LocalizationException(string message) : base(message) { } foreach (Locale l in locales.Values)
public LocalizationException(string message, Exception inner) : base(message, inner) { } {
protected LocalizationException( if (l.CommonNames.Contains(name))
SerializationInfo info, {
StreamingContext context) : base(info, context) { } log.ConditionalTrace("Found locale " + l.Name);
return l;
}
}
// Try to find as locale prefix
IEnumerable<Locale> matchinglocales = locales.Values.Where(x => x.Name.StartsWith(name + "-"));
if (matchinglocales.Count() > 1)
{
LocalizationException l = new LocalizationException("Too many locales");
l.Data["LocaleNames"] = matchinglocales.ToArray();
throw l;
}
else if (matchinglocales.Count() == 1)
return matchinglocales.First();
log.Warn("Failed to find locale " + name);
return null;
}
public bool ContainsLocale(string locale)
{
locale = locale.ToLowerInvariant();
return locales.ContainsKey(locale);
}
void Log(Exception e) => Console.WriteLine(e);
public IEnumerable<string> Keys => locales.Keys;
public IEnumerable<Locale> Values => locales.Values;
public int Count => locales.Count;
public bool ContainsKey(string key) => locales.ContainsKey(key.ToLower());
public bool TryGetValue(string key, out Locale value)
{
return locales.TryGetValue(key.ToLower(), out value);
}
public IEnumerator<KeyValuePair<string, Locale>> GetEnumerator()
{
return locales.GetEnumerator();
}
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
}
public class Locale : IReadOnlyDictionary<string, string>
{
private Dictionary<string, string> locale;
private string localeName;
private string[] commonNames;
private string note;
public Locale(JObject locale, string localeName)
{
this.locale = locale.Property("strings").Value.ToObject<Dictionary<string, string>>();
this.localeName = localeName;
this.commonNames = locale.Property("names").Value.ToObject<string[]>();
if (locale.ContainsKey("note")) this.note = locale.Property("note").Value.ToObject<string>();
}
public string[] CommonNames => commonNames;
public string Name => localeName;
public bool HasNote => note != null;
public string Note => note;
public IEnumerable<string> Keys => ((IReadOnlyDictionary<string, string>)locale).Keys;
public IEnumerable<string> Values => ((IReadOnlyDictionary<string, string>)locale).Values;
public int Count => ((IReadOnlyDictionary<string, string>)locale).Count;
public string this[string name] => locale.ContainsKey(name) ? locale[name] : "Unmapped locale key: " + name;
public bool ContainsKey(string key)
{
return ((IReadOnlyDictionary<string, string>)locale).ContainsKey(key);
}
public bool TryGetValue(string key, out string value)
{
return ((IReadOnlyDictionary<string, string>)locale).TryGetValue(key, out value);
}
public IEnumerator<KeyValuePair<string, string>> GetEnumerator()
{
return ((IReadOnlyDictionary<string, string>)locale).GetEnumerator();
}
IEnumerator IEnumerable.GetEnumerator()
{
return ((IReadOnlyDictionary<string, string>)locale).GetEnumerator();
} }
} }
[System.Serializable]
public class LocalizationException : Exception
{
public LocalizationException() { }
public LocalizationException(string message) : base(message) { }
public LocalizationException(string message, Exception inner) : base(message, inner) { }
protected LocalizationException(
SerializationInfo info,
StreamingContext context) : base(info, context) { }
}

View File

@ -1,22 +1,18 @@
using System;
using System.Linq;
using NLog; using NLog;
using Perfusion;
namespace JetKarmaBot namespace JetKarmaBot;
public class LogInfo : ObjectInfo
{ {
public class LogInfo : ObjectInfo public override ObjectInfo Clone() => new LogInfo();
{
public override ObjectInfo Clone() => new LogInfo();
public override object GetInstance(IContainer c, Type requester = null) public override object GetInstance(IContainer c, Type requester = null)
{ {
return LogManager.GetLogger(requester != null ? getTypeName(requester) : "<type unspecified>"); return LogManager.GetLogger(requester != null ? getTypeName(requester) : "<type unspecified>");
} }
private string getTypeName(Type t) private string getTypeName(Type t)
{ {
return (t.DeclaringType == null ? t.Namespace + "." + t.Name : getTypeName(t.DeclaringType) + "." + t.Name) return (t.DeclaringType == null ? t.Namespace + "." + t.Name : getTypeName(t.DeclaringType) + "." + t.Name)
+ (t.GenericTypeArguments.Length > 0 ? "<" + string.Join(",", t.GenericTypeArguments.Select(getTypeName)) + ">" : ""); + (t.GenericTypeArguments.Length > 0 ? "<" + string.Join(",", t.GenericTypeArguments.Select(getTypeName)) + ">" : "");
}
} }
} }

View File

@ -1,103 +1,100 @@
using System;
using Microsoft.Extensions.Logging; using Microsoft.Extensions.Logging;
using NLog; using NLog;
using Perfusion;
namespace JetKarmaBot namespace JetKarmaBot;
public class NLoggerFactory : ILoggerFactory
{ {
public class NLoggerFactory : ILoggerFactory [Inject]
private NLoggerProvider c;
public void AddProvider(ILoggerProvider provider)
{ {
[Inject]
private NLoggerProvider c;
public void AddProvider(ILoggerProvider provider)
{
}
public Microsoft.Extensions.Logging.ILogger CreateLogger(string categoryName) => c.CreateLogger(categoryName);
public void Dispose()
{
}
}
public class NLoggerProvider : ILoggerProvider
{
[Inject]
private Container c;
public Microsoft.Extensions.Logging.ILogger CreateLogger(string categoryName) => new LoggerVirtualizer(LogManager.GetLogger(categoryName));
public void Dispose()
{
}
} }
public class LoggerVirtualizer : Microsoft.Extensions.Logging.ILogger public Microsoft.Extensions.Logging.ILogger CreateLogger(string categoryName) => c.CreateLogger(categoryName);
public void Dispose()
{
}
}
public class NLoggerProvider : ILoggerProvider
{
[Inject]
private Container c;
public Microsoft.Extensions.Logging.ILogger CreateLogger(string categoryName) => new LoggerVirtualizer(LogManager.GetLogger(categoryName));
public void Dispose()
{
}
}
public class LoggerVirtualizer : Microsoft.Extensions.Logging.ILogger
{
private Logger logger;
public LoggerVirtualizer(Logger logger)
{
this.logger = logger;
}
private Microsoft.Extensions.Logging.LogLevel getAppropriate(NLog.LogLevel level)
{
if (level == NLog.LogLevel.Trace)
return Microsoft.Extensions.Logging.LogLevel.Trace;
else if (level == NLog.LogLevel.Debug)
return Microsoft.Extensions.Logging.LogLevel.Debug;
else if (level == NLog.LogLevel.Info)
return Microsoft.Extensions.Logging.LogLevel.Information;
else if (level == NLog.LogLevel.Warn)
return Microsoft.Extensions.Logging.LogLevel.Warning;
else if (level == NLog.LogLevel.Error)
return Microsoft.Extensions.Logging.LogLevel.Error;
else if (level == NLog.LogLevel.Fatal)
return Microsoft.Extensions.Logging.LogLevel.Critical;
else if (level == NLog.LogLevel.Off)
return Microsoft.Extensions.Logging.LogLevel.None;
else
return Microsoft.Extensions.Logging.LogLevel.None;
}
private NLog.LogLevel getAppropriate(Microsoft.Extensions.Logging.LogLevel level)
{
switch (level)
{
case Microsoft.Extensions.Logging.LogLevel.Trace:
return NLog.LogLevel.Trace;
case Microsoft.Extensions.Logging.LogLevel.Debug:
return NLog.LogLevel.Debug;
case Microsoft.Extensions.Logging.LogLevel.Information:
return NLog.LogLevel.Info;
case Microsoft.Extensions.Logging.LogLevel.Warning:
return NLog.LogLevel.Warn;
case Microsoft.Extensions.Logging.LogLevel.Error:
return NLog.LogLevel.Error;
case Microsoft.Extensions.Logging.LogLevel.Critical:
return NLog.LogLevel.Fatal;
default:
return NLog.LogLevel.Off;
}
}
public IDisposable BeginScope<TState>(TState state) => new SomeDisposable();
public bool IsEnabled(Microsoft.Extensions.Logging.LogLevel logLevel)
{
return logger.IsEnabled(getAppropriate(logLevel));
}
public void Log<TState>(Microsoft.Extensions.Logging.LogLevel logLevel, EventId eventId, TState state, Exception exception, Func<TState, Exception, string> formatter)
{
string message = state.ToString();
if (exception != null) logger.Log(getAppropriate(logLevel), exception, formatter(state, exception));
else logger.Log(getAppropriate(logLevel), state);
}
}
public class SomeDisposable : IDisposable
{
public void Dispose()
{ {
private Logger logger;
public LoggerVirtualizer(Logger logger)
{
this.logger = logger;
}
private Microsoft.Extensions.Logging.LogLevel getAppropriate(NLog.LogLevel level)
{
if (level == NLog.LogLevel.Trace)
return Microsoft.Extensions.Logging.LogLevel.Trace;
else if (level == NLog.LogLevel.Debug)
return Microsoft.Extensions.Logging.LogLevel.Debug;
else if (level == NLog.LogLevel.Info)
return Microsoft.Extensions.Logging.LogLevel.Information;
else if (level == NLog.LogLevel.Warn)
return Microsoft.Extensions.Logging.LogLevel.Warning;
else if (level == NLog.LogLevel.Error)
return Microsoft.Extensions.Logging.LogLevel.Error;
else if (level == NLog.LogLevel.Fatal)
return Microsoft.Extensions.Logging.LogLevel.Critical;
else if (level == NLog.LogLevel.Off)
return Microsoft.Extensions.Logging.LogLevel.None;
else
return Microsoft.Extensions.Logging.LogLevel.None;
}
private NLog.LogLevel getAppropriate(Microsoft.Extensions.Logging.LogLevel level)
{
switch (level)
{
case Microsoft.Extensions.Logging.LogLevel.Trace:
return NLog.LogLevel.Trace;
case Microsoft.Extensions.Logging.LogLevel.Debug:
return NLog.LogLevel.Debug;
case Microsoft.Extensions.Logging.LogLevel.Information:
return NLog.LogLevel.Info;
case Microsoft.Extensions.Logging.LogLevel.Warning:
return NLog.LogLevel.Warn;
case Microsoft.Extensions.Logging.LogLevel.Error:
return NLog.LogLevel.Error;
case Microsoft.Extensions.Logging.LogLevel.Critical:
return NLog.LogLevel.Fatal;
default:
return NLog.LogLevel.Off;
}
}
public IDisposable BeginScope<TState>(TState state) => new SomeDisposable();
public bool IsEnabled(Microsoft.Extensions.Logging.LogLevel logLevel)
{
return logger.IsEnabled(getAppropriate(logLevel));
}
public void Log<TState>(Microsoft.Extensions.Logging.LogLevel logLevel, EventId eventId, TState state, Exception exception, Func<TState, Exception, string> formatter)
{
string message = state.ToString();
if (exception != null) logger.Log(getAppropriate(logLevel), exception, formatter(state, exception));
else logger.Log(getAppropriate(logLevel), state);
}
}
public class SomeDisposable : IDisposable
{
public void Dispose()
{
}
} }
} }