using OnlineMsgServer.Core;
using WebSocketSharp.Server;

namespace OnlineMsgServer.Common
{
    class ForwardMessage : Message
    {
        public ForwardMessage()
        {
            Type = "forward";
            //收到客户端消息，并执行转发
        }
        public override Task Handler(string wsid, WebSocketSessionManager Sessions)
        {
            return Task.Run(() =>
            {
                try
                {
                    if (!UserService.IsAuthenticated(wsid))
                    {
                        Log.Security("forward_denied_unauthenticated", $"wsid={wsid}");
                        return;
                    }

                    if (string.IsNullOrWhiteSpace(Key))
                    {
                        Log.Security("forward_invalid_target", $"wsid={wsid}");
                        return;
                    }
                    string forwardPublickKey = Key.Trim();

                    if (!SignedMessagePayload.TryParse(Data, out SignedMessagePayload payload, out string parseError))
                    {
                        Log.Security("forward_payload_invalid", $"wsid={wsid} reason={parseError}");
                        return;
                    }

                    if (!SecurityValidator.VerifySignedMessage(wsid, Type, forwardPublickKey, payload, out string securityReason))
                    {
                        Log.Security("forward_security_failed", $"wsid={wsid} reason={securityReason}");
                        return;
                    }

                    string fromPublicKey = UserService.GetUserPublicKeyByID(wsid)!;

                    Message response = new()
                    {
                        Type = "forward",
                        Data = payload.Payload,
                        Key = fromPublicKey,
                    };

                    string jsonString = response.ToJsonString();
                    string encryptString = RsaService.EncryptForClient(forwardPublickKey, jsonString);

                    List<User> userList = UserService.GetUserListByPublicKey(forwardPublickKey);
                    if (userList.Count == 0)
                    {
                        Log.Security("forward_target_offline_or_untrusted", $"wsid={wsid}");
                        return;
                    }

                    foreach (IWebSocketSession session in Sessions.Sessions)
                    {
                        if (userList.Exists(u => u.ID == session.ID))
                        {
                            session.Context.WebSocket.Send(encryptString);
                            break;
                        }
                    }
                }
                catch (Exception ex)
                {
                    Log.Security("forward_error", $"wsid={wsid} error={ex.Message}");
                }
            });
        }
    }
}