From 400df4387594d9f9316e7a491afe5f410763b37f Mon Sep 17 00:00:00 2001 From: Master Kwoth Date: Sun, 13 Mar 2016 00:25:49 +0100 Subject: [PATCH] "e" command, finished twitch and hitbox stream notifications (hopefuly) --- NadekoBot/Classes/JSONModels/Configuration.cs | 11 +- NadekoBot/Commands/StreamNotifications.cs | 116 +++++++++++------- NadekoBot/Modules/Conversations.cs | 5 + 3 files changed, 90 insertions(+), 42 deletions(-) diff --git a/NadekoBot/Classes/JSONModels/Configuration.cs b/NadekoBot/Classes/JSONModels/Configuration.cs index e38ac370..bb85995e 100644 --- a/NadekoBot/Classes/JSONModels/Configuration.cs +++ b/NadekoBot/Classes/JSONModels/Configuration.cs @@ -81,7 +81,7 @@ namespace NadekoBot.Classes.JSONModels { public static bool IsUserBlacklisted(ulong id) => NadekoBot.Config.UserBlacklist.Contains(id); } - public class StreamNotificationConfig { + public class StreamNotificationConfig : IEquatable { public string Username { get; set; } public StreamType Type { get; set; } public ulong ServerId { get; set; } @@ -93,5 +93,14 @@ namespace NadekoBot.Classes.JSONModels { Hitbox, YoutubeGaming } + + public bool Equals(StreamNotificationConfig other) => + this.Username.ToLower().Trim() == other.Username.ToLower().Trim() && + this.Type == other.Type && + this.ServerId == other.ServerId; + + public override int GetHashCode() { + return (int) ((int) ServerId + Username.Length + (int) Type); + } } } diff --git a/NadekoBot/Commands/StreamNotifications.cs b/NadekoBot/Commands/StreamNotifications.cs index 2c50e078..8a981a74 100644 --- a/NadekoBot/Commands/StreamNotifications.cs +++ b/NadekoBot/Commands/StreamNotifications.cs @@ -15,7 +15,7 @@ namespace NadekoBot.Commands { internal class StreamNotifications : DiscordCommand { private readonly Timer checkTimer = new Timer { - Interval = new TimeSpan(0, 0, 30).TotalMilliseconds, + Interval = new TimeSpan(0, 0, 15).TotalMilliseconds, }; public StreamNotifications(DiscordModule module) : base(module) { @@ -25,7 +25,12 @@ namespace NadekoBot.Commands { if (streams == null || !streams.Any()) return; foreach (var stream in streams) { - var data = await GetStreamStatus(stream); + Tuple data; + try { + data = await GetStreamStatus(stream); + } catch { + continue; + } if (data.Item1 != stream.LastStatus) { stream.LastStatus = data.Item1; @@ -37,8 +42,12 @@ namespace NadekoBot.Commands { $"**{(data.Item1 ? "ONLINE" : "OFFLINE")}** with " + $"**{data.Item2}** viewers."; if (stream.LastStatus) - msg += $"\n`Here is the Link:`【http://www.hitbox.tv/{stream.Username}】"; - + if (stream.Type == StreamNotificationConfig.StreamType.Hitbox) + msg += $"\n`Here is the Link:`【http://www.hitbox.tv/{stream.Username}】"; + else if (stream.Type == StreamNotificationConfig.StreamType.Twitch) + msg += $"\n`Here is the Link:`【http://www.twitch.tv/channels/{stream.Username}】"; + else if (stream.Type == StreamNotificationConfig.StreamType.YoutubeGaming) + msg += $"\n`Here is the Link:`【not implemented yet - {stream.Username}】"; await channel.SendMessage(msg); } } @@ -51,13 +60,19 @@ namespace NadekoBot.Commands { private async Task> GetStreamStatus(StreamNotificationConfig stream) { bool isLive; + string response; + JObject data; switch (stream.Type) { case StreamNotificationConfig.StreamType.Hitbox: - var response = await SearchHelper.GetResponseStringAsync($"https://api.hitbox.tv/media/status/{stream.Username}"); - var data = JObject.Parse(response); + response = await SearchHelper.GetResponseStringAsync($"https://api.hitbox.tv/media/status/{stream.Username}"); + data = JObject.Parse(response); isLive = data["media_is_live"].ToString() == "1"; return new Tuple(isLive, data["media_views"].ToString()); - break; + case StreamNotificationConfig.StreamType.Twitch: + response = await SearchHelper.GetResponseStringAsync($"https://api.twitch.tv/kraken/streams/{Uri.EscapeUriString(stream.Username)}"); + data = JObject.Parse(response); + isLive = !string.IsNullOrWhiteSpace(data["stream"].ToString()); + return new Tuple(isLive, isLive ? data["stream"]["viewers"].ToString() : "0"); default: break; } @@ -70,36 +85,19 @@ namespace NadekoBot.Commands { .Description("Notifies this channel when a certain user starts streaming." + "\n**Usage**: ~hitbox SomeStreamer") .Parameter("username", ParameterType.Unparsed) - .Do(async e => { - var username = e.GetArg("username"); - if (string.IsNullOrWhiteSpace(username)) - return; + .Do(TrackStream(StreamNotificationConfig.StreamType.Hitbox)); - var stream = new StreamNotificationConfig { - ServerId = e.Server.Id, - ChannelId = e.Channel.Id, - Username = username, - Type = StreamNotificationConfig.StreamType.Hitbox, - }; - Tuple data; - try { - data = await GetStreamStatus(stream); - } catch { - await e.Channel.SendMessage(":anger: Stream probably doesn't exist."); - return; - } - var msg = $"Stream is currently **{(data.Item1 ? "ONLINE" : "OFFLINE")}**"; - if (data.Item1) - msg += $"\n`Here is the Link:` http://www.hitbox.tv/{stream.Username}"; - await e.Channel.SendMessage($":ok: I will notify this channel when status changes.\n{msg}"); - NadekoBot.Config.ObservingStreams.Add(stream); - ConfigHandler.SaveConfig(); - }); + cgb.CreateCommand(Module.Prefix + "twitch") + .Alias(Module.Prefix + "tw") + .Description("Notifies this channel when a certain user starts streaming." + + "\n**Usage**: ~twitch SomeStreamer") + .Parameter("username", ParameterType.Unparsed) + .Do(TrackStream(StreamNotificationConfig.StreamType.Twitch)); - cgb.CreateCommand(Module.Prefix + "hitboxremove") - .Alias(Module.Prefix + "hbr") - .Description("Removes hitbox notifications of a certain user on this channel." + - "\n**Usage**: ~hbr SomeGuy") + cgb.CreateCommand(Module.Prefix + "removestream") + .Alias(Module.Prefix + "rms") + .Description("Removes notifications of a certain streamer on this channel." + + "\n**Usage**: ~srm SomeGuy") .Parameter("username", ParameterType.Unparsed) .Do(async e => { var username = e.GetArg("username")?.ToLower().Trim(); @@ -116,12 +114,12 @@ namespace NadekoBot.Commands { NadekoBot.Config.ObservingStreams.Remove(toRemove); ConfigHandler.SaveConfig(); - await e.Channel.SendMessage($":ok: Removed `{toRemove.Username}`'s stream from notifications"); + await e.Channel.SendMessage($":ok: Removed `{toRemove.Username}`'s stream from hitbox notifications"); }); - cgb.CreateCommand(Module.Prefix + "hitboxlist") - .Alias(Module.Prefix + "hbl") - .Description("Lists all hitbox streams you are following on this server." + + cgb.CreateCommand(Module.Prefix + "liststreams") + .Alias(Module.Prefix + "ls") + .Description("Lists all streams you are following on this server." + "\n**Usage**: ~hbl") .Do(async e => { var streams = NadekoBot.Config.ObservingStreams.Where(snc => @@ -136,13 +134,49 @@ namespace NadekoBot.Commands { var text = string.Join("\n", streamsArray.Select(snc => { try { - return $"{snc.Username}'s stream on {e.Server.GetChannel(e.Channel.Id).Name} channel"; + return $"`{snc.Username}`'s stream on **{e.Server.GetChannel(e.Channel.Id).Name}** channel. 【`{snc.Type.ToString()}`】"; } catch { } return ""; })); - await e.Channel.SendMessage($"You are following **{streamsArray.Length}** hitbox streamers on this server.\n" + text); + await e.Channel.SendMessage($"You are following **{streamsArray.Length}** hitbox streamers on this server.\n\n" + text); }); } + + private Func TrackStream(StreamNotificationConfig.StreamType type) => + async e => { + var username = e.GetArg("username"); + if (string.IsNullOrWhiteSpace(username)) + return; + + var stream = new StreamNotificationConfig { + ServerId = e.Server.Id, + ChannelId = e.Channel.Id, + Username = username, + Type = type, + }; + if (NadekoBot.Config.ObservingStreams.Contains(stream)) { + await e.Channel.SendMessage(":anger: I am already notifying that stream on this channel."); + } + Tuple data; + try { + data = await GetStreamStatus(stream); + } catch { + await e.Channel.SendMessage(":anger: Stream probably doesn't exist."); + return; + } + var msg = $"Stream is currently **{(data.Item1 ? "ONLINE" : "OFFLINE")}** with **{data.Item2}** viewers"; + if (data.Item1) + if (type == StreamNotificationConfig.StreamType.Hitbox) + msg += $"\n`Here is the Link:` http://www.hitbox.tv/{stream.Username}"; + else if (type == StreamNotificationConfig.StreamType.Twitch) + msg += $"\n`Here is the Link:` http://www.twitch.tv/channels/{stream.Username}"; + else if (type == StreamNotificationConfig.StreamType.YoutubeGaming) + msg += $"\n`Here is the Link:` not implemented yet - {stream.Username}"; + stream.LastStatus = data.Item1; + await e.Channel.SendMessage($":ok: I will notify this channel when status changes.\n{msg}"); + NadekoBot.Config.ObservingStreams.Add(stream); + ConfigHandler.SaveConfig(); + }; } } diff --git a/NadekoBot/Modules/Conversations.cs b/NadekoBot/Modules/Conversations.cs index c191b5aa..64c3b3df 100644 --- a/NadekoBot/Modules/Conversations.cs +++ b/NadekoBot/Modules/Conversations.cs @@ -28,6 +28,11 @@ namespace NadekoBot.Modules { manager.CreateCommands("", cgb => { cgb.AddCheck(Classes.Permissions.PermissionChecker.Instance); + + cgb.CreateCommand("e") + .Do(async e => { + await e.Channel.SendMessage($"{e.User.Name} did it. 😒 🔫"); + }); cgb.CreateCommand("\\o\\") .Description("Nadeko replies with /o/")