309 lines
11 KiB
C#
309 lines
11 KiB
C#
using System.Text.RegularExpressions;
|
|
using TwitchLib.Client.Events;
|
|
using TwitchChatTTS.OBS.Socket;
|
|
using CommonSocketLibrary.Abstract;
|
|
using CommonSocketLibrary.Common;
|
|
using Serilog;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using TwitchChatTTS;
|
|
using TwitchChatTTS.Seven;
|
|
using TwitchChatTTS.Chat.Commands;
|
|
using TwitchChatTTS.Hermes.Socket;
|
|
using HermesSocketLibrary.Socket.Data;
|
|
|
|
|
|
public class ChatMessageHandler
|
|
{
|
|
private readonly User _user;
|
|
private readonly Configuration _configuration;
|
|
private readonly EmoteDatabase _emotes;
|
|
private readonly TTSPlayer _player;
|
|
private readonly ChatCommandManager _commands;
|
|
private readonly OBSSocketClient? _obsClient;
|
|
private readonly HermesSocketClient? _hermesClient;
|
|
private readonly ILogger _logger;
|
|
|
|
private Regex sfxRegex;
|
|
private HashSet<long> _chatters;
|
|
|
|
public HashSet<long> Chatters { get => _chatters; set => _chatters = value; }
|
|
|
|
|
|
public ChatMessageHandler(
|
|
User user,
|
|
TTSPlayer player,
|
|
ChatCommandManager commands,
|
|
EmoteDatabase emotes,
|
|
[FromKeyedServices("obs")] SocketClient<WebSocketMessage> obsClient,
|
|
[FromKeyedServices("hermes")] SocketClient<WebSocketMessage> hermesClient,
|
|
Configuration configuration,
|
|
ILogger logger
|
|
)
|
|
{
|
|
_user = user;
|
|
_player = player;
|
|
_commands = commands;
|
|
_emotes = emotes;
|
|
_obsClient = obsClient as OBSSocketClient;
|
|
_hermesClient = hermesClient as HermesSocketClient;
|
|
_configuration = configuration;
|
|
_logger = logger;
|
|
|
|
_chatters = null;
|
|
sfxRegex = new Regex(@"\(([A-Za-z0-9_-]+)\)");
|
|
}
|
|
|
|
|
|
public async Task<MessageResult> Handle(OnMessageReceivedArgs e)
|
|
{
|
|
if (_obsClient == null || _hermesClient == null || _obsClient.Connected && _chatters == null)
|
|
return new MessageResult(MessageStatus.NotReady, -1, -1);
|
|
if (_configuration.Twitch?.TtsWhenOffline != true && _obsClient.Live == false)
|
|
return new MessageResult(MessageStatus.NotReady, -1, -1);
|
|
|
|
var m = e.ChatMessage;
|
|
var msg = e.ChatMessage.Message;
|
|
var chatterId = long.Parse(m.UserId);
|
|
var tasks = new List<Task>();
|
|
|
|
var blocked = _user.ChatterFilters.TryGetValue(m.Username, out TTSUsernameFilter? filter) && filter.Tag == "blacklisted";
|
|
if (!blocked || m.IsBroadcaster)
|
|
{
|
|
try
|
|
{
|
|
var commandResult = await _commands.Execute(msg, m);
|
|
if (commandResult != ChatCommandResult.Unknown)
|
|
return new MessageResult(MessageStatus.Command, -1, -1);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Error(ex, $"Failed executing a chat command [message: {msg}][chatter: {m.Username}][cid: {m.UserId}][mid: {m.Id}]");
|
|
}
|
|
}
|
|
|
|
if (blocked)
|
|
{
|
|
_logger.Debug($"Blocked message by {m.Username}: {msg}");
|
|
return new MessageResult(MessageStatus.Blocked, -1, -1);
|
|
}
|
|
|
|
if (_obsClient.Connected && !_chatters.Contains(chatterId))
|
|
{
|
|
tasks.Add(_hermesClient.Send(6, new ChatterMessage()
|
|
{
|
|
Id = chatterId,
|
|
Name = m.Username
|
|
}));
|
|
_chatters.Add(chatterId);
|
|
}
|
|
|
|
// Filter highly repetitive words (like emotes) from the message.
|
|
int totalEmoteUsed = 0;
|
|
var emotesUsed = new HashSet<string>();
|
|
var words = msg.Split(" ");
|
|
var wordCounter = new Dictionary<string, int>();
|
|
string filteredMsg = string.Empty;
|
|
var newEmotes = new Dictionary<string, string>();
|
|
foreach (var w in words)
|
|
{
|
|
if (wordCounter.ContainsKey(w))
|
|
wordCounter[w]++;
|
|
else
|
|
wordCounter.Add(w, 1);
|
|
|
|
var emoteId = _emotes.Get(w);
|
|
if (emoteId == null)
|
|
{
|
|
emoteId = m.EmoteSet.Emotes.FirstOrDefault(e => e.Name == w)?.Id;
|
|
if (emoteId != null)
|
|
{
|
|
newEmotes.Add(emoteId, w);
|
|
_emotes.Add(w, emoteId);
|
|
}
|
|
}
|
|
if (emoteId != null)
|
|
{
|
|
emotesUsed.Add(emoteId);
|
|
totalEmoteUsed++;
|
|
}
|
|
|
|
if (wordCounter[w] <= 4 && (emoteId == null || totalEmoteUsed <= 5))
|
|
filteredMsg += w + " ";
|
|
}
|
|
if (_obsClient.Connected && newEmotes.Any())
|
|
tasks.Add(_hermesClient.Send(7, new EmoteDetailsMessage()
|
|
{
|
|
Emotes = newEmotes
|
|
}));
|
|
msg = filteredMsg;
|
|
|
|
// Replace filtered words.
|
|
if (_user.RegexFilters != null)
|
|
{
|
|
foreach (var wf in _user.RegexFilters)
|
|
{
|
|
if (wf.Search == null || wf.Replace == null)
|
|
continue;
|
|
|
|
if (wf.IsRegex)
|
|
{
|
|
try
|
|
{
|
|
var regex = new Regex(wf.Search);
|
|
msg = regex.Replace(msg, wf.Replace);
|
|
continue;
|
|
}
|
|
catch (Exception)
|
|
{
|
|
wf.IsRegex = false;
|
|
}
|
|
}
|
|
|
|
msg = msg.Replace(wf.Search, wf.Replace);
|
|
}
|
|
}
|
|
|
|
// Determine the priority of this message
|
|
int priority = 0;
|
|
if (m.IsStaff)
|
|
priority = int.MinValue;
|
|
else if (filter?.Tag == "priority")
|
|
priority = int.MinValue + 1;
|
|
else if (m.IsModerator)
|
|
priority = -100;
|
|
else if (m.IsVip)
|
|
priority = -10;
|
|
else if (m.IsPartner)
|
|
priority = -5;
|
|
else if (m.IsHighlighted)
|
|
priority = -1;
|
|
priority = Math.Min(priority, -m.SubscribedMonthCount * (m.IsSubscriber ? 2 : 1));
|
|
|
|
// Determine voice selected.
|
|
string voiceSelected = _user.DefaultTTSVoice;
|
|
if (long.TryParse(e.ChatMessage.UserId, out long userId) && _user.VoicesSelected?.ContainsKey(userId) == true)
|
|
{
|
|
var voiceId = _user.VoicesSelected[userId];
|
|
if (_user.VoicesAvailable.TryGetValue(voiceId, out string? voiceName) && voiceName != null)
|
|
{
|
|
if (_user.VoicesEnabled.Contains(voiceName) || chatterId == _user.OwnerId || m.IsStaff)
|
|
{
|
|
voiceSelected = voiceName;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Determine additional voices used
|
|
var matches = _user.WordFilterRegex?.Matches(msg).ToArray();
|
|
if (matches == null || matches.FirstOrDefault() == null || matches.First().Index < 0)
|
|
{
|
|
HandlePartialMessage(priority, voiceSelected, msg.Trim(), e);
|
|
return new MessageResult(MessageStatus.None, _user.TwitchUserId, chatterId, emotesUsed);
|
|
}
|
|
|
|
HandlePartialMessage(priority, voiceSelected, msg.Substring(0, matches.First().Index).Trim(), e);
|
|
foreach (Match match in matches)
|
|
{
|
|
var message = match.Groups[2].ToString();
|
|
if (string.IsNullOrWhiteSpace(message))
|
|
continue;
|
|
|
|
var voice = match.Groups[1].ToString();
|
|
voice = voice[0].ToString().ToUpper() + voice.Substring(1).ToLower();
|
|
HandlePartialMessage(priority, voice, message.Trim(), e);
|
|
}
|
|
|
|
if (tasks.Any())
|
|
await Task.WhenAll(tasks);
|
|
|
|
return new MessageResult(MessageStatus.None, _user.TwitchUserId, chatterId, emotesUsed);
|
|
}
|
|
|
|
private void HandlePartialMessage(int priority, string voice, string message, OnMessageReceivedArgs e)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(message))
|
|
return;
|
|
|
|
var m = e.ChatMessage;
|
|
var parts = sfxRegex.Split(message);
|
|
var badgesString = string.Join(", ", e.ChatMessage.Badges.Select(b => b.Key + " = " + b.Value));
|
|
|
|
if (parts.Length == 1)
|
|
{
|
|
_logger.Information($"Username: {m.Username}; User ID: {m.UserId}; Voice: {voice}; Priority: {priority}; Message: {message}; Month: {m.SubscribedMonthCount}; {badgesString}");
|
|
_player.Add(new TTSMessage()
|
|
{
|
|
Voice = voice,
|
|
Message = message,
|
|
Moderator = m.IsModerator,
|
|
Timestamp = DateTime.UtcNow,
|
|
Username = m.Username,
|
|
Bits = m.Bits,
|
|
Badges = m.Badges,
|
|
Priority = priority
|
|
});
|
|
return;
|
|
}
|
|
|
|
var sfxMatches = sfxRegex.Matches(message);
|
|
var sfxStart = sfxMatches.FirstOrDefault()?.Index ?? message.Length;
|
|
|
|
for (var i = 0; i < sfxMatches.Count; i++)
|
|
{
|
|
var sfxMatch = sfxMatches[i];
|
|
var sfxName = sfxMatch.Groups[1]?.ToString()?.ToLower();
|
|
|
|
if (!File.Exists("sfx/" + sfxName + ".mp3"))
|
|
{
|
|
parts[i * 2 + 2] = parts[i * 2] + " (" + parts[i * 2 + 1] + ")" + parts[i * 2 + 2];
|
|
continue;
|
|
}
|
|
|
|
if (!string.IsNullOrWhiteSpace(parts[i * 2]))
|
|
{
|
|
_logger.Information($"Username: {m.Username}; User ID: {m.UserId}; Voice: {voice}; Priority: {priority}; Message: {parts[i * 2]}; Month: {m.SubscribedMonthCount}; {badgesString}");
|
|
_player.Add(new TTSMessage()
|
|
{
|
|
Voice = voice,
|
|
Message = parts[i * 2],
|
|
Moderator = m.IsModerator,
|
|
Timestamp = DateTime.UtcNow,
|
|
Username = m.Username,
|
|
Bits = m.Bits,
|
|
Badges = m.Badges,
|
|
Priority = priority
|
|
});
|
|
}
|
|
|
|
_logger.Information($"Username: {m.Username}; User ID: {m.UserId}; Voice: {voice}; Priority: {priority}; SFX: {sfxName}; Month: {m.SubscribedMonthCount}; {badgesString}");
|
|
_player.Add(new TTSMessage()
|
|
{
|
|
Voice = voice,
|
|
Message = sfxName,
|
|
File = $"sfx/{sfxName}.mp3",
|
|
Moderator = m.IsModerator,
|
|
Timestamp = DateTime.UtcNow,
|
|
Username = m.Username,
|
|
Bits = m.Bits,
|
|
Badges = m.Badges,
|
|
Priority = priority
|
|
});
|
|
}
|
|
|
|
if (!string.IsNullOrWhiteSpace(parts.Last()))
|
|
{
|
|
_logger.Information($"Username: {m.Username}; User ID: {m.UserId}; Voice: {voice}; Priority: {priority}; Message: {parts.Last()}; Month: {m.SubscribedMonthCount}; {badgesString}");
|
|
_player.Add(new TTSMessage()
|
|
{
|
|
Voice = voice,
|
|
Message = parts.Last(),
|
|
Moderator = m.IsModerator,
|
|
Timestamp = DateTime.UtcNow,
|
|
Username = m.Username,
|
|
Bits = m.Bits,
|
|
Badges = m.Badges,
|
|
Priority = priority
|
|
});
|
|
}
|
|
}
|
|
} |