LoginInfoService.cs 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232
  1. using IdentityModel;
  2. using Microsoft.AspNetCore.Http;
  3. using Microsoft.Extensions.Configuration;
  4. using Microsoft.Extensions.Options;
  5. using System;
  6. using System.Collections.Generic;
  7. using System.Linq;
  8. using System.Security.Claims;
  9. using System.Text;
  10. using System.Threading.Tasks;
  11. using TEAMModelOS.Model.Core.Dtos;
  12. using TEAMModelOS.Model.Core.Models;
  13. using TEAMModelOS.SDK.Context.Configuration;
  14. using TEAMModelOS.SDK.Context.Constant.Common;
  15. using TEAMModelOS.SDK.Context.Exception;
  16. using TEAMModelOS.SDK.Extension.DataResult.JsonRpcRequest;
  17. using TEAMModelOS.SDK.Extension.DataResult.JsonRpcResponse;
  18. using TEAMModelOS.SDK.Extension.HttpClient.Implements;
  19. using TEAMModelOS.SDK.Extension.JwtAuth.JwtHelper;
  20. using TEAMModelOS.SDK.Extension.JwtAuth.Models;
  21. using TEAMModelOS.SDK.Helper.Common.CollectionHelper;
  22. using TEAMModelOS.SDK.Helper.Common.DateTimeHelper;
  23. using TEAMModelOS.SDK.Helper.Common.JsonHelper;
  24. using TEAMModelOS.SDK.Helper.Network.HttpHelper;
  25. using TEAMModelOS.SDK.Helper.Security.BCryptHelper;
  26. using TEAMModelOS.SDK.Module.AzureTable.Implements;
  27. using TEAMModelOS.SDK.Module.AzureTable.Interfaces;
  28. using TEAMModelOS.Service.Core.Interfaces;
  29. namespace TEAMModelOS.Service.Core.Implements
  30. {
  31. public class LoginInfoService : BaseService, ILoginInfoService
  32. {
  33. //private IAzureTableDBRepository _repository;
  34. private readonly IOptions<JwtSetting> _options;
  35. private readonly HttpClientUserInfo _httpClientService;
  36. public LoginInfoService( IOptions<JwtSetting> options, HttpClientUserInfo httpClientService)
  37. {
  38. _options = options;
  39. // _repository = repository;
  40. _httpClientService = httpClientService;
  41. }
  42. public async Task<LoginResult> CheckLoginAsync(TicketInfo ticketInfo) {
  43. // string jtoken = HttpContextHelper.GetValueInHttp(_httpContextAccessor.HttpContext.Request, Constants.AUTHORIZATION);
  44. if (string.IsNullOrEmpty(ticketInfo.Token))
  45. {
  46. LoginResult result = new LoginResult();
  47. if (string.IsNullOrEmpty(ticketInfo.Ticket))
  48. {
  49. result.CheckTicket = false;
  50. return result;
  51. }
  52. string code = BCryptHelper.Ecrypt(ticketInfo.Ticket + ticketInfo.TeamModelId);
  53. bool f = BCryptHelper.Verify(ticketInfo.Ticket + ticketInfo.TeamModelId, ticketInfo.Sign);
  54. LoginInfo login = FindOneByKey<LoginInfo>("Ticket", ticketInfo.Ticket).Result;
  55. if (login != null && !string.IsNullOrEmpty(login.Token))
  56. {
  57. result.CheckTicket = true;
  58. JwtResponse token = await CreateJwtToken(login);
  59. result.JwtToken = token;
  60. login.Token = token.Access_token;
  61. result.JwtToken.Scope = login.Scope;
  62. await Update<LoginInfo>(login);
  63. return result;
  64. }
  65. ///不验证TmdID
  66. #region
  67. result.CheckTicket = true;
  68. LoginInfo loginInfo = new LoginInfo
  69. {
  70. PartitionKey = "15283771540",
  71. Phone = "15283771540",
  72. RowKey = Guid.NewGuid().ToString(),
  73. TeamModelId = ticketInfo.TeamModelId,
  74. Name = ticketInfo.Name,
  75. Ticket = ticketInfo.Ticket,
  76. CountryCode ="86"
  77. };
  78. TeamModelUser user = await FindOneByKey<TeamModelUser>("TeamModelId", ticketInfo.TeamModelId);
  79. if (user == null || string.IsNullOrEmpty(user.RowKey))
  80. {
  81. user = new TeamModelUser { RowKey = Guid.NewGuid().ToString(), PartitionKey = loginInfo.CountryCode, RegisterTime = DateTimeHelper.ConvertToTimeStamp13(DateTime.Now) };
  82. }
  83. user.Cellphone = "15283771540";
  84. user.NickName = ticketInfo.Name;
  85. if (string.IsNullOrEmpty(user.FullName))
  86. {
  87. user.FullName = ticketInfo.Name;
  88. }
  89. user.TeamModelId = ticketInfo.TeamModelId;
  90. user.CountryCode = "86";
  91. JwtResponse jwtToken = await CreateJwtToken(loginInfo);
  92. loginInfo.Token = jwtToken.Access_token;
  93. loginInfo.Scope = jwtToken.Scope;
  94. result.JwtToken = jwtToken;
  95. await Save<LoginInfo>(loginInfo);
  96. await SaveOrUpdate<TeamModelUser>(user);
  97. // JosnRPCResponse<TeamModelIdInfo> response = MessagePackHelper.JsonToObject<JosnRPCResponse<TeamModelIdInfo>>(jsonStr);
  98. return result;
  99. #endregion
  100. /*
  101. JosnRPCRequest<Dictionary<string, object>> request = new JosnRPCRequest<Dictionary<string, object>>
  102. {
  103. method = "UserInfo"
  104. };
  105. Dictionary<string, object> ticket = new Dictionary<string, object>
  106. {
  107. { "ticket", ticketInfo.Ticket }
  108. };
  109. request.@params = ticket;
  110. string data = MessagePackHelper.ObjectToJson(request);
  111. string jsonStr = _httpClientService.HttpPost(BaseConfigModel.Configuration["HaBookAuth:AccountUrl"], data, Constants.CONTENT_TYPE_JSON, Encoding.UTF8);
  112. if (!string.IsNullOrEmpty(jsonStr))
  113. {
  114. JosnRPCResponse<TeamModelIdInfo> response = MessagePackHelper.JsonToObject<JosnRPCResponse<TeamModelIdInfo>>(jsonStr);
  115. if (response.error == null && response != null)
  116. {
  117. result.CheckTicket = true;
  118. LoginInfo loginInfo = new LoginInfo
  119. {
  120. PartitionKey = response.result.cellphone,
  121. Phone = response.result.cellphone,
  122. RowKey = Guid.NewGuid().ToString(),
  123. TeamModelId = response.result.id,
  124. Name = response.result.name,
  125. Ticket = ticketInfo.Ticket,
  126. CountryCode = response.result.countryCode
  127. };
  128. TeamModelUser user= await FindOneByKey<TeamModelUser>("TeamModelId", response.result.id);
  129. if (user == null || string.IsNullOrEmpty(user.RowKey))
  130. {
  131. user = new TeamModelUser { RowKey = Guid.NewGuid().ToString(), PartitionKey = loginInfo.CountryCode ,RegisterTime=DateTimeHelper.ConvertToTimeStamp13(DateTime.Now) };
  132. }
  133. user.Cellphone = response.result.cellphone;
  134. user.NickName = response.result.name;
  135. if (string.IsNullOrEmpty(user.FullName)) {
  136. user.FullName = response.result.name;
  137. }
  138. user.TeamModelId = response.result.id;
  139. user.CountryCode = response.result.countryCode;
  140. JwtResponse jwtToken = await CreateJwtToken(loginInfo);
  141. loginInfo.Token = jwtToken.Access_token;
  142. loginInfo.Scope = jwtToken.Scope;
  143. result.JwtToken = jwtToken;
  144. await Save<LoginInfo>(loginInfo);
  145. await SaveOrUpdate<TeamModelUser>(user);
  146. return result;
  147. }
  148. else
  149. {
  150. result.CheckTicket = false;
  151. return result;
  152. }
  153. }
  154. else
  155. {
  156. result.CheckTicket = false;
  157. return result;
  158. }
  159. */
  160. }
  161. else
  162. {
  163. ClaimModel claimModel = JwtHelper.SerializeJWT(ticketInfo.Token);
  164. var dateTime = DateTimeHelper.ConvertToTimeStamp10(DateTime.Now);
  165. var expExt=claimModel.Claim.TryGetValue("exp",out var exp);
  166. if (expExt==false || dateTime > long.Parse(exp.ToString()))
  167. {
  168. throw new BizException(401, "Unauthorized");
  169. }
  170. Dictionary<string, object> msp = new Dictionary<string, object>
  171. {
  172. { "Token", ticketInfo.Token }
  173. };
  174. LoginInfo loginInfo = FindOneByDict<LoginInfo>(msp).Result;
  175. if (loginInfo != null && !string.IsNullOrEmpty(loginInfo.Token))
  176. {
  177. return new LoginResult { JwtToken = new JwtResponse { Access_token=loginInfo.Token ,Scope=loginInfo.Scope}, CheckTicket = true };
  178. }
  179. else
  180. {
  181. throw new BizException(401, "Unauthorized");
  182. }
  183. }
  184. }
  185. public async Task<JwtResponse> CreateJwtToken(LoginInfo loginInfo)
  186. {
  187. Dictionary<string, object> dict = new Dictionary<string, object>
  188. {
  189. { "Phone", loginInfo.Phone },
  190. { "AreaCode", loginInfo.CountryCode },
  191. { "TeamModelId", loginInfo.TeamModelId }
  192. };
  193. string role = "";
  194. List<RoleUser> roleUsers = await FindListByDict<RoleUser>(dict);
  195. if (roleUsers.IsNotEmpty())
  196. {
  197. foreach (RoleUser roleUser in roleUsers)
  198. {
  199. role = role + roleUser.RoleCode + ",";
  200. }
  201. role = role.Substring(0, role.Length - 1);
  202. }
  203. else {
  204. role = "Teacher";
  205. }
  206. ClaimModel model = new ClaimModel
  207. {
  208. Scope = "WebApp"
  209. };
  210. model.Claims.Add(new Claim(JwtClaimTypes.Name, loginInfo.Name));
  211. model.Claims.Add(new Claim(JwtClaimTypes.Id, loginInfo.TeamModelId));
  212. ////保护隐私
  213. //model.Claims.Add(new Claim(JwtClaimTypes.PhoneNumber, loginInfo.Phone));
  214. model.Claims.AddRange(role.Split(',').Select(s => new Claim(JwtClaimTypes.Role, s)));
  215. model.Roles.Add(role);
  216. JwtResponse jwtResponse = JwtHelper.IssueJWT(model, _options.Value);
  217. return jwtResponse;
  218. }
  219. public Task<LoginInfo> SaveLoginInfoAsync(LoginInfo loginInfo)
  220. {
  221. return Save<LoginInfo>(loginInfo);
  222. }
  223. }
  224. }