import com.zs.springboot.model.User;
import com.zs.springboot.service.UserService;
import com.zs.springboot.util.ResultUtil;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.util.ByteSource;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;
import java.util.Map;

/**
 * @author zs
 * 登錄realm
 */
public class LoginRealm extends AuthorizingRealm {

  @Autowired
  private UserService userService;
  /**
   * 授權方法，只有頁面出現(xiàn)關于權限標簽的時候或者項目中關于授權注解的時候，才可以被使用
   * @param principal
   * @return
   */
  @Override
  protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principal) {　
    User user = (User) principal.getPrimaryPrincipal();
    Map<String, Object> resultMap1 = userService.getRoleByUid(user.getUid());
    //獲取角色，通過多表查詢獲取用戶對應的角色
    List<String> roleList = (List<String>) ResultUtil.getResult(resultMap1);
    SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
    info.addRoles(roleList);
    //獲取權限，通過關聯(lián)中間表，多表查詢出該用戶對應的角色所擁有的權限
    Map<String, Object> resultMap2 = userService.getPromessionByUid(user.getUid());
    List<String> permissionList = (List<String>) ResultUtil.getResult(resultMap2);
    info.addStringPermissions(permissionList);
    return info;
  }

  /**
   * 認證方法
   * @param token
   * @return
   * @throws AuthenticationException
   */
  @Override
  protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
    //獲取當前登錄的用戶名
    String username = (String) token.getPrincipal();
    //根據(jù)用戶到數(shù)據(jù)庫搜索用戶信息
    Map<String, Object> login = userService.login(username);
    //如果用戶不存在則拋出異常
    if ((Integer) login.get("code") == 404) {
      throw new UnknownAccountException("用戶不存在");
    }
    //如果用戶存在，獲取用戶信息
    User user = (User) login.get("result");
    //進行認證
    SimpleAuthenticationInfo info = new SimpleAuthenticationInfo(user, user.getPassword(), ByteSource.Util.bytes(user.getSalt()), this.getName());
    //將用戶信息放入session中，密碼制空
    Session session = SecurityUtils.getSubject().getSession();
    user.setPassword(null);
    session.setAttribute("user", user);
    return info;
  }
}