mattburchett/NadekoBot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

improvements to getresponse, prune and clr (performance/ratelimit)

Browse Source
This commit is contained in:
Master Kwoth 2016-02-05 18:17:48 +01:00
parent 95cc71b787
commit 62b2f0c869
3 changed files with 74 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
NadekoBot/Classes/NadekoStats.cs
View File

@ -12,7 +12,7 @@ namespace NadekoBot
{ {
public class NadekoStats public class NadekoStats
{ {
public string BotVersion = "0.8-beta7"; public string BotVersion = "0.8-beta8";
private static readonly NadekoStats _instance = new NadekoStats(); private static readonly NadekoStats _instance = new NadekoStats();
public static NadekoStats Instance => _instance; public static NadekoStats Instance => _instance;

38
NadekoBot/Modules/Administration.cs
View File

@ -9,12 +9,13 @@ using NadekoBot.Extensions;
using System.Threading.Tasks; using System.Threading.Tasks;
using NadekoBot.Commands; using NadekoBot.Commands;
using System.IO; using System.IO;
using System.Collections.Concurrent;
namespace NadekoBot.Modules { namespace NadekoBot.Modules {
class Administration : DiscordModule { class Administration : DiscordModule {
public Administration() : base() { public Administration() : base() {
commands.Add(new HelpCommand()); commands.Add(new HelpCommand());
if(NadekoBot.ParseActive) if (NadekoBot.ParseActive)
commands.Add(new ServerGreetCommand()); commands.Add(new ServerGreetCommand());
else else
Console.WriteLine("Parse not active. Server greet disabled."); Console.WriteLine("Parse not active. Server greet disabled.");
@ -148,7 +149,7 @@ namespace NadekoBot.Modules {
cgb.CreateCommand(".roles") cgb.CreateCommand(".roles")
.Description("List all roles on this server") .Description("List all roles on this server")
.Do(async e => { .Do(async e => {
await e.Send("`List of roles:` \n• " + string.Join("\n• ", e.Server.Roles).Replace("@everyone","[everyone]")); await e.Send("`List of roles:` \n• " + string.Join("\n• ", e.Server.Roles).Replace("@everyone", "[everyone]"));
}); });
cgb.CreateCommand(".b").Alias(".ban") cgb.CreateCommand(".b").Alias(".ban")
@ -375,28 +376,37 @@ namespace NadekoBot.Modules {
}); });
ConcurrentDictionary<Server, bool> pruneDict = new ConcurrentDictionary<Server, bool>();
cgb.CreateCommand(".prune") cgb.CreateCommand(".prune")
.Parameter("num", ParameterType.Required) .Parameter("num", ParameterType.Required)
.Description("Prunes a number of messages from the current channel.\n**Usage**: .prune 50") .Description("Prunes a number of messages from the current channel.\n**Usage**: .prune 50")
.Do(async e => { .Do(async e => {
if (!e.User.ServerPermissions.ManageMessages) return; if (!e.User.ServerPermissions.ManageMessages) return;
if (pruneDict.ContainsKey(e.Server))
return;
int num; int num;
if (!Int32.TryParse(e.GetArg("num"), out num) || num < 1) { if (!Int32.TryParse(e.GetArg("num"), out num) || num < 1) {
await e.Send("Incorrect amount."); await e.Send("Incorrect amount.");
return; return;
} }
pruneDict.TryAdd(e.Server, true);
await Task.Factory.StartNew(async () => {
try { try {
Message last = null; Message last = null;
while (num > 0) { while (num > 0) {
var msgs = await e.Channel.DownloadMessages(num, last?.Id); var msgs = await e.Channel.DownloadMessages(num, last?.Id);
last = msgs.LastOrDefault(); last = msgs.LastOrDefault();
msgs.ForEach(async m => await m.Delete()); foreach (var m in msgs) {
await m.Delete();
await Task.Delay(500);
}
num -= 100; num -= 100;
} }
} catch (Exception) { await e.Send("Failed pruning. Make sure the bot has correct permissions."); } } catch (Exception) { await e.Send("Failed pruning. Make sure the bot has correct permissions."); }
}, TaskCreationOptions.LongRunning);
bool throwAway;
pruneDict.TryRemove(e.Server, out throwAway);
}); });
cgb.CreateCommand(".die") cgb.CreateCommand(".die")
@ -412,13 +422,25 @@ namespace NadekoBot.Modules {
} }
}); });
ConcurrentDictionary<Server, bool> clearDictionary = new ConcurrentDictionary<Server, bool>();
cgb.CreateCommand(".clr") cgb.CreateCommand(".clr")
.Description("Clears some of nadeko's messages from the current channel.") .Description("Clears some of nadeko's messages from the current channel.")
.Do(async e => { .Do(async e => {
try {
if (clearDictionary.ContainsKey(e.Server))
return;
clearDictionary.TryAdd(e.Server, true);
var msgs = await e.Channel.DownloadMessages(100); var msgs = await e.Channel.DownloadMessages(100);
await Task.Run(async () => {
msgs.Where(msg => msg.User.Id == client.CurrentUser.Id) var ms = msgs.Where(msg => msg.User.Id == client.CurrentUser.Id);
.ForEach(async m => { try { await m.Delete(); } catch (Exception) { } }); foreach (var m in ms) {
try { await m.Delete(); } catch (Exception) { }
await Task.Delay(500);
}
});
} catch (Exception) {}
bool throwaway;
clearDictionary.TryRemove(e.Server, out throwaway);
}); });
cgb.CreateCommand(".newname") cgb.CreateCommand(".newname")
.Description("Give the bot a new name.") .Description("Give the bot a new name.")

20
NadekoBot/Modules/Searches.cs
View File

@ -183,14 +183,14 @@ namespace NadekoBot.Modules {
await e.Send($":anger: Error {ex}"); await e.Send($":anger: Error {ex}");
} }
}); });
/*
cgb.CreateCommand("~osu") cgb.CreateCommand("~osu")
.Description("desc") .Description("desc")
.Parameter("arg", ParameterType.Required) .Parameter("arg", ParameterType.Required)
.Do(async e => { .Do(async e => {
var arg = e.GetArg("arg"); var arg = e.GetArg("arg");
//make request to osu var res = await GetResponseStream($"http://lemmmy.pw/osusig/sig.php?uname=kwoth&flagshadow&xpbar&xpbarhex&pp=2");
//print useful data await e.Channel.SendFile($"_{e.GetArg("arg")}.png", res);
}); });
cgb.CreateCommand("~osubind") cgb.CreateCommand("~osubind")
@ -209,11 +209,21 @@ namespace NadekoBot.Modules {
//if exists save bind pair to parse.com //if exists save bind pair to parse.com
//if not valid error //if not valid error
}); });
*/
}); });
} }
public static async Task<Stream> GetResponseStream(string v) => public static async Task<Stream> GetResponseStream(string v) {
(await ((HttpWebRequest)WebRequest.Create(v)).GetResponseAsync()).GetResponseStream(); var wr = (HttpWebRequest)WebRequest.Create(v);
try {
ServicePointManager.ServerCertificateValidationCallback = delegate { return true; };
wr.UserAgent = @"Mozilla/5.0 (Windows NT 6.2; Win64; x64)";
return (await (wr).GetResponseAsync()).GetResponseStream();
} catch (Exception ex) {
Console.WriteLine("error in getresponse stream " + ex);
return null;
}
}
public static async Task<string> GetResponseAsync(string v) => public static async Task<string> GetResponseAsync(string v) =>
await new StreamReader((await ((HttpWebRequest)WebRequest.Create(v)).GetResponseAsync()).GetResponseStream()).ReadToEndAsync(); await new StreamReader((await ((HttpWebRequest)WebRequest.Create(v)).GetResponseAsync()).GetResponseStream()).ReadToEndAsync();