SessionWebSocketListener.cs 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479
  1. using MediaBrowser.Controller.Net;
  2. using MediaBrowser.Controller.Session;
  3. using MediaBrowser.Model.Events;
  4. using MediaBrowser.Model.Logging;
  5. using MediaBrowser.Model.Serialization;
  6. using MediaBrowser.Model.Session;
  7. using System;
  8. using System.Collections.Specialized;
  9. using System.Globalization;
  10. using System.Linq;
  11. using System.Threading.Tasks;
  12. namespace MediaBrowser.Server.Implementations.Session
  13. {
  14. /// <summary>
  15. /// Class SessionWebSocketListener
  16. /// </summary>
  17. public class SessionWebSocketListener : IWebSocketListener, IDisposable
  18. {
  19. /// <summary>
  20. /// The _true task result
  21. /// </summary>
  22. private readonly Task _trueTaskResult = Task.FromResult(true);
  23. /// <summary>
  24. /// The _session manager
  25. /// </summary>
  26. private readonly ISessionManager _sessionManager;
  27. /// <summary>
  28. /// The _logger
  29. /// </summary>
  30. private readonly ILogger _logger;
  31. /// <summary>
  32. /// The _dto service
  33. /// </summary>
  34. private readonly IJsonSerializer _json;
  35. private readonly IHttpServer _httpServer;
  36. private readonly IServerManager _serverManager;
  37. /// <summary>
  38. /// Initializes a new instance of the <see cref="SessionWebSocketListener" /> class.
  39. /// </summary>
  40. /// <param name="sessionManager">The session manager.</param>
  41. /// <param name="logManager">The log manager.</param>
  42. /// <param name="json">The json.</param>
  43. /// <param name="httpServer">The HTTP server.</param>
  44. /// <param name="serverManager">The server manager.</param>
  45. public SessionWebSocketListener(ISessionManager sessionManager, ILogManager logManager, IJsonSerializer json, IHttpServer httpServer, IServerManager serverManager)
  46. {
  47. _sessionManager = sessionManager;
  48. _logger = logManager.GetLogger(GetType().Name);
  49. _json = json;
  50. _httpServer = httpServer;
  51. _serverManager = serverManager;
  52. httpServer.WebSocketConnecting += _httpServer_WebSocketConnecting;
  53. serverManager.WebSocketConnected += _serverManager_WebSocketConnected;
  54. }
  55. async void _serverManager_WebSocketConnected(object sender, GenericEventArgs<IWebSocketConnection> e)
  56. {
  57. var session = await GetSession(e.Argument.QueryString, e.Argument.RemoteEndPoint).ConfigureAwait(false);
  58. if (session != null)
  59. {
  60. var controller = session.SessionController as WebSocketController;
  61. if (controller == null)
  62. {
  63. controller = new WebSocketController(session, _logger, _sessionManager);
  64. }
  65. controller.AddWebSocket(e.Argument);
  66. session.SessionController = controller;
  67. }
  68. else
  69. {
  70. _logger.Warn("Unable to determine session based on url: {0}", e.Argument.Url);
  71. }
  72. }
  73. async void _httpServer_WebSocketConnecting(object sender, WebSocketConnectingEventArgs e)
  74. {
  75. //var token = e.QueryString["api_key"];
  76. //if (!string.IsNullOrWhiteSpace(token))
  77. //{
  78. // try
  79. // {
  80. // var session = await GetSession(e.QueryString, e.Endpoint).ConfigureAwait(false);
  81. // if (session == null)
  82. // {
  83. // e.AllowConnection = false;
  84. // }
  85. // }
  86. // catch (Exception ex)
  87. // {
  88. // _logger.ErrorException("Error getting session info", ex);
  89. // }
  90. //}
  91. }
  92. private Task<SessionInfo> GetSession(NameValueCollection queryString, string remoteEndpoint)
  93. {
  94. if (queryString == null)
  95. {
  96. throw new ArgumentNullException("queryString");
  97. }
  98. var token = queryString["api_key"];
  99. if (string.IsNullOrWhiteSpace(token))
  100. {
  101. return Task.FromResult<SessionInfo>(null);
  102. }
  103. var deviceId = queryString["deviceId"];
  104. return _sessionManager.GetSessionByAuthenticationToken(token, deviceId, remoteEndpoint);
  105. }
  106. public void Dispose()
  107. {
  108. _httpServer.WebSocketConnecting -= _httpServer_WebSocketConnecting;
  109. _serverManager.WebSocketConnected -= _serverManager_WebSocketConnected;
  110. }
  111. /// <summary>
  112. /// Processes the message.
  113. /// </summary>
  114. /// <param name="message">The message.</param>
  115. /// <returns>Task.</returns>
  116. public Task ProcessMessage(WebSocketMessageInfo message)
  117. {
  118. if (string.Equals(message.MessageType, "Identity", StringComparison.OrdinalIgnoreCase))
  119. {
  120. ProcessIdentityMessage(message);
  121. }
  122. else if (string.Equals(message.MessageType, "Context", StringComparison.OrdinalIgnoreCase))
  123. {
  124. ProcessContextMessage(message);
  125. }
  126. else if (string.Equals(message.MessageType, "PlaybackStart", StringComparison.OrdinalIgnoreCase))
  127. {
  128. OnPlaybackStart(message);
  129. }
  130. else if (string.Equals(message.MessageType, "PlaybackProgress", StringComparison.OrdinalIgnoreCase))
  131. {
  132. OnPlaybackProgress(message);
  133. }
  134. else if (string.Equals(message.MessageType, "PlaybackStopped", StringComparison.OrdinalIgnoreCase))
  135. {
  136. OnPlaybackStopped(message);
  137. }
  138. else if (string.Equals(message.MessageType, "ReportPlaybackStart", StringComparison.OrdinalIgnoreCase))
  139. {
  140. ReportPlaybackStart(message);
  141. }
  142. else if (string.Equals(message.MessageType, "ReportPlaybackProgress", StringComparison.OrdinalIgnoreCase))
  143. {
  144. ReportPlaybackProgress(message);
  145. }
  146. else if (string.Equals(message.MessageType, "ReportPlaybackStopped", StringComparison.OrdinalIgnoreCase))
  147. {
  148. ReportPlaybackStopped(message);
  149. }
  150. return _trueTaskResult;
  151. }
  152. /// <summary>
  153. /// Processes the identity message.
  154. /// </summary>
  155. /// <param name="message">The message.</param>
  156. private async void ProcessIdentityMessage(WebSocketMessageInfo message)
  157. {
  158. _logger.Debug("Received Identity message: " + message.Data);
  159. var vals = message.Data.Split('|');
  160. if (vals.Length < 3)
  161. {
  162. _logger.Error("Client sent invalid identity message.");
  163. return;
  164. }
  165. var client = vals[0];
  166. var deviceId = vals[1];
  167. var version = vals[2];
  168. var deviceName = vals.Length > 3 ? vals[3] : string.Empty;
  169. var session = _sessionManager.GetSession(deviceId, client, version);
  170. if (session == null && !string.IsNullOrEmpty(deviceName))
  171. {
  172. _logger.Debug("Logging session activity");
  173. session = await _sessionManager.LogSessionActivity(client, version, deviceId, deviceName, message.Connection.RemoteEndPoint, null).ConfigureAwait(false);
  174. }
  175. if (session != null)
  176. {
  177. var controller = session.SessionController as WebSocketController;
  178. if (controller == null)
  179. {
  180. controller = new WebSocketController(session, _logger, _sessionManager);
  181. }
  182. controller.AddWebSocket(message.Connection);
  183. session.SessionController = controller;
  184. }
  185. else
  186. {
  187. _logger.Warn("Unable to determine session based on identity message: {0}", message.Data);
  188. }
  189. }
  190. /// <summary>
  191. /// Processes the context message.
  192. /// </summary>
  193. /// <param name="message">The message.</param>
  194. private void ProcessContextMessage(WebSocketMessageInfo message)
  195. {
  196. var session = GetSessionFromMessage(message);
  197. if (session != null)
  198. {
  199. var vals = message.Data.Split('|');
  200. _sessionManager.ReportNowViewingItem(session.Id, vals[1]);
  201. }
  202. }
  203. /// <summary>
  204. /// Gets the session from message.
  205. /// </summary>
  206. /// <param name="message">The message.</param>
  207. /// <returns>SessionInfo.</returns>
  208. private SessionInfo GetSessionFromMessage(WebSocketMessageInfo message)
  209. {
  210. var result = _sessionManager.Sessions.FirstOrDefault(i =>
  211. {
  212. var controller = i.SessionController as WebSocketController;
  213. if (controller != null)
  214. {
  215. if (controller.Sockets.Any(s => s.Id == message.Connection.Id))
  216. {
  217. return true;
  218. }
  219. }
  220. return false;
  221. });
  222. if (result == null)
  223. {
  224. _logger.Error("Unable to find session based on web socket message");
  225. }
  226. return result;
  227. }
  228. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  229. /// <summary>
  230. /// Reports the playback start.
  231. /// </summary>
  232. /// <param name="message">The message.</param>
  233. private void OnPlaybackStart(WebSocketMessageInfo message)
  234. {
  235. _logger.Debug("Received PlaybackStart message");
  236. var session = GetSessionFromMessage(message);
  237. if (session != null && session.UserId.HasValue)
  238. {
  239. var vals = message.Data.Split('|');
  240. var itemId = vals[0];
  241. var queueableMediaTypes = string.Empty;
  242. var canSeek = true;
  243. if (vals.Length > 1)
  244. {
  245. canSeek = string.Equals(vals[1], "true", StringComparison.OrdinalIgnoreCase);
  246. }
  247. if (vals.Length > 2)
  248. {
  249. queueableMediaTypes = vals[2];
  250. }
  251. var info = new PlaybackStartInfo
  252. {
  253. CanSeek = canSeek,
  254. ItemId = itemId,
  255. SessionId = session.Id,
  256. QueueableMediaTypes = queueableMediaTypes.Split(',').ToList()
  257. };
  258. if (vals.Length > 3)
  259. {
  260. info.MediaSourceId = vals[3];
  261. }
  262. if (vals.Length > 4 && !string.IsNullOrWhiteSpace(vals[4]))
  263. {
  264. info.AudioStreamIndex = int.Parse(vals[4], _usCulture);
  265. }
  266. if (vals.Length > 5 && !string.IsNullOrWhiteSpace(vals[5]))
  267. {
  268. info.SubtitleStreamIndex = int.Parse(vals[5], _usCulture);
  269. }
  270. _sessionManager.OnPlaybackStart(info);
  271. }
  272. }
  273. private void ReportPlaybackStart(WebSocketMessageInfo message)
  274. {
  275. _logger.Debug("Received ReportPlaybackStart message");
  276. var session = GetSessionFromMessage(message);
  277. if (session != null && session.UserId.HasValue)
  278. {
  279. var info = _json.DeserializeFromString<PlaybackStartInfo>(message.Data);
  280. info.SessionId = session.Id;
  281. _sessionManager.OnPlaybackStart(info);
  282. }
  283. }
  284. private void ReportPlaybackProgress(WebSocketMessageInfo message)
  285. {
  286. //_logger.Debug("Received ReportPlaybackProgress message");
  287. var session = GetSessionFromMessage(message);
  288. if (session != null && session.UserId.HasValue)
  289. {
  290. var info = _json.DeserializeFromString<PlaybackProgressInfo>(message.Data);
  291. info.SessionId = session.Id;
  292. _sessionManager.OnPlaybackProgress(info);
  293. }
  294. }
  295. /// <summary>
  296. /// Reports the playback progress.
  297. /// </summary>
  298. /// <param name="message">The message.</param>
  299. private void OnPlaybackProgress(WebSocketMessageInfo message)
  300. {
  301. var session = GetSessionFromMessage(message);
  302. if (session != null && session.UserId.HasValue)
  303. {
  304. var vals = message.Data.Split('|');
  305. var itemId = vals[0];
  306. long? positionTicks = null;
  307. if (vals.Length > 1)
  308. {
  309. long pos;
  310. if (long.TryParse(vals[1], out pos))
  311. {
  312. positionTicks = pos;
  313. }
  314. }
  315. var isPaused = vals.Length > 2 && string.Equals(vals[2], "true", StringComparison.OrdinalIgnoreCase);
  316. var isMuted = vals.Length > 3 && string.Equals(vals[3], "true", StringComparison.OrdinalIgnoreCase);
  317. var info = new PlaybackProgressInfo
  318. {
  319. ItemId = itemId,
  320. PositionTicks = positionTicks,
  321. IsMuted = isMuted,
  322. IsPaused = isPaused,
  323. SessionId = session.Id
  324. };
  325. if (vals.Length > 4)
  326. {
  327. info.MediaSourceId = vals[4];
  328. }
  329. if (vals.Length > 5 && !string.IsNullOrWhiteSpace(vals[5]))
  330. {
  331. info.VolumeLevel = int.Parse(vals[5], _usCulture);
  332. }
  333. if (vals.Length > 5 && !string.IsNullOrWhiteSpace(vals[6]))
  334. {
  335. info.AudioStreamIndex = int.Parse(vals[6], _usCulture);
  336. }
  337. if (vals.Length > 7 && !string.IsNullOrWhiteSpace(vals[7]))
  338. {
  339. info.SubtitleStreamIndex = int.Parse(vals[7], _usCulture);
  340. }
  341. _sessionManager.OnPlaybackProgress(info);
  342. }
  343. }
  344. private void ReportPlaybackStopped(WebSocketMessageInfo message)
  345. {
  346. _logger.Debug("Received ReportPlaybackStopped message");
  347. var session = GetSessionFromMessage(message);
  348. if (session != null && session.UserId.HasValue)
  349. {
  350. var info = _json.DeserializeFromString<PlaybackStopInfo>(message.Data);
  351. info.SessionId = session.Id;
  352. _sessionManager.OnPlaybackStopped(info);
  353. }
  354. }
  355. /// <summary>
  356. /// Reports the playback stopped.
  357. /// </summary>
  358. /// <param name="message">The message.</param>
  359. private void OnPlaybackStopped(WebSocketMessageInfo message)
  360. {
  361. _logger.Debug("Received PlaybackStopped message");
  362. var session = GetSessionFromMessage(message);
  363. if (session != null && session.UserId.HasValue)
  364. {
  365. var vals = message.Data.Split('|');
  366. var itemId = vals[0];
  367. long? positionTicks = null;
  368. if (vals.Length > 1)
  369. {
  370. long pos;
  371. if (long.TryParse(vals[1], out pos))
  372. {
  373. positionTicks = pos;
  374. }
  375. }
  376. var info = new PlaybackStopInfo
  377. {
  378. ItemId = itemId,
  379. PositionTicks = positionTicks,
  380. SessionId = session.Id
  381. };
  382. if (vals.Length > 2)
  383. {
  384. info.MediaSourceId = vals[2];
  385. }
  386. _sessionManager.OnPlaybackStopped(info);
  387. }
  388. }
  389. }
  390. }