75 lines
2.6 KiB
C#
75 lines
2.6 KiB
C#
|
|
||
|
|
using System.Collections.Concurrent;
|
||
|
|
using Serilog;
|
||
|
|
|
||
|
|
namespace TwitchChatTTS.Chat.Groups
|
||
|
|
{
|
||
|
|
public class ChatterGroupManager : IChatterGroupManager
|
||
|
|
{
|
||
|
|
private readonly IDictionary<string, Group> _groups;
|
||
|
|
private readonly IDictionary<long, ICollection<string>> _chatters;
|
||
|
|
private readonly ILogger _logger;
|
||
|
|
|
||
|
|
|
||
|
|
public ChatterGroupManager(ILogger logger) {
|
||
|
|
_logger = logger;
|
||
|
|
_groups = new ConcurrentDictionary<string, Group>();
|
||
|
|
_chatters = new ConcurrentDictionary<long, ICollection<string>>();
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Add(Group group) {
|
||
|
|
_groups.Add(group.Name, group);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Add(long chatter, string groupName) {
|
||
|
|
_chatters.Add(chatter, new List<string>() { groupName });
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Add(long chatter, ICollection<string> groupNames) {
|
||
|
|
if (_chatters.TryGetValue(chatter, out var list)) {
|
||
|
|
foreach (var group in groupNames)
|
||
|
|
list.Add(group);
|
||
|
|
} else
|
||
|
|
_chatters.Add(chatter, groupNames);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Clear() {
|
||
|
|
_groups.Clear();
|
||
|
|
_chatters.Clear();
|
||
|
|
}
|
||
|
|
|
||
|
|
public Group? Get(string groupName) {
|
||
|
|
if (_groups.TryGetValue(groupName, out var group))
|
||
|
|
return group;
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
public IEnumerable<string> GetGroupNamesFor(long chatter) {
|
||
|
|
if (_chatters.TryGetValue(chatter, out var groups))
|
||
|
|
return groups.Select(g => _groups[g].Name);
|
||
|
|
|
||
|
|
return Array.Empty<string>();
|
||
|
|
}
|
||
|
|
|
||
|
|
public int GetPriorityFor(long chatter) {
|
||
|
|
if (!_chatters.TryGetValue(chatter, out var groups))
|
||
|
|
return 0;
|
||
|
|
|
||
|
|
return GetPriorityFor(groups);
|
||
|
|
}
|
||
|
|
|
||
|
|
public int GetPriorityFor(IEnumerable<string> groupNames) {
|
||
|
|
return groupNames.Select(g => _groups.TryGetValue(g, out var group) ? group : null).Where(g => g != null).Max(g => g.Priority);
|
||
|
|
}
|
||
|
|
|
||
|
|
public bool Remove(long chatterId, string groupId) {
|
||
|
|
if (_chatters.TryGetValue(chatterId, out var groups)) {
|
||
|
|
groups.Remove(groupId);
|
||
|
|
_logger.Debug($"Removed chatter from group [chatter id: {chatterId}][group name: {_groups[groupId]}][group id: {groupId}]");
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
_logger.Debug($"Failed to remove chatter from group [chatter id: {chatterId}][group name: {_groups[groupId]}][group id: {groupId}]");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|