若依 1.0

RuoYi

2020-05-24 3da7aea540b1c381303769607a0b11ed5ba98128

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package com.ruoyi.common.security.config; 
 
import java.util.Collection; 
import java.util.LinkedHashMap; 
import java.util.Map; 
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; 
import org.springframework.security.core.Authentication; 
import org.springframework.security.core.GrantedAuthority; 
import org.springframework.security.core.authority.AuthorityUtils; 
import org.springframework.security.oauth2.provider.token.UserAuthenticationConverter; 
import org.springframework.util.StringUtils; 
import com.ruoyi.common.core.constant.SecurityConstants; 
import com.ruoyi.common.core.text.Convert; 
import com.ruoyi.common.security.domain.LoginUser; 
 
/** 
 * https://my.oschina.net/giegie/blog/3023768 根据checktoken 的结果转化用户信息 
 *  
 * @author lengleng 
 */ 
public class CommonUserConverter implements UserAuthenticationConverter 
{ 
    private static final String N_A = "N/A"; 
 
    /** 
     * 将授权信息返回到资源服务 
     */ 
    @Override 
    public Map<String, ?> convertUserAuthentication(Authentication userAuthentication) 
    { 
        Map<String, Object> authMap = new LinkedHashMap<>(); 
        authMap.put(USERNAME, userAuthentication.getName()); 
        if (userAuthentication.getAuthorities() != null && !userAuthentication.getAuthorities().isEmpty()) 
        { 
            authMap.put(AUTHORITIES, AuthorityUtils.authorityListToSet(userAuthentication.getAuthorities())); 
        } 
        return authMap; 
    } 
 
    /** 
     * 获取用户认证信息 
     */ 
    @Override 
    public Authentication extractAuthentication(Map<String, ?> map) 
    { 
        if (map.containsKey(USERNAME)) 
        { 
            Collection<? extends GrantedAuthority> authorities = getAuthorities(map); 
 
            Long userId = Convert.toLong(map.get(SecurityConstants.DETAILS_USER_ID)); 
            String username = (String) map.get(SecurityConstants.DETAILS_USERNAME); 
            LoginUser user = new LoginUser(userId, username, N_A, true, true, true, true, authorities); 
            return new UsernamePasswordAuthenticationToken(user, N_A, authorities); 
        } 
        return null; 
    } 
 
    /** 
     * 获取权限资源信息 
     */ 
    private Collection<? extends GrantedAuthority> getAuthorities(Map<String, ?> map) 
    { 
        Object authorities = map.get(AUTHORITIES); 
        if (authorities instanceof String) 
        { 
            return AuthorityUtils.commaSeparatedStringToAuthorityList((String) authorities); 
        } 
        if (authorities instanceof Collection) 
        { 
            return AuthorityUtils.commaSeparatedStringToAuthorityList( 
                    StringUtils.collectionToCommaDelimitedString((Collection<?>) authorities)); 
        } 
        throw new IllegalArgumentException("Authorities must be either a String or a Collection"); 
    } 
}