SessionWebSocketListener.cs 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. using System;
  2. using System.Threading.Tasks;
  3. using MediaBrowser.Controller.Net;
  4. using MediaBrowser.Controller.Session;
  5. using MediaBrowser.Model.Events;
  6. using MediaBrowser.Model.Serialization;
  7. using Microsoft.AspNetCore.Http;
  8. using Microsoft.Extensions.Logging;
  9. namespace Emby.Server.Implementations.Session
  10. {
  11. /// <summary>
  12. /// Class SessionWebSocketListener
  13. /// </summary>
  14. public class SessionWebSocketListener : IWebSocketListener, IDisposable
  15. {
  16. /// <summary>
  17. /// The _session manager
  18. /// </summary>
  19. private readonly ISessionManager _sessionManager;
  20. /// <summary>
  21. /// The _logger
  22. /// </summary>
  23. private readonly ILogger _logger;
  24. /// <summary>
  25. /// The _dto service
  26. /// </summary>
  27. private readonly IJsonSerializer _json;
  28. private readonly IHttpServer _httpServer;
  29. /// <summary>
  30. /// Initializes a new instance of the <see cref="SessionWebSocketListener" /> class.
  31. /// </summary>
  32. /// <param name="sessionManager">The session manager.</param>
  33. /// <param name="loggerFactory">The logger factory.</param>
  34. /// <param name="json">The json.</param>
  35. /// <param name="httpServer">The HTTP server.</param>
  36. public SessionWebSocketListener(ISessionManager sessionManager, ILoggerFactory loggerFactory, IJsonSerializer json, IHttpServer httpServer)
  37. {
  38. _sessionManager = sessionManager;
  39. _logger = loggerFactory.CreateLogger(GetType().Name);
  40. _json = json;
  41. _httpServer = httpServer;
  42. httpServer.WebSocketConnected += _serverManager_WebSocketConnected;
  43. }
  44. void _serverManager_WebSocketConnected(object sender, GenericEventArgs<IWebSocketConnection> e)
  45. {
  46. var session = GetSession(e.Argument.QueryString, e.Argument.RemoteEndPoint);
  47. if (session != null)
  48. {
  49. EnsureController(session, e.Argument);
  50. }
  51. else
  52. {
  53. _logger.LogWarning("Unable to determine session based on url: {0}", e.Argument.Url);
  54. }
  55. }
  56. private SessionInfo GetSession(IQueryCollection queryString, string remoteEndpoint)
  57. {
  58. if (queryString == null)
  59. {
  60. return null;
  61. }
  62. var token = queryString["api_key"];
  63. if (string.IsNullOrWhiteSpace(token))
  64. {
  65. return null;
  66. }
  67. var deviceId = queryString["deviceId"];
  68. return _sessionManager.GetSessionByAuthenticationToken(token, deviceId, remoteEndpoint);
  69. }
  70. public void Dispose()
  71. {
  72. _httpServer.WebSocketConnected -= _serverManager_WebSocketConnected;
  73. }
  74. /// <summary>
  75. /// Processes the message.
  76. /// </summary>
  77. /// <param name="message">The message.</param>
  78. /// <returns>Task.</returns>
  79. public Task ProcessMessageAsync(WebSocketMessageInfo message)
  80. => Task.CompletedTask;
  81. private void EnsureController(SessionInfo session, IWebSocketConnection connection)
  82. {
  83. var controllerInfo = session.EnsureController<WebSocketController>(s => new WebSocketController(s, _logger, _sessionManager));
  84. var controller = (WebSocketController)controllerInfo.Item1;
  85. controller.AddWebSocket(connection);
  86. }
  87. }
  88. }