OA2023/OASystem/OASystem.Api/Controllers/AuthController.cs


using Microsoft.Extensions.Caching.Distributed;
using OASystem.Infrastructure.Repositories.Login;
using System.IdentityModel.Tokens.Jwt;
using System.Security.Claims;
using StackExchange.Redis;

namespace OASystem.API.Controllers
{
    /// <summary>
    /// 鉴权相关
    /// </summary>
    public class AuthController : ControllerBase
    {
        private readonly IConfiguration _config;
        private readonly LoginRepository _loginRep;
        private readonly IMapper _mapper;
        private IDatabase _redis;
        private RedisHelper _redisHelper;
        public AuthController(IConfiguration config, LoginRepository loginRep, IMapper mapper, RedisHelper client)
        {
            _config = config;
            _loginRep = loginRep;
            _mapper = mapper;
            //_redis = client.GetDatabase(RedisEnum.Common);
            //_redisHelper = client("132.232.92.186", "7369", "123456");
        }

        /// <summary>
        /// 用户登录
        /// </summary>
        /// <param name="dto"></param>
        /// <returns></returns>
        [AllowAnonymous]
        [HttpPost("login")]
        [ProducesResponseType(typeof(LoginView), StatusCodes.Status200OK)]
        public async Task<IActionResult> LoginAsync(LoginDto dto)
        {
            #region 校验用户信息，假设此处我们已经校验成功
            var userData = OAMethodLib.LoginLib.Api_Login(_loginRep,dto);
            if (userData.Result.Code != 0)
            {
                if (userData.Result.Code != 0) { return Ok(JsonView(false, userData.Result.Message)); }

                return Ok(JsonView(false,"暂无该员工信息!"));
            }

            #endregion
            var view = new LoginView
            {
                Expires = DateTime.Now.AddMinutes(30)
            };
           
            string authorId = dto.Number + "Token";

            var claims = new[] { new Claim(ClaimTypes.NameIdentifier, "Future") };
            var key = new SymmetricSecurityKey(Encoding.UTF8.GetBytes(_config["JwtSecurityKey"]));
            var creds = new SigningCredentials(key, SecurityAlgorithms.HmacSha256);
            var token = new JwtSecurityToken(
                issuer: "OASystem.com",
                audience: "OASystem.com",
                claims: claims,
                expires: view.Expires,
                signingCredentials: creds);
            view.Token = new JwtSecurityTokenHandler().WriteToken(token);


            // 往Redis里面存入数据
            _redis.StringSet(authorId, view.Token);


            // 从Redis里面取数据
            string name = _redis.StringGet(authorId);

            return Ok(JsonView(view));
        }


        /// <summary>
        /// 测试auth
        /// </summary>
        /// <param name="dto"></param>
        /// <returns></returns>
        [Authorize]
        [HttpPost("TestToken")]
        [ProducesResponseType(typeof(LoginView), StatusCodes.Status200OK)]
        public async Task<IActionResult> TestToken(LoginDto dto)
        {
            string authorId = dto.Number + "Token";
            // 从Redis里面取数据
            string userToken = _redis.StringGet(authorId);

            var view = new LoginView
            {
                Token = authorId + ":" + userToken
            };

            return Ok(JsonView(view));
        }
    }
}