TEAMModelOS/TEAMModelOS.SDK/Extension/HttpContextExtensions.cs

using System;
using System.Collections.Generic;
using System.Text;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Primitives;

namespace TEAMModelOS.SDK.Extension
{
    public static class HttpContextExtensions
    {
        /// <summary>
        /// 取得驗證金鑰，Authorization
        /// </summary>        
        public static string GetToken(this HttpContext httpContext)
        {
            return httpContext.Request.Headers["Authorization"].ToString();
        }

        /// <summary>
        /// 取得JWT驗證金鑰，Authorization Bearer
        /// </summary>
        /// <param name="httpContext"></param>
        /// <returns></returns>
        public static string GetJwtToken(this HttpContext httpContext)
        {
            var token = string.Empty;
            string authorization = httpContext.Request.Headers["Authorization"].ToString();
            if (!string.IsNullOrWhiteSpace(authorization) && authorization.StartsWith("Bearer ", StringComparison.OrdinalIgnoreCase))
            {
                token = authorization.Substring("Bearer ".Length).Trim();
            }
            return token;
        }

        /// <summary>
        /// 取得遠端呼叫的IP
        /// </summary>        
        public static string GetRemoteIP(this HttpContext httpContext)
        {
            return httpContext?.Connection?.RemoteIpAddress?.ToString();
        }

        /// <summary>
        /// 取得X-Auth-Key值
        /// </summary>        
        /// <param name="key">Key Name</param>
        /// <returns></returns>
        public static string GetXAuth(this HttpContext httpContext, string key = null)
        {
            try
            {
                if (httpContext.Request.Headers.TryGetValue($"X-Auth-{key}", out StringValues value))
                    return value.ToString();
                else
                    return null;
            }
            catch
            {
                return null;
            }
        }
        /// <summary>
        /// 取得AuthToken權杖資訊
        /// </summary>        
        /// <param name="key">Key Name</param>
        /// <returns></returns>
        public static (string id, string school) GetApiTokenInfo(this HttpContext httpContext, string key = null)
        {
            object id = null, school = null;
            httpContext?.Items.TryGetValue("ID", out id);
            httpContext?.Items.TryGetValue("School", out school);
            return (id?.ToString(), school?.ToString());
        }
        /// <summary>
        /// 取得AuthToken權杖資訊
        /// </summary>        
        /// <param name="key">Key Name</param>
        /// <returns></returns>
        public static (string id, string name, string picture, string school) GetAuthTokenInfo(this HttpContext httpContext, string key = null)
        {
            object id = null, name = null, picture = null, school = null;
            httpContext?.Items.TryGetValue("ID", out id);
            httpContext?.Items.TryGetValue("Name", out name);
            httpContext?.Items.TryGetValue("Picture", out picture);
            httpContext?.Items.TryGetValue("School", out school);

            return (id?.ToString(), name?.ToString(), picture?.ToString(), school?.ToString());
        }
        /// <summary>
        /// 取得AuthToken權杖資訊
        /// </summary>        
        /// <param name="key">Key Name</param>
        /// <returns></returns>
        public static (string id, string name, string picture, string school, string area, string keyData) GetAuthTokenKey(this HttpContext httpContext, string key = null)
        {
            object id = null, name = null, picture = null, school = null, area = null, keyData = null;
            httpContext?.Items.TryGetValue("ID", out id);
            httpContext?.Items.TryGetValue("Name", out name);
            httpContext?.Items.TryGetValue("Picture", out picture);
            httpContext?.Items.TryGetValue("School", out school);
            httpContext?.Items.TryGetValue("Area", out area);
            if (!string.IsNullOrWhiteSpace(key))
            {
                httpContext?.Items.TryGetValue(key, out keyData);
            }
            return (id?.ToString(), name?.ToString(), picture?.ToString(), school?.ToString(), area?.ToString(), keyData?.ToString());
        }
        /// <summary>
        /// 取得User-Agent值
        /// </summary>       
        public static string GetUserAgent(this HttpContext httpContext)
        {
            try
            {
                return httpContext.Request.Headers["User-Agent"].ToString();
            }
            catch
            {
                return null;
            }
        }

        /// <summary>
        /// 取得Scheme值
        /// </summary>      
        public static string GetScheme(this HttpContext httpContext)
        {
            return httpContext?.Request?.Scheme;
        }

        /// <summary>
        /// 取得HostName值
        /// </summary>        
        public static string GetHostName(this HttpContext httpContext)
        {
            return httpContext?.Request?.Host.ToString();
        }

        /// <summary>
        /// 设置本地cookie
        /// </summary>
        /// <param name="key">键</param>
        /// <param name="value">值</param>  
        /// <param name="minutes">过期时长，单位：分钟</param>      
        public static void SetCookies(HttpResponse Response, string key, string value, int minutes = 30)
        {
            Response.Cookies.Append(key, value, new CookieOptions
            {
                Expires = DateTimeOffset.Now.AddMinutes(minutes)
            });
        }
        /// <summary>
        /// 删除指定的cookie
        /// </summary>
        /// <param name="key">键</param>
        public static void DeleteCookies(HttpContext httpContext, string key)
        {
            httpContext.Response.Cookies.Delete(key);
        }
    }
}