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

Async Stuff, !rank doesn't mention anymore, setup finalizer, minor tweaks

Browse source
This commit is contained in:
Runebaas 2017-09-29 20:30:00 +02:00
parent b1f4e8ad93
commit a549950ad9
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
4 changed files with 59 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
Geekbot.net/Modules/GuildInfo.cs
View file

@ -22,12 +22,10 @@ namespace Geekbot.net.Modules
public async Task getInfo()
{
var eb = new EmbedBuilder();
// eb.WithAuthor(new EmbedAuthorBuilder()
// .WithIconUrl(Context.Guild.IconUrl)
// .WithName(Context.Guild.Name));
eb.WithAuthor(new EmbedAuthorBuilder()
.WithIconUrl(Context.Guild.IconUrl)
.WithName(Context.Guild.Name));
eb.WithColor(new Color(110, 204, 147));
eb.Title = $":bar_chart: {Context.Guild.Name}";
eb.ThumbnailUrl = Context.Guild.IconUrl;
var created = Context.Guild.CreatedAt;
var age = Math.Floor((DateTime.Now - created).TotalDays);

7
Geekbot.net/Modules/Quote.cs
View file

@ -53,7 +53,8 @@ namespace Geekbot.net.Modules
var quote = createQuoteObject(lastMessage);
var quoteStore = JsonConvert.SerializeObject(quote);
redis.SetAdd($"{Context.Guild.Id}:Quotes", quoteStore);
await ReplyAsync("Quote Added");
var embed = quoteBuilder(quote);
await ReplyAsync("**Quote Added**", false, embed.Build());
}
catch (Exception e)
{
@ -71,7 +72,9 @@ namespace Geekbot.net.Modules
var quote = createQuoteObject(message);
var quoteStore = JsonConvert.SerializeObject(quote);
redis.SetAdd($"{Context.Guild.Id}:Quotes", quoteStore);
await ReplyAsync("Quote Added");
var embed = quoteBuilder(quote);
await ReplyAsync("**Quote Added**", false, embed.Build());
}
catch (Exception e)
{

12
Geekbot.net/Modules/UserInfo.cs
View file

@ -39,11 +39,9 @@ namespace Geekbot.net.Modules
var percent = Math.Round((double) (100 * messages) / guildMessages, 2);
var eb = new EmbedBuilder();
// eb.WithAuthor(new EmbedAuthorBuilder()
// .WithIconUrl(userInfo.GetAvatarUrl())
// .WithName(userInfo.Username));
eb.Title = $":bar_chart: {userInfo.Username}#{userInfo.Discriminator}";
eb.ThumbnailUrl = userInfo.GetAvatarUrl();
eb.WithAuthor(new EmbedAuthorBuilder()
.WithIconUrl(userInfo.GetAvatarUrl())
.WithName(userInfo.Username));
eb.WithColor(new Color(221, 255, 119));
eb.AddField("Discordian Since",
@ -109,7 +107,7 @@ namespace Geekbot.net.Modules
}
var highScore = new StringBuilder();
if (failedToRetrieveUser) highScore.AppendLine(":warning: I couldn't get all userdata, i mentioned the missing ones, sorry!\n");
if (failedToRetrieveUser) highScore.AppendLine(":warning: I couldn't get all userdata, sorry! (bugfix coming soon:tm:)\n");
highScore.AppendLine($":bar_chart: **Highscore for {Context.Guild.Name}**");
var highscorePlace = 1;
foreach (var user in highscoreUsers)
@ -123,7 +121,7 @@ namespace Geekbot.net.Modules
else
{
highScore.AppendLine(
$"{NumerToEmoji(highscorePlace)} **<@{user.Key.Id}>** - {percent}% of total - {user.Value} messages");
$"{NumerToEmoji(highscorePlace)} **{user.Key.Id}** - {percent}% of total - {user.Value} messages");
}
highscorePlace++;
}

72
Geekbot.net/Program.cs
View file

@ -1,6 +1,8 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
@ -24,7 +26,8 @@ namespace Geekbot.net
private IServiceProvider servicesProvider;
private RedisValue token;
private ILogger logger;
private ulong botOwnerId;
private string[] args;
private bool firstStart = false;
private static void Main(string[] args)
{
@ -44,6 +47,7 @@ namespace Geekbot.net
private async Task MainAsync(string[] args, ILogger logger)
{
this.logger = logger;
this.args = args;
logger.Information("[Geekbot] Initing Stuff");
client = new DiscordSocketClient(new DiscordSocketConfig
@ -91,17 +95,9 @@ namespace Geekbot.net
redis.StringSet("discordToken", newToken);
redis.StringSet("Game", "Ping Pong");
token = newToken;
firstStart = true;
}
var botOwner = redis.StringGet("botOwner");
if (botOwner.IsNullOrEmpty)
{
Console.Write("Bot Owner User ID: ");
botOwner = Console.ReadLine();
redis.StringSet("botOwner", botOwner);
}
botOwnerId = (ulong) botOwner;
services = new ServiceCollection();
var RandomClient = new Random();
var fortunes = new FortunesProvider(RandomClient, logger);
@ -144,7 +140,15 @@ namespace Geekbot.net
await commands.AddModulesAsync(Assembly.GetEntryAssembly());
services.AddSingleton(commands);
services.AddSingleton<DiscordSocketClient>(client);
this.servicesProvider = services.BuildServiceProvider();
servicesProvider = services.BuildServiceProvider();
if (firstStart || (args.Length != 0 && args.Contains("--reset")))
{
logger.Information("[Geekbot] Finishing setup");
await FinishSetup();
logger.Information("[Geekbot] Setup finished");
}
logger.Information("[Geekbot] Done and ready for use\n");
}
}
@ -162,44 +166,46 @@ namespace Geekbot.net
return true;
}
private async Task HandleCommand(SocketMessage messageParam)
private Task HandleCommand(SocketMessage messageParam)
{
var message = messageParam as SocketUserMessage;
if (message == null) return;
if (message.Author.IsBot) return;
if (message == null) return Task.CompletedTask;
if (message.Author.IsBot) return Task.CompletedTask;
var argPos = 0;
var lowCaseMsg = message.ToString().ToLower();
if (lowCaseMsg.StartsWith("ping"))
{
await message.Channel.SendMessageAsync("pong");
return;
message.Channel.SendMessageAsync("pong");
return Task.CompletedTask;
}
if (lowCaseMsg.StartsWith("hui"))
{
await message.Channel.SendMessageAsync("hui!!!");
return;
message.Channel.SendMessageAsync("hui!!!");
return Task.CompletedTask;
}
if (!(message.HasCharPrefix('!', ref argPos) ||
message.HasMentionPrefix(client.CurrentUser, ref argPos))) return;
message.HasMentionPrefix(client.CurrentUser, ref argPos))) return Task.CompletedTask;
var context = new CommandContext(client, message);
var commandExec = commands.ExecuteAsync(context, argPos, servicesProvider);
return Task.CompletedTask;
}
private async Task HandleMessageReceived(SocketMessage messsageParam)
private Task HandleMessageReceived(SocketMessage messsageParam)
{
var message = messsageParam;
if (message == null) return;
if (message == null) return Task.CompletedTask;
var channel = (SocketGuildChannel) message.Channel;
await redis.HashIncrementAsync($"{channel.Guild.Id}:Messages", message.Author.Id.ToString());
await redis.HashIncrementAsync($"{channel.Guild.Id}:Messages", 0.ToString());
redis.HashIncrementAsync($"{channel.Guild.Id}:Messages", message.Author.Id.ToString());
redis.HashIncrementAsync($"{channel.Guild.Id}:Messages", 0.ToString());
if (message.Author.IsBot) return;
if (message.Author.IsBot) return Task.CompletedTask;
logger.Information($"[Message] {channel.Guild.Name} - {message.Channel} - {message.Author.Username} - {message.Content}");
return Task.CompletedTask;
}
private async Task HandleUserJoined(SocketGuildUser user)
private Task HandleUserJoined(SocketGuildUser user)
{
if (!user.IsBot)
{
@ -207,9 +213,23 @@ namespace Geekbot.net
if (!message.IsNullOrEmpty)
{
message = message.ToString().Replace("$user", user.Mention);
await user.Guild.DefaultChannel.SendMessageAsync(message);
user.Guild.DefaultChannel.SendMessageAsync(message);
}
}
return Task.CompletedTask;
}
private async Task<Task> FinishSetup()
{
var appInfo = await client.GetApplicationInfoAsync();
redis.StringSet("botOwner", appInfo.Owner.Id);
var req = HttpWebRequest.Create(appInfo.IconUrl);
using (Stream stream = req.GetResponse().GetResponseStream() )
{
await client.CurrentUser.ModifyAsync(Avatar => new Image(stream));
}
return Task.CompletedTask;
}
private Task DiscordLogger(LogMessage message)