2017-04-12 21:49:04 +02:00
|
|
|
|
using System;
|
|
|
|
|
using System.Reflection;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
using Discord;
|
|
|
|
|
using Discord.Commands;
|
|
|
|
|
using Discord.WebSocket;
|
2017-04-13 00:03:03 +02:00
|
|
|
|
using Geekbot.net.Lib;
|
2017-04-13 13:22:30 +02:00
|
|
|
|
using Geekbot.net.Modules;
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
|
|
|
|
namespace Geekbot.net
|
|
|
|
|
{
|
|
|
|
|
class Program
|
|
|
|
|
{
|
2017-04-17 23:58:43 +02:00
|
|
|
|
private CommandService commands;
|
2017-04-12 21:49:04 +02:00
|
|
|
|
private DiscordSocketClient client;
|
|
|
|
|
private DependencyMap map;
|
2017-04-18 11:00:38 +02:00
|
|
|
|
private IRedisClient redis;
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
|
|
|
|
private static void Main(string[] args)
|
|
|
|
|
{
|
2017-04-17 16:58:48 +02:00
|
|
|
|
Console.WriteLine(@" ____ _____ _____ _ ______ ___ _____");
|
|
|
|
|
Console.WriteLine(@" / ___| ____| ____| |/ / __ ) / _ \\_ _|");
|
|
|
|
|
Console.WriteLine(@"| | _| _| | _| | ' /| _ \| | | || |");
|
|
|
|
|
Console.WriteLine(@"| |_| | |___| |___| . \| |_) | |_| || |");
|
|
|
|
|
Console.WriteLine(@" \____|_____|_____|_|\_\____/ \___/ |_|");
|
2017-04-12 22:43:52 +02:00
|
|
|
|
Console.WriteLine("=========================================");
|
|
|
|
|
Console.WriteLine("Starting...");
|
2017-04-17 12:55:20 +02:00
|
|
|
|
|
2017-04-17 16:58:48 +02:00
|
|
|
|
//Task.WaitAll(BootTasks.CheckSettingsFile());
|
2017-04-18 14:08:56 +02:00
|
|
|
|
if (args.Length == 2 && args[0] == "--parse-db")
|
|
|
|
|
{
|
|
|
|
|
BootTasks.ParseOldDatabase(args[1]);
|
|
|
|
|
Environment.Exit(1);
|
|
|
|
|
}
|
2017-04-13 13:22:30 +02:00
|
|
|
|
Task.WaitAll(new Program().MainAsync());
|
2017-04-12 21:49:04 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task MainAsync()
|
|
|
|
|
{
|
|
|
|
|
client = new DiscordSocketClient();
|
|
|
|
|
commands = new CommandService();
|
2017-04-18 11:00:38 +02:00
|
|
|
|
redis = new RedisClient();
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
2017-04-18 11:00:38 +02:00
|
|
|
|
var token = redis.Client.StringGet("discordToken");
|
2017-04-17 23:58:43 +02:00
|
|
|
|
if (token.IsNullOrEmpty)
|
|
|
|
|
{
|
|
|
|
|
Console.Write("Your bot Token: ");
|
|
|
|
|
var newToken = Console.ReadLine();
|
2017-04-18 11:00:38 +02:00
|
|
|
|
redis.Client.StringSet("discordToken", newToken);
|
2017-04-17 23:58:43 +02:00
|
|
|
|
token = newToken;
|
|
|
|
|
|
|
|
|
|
Console.Write("Bot Owner User ID: ");
|
|
|
|
|
var ownerId = Console.ReadLine();
|
2017-04-18 11:00:38 +02:00
|
|
|
|
redis.Client.StringSet("botOwner", ownerId);
|
2017-04-17 23:58:43 +02:00
|
|
|
|
}
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
|
|
|
|
map = new DependencyMap();
|
2017-04-13 13:22:30 +02:00
|
|
|
|
map.Add<ICatClient>(new CatClient());
|
2017-04-18 14:08:56 +02:00
|
|
|
|
map.Add(redis);
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
|
|
|
|
await InstallCommands();
|
2017-04-12 22:43:52 +02:00
|
|
|
|
Console.WriteLine("Connecting to Discord...");
|
2017-04-12 21:49:04 +02:00
|
|
|
|
await client.LoginAsync(TokenType.Bot, token);
|
|
|
|
|
await client.StartAsync();
|
2017-04-12 22:43:52 +02:00
|
|
|
|
Console.WriteLine("Done and ready for use...\n");
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
|
|
|
|
await Task.Delay(-1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task InstallCommands()
|
|
|
|
|
{
|
|
|
|
|
client.MessageReceived += HandleCommand;
|
|
|
|
|
client.MessageReceived += HandleMessageReceived;
|
2017-04-12 22:43:52 +02:00
|
|
|
|
|
2017-04-14 21:36:01 +02:00
|
|
|
|
client.UserJoined += HandleUserJoined;
|
|
|
|
|
|
2017-04-12 21:49:04 +02:00
|
|
|
|
await commands.AddModulesAsync(Assembly.GetEntryAssembly());
|
|
|
|
|
}
|
|
|
|
|
public async Task HandleCommand(SocketMessage messageParam)
|
|
|
|
|
{
|
|
|
|
|
var message = messageParam as SocketUserMessage;
|
|
|
|
|
if (message == null) return;
|
|
|
|
|
int argPos = 0;
|
2017-04-17 16:58:48 +02:00
|
|
|
|
if (message.ToString().ToLower().Equals("ping"))
|
|
|
|
|
{
|
|
|
|
|
await message.Channel.SendMessageAsync("pong");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (message.ToString().ToLower().Equals("hui"))
|
2017-04-12 21:49:04 +02:00
|
|
|
|
{
|
2017-04-17 16:58:48 +02:00
|
|
|
|
await message.Channel.SendMessageAsync("hui!!!");
|
|
|
|
|
return;
|
2017-04-12 21:49:04 +02:00
|
|
|
|
}
|
2017-04-17 16:58:48 +02:00
|
|
|
|
if (!(message.HasCharPrefix('!', ref argPos) || message.HasMentionPrefix(client.CurrentUser, ref argPos))) return;
|
|
|
|
|
var context = new CommandContext(client, message);
|
|
|
|
|
commands.ExecuteAsync(context, argPos, map);
|
|
|
|
|
//var result = await commands.ExecuteAsync(context, argPos, map);
|
|
|
|
|
//if (!result.IsSuccess)
|
|
|
|
|
//{
|
|
|
|
|
// await context.Channel.SendMessageAsync(result.ErrorReason);
|
|
|
|
|
//}
|
2017-04-12 21:49:04 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task HandleMessageReceived(SocketMessage messsageParam)
|
|
|
|
|
{
|
|
|
|
|
var message = messsageParam;
|
|
|
|
|
if (message == null) return;
|
2017-04-17 23:58:43 +02:00
|
|
|
|
if (message.Author.Username.Equals(client.CurrentUser.Username)) return;
|
2017-04-12 21:49:04 +02:00
|
|
|
|
|
2017-04-17 16:58:48 +02:00
|
|
|
|
var channel = (SocketGuildChannel)message.Channel;
|
2017-04-13 00:03:03 +02:00
|
|
|
|
|
|
|
|
|
Console.WriteLine(channel.Guild.Name + " - " + message.Channel + " - " + message.Author.Username + " - " + message.Content);
|
|
|
|
|
|
2017-04-18 11:00:38 +02:00
|
|
|
|
var statsRecorder = new StatsRecorder(message, redis);
|
2017-04-14 22:18:22 +02:00
|
|
|
|
await statsRecorder.UpdateUserRecordAsync();
|
|
|
|
|
await statsRecorder.UpdateGuildRecordAsync();
|
2017-04-14 21:36:01 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task HandleUserJoined(SocketGuildUser user)
|
|
|
|
|
{
|
|
|
|
|
if (!user.IsBot)
|
|
|
|
|
{
|
2017-04-18 11:00:38 +02:00
|
|
|
|
var message = redis.Client.StringGet(user.Guild.Id + "-welcomeMsg");
|
2017-04-14 22:18:22 +02:00
|
|
|
|
if (!message.IsNullOrEmpty)
|
|
|
|
|
{
|
|
|
|
|
message = message.ToString().Replace("$user", user.Mention);
|
|
|
|
|
await user.Guild.DefaultChannel.SendMessageAsync(message);
|
|
|
|
|
}
|
2017-04-14 21:36:01 +02:00
|
|
|
|
}
|
2017-04-12 21:49:04 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|