AdminProvider.cs 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. using Microsoft.AspNetCore.Http;
  2. using Microsoft.AspNetCore.Mvc.Controllers;
  3. using Microsoft.AspNetCore.Mvc.Filters;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Linq;
  7. using System.Text;
  8. using dodohold.core;
  9. namespace molilian.core
  10. {
  11. public class AdminProvider : IAuthorizationProvider
  12. {
  13. public string Domain => "admin";
  14. AccessTokenCore tokenCore = new AccessTokenCore();
  15. AdminPermissionCore permissionCore = new AdminPermissionCore();
  16. /// <summary>
  17. /// 验证是否未授权
  18. /// </summary>
  19. /// <param name="request">客户请求的token和scope(域)</param>
  20. /// <param name="logicAttr">受访问的权限属性</param>
  21. /// <returns></returns>
  22. public bool Verified(ActionExecutingContext context, (string ticket, string domain) request, MyAuthorizeAttribute logicAttr)
  23. {
  24. //tikect 为空,直接返回 false
  25. if (string.IsNullOrEmpty(request.ticket)) { return false; }
  26. //如果接口里面要求验证domain,就判断是否匹配当前domain
  27. if (!string.IsNullOrEmpty(logicAttr.Domain) && !logicAttr.Domain.Equals(request.domain)) { return false; }
  28. AccessTokenDTO accessTicket = (AccessTokenDTO)Get(request.domain, request.ticket);
  29. //从内存中获取ticket,没有就返回失败
  30. if (accessTicket == null) { return false; }
  31. //tikcet 设置永不过期,就直接返回 true
  32. if (accessTicket.ExpiresIn == 0) { return true; }
  33. DateTime expiresTime = (accessTicket.create_time + accessTicket.ExpiresIn).Convert2Datetime();
  34. //ticket 有效期还没到也返回true
  35. if (expiresTime > DateTime.Now) { return true; }
  36. //这里拿到 actionName 和 actionMethod ,但是实际上并没有去验证
  37. string actionName = null;
  38. if (context.ActionDescriptor is ControllerActionDescriptor actionDescriptor)
  39. {
  40. actionName = actionDescriptor.ActionName;
  41. }
  42. string actionMethod = context.HttpContext.Request.Method;
  43. //permission == null 就是没授权咯
  44. var permission = accessTicket.Permissions.Where(s => s.ResourceFlag.Equals(actionName) && s.ActionFlag.Equals(actionMethod)).FirstOrDefault();
  45. if (permission == null) { return false; }
  46. return false;
  47. }
  48. public IAccessTokenDTO Get(HttpContext context)
  49. {
  50. var (accessToken, domain) = context.GetAccessToken();
  51. return Get(domain, accessToken);
  52. }
  53. /// <summary>
  54. /// 获取token缓存(从内存缓存中)
  55. /// </summary>
  56. /// <param name="token">客户请求的token</param>
  57. /// <returns></returns>
  58. public IAccessTokenDTO Get(string domain, string accessTicket)
  59. {
  60. var accessToken = tokenCore.Get(domain, accessTicket);
  61. if (accessToken == null) { return null; }
  62. //todo 这里除了获取accesstoken以外 还得获取一下授权
  63. string roleIds = accessToken.AccessRoles;
  64. //if (!string.IsNullOrEmpty(roleIds))
  65. //{
  66. // accessToken.Permissions = permissionCore.GetList(new { id = roleIds.ConvertToIntArray() })?.List;
  67. //}
  68. return accessToken;
  69. }
  70. /// <summary>
  71. /// 注销授权
  72. /// </summary>
  73. /// <param name="accessTicket"></param>
  74. /// <returns></returns>
  75. public int Revoke(string doamin, string accessTicket)
  76. {
  77. return tokenCore.Delete(doamin, accessTicket);
  78. }
  79. /// <summary>
  80. /// 创建一个授权
  81. /// </summary>
  82. /// <param name="key"></param>
  83. /// <param name="secret"></param>
  84. /// <param name="roles"></param>
  85. /// <param name="domain"></param>
  86. /// <param name="clientIp"></param>
  87. /// <returns></returns>
  88. public IAccessTokenDTO Create(long id, string key, string secret, string roles, string domain, string clientIp)
  89. {
  90. AccessTokenDTO token = Generate(id, key, secret, roles, domain, clientIp);
  91. tokenCore.Create(token);
  92. return token;
  93. }
  94. /// <summary>
  95. /// 工具类 ,产生一个 token
  96. /// </summary>
  97. /// <param name="accessKey"></param>
  98. /// <param name="secret"></param>
  99. /// <param name="accessRoles"></param>
  100. /// <param name="domain"></param>
  101. /// <param name="clientIp"></param>
  102. /// <returns></returns>
  103. public AccessTokenDTO Generate(long id, string accessKey, string secret, string roles, string domain, string clientIp)
  104. {
  105. DateTime now = DateTime.Now;
  106. AccessTokenDTO token = new AccessTokenDTO
  107. {
  108. AccessId = id,
  109. AccessKey = accessKey,
  110. AccessRoles = roles,
  111. create_time = now.Convert2UnixTimestamp(),
  112. ExpiresIn = 0,
  113. Domain = domain,
  114. ClientIp = clientIp,
  115. AccessTicket = $"{accessKey}|{secret}|{roles}|{domain}|{now}|{clientIp}".ToAES256()
  116. };
  117. return token;
  118. }
  119. }
  120. }