using JetKarmaBot.Commands; using NLog; using Perfusion; using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using Telegram.Bot; namespace JetKarmaBot.Services.Handling { public class ChatCommandRouter : IRequestHandler { public class Feature { public Type CommandType; public ChatCommandRouter Router; public bool Succeded; } public Telegram.Bot.Types.User Me { get; private set; } [Inject] private Logger log; [Inject] private TelegramBotClient Client { get; set; } public async Task Start() { Me = await Client.GetMeAsync(); } public Task Handle(RequestContext ctx, Func next) { log.Debug("Message received"); CommandString cmd = ctx.Command; Feature feature = new Feature() { Router = this }; ctx.AddFeature(feature); try { if (commands.ContainsKey(cmd.Command)) { feature.CommandType = commands[cmd.Command].GetType(); log.Debug($"Handling message via {feature.CommandType.Name}"); 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; } } 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(); } }