DocSpace-client/common/services/ASC.TelegramService/Core/Core.cs

namespace ASC.TelegramService.Core;

public class TelegramCommand
{
    public string CommandName { get; private set; }
    public string[] Args { get; private set; }
    public Message Message { get; private set; }
    public User User { get { return Message.From; } }
    public Chat Chat { get { return Message.Chat; } }

    public TelegramCommand(Message msg, string cmdName, string[] args = null)
    {
        Message = msg;
        CommandName = cmdName;
        Args = args;
    }
}

[Singletone]
public class CommandModule
{
    private readonly Regex _cmdReg = new Regex(@"^\/([^\s]+)\s?(.*)");
    private readonly Regex _argsReg = new Regex(@"[^""\s]\S*|"".+?""");

    private readonly Dictionary<string, MethodInfo> _commands = new Dictionary<string, MethodInfo>();
    private readonly Dictionary<string, Type> _contexts = new Dictionary<string, Type>();
    private readonly Dictionary<Type, ParamParser> _parsers = new Dictionary<Type, ParamParser>();

    private readonly IServiceScopeFactory _scopeFactory;
    private readonly ILog _log;

    public CommandModule(IOptionsMonitor<ILog> options, IServiceScopeFactory scopeFactory)
    {
        _scopeFactory = scopeFactory;
        _log = options.CurrentValue;

        var assembly = Assembly.GetExecutingAssembly();

        foreach (var t in assembly.GetExportedTypes())
        {
            if (t.IsAbstract) continue;

            if (t.IsSubclassOf(typeof(CommandContext)))
            {
                foreach (var method in t.GetRuntimeMethods())
                {
                    if (method.IsPublic && Attribute.IsDefined(method, typeof(CommandAttribute)))
                    {
                        var attr = method.GetCustomAttribute<CommandAttribute>();
                        _commands.Add(attr.Name, method);
                        _contexts.Add(attr.Name, t);
                    }
                }
            }

            if (t.IsSubclassOf(typeof(ParamParser)) && Attribute.IsDefined(t, typeof(ParamParserAttribute)))
            {
                _parsers.Add(t.GetCustomAttribute<ParamParserAttribute>().Type, (ParamParser)Activator.CreateInstance(t));
            }
        }
    }

    private TelegramCommand ParseCommand(Message msg)
    {
        var reg = _cmdReg.Match(msg.Text);
        var args = _argsReg.Matches(reg.Groups[2].Value);

        return new TelegramCommand(msg, reg.Groups[1].Value.ToLowerInvariant(), args.Count > 0 ? args.Select(a => a.Value).ToArray() : null);
    }

    private object[] ParseParams(MethodInfo cmd, string[] args)
    {
        var parsedParams = new List<object>();

        var cmdArgs = cmd.GetParameters();

        if (cmdArgs.Length > 0 && args == null || cmdArgs.Length != args.Length) throw new Exception("Wrong parameters count");
        for (var i = 0; i < cmdArgs.Length; i++)
        {
            var type = cmdArgs[i].ParameterType;
            var arg = args[i];
            if (type == typeof(string))
            {
                parsedParams.Add(arg);
                continue;
            }

            if (!_parsers.ContainsKey(type)) throw new Exception(string.Format("No parser found for type '{0}'", type));

            parsedParams.Add(_parsers[type].FromString(arg));
        }

        return parsedParams.ToArray();
    }

    public async Task HandleCommand(Message msg, TelegramBotClient client, int tenantId)
    {
        try
        {
            var cmd = ParseCommand(msg);

            if (!_commands.ContainsKey(cmd.CommandName)) throw new Exception($"No handler found for command '{cmd.CommandName}'");

            var command = _commands[cmd.CommandName];
            var context = (CommandContext)_scopeFactory.CreateScope().ServiceProvider.GetService(_contexts[cmd.CommandName]);
            var param = ParseParams(command, cmd.Args);

            context.Context = cmd;
            context.Client = client;
            context.TenantId = tenantId;
            await Task.FromResult(command.Invoke(context, param));
        }
        catch (Exception ex)
        {
            _log.DebugFormat("Couldn't handle ({0}) message ({1})", msg.Text, ex.Message);
        }
    }
}

public abstract class CommandContext
{
    public ITelegramBotClient Client { get; set; }
    public TelegramCommand Context { get; set; }
    public int TenantId { get; set; }

    protected async Task ReplyAsync(string message)
    {
        await Client.SendTextMessageAsync(Context.Chat, message);
    }
}

public abstract class ParamParser
{
    protected Type type;

    protected ParamParser(Type type)
    {
        this.type = type;
    }

    public abstract object FromString(string arg);
    public abstract string ToString(object arg);
}

public abstract class ParamParser<T> : ParamParser
{
    protected ParamParser() : base(typeof(T)) { }

    public override abstract object FromString(string arg);
    public override abstract string ToString(object arg);
}