using Govor.Core.Models.Users;

Govor.API.Tests/UnitTests/Infrastructure/Validators/MessageValidatorTests.cs

@@ -1,6 +1,7 @@
 using AutoFixture;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.API.Tests.UnitTests.Infrastructure.Validators;
 

Govor.API.Tests/UnitTests/Infrastructure/Validators/UserValidatorTests.cs

@@ -1,6 +1,7 @@
 using AutoFixture;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.API.Tests.UnitTests.Infrastructure.Validators;
 

Govor.API/Controllers/AdminStuff/FriendshipsController.cs

@@ -1,5 +1,6 @@
 using Govor.Contracts.DTOs;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Friendships;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.AspNetCore.Authorization;

Govor.API/Controllers/AdminStuff/UsersController.cs

@@ -1,6 +1,7 @@
 using Govor.API.Services.AdminsStuff.Interfaces;
+using Govor.Application.Interfaces;
 using Govor.Contracts.Responses.Admins;
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Mvc;

Govor.API/Controllers/Friends/FriendshipController.cs

@@ -2,7 +2,7 @@ using Govor.Application.Exceptions.FriendsService;
 using Govor.Application.Interfaces.Friends;
 using Govor.Application.Interfaces.Infrastructure.Extensions;
 using Govor.Contracts.DTOs;
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Microsoft.AspNetCore.Mvc;
 
 namespace Govor.API.Controllers.Friends;

Govor.API/Extensions/ConfigurationProgramExtensions.cs

@@ -4,7 +4,6 @@ using Govor.Application.Infrastructure.AdminsStuff;
 using Govor.Application.Infrastructure.Extensions;
 using Govor.Application.Infrastructure.Validators;
 using Govor.Application.Interfaces;
-using Govor.Application.Interfaces.AdminsStuff;
 using Govor.Application.Interfaces.Authentication;
 using Govor.Application.Interfaces.Friends;
 using Govor.Application.Interfaces.Infrastructure.Extensions;
@@ -16,6 +15,8 @@ using Govor.Application.Services.Messages;
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Admins;
 using Govor.Core.Repositories.Friendships;
 using Govor.Core.Repositories.Groups;

Govor.API/Hubs/ChatsHub.cs

@@ -1,10 +1,10 @@
-using Govor.API.Services;
+using Govor.Application.Exceptions.VerifyFriendship;
 using Govor.Application.Interfaces;
 using Govor.Application.Interfaces.Messages;
 using Govor.Application.Interfaces.Messages.Parameters;
 using Govor.Contracts.Requests.SignalR;
 using Govor.Contracts.Responses.SignalR;
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.SignalR;
 
@@ -85,142 +85,190 @@ public class ChatsHub : Hub
         await base.OnDisconnectedAsync(exception);
     }
 
-    public async Task Send(MessageRequest request)
+    public async Task<HubResult<UserMessageResponse>> Send(MessageRequest request)
     {
         var senderId = GetUserId();
 
         if (string.IsNullOrWhiteSpace(request.EncryptedContent) &&
             (request.MediaAttachments == null || !request.MediaAttachments.Any()))
         {
-            _logger.LogWarning("Empty message (no content and no attachments) received from user {UserId}", senderId);
+            _logger.LogWarning("Empty message received from user {UserId}", senderId);
-            throw new ArgumentException("Message cannot be empty (must have content or attachments).");
+            return HubResult<UserMessageResponse>.BadRequest("Message must contain content or media.");
         }
 
-        _logger.LogInformation("Message send initiated by {SenderId} to {RecipientId} of type {RecipientType}",
+        _logger.LogInformation("Sending message from {SenderId} to {RecipientId} ({RecipientType})",
             senderId, request.RecipientId, request.RecipientType);
 
-        var sendMessageParams = new SendMessage(EncryptContent: request.EncryptedContent,
+        var sendMessageParams = new SendMessage(
-            ReplyToMessageId: request.ReplyToMessageId, FromUserId: senderId, RecipientId: request.RecipientId,
+            EncryptContent: request.EncryptedContent,
-            RecipientType: request.RecipientType, SendAt: DateTime.UtcNow,
+            ReplyToMessageId: request.ReplyToMessageId,
+            FromUserId: senderId,
+            RecipientId: request.RecipientId,
+            RecipientType: request.RecipientType,
+            SendAt: DateTime.UtcNow,
             Media: request.MediaAttachments?.Select(f => new SendMedia(f.MediaId, f.EncryptedKey)) ??
-                   Array.Empty<SendMedia>());
+                   Array.Empty<SendMedia>()
-
-        var result = await _messageCommandService.SendMessageAsync(sendMessageParams);
-
-        if (!result.IsSuccess || result.Message.Id == Guid.Empty)
-        {
-            _logger.LogError(result.Exception,
-                "Failed to send message from {SenderId} to {RecipientId}. Error: {ErrorMessage}", senderId,
-                request.RecipientId, result.Exception?.Message ?? "Unknown error");
-            if (result.Exception != null)
-                throw result.Exception;
-            throw new HubException("Failed to send message due to an internal error.");
-        }
-
-        var messageResponse = new UserMessageResponse // Assuming a response DTO
-        {
-            MessageId = result.Message.Id,
-            SenderId = result.Message.SenderId,
-            RecipientId = result.Message.RecipientId,
-            RecipientType = result.Message.RecipientType,
-            EncryptedContent = result.Message.EncryptedContent,
-            SentAt = result.Message.SentAt,
-            IsEdited = false,
-            MediaAttachments = result.Message.MediaAttachments.Select(m => m.MediaFile).ToList(),
-            ReplyToMessageId = request.ReplyToMessageId
-        };
-
-        // Notify recipient (user or group)
-        if (request.RecipientType == RecipientType.User)
-        {
-            // Send to the recipient's personal group
-            await Clients.Group(request.RecipientId.ToString()).SendAsync("ReceiveMessage", messageResponse);
-        }
-        else if (request.RecipientType == RecipientType.Group)
-        {
-            // Send to all members of the group, including the sender if they are part of the group via a different connection
-            await Clients.Group($"group_{request.RecipientId}").SendAsync("ReceiveMessage", messageResponse);
-        }
-
-        // Notify sender (confirmation) on their connection
-        await Clients.Caller.SendAsync("MessageSent",
-            messageResponse); // Or use "ReceiveMessage" if the sender should also just get it like anyone else
-
-        _logger.LogInformation(
-            "Message {MessageId} sent successfully from {SenderId} to {RecipientId} ({RecipientType})",
-            result.Message.Id, senderId, request.RecipientId, request.RecipientType);
-    }
-    
-    public async Task Remove(RemoveMessageRequest request)
-    {
-        var removerId = GetUserId();
-        _logger.LogInformation("Message removal initiated by {RemoverId} for message {MessageId}", removerId, request.MessageId);
-        
-        var removeParams = new DeleteMessage(
-            DeleterId: removerId,
-            MessageId: request.MessageId
         );
 
         try
         {
-            var result = await _messageCommandService.DeleteMessageAsync(removeParams);
+            var result = await _messageCommandService.SendMessageAsync(sendMessageParams);
-            if (!result.IsSuccess)
-            {
-                _logger.LogError(result.Exception, "Failed to remove message {MessageId} by {RemoverId}. Error: {ErrorMessage}", request.MessageId, removerId, result.Exception?.Message ?? "Unknown error");
-                if (result.Exception != null) throw result.Exception;
-                throw new HubException("Failed to remove message.");
-            }
 
-            var originalMessage = result.OriginalMessage; // Assuming service returns this
+            if (!result.IsSuccess || result.Message.Id == Guid.Empty)
-             if (originalMessage == null)
+                return LogAndError<UserMessageResponse>(senderId, request.RecipientId, "Failed to send message", result.Exception);
-            {
-                _logger.LogError("DeleteMessageAsync succeeded but did not return the original message details for message {MessageId}", request.MessageId);
-                throw new HubException("Failed to process message deletion due to missing message details.");
-            }
 
-            var removalNotification = new MessageRemovedResponse
+            var response = BuildUserMessageResponse(result.Message, request.ReplyToMessageId);
-            {
-                MessageId = request.MessageId,
-                SenderId = originalMessage.SenderId,
-                RecipientId = originalMessage.RecipientId,
-                RecipientType = originalMessage.RecipientType
-            };
 
-            // Notify relevant clients
+            await NotifyClientsAboutMessage(response);
-            if (originalMessage.RecipientType == RecipientType.User)
+
-            {
+            return HubResult<UserMessageResponse>.Ok(response);
-                await Clients.Group(originalMessage.SenderId.ToString()).SendAsync("MessageRemoved", removalNotification);
-                if (originalMessage.SenderId != originalMessage.RecipientId)
-                {
-                    await Clients.Group(originalMessage.RecipientId.ToString()).SendAsync("MessageRemoved", removalNotification);
-                }
-            }
-            else if (originalMessage.RecipientType == RecipientType.Group)
-            {
-                await Clients.Group($"group_{originalMessage.RecipientId}").SendAsync("MessageRemoved", removalNotification);
-            }
-            _logger.LogInformation("Message {MessageId} removed successfully by {RemoverId}", request.MessageId, removerId);
         }
         catch (UnauthorizedAccessException ex)
         {
-            _logger.LogWarning(ex, "Unauthorized attempt to remove message {MessageId} by user {RemoverId}", request.MessageId, removerId);
+            return LogAndUnauthorized<UserMessageResponse>(ex, "Unauthorized sending attempt", senderId,
-            throw new HubException("You are not authorized to remove this message.", ex);
+                request.RecipientId);
         }
-        catch (KeyNotFoundException ex) // Or a custom NotFoundException
+        catch (FriendshipException)
         {
-            _logger.LogWarning(ex, "Attempt to remove non-existent message {MessageId} by user {RemoverId}", request.MessageId, removerId);
+            return HubResult<UserMessageResponse>.Unauthorized(
-            throw new HubException("Message not found.", ex);
+                "You cannot send this message because you are not friends.");
+        }
+        catch (ArgumentException)
+        {
+            return HubResult<UserMessageResponse>.BadRequest("Invalid message content.");
         }
         catch (Exception ex)
         {
-            _logger.LogError(ex, "Error removing message {MessageId} by {RemoverId}", request.MessageId, removerId);
+            return LogAndError<UserMessageResponse>(senderId, request.RecipientId, "Unhandled exception", ex);
-            throw new HubException("An error occurred while removing the message.", ex);
         }
     }
 
-    public async Task Edit(EditMessageRequest request)
+    
+    public async Task<HubResult<MessageRemovedResponse>> Remove(RemoveMessageRequest request)
     {
+        var removerId = GetUserId();
+        _logger.LogInformation("Removing message {MessageId} by user {RemoverId}", request.MessageId, removerId);
+
+        try
+        {
+            var result = await _messageCommandService.DeleteMessageAsync(new DeleteMessage(removerId, request.MessageId));
+
+            if (!result.IsSuccess || result.OriginalMessage == null)
+                return LogAndError<MessageRemovedResponse>(removerId, request.MessageId, "Message deletion failed", result.Exception);
+
+            var notification = new MessageRemovedResponse
+            {
+                MessageId = request.MessageId,
+                SenderId = result.OriginalMessage.SenderId,
+                RecipientId = result.OriginalMessage.RecipientId,
+                RecipientType = result.OriginalMessage.RecipientType
+            };
+
+            await NotifyClientsAboutRemoval(notification);
+
+            _logger.LogInformation("Message {MessageId} removed successfully by {RemoverId}", request.MessageId, removerId);
+            return HubResult<MessageRemovedResponse>.Ok(notification);
         }
+        catch (UnauthorizedAccessException ex)
+        {
+            return LogAndUnauthorized<MessageRemovedResponse>(ex, "Unauthorized removal", removerId, request.MessageId);
+        }
+        catch (KeyNotFoundException ex)
+        {
+            return LogAndNotFound<MessageRemovedResponse>(ex, "Message not found", removerId, request.MessageId);
+        }
+        catch (Exception ex)
+        {
+            return LogAndError<MessageRemovedResponse>(removerId, request.MessageId, "Unhandled deletion error", ex);
+        }
+    }
+
+
+    public async Task<HubResult<MessageEditResponse>> Edit(EditMessageRequest request)
+    {
+        var editor = GetUserId();
+        _logger.LogInformation("Editing message {MessageId} by user {EditorId}", request.MessageId, editor);
+
+        var editMessageParam = new EditMessage(editor,
+            request.MessageId,
+            request.NewEncryptedContent,
+            DateTime.UtcNow);
+        
+        try
+        {
+            var result = await _messageCommandService.EditMessageAsync(editMessageParam);
+            
+            if(!result.IsSuccess)
+                return LogAndError<MessageEditResponse>(editor, request.MessageId, "Edit message error", result.Exception);
+                
+            return HubResult<MessageEditResponse>.Ok();
+        }
+        catch (Exception ex)
+        {
+            return LogAndError<MessageEditResponse>(editor, request.MessageId, "Unhandled exception error", ex);
+        }
+    }
+
+
+    private UserMessageResponse BuildUserMessageResponse(Message message, Guid? replyToId)
+    {
+        return new UserMessageResponse
+        {
+            MessageId = message.Id,
+            SenderId = message.SenderId,
+            RecipientId = message.RecipientId,
+            RecipientType = message.RecipientType,
+            EncryptedContent = message.EncryptedContent,
+            SentAt = message.SentAt,
+            IsEdited = false,
+            MediaAttachments = message.MediaAttachments.Select(m => m.MediaFile).ToList(),
+            ReplyToMessageId = replyToId
+        };
+    }
+
+    private async Task NotifyClientsAboutMessage(UserMessageResponse response)
+    {
+        string group = response.RecipientType == RecipientType.User
+            ? response.RecipientId.ToString()
+            : $"group_{response.RecipientId}";
+
+        await Clients.Group(group).SendAsync("ReceiveMessage", response);
+        await Clients.Caller.SendAsync("MessageSent", response);
+    }
+
+    private async Task NotifyClientsAboutRemoval(MessageRemovedResponse response)
+    {
+        if (response.RecipientType == RecipientType.User)
+        {
+            await Clients.Group(response.SenderId.ToString()).SendAsync("MessageRemoved", response);
+            if (response.SenderId != response.RecipientId)
+                await Clients.Group(response.RecipientId.ToString()).SendAsync("MessageRemoved", response);
+        }
+        else
+        {
+            await Clients.Group($"group_{response.RecipientId}").SendAsync("MessageRemoved", response);
+        }
+    }
+
+    // Logging helpers
+    private HubResult<T> LogAndError<T>(Guid userId, Guid targetId, string message, Exception ex)
+    {
+        _logger.LogError(ex, "{Message} from {UserId} to {TargetId}", message, userId, targetId);
+        return HubResult<T>.Error(ex?.Message ?? "Internal server error");
+    }
+
+    private HubResult<T> LogAndUnauthorized<T>(Exception ex, string msg, Guid userId, Guid targetId)
+    {
+        _logger.LogWarning(ex, "{Msg}: {UserId} -> {TargetId}", msg, userId, targetId);
+        return HubResult<T>.Unauthorized("You are not authorized to perform this action.");
+    }
+
+    private HubResult<T> LogAndNotFound<T>(Exception ex, string msg, Guid userId, Guid targetId)
+    {
+        _logger.LogWarning(ex, "{Msg}: {UserId} -> {TargetId}", msg, userId, targetId);
+        return HubResult<T>.NotFound("Message not found.");
+    }
+
 
     private Guid GetUserId(bool suppressException = false)
     {

Govor.API/Program.cs

@@ -121,7 +121,7 @@ app.UseAuthorization();
 
 app.MapControllers();
 
-app.MapHub<ChatsHub>("/hubs/friends"); 
+app.MapHub<ChatsHub>("/hubs/chats"); 
 app.MapHub<FriendsHub>("/hubs/friends");
 
 app.MapSwagger().RequireAuthorization();

Govor.Application.Tests/Services/Authentication/AuthServiceTests.cs

@@ -7,6 +7,7 @@ using Govor.Application.Exceptions.AuthService;
 using Govor.Application.Interfaces.Authentication;
 using Govor.Application.Services;
 using Govor.Application.Services.Authentication;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Admins;
 using Moq;
 

Govor.Application.Tests/Services/Authentication/JwtServiceTests.cs

@@ -1,9 +1,9 @@
 using System.IdentityModel.Tokens.Jwt;
 using AutoFixture;
 using Govor.API.Services.Authentication.Interfaces;
-using Govor.Application.Services;
+using Govor.Application.Interfaces.Authentication;
 using Govor.Application.Services.Authentication;
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Microsoft.Extensions.Options;
 using Moq;
 

Govor.Application.Tests/Services/Friends/FriendRequestQueryServiceTests.cs

@@ -2,6 +2,7 @@ using AutoFixture;
 using Govor.Application.Interfaces.Friends;
 using Govor.Application.Services.Friends;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Friendships;
 using Govor.Core.Repositories.Users;
 using Govor.Data.Repositories.Exceptions;

Govor.Application.Tests/Services/Friends/FriendshipServiceTests.cs

@@ -3,6 +3,7 @@ using Govor.Application.Exceptions.FriendsService;
 using Govor.Application.Interfaces.Friends;
 using Govor.Application.Services.Friends;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Friendships;
 using Govor.Core.Repositories.Users;
 using Govor.Data.Repositories.Exceptions;

Govor.Application.Tests/Services/Messages/MessageCommandServiceTests.cs

@@ -3,6 +3,7 @@ using Govor.Application.Interfaces;
 using Govor.Application.Interfaces.Messages.Parameters;
 using Govor.Application.Services.Messages;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Core.Repositories.Groups;
 using Govor.Core.Repositories.Messages;
 using Govor.Core.Repositories.PrivateChats;
@@ -296,7 +297,7 @@ public class MessageCommandServiceTests
         Assert.That(result, Is.Not.Null);
         Assert.That(result.IsSuccess, Is.False);
         Assert.That(result.Exception, Is.Not.Null);
-        Assert.That(result.Exception,Is.TypeOf<NotFoundByKeyException<Guid>>());
+        Assert.That(result.Exception,Is.TypeOf<KeyNotFoundException>());
         Assert.That(result.OriginalMessage, Is.Null);
     }
 

Govor.Application.Tests/Services/PingHandlerServiceTests.cs

@@ -1,5 +1,5 @@
 using Govor.Application.Services;
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Data;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Caching.Memory;

Govor.Application/Infrastructure/AdminsStuff/UsersService.cs

@@ -1,9 +1,9 @@
-using Govor.API.Services.AdminsStuff.Interfaces;
+using Govor.Application.Interfaces;
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Users;
 using Govor.Data.Repositories.Exceptions;
 
-namespace Govor.Application.Interfaces.AdminsStuff;
+namespace Govor.Application.Infrastructure.AdminsStuff;
 
 public class UsersService : IUsersAdministration
 {

Govor.Application/Interfaces/Authentication/IInvitesService.cs

@@ -1,4 +1,5 @@
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.API.Services.Authentication.Interfaces;
 

Govor.Application/Interfaces/Authentication/IJwtService.cs

@@ -1,6 +1,6 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
-namespace Govor.API.Services.Authentication.Interfaces;
+namespace Govor.Application.Interfaces.Authentication;
 
 public interface IJwtService
 {

Govor.Application/Interfaces/Friends/IFriendshipService.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Application.Interfaces.Friends;
 

Govor.Application/Interfaces/IFriendsService.cs

@@ -1,4 +1,5 @@
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Application.Interfaces;
 

Govor.Application/Interfaces/IGroupService.cs

@@ -1,5 +1,6 @@
 using Govor.Application.Interfaces.Messages.Parameters;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Application.Interfaces;
 

Govor.Application/Interfaces/IMessagesLoader.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Application.Interfaces;
 

Govor.Application/Interfaces/IUsersAdministration.cs

@@ -1,6 +1,6 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
-namespace Govor.API.Services.AdminsStuff.Interfaces;
+namespace Govor.Application.Interfaces;
 
 public interface IUsersAdministration
 {

Govor.Application/Interfaces/Medias/IMediaService.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Application.Interfaces.Medias;
 

Govor.Application/Interfaces/Messages/IMessageCommandService.cs

@@ -1,5 +1,5 @@
 using Govor.Application.Interfaces.Messages.Parameters;
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Application.Interfaces.Messages;
 

Govor.Application/Interfaces/Messages/Parameters/SendMessage.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Application.Interfaces.Messages.Parameters;
 

Govor.Application/Services/Authentication/AuthService.cs

@@ -1,11 +1,9 @@
-using System.Text.RegularExpressions;
-using Govor.API.Services.Authentication.Interfaces;
 using Govor.Application.Exceptions.AuthService;
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Models;
 using Govor.Core.Repositories.Users;
 using Govor.Application.Interfaces.Authentication;
-using Govor.Core.Infrastructure.Validators;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Admins;
 
 namespace Govor.Application.Services.Authentication;

Govor.Application/Services/Authentication/JwtService.cs

@@ -2,7 +2,8 @@ using System.IdentityModel.Tokens.Jwt;
 using System.Security.Claims;
 using System.Text;
 using Govor.API.Services.Authentication.Interfaces;
-using Govor.Core.Models;
+using Govor.Application.Interfaces.Authentication;
+using Govor.Core.Models.Users;
 using Microsoft.Extensions.Options;
 using Microsoft.IdentityModel.Tokens;
 

Govor.Application/Services/Friends/FriendshipService.cs

@@ -1,6 +1,7 @@
 using Govor.Application.Exceptions.FriendsService;
 using Govor.Application.Interfaces.Friends;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Friendships;
 using Govor.Core.Repositories.Users;
 using Govor.Data.Repositories.Exceptions;

Govor.Application/Services/InvitesService.cs

@@ -1,6 +1,7 @@
 using Govor.API.Services.Authentication.Interfaces;
 using Govor.Application.Exceptions.InvitesService;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Invaites;
 using Govor.Data.Repositories.Exceptions;
 

Govor.Application/Services/Messages/MessageCommandService.cs

@@ -2,10 +2,12 @@ using Govor.Application.Interfaces;
 using Govor.Application.Interfaces.Messages;
 using Govor.Application.Interfaces.Messages.Parameters;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Core.Repositories.Groups;
 using Govor.Core.Repositories.Messages;
 using Govor.Core.Repositories.PrivateChats;
 using Govor.Core.Repositories.Users;
+using Govor.Data.Repositories.Exceptions;
 using Microsoft.Extensions.Logging;
 
 namespace Govor.Application.Services.Messages;
@@ -183,8 +185,11 @@ public class MessageCommandService : IMessageCommandService
                 //        return new DeleteMessageResult(false, new UnauthorizedAccessException("User is not authorized to delete this message."), null);
                 //    }
                 // } else {
-                   _logger.LogWarning("User {DeleterId} attempted to delete message {MessageId} not sent by them (sender was {SenderId})", deleteParams.DeleterId, deleteParams.MessageId, message.SenderId);
+                _logger.LogWarning(
-                   return new DeleteMessageResult(false, new UnauthorizedAccessException("User is not authorized to delete this message."), default);
+                    "User {DeleterId} attempted to delete message {MessageId} not sent by them (sender was {SenderId})",
+                    deleteParams.DeleterId, deleteParams.MessageId, message.SenderId);
+                return new DeleteMessageResult(false,
+                    new UnauthorizedAccessException("User is not authorized to delete this message."), default);
                 // }
             }
 
@@ -198,9 +203,14 @@ public class MessageCommandService : IMessageCommandService
 
             await _messagesRepository.RemoveAsync(deleteParams.MessageId);
 
-            _logger.LogInformation("Message {MessageId} deleted successfully by user {DeleterId}", deleteParams.MessageId, deleteParams.DeleterId);
+            _logger.LogInformation("Message {MessageId} deleted successfully by user {DeleterId}",
+                deleteParams.MessageId, deleteParams.DeleterId);
             return new DeleteMessageResult(true, default, originalMessageForNotification);
         }
+        catch (NotFoundByKeyException<Guid> ex)
+        {
+            return new DeleteMessageResult(false, new KeyNotFoundException("Message not found", ex), default);
+        }
         catch (Exception ex)
         {
             _logger.LogError(ex, "Error deleting message {MessageId} by user {DeleterId}", deleteParams.MessageId, deleteParams.DeleterId);

Govor.Application/Services/Messages/MessagesLoader.cs

@@ -1,5 +1,5 @@
 using Govor.Application.Interfaces;
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Core.Repositories.Groups;
 using Govor.Core.Repositories.Messages;
 using Govor.Data.Repositories.Exceptions;

Govor.Console/Program.cs

@@ -10,6 +10,7 @@ using Govor.ConsoleClient.Commands;
 using Govor.Contracts.Requests.SignalR;
 using Govor.Contracts.Responses.SignalR;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 /*====================================
  *Личные сообщения| Егор 

Govor.Contracts/Requests/MediaUploadRequest.cs

@@ -1,5 +1,4 @@
-using System.ComponentModel.DataAnnotations;
+using Govor.Core.Models.Messages;
-using Govor.Core.Models;
 
 namespace Govor.Contracts.Requests;
 

Govor.Contracts/Requests/SignalR/MessageRequest.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Contracts.Requests.SignalR;
 

Govor.Contracts/Responses/SignalR/MessageEditResponse.cs

@@ -0,0 +1,6 @@
+namespace Govor.Contracts.Responses.SignalR;
+
+public class MessageEditResponse
+{
+    
+}

Govor.Contracts/Responses/SignalR/MessageRemovedResponse.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Contracts.Responses.SignalR;
 

Govor.Contracts/Responses/SignalR/UserMessageResponse.cs

@@ -1,5 +1,5 @@
-using Govor.Contracts.Requests.SignalR;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Contracts.Responses.SignalR;
 

Govor.Core/Infrastructure/Validators/AdminValidator.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Infrastructure.Validators;
 

Govor.Core/Infrastructure/Validators/MediaAttachmentsValidator.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Exception = System.Exception;
 
 namespace Govor.Core.Infrastructure.Validators;

Govor.Core/Infrastructure/Validators/MessageValidator.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Infrastructure.Validators;
 

Govor.Core/Infrastructure/Validators/UserValidator.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using ArgumentNullException = System.ArgumentNullException;
 
 namespace Govor.Core.Infrastructure.Validators;

Govor.Core/Models/Friendship.cs

@@ -1,3 +1,5 @@
+using Govor.Core.Models.Users;
+
 namespace Govor.Core.Models;
 
 public class Friendship

Govor.Core/Models/GroupInvitation.cs

@@ -1,3 +1,5 @@
+using Govor.Core.Models.Users;
+
 namespace Govor.Core.Models;
 
 public class GroupInvitation

Govor.Core/Models/Invitation.cs

@@ -1,3 +1,5 @@
+using Govor.Core.Models.Users;
+
 namespace Govor.Core.Models;
 
 public class Invitation

Govor.Core/Models/MediaFile.cs

@@ -1,3 +1,5 @@
+using Govor.Core.Models.Messages;
+
 namespace Govor.Core.Models;
 
 public class MediaFile

Govor.Core/Models/Messages/MediaAttachments.cs

@@ -1,4 +1,4 @@
-namespace Govor.Core.Models;
+namespace Govor.Core.Models.Messages;
 
 public class MediaAttachments
 {

Govor.Core/Models/Messages/Message.cs

@@ -1,4 +1,4 @@
-namespace Govor.Core.Models;
+namespace Govor.Core.Models.Messages;
 
 public class Message
 {

Govor.Core/Models/Messages/MessageReaction.cs

@@ -1,4 +1,6 @@
-namespace Govor.Core.Models;
+using Govor.Core.Models.Users;
+
+namespace Govor.Core.Models.Messages;
 
 public class MessageReaction
 {

Govor.Core/Models/Messages/MessageView.cs

@@ -1,4 +1,4 @@
-namespace Govor.Core.Models;
+namespace Govor.Core.Models.Messages;
 
 public class MessageView
 {

Govor.Core/Models/PrivateChat.cs

@@ -1,3 +1,5 @@
+using Govor.Core.Models.Messages;
+
 namespace Govor.Core.Models;
 
 public class PrivateChat

Govor.Core/Models/Users/Admin.cs

@@ -1,6 +1,6 @@
 using System.ComponentModel.DataAnnotations;
 
-namespace Govor.Core.Models;
+namespace Govor.Core.Models.Users;
 
 public class Admin
 {

Govor.Core/Models/Users/PrivacyUserSettings.cs

@@ -0,0 +1,39 @@
+namespace Govor.Core.Models.Users;
+
+public class PrivacyUserSettings
+{
+    public Guid UserId { get; set; }
+    public bool IsGlobalAccount  { get; set; }   
+    
+    public WhoCan CanSend { get; set; }
+    public List<Guid>? WhitelistSent { get; set; }
+    public List<Guid>? BlacklistSent { get; set; }
+    
+    public WhoCan CanSeeTimeWas  { get; set; }
+    public List<Guid>? WhitelistTimeWas { get; set; }
+    public List<Guid>? BlacklistTimeWas { get; set; }
+    
+    public WhoCan CanSeeImage  { get; set; }
+    public List<Guid>? WhitelistSeeImage { get; set; }
+    public List<Guid>? BlacklistSeeImage{ get; set; }
+    
+    public DeletingMessagesVia Via { get; set; } // if min value = none 
+    public int DeletingIn { get; set; }
+}
+
+public enum WhoCan
+{
+    None = 0,
+    OnlyFriends = 1,
+    EveryoneCanSend = 2,
+}
+
+public enum DeletingMessagesVia
+{
+    None = 0,
+    Hours = 1,
+    Days = 2,
+    Months = 3,
+    Years = 4
+}
+

Govor.Core/Models/Users/User.cs

@@ -1,6 +1,6 @@
 using System.ComponentModel.DataAnnotations;
 
-namespace Govor.Core.Models;
+namespace Govor.Core.Models.Users;
 
 public class User
 {

Govor.Core/Repositories/Admins/IAdminsExist.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Repositories.Admins;
 

Govor.Core/Repositories/Admins/IAdminsReader.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Repositories.Admins;
 

Govor.Core/Repositories/Admins/IAdminsWriter.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Repositories.Admins;
 

Govor.Core/Repositories/Groups/IGroupMessagesReader.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.Groups;
 

Govor.Core/Repositories/MediasAttachments/IMediaAttachmentsExist.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.MediasAttachments;
 

Govor.Core/Repositories/MediasAttachments/IMediaAttachmentsReader.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.MediasAttachments;
 

Govor.Core/Repositories/MediasAttachments/IMediaAttachmentsWriter.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.MediasAttachments;
 

Govor.Core/Repositories/Messages/IMessagesExist.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.Messages;
 

Govor.Core/Repositories/Messages/IMessagesReader.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.Messages;
 

Govor.Core/Repositories/Messages/IMessagesWriter.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 
 namespace Govor.Core.Repositories.Messages;
 

Govor.Core/Repositories/Users/IUsersExist.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Repositories.Users;
 

Govor.Core/Repositories/Users/IUsersReader.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Repositories.Users;
 

Govor.Core/Repositories/Users/IUsersWriter.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 
 namespace Govor.Core.Repositories.Users;
 

Govor.Data.Tests/Repositories/AdminsRepositoryTests.cs

@@ -1,6 +1,7 @@
 using AutoFixture;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Data;
 using Govor.Data.Repositories;
 using Govor.Data.Repositories.Exceptions;

Govor.Data.Tests/Repositories/MediaAttachmentsTests.cs

@@ -1,6 +1,7 @@
 using AutoFixture;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Data;
 using Govor.Data.Repositories;
 using Govor.Data.Repositories.Exceptions;

Govor.Data.Tests/Repositories/MessagesRepositoryTests.cs

@@ -1,6 +1,7 @@
 using AutoFixture;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Data;
 using Govor.Data.Repositories;
 using Govor.Data.Repositories.Exceptions;

Govor.Data.Tests/Repositories/UsersRepositoryTests.cs

@@ -1,6 +1,7 @@
 using AutoFixture;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Data;
 using Govor.Data.Repositories;
 using Govor.Data.Repositories.Exceptions;

Govor.Data/Configurations/AdminConfiguration.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Govor.Data/Configurations/MediaAttachmentsConfiguration.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Govor.Data/Configurations/MessageReactionConfiguration.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Govor.Data/Configurations/MessageViewConfiguration.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Govor.Data/Configurations/MessagesConfiguration.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Govor.Data/Configurations/UserConfiguration.cs

@@ -1,4 +1,4 @@
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Govor.Data/GovorDbContext.cs

@@ -1,5 +1,7 @@
 using System.Text.RegularExpressions;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
+using Govor.Core.Models.Users;
 using Govor.Data.Configurations;
 using Microsoft.EntityFrameworkCore;
 

Govor.Data/Repositories/AdminsRepository.cs

@@ -1,6 +1,7 @@
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Admins;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;

Govor.Data/Repositories/FriendshipsRepository.cs

@@ -1,6 +1,7 @@
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Friendships;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;

Govor.Data/Repositories/GroupRepository.cs

@@ -1,7 +1,7 @@
-using System.Text.RegularExpressions;
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Groups;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;

Govor.Data/Repositories/InvitesRepository.cs

@@ -1,6 +1,7 @@
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Users;
 using Govor.Core.Repositories.Invaites;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;

Govor.Data/Repositories/MediaAttachmentsRepository.cs

@@ -1,6 +1,6 @@
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
-using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Core.Repositories.MediasAttachments;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;

Govor.Data/Repositories/MessagesRepository.cs

@@ -1,6 +1,7 @@
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
 using Govor.Core.Models;
+using Govor.Core.Models.Messages;
 using Govor.Core.Repositories.Messages;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;

Govor.Data/Repositories/UsersRepository.cs

@@ -1,7 +1,6 @@
 using Govor.Core.Infrastructure.Extensions;
 using Govor.Core.Infrastructure.Validators;
-using Govor.Core.Models;
+using Govor.Core.Models.Users;
-using Govor.Core.Repositories;
 using Govor.Core.Repositories.Users;
 using Govor.Data.Repositories.Exceptions;
 using Microsoft.EntityFrameworkCore;