Fixed issues with TTS voice changes. Added proper handling for slave clients. Fixed several stores. Fixed database saving to safely save foreign keys.

This commit is contained in:
Tom
2025-03-06 16:11:36 +00:00
parent 3e717522c2
commit fd0bca5c7c
13 changed files with 139 additions and 96 deletions

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

@ -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,28 +7,27 @@ 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))
return RequestResult.Failed("Chatter should be an integer, representing the Twitch User Id of the chatter."); return Task.FromResult(RequestResult.Failed("Invalid Twitch user id."));
data["user"] = channel.Id;
var voiceId = 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 either non-existent or disabled on this channel."); return Task.FromResult(RequestResult.Failed("Voice is either non-existent or disabled on this channel."));
var voice = new ChatterVoice() var voice = new ChatterVoice()
{ {
@ -43,9 +40,9 @@ namespace HermesSocketServer.Requests
if (result) if (result)
{ {
_logger.Information($"Updated chatter's selected tts voice on channel [chatter id: {chatterId}][voice id: {voiceId}][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(voice); 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

@ -20,20 +20,20 @@ namespace HermesSocketServer.Requests
public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data) public Task<RequestResult> Grant(Channel channel, IDictionary<string, object> data)
{ {
var id = data["voice"].ToString()!; var voiceId = data["voice"].ToString()!;
var state = data["state"].ToString() == "True"; var state = data["state"].ToString()?.ToLower() == "true";
var voiceState = new TTSVoiceState() var voiceState = new TTSVoiceState()
{ {
Id = id, Id = voiceId,
UserId = channel.Id, UserId = channel.Id,
Enabled = state, Enabled = state,
}; };
var result = channel.VoiceStates.Set(id, voiceState); var result = channel.VoiceStates.Set(voiceId, voiceState);
if (result) if (result)
{ {
_logger.Information($"Updated voice state on channel [voice id: {id}][state: {state}][channel: {channel.Id}]"); _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.Successful(voiceState));
} }
return Task.FromResult(RequestResult.Failed("Something went wrong when updating the database.")); return Task.FromResult(RequestResult.Failed("Something went wrong when updating the database."));

View File

@ -2,6 +2,7 @@ using System.Net.WebSockets;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
using CommonSocketLibrary.Common; using CommonSocketLibrary.Common;
using HermesSocketLibrary.Requests.Messages;
using HermesSocketLibrary.Socket.Data; using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Socket; using HermesSocketServer.Socket;
using ILogger = Serilog.ILogger; using ILogger = Serilog.ILogger;
@ -32,7 +33,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 +45,71 @@ 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($"rxm: {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;
}
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}]");
return; return;
} }
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; return;
} }
/** 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) if (message.OpCode == 0)
obj.Data = JsonSerializer.Deserialize<HeartbeatMessage>(data, _options); message.Data = JsonSerializer.Deserialize<HeartbeatMessage>(data, _options);
else if (obj.OpCode == 1) else if (message.OpCode == 1)
obj.Data = JsonSerializer.Deserialize<HermesLoginMessage>(data, _options); message.Data = JsonSerializer.Deserialize<HermesLoginMessage>(data, _options);
else if (obj.OpCode == 3) else if (message.OpCode == 3)
obj.Data = JsonSerializer.Deserialize<RequestMessage>(data, _options); message.Data = JsonSerializer.Deserialize<RequestMessage>(data, _options);
else if (obj.OpCode == 5) else if (message.OpCode == 5)
obj.Data = JsonSerializer.Deserialize<LoggingMessage>(data, _options); message.Data = JsonSerializer.Deserialize<LoggingMessage>(data, _options);
else if (obj.OpCode == 6) else if (message.OpCode == 6)
obj.Data = JsonSerializer.Deserialize<ChatterMessage>(data, _options); message.Data = JsonSerializer.Deserialize<ChatterMessage>(data, _options);
else if (obj.OpCode == 7) else if (message.OpCode == 7)
obj.Data = JsonSerializer.Deserialize<EmoteDetailsMessage>(data, _options); message.Data = JsonSerializer.Deserialize<EmoteDetailsMessage>(data, _options);
else if (obj.OpCode == 8) else if (message.OpCode == 8)
obj.Data = JsonSerializer.Deserialize<EmoteUsageMessage>(data, _options); message.Data = JsonSerializer.Deserialize<EmoteUsageMessage>(data, _options);
//else if (message.OpCode == 9)
// message.Data = JsonSerializer.Deserialize<SlaveMessage>(data, _options);
else else
{ {
await socket.Send(5, new LoggingMessage("Received an invalid message: " + message, HermesLoggingLevel.Error)); await socket.Send(5, new LoggingMessage("Received a message with invalid data: " + messageString, HermesLoggingLevel.Error));
continue; continue;
} }
await _handlers.Execute(socket, obj.OpCode, obj.Data); await _handlers.Execute(socket, message.OpCode, message.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 +120,26 @@ 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.Information(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.
// TODO: Ensure one of the clients is always a non-slave.
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;
}
}
} }
} }
} }

View File

@ -38,8 +38,8 @@ namespace HermesSocketServer.Services
{ {
lock (_lock) lock (_lock)
{ {
if (_channels.ContainsKey(userId)) if (_channels.TryGetValue(userId, out var channel))
return Task.FromResult<Channel?>(null); return Task.FromResult<Channel?>(channel);
var actionTable = _configuration.Database.Tables["Action"]; var actionTable = _configuration.Database.Tables["Action"];
var chatterTable = _configuration.Database.Tables["Chatter"]; var chatterTable = _configuration.Database.Tables["Chatter"];
@ -62,7 +62,7 @@ namespace HermesSocketServer.Services
var redemptions = new RedemptionStore(userId, redemptionTable, actions, _database, _logger); var redemptions = new RedemptionStore(userId, redemptionTable, actions, _database, _logger);
var voiceStates = new VoiceStateStore(userId, ttsVoiceStateTable, _voices, _database, _logger); var voiceStates = new VoiceStateStore(userId, ttsVoiceStateTable, _voices, _database, _logger);
var channel = new Channel() channel = new Channel()
{ {
Id = userId, Id = userId,
User = user, User = user,

View File

@ -45,16 +45,24 @@ namespace HermesSocketServer.Socket.Handlers
if (userId == null) if (userId == null)
return; return;
IEnumerable<WebSocketUser?> recipients = Enumerable.Empty<WebSocketUser?>();
lock (_lock) lock (_lock)
{ {
if (sender.Id != null) if (sender.Id != null)
return; throw new Exception("User already logged in.");
sender.Id = userId; sender.Id = userId;
sender.ApiKey = data.ApiKey;
sender.WebLogin = data.WebLogin; if (string.IsNullOrWhiteSpace(sender.Id))
throw new Exception("Credentials do not match.");
recipients = _sockets.GetSockets(userId).ToList().Where(s => s.SessionId != sender.SessionId);
sender.Slave = data.WebLogin || recipients.Where(r => r?.WebLogin != true).Any();
} }
sender.ApiKey = data.ApiKey;
sender.WebLogin = data.WebLogin;
var channel = _manager.Get(userId); var channel = _manager.Get(userId);
if (channel == null) if (channel == null)
{ {
@ -73,11 +81,12 @@ namespace HermesSocketServer.Socket.Handlers
} }
if (string.IsNullOrEmpty(channel.User.DefaultVoice)) if (string.IsNullOrEmpty(channel.User.DefaultVoice))
_logger.Warning($"No default voice was set for an user [user id: {userId}][api key: {data.ApiKey}]"); _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"; 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 result2 = await _database.ExecuteScalar(sql, new Dictionary<string, object>() { { "user", userId }, { "provider", "twitch" } });
var providerId = result2?.ToString(); var providerId = result2?.ToString();
if (providerId == null) { if (providerId == null)
{
_logger.Warning($"Could not find the Provider Account Id [user id: {userId}][provider: twitch]"); _logger.Warning($"Could not find the Provider Account Id [user id: {userId}][provider: twitch]");
return; return;
} }
@ -86,7 +95,7 @@ namespace HermesSocketServer.Socket.Handlers
{ {
UserId = userId, UserId = userId,
ProviderAccountId = providerId, ProviderAccountId = providerId,
SessionId = sender.UID, SessionId = sender.SessionId,
UserName = channel.User.Name, UserName = channel.User.Name,
OwnerId = _configuration.Tts.OwnerId, OwnerId = _configuration.Tts.OwnerId,
Admin = sender.Admin, Admin = sender.Admin,
@ -94,6 +103,7 @@ namespace HermesSocketServer.Socket.Handlers
WordFilters = channel.Filters.Get().Values, WordFilters = channel.Filters.Get().Values,
DefaultTTSVoice = channel.User.DefaultVoice ?? _configuration.Tts.DefaultTtsVoice, DefaultTTSVoice = channel.User.DefaultVoice ?? _configuration.Tts.DefaultTtsVoice,
TTSVoicesAvailable = _voices.Get().ToDictionary(v => v.Key, v => v.Value.Name), TTSVoicesAvailable = _voices.Get().ToDictionary(v => v.Key, v => v.Value.Name),
Slave = sender.Slave,
}; };
var userIdDict = new Dictionary<string, object>() { { "user", userId } }; var userIdDict = new Dictionary<string, object>() { { "user", userId } };
@ -122,7 +132,7 @@ namespace HermesSocketServer.Socket.Handlers
await sender.Send(2, ack); await sender.Send(2, ack);
string version = data.MajorVersion == null ? "unknown" : $"{data.MajorVersion}.{data.MinorVersion}"; 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}]"); _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() ack = new LoginAckMessage()
@ -133,7 +143,6 @@ namespace HermesSocketServer.Socket.Handlers
WebLogin = data.WebLogin WebLogin = data.WebLogin
}; };
var recipients = _sockets.GetSockets(userId).ToList().Where(s => s.UID != sender.UID);
var tasks = new List<Task>(); var tasks = new List<Task>();
foreach (var socket in recipients) foreach (var socket in recipients)
{ {

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

@ -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

@ -47,6 +47,9 @@ namespace HermesSocketServer.Store
ArgumentException.ThrowIfNullOrWhiteSpace(value.Name, nameof(value.Name)); ArgumentException.ThrowIfNullOrWhiteSpace(value.Name, nameof(value.Name));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Type, nameof(value.Type)); ArgumentException.ThrowIfNullOrWhiteSpace(value.Type, nameof(value.Type));
ArgumentNullException.ThrowIfNull(value.Data, nameof(value.Data)); 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 oldValue, RedeemableAction newValue) protected override void OnInitialModify(string key, RedeemableAction oldValue, RedeemableAction newValue)

View File

@ -54,6 +54,9 @@ namespace HermesSocketServer.Store
ArgumentException.ThrowIfNullOrWhiteSpace(value.Scope, nameof(value.Scope)); ArgumentException.ThrowIfNullOrWhiteSpace(value.Scope, nameof(value.Scope));
ArgumentNullException.ThrowIfNull(value.ExpiresAt, nameof(value.ExpiresAt)); ArgumentNullException.ThrowIfNull(value.ExpiresAt, nameof(value.ExpiresAt));
ArgumentNullException.ThrowIfNull(value.Default, nameof(value.Default)); 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) protected override void OnInitialModify(string key, Connection oldValue, Connection newValue)
@ -70,6 +73,8 @@ namespace HermesSocketServer.Store
ArgumentNullException.ThrowIfNull(newValue.Default, nameof(newValue.Default)); ArgumentNullException.ThrowIfNull(newValue.Default, nameof(newValue.Default));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId)); ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.Name, newValue.Name, nameof(oldValue.Name)); 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));
} }
} }
} }

View File

@ -48,6 +48,8 @@ namespace HermesSocketServer.Store
ArgumentNullException.ThrowIfNull(value.GroupId, nameof(value.GroupId)); ArgumentNullException.ThrowIfNull(value.GroupId, nameof(value.GroupId));
ArgumentException.ThrowIfNullOrWhiteSpace(value.Path, nameof(value.Path)); 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) if (_groups.Get(value.GroupId.ToString()) == null)
throw new ArgumentException("The group id does not exist."); throw new ArgumentException("The group id does not exist.");
} }
@ -60,6 +62,9 @@ namespace HermesSocketServer.Store
ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Path, nameof(newValue.Path)); ArgumentException.ThrowIfNullOrWhiteSpace(newValue.Path, nameof(newValue.Path));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId)); ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.UserId, newValue.UserId, nameof(oldValue.UserId));
ArgumentOutOfRangeException.ThrowIfNotEqual(oldValue.GroupId, newValue.GroupId, nameof(oldValue.GroupId)); 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) protected override void OnPostRemove(string key, GroupPermission value)

View File

@ -50,7 +50,9 @@ namespace HermesSocketServer.Store
ArgumentException.ThrowIfNullOrWhiteSpace(value.Id, nameof(value.Id)); ArgumentException.ThrowIfNullOrWhiteSpace(value.Id, nameof(value.Id));
ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId)); ArgumentException.ThrowIfNullOrWhiteSpace(value.UserId, nameof(value.UserId));
ArgumentNullException.ThrowIfNull(value.Enabled, nameof(value.Enabled)); ArgumentNullException.ThrowIfNull(value.Enabled, nameof(value.Enabled));
ArgumentNullException.ThrowIfNull(_voices.Get(value.Id));
if (_voices.Get(value.Id) == null)
throw new ArgumentException("The voice does not exist.");
} }
protected override void OnInitialModify(string key, TTSVoiceState oldValue, TTSVoiceState newValue) protected override void OnInitialModify(string key, TTSVoiceState oldValue, TTSVoiceState newValue)