started work on stream notifications, hitbox semi-supported
This commit is contained in:
Master Kwoth
parent
7631fa5408
commit
c361200178
@ -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
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
148
NadekoBot/Commands/StreamNotifications.cs
Normal file
148
NadekoBot/Commands/StreamNotifications.cs
Normal file
@ -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);
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -3,6 +3,7 @@
|
||||
"ForwardMessages": true,
|
||||
"IsRotatingStatus": false,
|
||||
"SendPrivateMessageOnMention": false,
|
||||
"ObservingStreams": [],
|
||||
"RotatingStatuses": [],
|
||||
"CommandPrefixes": {
|
||||
"Administration": ".",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user