46 lines
1.7 KiB
C#
46 lines
1.7 KiB
C#
using FastGithub.Scanner;
|
|
using Microsoft.AspNetCore.Builder;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using System.Net.Http;
|
|
using Yarp.ReverseProxy.Forwarder;
|
|
|
|
namespace FastGithub
|
|
{
|
|
/// <summary>
|
|
/// gitub反向代理的中间件扩展
|
|
/// </summary>
|
|
public static class ReverseProxyApplicationBuilderExtensions
|
|
{
|
|
/// <summary>
|
|
/// 使用gitub反向代理中间件
|
|
/// </summary>
|
|
/// <param name="app"></param>
|
|
/// <returns></returns>
|
|
public static IApplicationBuilder UseGithubReverseProxy(this IApplicationBuilder app)
|
|
{
|
|
var httpForwarder = app.ApplicationServices.GetRequiredService<IHttpForwarder>();
|
|
var httpClientHanlder = app.ApplicationServices.GetRequiredService<GithubHttpClientHanlder>();
|
|
var githubResolver = app.ApplicationServices.GetRequiredService<IGithubResolver>();
|
|
|
|
app.Use(next => async context =>
|
|
{
|
|
var host = context.Request.Host.Host;
|
|
if (githubResolver.IsSupported(host) == false)
|
|
{
|
|
await context.Response.WriteAsJsonAsync(new { message = $"不支持以{host}访问" });
|
|
}
|
|
else
|
|
{
|
|
var port = context.Request.Host.Port ?? 443;
|
|
var destinationPrefix = $"http://{host}:{port}/";
|
|
var httpClient = new HttpMessageInvoker(httpClientHanlder, disposeHandler: false);
|
|
await httpForwarder.SendAsync(context, destinationPrefix, httpClient);
|
|
}
|
|
});
|
|
|
|
return app;
|
|
}
|
|
}
|
|
}
|