49 lines
1.7 KiB
C#
49 lines
1.7 KiB
C#
using FastGithub.ReverseProxy;
|
|
using FastGithub.Scanner;
|
|
using Microsoft.AspNetCore.Builder;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Options;
|
|
using System;
|
|
using System.Linq;
|
|
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 httpClientFactory = app.ApplicationServices.GetRequiredService<NoneSniHttpClientFactory>();
|
|
var options = app.ApplicationServices.GetRequiredService<IOptionsMonitor<GithubLookupFactoryOptions>>();
|
|
|
|
app.Use(next => async context =>
|
|
{
|
|
var host = context.Request.Host.Host;
|
|
if (options.CurrentValue.Domains.Contains(host) == false)
|
|
{
|
|
await context.Response.WriteAsJsonAsync(new { message = $"不支持以{host}访问" });
|
|
}
|
|
else
|
|
{
|
|
var port = context.Request.Host.Port ?? 443;
|
|
var destinationPrefix = $"http://{host}:{port}/";
|
|
var httpClient = httpClientFactory.CreateHttpClient();
|
|
await httpForwarder.SendAsync(context, destinationPrefix, httpClient);
|
|
}
|
|
});
|
|
|
|
return app;
|
|
}
|
|
}
|
|
}
|