using JetKarmaBot.Commands; using JetKarmaBot.Services; using Microsoft.EntityFrameworkCore; using NLog; using Perfusion; using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using Telegram.Bot; using Telegram.Bot.Args; namespace JetKarmaBot { public class ChatCommandRouter { Telegram.Bot.Types.User BotUser { get; } [Inject] private Logger log; [Inject] private KarmaContextFactory Db; [Inject] private TelegramBotClient Client { get; set; } public ChatCommandRouter(Telegram.Bot.Types.User botUser) { BotUser = botUser; } public async Task Execute(object sender, MessageEventArgs args) { log.Debug("Message received"); var text = args.Message.Text; if (CommandString.TryParse(text, out var cmd)) { if (cmd.UserName != null && cmd.UserName != BotUser.Username) { // directed not at us! log.Debug("Message not directed at us"); return false; } try { if (commands.ContainsKey(cmd.Command)) { log.Debug($"Handling message via {commands[cmd.Command].GetType().Name}"); return await commands[cmd.Command].Execute(cmd, args); } } catch (Exception e) { log.Error($"Error while handling command {cmd.Command}!"); log.Error(e); await ReportToAdministratorChats($"Error while handling command {cmd.Command}!\n{e.ToString()}"); } } return false; } public async Task ReportToAdministratorChats(string text) { using (var db = Db.GetContext()) { await Task.WhenAll( (await db.Chats .Where(x => x.IsAdministrator) .Select(x => x.ChatId) .ToArrayAsync()) .Select(x => Client.SendTextMessageAsync(x, text))); foreach (long chatid in db.Chats.Where(x => x.IsAdministrator).Select(x => x.ChatId)) { await Client.SendTextMessageAsync(chatid, text); } } } 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 pieces = new List(); foreach (IChatCommand c in commands.Values.Distinct()) { string build = ""; List 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 ? "]" : ""))) + " " + getLocalizedCMDDesc(c, loc) + ""; pieces.Add(build); } return string.Join("\n", pieces); } internal string GetHelpTextFor(string commandname, Locale loc) { IChatCommand c = commands[commandname]; string build = ""; List 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 ? "]" : ""))) + " " + getLocalizedCMDDesc(c, loc) + "\n"; build += string.Join("\n", c.Arguments.Select(ca => (!ca.Required ? "[" : "") + ca.Name + (!ca.Required ? "]" : "") + ": " + getLocalizedCMDArgDesc(ca, loc) + "")); 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 commands = new Dictionary(); } }