Blame view

netcore/src/Infrastructure/NCC/AspNetMvc/Extensions/HttpContextExtensions.cs 4.12 KB
de2bd2f9   “wangming”   项目初始化
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
  using NCC.Dependency;
  using Microsoft.AspNetCore.Mvc.Controllers;
  using System.Linq;
  using System.Text;
  
  namespace Microsoft.AspNetCore.Http
  {
      /// <summary>
      /// Http 拓展类
      /// </summary>
      [SuppressSniffer]
      public static class HttpContextExtensions
      {
          /// <summary>
          /// 获取 Action 特性
          /// </summary>
          /// <typeparam name="TAttribute"></typeparam>
          /// <param name="httpContext"></param>
          /// <returns></returns>
          public static TAttribute GetMetadata<TAttribute>(this HttpContext httpContext)
              where TAttribute : class
          {
              return httpContext.GetEndpoint()?.Metadata?.GetMetadata<TAttribute>();
          }
  
          /// <summary>
          /// 获取 控制器/Action 描述器
          /// </summary>
          /// <param name="httpContext"></param>
          /// <returns></returns>
          public static ControllerActionDescriptor GetControllerActionDescriptor(this HttpContext httpContext)
          {
              return httpContext.GetEndpoint()?.Metadata?.FirstOrDefault(u => u is ControllerActionDescriptor) as ControllerActionDescriptor;
          }
  
          /// <summary>
          /// 设置规范化文档自动登录
          /// </summary>
          /// <param name="httpContext"></param>
          /// <param name="accessToken"></param>
          public static void SigninToSwagger(this HttpContext httpContext, string accessToken)
          {
              // 设置 Swagger 刷新自动授权
              httpContext.Response.Headers["access-token"] = accessToken;
          }
  
          /// <summary>
          /// 设置规范化文档退出登录
          /// </summary>
          /// <param name="httpContext"></param>
          public static void SignoutToSwagger(this HttpContext httpContext)
          {
              httpContext.Response.Headers["access-token"] = "invalid_token";
          }
  
          /// <summary>
          /// 获取本机 IPv4地址
          /// </summary>
          /// <param name="context"></param>
          /// <returns></returns>
          public static string GetLocalIpAddressToIPv4(this HttpContext context)
          {
              return context.Connection.LocalIpAddress?.MapToIPv4()?.ToString();
          }
  
          /// <summary>
          /// 获取本机 IPv6地址
          /// </summary>
          /// <param name="context"></param>
          /// <returns></returns>
          public static string GetLocalIpAddressToIPv6(this HttpContext context)
          {
              return context.Connection.LocalIpAddress?.MapToIPv6()?.ToString();
          }
  
          /// <summary>
          /// 获取远程 IPv4地址
          /// </summary>
          /// <param name="context"></param>
          /// <returns></returns>
          public static string GetRemoteIpAddressToIPv4(this HttpContext context)
          {
              return context.Connection.RemoteIpAddress?.MapToIPv4()?.ToString();
          }
  
          /// <summary>
          /// 获取远程 IPv6地址
          /// </summary>
          /// <param name="context"></param>
          /// <returns></returns>
          public static string GetRemoteIpAddressToIPv6(this HttpContext context)
          {
              return context.Connection.RemoteIpAddress?.MapToIPv6()?.ToString();
          }
  
          /// <summary>
          /// 获取完整请求地址
          /// </summary>
          /// <param name="request"></param>
          /// <returns></returns>
          public static string GetRequestUrlAddress(this HttpRequest request)
          {
              return new StringBuilder()
                      .Append(request.Scheme)
                      .Append("://")
                      .Append(request.Host)
                      .Append(request.PathBase)
                      .Append(request.Path)
                      .Append(request.QueryString)
                      .ToString();
          }
  
          /// <summary>
          /// 获取来源地址
          /// </summary>
          /// <param name="request"></param>
          /// <param name="refererHeaderKey"></param>
          /// <returns></returns>
          public static string GetRefererUrlAddress(this HttpRequest request, string refererHeaderKey = "Referer")
          {
              return request.Headers[refererHeaderKey].ToString();
          }
      }
  }