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
76
77
78
79
80
81
82
83
84
85
86
87
88
|
package com.juick.service.security;
import com.juick.User;
import com.juick.server.security.entities.JuickUser;
import com.juick.service.UserService;
import com.juick.util.UserUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.core.env.Environment;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.security.web.authentication.RememberMeServices;
import org.springframework.security.web.authentication.rememberme.AbstractRememberMeServices;
import org.springframework.security.web.authentication.rememberme.InvalidCookieException;
import org.springframework.security.web.authentication.rememberme.RememberMeAuthenticationException;
import org.springframework.util.Assert;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Optional;
/**
* Created by aalexeev on 11/28/16.
*/
public class SimpleRememberMeServices extends AbstractRememberMeServices implements RememberMeServices {
private final UserService userService;
public SimpleRememberMeServices(
final String key, final UserService userService, final Environment environment) {
super(key, new NullUserDetailsService());
Assert.notNull(userService);
Assert.notNull(environment);
this.userService = userService;
setCookieName(environment.getProperty("auth_cookie_name", "hash"));
setCookieDomain(environment.getProperty("web_domain", "juick.com"));
}
@Override
public void logout(HttpServletRequest request, HttpServletResponse response, Authentication authentication) {
super.logout(request, response, authentication);
userService.deleteLoginForUser(authentication.getName());
}
@Override
protected void onLoginSuccess(
HttpServletRequest request, HttpServletResponse response, Authentication successfulAuthentication) {
String username = successfulAuthentication.getName();
logger.debug("Creating new persistent login for user " + username);
try {
int uid = userService.getUIDbyName(username);
Assert.isTrue(uid > 0);
String hash = UserUtils.generateHash(16);
userService.setLoginForUser(uid, hash);
setCookie(new String[]{hash}, getTokenValiditySeconds(), request, response);
} catch (Exception e) {
logger.error("Failed to save cookies ", e);
}
}
@Override
protected UserDetails processAutoLoginCookie(
String[] cookieTokens, HttpServletRequest request, HttpServletResponse response)
throws RememberMeAuthenticationException, UsernameNotFoundException {
String hash = cookieTokens[0];
if (StringUtils.isBlank(hash))
throw new InvalidCookieException("Cookie is invalid, cookies " + cookieTokens);
int uid = userService.getUIDbyHash(cookieTokens[0]);
if (uid <= 0)
throw new UsernameNotFoundException("User not found bash hash, cookies" + cookieTokens);
Optional<User> userOptional = userService.getUserByUID(uid);
Assert.isTrue(userOptional.isPresent());
return new JuickUser(userOptional.get());
}
}
|
