package com.dai.party.config.realm;

import com.dai.party.config.CustomCredentialsMatcher;
import com.dai.party.config.SpringBeanFactoryUtils;
import com.dai.party.pojo.Backstage.BackstageUser;
import com.dai.party.service.backstageService.BackstageUserService;
import lombok.extern.slf4j.Slf4j;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @program: partyAffairs
 * @description: 项目后台配置Shiro继承AuthenticatingRealm类
 * @author: Dai Yuanchuan
 * @create: 2019-01-11 00:24
 **/
@Slf4j
@Component(value = "ShiroRealm")
public class BackstageShiroRealm extends AuthorizingRealm {

    /**
     * 项目后台用户服务注入
     */
    @Autowired
    private BackstageUserService backstageUserService;

    /**
     * 这是授权方法
     * @param principalCollection
     * @return
     */
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
        String username = principalCollection.getPrimaryPrincipal().toString();
        log.info("============================  username:" + username + "  ============================");
        return null;
    }

    /**
     * 这是认证方法
     * @param token
     * @return
     * @throws AuthenticationException
     */
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
        log.info("doGetAuthenticationInfo------------------登录------------------");
        // 手动注入服务
        if (backstageUserService == null) {
            backstageUserService = SpringBeanFactoryUtils.getBean(BackstageUserService.class);
        }
        UsernamePasswordToken upToken = (UsernamePasswordToken) token;
        BackstageUser backstageUser = backstageUserService.lookupToUserName(upToken.getUsername());
        if (backstageUser == null) {
            /* 抛出用户名不存在的异常 */
            throw new UnknownAccountException();
        }
        SimpleAuthenticationInfo simpleAuthenticationInfo = new SimpleAuthenticationInfo(
                backstageUser.getUserName(), backstageUser.getPassword(), getName());
        return simpleAuthenticationInfo;
    }

    /**
     * 这是重写加密规则
     */
    @PostConstruct
    public void initCredentialsMatcher() {
        setCredentialsMatcher(new CustomCredentialsMatcher());
    }
}