started work on stream notifications, hitbox semi-supported

NadekoBot/Classes/JSONModels/Configuration.cs

@@ -10,6 +10,7 @@ namespace NadekoBot.Classes.JSONModels {
         public bool ForwardMessages { get; set; } = true;
         public bool IsRotatingStatus { get; set; } = false;
         public bool SendPrivateMessageOnMention { get; set; } = false;
+        public HashSet<StreamNotificationConfig> ObservingStreams { get; set; } = new HashSet<StreamNotificationConfig>();
         public List<string> RotatingStatuses { get; set; } = new List<string>();
         public CommandPrefixesModel CommandPrefixes { get; set; } = new CommandPrefixesModel();
         public HashSet<ulong> ServerBlacklist { get; set; } = new HashSet<ulong>();
@@ -47,8 +48,7 @@ namespace NadekoBot.Classes.JSONModels {
         };
     }
 
-    public class CommandPrefixesModel
+    public class CommandPrefixesModel {
-    {
         public string Administration { get; set; } = ".";
         public string Searches { get; set; } = "~";
         public string NSFW { get; set; } = "~";
@@ -80,4 +80,18 @@ namespace NadekoBot.Classes.JSONModels {
 
         public static bool IsUserBlacklisted(ulong id) => NadekoBot.Config.UserBlacklist.Contains(id);
     }
+
+    public class StreamNotificationConfig {
+        public string Username { get; set; }
+        public StreamType Type { get; set; }
+        public ulong ServerId { get; set; }
+        public ulong ChannelId { get; set; }
+        public bool LastStatus { get; set; }
+
+        public enum StreamType {
+            Twitch,
+            Hitbox,
+            YoutubeGaming
+        }
+    }
 }

NadekoBot/Commands/StreamNotifications.cs

@@ -0,0 +1,148 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.CompilerServices;
+using System.Text;
+using System.Threading.Tasks;
+using System.Timers;
+using Discord.Commands;
+using NadekoBot.Classes;
+using NadekoBot.Classes.JSONModels;
+using NadekoBot.Modules;
+using Newtonsoft.Json.Linq;
+
+namespace NadekoBot.Commands {
+    internal class StreamNotifications : DiscordCommand {
+
+        private readonly Timer checkTimer = new Timer {
+            Interval = new TimeSpan(0, 0, 30).TotalMilliseconds,
+        };
+        public StreamNotifications(DiscordModule module) : base(module) {
+
+            checkTimer.Elapsed += async (s, e) => {
+                try {
+                    var streams = NadekoBot.Config.ObservingStreams;
+                    if (streams == null || !streams.Any()) return;
+
+                    foreach (var stream in streams) {
+                        var data = await GetStreamStatus(stream);
+
+                        if (data.Item1 != stream.LastStatus) {
+                            stream.LastStatus = data.Item1;
+                            var server = NadekoBot.Client.GetServer(stream.ServerId);
+                            var channel = server?.GetChannel(stream.ChannelId);
+                            if (channel == null)
+                                continue;
+                            var msg = $"`{stream.Username}`'s stream is now " +
+                                      $"**{(data.Item1 ? "ONLINE" : "OFFLINE")}** with " +
+                                      $"**{data.Item2}** viewers.";
+                            if (stream.LastStatus)
+                                msg += $"\n`Here is the Link:`【http://www.hitbox.tv/{stream.Username}】";
+
+                            await channel.SendMessage(msg);
+                        }
+                    }
+                } catch { }
+                ConfigHandler.SaveConfig();
+            };
+
+            checkTimer.Start();
+        }
+
+        private async Task<Tuple<bool, string>> GetStreamStatus(StreamNotificationConfig stream) {
+            bool isLive;
+            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);
+                    isLive = data["media_is_live"].ToString() == "1";
+                    return new Tuple<bool, string>(isLive, data["media_views"].ToString());
+                    break;
+                default:
+                    break;
+            }
+            return new Tuple<bool, string>(false, "0");
+        }
+
+        internal override void Init(CommandGroupBuilder cgb) {
+            cgb.CreateCommand(Module.Prefix + "hitbox")
+                .Alias(Module.Prefix + "hb")
+                .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;
+
+                    var stream = new StreamNotificationConfig {
+                        ServerId = e.Server.Id,
+                        ChannelId = e.Channel.Id,
+                        Username = username,
+                        Type = StreamNotificationConfig.StreamType.Hitbox,
+                    };
+                    Tuple<bool, string> 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 + "hitboxremove")
+                .Alias(Module.Prefix + "hbr")
+                .Description("Removes hitbox notifications of a certain user on this channel." +
+                             "\n**Usage**: ~hbr SomeGuy")
+                .Parameter("username", ParameterType.Unparsed)
+                .Do(async e => {
+                    var username = e.GetArg("username")?.ToLower().Trim();
+                    if (string.IsNullOrWhiteSpace(username))
+                        return;
+
+                    var toRemove = NadekoBot.Config.ObservingStreams
+                        .FirstOrDefault(snc => snc.ChannelId == e.Channel.Id &&
+                                        snc.Username.ToLower().Trim() == username);
+                    if (toRemove == null) {
+                        await e.Channel.SendMessage(":anger: No such stream.");
+                        return;
+                    }
+
+                    NadekoBot.Config.ObservingStreams.Remove(toRemove);
+                    ConfigHandler.SaveConfig();
+                    await e.Channel.SendMessage($":ok: Removed `{toRemove.Username}`'s stream from notifications");
+                });
+
+            cgb.CreateCommand(Module.Prefix + "hitboxlist")
+                .Alias(Module.Prefix + "hbl")
+                .Description("Lists all hitbox streams you are following on this server." +
+                             "\n**Usage**: ~hbl")
+                .Do(async e => {
+                    var streams = NadekoBot.Config.ObservingStreams.Where(snc =>
+                        snc.ServerId == e.Server.Id);
+
+                    var streamsArray = streams as StreamNotificationConfig[] ?? streams.ToArray();
+
+                    if (streamsArray.Length == 0) {
+                        await e.Channel.SendMessage("You are not following any streams on this server.");
+                        return;
+                    }
+
+                    var text = string.Join("\n", streamsArray.Select(snc => {
+                        try {
+                            return $"{snc.Username}'s stream on {e.Server.GetChannel(e.Channel.Id).Name} channel";
+                        } catch { }
+                        return "";
+                    }));
+
+                    await e.Channel.SendMessage($"You are following **{streamsArray.Length}** hitbox streamers on this server.\n" + text);
+                });
+        }
+    }
+}

NadekoBot/bin/Debug/data/config_example.json

@@ -3,6 +3,7 @@
   "ForwardMessages": true,
   "IsRotatingStatus": false,
   "SendPrivateMessageOnMention": false,
+  "ObservingStreams": [],
   "RotatingStatuses": [],
   "CommandPrefixes": {
     "Administration": ".",