aboutsummaryrefslogtreecommitdiff
path: root/juick-server/src/main/java/com/juick/service/security/SimpleRememberMeServices.java
blob: d5d5400515228d236f64b216812b558cf1dc7e75 (plain) (blame)
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
89
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.UserDetailsService;
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 UserDetailsService userDetailsService, final UserService userService, final Environment environment) {
        super(key, userDetailsService);

        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());
    }
}