Compare commits

...

14 Commits

Author SHA1 Message Date
Tom
3009906b2a Added support for StreamElements' Overlay Key for TTS. Added emote support for stores. Reduced emotes sent to users. 2026-01-03 05:18:03 +00:00
Tom
78b6d4b789 Fixed a few issues. 2025-04-10 17:04:27 +00:00
Tom
c7904f120d Changed various locking mechanism. 2025-03-29 20:18:09 +00:00
Tom
c80f1f2aa0 Added checks for existing permissions when creating, updating and deleting permissions. 2025-03-29 20:11:12 +00:00
Tom
07cb396dd7 Fixed issues with creating/deleting connection messages. 2025-03-29 20:09:22 +00:00
Tom
de3455ee8a Fixed type mapping for database handling. 2025-03-19 23:36:59 +00:00
Tom
6cd8997301 Fixed saving for groups. 2025-03-19 23:36:39 +00:00
Tom
4d0743c4aa Fixed emote duplicate issue. Improved chatter tracking. Used stores in Login Handler in cases where databases was used instead. 2025-03-07 17:34:27 +00:00
Tom
fd0bca5c7c Fixed issues with TTS voice changes. Added proper handling for slave clients. Fixed several stores. Fixed database saving to safely save foreign keys. 2025-03-06 16:11:36 +00:00
Tom
3e717522c2 Added checks for non-user foreign keys in stores. Load/Saving stores' order is now based on table dependencies. Added ability to use chat message when using redemption. 2025-01-28 19:12:14 +00:00
Tom
6d955f245a Added stores for connections. Added requests for groups, group chatters, group permissions & connections. Using TTS Voice State store. 2025-01-17 04:32:31 +00:00
Tom
422cd91db2 Added stores for groups, group chatters and group permissions. Fixed some logging text. 2025-01-16 23:17:04 +00:00
Tom
ee3f128a9f Changed the default way to modify values in store. Added basic validation to stores. Using interfaces to DI store objects. 2025-01-16 19:49:02 +00:00
Tom
b00c72ec2a Added TTS flags. Fixed adding, modifying and loading TTS Filters. Changed TTS Filters loading from database to TTS Fitler Store on login ack. 2025-01-15 14:11:44 +00:00
66 changed files with 1881 additions and 549 deletions

View File

@@ -7,9 +7,14 @@ namespace HermesSocketServer.Models
public required string Id { get; set; } public required string Id { get; set; }
public required User User { get; set; } public required User User { get; set; }
public required ChatterStore Chatters { get; set; } public required ChatterStore Chatters { get; set; }
public required ConnectionStore Connections { get; set; }
public required EmoteStore Emotes { get; set; }
public required GroupStore Groups { get; set; }
public required GroupPermissionStore GroupPermissions { get; set; }
public required PolicyStore Policies { get; set; } public required PolicyStore Policies { get; set; }
public required TTSFilterStore Filters { get; set; } public required TTSFilterStore Filters { get; set; }
public required ActionStore Actions { get; set; } public required ActionStore Actions { get; set; }
public required RedemptionStore Redemptions { get; set; } public required RedemptionStore Redemptions { get; set; }
public required VoiceStateStore VoiceStates { get; set; }
} }
} }

View File

@@ -7,5 +7,6 @@ namespace HermesSocketServer.Models
public required string Email { get; set; } public required string Email { get; set; }
public required string Role { get; set; } public required string Role { get; set; }
public required string DefaultVoice { get; set; } public required string DefaultVoice { get; set; }
public required string StreamElementsOverlayKey { get; set; }
} }
} }

View File

@@ -0,0 +1,53 @@
using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class CreateConnection : IRequest
{
public string Name => "create_connection";
public string[] RequiredKeys => ["name", "type", "client_id", "access_token", "grant_type", "scope", "expiration"];
private ILogger _logger;
public CreateConnection(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
string name = data["name"].ToString()!;
string type = data["type"].ToString()!;
string clientId = data["client_id"].ToString()!;
string accessToken = data["access_token"].ToString()!;
string grantType = data["grant_type"].ToString()!;
string scope = data["scope"].ToString()!;
if (data["expiration"] == null || !DateTime.TryParse(data["expiration"].ToString(), out var expiresAt))
return Task.FromResult(RequestResult.Failed("Expiration needs to be a date time string."));
var previous = channel.Connections.Get(name);
var connection = new Connection()
{
UserId = channel.Id,
Name = name,
Type = type,
ClientId = clientId,
AccessToken = accessToken,
GrantType = grantType,
Scope = scope,
ExpiresAt = expiresAt,
Default = previous?.Default ?? false,
};
bool result = channel.Connections.Set(name, connection);
if (result)
{
_logger.Information($"Added or updated connection to channel [name: {name}][type: {type}][scope: {scope}][expiration: {expiresAt}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(connection));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

51
Requests/CreateGroup.cs Normal file
View File

@@ -0,0 +1,51 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using HermesSocketServer.Store;
using HermesSocketServer.Store.Internal;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class CreateGroup : IRequest
{
public string Name => "create_group";
public string[] RequiredKeys => ["name", "priority"];
private readonly DatabaseTable _table;
private readonly Database _database;
private ILogger _logger;
public CreateGroup([FromKeyedServices("ChatterGroup")] DatabaseTable table, Database database, ILogger logger)
{
_table = table;
_database = database;
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = Guid.NewGuid();
string name = data["name"].ToString()!;
if (!int.TryParse(data["priority"].ToString()!, out var priority))
return Task.FromResult(RequestResult.Failed("Priority needs to be an integer."));
var group = new Group()
{
Id = id.ToString(),
UserId = channel.Id,
Name = name,
Priority = priority,
};
bool result = channel.Groups.Set(id.ToString(), group);
if (result)
{
var store = new ChatterGroupStore(channel.Id, group.Id, _table, _database, _logger);
channel.Groups.Chatters.Add(group.Id, store);
_logger.Information($"Added group to channel [group id: {id}][name: {name}][priority: {priority}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(group));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -0,0 +1,46 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class CreateGroupChatter : IRequest
{
public string Name => "create_group_chatter";
public string[] RequiredKeys => ["group", "chatter", "label"];
private ILogger _logger;
public CreateGroupChatter(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = Guid.NewGuid();
string groupId = data["group"].ToString()!;
if (!int.TryParse(data["chatter"].ToString()!, out var chatterId))
return Task.FromResult(RequestResult.Failed("Priority needs to be an integer."));
string chatterLabel = data["label"].ToString()!;
if (!channel.Groups.Chatters.TryGetValue(groupId, out var chatters))
return Task.FromResult(RequestResult.Failed($"The group does not exist."));
var groupChatter = new GroupChatter()
{
UserId = channel.Id,
GroupId = groupId,
ChatterId = chatterId,
ChatterLabel = chatterLabel,
};
bool result = chatters.Set(chatterId.ToString(), groupChatter);
if (result)
{
_logger.Information($"Added group chatter to channel [group id: {id}][group id: {groupId}][chatter id: {chatterId}][chatter label: {chatterLabel}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(groupChatter));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -0,0 +1,48 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class CreateGroupPermission : IRequest
{
public string Name => "create_group_permission";
public string[] RequiredKeys => ["group", "path", "allow"];
private ILogger _logger;
public CreateGroupPermission(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = Guid.NewGuid();
Guid groupId = new Guid(data["group"].ToString()!);
string path = data["path"].ToString()!;
bool? allow = bool.TryParse(data["allow"].ToString()!, out bool a) ? a : null;
var permission = new GroupPermission()
{
Id = id.ToString(),
UserId = channel.Id,
GroupId = groupId,
Path = path,
Allow = allow,
};
if (channel.GroupPermissions.Get().Values.Any(p => p.GroupId == groupId && p.Path == path))
{
return Task.FromResult(RequestResult.Failed("Permission exists already."));
}
bool result = channel.GroupPermissions.Set(id.ToString(), permission);
if (result)
{
_logger.Information($"Added group permission to channel [permission id: {id}][group id: {groupId}][path: {path}][allow: {allow}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(permission));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -8,7 +8,7 @@ namespace HermesSocketServer.Requests
public class CreateRedeemableAction : IRequest public class CreateRedeemableAction : IRequest
{ {
public string Name => "create_redeemable_action"; public string Name => "create_redeemable_action";
public string[] RequiredKeys => ["name", "data", "type"]; public string[] RequiredKeys => ["name", "has_message", "type", "data"];
private ILogger _logger; private ILogger _logger;
public CreateRedeemableAction(ILogger logger) public CreateRedeemableAction(ILogger logger)
@@ -19,8 +19,9 @@ namespace HermesSocketServer.Requests
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
string name = data["name"].ToString()!; string name = data["name"].ToString()!;
string d = data["data"].ToString()!;
string type = data["type"].ToString()!; string type = data["type"].ToString()!;
bool hasMessage = data["has_message"].ToString()!.ToLower() == "true";
string d = data["data"].ToString()!;
IDictionary<string, string> dict = new Dictionary<string, string>(); IDictionary<string, string> dict = new Dictionary<string, string>();
try try
@@ -29,7 +30,7 @@ namespace HermesSocketServer.Requests
} }
catch (Exception ex) catch (Exception ex)
{ {
_logger.Error(ex, $"Failed to parse data on redeemable action while creating action [name: {name}][type: {type}][data: {d}]"); _logger.Error(ex, $"Failed to parse data on redeemable action while creating action [name: {name}][type: {type}][has message: {hasMessage}][data: {d}]");
return Task.FromResult(RequestResult.Failed("Could not parse the data on this action.")); return Task.FromResult(RequestResult.Failed("Could not parse the data on this action."));
} }
@@ -37,14 +38,15 @@ namespace HermesSocketServer.Requests
{ {
UserId = channel.Id, UserId = channel.Id,
Name = name, Name = name,
Data = dict,
Type = type, Type = type,
HasMessage = hasMessage,
Data = dict,
}; };
bool result = channel.Actions.Set(name, action); bool result = channel.Actions.Set(name, action);
if (result) if (result)
{ {
_logger.Information($"Added redeemable action to channel [name: {name}][type: {type}][channel: {channel.Id}]"); _logger.Information($"Added redeemable action to channel [name: {name}][type: {type}][has message: {hasMessage}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(action)); return Task.FromResult(RequestResult.Successful(action));
} }
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));

View File

@@ -38,7 +38,7 @@ namespace HermesSocketServer.Requests
bool result = channel.Redemptions.Set(id.ToString(), redemption); bool result = channel.Redemptions.Set(id.ToString(), redemption);
if (result) if (result)
{ {
_logger.Information($"Added redemption to channel [id: {id}][redemption id: {redemptionId}][action: {actionName}][order: {order}][channel: {channel.Id}]"); _logger.Information($"Added redemption to channel [redemption id: {id}][twitch redemption id: {redemptionId}][action: {actionName}][order: {order}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(redemption)); return Task.FromResult(RequestResult.Successful(redemption));
} }
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));

View File

@@ -20,6 +20,10 @@ namespace HermesSocketServer.Requests
var id = Guid.NewGuid(); var id = Guid.NewGuid();
string search = data["search"].ToString()!; string search = data["search"].ToString()!;
string replace = data["replace"].ToString()!; string replace = data["replace"].ToString()!;
int flag = 0;
if (data.TryGetValue("flag", out var flagObject)) {
int.TryParse(flagObject.ToString(), out flag);
}
var filter = new TTSWordFilter() var filter = new TTSWordFilter()
{ {
@@ -27,6 +31,7 @@ namespace HermesSocketServer.Requests
UserId = channel.Id, UserId = channel.Id,
Search = search, Search = search,
Replace = replace, Replace = replace,
Flag = flag,
}; };
bool result = channel.Filters.Set(id.ToString(), filter); bool result = channel.Filters.Set(id.ToString(), filter);

View File

@@ -1,4 +1,3 @@
using HermesSocketLibrary.db;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@@ -8,28 +7,26 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "create_tts_user"; public string Name => "create_tts_user";
public string[] RequiredKeys => ["chatter", "voice"]; public string[] RequiredKeys => ["chatter", "voice"];
private Database _database;
private readonly ServerConfiguration _configuration; private readonly ServerConfiguration _configuration;
private ILogger _logger; private ILogger _logger;
public CreateTTSUser(Database database, ServerConfiguration configuration, ILogger logger) public CreateTTSUser(ServerConfiguration configuration, ILogger logger)
{ {
_database = database;
_configuration = configuration; _configuration = configuration;
_logger = logger; _logger = logger;
} }
public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
if (!long.TryParse(data["chatter"].ToString(), out long chatterId)) if (!long.TryParse(data["chatter"].ToString(), out long chatterId))
return RequestResult.Failed("Invalid Twitch user id"); return Task.FromResult(RequestResult.Failed("Invalid Twitch user id."));
data["user"] = channel.Id; data["user"] = channel.Id;
data["voice"] = data["voice"].ToString()!; var voiceId = data["voice"].ToString()!;
var check = await _database.ExecuteScalar("SELECT state FROM \"TtsVoiceState\" WHERE \"userId\" = @user AND \"ttsVoiceId\" = @voice", data) ?? false; var check = channel.VoiceStates.Get(voiceId)?.Enabled ?? false;
if ((check is not bool state || !state) && chatterId != _configuration.Tts.OwnerId) if (!check && chatterId != _configuration.Tts.OwnerId)
return RequestResult.Failed("Voice is disabled on this channel."); return Task.FromResult(RequestResult.Failed("Voice is disabled on this channel."));
bool result = channel.Chatters.Set(chatterId.ToString(), new ChatterVoice() bool result = channel.Chatters.Set(chatterId.ToString(), new ChatterVoice()
{ {
@@ -40,10 +37,10 @@ namespace HermesSocketServer.Requests
if (result) if (result)
{ {
_logger.Information($"Selected a tts voice [voice: {data["voice"]}] for user [chatter: {data["chatter"]}] in channel [channel: {data["user"]}]"); _logger.Information($"Selected a tts voice [voice: {voiceId}] for user [chatter: {chatterId}] in channel [channel: {channel.Id}]");
return RequestResult.Successful(null); return Task.FromResult(RequestResult.Successful(null));
} }
return RequestResult.Failed("Something went wrong when updating the cache."); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
} }
} }
} }

View File

@@ -13,7 +13,7 @@ namespace HermesSocketServer.Requests
private ILogger _logger; private ILogger _logger;
private Random _random; private Random _random;
public CreateTTSVoice(VoiceStore voices, ILogger logger) public CreateTTSVoice(IStore<string, TTSVoice> voices, ILogger logger)
{ {
_voices = voices; _voices = voices;
_logger = logger; _logger = logger;

View File

@@ -0,0 +1,32 @@
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class DeleteConnection : IRequest
{
public string Name => "delete_connection";
public string[] RequiredKeys => ["name"];
private ILogger _logger;
public DeleteConnection(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var connectionName = data["name"].ToString()!;
var result = channel.Connections.Remove(connectionName);
if (result)
{
_logger.Information($"Deleted a connection by name [connection name: {connectionName}]");
return Task.FromResult(RequestResult.Successful(null));
}
_logger.Warning($"Connection Name does not exist [connection name: {connectionName}]");
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

66
Requests/DeleteGroup.cs Normal file
View File

@@ -0,0 +1,66 @@
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class DeleteGroup : IRequest
{
public string Name => "delete_group";
public string[] RequiredKeys => ["id"];
private ILogger _logger;
public DeleteGroup(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var groupIdString = data["id"].ToString()!;
var groupId = new Guid(groupIdString);
var result = channel.Groups.Remove(groupIdString);
if (result)
{
if (channel.Groups.Chatters.TryGetValue(groupId.ToString(), out var chatters))
{
var filteredChatters = chatters.Get().Values;
foreach (var chatter in filteredChatters)
{
var res = chatters.Remove(chatter.ChatterId.ToString(), fromCascade: true);
if (!res)
_logger.Warning($"Failed to delete group chatter by id from cascade [group chatter id: {chatter.ChatterId}]");
}
}
var permissions = channel.GroupPermissions.Get().Values
.Where(p => p.GroupId == groupId);
foreach (var permission in permissions)
{
var res = channel.GroupPermissions.Remove(permission.Id, fromCascade: true);
if (!res)
_logger.Warning($"Failed to delete group permission by id from cascade [group permission id: {permission.Id}]");
}
var policies = channel.Policies.Get().Values
.Where(c => c.GroupId == groupId).ToArray();
foreach (var policy in policies)
{
var res = channel.Policies.Remove(policy.Id.ToString(), fromCascade: true);
if (!res)
_logger.Warning($"Failed to delete group policy by id from cascade [group policy id: {policy.Id}]");
}
if (!channel.Groups.Chatters.Remove(groupIdString))
_logger.Warning($"Failed to delete group chatters from inner store [group id: {groupId}]");
_logger.Information($"Deleted a group by id [group id: {groupId}]");
return Task.FromResult(RequestResult.Successful(null));
}
_logger.Warning($"Group Id does not exist [group id: {groupId}]");
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -0,0 +1,36 @@
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class DeleteGroupChatter : IRequest
{
public string Name => "delete_group_chatter";
public string[] RequiredKeys => ["chatter", "group"];
private ILogger _logger;
public DeleteGroupChatter(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var chatterId = data["chatter"].ToString()!;
var groupId = data["group"].ToString()!;
if (!channel.Groups.Chatters.TryGetValue(groupId, out var chatters))
return Task.FromResult(RequestResult.Failed($"The group does not exist."));
var result = chatters.Remove(chatterId);
if (result)
{
_logger.Information($"Deleted a group chatter by id [group id: {chatterId}]");
return Task.FromResult(RequestResult.Successful(null));
}
_logger.Warning($"Group Chatter Id does not exist [group id: {chatterId}]");
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -0,0 +1,33 @@
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class DeleteGroupPermission : IRequest
{
public string Name => "delete_group_permission";
public string[] RequiredKeys => ["id"];
private ILogger _logger;
public DeleteGroupPermission(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = data["id"].ToString()!;
var permission = channel.GroupPermissions.Get(id);
var result = channel.GroupPermissions.Remove(id);
if (result)
{
_logger.Information($"Deleted a group permission by id [group permission id: {id}]");
return Task.FromResult(RequestResult.Successful(permission));
}
_logger.Warning($"Group Permission Id does not exist [group permission id: {id}]");
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -21,6 +21,15 @@ namespace HermesSocketServer.Requests
if (result) if (result)
{ {
var redemptions = channel.Redemptions.Get().Values
.Where(r => r.ActionName == name);
foreach (var redemption in redemptions)
{
var res = channel.Redemptions.Remove(redemption.Id, fromCascade: true);
if (!res)
_logger.Warning($"Failed to delete redemption by id from cascade [redemption id: {redemption.Id}]");
}
_logger.Information($"Deleted a redeemable action by name [name: {name}]"); _logger.Information($"Deleted a redeemable action by name [name: {name}]");
return Task.FromResult(RequestResult.Successful(null)); return Task.FromResult(RequestResult.Successful(null));
} }

View File

@@ -12,7 +12,7 @@ namespace HermesSocketServer.Requests
private IStore<string, TTSVoice> _voices; private IStore<string, TTSVoice> _voices;
private ILogger _logger; private ILogger _logger;
public DeleteTTSVoice(VoiceStore voices, ILogger logger) public DeleteTTSVoice(IStore<string, TTSVoice> voices, ILogger logger)
{ {
_voices = voices; _voices = voices;
_logger = logger; _logger = logger;

View File

@@ -1,5 +1,3 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@@ -9,35 +7,18 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "get_connections"; public string Name => "get_connections";
public string[] RequiredKeys => []; public string[] RequiredKeys => [];
private Database _database;
private ILogger _logger; private ILogger _logger;
public GetConnections(Database database, ILogger logger) public GetConnections(ILogger logger)
{ {
_database = database;
_logger = logger; _logger = logger;
} }
public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
var temp = new Dictionary<string, object>() { { "user", channel.Id } }; var connections = channel.Connections.Get().Values;
_logger.Information($"Fetched all connections for channel [channel: {channel.Id}]");
var connections = new List<Connection>(); return Task.FromResult(RequestResult.Successful(connections, notifyClientsOnAccount: false));
string sql = "select \"name\", \"type\", \"clientId\", \"accessToken\", \"grantType\", \"scope\", \"expiresAt\", \"default\" from \"Connection\" where \"userId\" = @user";
await _database.Execute(sql, temp, sql =>
connections.Add(new Connection()
{
Name = sql.GetString(0),
Type = sql.GetString(1),
ClientId = sql.GetString(2),
AccessToken = sql.GetString(3),
GrantType = sql.GetString(4),
Scope = sql.GetString(5),
ExpiresAt = sql.GetDateTime(6),
Default = sql.GetBoolean(7)
})
);
return RequestResult.Successful(connections, notifyClientsOnAccount: false);
} }
} }
} }

View File

@@ -8,11 +8,11 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "get_default_tts_voice"; public string Name => "get_default_tts_voice";
public string[] RequiredKeys => []; public string[] RequiredKeys => [];
private readonly UserStore _users; private readonly IStore<string, User> _users;
private readonly ServerConfiguration _configuration; private readonly ServerConfiguration _configuration;
private readonly ILogger _logger; private readonly ILogger _logger;
public GetDefaultTTSVoice(UserStore users, ServerConfiguration configuration, ILogger logger) public GetDefaultTTSVoice(IStore<string, User> users, ServerConfiguration configuration, ILogger logger)
{ {
_users = users; _users = users;
_configuration = configuration; _configuration = configuration;

View File

@@ -1,5 +1,3 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@@ -9,24 +7,16 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "get_emotes"; public string Name => "get_emotes";
public string[] RequiredKeys => []; public string[] RequiredKeys => [];
private Database _database;
private ILogger _logger; private ILogger _logger;
public GetEmotes(Database database, ILogger logger) public GetEmotes(ILogger logger)
{ {
_database = database;
_logger = logger; _logger = logger;
} }
public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
IList<EmoteInfo> emotes = new List<EmoteInfo>(); var emotes = channel.Emotes.Get().Values;
string sql = $"SELECT id, name FROM \"Emote\"";
await _database.Execute(sql, (IDictionary<string, object>?) null, (r) => emotes.Add(new EmoteInfo()
{
Id = r.GetString(0),
Name = r.GetString(1)
}));
_logger.Information($"Fetched all emotes for channel [channel: {channel.Id}]"); _logger.Information($"Fetched all emotes for channel [channel: {channel.Id}]");
return RequestResult.Successful(emotes, notifyClientsOnAccount: false); return RequestResult.Successful(emotes, notifyClientsOnAccount: false);
} }

View File

@@ -0,0 +1,24 @@
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class GetGroupPermissions : IRequest
{
public string Name => "get_group_permissions";
public string[] RequiredKeys => [];
private readonly ILogger _logger;
public GetGroupPermissions(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var permissions = channel.GroupPermissions.Get().Values;
_logger.Information($"Fetched all group permissions for channel [channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(permissions, notifyClientsOnAccount: false));
}
}
}

37
Requests/GetGroups.cs Normal file
View File

@@ -0,0 +1,37 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class GetGroups : IRequest
{
public string Name => "get_groups";
public string[] RequiredKeys => [];
private readonly ILogger _logger;
public GetGroups(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var groups = channel.Groups.Get().Values;
var chatters = channel.Groups.Chatters;
var all = groups.Select(g => new GroupDetails()
{
Group = g,
Chatters = chatters[g.Id].Get().Values,
});
_logger.Information($"Fetched all groups for channel [channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(all, notifyClientsOnAccount: false));
}
private class GroupDetails
{
public required Group Group { get; set; }
public required IEnumerable<GroupChatter> Chatters { get; set; }
}
}
}

View File

@@ -1,4 +1,3 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages; using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@@ -9,54 +8,27 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "get_permissions"; public string Name => "get_permissions";
public string[] RequiredKeys => []; public string[] RequiredKeys => [];
private readonly Database _database;
private readonly ILogger _logger; private readonly ILogger _logger;
public GetPermissions(Database database, ILogger logger) public GetPermissions(ILogger logger)
{ {
_database = database;
_logger = logger; _logger = logger;
} }
public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
var temp = new Dictionary<string, object>() { { "user", channel.Id } }; var groups = channel.Groups.Get().Values;
var groupChatters = channel.Groups.Chatters.Values.SelectMany(g => g.Get().Values);
var groups = new List<Group>(); var groupPermissions = channel.GroupPermissions.Get().Values;
string sql = $"SELECT id, name, priority FROM \"Group\" WHERE \"userId\" = @user";
await _database.Execute(sql, temp, (r) => groups.Add(new Group()
{
Id = r.GetGuid(0).ToString("D"),
Name = r.GetString(1),
Priority = r.GetInt32(2)
}));
var groupChatters = new List<GroupChatter>();
sql = $"SELECT \"groupId\", \"chatterId\", \"chatterId\" FROM \"ChatterGroup\" WHERE \"userId\" = @user";
await _database.Execute(sql, temp, (r) => groupChatters.Add(new GroupChatter()
{
GroupId = r.GetGuid(0).ToString("D"),
ChatterId = r.GetInt32(1)
}));
var groupPermissions = new List<GroupPermission>();
sql = $"SELECT id, \"groupId\", \"path\", \"allow\" FROM \"GroupPermission\" WHERE \"userId\" = @user";
await _database.Execute(sql, temp, (r) => groupPermissions.Add(new GroupPermission()
{
Id = r.GetGuid(0).ToString("D"),
GroupId = r.GetGuid(1).ToString("D"),
Path = r.GetString(2),
Allow = r.GetBoolean(3)
}));
_logger.Information($"Fetched all permissions for channel [channel: {channel.Id}]"); _logger.Information($"Fetched all permissions for channel [channel: {channel.Id}]");
var info = new GroupInfo() var info = new GroupInfo()
{ {
Groups = groups, Groups = groups,
GroupChatters = groupChatters, GroupChatters = groupChatters,
GroupPermissions = groupPermissions GroupPermissions = groupPermissions,
}; };
return RequestResult.Successful(info, notifyClientsOnAccount: false); return Task.FromResult(RequestResult.Successful(info, notifyClientsOnAccount: false));
} }
} }
} }

View File

@@ -9,10 +9,10 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "get_tts_voices"; public string Name => "get_tts_voices";
public string[] RequiredKeys => []; public string[] RequiredKeys => [];
private VoiceStore _voices; private IStore<string, TTSVoice> _voices;
private ILogger _logger; private ILogger _logger;
public GetTTSVoices(VoiceStore voices, ILogger logger) public GetTTSVoices(IStore<string, TTSVoice> voices, ILogger logger)
{ {
_voices = voices; _voices = voices;
_logger = logger; _logger = logger;

View File

@@ -0,0 +1,50 @@
using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class UpdateConnection : IRequest
{
public string Name => "update_connection";
public string[] RequiredKeys => ["name", "type", "clientId", "accessToken", "grantType", "scope", "expiration"];
private ILogger _logger;
public UpdateConnection(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
string name = data["name"].ToString()!;
string type = data["type"].ToString()!;
string clientId = data["clientId"].ToString()!;
string accessToken = data["accessToken"].ToString()!;
string grantType = data["grantType"].ToString()!;
string scope = data["scope"].ToString()!;
if (!DateTime.TryParse(data["expiration"].ToString()!, out var expiresAt))
return Task.FromResult(RequestResult.Failed("Expiration needs to be a date time string."));
var connection = new Connection()
{
UserId = channel.Id,
Name = name,
Type = type,
ClientId = clientId,
AccessToken = accessToken,
GrantType = grantType,
Scope = scope,
ExpiresAt = expiresAt,
};
bool result = channel.Connections.Modify(name, connection);
if (result)
{
_logger.Information($"Added connection to channel [name: {name}][type: {type}][scope: {scope}][expiration: {expiresAt}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(connection));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -8,10 +8,10 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "update_default_tts_voice"; public string Name => "update_default_tts_voice";
public string[] RequiredKeys => ["user", "voice"]; public string[] RequiredKeys => ["user", "voice"];
private UserStore _users; private IStore<string, User> _users;
private ILogger _logger; private ILogger _logger;
public UpdateDefaultTTSVoice(UserStore users, ILogger logger) public UpdateDefaultTTSVoice(IStore<string, User> users, ILogger logger)
{ {
_users = users; _users = users;
_logger = logger; _logger = logger;

42
Requests/UpdateGroup.cs Normal file
View File

@@ -0,0 +1,42 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class UpdateGroup : IRequest
{
public string Name => "update_group";
public string[] RequiredKeys => ["id", "name", "priority"];
private ILogger _logger;
public UpdateGroup(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = data["id"].ToString()!;
string name = data["name"].ToString()!;
if (!int.TryParse(data["priority"].ToString()!, out var priority))
return Task.FromResult(RequestResult.Failed("Priority needs to be an integer."));
var group = new Group()
{
Id = id,
UserId = channel.Id,
Name = name,
Priority = priority,
};
bool result = channel.Groups.Modify(id, group);
if (result)
{
_logger.Information($"Updated group on channel [group id: {id}][name: {name}][priority: {priority}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(group));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -0,0 +1,46 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class UpdateGroupChatter : IRequest
{
public string Name => "update_group_chatter";
public string[] RequiredKeys => ["group", "chatter", "label"];
private ILogger _logger;
public UpdateGroupChatter(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = Guid.NewGuid();
string groupId = data["group"].ToString()!;
if (!int.TryParse(data["chatter"].ToString()!, out var chatterId))
return Task.FromResult(RequestResult.Failed("Chatter Id needs to be an integer."));
string chatterLabel = data["label"].ToString()!;
var groupChatter = new GroupChatter()
{
UserId = channel.Id,
GroupId = groupId,
ChatterId = chatterId,
ChatterLabel = chatterLabel,
};
if (!channel.Groups.Chatters.TryGetValue(groupId, out var chatters))
return Task.FromResult(RequestResult.Failed($"The group does not exist."));
bool result = chatters.Modify(chatterId.ToString(), groupChatter);
if (result)
{
_logger.Information($"Updated group chatter on channel [group id: {id}][group id: {groupId}][chatter id: {chatterId}][chatter label: {chatterLabel}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(groupChatter));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -0,0 +1,48 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests
{
public class UpdateGroupPermission : IRequest
{
public string Name => "update_group_permission";
public string[] RequiredKeys => ["id", "group", "path", "allow"];
private ILogger _logger;
public UpdateGroupPermission(ILogger logger)
{
_logger = logger;
}
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{
var id = data["id"].ToString()!;
Guid groupId = new Guid(data["group"].ToString()!);
string path = data["path"].ToString()!;
bool? allow = bool.TryParse(data["allow"].ToString()!, out bool a) ? a : null;
var permission = new GroupPermission()
{
Id = id.ToString(),
UserId = channel.Id,
GroupId = groupId,
Path = path,
Allow = allow,
};
if (!channel.GroupPermissions.Get().Values.Any(p => p.GroupId == groupId && p.Path == path))
{
return Task.FromResult(RequestResult.Failed("Permission does not exist."));
}
bool result = channel.GroupPermissions.Modify(id.ToString(), permission);
if (result)
{
_logger.Information($"Updated group permission on channel [permission id: {id}][group id: {groupId}][path: {path}][allow: {allow}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(permission));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
}
}
}

View File

@@ -23,7 +23,7 @@ namespace HermesSocketServer.Requests
int count = int.Parse(data["count"].ToString()!); int count = int.Parse(data["count"].ToString()!);
int span = int.Parse(data["span"].ToString()!); int span = int.Parse(data["span"].ToString()!);
bool result = channel.Policies.Set(id.ToString(), new Policy() var policy = new Policy()
{ {
Id = id, Id = id,
UserId = channel.Id, UserId = channel.Id,
@@ -31,12 +31,12 @@ namespace HermesSocketServer.Requests
Path = path, Path = path,
Usage = count, Usage = count,
Span = span, Span = span,
}); };
bool result = channel.Policies.Modify(id.ToString(), policy);
if (result) if (result)
{ {
var policy = channel.Policies.Get(id.ToString()); _logger.Information($"Updated policy on channel [policy id: {id}][group id: {groupId}][path: {path}][count: {count}][span: {span}][channel: {channel.Id}]");
_logger.Information($"Updated policy to channel [policy id: {id}][group id: {groupId}][path: {path}][count: {count}][span: {span}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(policy)); return Task.FromResult(RequestResult.Successful(policy));
} }
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));

View File

@@ -19,8 +19,9 @@ namespace HermesSocketServer.Requests
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
string name = data["name"].ToString()!; string name = data["name"].ToString()!;
string d = data["data"].ToString()!;
string type = data["type"].ToString()!; string type = data["type"].ToString()!;
bool hasMessage = data["has_message"].ToString()!.ToLower() == "true";
string d = data["data"].ToString()!;
IDictionary<string, string> dict = new Dictionary<string, string>(); IDictionary<string, string> dict = new Dictionary<string, string>();
try try
@@ -29,7 +30,7 @@ namespace HermesSocketServer.Requests
} }
catch (Exception ex) catch (Exception ex)
{ {
_logger.Error(ex, $"Failed to parse data on redeemable action while updating action [name: {name}][type: {type}][data: {d}]"); _logger.Error(ex, $"Failed to parse data on redeemable action while updating action [name: {name}][type: {type}][has message: {hasMessage}][data: {d}]");
return Task.FromResult(RequestResult.Failed("Could not parse the data on this action.")); return Task.FromResult(RequestResult.Failed("Could not parse the data on this action."));
} }
@@ -37,18 +38,15 @@ namespace HermesSocketServer.Requests
{ {
UserId = channel.Id, UserId = channel.Id,
Name = name, Name = name,
Data = dict,
Type = type, Type = type,
HasMessage = hasMessage,
Data = dict,
}; };
bool result = channel.Actions.Modify(name, action => bool result = channel.Actions.Modify(name, action);
{
action.Type = type;
action.Data = dict;
});
if (result) if (result)
{ {
_logger.Information($"Added redeemable action to channel [name: {name}][type: {type}][channel: {channel.Id}]"); _logger.Information($"Updated redeemable action on channel [name: {name}][type: {type}][has message: {hasMessage}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(action)); return Task.FromResult(RequestResult.Successful(action));
} }
if (channel.Actions.Get(name) == null) if (channel.Actions.Get(name) == null)

View File

@@ -1,3 +1,4 @@
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@@ -6,7 +7,7 @@ namespace HermesSocketServer.Requests
public class UpdateRedemption : IRequest public class UpdateRedemption : IRequest
{ {
public string Name => "update_redemption"; public string Name => "update_redemption";
public string[] RequiredKeys => ["id", "redemption", "action", "order"]; public string[] RequiredKeys => ["id", "redemption", "action", "order", "state"];
private ILogger _logger; private ILogger _logger;
public UpdateRedemption(ILogger logger) public UpdateRedemption(ILogger logger)
@@ -25,26 +26,21 @@ namespace HermesSocketServer.Requests
return Task.FromResult(RequestResult.Failed("Order must be an integer.")); return Task.FromResult(RequestResult.Failed("Order must be an integer."));
bool state = data["state"].ToString()?.ToLower() == "true"; bool state = data["state"].ToString()?.ToLower() == "true";
bool result = channel.Redemptions.Modify(id, r => Redemption redemption = new Redemption() {
{ Id = id,
if (r.UserId != channel.Id) UserId = channel.Id,
return; RedemptionId = redemptionId,
ActionName = actionName,
Order = order,
State = state,
};
r.RedemptionId = redemptionId; bool result = channel.Redemptions.Modify(id, redemption);
r.ActionName = actionName;
r.Order = order;
r.State = state;
});
var r = channel.Redemptions.Get(id);
if (result) if (result)
{ {
_logger.Information($"Updated redemption to channel [id: {id}][redemption id: {redemptionId}][action: {actionName}][order: {order}][channel: {channel.Id}]"); _logger.Information($"Updated redemption on channel [id: {id}][redemption id: {redemptionId}][action: {actionName}][order: {order}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(r)); return Task.FromResult(RequestResult.Successful(redemption));
} }
if (r == null || r.UserId != channel.Id)
return Task.FromResult(RequestResult.Failed("Redemption does not exist."));
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
} }
} }

View File

@@ -21,6 +21,10 @@ namespace HermesSocketServer.Requests
var id = data["id"].ToString()!; var id = data["id"].ToString()!;
string search = data["search"].ToString()!; string search = data["search"].ToString()!;
string replace = data["replace"].ToString()!; string replace = data["replace"].ToString()!;
int flag = -1;
if (data.TryGetValue("flag", out var flagObject)) {
int.TryParse(flagObject.ToString(), out flag);
}
var filter = new TTSWordFilter() var filter = new TTSWordFilter()
{ {
@@ -28,13 +32,13 @@ namespace HermesSocketServer.Requests
UserId = channel.Id, UserId = channel.Id,
Search = search, Search = search,
Replace = replace, Replace = replace,
Flag = flag,
}; };
bool result = channel.Filters.Set(id, filter); bool result = channel.Filters.Modify(id, filter);
if (result) if (result)
{ {
_logger.Information($"Updated filter to channel [filter id: {id}][search: {search}][replace: {replace}][channel: {channel.Id}]"); _logger.Information($"Updated filter on channel [filter id: {id}][search: {search}][replace: {replace}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(filter)); return Task.FromResult(RequestResult.Successful(filter));
} }
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));

View File

@@ -1,6 +1,4 @@
using HermesSocketLibrary.db;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using HermesSocketServer.Services;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Requests namespace HermesSocketServer.Requests
@@ -9,41 +7,42 @@ namespace HermesSocketServer.Requests
{ {
public string Name => "update_tts_user"; public string Name => "update_tts_user";
public string[] RequiredKeys => ["chatter", "voice"]; public string[] RequiredKeys => ["chatter", "voice"];
private Database _database;
private readonly ServerConfiguration _configuration; private readonly ServerConfiguration _configuration;
private ILogger _logger; private ILogger _logger;
public UpdateTTSUser(Database database, ServerConfiguration configuration, ILogger logger) public UpdateTTSUser(ServerConfiguration configuration, ILogger logger)
{ {
_database = database;
_configuration = configuration; _configuration = configuration;
_logger = logger; _logger = logger;
} }
public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
if (long.TryParse(data["chatter"].ToString(), out long chatterId)) if (!long.TryParse(data["chatter"].ToString(), out long chatterId))
data["chatter"] = chatterId; return Task.FromResult(RequestResult.Failed("Invalid Twitch user id."));
data["voice"] = data["voice"].ToString()!;
data["user"] = channel.Id; data["user"] = channel.Id;
var voiceId = data["voice"].ToString()!;
var check = await _database.ExecuteScalar("SELECT state FROM \"TtsVoiceState\" WHERE \"userId\" = @user AND \"ttsVoiceId\" = @voice", data) ?? false; var check = channel.VoiceStates.Get(voiceId)?.Enabled ?? false;
if ((check is not bool state || !state) && chatterId != _configuration.Tts.OwnerId) if (!check && chatterId != _configuration.Tts.OwnerId)
return RequestResult.Failed("Voice is either non-existent or disabled on this channel."); return Task.FromResult(RequestResult.Failed("Voice is either non-existent or disabled on this channel."));
var result = channel.Chatters.Set(chatterId.ToString(), new ChatterVoice() var voice = new ChatterVoice()
{ {
UserId = channel.Id, UserId = channel.Id,
ChatterId = chatterId, ChatterId = chatterId,
VoiceId = data["voice"].ToString()! VoiceId = voiceId
}); };
var result = channel.Chatters.Modify(chatterId.ToString(), voice);
if (result) if (result)
{ {
_logger.Information($"Updated chatter's [chatter: {data["chatter"]}] selected tts voice [voice: {data["voice"]}] in channel [channel: {channel.Id}]"); _logger.Information($"Updated chatter's selected tts voice on channel [chatter id: {chatterId}][voice id: {voiceId}][channel: {channel.Id}]");
return RequestResult.Successful(null); return Task.FromResult(RequestResult.Successful(voice));
} }
return RequestResult.Failed("Soemthing went wrong when updating the cache."); return Task.FromResult(RequestResult.Failed("Soemthing went wrong when updating the cache."));
} }
} }
} }

View File

@@ -12,7 +12,7 @@ namespace HermesSocketServer.Requests
private IStore<string, TTSVoice> _voices; private IStore<string, TTSVoice> _voices;
private ILogger _logger; private ILogger _logger;
public UpdateTTSVoice(VoiceStore voices, ILogger logger) public UpdateTTSVoice(IStore<string, TTSVoice> voices, ILogger logger)
{ {
_voices = voices; _voices = voices;
_logger = logger; _logger = logger;
@@ -23,15 +23,17 @@ namespace HermesSocketServer.Requests
string voiceName = data["voice"].ToString()!; string voiceName = data["voice"].ToString()!;
string voiceId = data["voiceid"].ToString()!; string voiceId = data["voiceid"].ToString()!;
var result = _voices.Set(voiceId, new TTSVoice() var voice = new TTSVoice()
{ {
Id = voiceId, Id = voiceId,
Name = voiceName Name = voiceName
}); };
var result = _voices.Modify(voiceId, voice);
if (result) if (result)
{ {
_logger.Information($"Updated voice's [voice id: {voiceId}] name [new name: {voiceName}]"); _logger.Information($"Updated voice's name on channel [voice id: {voiceId}][name: {voiceName}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(null)); return Task.FromResult(RequestResult.Successful(voice));
} }
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the cache."));
} }

View File

@@ -1,4 +1,5 @@
using HermesSocketLibrary.db; using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@@ -17,18 +18,25 @@ namespace HermesSocketServer.Requests
_logger = logger; _logger = logger;
} }
public async Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
data["voice"] = data["voice"].ToString()!; var voiceId = data["voice"].ToString()!;
data["state"] = data["state"].ToString() == "True"; var state = data["state"].ToString()?.ToLower() == "true";
data["user"] = channel.Id;
string sql = "INSERT INTO \"TtsVoiceState\" (\"userId\", \"ttsVoiceId\", state) VALUES (@user, @voice, @state) ON CONFLICT (\"userId\", \"ttsVoiceId\") DO UPDATE SET state = @state"; var voiceState = new TTSVoiceState()
var result = await _database.Execute(sql, data); {
_logger.Information($"Updated voice's [voice id: {data["voice"]}] state [new state: {data["state"]}][channel: {data["user"]}]"); Id = voiceId,
if (result > 0) UserId = channel.Id,
return RequestResult.Successful(null); Enabled = state,
return RequestResult.Failed("Something went wrong when updating the database."); };
var result = channel.VoiceStates.Set(voiceId, voiceState);
if (result)
{
_logger.Information($"Updated voice state on channel [voice id: {voiceId}][state: {state}][channel: {channel.Id}]");
return Task.FromResult(RequestResult.Successful(voiceState));
}
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the database."));
} }
} }
} }

106
Server.cs
View File

@@ -32,7 +32,7 @@ namespace HermesSocketLibrary
public async Task Handle(WebSocketUser socket, HttpContext context) public async Task Handle(WebSocketUser socket, HttpContext context)
{ {
_logger.Information($"Socket connected [ip: {socket.IPAddress}][uid: {socket.UID}]"); _logger.Information($"Socket connected [ip: {socket.IPAddress}][uid: {socket.SessionId}]");
_sockets.Add(socket); _sockets.Add(socket);
var buffer = new byte[1024 * 8]; var buffer = new byte[1024 * 8];
@@ -44,70 +44,51 @@ namespace HermesSocketLibrary
if (result == null || result.MessageType == WebSocketMessageType.Close || !socket.Connected) if (result == null || result.MessageType == WebSocketMessageType.Close || !socket.Connected)
break; break;
string message = Encoding.UTF8.GetString(buffer, 0, result.Count).TrimEnd('\0'); string messageString = Encoding.UTF8.GetString(buffer, 0, result.Count).TrimEnd('\0');
var obj = JsonSerializer.Deserialize<WebSocketMessage>(message, _options); var message = JsonSerializer.Deserialize<WebSocketMessage>(messageString, _options);
if (obj == null) if (message == null)
continue; continue;
if (obj.OpCode != 0) if (message.OpCode != 0)
_logger.Information($"rxm: {message} [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); _logger.Information($"receive: {messageString} [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
int[] nonProtectedOps = { 0, 1 }; if (message.OpCode < 0 || message.OpCode > 8 || message.OpCode == 2 || message.OpCode == 4)
if (string.IsNullOrEmpty(socket.Id) && !nonProtectedOps.Contains(obj.OpCode))
{ {
_logger.Warning($"An attempt was made to use protected routes while not logged in [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); await socket.Send(5, new LoggingMessage("Received an invalid message: " + messageString, HermesLoggingLevel.Error));
return; break;
}
bool loggedIn = !string.IsNullOrEmpty(socket.Id);
int[] nonProtectedOps = { 0, 1 };
if (!loggedIn && !nonProtectedOps.Contains(message.OpCode))
{
_logger.Warning($"An attempt was made to use protected routes while not logged in [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
break;
} }
int[] protectedOps = { 0, 3, 5, 6, 7, 8 }; int[] protectedOps = { 0, 3, 5, 6, 7, 8 };
if (!string.IsNullOrEmpty(socket.Id) && !protectedOps.Contains(obj.OpCode)) if (loggedIn && !protectedOps.Contains(message.OpCode))
{ {
_logger.Warning($"An attempt was made to use non-protected routes while logged in [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); _logger.Warning($"An attempt was made to use non-protected routes while logged in [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
return; break;
} }
/** if (message.Data == null)
* 0: Heartbeat
* 1: Login RX
* 2: Login Ack TX
* 3: Request RX
* 4: Request Ack TX
* 5: Logging RX/TX
*/
if (obj.Data == null)
{ {
await socket.Send(5, new LoggingMessage("Received no data in the message.", HermesLoggingLevel.Warn)); await socket.Send(5, new LoggingMessage("Received no data in the message.", HermesLoggingLevel.Warn));
continue; continue;
} }
string data = obj.Data.ToString()!; string data = message.Data.ToString()!;
if (obj.OpCode == 0) message.Data = DeserializeData(message.OpCode, data);
obj.Data = JsonSerializer.Deserialize<HeartbeatMessage>(data, _options); await _handlers.Execute(socket, message.OpCode, message.Data);
else if (obj.OpCode == 1)
obj.Data = JsonSerializer.Deserialize<HermesLoginMessage>(data, _options);
else if (obj.OpCode == 3)
obj.Data = JsonSerializer.Deserialize<RequestMessage>(data, _options);
else if (obj.OpCode == 5)
obj.Data = JsonSerializer.Deserialize<LoggingMessage>(data, _options);
else if (obj.OpCode == 6)
obj.Data = JsonSerializer.Deserialize<ChatterMessage>(data, _options);
else if (obj.OpCode == 7)
obj.Data = JsonSerializer.Deserialize<EmoteDetailsMessage>(data, _options);
else if (obj.OpCode == 8)
obj.Data = JsonSerializer.Deserialize<EmoteUsageMessage>(data, _options);
else
{
await socket.Send(5, new LoggingMessage("Received an invalid message: " + message, HermesLoggingLevel.Error));
continue;
}
await _handlers.Execute(socket, obj.OpCode, obj.Data);
} }
catch (WebSocketException wse) catch (WebSocketException wse)
{ {
_logger.Error(wse, $"Error trying to process a socket message [code: {wse.ErrorCode}][ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); _logger.Error(wse, $"Error trying to process a socket message [code: {wse.ErrorCode}][ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
} }
catch (Exception e) catch (Exception e)
{ {
_logger.Error(e, $"Error trying to process a socket message [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); _logger.Error(e, $"Error trying to process a socket message [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
} }
} }
@@ -118,14 +99,45 @@ namespace HermesSocketLibrary
} }
catch (Exception e) catch (Exception e)
{ {
_logger.Information(e, $"Client failed to disconnect [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); _logger.Warning(e, $"Client failed to disconnect [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
} }
finally finally
{ {
socket.Dispose(); socket.Dispose();
_sockets.Remove(socket); _sockets.Remove(socket);
} }
_logger.Information($"Client disconnected [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.UID}]"); _logger.Information($"Client disconnected [ip: {socket.IPAddress}][id: {socket.Id}][name: {socket.Name}][token: {socket.ApiKey}][uid: {socket.SessionId}]");
// Update slave status of another client from the same user if available.
if (socket.Id != null && !socket.Slave)
{
var client = _sockets.GetSockets(socket.Id).Where(s => !s.WebLogin).FirstOrDefault();
if (client != null)
{
await client.Send(9, new SlaveMessage() { Slave = false });
client.Slave = false;
socket.Slave = true;
}
}
}
private object? DeserializeData(int opcode, string data)
{
if (opcode == 0)
return JsonSerializer.Deserialize<HeartbeatMessage>(data, _options);
else if (opcode == 1)
return JsonSerializer.Deserialize<HermesLoginMessage>(data, _options);
else if (opcode == 3)
return JsonSerializer.Deserialize<RequestMessage>(data, _options);
else if (opcode == 5)
return JsonSerializer.Deserialize<LoggingMessage>(data, _options);
else if (opcode == 6)
return JsonSerializer.Deserialize<ChatterMessage>(data, _options);
else if (opcode == 7)
return JsonSerializer.Deserialize<EmoteDetailsMessage>(data, _options);
else if (opcode == 8)
return JsonSerializer.Deserialize<EmoteUsageMessage>(data, _options);
return null;
} }
} }
} }

View File

@@ -1,5 +1,6 @@
using System.Collections.Concurrent; using System.Collections.Concurrent;
using HermesSocketLibrary.db; using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models; using HermesSocketServer.Models;
using HermesSocketServer.Store; using HermesSocketServer.Store;
@@ -7,89 +8,146 @@ namespace HermesSocketServer.Services
{ {
public class ChannelManager public class ChannelManager
{ {
private readonly UserStore _users; private readonly IStore<string, User> _users;
private readonly Database _database; private readonly Database _database;
private readonly IStore<string, TTSVoice> _voices;
private readonly ServerConfiguration _configuration; private readonly ServerConfiguration _configuration;
private readonly Serilog.ILogger _logger; private readonly Serilog.ILogger _logger;
private readonly IDictionary<string, Channel> _channels; private readonly IDictionary<string, Channel> _channels;
private readonly object _lock; private readonly Mutex _mutex;
public ChannelManager(UserStore users, Database database, ServerConfiguration configuration, Serilog.ILogger logger) public ChannelManager(IStore<string, User> users, Database database, IStore<string, TTSVoice> voices, ServerConfiguration configuration, Serilog.ILogger logger)
{ {
_users = users; _users = users;
_database = database; _database = database;
_voices = voices;
_configuration = configuration; _configuration = configuration;
_logger = logger; _logger = logger;
_channels = new ConcurrentDictionary<string, Channel>(); _channels = new ConcurrentDictionary<string, Channel>();
_lock = new object(); _mutex = new Mutex();
} }
public Task<Channel?> Add(string userId) public async Task<Channel?> Add(string userId)
{ {
var user = _users.Get(userId); var user = _users.Get(userId);
if (user == null) if (user == null)
return Task.FromResult<Channel?>(null); return null;
lock (_lock) return await Task.Run(() =>
{ {
if (_channels.ContainsKey(userId)) try
return Task.FromResult<Channel?>(null);
var actionTable = _configuration.Database.Tables["Action"];
var chatterTable = _configuration.Database.Tables["Chatter"];
var policyTable = _configuration.Database.Tables["Policy"];
var redemptionTable = _configuration.Database.Tables["Redemption"];
var ttsFilterTable = _configuration.Database.Tables["TtsFilter"];
var chatters = new ChatterStore(userId, chatterTable, _database, _logger);
var policies = new PolicyStore(userId, policyTable, _database, _logger);
var filters = new TTSFilterStore(userId, ttsFilterTable, _database, _logger);
var actions = new ActionStore(userId, actionTable, _database, _logger);
var redemptions = new RedemptionStore(userId, redemptionTable, _database, _logger);
Task.WaitAll([
chatters.Load(),
policies.Load(),
filters.Load(),
actions.Load(),
redemptions.Load(),
]);
var channel = new Channel()
{ {
Id = userId, _mutex.WaitOne();
User = user, if (_channels.TryGetValue(userId, out var channel))
Chatters = chatters, return channel;
Policies = policies,
Filters = filters,
Actions = actions,
Redemptions = redemptions,
};
_channels.Add(userId, channel); var actionTable = _configuration.Database.Tables["Action"];
return Task.FromResult<Channel?>(channel); var chatterTable = _configuration.Database.Tables["Chatter"];
} var connectionTable = _configuration.Database.Tables["Connection"];
var connectionStateTable = _configuration.Database.Tables["ConnectionState"];
var emoteTable = _configuration.Database.Tables["Emote"];
var groupTable = _configuration.Database.Tables["Group"];
var groupPermissionTable = _configuration.Database.Tables["GroupPermission"];
var policyTable = _configuration.Database.Tables["Policy"];
var redemptionTable = _configuration.Database.Tables["Redemption"];
var ttsFilterTable = _configuration.Database.Tables["TtsFilter"];
var ttsVoiceStateTable = _configuration.Database.Tables["VoiceState"];
var chatters = new ChatterStore(userId, chatterTable, _database, _logger);
var connections = new ConnectionStore(userId, connectionTable, _database, _logger);
var emotes = new EmoteStore(userId, emoteTable, _database, _logger);
var groups = new GroupStore(userId, groupTable, _database, _configuration, _logger);
var groupPermissions = new GroupPermissionStore(userId, groupPermissionTable, groups, _database, _logger);
var policies = new PolicyStore(userId, policyTable, groups, _database, _logger);
var filters = new TTSFilterStore(userId, ttsFilterTable, _database, _logger);
var actions = new ActionStore(userId, actionTable, _database, _logger);
var redemptions = new RedemptionStore(userId, redemptionTable, actions, _database, _logger);
var voiceStates = new VoiceStateStore(userId, ttsVoiceStateTable, _voices, _database, _logger);
channel = new Channel()
{
Id = userId,
User = user,
Chatters = chatters,
Connections = connections,
Emotes = emotes,
Groups = groups,
GroupPermissions = groupPermissions,
Policies = policies,
Filters = filters,
Actions = actions,
Redemptions = redemptions,
VoiceStates = voiceStates,
};
Task.WaitAll([
channel.Actions.Load(),
channel.Chatters.Load(),
channel.Emotes.Load(),
channel.Connections.Load(),
channel.Groups.Load(),
channel.Filters.Load(),
channel.VoiceStates.Load(),
]);
Task.WaitAll([
channel.GroupPermissions.Load(),
channel.Policies.Load(),
channel.Redemptions.Load(),
]);
_channels.Add(userId, channel);
return channel;
}
finally
{
_mutex.ReleaseMutex();
}
});
} }
public Channel? Get(string channelId) public Channel? Get(string channelId)
{ {
if (_channels.TryGetValue(channelId, out var channel)) try
return channel; {
return null; _mutex.WaitOne();
if (_channels.TryGetValue(channelId, out var channel))
return channel;
return null;
}
finally
{
_mutex.ReleaseMutex();
}
} }
public async Task Save(string userId) public async Task Save(string userId)
{ {
if (!_channels.TryGetValue(userId, out var channel)) Channel? channel;
return; try
{
_mutex.WaitOne();
if (!_channels.TryGetValue(userId, out channel))
return;
}
finally
{
_mutex.ReleaseMutex();
}
_logger.Debug($"Saving channel data to database [channel id: {channel.Id}][channel name: {channel.User.Name}]");
await Task.WhenAll([ await Task.WhenAll([
channel.Chatters.Save(), channel.Chatters.Save(),
channel.Connections.Save(),
channel.Emotes.Save(),
channel.Groups.Save(),
channel.GroupPermissions.Save(),
channel.Policies.Save(), channel.Policies.Save(),
channel.Filters.Save(), channel.Filters.Save(),
channel.Actions.Save(), channel.Actions.Save(),
channel.Redemptions.Save(), channel.Redemptions.Save(),
channel.VoiceStates.Save(),
]); ]);
} }
@@ -98,13 +156,26 @@ namespace HermesSocketServer.Services
foreach (var channel in _channels.Values) foreach (var channel in _channels.Values)
{ {
_logger.Debug($"Saving channel data to database [channel id: {channel.Id}][channel name: {channel.User.Name}]"); _logger.Debug($"Saving channel data to database [channel id: {channel.Id}][channel name: {channel.User.Name}]");
await Task.WhenAll([ var genericTablesTask = Task.WhenAll([
channel.Chatters.Save(), channel.Chatters.Save(),
channel.Policies.Save(), channel.Connections.Save(),
channel.Emotes.Save(),
channel.Filters.Save(), channel.Filters.Save(),
channel.VoiceStates.Save(),
]).ConfigureAwait(false);
await Task.WhenAll([
channel.Actions.Save(), channel.Actions.Save(),
channel.Groups.Save(),
]).ConfigureAwait(false);
await Task.WhenAll([
channel.GroupPermissions.Save(),
channel.Policies.Save(),
channel.Redemptions.Save(), channel.Redemptions.Save(),
]); ]).ConfigureAwait(false);
await genericTablesTask;
} }
} }
} }

View File

@@ -1,4 +1,5 @@
using HermesSocketLibrary.db; using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
using HermesSocketServer.Store; using HermesSocketServer.Store;
namespace HermesSocketServer.Services namespace HermesSocketServer.Services
@@ -6,12 +7,12 @@ namespace HermesSocketServer.Services
public class DatabaseService : BackgroundService public class DatabaseService : BackgroundService
{ {
private readonly ChannelManager _channels; private readonly ChannelManager _channels;
private readonly VoiceStore _voices; private readonly IStore<string, TTSVoice> _voices;
private readonly UserStore _users; private readonly IStore<string, User> _users;
private readonly ServerConfiguration _configuration; private readonly ServerConfiguration _configuration;
private readonly Serilog.ILogger _logger; private readonly Serilog.ILogger _logger;
public DatabaseService(ChannelManager channels, VoiceStore voices, UserStore users, ServerConfiguration configuration, Serilog.ILogger logger) public DatabaseService(ChannelManager channels, IStore<string, TTSVoice> voices, IStore<string, User> users, ServerConfiguration configuration, Serilog.ILogger logger)
{ {
_channels = channels; _channels = channels;
_voices = voices; _voices = voices;
@@ -22,10 +23,10 @@ namespace HermesSocketServer.Services
protected override async Task ExecuteAsync(CancellationToken cancellationToken) protected override async Task ExecuteAsync(CancellationToken cancellationToken)
{ {
_logger.Information("Loading TTS voices...");
await _voices.Load();
_logger.Information("Loading users..."); _logger.Information("Loading users...");
await _users.Load(); await _users.Load();
_logger.Information("Loading TTS voices...");
await _voices.Load();
await Task.Run(async () => await Task.Run(async () =>
{ {
@@ -33,11 +34,10 @@ namespace HermesSocketServer.Services
while (true) while (true)
{ {
await Task.WhenAll([ await _users.Save();
_voices.Save(), await _voices.Save();
_users.Save(), await _channels.Save();
_channels.Save(),
]);
await Task.Delay(TimeSpan.FromSeconds(_configuration.Database.SaveDelayInSeconds)); await Task.Delay(TimeSpan.FromSeconds(_configuration.Database.SaveDelayInSeconds));
} }
}); });

View File

@@ -6,32 +6,40 @@ namespace HermesSocketServer.Socket.Handlers
{ {
public class ChatterHandler : ISocketHandler public class ChatterHandler : ISocketHandler
{ {
private const int CHATTER_BUFFER_SIZE = 2000;
public int OperationCode { get; } = 6; public int OperationCode { get; } = 6;
private readonly Database _database; private readonly Database _database;
private readonly HashSet<long> _chatters; private readonly HashSet<long> _chatters;
private readonly ChatterMessage[] _array; private readonly long[] _array;
private readonly ILogger _logger; private readonly ILogger _logger;
private readonly object _lock; private readonly Mutex _lock;
private int _index; private int _index;
public ChatterHandler(Database database, ILogger logger) public ChatterHandler(Database database, ILogger logger)
{ {
_database = database; _database = database;
_logger = logger; _logger = logger;
_chatters = new HashSet<long>(1001); _chatters = new HashSet<long>(CHATTER_BUFFER_SIZE);
_array = new ChatterMessage[1000]; _array = new long[CHATTER_BUFFER_SIZE];
_index = -1; _index = -1;
_lock = new object(); _lock = new Mutex();
} }
public async Task Execute<T>(WebSocketUser sender, T message, HermesSocketManager sockets) public async Task Execute<T>(WebSocketUser sender, T message, HermesSocketManager sockets)
{ {
if (message is not ChatterMessage data || sender.Id == null) if (message is not ChatterMessage data || sender.Id == null)
return; return;
if (sender.Slave && !sender.WebLogin)
lock (_lock)
{ {
_logger.Warning($"Received message from a slave client [message type: chatter details][sender id: {sender.Id}][sender session: {sender.SessionId}]");
return;
}
try
{
_lock.WaitOne();
if (_chatters.Contains(data.Id)) if (_chatters.Contains(data.Id))
return; return;
@@ -40,7 +48,16 @@ namespace HermesSocketServer.Socket.Handlers
if (_index == _array.Length - 1) if (_index == _array.Length - 1)
_index = -1; _index = -1;
_array[++_index] = data; var previous = _array[++_index];
if (previous != 0)
{
_chatters.Remove(previous);
}
_array[_index] = data.Id;
}
finally
{
_lock.ReleaseMutex();
} }
try try

View File

@@ -1,36 +1,48 @@
using HermesSocketLibrary.db; using HermesSocketLibrary.Requests.Messages;
using HermesSocketLibrary.Socket.Data; using HermesSocketLibrary.Socket.Data;
using Npgsql; using HermesSocketServer.Services;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
namespace HermesSocketServer.Socket.Handlers namespace HermesSocketServer.Socket.Handlers
{ {
public class EmoteDetailsHandler : ISocketHandler public class EmoteDetailsHandler : ISocketHandler
{ {
public int OperationCode { get; } = 7; private const int EMOTE_BUFFER_SIZE = 5000;
private readonly Database _database;
private readonly HashSet<string> _emotes;
private readonly ILogger _logger;
private readonly object _lock;
public EmoteDetailsHandler(Database database, ILogger logger) public int OperationCode { get; } = 7;
private readonly ChannelManager _manager;
private readonly HashSet<string> _emotes;
private readonly string[] _array;
private readonly ILogger _logger;
private readonly Mutex _mutex;
private int _index;
public EmoteDetailsHandler(ChannelManager manager, ILogger logger)
{ {
_database = database; _manager = manager;
_emotes = new HashSet<string>(EMOTE_BUFFER_SIZE);
_array = new string[EMOTE_BUFFER_SIZE];
_logger = logger; _logger = logger;
_emotes = new HashSet<string>(501); _mutex = new Mutex();
_lock = new object(); _index = -1;
} }
public async Task Execute<T>(WebSocketUser sender, T message, HermesSocketManager sockets) public async Task Execute<T>(WebSocketUser sender, T message, HermesSocketManager sockets)
{ {
if (message is not EmoteDetailsMessage data || sender.Id == null) if (message is not EmoteDetailsMessage data || sender.Id == null)
return; return;
if (data.Emotes == null || !data.Emotes.Any()) if (data.Emotes == null || !data.Emotes.Any())
return; return;
if (sender.Slave && !sender.WebLogin)
lock (_lock)
{ {
_logger.Warning($"Received message from a slave client [message type: emote details][sender id: {sender.Id}][sender session: {sender.SessionId}]");
return;
}
try
{
_mutex.WaitOne();
foreach (var entry in data.Emotes) foreach (var entry in data.Emotes)
{ {
if (_emotes.Contains(entry.Key)) if (_emotes.Contains(entry.Key))
@@ -40,36 +52,32 @@ namespace HermesSocketServer.Socket.Handlers
} }
_emotes.Add(entry.Key); _emotes.Add(entry.Key);
if (_index == _array.Length - 1)
_index = -1;
var previous = _array[++_index];
if (previous != null)
{
_emotes.Remove(previous);
}
_array[_index] = entry.Key;
} }
} }
finally
{
_mutex.ReleaseMutex();
}
if (!data.Emotes.Any()) if (!data.Emotes.Any())
return; return;
int rows = 0; var channel = _manager.Get(sender.Id);
string sql = "INSERT INTO \"Emote\" (id, name) VALUES (@idd, @name)"; if (channel == null)
using (var connection = await _database.DataSource.OpenConnectionAsync()) return;
{
using (var command = new NpgsqlCommand(sql, connection))
{
foreach (var entry in data.Emotes)
{
command.Parameters.Clear();
command.Parameters.AddWithValue("idd", entry.Key);
command.Parameters.AddWithValue("name", entry.Value);
await command.PrepareAsync(); foreach (var entry in data.Emotes)
try channel.Emotes.Set(entry.Key, new EmoteInfo() { Id = entry.Key, Name = entry.Value, UserId = channel.Id });
{
rows += await command.ExecuteNonQueryAsync();
}
catch (Exception e)
{
_logger.Error(e, "Failed to add emote detail: " + entry.Key + " -> " + entry.Value);
}
}
}
}
} }
} }
} }

View File

@@ -13,7 +13,7 @@ namespace HermesSocketServer.Socket.Handlers
private readonly HashSet<string> _history; private readonly HashSet<string> _history;
private readonly EmoteUsageMessage[] _array; private readonly EmoteUsageMessage[] _array;
private readonly ILogger _logger; private readonly ILogger _logger;
private readonly object _lock; private readonly Mutex _mutex;
private int _index; private int _index;
@@ -21,9 +21,9 @@ namespace HermesSocketServer.Socket.Handlers
{ {
_database = database; _database = database;
_logger = logger; _logger = logger;
_history = new HashSet<string>(101); _history = new HashSet<string>(1001);
_array = new EmoteUsageMessage[100]; _array = new EmoteUsageMessage[1000];
_lock = new object(); _mutex = new Mutex();
_index = -1; _index = -1;
} }
@@ -32,9 +32,15 @@ namespace HermesSocketServer.Socket.Handlers
{ {
if (message is not EmoteUsageMessage data || sender.Id == null) if (message is not EmoteUsageMessage data || sender.Id == null)
return; return;
if (sender.Slave && !sender.WebLogin)
lock (_lock)
{ {
_logger.Warning($"Received message from a slave client [message type: emote usage][sender id: {sender.Id}][sender session: {sender.SessionId}]");
return;
}
try
{
_mutex.WaitOne();
if (_history.Contains(data.MessageId)) if (_history.Contains(data.MessageId))
{ {
return; return;
@@ -50,7 +56,12 @@ namespace HermesSocketServer.Socket.Handlers
_array[_index] = data; _array[_index] = data;
} }
finally
{
_mutex.ReleaseMutex();
}
// TODO: multi-row inserts to increase database performance.
int rows = 0; int rows = 0;
string sql = "INSERT INTO \"EmoteUsageHistory\" (timestamp, \"broadcasterId\", \"emoteId\", \"chatterId\") VALUES (@time, @broadcaster, @emote, @chatter)"; string sql = "INSERT INTO \"EmoteUsageHistory\" (timestamp, \"broadcasterId\", \"emoteId\", \"chatterId\") VALUES (@time, @broadcaster, @emote, @chatter)";
using (var connection = await _database.DataSource.OpenConnectionAsync()) using (var connection = await _database.DataSource.OpenConnectionAsync())

View File

@@ -12,14 +12,14 @@ namespace HermesSocketServer.Socket.Handlers
public int OperationCode { get; } = 1; public int OperationCode { get; } = 1;
private readonly ChannelManager _manager; private readonly ChannelManager _manager;
private readonly VoiceStore _voices; private readonly IStore<string, TTSVoice> _voices;
private readonly ServerConfiguration _configuration; private readonly ServerConfiguration _configuration;
private readonly Database _database; private readonly Database _database;
private readonly HermesSocketManager _sockets; private readonly HermesSocketManager _sockets;
private readonly ILogger _logger; private readonly ILogger _logger;
private readonly object _lock; private readonly SemaphoreSlim _semaphore;
public HermesLoginHandler(ChannelManager manager, VoiceStore voices, ServerConfiguration configuration, Database database, HermesSocketManager sockets, ILogger logger) public HermesLoginHandler(ChannelManager manager, IStore<string, TTSVoice> voices, ServerConfiguration configuration, Database database, HermesSocketManager sockets, ILogger logger)
{ {
_manager = manager; _manager = manager;
_voices = voices; _voices = voices;
@@ -27,7 +27,7 @@ namespace HermesSocketServer.Socket.Handlers
_database = database; _database = database;
_sockets = sockets; _sockets = sockets;
_logger = logger; _logger = logger;
_lock = new object(); _semaphore = new SemaphoreSlim(1);
} }
@@ -42,119 +42,102 @@ namespace HermesSocketServer.Socket.Handlers
var result = await _database.ExecuteScalar(sql, new Dictionary<string, object>() { { "key", data.ApiKey } }); var result = await _database.ExecuteScalar(sql, new Dictionary<string, object>() { { "key", data.ApiKey } });
string? userId = result?.ToString(); string? userId = result?.ToString();
if (userId == null) if (string.IsNullOrWhiteSpace(userId))
return; return;
lock (_lock) IEnumerable<WebSocketUser?> recipients = Enumerable.Empty<WebSocketUser?>();
try
{ {
await _semaphore.WaitAsync();
if (sender.Id != null) if (sender.Id != null)
return; return;
sender.Id = userId; sender.Id = userId;
recipients = _sockets.GetSockets(userId).ToList().Where(s => s.SessionId != sender.SessionId);
sender.Slave = data.WebLogin || recipients.Any(r => r != null && !r.WebLogin);
sender.ApiKey = data.ApiKey; sender.ApiKey = data.ApiKey;
sender.WebLogin = data.WebLogin; sender.WebLogin = data.WebLogin;
}
var channel = _manager.Get(userId); // Fetch channel data.
if (channel == null) var channel = _manager.Get(userId);
{
channel = await _manager.Add(userId);
if (channel == null) if (channel == null)
{
channel = await _manager.Add(userId);
if (channel == null)
throw new Exception("Channel does not exist.");
}
sender.Name = channel.User.Name;
sender.Admin = channel.User.Role == "ADMIN";
if (string.IsNullOrEmpty(sender.Name))
{
_logger.Error($"Could not find username for a certain user [user id: {userId}][api key: {data.ApiKey}]");
return; return;
}
sender.Name = channel.User.Name;
sender.Admin = channel.User.Role == "ADMIN";
if (string.IsNullOrEmpty(sender.Name))
{
_logger.Error($"Could not find username for a certain user [user id: {userId}][api key: {data.ApiKey}]");
return;
}
if (string.IsNullOrEmpty(channel.User.DefaultVoice))
_logger.Warning($"No default voice was set for an user [user id: {userId}][api key: {data.ApiKey}]");
sql = "select \"providerAccountId\" from \"Account\" where \"userId\" = @user and provider = @provider";
var result2 = await _database.ExecuteScalar(sql, new Dictionary<string, object>() { { "user", userId }, { "provider", "twitch" } });
var providerId = result2?.ToString();
if (providerId == null) {
_logger.Warning($"Could not find the Provider Account Id [user id: {userId}][provider: twitch]");
return;
}
var ack = new LoginAckMessage()
{
UserId = userId,
ProviderAccountId = providerId,
SessionId = sender.UID,
UserName = channel.User.Name,
OwnerId = _configuration.Tts.OwnerId,
Admin = sender.Admin,
WebLogin = data.WebLogin,
};
var userIdDict = new Dictionary<string, object>() { { "user", userId } };
ack.Connections = new List<Connection>();
string sql3 = "select \"name\", \"type\", \"clientId\", \"accessToken\", \"grantType\", \"scope\", \"expiresAt\", \"default\" from \"Connection\" where \"userId\" = @user";
await _database.Execute(sql3, userIdDict, sql =>
ack.Connections.Add(new Connection()
{
Name = sql.GetString(0),
Type = sql.GetString(1),
ClientId = sql.GetString(2),
AccessToken = sql.GetString(3),
GrantType = sql.GetString(4),
Scope = sql.GetString(5),
ExpiresAt = sql.GetDateTime(6),
Default = sql.GetBoolean(7)
})
);
ack.TTSVoicesAvailable = _voices.Get().ToDictionary(v => v.Key, v => v.Value.Name);
ack.EnabledTTSVoices = new List<string>();
string sql5 = $"SELECT v.name FROM \"TtsVoiceState\" s "
+ "INNER JOIN \"TtsVoice\" v ON s.\"ttsVoiceId\" = v.id "
+ "WHERE \"userId\" = @user AND state = true";
await _database.Execute(sql5, userIdDict, (r) => ack.EnabledTTSVoices.Add(r.GetString(0)));
ack.WordFilters = new List<TTSWordFilter>();
string sql6 = $"SELECT id, search, replace FROM \"TtsWordFilter\" WHERE \"userId\" = @user";
await _database.Execute(sql6, userIdDict, (r) => ack.WordFilters.Add(new TTSWordFilter()
{
Id = r.GetString(0),
Search = r.GetString(1),
Replace = r.GetString(2)
}));
ack.DefaultTTSVoice = channel.User.DefaultVoice ?? _configuration.Tts.DefaultTtsVoice;
await sender.Send(2, ack);
string version = data.MajorVersion == null ? "unknown" : $"{data.MajorVersion}.{data.MinorVersion}";
_logger.Information($"Hermes client logged in {(sender.Admin ? "as administrator " : "")}[name: {sender.Name}][id: {userId}][ip: {sender.IPAddress}][version: {version}][web: {data.WebLogin}]");
ack = new LoginAckMessage()
{
AnotherClient = true,
UserId = userId,
OwnerId = _configuration.Tts.OwnerId,
WebLogin = data.WebLogin
};
var recipients = _sockets.GetSockets(userId).ToList().Where(s => s.UID != sender.UID);
var tasks = new List<Task>();
foreach (var socket in recipients)
{
try
{
tasks.Add(socket.Send(2, ack));
} }
catch (Exception)
sql = "select \"providerAccountId\" from \"Account\" where \"userId\" = @user and provider = @provider";
var result2 = await _database.ExecuteScalar(sql, new Dictionary<string, object>() { { "user", userId }, { "provider", "twitch" } });
var providerId = result2?.ToString();
if (providerId == null)
{ {
_logger.Warning($"Could not find the Provider Account Id [user id: {userId}][provider: twitch]");
return;
} }
var voices = _voices.Get();
var voiceStates = channel.VoiceStates.Get();
var voicesEnabled = voices.Values.Where(v => !voiceStates.TryGetValue(v.Id, out var voice) || voice.Enabled).Select(v => v.Name).ToList();
var ack = new LoginAckMessage()
{
UserId = userId,
ProviderAccountId = providerId,
SessionId = sender.SessionId,
UserName = channel.User.Name,
StreamElementsOverlayKey = channel.User.StreamElementsOverlayKey,
OwnerId = _configuration.Tts.OwnerId,
Admin = sender.Admin,
WebLogin = data.WebLogin,
WordFilters = channel.Filters.Get().Values,
DefaultTTSVoice = channel.User.DefaultVoice ?? _configuration.Tts.DefaultTtsVoice,
TTSVoicesAvailable = voices.ToDictionary(v => v.Key, v => v.Value.Name),
EnabledTTSVoices = voicesEnabled,
Connections = channel.Connections.Get().Values.ToList(),
Slave = sender.Slave,
};
await sender.Send(2, ack);
// Sending notification to other clients about another client logging in.
string version = $"{data.MajorVersion}.{data.MinorVersion}.{data.PatchVersion}";
_logger.Information($"Hermes client logged in {(sender.Admin ? "as administrator " : "")}[name: {sender.Name}][id: {userId}][ip: {sender.IPAddress}][version: {version}][web: {data.WebLogin}]");
ack = new LoginAckMessage()
{
AnotherClient = true,
UserId = userId,
OwnerId = _configuration.Tts.OwnerId,
WebLogin = data.WebLogin
};
var tasks = new List<Task>();
foreach (var socket in recipients)
{
try
{
tasks.Add(socket!.Send(2, ack));
}
catch (Exception)
{
}
}
await Task.WhenAll(tasks);
}
finally
{
_semaphore.Release();
} }
await Task.WhenAll(tasks);
} }
} }
} }

View File

@@ -38,7 +38,7 @@ namespace HermesSocketServer.Socket.Handlers
return Task.CompletedTask; return Task.CompletedTask;
} }
logging.Invoke(message.Exception, message.Message + $" [ip: {sender.IPAddress}][id: {sender.Id}][name: {sender.Name}][token: {sender.ApiKey}][uid: {sender.UID}]"); logging.Invoke(message.Exception, message.Message + $" [ip: {sender.IPAddress}][id: {sender.Id}][name: {sender.Name}][token: {sender.ApiKey}][uid: {sender.SessionId}]");
return Task.CompletedTask; return Task.CompletedTask;
} }
} }

View File

@@ -23,6 +23,11 @@ namespace HermesSocketServer.Socket.Handlers
{ {
if (message is not RequestMessage data || sender.Id == null) if (message is not RequestMessage data || sender.Id == null)
return; return;
if (sender.Slave && !sender.WebLogin && !data.Type.StartsWith("get"))
{
_logger.Warning($"Received a non-get request from a slave client [request type: {data.Type}][sender id: {sender.Id}][sender session: {sender.SessionId}]");
return;
}
RequestResult? result = null; RequestResult? result = null;
_logger.Debug("Executing request handler: " + data.Type); _logger.Debug("Executing request handler: " + data.Type);

View File

@@ -6,9 +6,10 @@ namespace HermesSocketServer.Socket
{ {
public class HermesSocketManager public class HermesSocketManager
{ {
private IList<WebSocketUser> _sockets; private readonly IList<WebSocketUser> _sockets;
private System.Timers.Timer _timer; private readonly System.Timers.Timer _timer;
private ILogger _logger; private readonly ILogger _logger;
private readonly object _lock = new object();
public HermesSocketManager(ILogger logger) public HermesSocketManager(ILogger logger)
@@ -18,31 +19,40 @@ namespace HermesSocketServer.Socket
_timer.Elapsed += async (sender, e) => await HandleHeartbeats(e); _timer.Elapsed += async (sender, e) => await HandleHeartbeats(e);
_timer.Enabled = true; _timer.Enabled = true;
_logger = logger; _logger = logger;
_lock = new object();
} }
public void Add(WebSocketUser socket) public void Add(WebSocketUser socket)
{ {
_sockets.Add(socket); lock (_lock)
{
_sockets.Add(socket);
}
} }
public IList<WebSocketUser> GetAllSockets() public IList<WebSocketUser> GetAllSockets()
{ {
return _sockets.AsReadOnly(); lock (_lock)
{
return _sockets.AsReadOnly();
}
} }
public IEnumerable<WebSocketUser> GetSockets(string userId) public IEnumerable<WebSocketUser> GetSockets(string userId)
{ {
foreach (var socket in _sockets) lock (_lock)
{ {
if (socket.Id == userId) return _sockets.Where(s => s.Id == userId);
yield return socket;
} }
} }
public bool Remove(WebSocketUser socket) public bool Remove(WebSocketUser socket)
{ {
return _sockets.Remove(socket); lock (_lock)
{
return _sockets.Remove(socket);
}
} }
private async Task HandleHeartbeats(ElapsedEventArgs e) private async Task HandleHeartbeats(ElapsedEventArgs e)

View File

@@ -22,7 +22,8 @@ namespace HermesSocketServer.Socket
public WebSocketState State { get => _socket.State; } public WebSocketState State { get => _socket.State; }
public IPAddress? IPAddress { get => _ipAddress; } public IPAddress? IPAddress { get => _ipAddress; }
public bool Connected { get => _connected; } public bool Connected { get => _connected; }
public string UID { get; } public bool Slave { get; set; }
public string SessionId { get; }
public string? ApiKey { get; set; } public string? ApiKey { get; set; }
public string? Id { get; set; } public string? Id { get; set; }
public string? Name { get; set; } public string? Name { get; set; }
@@ -43,7 +44,7 @@ namespace HermesSocketServer.Socket
Admin = false; Admin = false;
WebLogin = false; WebLogin = false;
_cts = new CancellationTokenSource(); _cts = new CancellationTokenSource();
UID = Guid.NewGuid().ToString("D"); SessionId = Guid.NewGuid().ToString("D");
LastHeartbeatReceived = DateTime.UtcNow; LastHeartbeatReceived = DateTime.UtcNow;
} }

View File

@@ -17,6 +17,8 @@ using HermesSocketServer.Store;
using HermesSocketServer.Services; using HermesSocketServer.Services;
using HermesSocketServer.Store.Internal; using HermesSocketServer.Store.Internal;
using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.DependencyInjection;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Models;
var yamlDeserializer = new DeserializerBuilder() var yamlDeserializer = new DeserializerBuilder()
@@ -88,19 +90,27 @@ if (configuration.Database.Tables != null)
} }
// Stores // Stores
s.AddSingleton<VoiceStore>(); s.AddSingleton<IStore<string, TTSVoice>, VoiceStore>();
s.AddSingleton<UserStore>(); s.AddSingleton<IStore<string, User>, UserStore>();
// Request handlers // Request handlers
s.AddSingleton<IRequest, CreateConnection>();
s.AddSingleton<IRequest, CreateGroup>();
s.AddSingleton<IRequest, CreateGroupChatter>();
s.AddSingleton<IRequest, CreateGroupPermission>();
s.AddSingleton<IRequest, CreatePolicy>(); s.AddSingleton<IRequest, CreatePolicy>();
s.AddSingleton<IRequest, CreateRedeemableAction>(); s.AddSingleton<IRequest, CreateRedeemableAction>();
s.AddSingleton<IRequest, CreateRedemption>(); s.AddSingleton<IRequest, CreateRedemption>();
s.AddSingleton<IRequest, CreateTTSFilter>(); s.AddSingleton<IRequest, CreateTTSFilter>();
s.AddSingleton<IRequest, CreateTTSUser>(); s.AddSingleton<IRequest, CreateTTSUser>();
s.AddSingleton<IRequest, CreateTTSVoice>(); s.AddSingleton<IRequest, CreateTTSVoice>();
s.AddSingleton<IRequest, DeleteConnection>();
s.AddSingleton<IRequest, DeleteGroup>();
s.AddSingleton<IRequest, DeleteGroupChatter>();
s.AddSingleton<IRequest, DeleteGroupPermission>();
s.AddSingleton<IRequest, DeletePolicy>();
s.AddSingleton<IRequest, DeleteRedeemableAction>(); s.AddSingleton<IRequest, DeleteRedeemableAction>();
s.AddSingleton<IRequest, DeleteRedemption>(); s.AddSingleton<IRequest, DeleteRedemption>();
s.AddSingleton<IRequest, DeletePolicy>();
s.AddSingleton<IRequest, DeleteTTSFilter>(); s.AddSingleton<IRequest, DeleteTTSFilter>();
s.AddSingleton<IRequest, DeleteTTSVoice>(); s.AddSingleton<IRequest, DeleteTTSVoice>();
s.AddSingleton<IRequest, GetChatterIds>(); s.AddSingleton<IRequest, GetChatterIds>();
@@ -108,6 +118,8 @@ s.AddSingleton<IRequest, GetConnections>();
s.AddSingleton<IRequest, GetDefaultTTSVoice>(); s.AddSingleton<IRequest, GetDefaultTTSVoice>();
s.AddSingleton<IRequest, GetEmotes>(); s.AddSingleton<IRequest, GetEmotes>();
s.AddSingleton<IRequest, GetEnabledTTSVoices>(); s.AddSingleton<IRequest, GetEnabledTTSVoices>();
s.AddSingleton<IRequest, GetGroups>();
s.AddSingleton<IRequest, GetGroupPermissions>();
s.AddSingleton<IRequest, GetPermissions>(); s.AddSingleton<IRequest, GetPermissions>();
s.AddSingleton<IRequest, GetRedeemableActions>(); s.AddSingleton<IRequest, GetRedeemableActions>();
s.AddSingleton<IRequest, GetRedemptions>(); s.AddSingleton<IRequest, GetRedemptions>();
@@ -115,7 +127,11 @@ s.AddSingleton<IRequest, GetPolicies>();
s.AddSingleton<IRequest, GetTTSUsers>(); s.AddSingleton<IRequest, GetTTSUsers>();
s.AddSingleton<IRequest, GetTTSVoices>(); s.AddSingleton<IRequest, GetTTSVoices>();
s.AddSingleton<IRequest, GetTTSWordFilters>(); s.AddSingleton<IRequest, GetTTSWordFilters>();
s.AddSingleton<IRequest, UpdateConnection>();
s.AddSingleton<IRequest, UpdateDefaultTTSVoice>(); s.AddSingleton<IRequest, UpdateDefaultTTSVoice>();
s.AddSingleton<IRequest, UpdateGroup>();
s.AddSingleton<IRequest, UpdateGroupChatter>();
s.AddSingleton<IRequest, UpdateGroupPermission>();
s.AddSingleton<IRequest, UpdatePolicy>(); s.AddSingleton<IRequest, UpdatePolicy>();
s.AddSingleton<IRequest, UpdateRedeemableAction>(); s.AddSingleton<IRequest, UpdateRedeemableAction>();
s.AddSingleton<IRequest, UpdateRedemption>(); s.AddSingleton<IRequest, UpdateRedemption>();
@@ -147,9 +163,7 @@ var wsOptions = new WebSocketOptions()
{ {
KeepAliveInterval = TimeSpan.FromSeconds(30) KeepAliveInterval = TimeSpan.FromSeconds(30)
}; };
// wsOptions.AllowedOrigins.Add("wss://tomtospeech.com");
//wsOptions.AllowedOrigins.Add("ws.tomtospeech.com");
//wsOptions.AllowedOrigins.Add("hermes-ws.goblincaves.com");
app.UseWebSockets(wsOptions); app.UseWebSockets(wsOptions);
var options = app.Services.GetRequiredService<JsonSerializerOptions>(); var options = app.Services.GetRequiredService<JsonSerializerOptions>();

View File

@@ -23,27 +23,44 @@ namespace HermesSocketServer.Store
public override async Task Load() public override async Task Load()
{ {
var data = new Dictionary<string, object>() { { "user", _userId } }; var data = new Dictionary<string, object>() { { "user", _userId } };
string sql = $"SELECT name, type, data FROM \"Action\" WHERE \"userId\" = @user"; string sql = $"SELECT name, type, has_message, data FROM \"Action\" WHERE \"userId\" = @user";
await _database.Execute(sql, data, (reader) => await _database.Execute(sql, data, (reader) =>
{ {
var name = reader.GetString(0); var name = reader.GetString(0);
_store.Add(name.ToString(), new RedeemableAction() _store.Add(name, new RedeemableAction()
{ {
UserId = _userId, UserId = _userId,
Name = name, Name = name,
Type = reader.GetString(1), Type = reader.GetString(1),
Data = JsonSerializer.Deserialize<IDictionary<string, string>>(reader.GetString(2))! HasMessage = reader.GetBoolean(2),
Data = JsonSerializer.Deserialize<IDictionary<string, string>>(reader.GetString(3))!
}); });
}); });
_logger.Information($"Loaded {_store.Count} TTS chatter voices from database."); _logger.Information($"Loaded {_store.Count} redeemable actions from database.");
} }
protected override void OnInitialAdd(string key, RedeemableAction value) protected override void OnInitialAdd(string key, RedeemableAction value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Name, nameof(value.Name));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Type, nameof(value.Type));
ArgumentNullException.ThrowIfNull(value.Data, nameof(value.Data));
if (value.Name.Length > 36)
throw new ArgumentException("Action name cannot be longer than 36 characters.");
} }
protected override void OnInitialModify(string key, RedeemableAction value) protected override void OnInitialModify(string key, RedeemableAction oldValue, RedeemableAction newValue)
{ {
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Name, nameof(newValue.Name));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Type, nameof(newValue.Type));
ArgumentNullException.ThrowIfNull(newValue.Data, nameof(newValue.Data));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Name, newValue.Name, nameof(oldValue.Name));
} }
} }
} }

View File

@@ -0,0 +1,71 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Store.Internal;
namespace HermesSocketServer.Store
{
public class ChatterGroupStore : ComplexAutoSavedStore<string, GroupChatter>
{
private readonly string _userId;
private readonly string _groupId;
private readonly Database _database;
private readonly Serilog.ILogger _logger;
public string GroupId { get => _groupId; }
public ChatterGroupStore(string userId, string groupId, DatabaseTable table, Database database, Serilog.ILogger logger)
: base(table, database, logger)
{
_userId = userId;
_groupId = groupId;
_database = database;
_logger = logger;
}
public override async Task Load()
{
var data = new Dictionary<string, object>() { { "user", _userId }, { "group", new Guid(_groupId) } };
string sql = $"SELECT \"chatterId\", \"chatterLabel\" FROM \"ChatterGroup\" WHERE \"userId\" = @user AND \"groupId\" = @group";
await _database.Execute(sql, data, (reader) =>
{
var chatterId = reader.GetInt32(0).ToString();
_rwls.EnterWriteLock();
try
{
_store.Add(chatterId, new GroupChatter()
{
UserId = _userId,
GroupId = _groupId,
ChatterId = reader.GetInt32(0),
ChatterLabel = reader.GetString(1),
});
}
finally
{
_rwls.ExitWriteLock();
}
});
_logger.Information($"Loaded {_store.Count} group chatters from database [group id: {_groupId}]");
}
protected override void OnInitialAdd(string key, GroupChatter value)
{
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.GroupId, nameof(value.GroupId));
ArgumentNullException.ThrowIfNull(value.ChatterId, nameof(value.ChatterId));
}
protected override void OnInitialModify(string key, GroupChatter oldValue, GroupChatter newValue)
{
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.GroupId, nameof(newValue.GroupId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.GroupId, nameof(newValue.GroupId));
ArgumentNullException.ThrowIfNull(newValue.ChatterId, nameof(newValue.ChatterId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.GroupId, newValue.GroupId, nameof(oldValue.GroupId));
}
}
}

View File

@@ -38,10 +38,21 @@ namespace HermesSocketServer.Store
protected override void OnInitialAdd(string key, ChatterVoice value) protected override void OnInitialAdd(string key, ChatterVoice value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentOutOfRangeException.ThrowIfNegativeOrZero(value.ChatterId, nameof(value.ChatterId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.VoiceId, nameof(value.VoiceId));
} }
protected override void OnInitialModify(string key, ChatterVoice value) protected override void OnInitialModify(string key, ChatterVoice oldValue, ChatterVoice newValue)
{ {
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentOutOfRangeException.ThrowIfNegativeOrZero(newValue.ChatterId, nameof(newValue.ChatterId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.VoiceId, nameof(newValue.VoiceId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.ChatterId, newValue.ChatterId, nameof(oldValue.ChatterId));
} }
} }
} }

80
Store/ConnectionStore.cs Normal file
View File

@@ -0,0 +1,80 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Store.Internal;
namespace HermesSocketServer.Store
{
public class ConnectionStore : ComplexAutoSavedStore<string, Connection>
{
private readonly string _userId;
private readonly Database _database;
private readonly Serilog.ILogger _logger;
public ConnectionStore(string userId, DatabaseTable table, Database database, Serilog.ILogger logger)
: base(table, database, logger)
{
_userId = userId;
_database = database;
_logger = logger;
}
public override async Task Load()
{
var data = new Dictionary<string, object>() { { "user", _userId } };
string sql = $"SELECT name, \"type\", \"clientId\", \"accessToken\", \"grantType\", \"scope\", \"expiresAt\", \"default\" FROM \"Connection\" WHERE \"userId\" = @user";
await _database.Execute(sql, data, (reader) =>
{
var name = reader.GetString(0);
_store.Add(name, new Connection()
{
Name = name,
UserId = _userId,
Type = reader.GetString(1),
ClientId = reader.GetString(2),
AccessToken = reader.GetString(3),
GrantType = reader.GetString(4),
Scope = reader.GetString(5),
ExpiresAt = reader.GetDateTime(6),
Default = reader.GetBoolean(7),
});
});
_logger.Information($"Loaded {_store.Count} connections from database.");
}
protected override void OnInitialAdd(string key, Connection value)
{
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Name, nameof(value.Name));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Type, nameof(value.Type));
ArgumentException.ThrowIfNullOrWhiteSpace(value.ClientId, nameof(value.ClientId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.AccessToken, nameof(value.AccessToken));
ArgumentException.ThrowIfNullOrWhiteSpace(value.GrantType, nameof(value.GrantType));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Scope, nameof(value.Scope));
ArgumentNullException.ThrowIfNull(value.ExpiresAt, nameof(value.ExpiresAt));
ArgumentNullException.ThrowIfNull(value.Default, nameof(value.Default));
if (value.Name.Length > 36)
throw new ArgumentException("Action name cannot be longer than 36 characters.");
}
protected override void OnInitialModify(string key, Connection oldValue, Connection newValue)
{
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Name, nameof(newValue.Name));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Type, nameof(newValue.Type));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.ClientId, nameof(newValue.ClientId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.AccessToken, nameof(newValue.AccessToken));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.GrantType, nameof(newValue.GrantType));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Scope, nameof(newValue.Scope));
ArgumentNullException.ThrowIfNull(newValue.ExpiresAt, nameof(newValue.ExpiresAt));
ArgumentNullException.ThrowIfNull(newValue.Default, nameof(newValue.Default));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Name, newValue.Name, nameof(oldValue.Name));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Type, newValue.Type, nameof(oldValue.Type));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.ClientId, newValue.ClientId, nameof(oldValue.ClientId));
}
}
}

62
Store/EmoteStore.cs Normal file
View File

@@ -0,0 +1,62 @@
using System.Collections.Concurrent;
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Store.Internal;
namespace HermesSocketServer.Store
{
public class EmoteStore : AutoSavedStore<string, EmoteInfo>
{
private readonly string _userId;
private readonly Database _database;
private readonly Serilog.ILogger _logger;
public EmoteStore(string userId, DatabaseTable table, Database database, Serilog.ILogger logger)
: base(table, database, logger)
{
_userId = userId;
_database = database;
_logger = logger;
}
public override async Task Load()
{
var emotes = new List<EmoteInfo>();
var data = new Dictionary<string, object>() { { "user", _userId } };
string sql = $"SELECT id, name FROM \"Emote\" WHERE \"userId\" = @user";
await _database.Execute(sql, data, (reader) =>
{
var id = reader.GetString(0);
var name = reader.GetString(1);
_store.Add(id, new EmoteInfo()
{
Id = id,
UserId = _userId,
Name = name,
});
emotes.Add(new EmoteInfo() { Id = id, Name = name, UserId = _userId });
});
_logger.Information($"Loaded {_store.Count} emotes from database.");
}
protected override void OnInitialAdd(string key, EmoteInfo value)
{
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Name, nameof(value.Name));
}
protected override void OnInitialModify(string key, EmoteInfo oldValue, EmoteInfo newValue)
{
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Name, nameof(newValue.Name));
}
protected override void OnPostRemove(string key, EmoteInfo value)
{
}
}
}

View File

@@ -0,0 +1,74 @@
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Store.Internal;
namespace HermesSocketServer.Store
{
public class GroupPermissionStore : AutoSavedStore<string, GroupPermission>
{
private readonly string _userId;
private readonly IStore<string, Group> _groups;
private readonly Database _database;
private readonly Serilog.ILogger _logger;
public GroupPermissionStore(string userId, DatabaseTable table, IStore<string, Group> groups, Database database, Serilog.ILogger logger)
: base(table, database, logger)
{
_userId = userId;
_groups = groups;
_database = database;
_logger = logger;
}
public override async Task Load()
{
var data = new Dictionary<string, object>() { { "user", _userId } };
string sql = $"SELECT id, \"groupId\", path, allow FROM \"GroupPermission\" WHERE \"userId\" = @user";
await _database.Execute(sql, data, async (reader) =>
{
var id = reader.GetGuid(0).ToString();
_store.Add(id, new GroupPermission()
{
Id = id,
UserId = _userId,
GroupId = reader.GetGuid(1),
Path = reader.GetString(2),
Allow = await reader.IsDBNullAsync(3) ? null : reader.GetBoolean(3),
});
});
_logger.Information($"Loaded {_store.Count} group permissions from database.");
}
protected override void OnInitialAdd(string key, GroupPermission value)
{
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentNullException.ThrowIfNull(value.GroupId, nameof(value.GroupId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Path, nameof(value.Path));
if (value.Path.Length > 100)
throw new ArgumentException("The path cannot be longer than 100 characters.");
if (_groups.Get(value.GroupId.ToString()) == null)
throw new ArgumentException("The group id does not exist.");
}
protected override void OnInitialModify(string key, GroupPermission oldValue, GroupPermission newValue)
{
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentNullException.ThrowIfNull(newValue.GroupId, nameof(newValue.GroupId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Path, nameof(newValue.Path));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.GroupId, newValue.GroupId, nameof(oldValue.GroupId));
if (oldValue.Path != newValue.Path && newValue.Path.Length > 100)
throw new ArgumentException("The path cannot be longer than 100 characters.");
}
protected override void OnPostRemove(string key, GroupPermission value)
{
}
}
}

97
Store/GroupStore.cs Normal file
View File

@@ -0,0 +1,97 @@
using System.Collections.Concurrent;
using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Store.Internal;
namespace HermesSocketServer.Store
{
public class GroupStore : AutoSavedStore<string, Group>
{
public static readonly string[] AUTO_GENERATED_GROUP_NAMES = ["everyone", "subscribers", "vip", "moderators", "broadcaster"];
private IDictionary<string, ChatterGroupStore> _chatters;
private readonly string _userId;
private readonly Database _database;
private readonly ServerConfiguration _configuration;
private readonly Serilog.ILogger _logger;
public IDictionary<string, ChatterGroupStore> Chatters { get => _chatters; }
public GroupStore(string userId, DatabaseTable table, Database database, ServerConfiguration configuration, Serilog.ILogger logger)
: base(table, database, logger)
{
_userId = userId;
_database = database;
_logger = logger;
_configuration = configuration;
_chatters = new ConcurrentDictionary<string, ChatterGroupStore>();
}
public override async Task Load()
{
var groups = new List<(string, string)>();
var data = new Dictionary<string, object>() { { "user", _userId } };
string sql = $"SELECT id, name, priority FROM \"Group\" WHERE \"userId\" = @user";
await _database.Execute(sql, data, (reader) =>
{
var id = reader.GetGuid(0).ToString();
var name = reader.GetString(1);
_store.Add(id, new Group()
{
Id = id,
UserId = _userId,
Name = name,
Priority = reader.GetInt32(2),
});
groups.Add((id, name));
});
_logger.Information($"Loaded {_store.Count} groups from database.");
// Load Chatter Groups
var chatterGroupTable = _configuration.Database.Tables["ChatterGroup"];
var groupChatters = groups.Select(group => new ChatterGroupStore(_userId, group.Item1, chatterGroupTable, _database, _logger)).ToArray();
List<Task> tasks = new List<Task>();
for (var i = 0; i < groups.Count; i++)
{
var store = groupChatters[i];
if (AUTO_GENERATED_GROUP_NAMES.All(n => n != groups[i].Item2))
tasks.Add(store.Load());
_chatters.Add(store.GroupId, store);
}
await Task.WhenAll(tasks);
}
public override async Task Save()
{
List<Task> tasks = _chatters.Values.Select(c => c.Save()).ToList();
tasks.Add(base.Save());
await Task.WhenAll(tasks);
}
protected override void OnInitialAdd(string key, Group value)
{
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Name, nameof(value.Name));
ArgumentNullException.ThrowIfNull(value.Priority, nameof(value.Priority));
}
protected override void OnInitialModify(string key, Group oldValue, Group newValue)
{
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Name, nameof(newValue.Name));
ArgumentNullException.ThrowIfNull(newValue.Priority, nameof(newValue.Priority));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
if (AUTO_GENERATED_GROUP_NAMES.Any(s => s == oldValue.Name))
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Name, newValue.Name, nameof(oldValue.Name));
}
protected override void OnPostRemove(string key, Group value)
{
}
}
}

View File

@@ -2,12 +2,14 @@ namespace HermesSocketServer.Store
{ {
public interface IStore<K, V> public interface IStore<K, V>
{ {
bool Exists(K key);
V? Get(K key); V? Get(K key);
IDictionary<K, V> Get(); IDictionary<K, V> Get();
Task Load(); Task Load();
bool Modify(K? key, Action<V> action); bool Modify(K? key, Action<V> modify);
bool Remove(K? key); bool Modify(K? key, V value);
bool Remove(K? key, bool fromCascade = false);
Task Save(); Task Save();
bool Set(K? key, V? value); bool Set(K? key, V value);
} }
} }

View File

@@ -46,13 +46,26 @@ namespace HermesSocketServer.Store.Internal
private async Task GenerateQuery(IList<K> keys, Func<int, string> generate, Func<string, IEnumerable<K>, IEnumerable<V>, Task<int>> execute) private async Task GenerateQuery(IList<K> keys, Func<int, string> generate, Func<string, IEnumerable<K>, IEnumerable<V>, Task<int>> execute)
{ {
ImmutableList<K>? list = null; ImmutableList<K>? list = null;
lock (_lock) _rwls.EnterUpgradeableReadLock();
try
{ {
if (!keys.Any()) if (!keys.Any())
return; return;
list = keys.ToImmutableList(); _rwls.EnterWriteLock();
keys.Clear(); try
{
list = keys.ToImmutableList();
keys.Clear();
}
finally
{
_rwls.ExitWriteLock();
}
}
finally
{
_rwls.ExitUpgradeableReadLock();
} }
var query = generate(list.Count); var query = generate(list.Count);

View File

@@ -48,16 +48,29 @@ namespace HermesSocketServer.Store.Internal
async (query, list) => await _generator.DoPreparedStatement(_database, query, list, _table.KeyColumns)); async (query, list) => await _generator.DoPreparedStatement(_database, query, list, _table.KeyColumns));
} }
private async Task GenerateQuery(IList<K> keys, Func<int, string> generate, Func<string, IEnumerable<K>, IEnumerable<V>, Task<int>> execute) private async Task GenerateQuery(IList<K> keys, Func<int, string> generate, Func<string, IEnumerable<K>, IEnumerable<V?>, Task<int>> execute)
{ {
ImmutableList<K>? list = null; ImmutableList<K>? list = null;
lock (_lock) _rwls.EnterUpgradeableReadLock();
try
{ {
if (!keys.Any()) if (!keys.Any())
return; return;
list = keys.ToImmutableList(); _rwls.EnterWriteLock();
keys.Clear(); try
{
list = keys.ToImmutableList();
keys.Clear();
}
finally
{
_rwls.ExitWriteLock();
}
}
finally
{
_rwls.ExitUpgradeableReadLock();
} }
var query = generate(list.Count); var query = generate(list.Count);
@@ -77,15 +90,27 @@ namespace HermesSocketServer.Store.Internal
private async Task GenerateDeleteQuery(IList<K> keys, IList<V> values, Func<int, string> generate, Func<string, IEnumerable<V>, Task<int>> execute) private async Task GenerateDeleteQuery(IList<K> keys, IList<V> values, Func<int, string> generate, Func<string, IEnumerable<V>, Task<int>> execute)
{ {
ImmutableList<V>? list = null; ImmutableList<V>? list = null;
lock (_lock) _rwls.EnterUpgradeableReadLock();
try
{ {
if (!keys.Any() || !values.Any()) { if (!keys.Any() || !values.Any())
return; return;
}
list = values.ToImmutableList(); _rwls.EnterWriteLock();
values.Clear(); try
keys.Clear(); {
list = values.ToImmutableList();
values.Clear();
keys.Clear();
}
finally
{
_rwls.ExitWriteLock();
}
}
finally
{
_rwls.ExitUpgradeableReadLock();
} }
var query = generate(list.Count); var query = generate(list.Count);

View File

@@ -18,7 +18,8 @@ namespace HermesSocketServer.Store.Internal
public GroupSaveSqlGenerator(IDictionary<string, string> columnsToProperties, IDictionary<string, string> columnTypes, Serilog.ILogger logger) public GroupSaveSqlGenerator(IDictionary<string, string> columnsToProperties, IDictionary<string, string> columnTypes, Serilog.ILogger logger)
{ {
_columnPropertyRelations = columnsToProperties.ToDictionary(p => p.Key, p => typeof(T).GetProperty(p.Value)); var type = typeof(T);
_columnPropertyRelations = columnsToProperties.ToDictionary(p => p.Key, p => type.GetProperty(p.Value));
_columnTypes = columnTypes; _columnTypes = columnTypes;
_logger = logger; _logger = logger;
@@ -94,15 +95,14 @@ namespace HermesSocketServer.Store.Internal
var ctp = columns.ToDictionary(c => c, c => _columnPropertyRelations[c]); var ctp = columns.ToDictionary(c => c, c => _columnPropertyRelations[c]);
var sb = new StringBuilder(); var sb = new StringBuilder();
var columnsLower = columns.Select(c => c.ToLower());
sb.Append($"INSERT INTO \"{table}\" (\"{string.Join("\", \"", columns)}\") VALUES "); sb.Append($"INSERT INTO \"{table}\" (\"{string.Join("\", \"", columns)}\") VALUES ");
for (var row = 0; row < rows; row++) for (var row = 0; row < rows; row++)
{ {
sb.Append("("); sb.Append("(");
foreach (var column in columnsLower) foreach (var column in columns)
{ {
sb.Append('@') sb.Append('@')
.Append(column) .Append(column.ToLower())
.Append(row); .Append(row);
if (typeMapping.TryGetValue(column, out var type)) if (typeMapping.TryGetValue(column, out var type))

View File

@@ -10,7 +10,7 @@ namespace HermesSocketServer.Store.Internal
protected readonly IList<K> _added; protected readonly IList<K> _added;
protected readonly IList<K> _modified; protected readonly IList<K> _modified;
protected readonly IList<K> _deleted; protected readonly IList<K> _deleted;
protected readonly object _lock; protected readonly ReaderWriterLockSlim _rwls;
public GroupSaveStore() public GroupSaveStore()
@@ -19,96 +19,71 @@ namespace HermesSocketServer.Store.Internal
_added = new List<K>(); _added = new List<K>();
_modified = new List<K>(); _modified = new List<K>();
_deleted = new List<K>(); _deleted = new List<K>();
_lock = new object(); _rwls = new ReaderWriterLockSlim();
} }
public abstract Task Load(); public abstract Task Load();
protected abstract void OnInitialAdd(K key, V value); protected abstract void OnInitialAdd(K key, V value);
protected abstract void OnInitialModify(K key, V value); protected abstract void OnInitialModify(K key, V value, V newValue);
protected abstract void OnPostRemove(K key, V value); protected abstract void OnPostRemove(K key, V value);
public abstract Task Save(); public abstract Task Save();
public bool Exists(K key)
{
_rwls.EnterReadLock();
try
{
return _store.ContainsKey(key);
}
finally
{
_rwls.ExitReadLock();
}
}
public V? Get(K key) public V? Get(K key)
{ {
lock (_lock) _rwls.EnterReadLock();
try
{ {
if (_store.TryGetValue(key, out var value)) if (_store.TryGetValue(key, out var value))
return value; return value;
} }
finally
{
_rwls.ExitReadLock();
}
return null; return null;
} }
public IDictionary<K, V> Get() public IDictionary<K, V> Get()
{ {
lock (_lock) _rwls.EnterReadLock();
try
{ {
return _store.ToImmutableDictionary(); return _store.ToImmutableDictionary();
} }
finally
{
_rwls.ExitReadLock();
}
} }
public bool Modify(K? key, Action<V> action) public bool Modify(K? key, V value)
{ {
if (key == null) if (key == null)
return false; return false;
lock (_lock) _rwls.EnterUpgradeableReadLock();
try
{ {
if (_store.TryGetValue(key, out V? value)) if (_store.TryGetValue(key, out V? oldValue))
{ {
if (value == null) _rwls.EnterWriteLock();
return false; try
OnInitialModify(key, value);
action(value);
if (!_added.Contains(key) && !_modified.Contains(key))
{ {
_modified.Add(key); OnInitialModify(key, oldValue, value);
}
return true;
}
}
return false;
}
public bool Remove(K? key)
{
if (key == null)
return false;
lock (_lock)
{
if (_store.TryGetValue(key, out var value))
{
if (_store.Remove(key))
{
OnPostRemove(key, value);
if (!_added.Remove(key))
{
_modified.Remove(key);
if (!_deleted.Contains(key))
{
_deleted.Add(key);
}
}
return true;
}
}
}
return false;
}
public bool Set(K? key, V? value)
{
if (key == null || value == null)
return false;
lock (_lock)
{
if (_store.TryGetValue(key, out V? fetched))
{
if (fetched != value)
{
OnInitialModify(key, value);
_store[key] = value; _store[key] = value;
if (!_added.Contains(key) && !_modified.Contains(key)) if (!_added.Contains(key) && !_modified.Contains(key))
{ {
@@ -116,6 +91,99 @@ namespace HermesSocketServer.Store.Internal
} }
return true; return true;
} }
finally
{
_rwls.ExitWriteLock();
}
}
}
finally
{
_rwls.ExitUpgradeableReadLock();
}
return false;
}
public bool Modify(K? key, Action<V> modify)
{
if (key == null)
return false;
_rwls.EnterUpgradeableReadLock();
try
{
if (_store.TryGetValue(key, out V? value))
{
_rwls.EnterWriteLock();
try
{
modify(value);
if (!_added.Contains(key) && !_modified.Contains(key))
{
_modified.Add(key);
}
return true;
}
finally
{
_rwls.ExitWriteLock();
}
}
}
finally
{
_rwls.ExitUpgradeableReadLock();
}
return false;
}
public bool Remove(K? key, bool fromCascade = false)
{
if (key == null)
return false;
_rwls.EnterWriteLock();
try
{
if (_store.TryGetValue(key, out var value))
{
_store.Remove(key);
OnPostRemove(key, value);
if (!_added.Remove(key))
{
_modified.Remove(key);
if (!fromCascade && !_deleted.Contains(key))
{
_deleted.Add(key);
}
}
return true;
}
}
finally
{
_rwls.ExitWriteLock();
}
return false;
}
public bool Set(K? key, V value)
{
if (key == null)
return false;
_rwls.EnterWriteLock();
try
{
if (_store.TryGetValue(key, out V? fetched))
{
OnInitialModify(key, fetched, value);
_store[key] = value;
if (!_added.Contains(key) && !_modified.Contains(key))
{
_modified.Add(key);
}
return true;
} }
else else
{ {
@@ -128,7 +196,10 @@ namespace HermesSocketServer.Store.Internal
return true; return true;
} }
} }
return false; finally
{
_rwls.ExitWriteLock();
}
} }
} }
} }

View File

@@ -1,4 +1,5 @@
using HermesSocketLibrary.db; using HermesSocketLibrary.db;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketServer.Messages; using HermesSocketServer.Messages;
using HermesSocketServer.Store.Internal; using HermesSocketServer.Store.Internal;
@@ -7,14 +8,16 @@ namespace HermesSocketServer.Store
public class PolicyStore : AutoSavedStore<string, Policy> public class PolicyStore : AutoSavedStore<string, Policy>
{ {
private readonly string _userId; private readonly string _userId;
private readonly IStore<string, Group> _groups;
private readonly Database _database; private readonly Database _database;
private readonly Serilog.ILogger _logger; private readonly Serilog.ILogger _logger;
public PolicyStore(string userId, DatabaseTable table, Database database, Serilog.ILogger logger) public PolicyStore(string userId, DatabaseTable table, IStore<string, Group> groups, Database database, Serilog.ILogger logger)
: base(table, database, logger) : base(table, database, logger)
{ {
_userId = userId; _userId = userId;
_groups = groups;
_database = database; _database = database;
_logger = logger; _logger = logger;
} }
@@ -41,10 +44,32 @@ namespace HermesSocketServer.Store
protected override void OnInitialAdd(string key, Policy value) protected override void OnInitialAdd(string key, Policy value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentOutOfRangeException.ThrowIfEqual(value.GroupId, default, nameof(value.GroupId));
ArgumentOutOfRangeException.ThrowIfNegativeOrZero(value.Usage, nameof(value.Usage));
ArgumentOutOfRangeException.ThrowIfGreaterThan(value.Usage, 99, nameof(value.Usage));
ArgumentOutOfRangeException.ThrowIfNegativeOrZero(value.Span, nameof(value.Span));
ArgumentOutOfRangeException.ThrowIfLessThan(value.Span, 1000, nameof(value.Span));
ArgumentOutOfRangeException.ThrowIfGreaterThan(value.Span, 86400, nameof(value.Span));
if (_groups.Get(value.GroupId.ToString()) == null)
throw new ArgumentException("The group id does not exist.");
} }
protected override void OnInitialModify(string key, Policy value) protected override void OnInitialModify(string key, Policy oldValue, Policy newValue)
{ {
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentOutOfRangeException.ThrowIfEqual(newValue.GroupId, default, nameof(newValue.GroupId));
ArgumentOutOfRangeException.ThrowIfNegativeOrZero(newValue.Usage, nameof(newValue.Usage));
ArgumentOutOfRangeException.ThrowIfGreaterThan(newValue.Usage, 99, nameof(newValue.Usage));
ArgumentOutOfRangeException.ThrowIfNegativeOrZero(newValue.Span, nameof(newValue.Span));
ArgumentOutOfRangeException.ThrowIfLessThan(newValue.Span, 1000, nameof(newValue.Span));
ArgumentOutOfRangeException.ThrowIfGreaterThan(newValue.Span, 86400, nameof(newValue.Span));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Id, newValue.Id, nameof(oldValue.Id));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
} }
protected override void OnPostRemove(string key, Policy value) protected override void OnPostRemove(string key, Policy value)

View File

@@ -7,14 +7,16 @@ namespace HermesSocketServer.Store
public class RedemptionStore : AutoSavedStore<string, Redemption> public class RedemptionStore : AutoSavedStore<string, Redemption>
{ {
private readonly string _userId; private readonly string _userId;
private readonly IStore<string, RedeemableAction> _actions;
private readonly Database _database; private readonly Database _database;
private readonly Serilog.ILogger _logger; private readonly Serilog.ILogger _logger;
public RedemptionStore(string userId, DatabaseTable table, Database database, Serilog.ILogger logger) public RedemptionStore(string userId, DatabaseTable table, IStore<string, RedeemableAction> actions, Database database, Serilog.ILogger logger)
: base(table, database, logger) : base(table, database, logger)
{ {
_userId = userId; _userId = userId;
_actions = actions;
_database = database; _database = database;
_logger = logger; _logger = logger;
} }
@@ -36,18 +38,43 @@ namespace HermesSocketServer.Store
ActionName = reader.GetString(4), ActionName = reader.GetString(4),
}); });
}); });
_logger.Information($"Loaded {_store.Count} TTS chatter voices from database."); _logger.Information($"Loaded {_store.Count} redemptions from database.");
} }
protected override void OnInitialAdd(string key, Redemption value) protected override void OnInitialAdd(string key, Redemption value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.RedemptionId, nameof(value.RedemptionId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.ActionName, nameof(value.ActionName));
ArgumentNullException.ThrowIfNull(value.State, nameof(value.State));
ArgumentNullException.ThrowIfNull(value.Order, nameof(value.Order));
ArgumentOutOfRangeException.ThrowIfNegative(value.Order, nameof(value.Order));
ArgumentOutOfRangeException.ThrowIfGreaterThan(value.Order, 99, nameof(value.Order));
if (_actions.Get(value.ActionName) == null)
throw new ArgumentException("The action name does not exist.");
} }
protected override void OnInitialModify(string key, Redemption value) protected override void OnInitialModify(string key, Redemption oldValue, Redemption newValue)
{ {
ArgumentNullException.ThrowIfNull(newValue);
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.RedemptionId, nameof(newValue.RedemptionId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.ActionName, nameof(newValue.ActionName));
ArgumentNullException.ThrowIfNull(newValue.State, nameof(newValue.State));
ArgumentNullException.ThrowIfNull(newValue.Order, nameof(newValue.Order));
ArgumentOutOfRangeException.ThrowIfNegative(newValue.Order, nameof(newValue.Order));
ArgumentOutOfRangeException.ThrowIfGreaterThan(newValue.Order, 99, nameof(newValue.Order));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Id, newValue.Id, nameof(newValue.Id));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(newValue.UserId));
if (oldValue.ActionName != newValue.ActionName && _actions.Get(newValue.ActionName) == null)
throw new ArgumentException("The action name does not exist.");
} }
protected override void OnPostRemove(string key, Redemption value) protected override void OnPostRemove(string key, Redemption? value)
{ {
} }
} }

View File

@@ -22,26 +22,45 @@ namespace HermesSocketServer.Store
public override async Task Load() public override async Task Load()
{ {
var data = new Dictionary<string, object>() { { "user", _userId } }; var data = new Dictionary<string, object>() { { "user", _userId } };
string sql = $"SELECT id, search, replace FROM \"TtsWordFilter\" WHERE \"userId\" = @user"; string sql = $"SELECT id, search, replace, flag FROM \"TtsWordFilter\" WHERE \"userId\" = @user";
await _database.Execute(sql, data, (reader) => await _database.Execute(sql, data, (reader) =>
{ {
var id = reader.GetString(0); var id = reader.GetString(0);
_store.Add(id.ToString(), new TTSWordFilter() _store.Add(id.ToString(), new TTSWordFilter()
{ {
Id = id, Id = id,
UserId = _userId,
Search = reader.GetString(1), Search = reader.GetString(1),
Replace = reader.GetString(2) Replace = reader.GetString(2),
Flag = reader.GetInt32(3),
}); });
}); });
_logger.Information($"Loaded {_store.Count} TTS chatter voices from database."); _logger.Information($"Loaded {_store.Count} TTS filters from database.");
} }
protected override void OnInitialAdd(string key, TTSWordFilter value) protected override void OnInitialAdd(string key, TTSWordFilter value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Id, nameof(value.Id));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Search, nameof(value.Search));
ArgumentNullException.ThrowIfNull(value.Replace, nameof(value.Replace));
ArgumentOutOfRangeException.ThrowIfNegative(value.Flag, nameof(value.Flag));
ArgumentOutOfRangeException.ThrowIfGreaterThan(value.Flag, 2047, nameof(value.Flag));
} }
protected override void OnInitialModify(string key, TTSWordFilter value) protected override void OnInitialModify(string key, TTSWordFilter oldValue, TTSWordFilter newValue)
{ {
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Id, nameof(newValue.Id));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Search, nameof(newValue.Search));
ArgumentNullException.ThrowIfNull(newValue.Replace, nameof(newValue.Replace));
ArgumentOutOfRangeException.ThrowIfNegative(newValue.Flag, nameof(newValue.Flag));
ArgumentOutOfRangeException.ThrowIfGreaterThan(newValue.Flag, 2047, nameof(newValue.Flag));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Id, newValue.Id, nameof(oldValue.Id));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
} }
protected override void OnPostRemove(string key, TTSWordFilter value) protected override void OnPostRemove(string key, TTSWordFilter value)

View File

@@ -19,7 +19,7 @@ namespace HermesSocketServer.Store
public override async Task Load() public override async Task Load()
{ {
string sql = "SELECT id, name, email, role, \"ttsDefaultVoice\" FROM \"User\";"; string sql = "SELECT id, name, email, role, \"ttsDefaultVoice\", seolkey FROM \"User\";";
await _database.Execute(sql, new Dictionary<string, object>(), (reader) => await _database.Execute(sql, new Dictionary<string, object>(), (reader) =>
{ {
string id = reader.GetString(0); string id = reader.GetString(0);
@@ -30,6 +30,7 @@ namespace HermesSocketServer.Store
Email = reader.GetString(2), Email = reader.GetString(2),
Role = reader.GetString(3), Role = reader.GetString(3),
DefaultVoice = reader.GetString(4), DefaultVoice = reader.GetString(4),
StreamElementsOverlayKey = reader.GetString(5),
}); });
}); });
_logger.Information($"Loaded {_store.Count} users from database."); _logger.Information($"Loaded {_store.Count} users from database.");
@@ -39,7 +40,7 @@ namespace HermesSocketServer.Store
{ {
} }
protected override void OnInitialModify(string key, User value) protected override void OnInitialModify(string key, User oldValue, User newValue)
{ {
} }

View File

@@ -8,18 +8,21 @@ namespace HermesSocketServer.Store
public class VoiceStateStore : ComplexAutoSavedStore<string, TTSVoiceState> public class VoiceStateStore : ComplexAutoSavedStore<string, TTSVoiceState>
{ {
private readonly string _userId; private readonly string _userId;
private readonly IStore<string, TTSVoice> _voices;
private readonly VoiceIdValidator _idValidator; private readonly VoiceIdValidator _idValidator;
private readonly Database _database; private readonly Database _database;
private readonly Serilog.ILogger _logger; private readonly Serilog.ILogger _logger;
public VoiceStateStore(string userId, VoiceIdValidator voiceIdValidator, DatabaseTable table, Database database, Serilog.ILogger logger) public VoiceStateStore(string userId, DatabaseTable table, IStore<string, TTSVoice> voices, Database database, Serilog.ILogger logger)
: base(table, database, logger) : base(table, database, logger)
{ {
_userId = userId; _userId = userId;
_idValidator = voiceIdValidator; _voices = voices;
_database = database; _database = database;
_logger = logger; _logger = logger;
_idValidator = new VoiceIdValidator();
} }
public override async Task Load() public override async Task Load()
@@ -41,11 +44,26 @@ namespace HermesSocketServer.Store
protected override void OnInitialAdd(string key, TTSVoiceState value) protected override void OnInitialAdd(string key, TTSVoiceState value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
_idValidator.Check(value.Id); _idValidator.Check(value.Id);
ArgumentNullException.ThrowIfNull(value, nameof(value));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Id, nameof(value.Id));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentNullException.ThrowIfNull(value.Enabled, nameof(value.Enabled));
if (_voices.Get(value.Id) == null)
throw new ArgumentException("The voice does not exist.");
} }
protected override void OnInitialModify(string key, TTSVoiceState value) protected override void OnInitialModify(string key, TTSVoiceState oldValue, TTSVoiceState newValue)
{ {
ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Id, nameof(newValue.Id));
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.UserId, nameof(newValue.UserId));
ArgumentNullException.ThrowIfNull(newValue.Enabled, nameof(newValue.Enabled));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Id, newValue.Id, nameof(oldValue.Id));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfEqual(oldValue.Enabled, newValue.Enabled, nameof(oldValue.Enabled));
} }
protected override void OnPostRemove(string key, TTSVoiceState value) protected override void OnPostRemove(string key, TTSVoiceState value)

View File

@@ -39,13 +39,19 @@ namespace HermesSocketServer.Store
protected override void OnInitialAdd(string key, TTSVoice value) protected override void OnInitialAdd(string key, TTSVoice value)
{ {
ArgumentException.ThrowIfNullOrWhiteSpace(key, nameof(key));
ArgumentNullException.ThrowIfNull(value, nameof(value));
_idValidator.Check(value.Id); _idValidator.Check(value.Id);
_nameValidator.Check(value.Name); _nameValidator.Check(value.Name);
} }
protected override void OnInitialModify(string key, TTSVoice value) protected override void OnInitialModify(string key, TTSVoice oldValue, TTSVoice newValue)
{ {
_nameValidator.Check(value.Name); ArgumentNullException.ThrowIfNull(newValue, nameof(newValue));
_idValidator.Check(newValue.Id);
_nameValidator.Check(newValue.Name);
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Id, newValue.Id, nameof(oldValue.Id));
ArgumentOutOfRangeException.ThrowIfEqual(oldValue.Name, newValue.Name, nameof(oldValue.Name));
} }
protected override void OnPostRemove(string key, TTSVoice value) protected override void OnPostRemove(string key, TTSVoice value)