-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathBot.cs
150 lines (129 loc) · 5.61 KB
/
Bot.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
using DSharpPlus;
using DSharpPlus.EventArgs;
using DSharpPlus.SlashCommands;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using SchedulingAssistant.Services;
using Serilog;
using System;
namespace SchedulingAssistant
{
public class Bot
{
private DiscordClient? _Client;
private ILogger<Bot>? _Logger;
private static ServiceProvider ConfigureServices(string Token, LogLevel LogLevel = LogLevel.Debug)
{
if (Token.Count() == 0)
{
throw new Exception("Missing Discord Bot Token. Verify Configuration");
}
var logFactory = new LoggerFactory().AddSerilog();
var config = new DiscordConfiguration()
{
Token = Token,
TokenType = TokenType.Bot,
Intents = DiscordIntents.AllUnprivileged,
MinimumLogLevel = LogLevel.Debug,
AutoReconnect = true,
LoggerFactory = logFactory
};
var services = new ServiceCollection()
.AddSingleton(x => new DiscordClient(config))
.AddSingleton<CommandHandler>()
.AddSingleton<DatabaseHandler>()
.AddSingleton<InteractivityHandler>()
.AddSingleton<Scheduler>()
.AddLogging(configure => configure.AddSerilog());
switch (LogLevel)
{
case LogLevel.Information:
{
services.Configure<LoggerFilterOptions>(options => options.MinLevel = LogLevel.Information);
break;
}
case LogLevel.Error:
{
services.Configure<LoggerFilterOptions>(options => options.MinLevel = LogLevel.Error);
break;
}
case LogLevel.Debug:
{
services.Configure<LoggerFilterOptions>(options => options.MinLevel = LogLevel.Debug);
break;
}
default:
{
services.Configure<LoggerFilterOptions>(options => options.MinLevel = LogLevel.Error);
break;
}
}
var serviceProvider = services.BuildServiceProvider();
return serviceProvider;
}
// this method logs in and starts the client
public async Task StartAsync(string Token)
{
await using var services = ConfigureServices(Token);
_Client = services.GetRequiredService<DiscordClient>();
_Logger = services.GetRequiredService<ILogger<Bot>>();
if (_Client == null)
{
throw new Exception("DiscordClient has not been configured");
}
await services.GetRequiredService<DatabaseHandler>().Initalize();
await services.GetRequiredService<CommandHandler>().Initalize();
await services.GetRequiredService<InteractivityHandler>().Initalize();
await services.GetRequiredService<Scheduler>().Initalize();
_Client.Ready += OnReady;
_Client.GuildAvailable += OnGuildAvailable;
_Client.GuildUnavailable += OnGuildUnavailable; ;
_Client.GuildDownloadCompleted += OnGuildDownloadComplete;
_Client.Heartbeated += OnHeartbeat;
_Client.Zombied += OnZombied;
await _Client.ConnectAsync();
await Task.Delay(Timeout.Infinite);
}
private Task OnZombied(DiscordClient sender, ZombiedEventArgs e)
{
_Logger?.LogCritical($"{sender.CurrentUser.Username} is still dead! Failed to respond to heartbeat. Recieved {e.Failures} Failed Heartbeats.");
throw new Exception($"{sender.CurrentUser.Username} is still dead! Failed to respond to heartbeat. Recieved {e.Failures} Failed Heartbeats.");
}
private Task OnHeartbeat(DiscordClient sender, HeartbeatEventArgs e)
{
_Logger?.LogInformation($"{sender.CurrentUser.Username} is still alive! Ping is {e.Ping} ms to Discord servers");
return Task.CompletedTask;
}
private Task OnGuildUnavailable(DiscordClient sender, GuildDeleteEventArgs e)
{
_Logger?.LogDebug($"Guild Id: {e.Guild.Id} Guild Name: {e.Guild.Name} is Unavailable!");
return Task.CompletedTask;
}
private Task OnGuildDownloadComplete(DiscordClient sender, GuildDownloadCompletedEventArgs e)
{
// Logs the bot name and all the servers that it's connected to
_Logger?.LogInformation($"Connected to these servers as '{sender.CurrentUser.Username}': ");
foreach (var guild in sender.Guilds)
_Logger?.LogInformation($"\t Guild Id: {guild.Key} Guild Name: {guild.Value.Name}");
return Task.CompletedTask;
}
private Task OnGuildAvailable(DiscordClient sender, GuildCreateEventArgs e)
{
_Logger?.LogDebug($"Guild Id: {e.Guild.Id} Guild Name: {e.Guild.Name} is Available!");
return Task.CompletedTask;
}
private Task OnReady(DiscordClient sender, ReadyEventArgs e)
{
_Logger?.LogInformation($"{sender.CurrentUser.Username} is ready!");
return Task.CompletedTask;
}
public Task StopAsync()
{
if (_Client != null)
{
this._Client.DisconnectAsync();
}
return Task.CompletedTask;
}
}
}