支持由代理软件设置的 Protocol 与 Hostname

src/Discussion.Web/Services/UserManagement/Avatar/UrlGenerators/StorageFileAvatarUrlGenerator.cs

@@ -1,25 +1,30 @@
 using Discussion.Core.Models.UserAvatar;
+using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
 
 namespace Discussion.Web.Services.UserManagement.Avatar.UrlGenerators
 {
     public class StorageFileAvatarUrlGenerator : IUserAvatarUrlGenerator<StorageFileAvatar>
     {
         private readonly IUrlHelper _urlHelper;
+        private readonly IHttpContextAccessor _httpContextAccessor;
 
-        public StorageFileAvatarUrlGenerator(IUrlHelper urlHelper)
+        public StorageFileAvatarUrlGenerator(IUrlHelper urlHelper, IHttpContextAccessor httpContextAccessor)
         {
             _urlHelper = urlHelper;
+            _httpContextAccessor = httpContextAccessor;
         }
 
         public string GetUserAvatarUrl(StorageFileAvatar avatar)
         {
+            var request = _httpContextAccessor.HttpContext.Request;
             // ReSharper disable Mvc.ActionNotResolved
             // ReSharper disable Mvc.ControllerNotResolved
             return _urlHelper.Action("DownloadFile", 
                 "Common", 
                 new {slug = avatar.StorageFileSlug}, 
-                _urlHelper.ActionContext.HttpContext.Request.Scheme);
+                request.Scheme, 
+                request.Host.ToString());
         }
     }
 }

src/Discussion.Web/Startup.cs

@@ -28,6 +28,7 @@
 using Discussion.Core.Logging;
 using Discussion.Core.Middleware;
 using Discussion.Core.Utilities;
+using Microsoft.AspNetCore.HttpOverrides;
 
 namespace Discussion.Web
 {
@@ -57,6 +58,12 @@ private static void Main()
         // ConfigureServices is invoked before Configure
         public void ConfigureServices(IServiceCollection services)
         {
+            services.Configure<ForwardedHeadersOptions>(options =>
+            {
+                options.ForwardedHeaders = ForwardedHeaders.XForwardedFor | ForwardedHeaders.XForwardedProto | ForwardedHeaders.XForwardedHost;
+                options.KnownNetworks.Clear();
+                options.KnownProxies.Clear();
+            });
             services.AddLazySupport();
             services.AddLogging();
             services.AddSingleton(HtmlEncoder.Create(UnicodeRanges.BasicLatin, UnicodeRanges.CjkUnifiedIdeographs));
@@ -107,6 +114,7 @@ public void ConfigureServices(IServiceCollection services)
 
         public void Configure(IApplicationBuilder app)
         {
+            app.UseForwardedHeaders();
             app.UseTracingId();
             SetupGlobalExceptionHandling(app);
             SetupHttpsSupport(app);

test/Discussion.Web.Tests/Specs/Services/UserAvatarServiceSpecs.cs

@@ -80,9 +80,12 @@ private static IUrlHelper CreateMockUrlHelper(string slug = null)
         private static IServiceProvider CreateServices(IUrlHelper urlHelper)
         {
             var services = new ServiceCollection();
+            var mockAccessor = new Mock<IHttpContextAccessor>();
+            mockAccessor.SetupGet(m => m.HttpContext).Returns(new DefaultHttpContext());
 
             services.AddScoped(sp => urlHelper);
             services.AddSingleton<IAvatarUrlService, DispatchAvatarUrlService>();
+            services.AddSingleton(mockAccessor.Object);
             services.AddScoped<IUserAvatarUrlGenerator<DefaultAvatar>, DefaultAvatarUrlGenerator>();
             services.AddScoped<IUserAvatarUrlGenerator<StorageFileAvatar>, StorageFileAvatarUrlGenerator>();
             services.AddScoped<IUserAvatarUrlGenerator<GravatarAvatar>, GravatarAvatarUrlGenerator>();