UserAgentDelegatingHandler.cs 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Net.Http;
  4. using System.Net.Http.Headers;
  5. using System.Threading;
  6. using System.Threading.Tasks;
  7. namespace MediaBrowser.Common.Net
  8. {
  9. /// <summary>
  10. /// User agent delegating handler.
  11. /// Adds User-Agent header to all requests.
  12. /// </summary>
  13. public class UserAgentDelegatingHandler : DelegatingHandler
  14. {
  15. /// <inheritdoc />
  16. public UserAgentDelegatingHandler(IApplicationHost applicationHost)
  17. {
  18. UserAgentValues = new List<ProductInfoHeaderValue>
  19. {
  20. new ProductInfoHeaderValue(applicationHost.Name.Replace(' ', '-'), applicationHost.ApplicationVersionString),
  21. new ProductInfoHeaderValue($"({Environment.OSVersion}; {applicationHost.ApplicationUserAgentAddress})")
  22. };
  23. }
  24. /// <summary>
  25. /// Gets or sets the user agent values.
  26. /// </summary>
  27. public List<ProductInfoHeaderValue> UserAgentValues { get; set; }
  28. /// <summary>
  29. /// Send request message.
  30. /// </summary>
  31. /// <param name="request">The request message.</param>
  32. /// <param name="cancellationToken">The cancellation token.</param>
  33. /// <returns>A <see cref="Task"/> containing the <see cref="HttpResponseMessage"/>.</returns>
  34. protected override Task<HttpResponseMessage> SendAsync(
  35. HttpRequestMessage request,
  36. CancellationToken cancellationToken)
  37. {
  38. if (request.Headers.UserAgent.Count == 0)
  39. {
  40. foreach (var userAgentValue in UserAgentValues)
  41. {
  42. request.Headers.UserAgent.Add(userAgentValue);
  43. }
  44. }
  45. return base.SendAsync(request, cancellationToken);
  46. }
  47. }
  48. }