public static IHostBuilder ConfigureDiscordBot <TDiscordBot, TDiscordBotConfiguration>(this IHostBuilder builder, Action <HostBuilderContext, DiscordBotHostingContext> configure = null)
where TDiscordBot : DiscordBot
where TDiscordBotConfiguration : DiscordBotBaseConfiguration, new()
{
builder.ConfigureServices((context, services) =>
{
var discordContext = new DiscordBotHostingContext();
configure?.Invoke(context, discordContext);
services.AddDiscordBot <TDiscordBot>();
services.ConfigureDiscordClient(context, discordContext);
services.ConfigureDiscordBot <TDiscordBotConfiguration>(context, discordContext);
});
return(builder);
}
public static void ConfigureDiscordBot <TBotConfiguration>(this IServiceCollection services, HostBuilderContext context, DiscordBotHostingContext discordContext)
where TBotConfiguration : DiscordBotBaseConfiguration, new()
{
services.Configure <CommandServiceConfiguration>(x => x.CooldownBucketKeyGenerator = CooldownBucketKeyGenerator.Instance);
if (discordContext.OwnerIds != null)
{
services.Configure <TBotConfiguration>(x => x.OwnerIds = discordContext.OwnerIds);
}
if (!discordContext.UseMentionPrefix && (discordContext.Prefixes == null || !discordContext.Prefixes.Any()))
{
throw new InvalidOperationException($"No prefixes were specified for the {nameof(DefaultPrefixProvider)}.");
}
services.AddSingleton <IConfigureOptions <DefaultPrefixProviderConfiguration> >(services => new ConfigureOptions <DefaultPrefixProviderConfiguration>(x =>
{
var prefixes = new List <IPrefix>();
if (discordContext.UseMentionPrefix)
{
if (services.GetService <Token>() is not BotToken botToken)
{
throw new InvalidOperationException("The mention prefix cannot be used without a bot token set.");
}
prefixes.Add(new MentionPrefix(botToken.Id));
}
if (discordContext.Prefixes != null)
{
prefixes.AddRange(discordContext.Prefixes.Select(x => new StringPrefix(x)));
}
x.Prefixes = prefixes;
}));
services.AddHostedService <DiscordBotRunnerService>();
}
