nadekobot/src/NadekoBot/Services/Impl/BehaviorExecutor.cs

using NadekoBot.Common.ModuleBehaviors;
using NadekoBot.Extensions;
using Microsoft.Extensions.DependencyInjection;

namespace NadekoBot.Services;

public sealed class BehaviorExecutor : IBehaviourExecutor, INService
{
    private readonly IServiceProvider _services;
    private IEnumerable<ILateExecutor> _lateExecutors;
    private IEnumerable<ILateBlocker> _lateBlockers;
    private IEnumerable<IEarlyBehavior> _earlyBehaviors;
    private IEnumerable<IInputTransformer> _transformers;

    public BehaviorExecutor(IServiceProvider services)
    {
        _services = services;
    }

    public void Initialize()
    {
        _lateExecutors = _services.GetServices<ILateExecutor>();
        _lateBlockers = _services.GetServices<ILateBlocker>();
        _earlyBehaviors = _services.GetServices<IEarlyBehavior>()
            .OrderByDescending(x => x.Priority);
        _transformers = _services.GetServices<IInputTransformer>();
    }

    public async Task<bool> RunEarlyBehavioursAsync(SocketGuild guild, IUserMessage usrMsg)
    {
        foreach (var beh in _earlyBehaviors)
        {
            if (await beh.RunBehavior(guild, usrMsg))
            {
                return true;
            }
        }

        return false;
    }

    public async Task<string> RunInputTransformersAsync(SocketGuild guild, IUserMessage usrMsg)
    {
        var messageContent = usrMsg.Content;
        foreach (var exec in _transformers)
        {
            string newContent;
            if ((newContent = await exec.TransformInput(guild, usrMsg.Channel, usrMsg.Author, messageContent)) 
                != messageContent.ToLowerInvariant())
            {
                messageContent = newContent;
                break;
            }
        }

        return messageContent;
    }

    public async Task<bool> RunLateBlockersAsync(ICommandContext ctx, CommandInfo cmd)
    {
        foreach (var exec in _lateBlockers)
        {
            if (await exec.TryBlockLate(ctx, cmd.Module.GetTopLevelModule().Name, cmd))
            {
                Log.Information("Late blocking User [{0}] Command: [{1}] in [{2}]", 
                    ctx.User,
                    cmd.Aliases[0],
                    exec.GetType().Name);
                return true;
            }
        }

        return false;
    }

    public async Task RunLateExecutorsAsync(SocketGuild guild, IUserMessage usrMsg)
    {
        foreach (var exec in _lateExecutors)
        {
            try
            {
                await exec.LateExecute(guild, usrMsg).ConfigureAwait(false);
            }
            catch (Exception ex)
            {
                Log.Error(ex, "Error in {TypeName} late executor: {ErrorMessage}",
                    exec.GetType().Name,
                    ex.Message);
            }
        }
    }
}