Compare commits

..

No commits in common. "f47685a17d9bb9ecb1f8b2cc13d6606db6269cd0" and "9fd43502f56f0eed7d94b9d00fb349006ce62723" have entirely different histories.

21 changed files with 171 additions and 461 deletions

View File

@ -9,6 +9,5 @@ namespace TwitchChatTTS.Chat.Commands
Syntax = 4, Syntax = 4,
Fail = 5, Fail = 5,
OtherRoom = 6, OtherRoom = 6,
RateLimited = 7
} }
} }

View File

@ -3,7 +3,6 @@ using CommonSocketLibrary.Abstract;
using CommonSocketLibrary.Common; using CommonSocketLibrary.Common;
using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.DependencyInjection;
using Serilog; using Serilog;
using TwitchChatTTS.Chat.Commands.Limits;
using TwitchChatTTS.Chat.Groups.Permissions; using TwitchChatTTS.Chat.Groups.Permissions;
using TwitchChatTTS.Hermes.Socket; using TwitchChatTTS.Hermes.Socket;
using TwitchChatTTS.Twitch.Socket.Messages; using TwitchChatTTS.Twitch.Socket.Messages;
@ -18,7 +17,6 @@ namespace TwitchChatTTS.Chat.Commands
private readonly HermesSocketClient _hermes; private readonly HermesSocketClient _hermes;
//private readonly TwitchWebsocketClient _twitch; //private readonly TwitchWebsocketClient _twitch;
private readonly IGroupPermissionManager _permissionManager; private readonly IGroupPermissionManager _permissionManager;
private readonly IUsagePolicy<long> _permissionPolicy;
private readonly ILogger _logger; private readonly ILogger _logger;
private string CommandStartSign { get; } = "!"; private string CommandStartSign { get; } = "!";
@ -28,7 +26,6 @@ namespace TwitchChatTTS.Chat.Commands
[FromKeyedServices("hermes")] SocketClient<WebSocketMessage> hermes, [FromKeyedServices("hermes")] SocketClient<WebSocketMessage> hermes,
//[FromKeyedServices("twitch")] SocketClient<TwitchWebsocketMessage> twitch, //[FromKeyedServices("twitch")] SocketClient<TwitchWebsocketMessage> twitch,
IGroupPermissionManager permissionManager, IGroupPermissionManager permissionManager,
IUsagePolicy<long> limitManager,
ILogger logger ILogger logger
) )
{ {
@ -36,7 +33,6 @@ namespace TwitchChatTTS.Chat.Commands
_hermes = (hermes as HermesSocketClient)!; _hermes = (hermes as HermesSocketClient)!;
//_twitch = (twitch as TwitchWebsocketClient)!; //_twitch = (twitch as TwitchWebsocketClient)!;
_permissionManager = permissionManager; _permissionManager = permissionManager;
_permissionPolicy = limitManager;
_logger = logger; _logger = logger;
} }
@ -73,10 +69,9 @@ namespace TwitchChatTTS.Chat.Commands
// Check if command can be executed by this chatter. // Check if command can be executed by this chatter.
var command = selectorResult.Command; var command = selectorResult.Command;
long chatterId = long.Parse(message.ChatterUserId); long chatterId = long.Parse(message.ChatterUserId);
var path = $"tts.commands.{com}";
if (chatterId != _user.OwnerId) if (chatterId != _user.OwnerId)
{ {
bool executable = command.AcceptCustomPermission ? CanExecute(chatterId, groups, path, selectorResult.Permissions) : false; bool executable = command.AcceptCustomPermission ? CanExecute(chatterId, groups, $"tts.commands.{com}", selectorResult.Permissions) : false;
if (!executable) if (!executable)
{ {
_logger.Warning($"Denied permission to use command [chatter id: {chatterId}][args: {arg}][command type: {command.GetType().Name}]"); _logger.Warning($"Denied permission to use command [chatter id: {chatterId}][args: {arg}][command type: {command.GetType().Name}]");
@ -84,12 +79,6 @@ namespace TwitchChatTTS.Chat.Commands
} }
} }
if (!_permissionPolicy.TryUse(chatterId, groups, path))
{
_logger.Warning($"Chatter reached usage limit on command [command type: {command.GetType().Name}][chatter id: {chatterId}][path: {path}][groups: {string.Join("|", groups)}]");
return ChatCommandResult.RateLimited;
}
// Check if the arguments are valid. // Check if the arguments are valid.
var arguments = _commandSelector.GetNonStaticArguments(args, selectorResult.Path); var arguments = _commandSelector.GetNonStaticArguments(args, selectorResult.Path);
foreach (var entry in arguments) foreach (var entry in arguments)
@ -99,7 +88,7 @@ namespace TwitchChatTTS.Chat.Commands
// Optional parameters were validated while fetching this command. // Optional parameters were validated while fetching this command.
if (!parameter.Optional && !parameter.Validate(argument, message.Message.Fragments)) if (!parameter.Optional && !parameter.Validate(argument, message.Message.Fragments))
{ {
_logger.Warning($"Command failed due to an argument being invalid [argument name: {parameter.Name}][argument value: {argument}][parameter type: {parameter.GetType().Name}][arguments: {arg}][command type: {command.GetType().Name}][chatter: {message.ChatterUserLogin}][chatter id: {message.ChatterUserId}]"); _logger.Warning($"Command failed due to an argument being invalid [argument name: {parameter.Name}][argument value: {argument}][arguments: {arg}][command type: {command.GetType().Name}][chatter: {message.ChatterUserLogin}][chatter id: {message.ChatterUserId}]");
return ChatCommandResult.Syntax; return ChatCommandResult.Syntax;
} }
} }

View File

@ -0,0 +1,88 @@
namespace TwitchChatTTS.Chat.Commands.Limits
{
public interface ICommandLimitManager
{
bool HasReachedLimit(long chatterId, string name, string group);
void RemoveUsageLimit(string name, string group);
void SetUsageLimit(int count, TimeSpan span, string name, string group);
bool TryUse(long chatterId, string name, string group);
}
public class CommandLimitManager : ICommandLimitManager
{
// group + name -> chatter id -> usage
private readonly IDictionary<string, IDictionary<long, Usage>> _usages;
// group + name -> limit
private readonly IDictionary<string, Limit> _limits;
public CommandLimitManager()
{
_usages = new Dictionary<string, IDictionary<long, Usage>>();
_limits = new Dictionary<string, Limit>();
}
public bool HasReachedLimit(long chatterId, string name, string group)
{
throw new NotImplementedException();
}
public void RemoveUsageLimit(string name, string group)
{
throw new NotImplementedException();
}
public void SetUsageLimit(int count, TimeSpan span, string name, string group)
{
throw new NotImplementedException();
}
public bool TryUse(long chatterId, string name, string group)
{
var path = $"{group}.{name}";
if (!_limits.TryGetValue(path, out var limit))
return true;
if (!_usages.TryGetValue(path, out var groupUsage))
{
groupUsage = new Dictionary<long, Usage>();
_usages.Add(path, groupUsage);
}
if (!groupUsage.TryGetValue(chatterId, out var usage))
{
usage = new Usage()
{
Usages = new long[limit.Count],
Index = 0
};
groupUsage.Add(chatterId, usage);
}
int first = (usage.Index + 1) % limit.Count;
long timestamp = DateTime.UtcNow.Ticks / TimeSpan.TicksPerMillisecond;
if (timestamp - usage.Usages[first] < limit.Span)
{
return false;
}
usage.Usages[usage.Index] = timestamp;
usage.Index = first;
return true;
}
private class Usage
{
public long[] Usages { get; set; }
public int Index { get; set; }
}
private struct Limit
{
public int Count { get; set; }
public int Span { get; set; }
}
}
}

View File

@ -1,10 +0,0 @@
namespace TwitchChatTTS.Chat.Commands.Limits
{
public interface IUsagePolicy<K>
{
void Remove(string group, string policy);
void Set(string group, string policy, int count, TimeSpan span);
bool TryUse(K key, string group, string policy);
public bool TryUse(K key, IEnumerable<string> groups, string policy);
}
}

View File

@ -1,216 +0,0 @@
using Serilog;
namespace TwitchChatTTS.Chat.Commands.Limits
{
public class UsagePolicy<K> : IUsagePolicy<K> where K : notnull
{
private readonly ILogger _logger;
private readonly UsagePolicyNode<K> _root;
public UsagePolicy(ILogger logger)
{
_logger = logger;
_root = new UsagePolicyNode<K>(string.Empty, null, null, logger);
}
public void Remove(string group, string policy)
{
ArgumentException.ThrowIfNullOrWhiteSpace(group, nameof(group));
ArgumentException.ThrowIfNullOrWhiteSpace(policy, nameof(policy));
string[] path = (group + '.' + policy).Split('.');
_root.Remove(path);
}
public void Set(string group, string policy, int count, TimeSpan span)
{
ArgumentException.ThrowIfNullOrWhiteSpace(group, nameof(group));
ArgumentException.ThrowIfNullOrWhiteSpace(policy, nameof(policy));
if (count <= 0)
throw new InvalidOperationException("Count cannot be 0 or lower.");
if (span.TotalMilliseconds == 0)
throw new InvalidOperationException("Time span cannot be 0 milliseconds.");
string[] path = (group + '.' + policy).Split('.');
_root.Set(path, count, span);
}
public bool TryUse(K key, string group, string policy)
{
ArgumentException.ThrowIfNullOrWhiteSpace(group, nameof(group));
ArgumentException.ThrowIfNullOrWhiteSpace(policy, nameof(policy));
string[] path = (group + '.' + policy).Split('.');
UsagePolicyNode<K>? node = _root.Get(path);
_logger.Debug($"Fetched policy node [is null: {node == null}]");
if (node == null)
return false;
return node.TryUse(key, DateTime.UtcNow);
}
public bool TryUse(K key, IEnumerable<string> groups, string policy)
{
ArgumentNullException.ThrowIfNull(groups, nameof(groups));
ArgumentException.ThrowIfNullOrWhiteSpace(policy, nameof(policy));
foreach (string group in groups)
{
if (TryUse(key, group, policy))
{
_logger.Debug($"Checking policy node [policy: {group}.{policy}][result: True]");
return true;
}
_logger.Debug($"Checking policy node [policy: {group}.{policy}][result: False]");
}
return false;
}
private class UsagePolicyLimit
{
public int Count { get; set; }
public TimeSpan Span { get; set; }
public UsagePolicyLimit(int count, TimeSpan span)
{
Count = count;
Span = span;
}
}
private class UserUsageData
{
public DateTime[] Uses { get; set; }
public int Index { get; set; }
public UserUsageData(int size, int index)
{
Uses = new DateTime[size];
Index = index;
}
}
private class UsagePolicyNode<T> where T : notnull
{
public string Name { get; set; }
public UsagePolicyLimit? Limit { get; private set; }
private UsagePolicyNode<T>? _parent { get; }
private IDictionary<T, UserUsageData> _usages { get; }
private IList<UsagePolicyNode<T>> _children { get; }
private ILogger _logger;
private object _lock { get; }
public UsagePolicyNode(string name, UsagePolicyLimit? data, UsagePolicyNode<T>? parent, ILogger logger)
{
//ArgumentException.ThrowIfNullOrWhiteSpace(name, nameof(name));
Name = name;
Limit = data;
_parent = parent;
_usages = new Dictionary<T, UserUsageData>();
_children = new List<UsagePolicyNode<T>>();
_logger = logger;
_lock = new object();
}
public UsagePolicyNode<T>? Get(IEnumerable<string> path)
{
if (!path.Any())
return this;
var nextName = path.First();
var next = _children.FirstOrDefault(c => c.Name == nextName);
if (next == null)
return this;
return next.Get(path.Skip(1));
}
public UsagePolicyNode<T>? Remove(IEnumerable<string> path)
{
if (!path.Any())
{
if (_parent == null)
throw new InvalidOperationException("Cannot remove root node");
_parent._children.Remove(this);
return this;
}
var nextName = path.First();
var next = _children.FirstOrDefault(c => c.Name == nextName);
_logger.Debug($"internal remove node [is null: {next == null}][path: {string.Join('.', path)}]");
if (next == null)
return null;
return next.Remove(path.Skip(1));
}
public void Set(IEnumerable<string> path, int count, TimeSpan span)
{
if (!path.Any())
{
Limit = new UsagePolicyLimit(count, span);
return;
}
var nextName = path.First();
var next = _children.FirstOrDefault(c => c.Name == nextName);
_logger.Debug($"internal set node [is null: {next == null}][path: {string.Join('.', path)}]");
if (next == null)
{
next = new UsagePolicyNode<T>(nextName, null, this, _logger);
_children.Add(next);
}
next.Set(path.Skip(1), count, span);
}
public bool TryUse(T key, DateTime timestamp)
{
if (_parent == null)
return false;
if (Limit == null || Limit.Count <= 0)
return _parent.TryUse(key, timestamp);
UserUsageData? usage;
lock (_lock)
{
if (!_usages.TryGetValue(key, out usage))
{
usage = new UserUsageData(Limit.Count, 1 % Limit.Count);
usage.Uses[0] = timestamp;
_usages.Add(key, usage);
_logger.Debug($"internal use node create");
return true;
}
if (usage.Uses.Length != Limit.Count)
{
var sizeDiff = Math.Max(0, usage.Uses.Length - Limit.Count);
var temp = usage.Uses.Skip(sizeDiff);
var tempSize = usage.Uses.Length - sizeDiff;
usage.Uses = temp.Union(new DateTime[Math.Max(0, Limit.Count - tempSize)]).ToArray();
}
}
// Attempt on parent node if policy has been abused.
if (timestamp - usage.Uses[usage.Index] < Limit.Span)
{
_logger.Debug($"internal use node spam [span: {(timestamp - usage.Uses[usage.Index]).TotalMilliseconds}][index: {usage.Index}]");
return _parent.TryUse(key, timestamp);
}
_logger.Debug($"internal use node normal [span: {(timestamp - usage.Uses[usage.Index]).TotalMilliseconds}][index: {usage.Index}]");
lock (_lock)
{
usage.Uses[usage.Index] = timestamp;
usage.Index = (usage.Index + 1) % Limit.Count;
}
return true;
}
}
}
}

View File

@ -36,13 +36,13 @@ namespace TwitchChatTTS.Chat.Commands
{ {
b.CreateStaticInputParameter("add", b => b.CreateStaticInputParameter("add", b =>
{ {
b.CreateUnvalidatedParameter("voiceName") b.CreateVoiceNameParameter("voiceName", false)
.CreateCommand(new AddTTSVoiceCommand(_user, _logger)); .CreateCommand(new AddTTSVoiceCommand(_user, _logger));
}) })
.AddAlias("insert", "add") .AddAlias("insert", "add")
.CreateStaticInputParameter("delete", b => .CreateStaticInputParameter("delete", b =>
{ {
b.CreateVoiceNameParameter("voiceName", false) b.CreateVoiceNameParameter("voiceName", true)
.CreateCommand(new DeleteTTSVoiceCommand(_user, _logger)); .CreateCommand(new DeleteTTSVoiceCommand(_user, _logger));
}) })
.AddAlias("del", "delete") .AddAlias("del", "delete")

View File

@ -12,7 +12,7 @@ using TwitchChatTTS.Twitch.Socket.Messages;
namespace TwitchChatTTS.Chat.Messaging namespace TwitchChatTTS.Chat.Messaging
{ {
public class ChatMessageReader : IChatMessageReader public class ChatMessageReader
{ {
private readonly User _user; private readonly User _user;
private readonly TTSPlayer _player; private readonly TTSPlayer _player;
@ -94,6 +94,7 @@ namespace TwitchChatTTS.Chat.Messaging
private IEnumerable<TTSMessage> HandlePartialMessage(string voice, string message) private IEnumerable<TTSMessage> HandlePartialMessage(string voice, string message)
{ {
var parts = _sfxRegex.Split(message); var parts = _sfxRegex.Split(message);
if (parts.Length == 1) if (parts.Length == 1)
{ {
return [new TTSMessage() return [new TTSMessage()

View File

@ -1,10 +0,0 @@
using TwitchChatTTS.Twitch.Socket;
using TwitchChatTTS.Twitch.Socket.Messages;
namespace TwitchChatTTS.Chat.Messaging
{
public interface IChatMessageReader
{
Task Read(TwitchWebsocketClient sender, long broadcasterId, long? chatterId, string? chatterLogin, string? messageId, TwitchReplyInfo? reply, TwitchChatFragment[] fragments, int priority);
}
}

View File

@ -68,7 +68,6 @@ namespace TwitchChatTTS.Hermes.Socket.Handlers
await client.FetchEmotes(); await client.FetchEmotes();
await client.FetchRedemptions(); await client.FetchRedemptions();
await client.FetchPermissions(); await client.FetchPermissions();
await client.FetchPolicies();
if (_user.NightbotConnection != null) if (_user.NightbotConnection != null)
{ {

View File

@ -6,10 +6,8 @@ using CommonSocketLibrary.Common;
using HermesSocketLibrary.Requests.Callbacks; using HermesSocketLibrary.Requests.Callbacks;
using HermesSocketLibrary.Requests.Messages; using HermesSocketLibrary.Requests.Messages;
using HermesSocketLibrary.Socket.Data; using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Models;
using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.DependencyInjection;
using Serilog; using Serilog;
using TwitchChatTTS.Chat.Commands.Limits;
using TwitchChatTTS.Chat.Emotes; using TwitchChatTTS.Chat.Emotes;
using TwitchChatTTS.Chat.Groups; using TwitchChatTTS.Chat.Groups;
using TwitchChatTTS.Chat.Groups.Permissions; using TwitchChatTTS.Chat.Groups.Permissions;
@ -21,8 +19,6 @@ namespace TwitchChatTTS.Hermes.Socket.Handlers
{ {
private User _user; private User _user;
private readonly ICallbackManager<HermesRequestData> _callbackManager; private readonly ICallbackManager<HermesRequestData> _callbackManager;
private readonly IChatterGroupManager _groups;
private readonly IUsagePolicy<long> _policies;
private readonly TwitchApiClient _twitch; private readonly TwitchApiClient _twitch;
private readonly NightbotApiClient _nightbot; private readonly NightbotApiClient _nightbot;
private readonly IServiceProvider _serviceProvider; private readonly IServiceProvider _serviceProvider;
@ -36,8 +32,6 @@ namespace TwitchChatTTS.Hermes.Socket.Handlers
public RequestAckHandler( public RequestAckHandler(
ICallbackManager<HermesRequestData> callbackManager, ICallbackManager<HermesRequestData> callbackManager,
IChatterGroupManager groups,
IUsagePolicy<long> policies,
TwitchApiClient twitch, TwitchApiClient twitch,
NightbotApiClient nightbot, NightbotApiClient nightbot,
IServiceProvider serviceProvider, IServiceProvider serviceProvider,
@ -47,8 +41,6 @@ namespace TwitchChatTTS.Hermes.Socket.Handlers
) )
{ {
_callbackManager = callbackManager; _callbackManager = callbackManager;
_groups = groups;
_policies = policies;
_twitch = twitch; _twitch = twitch;
_nightbot = nightbot; _nightbot = nightbot;
_serviceProvider = serviceProvider; _serviceProvider = serviceProvider;
@ -367,74 +359,6 @@ namespace TwitchChatTTS.Hermes.Socket.Handlers
else else
_logger.Warning("Failed to update default TTS voice via request."); _logger.Warning("Failed to update default TTS voice via request.");
} }
else if (message.Request.Type == "get_policies")
{
var policies = JsonSerializer.Deserialize<IEnumerable<PolicyMessage>>(message.Data!.ToString()!, _options);
if (policies == null || !policies.Any())
{
_logger.Information($"Policies have been set to default.");
_policies.Set("everyone", "tts", 100, TimeSpan.FromSeconds(15));
return;
}
foreach (var policy in policies)
{
var group = _groups.Get(policy.GroupId.ToString());
if (policy == null)
{
_logger.Debug($"Policy data failed");
continue;
}
_logger.Debug($"Policy data [policy id: {policy.Id}][path: {policy.Path}][group id: {policy.GroupId}][group name: {group?.Name}]");
_policies.Set(group?.Name ?? string.Empty, policy.Path, policy.Usage, TimeSpan.FromMilliseconds(policy.Span));
}
_logger.Information($"Policies have been loaded, a total of {policies.Count()} policies.");
}
else if (message.Request.Type == "update_policy")
{
var policy = JsonSerializer.Deserialize<PolicyMessage>(message.Data!.ToString()!, _options);
var group = _groups.Get(policy.GroupId.ToString());
if (policy == null || group == null)
{
_logger.Debug($"Policy data failed");
return;
}
_logger.Debug($"Policy data [policy id: {policy.Id}][path: {policy.Path}][group id: {policy.GroupId}][group name: {group?.Name}]");
_policies.Set(group?.Name ?? string.Empty, policy.Path, policy.Usage, TimeSpan.FromMilliseconds(policy.Span));
_logger.Information($"Policy has been updated [policy id: {policy.Id}]");
}
else if (message.Request.Type == "create_policy")
{
var policy = JsonSerializer.Deserialize<PolicyMessage>(message.Data!.ToString()!, _options);
if (policy == null)
{
_logger.Debug($"Policy data failed");
return;
}
var group = _groups.Get(policy.GroupId.ToString());
if (group == null)
{
_logger.Debug($"Group data failed");
return;
}
_logger.Debug($"Policy data [policy id: {policy.Id}][path: {policy.Path}][group id: {policy.GroupId}][group name: {group?.Name}]");
_policies.Set(group?.Name, policy.Path, policy.Usage, TimeSpan.FromMilliseconds(policy.Span));
_logger.Information($"Policy has been updated [policy id: {policy.Id}]");
}
else if (message.Request.Type == "update_policies")
{
var policy = JsonSerializer.Deserialize<PolicyMessage>(message.Data!.ToString()!, _options);
var group = _groups.Get(policy.GroupId.ToString());
if (policy == null)
{
_logger.Debug($"Policy data failed");
return;
}
_logger.Debug($"Policy data [policy id: {policy.Id}][path: {policy.Path}][group id: {policy.GroupId}][group name: {group?.Name}]");
_policies.Set(group?.Name ?? string.Empty, policy.Path, policy.Usage, TimeSpan.FromMilliseconds(policy.Span));
_logger.Information($"Policy has been updated [policy id: {policy.Id}]");
}
else else
{ {
_logger.Warning($"Found unknown request type when acknowledging [type: {message.Request.Type}]"); _logger.Warning($"Found unknown request type when acknowledging [type: {message.Request.Type}]");

View File

@ -215,15 +215,6 @@ namespace TwitchChatTTS.Hermes.Socket
}); });
} }
public async Task FetchPolicies()
{
await Send(3, new RequestMessage()
{
Type = "get_policies",
Data = null
});
}
public async Task FetchConnections() public async Task FetchConnections()
{ {
await Send(3, new RequestMessage() await Send(3, new RequestMessage()

View File

@ -8,7 +8,7 @@ namespace TwitchChatTTS.Seven
public string Platform { get; set; } public string Platform { get; set; }
public string Username { get; set; } public string Username { get; set; }
public int EmoteCapacity { get; set; } public int EmoteCapacity { get; set; }
public string EmoteSetId { get; set; } public int? EmoteSetId { get; set; }
public EmoteSet EmoteSet { get; set; } public EmoteSet EmoteSet { get; set; }
public SevenUser User { get; set; } public SevenUser User { get; set; }
} }

View File

@ -80,8 +80,8 @@ s.AddSingleton<IChatCommand, VersionCommand>();
s.AddSingleton<ICommandBuilder, CommandBuilder>(); s.AddSingleton<ICommandBuilder, CommandBuilder>();
s.AddSingleton<IChatterGroupManager, ChatterGroupManager>(); s.AddSingleton<IChatterGroupManager, ChatterGroupManager>();
s.AddSingleton<IGroupPermissionManager, GroupPermissionManager>(); s.AddSingleton<IGroupPermissionManager, GroupPermissionManager>();
s.AddSingleton<ICommandFactory, CommandFactory>();
s.AddSingleton<ICommandManager, CommandManager>(); s.AddSingleton<ICommandManager, CommandManager>();
s.AddTransient<ICommandFactory, CommandFactory>();
s.AddSingleton<TTSPlayer>(); s.AddSingleton<TTSPlayer>();
s.AddSingleton<IRedemptionManager, RedemptionManager>(); s.AddSingleton<IRedemptionManager, RedemptionManager>();
@ -94,8 +94,7 @@ s.AddSingleton<IEmoteDatabase, EmoteDatabase>();
s.AddSingleton<TTSConsumer>(); s.AddSingleton<TTSConsumer>();
s.AddSingleton<TTSPublisher>(); s.AddSingleton<TTSPublisher>();
s.AddSingleton<IChatMessageReader, ChatMessageReader>(); s.AddSingleton<ChatMessageReader>();
s.AddSingleton<IUsagePolicy<long>, UsagePolicy<long>>();
// OBS websocket // OBS websocket
s.AddKeyedSingleton<IWebSocketHandler, HelloHandler>("obs"); s.AddKeyedSingleton<IWebSocketHandler, HelloHandler>("obs");
@ -138,7 +137,7 @@ s.AddKeyedSingleton<ITwitchSocketHandler, SessionWelcomeHandler>("twitch");
s.AddKeyedSingleton<ITwitchSocketHandler, SessionReconnectHandler>("twitch"); s.AddKeyedSingleton<ITwitchSocketHandler, SessionReconnectHandler>("twitch");
s.AddKeyedSingleton<ITwitchSocketHandler, NotificationHandler>("twitch"); s.AddKeyedSingleton<ITwitchSocketHandler, NotificationHandler>("twitch");
s.AddKeyedSingleton<ITwitchSocketHandler, ChannelAdBreakBeginHandler>("twitch-notifications"); s.AddKeyedSingleton<ITwitchSocketHandler, ChannelAdBreakHandler>("twitch-notifications");
s.AddKeyedSingleton<ITwitchSocketHandler, ChannelBanHandler>("twitch-notifications"); s.AddKeyedSingleton<ITwitchSocketHandler, ChannelBanHandler>("twitch-notifications");
s.AddKeyedSingleton<ITwitchSocketHandler, ChannelChatMessageHandler>("twitch-notifications"); s.AddKeyedSingleton<ITwitchSocketHandler, ChannelChatMessageHandler>("twitch-notifications");
s.AddKeyedSingleton<ITwitchSocketHandler, ChannelChatClearHandler>("twitch-notifications"); s.AddKeyedSingleton<ITwitchSocketHandler, ChannelChatClearHandler>("twitch-notifications");

2
TTS.cs
View File

@ -20,7 +20,7 @@ namespace TwitchChatTTS
public class TTS : IHostedService public class TTS : IHostedService
{ {
public const int MAJOR_VERSION = 4; public const int MAJOR_VERSION = 4;
public const int MINOR_VERSION = 4; public const int MINOR_VERSION = 3;
private readonly User _user; private readonly User _user;
private readonly HermesApiClient _hermesApiClient; private readonly HermesApiClient _hermesApiClient;

View File

@ -1,83 +0,0 @@
using Serilog;
using TwitchChatTTS.Twitch.Redemptions;
using TwitchChatTTS.Twitch.Socket.Messages;
namespace TwitchChatTTS.Twitch.Socket.Handlers
{
public class ChannelAdBreakBeginHandler : ITwitchSocketHandler
{
public string Name => "channel.ad_break.begin";
private readonly IRedemptionManager _redemptionManager;
private readonly ILogger _logger;
public ChannelAdBreakBeginHandler(IRedemptionManager redemptionManager, ILogger logger)
{
_redemptionManager = redemptionManager;
_logger = logger;
}
public async Task Execute(TwitchWebsocketClient sender, object data)
{
if (data is not ChannelAdBreakMessage message)
return;
if (message.IsAutomatic)
_logger.Information($"Ad break has begun [duration: {message.DurationSeconds} seconds][automatic: true]");
else
_logger.Information($"Ad break has begun [duration: {message.DurationSeconds} seconds][requester: {message.RequesterUserLogin}][requester id: {message.RequesterUserId}]");
try
{
var actions = _redemptionManager.Get("adbreak_begin");
if (!actions.Any())
{
_logger.Debug($"Found {actions.Count} actions for this Twitch ad break");
foreach (var action in actions)
try
{
await _redemptionManager.Execute(action, message.RequesterUserLogin, long.Parse(message.RequesterUserId));
}
catch (Exception ex)
{
_logger.Error(ex, $"Failed to execute redeemable action [action: {action.Name}][action type: {action.Type}][redeem: ad break begin]");
}
}
else
_logger.Debug($"No redeemable actions for ad break begin was found");
Task.Run(async () =>
{
await Task.Delay(TimeSpan.FromSeconds(message.DurationSeconds));
if (message.IsAutomatic)
_logger.Information($"Ad break has ended [duration: {message.DurationSeconds} seconds][automatic: true]");
else
_logger.Information($"Ad break has ended [duration: {message.DurationSeconds} seconds][requester: {message.RequesterUserLogin}][requester id: {message.RequesterUserId}]");
actions = _redemptionManager.Get("adbreak_end");
if (!actions.Any())
{
_logger.Debug($"Found {actions.Count} actions for this Twitch ad break");
foreach (var action in actions)
try
{
await _redemptionManager.Execute(action, message.RequesterUserLogin, long.Parse(message.RequesterUserId));
}
catch (Exception ex)
{
_logger.Error(ex, $"Failed to execute redeemable action [action: {action.Name}][action type: {action.Type}][redeem: ad break end]");
}
}
else
_logger.Debug($"No redeemable actions for ad break end was found");
});
}
catch (Exception ex)
{
_logger.Error(ex, $"Failed to fetch the redeemable actions for ad break begin");
}
}
}
}

View File

@ -0,0 +1,57 @@
using Serilog;
using TwitchChatTTS.Twitch.Redemptions;
using TwitchChatTTS.Twitch.Socket.Messages;
namespace TwitchChatTTS.Twitch.Socket.Handlers
{
public class ChannelAdBreakHandler : ITwitchSocketHandler
{
public string Name => "channel.ad_break.begin";
private readonly IRedemptionManager _redemptionManager;
private readonly ILogger _logger;
public ChannelAdBreakHandler(IRedemptionManager redemptionManager, ILogger logger)
{
_redemptionManager = redemptionManager;
_logger = logger;
}
public async Task Execute(TwitchWebsocketClient sender, object data)
{
if (data is not ChannelAdBreakMessage message)
return;
bool isAutomatic = message.IsAutomatic == "true";
if (isAutomatic)
_logger.Information($"Ad break has begun [duration: {message.DurationSeconds} seconds][automatic: {isAutomatic}]");
else
_logger.Information($"Ad break has begun [duration: {message.DurationSeconds} seconds][requester: {message.RequesterUserLogin}][requester id: {message.RequesterUserId}]");
try
{
var actions = _redemptionManager.Get("adbreak");
if (!actions.Any())
{
_logger.Debug($"No redeemable actions for ad break was found");
return;
}
_logger.Debug($"Found {actions.Count} actions for this Twitch ad break");
foreach (var action in actions)
try
{
await _redemptionManager.Execute(action, message.RequesterUserLogin, long.Parse(message.RequesterUserId));
}
catch (Exception ex)
{
_logger.Error(ex, $"Failed to execute redeemable action [action: {action.Name}][action type: {action.Type}][redeem: ad break]");
}
}
catch (Exception ex)
{
_logger.Error(ex, $"Failed to fetch the redeemable actions for ad break");
}
}
}
}

View File

@ -1,6 +1,5 @@
using Serilog; using Serilog;
using TwitchChatTTS.Chat.Commands; using TwitchChatTTS.Chat.Commands;
using TwitchChatTTS.Chat.Commands.Limits;
using TwitchChatTTS.Chat.Groups; using TwitchChatTTS.Chat.Groups;
using TwitchChatTTS.Chat.Groups.Permissions; using TwitchChatTTS.Chat.Groups.Permissions;
using TwitchChatTTS.Chat.Messaging; using TwitchChatTTS.Chat.Messaging;
@ -12,20 +11,18 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
{ {
public string Name => "channel.chat.message"; public string Name => "channel.chat.message";
private readonly IChatMessageReader _reader; private readonly ChatMessageReader _reader;
private readonly User _user; private readonly User _user;
private readonly ICommandManager _commands; private readonly ICommandManager _commands;
private readonly IGroupPermissionManager _permissionManager; private readonly IGroupPermissionManager _permissionManager;
private readonly IUsagePolicy<long> _permissionPolicy;
private readonly IChatterGroupManager _chatterGroupManager; private readonly IChatterGroupManager _chatterGroupManager;
private readonly ILogger _logger; private readonly ILogger _logger;
public ChannelChatMessageHandler( public ChannelChatMessageHandler(
IChatMessageReader reader, ChatMessageReader reader,
ICommandManager commands, ICommandManager commands,
IGroupPermissionManager permissionManager, IGroupPermissionManager permissionManager,
IUsagePolicy<long> permissionPolicy,
IChatterGroupManager chatterGroupManager, IChatterGroupManager chatterGroupManager,
User user, User user,
ILogger logger ILogger logger
@ -35,13 +32,8 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
_user = user; _user = user;
_commands = commands; _commands = commands;
_permissionManager = permissionManager; _permissionManager = permissionManager;
_permissionPolicy = permissionPolicy;
_chatterGroupManager = chatterGroupManager; _chatterGroupManager = chatterGroupManager;
_logger = logger; _logger = logger;
_permissionPolicy.Set("everyone", "tts", 100, TimeSpan.FromSeconds(15));
_permissionPolicy.Set("everyone", "tts.chat.messages.read", 3, TimeSpan.FromMilliseconds(15000));
} }
@ -52,8 +44,10 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
if (data is not ChannelChatMessage message) if (data is not ChannelChatMessage message)
return; return;
var broadcasterId = long.Parse(message.BroadcasterUserId);
var chatterId = long.Parse(message.ChatterUserId); var chatterId = long.Parse(message.ChatterUserId);
var chatterLogin = message.ChatterUserLogin; var chatterLogin = message.ChatterUserLogin;
var messageId = message.MessageId;
var fragments = message.Message.Fragments; var fragments = message.Message.Fragments;
var groups = GetGroups(message.Badges, chatterId); var groups = GetGroups(message.Badges, chatterId);
var bits = GetTotalBits(fragments); var bits = GetTotalBits(fragments);
@ -62,22 +56,14 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
if (commandResult != ChatCommandResult.Unknown) if (commandResult != ChatCommandResult.Unknown)
return; return;
string permission = GetPermissionPath(message.ChannelPointsCustomRewardId, bits); if (!HasPermission(message.ChannelPointsCustomRewardId, chatterId, groups, bits))
if (!HasPermission(chatterId, groups, permission))
{ {
_logger.Debug($"Blocked message [chatter: {chatterLogin}][message: {message}]"); _logger.Debug($"Blocked message by {chatterLogin}: {message}");
return; return;
} }
if (!_permissionPolicy.TryUse(chatterId, groups, permission))
{
_logger.Debug($"Chatter has been rate limited from TTS [chatter: {chatterLogin}][chatter id: {chatterId}][message: {message}]");
return;
}
var broadcasterId = long.Parse(message.BroadcasterUserId);
int priority = _chatterGroupManager.GetPriorityFor(groups); int priority = _chatterGroupManager.GetPriorityFor(groups);
await _reader.Read(sender, broadcasterId, chatterId, chatterLogin, message.MessageId, message.Reply, fragments, priority); await _reader.Read(sender, broadcasterId, chatterId, chatterLogin, messageId, message.Reply, fragments, priority);
} }
private async Task<ChatCommandResult> CheckForChatCommand(string arguments, ChannelChatMessage message, IEnumerable<string> groups) private async Task<ChatCommandResult> CheckForChatCommand(string arguments, ChannelChatMessage message, IEnumerable<string> groups)
@ -109,7 +95,7 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
var customGroups = _chatterGroupManager.GetGroupNamesFor(chatterId); var customGroups = _chatterGroupManager.GetGroupNamesFor(chatterId);
return defaultGroups.Union(badgesGroups).Union(customGroups); return defaultGroups.Union(badgesGroups).Union(customGroups);
} }
private int GetTotalBits(TwitchChatFragment[] fragments) private int GetTotalBits(TwitchChatFragment[] fragments)
{ {
return fragments.Where(f => f.Type == "cheermote" && f.Cheermote != null) return fragments.Where(f => f.Type == "cheermote" && f.Cheermote != null)
@ -117,18 +103,14 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
.Sum(); .Sum();
} }
private string GetPermissionPath(string? customRewardId, int bits) private bool HasPermission(string? customRewardId, long chatterId, IEnumerable<string> groups, int bits)
{ {
var permissionPath = "tts.chat.messages.read"; var permissionPath = "tts.chat.messages.read";
if (!string.IsNullOrWhiteSpace(customRewardId)) if (!string.IsNullOrWhiteSpace(customRewardId))
permissionPath = "tts.chat.redemptions.read"; permissionPath = "tts.chat.redemptions.read";
else if (bits > 0) else if (bits > 0)
permissionPath = "tts.chat.bits.read"; permissionPath = "tts.chat.bits.read";
return permissionPath;
}
private bool HasPermission(long chatterId, IEnumerable<string> groups, string permissionPath)
{
return chatterId == _user.OwnerId ? true : _permissionManager.CheckIfAllowed(groups, permissionPath) == true; return chatterId == _user.OwnerId ? true : _permissionManager.CheckIfAllowed(groups, permissionPath) == true;
} }
} }

View File

@ -30,7 +30,7 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
if (!_user.Raids.ContainsKey(message.ToBroadcasterUserId)) if (!_user.Raids.ContainsKey(message.ToBroadcasterUserId))
{ {
chatters = await _api.GetChatters(message.ToBroadcasterUserId.ToString(), _user.TwitchUserId.ToString()); chatters = await _api.GetChatters(_user.TwitchUserId.ToString(), _user.TwitchUserId.ToString());
if (chatters?.Data == null) if (chatters?.Data == null)
{ {
var extraErrorInfo = _user.TwitchUserId.ToString() != message.ToBroadcasterUserId ? " Ensure you have moderator status in your joined channel(s) to prevent raid spam." : string.Empty; var extraErrorInfo = _user.TwitchUserId.ToString() != message.ToBroadcasterUserId ? " Ensure you have moderator status in your joined channel(s) to prevent raid spam." : string.Empty;

View File

@ -9,11 +9,11 @@ namespace TwitchChatTTS.Twitch.Socket.Handlers
{ {
public string Name => "channel.subscription.message"; public string Name => "channel.subscription.message";
private readonly IChatMessageReader _reader; private readonly ChatMessageReader _reader;
private readonly IRedemptionManager _redemptionManager; private readonly IRedemptionManager _redemptionManager;
private readonly ILogger _logger; private readonly ILogger _logger;
public ChannelResubscriptionHandler(IChatMessageReader reader, IRedemptionManager redemptionManager, ILogger logger) public ChannelResubscriptionHandler(ChatMessageReader reader, IRedemptionManager redemptionManager, ILogger logger)
{ {
_reader = reader; _reader = reader;
_redemptionManager = redemptionManager; _redemptionManager = redemptionManager;

View File

@ -2,9 +2,9 @@ namespace TwitchChatTTS.Twitch.Socket.Messages
{ {
public class ChannelAdBreakMessage public class ChannelAdBreakMessage
{ {
public int DurationSeconds { get; set; } public string DurationSeconds { get; set; }
public DateTime StartedAt { get; set; } public DateTime StartedAt { get; set; }
public bool IsAutomatic { get; set; } public string IsAutomatic { get; set; }
public string BroadcasterUserId { get; set; } public string BroadcasterUserId { get; set; }
public string BroadcasterUserLogin { get; set; } public string BroadcasterUserLogin { get; set; }
public string BroadcasterUserName { get; set; } public string BroadcasterUserName { get; set; }

View File

@ -10,6 +10,6 @@ namespace TwitchChatTTS.Twitch.Socket.Messages
} }
public class EventResponsePagination { public class EventResponsePagination {
public string? Cursor { get; set; } public string Cursor { get; set; }
} }
} }