daan/geekbot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Now recovers from a disconnect

Browse source
This commit is contained in:
Runebaas 2017-04-25 21:56:09 +02:00
parent 2af916e4d1
commit cb8423373a
1 changed files with 24 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
Geekbot.net/Program.cs
View file

@ -1,5 +1,6 @@
using System; using System;
using System.Reflection; using System.Reflection;
using System.Runtime.InteropServices.ComTypes;
using System.Threading.Tasks; using System.Threading.Tasks;
using Discord; using Discord;
using Discord.Commands; using Discord.Commands;
@ -7,6 +8,7 @@ using Discord.WebSocket;
using Geekbot.net.Lib; using Geekbot.net.Lib;
using Geekbot.net.Lib.IClients; using Geekbot.net.Lib.IClients;
using Geekbot.net.Modules; using Geekbot.net.Modules;
using StackExchange.Redis;
namespace Geekbot.net namespace Geekbot.net
{ {
@ -16,6 +18,7 @@ namespace Geekbot.net
private DiscordSocketClient client; private DiscordSocketClient client;
private DependencyMap map; private DependencyMap map;
private IRedisClient redis; private IRedisClient redis;
private RedisValue token;
private static void Main(string[] args) private static void Main(string[] args)
{ {
@ -36,7 +39,7 @@ namespace Geekbot.net
commands = new CommandService(); commands = new CommandService();
redis = new RedisClient(); redis = new RedisClient();
var token = redis.Client.StringGet("discordToken"); token = redis.Client.StringGet("discordToken");
if (token.IsNullOrEmpty) if (token.IsNullOrEmpty)
{ {
Console.Write("Your bot Token: "); Console.Write("Your bot Token: ");
@ -55,6 +58,14 @@ namespace Geekbot.net
map.Add<IRandomClient>(new RandomClient()); map.Add<IRandomClient>(new RandomClient());
Console.WriteLine("Connecting to Discord..."); Console.WriteLine("Connecting to Discord...");
await Login();
await Task.Delay(-1);
}
public async Task Login()
{
try try
{ {
await client.LoginAsync(TokenType.Bot, token); await client.LoginAsync(TokenType.Bot, token);
@ -66,8 +77,18 @@ namespace Geekbot.net
Console.WriteLine("Could not connect to discord..."); Console.WriteLine("Could not connect to discord...");
Environment.Exit(1); Environment.Exit(1);
} }
}
await Task.Delay(-1); public async Task Reconnect(Exception exception)
{
Console.WriteLine("=========================================");
Console.WriteLine("Geekbot Disconnected from the Discord Gateway...");
Console.WriteLine(exception.Message);
Console.WriteLine("Attempting Reconnect...");
Console.WriteLine("=========================================");
await client.StopAsync();
System.Threading.Thread.Sleep(10000);
await Login();
} }
public async Task FinishStartup() public async Task FinishStartup()
@ -80,6 +101,7 @@ namespace Geekbot.net
client.MessageReceived += HandleCommand; client.MessageReceived += HandleCommand;
client.MessageReceived += HandleMessageReceived; client.MessageReceived += HandleMessageReceived;
client.UserJoined += HandleUserJoined; client.UserJoined += HandleUserJoined;
client.Disconnected += Reconnect;
await commands.AddModulesAsync(Assembly.GetEntryAssembly()); await commands.AddModulesAsync(Assembly.GetEntryAssembly());
Console.WriteLine("Done and ready for use...\n"); Console.WriteLine("Done and ready for use...\n");