using Fleck; using System; using System.Collections.Generic; using System.Net; namespace SharpChat { public class ChatUserSession : IDisposable { public const int ID_LENGTH = 32; #if DEBUG public static TimeSpan SessionTimeOut { get; } = TimeSpan.FromMinutes(1); #else public static TimeSpan SessionTimeOut { get; } = TimeSpan.FromMinutes(5); #endif public IWebSocketConnection Connection { get; } public string Id { get; private set; } public bool IsDisposed { get; private set; } public DateTimeOffset LastPing { get; set; } = DateTimeOffset.MinValue; public ChatUser User { get; set; } private int CloseCode { get; set; } = 1000; private IPAddress _RemoteAddress = null; public IPAddress RemoteAddress { get { if(_RemoteAddress == null) { if((Connection.ConnectionInfo.ClientIpAddress == "127.0.0.1" || Connection.ConnectionInfo.ClientIpAddress == "::1") && Connection.ConnectionInfo.Headers.ContainsKey("X-Real-IP")) _RemoteAddress = IPAddress.Parse(Connection.ConnectionInfo.Headers["X-Real-IP"]); else _RemoteAddress = IPAddress.Parse(Connection.ConnectionInfo.ClientIpAddress); } return _RemoteAddress; } } public ChatUserSession(IWebSocketConnection ws) { Connection = ws; Id = RNG.SecureRandomString(ID_LENGTH); } public void Send(IServerPacket packet) { if(!Connection.IsAvailable) return; IEnumerable data = packet.Pack(); if(data != null) foreach(string line in data) if(!string.IsNullOrWhiteSpace(line)) Connection.Send(line); } public void BumpPing() { LastPing = DateTimeOffset.Now; } public bool HasTimedOut => DateTimeOffset.Now - LastPing > SessionTimeOut; public void PrepareForRestart() { CloseCode = 1012; } ~ChatUserSession() { DoDispose(); } public void Dispose() { DoDispose(); GC.SuppressFinalize(this); } private void DoDispose() { if(IsDisposed) return; IsDisposed = true; Connection.Close(CloseCode); } } }