iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #132 from Doha2012/master 

small modifications
This commit is contained in:
Eugen 2015-02-09 19:05:14 +02:00
parent 5a9dc51ea7 326695b056
commit 7539be6c78
2 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spring-security-login-and-registration/src/main/java/org/baeldung/registration/listener/RegistrationListener.java
View File

@ -40,7 +40,7 @@ public class RegistrationListener implements ApplicationListener<OnRegistrationC
SimpleMailMessage email = new SimpleMailMessage(); SimpleMailMessage email = new SimpleMailMessage();
email.setTo(recipientAddress); email.setTo(recipientAddress);
email.setSubject(subject); email.setSubject(subject);
email.setText(message + " \r\n" + "http://localhost:8080" + confirmationUrl); email.setText(message + " \r\n" + confirmationUrl);
mailSender.send(email); mailSender.send(email);
} }
} }

32
spring-security-login-and-registration/src/main/java/org/baeldung/web/controller/RegistrationController.java
View File

@ -4,6 +4,7 @@ import java.util.Calendar;
import java.util.Locale; import java.util.Locale;
import java.util.UUID; import java.util.UUID;
import javax.servlet.http.HttpServletRequest;
import javax.validation.Valid; import javax.validation.Valid;
import org.baeldung.persistence.model.PasswordResetToken; import org.baeldung.persistence.model.PasswordResetToken;
@ -34,7 +35,6 @@ import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.context.request.WebRequest;
import org.springframework.web.servlet.ModelAndView; import org.springframework.web.servlet.ModelAndView;
@Controller @Controller
@ -63,7 +63,7 @@ public class RegistrationController {
// API // API
@RequestMapping(value = "/user/registration", method = RequestMethod.GET) @RequestMapping(value = "/user/registration", method = RequestMethod.GET)
public String showRegistrationForm(final WebRequest request, final Model model) { public String showRegistrationForm(final HttpServletRequest request, final Model model) {
LOGGER.debug("Rendering registration page."); LOGGER.debug("Rendering registration page.");
final UserDto accountDto = new UserDto(); final UserDto accountDto = new UserDto();
model.addAttribute("user", accountDto); model.addAttribute("user", accountDto);
@ -71,7 +71,7 @@ public class RegistrationController {
} }
@RequestMapping(value = "/regitrationConfirm", method = RequestMethod.GET) @RequestMapping(value = "/regitrationConfirm", method = RequestMethod.GET)
public String confirmRegistration(final WebRequest request, final Model model, @RequestParam("token") final String token) { public String confirmRegistration(final HttpServletRequest request, final Model model, @RequestParam("token") final String token) {
final Locale locale = request.getLocale(); final Locale locale = request.getLocale();
final VerificationToken verificationToken = userService.getVerificationToken(token); final VerificationToken verificationToken = userService.getVerificationToken(token);
@ -83,7 +83,6 @@ public class RegistrationController {
final User user = verificationToken.getUser(); final User user = verificationToken.getUser();
final Calendar cal = Calendar.getInstance(); final Calendar cal = Calendar.getInstance();
if ((verificationToken.getExpiryDate().getTime() - cal.getTime().getTime()) <= 0) { if ((verificationToken.getExpiryDate().getTime() - cal.getTime().getTime()) <= 0) {
model.addAttribute("message", messages.getMessage("auth.message.expired", null, locale)); model.addAttribute("message", messages.getMessage("auth.message.expired", null, locale));
model.addAttribute("expired", true); model.addAttribute("expired", true);
@ -97,7 +96,7 @@ public class RegistrationController {
} }
@RequestMapping(value = "/user/registration", method = RequestMethod.POST) @RequestMapping(value = "/user/registration", method = RequestMethod.POST)
public ModelAndView registerUserAccount(@ModelAttribute("user") @Valid final UserDto accountDto, final BindingResult result, final WebRequest request, final Errors errors) { public ModelAndView registerUserAccount(@ModelAttribute("user") @Valid final UserDto accountDto, final BindingResult result, final HttpServletRequest request, final Errors errors) {
LOGGER.debug("Registering user account with information: {}", accountDto); LOGGER.debug("Registering user account with information: {}", accountDto);
if (result.hasErrors()) { if (result.hasErrors()) {
return new ModelAndView("registration", "user", accountDto); return new ModelAndView("registration", "user", accountDto);
@ -106,9 +105,10 @@ public class RegistrationController {
final User registered = createUserAccount(accountDto); final User registered = createUserAccount(accountDto);
if (registered == null) { if (registered == null) {
result.rejectValue("email", "message.regError"); result.rejectValue("email", "message.regError");
return new ModelAndView("registration", "user", accountDto);
} }
try { try {
final String appUrl = request.getContextPath(); final String appUrl = request.getServerName() + ":" + request.getServerPort() + request.getContextPath();
eventPublisher.publishEvent(new OnRegistrationCompleteEvent(registered, request.getLocale(), appUrl)); eventPublisher.publishEvent(new OnRegistrationCompleteEvent(registered, request.getLocale(), appUrl));
} catch (final Exception ex) { } catch (final Exception ex) {
LOGGER.warn("Unable to register user", ex); LOGGER.warn("Unable to register user", ex);
@ -118,12 +118,13 @@ public class RegistrationController {
} }
@RequestMapping(value = "/user/resendRegistrationToken", method = RequestMethod.GET) @RequestMapping(value = "/user/resendRegistrationToken", method = RequestMethod.GET)
public String resendRegistrationToken(final WebRequest request, final Model model, @RequestParam("token") final String existingToken) { public String resendRegistrationToken(final HttpServletRequest request, final Model model, @RequestParam("token") final String existingToken) {
final Locale locale = request.getLocale(); final Locale locale = request.getLocale();
final VerificationToken newToken = userService.generateNewVerificationToken(existingToken); final VerificationToken newToken = userService.generateNewVerificationToken(existingToken);
final User user = userService.getUser(newToken.getToken()); final User user = userService.getUser(newToken.getToken());
try { try {
final SimpleMailMessage email = constructResetVerificationTokenEmail(request.getContextPath(), request.getLocale(), newToken, user); final String appUrl = request.getServerName() + ":" + request.getServerPort() + request.getContextPath();
final SimpleMailMessage email = constructResetVerificationTokenEmail(appUrl, request.getLocale(), newToken, user);
mailSender.send(email); mailSender.send(email);
} catch (final MailAuthenticationException e) { } catch (final MailAuthenticationException e) {
LOGGER.debug("MailAuthenticationException", e); LOGGER.debug("MailAuthenticationException", e);
@ -138,7 +139,7 @@ public class RegistrationController {
} }
@RequestMapping(value = "/user/resetPassword", method = RequestMethod.POST) @RequestMapping(value = "/user/resetPassword", method = RequestMethod.POST)
public String resetPassword(final WebRequest request, final Model model, @RequestParam("email") final String userEmail) { public String resetPassword(final HttpServletRequest request, final Model model, @RequestParam("email") final String userEmail) {
final User user = userService.findUserByEmail(userEmail); final User user = userService.findUserByEmail(userEmail);
if (user == null) { if (user == null) {
model.addAttribute("message", messages.getMessage("message.userNotFound", null, request.getLocale())); model.addAttribute("message", messages.getMessage("message.userNotFound", null, request.getLocale()));
@ -148,9 +149,8 @@ public class RegistrationController {
final String token = UUID.randomUUID().toString(); final String token = UUID.randomUUID().toString();
userService.createPasswordResetTokenForUser(user, token); userService.createPasswordResetTokenForUser(user, token);
try { try {
final SimpleMailMessage email = constructResetTokenEmail(request.getContextPath(), request.getLocale(), token, user); final String appUrl = request.getServerName() + ":" + request.getServerPort() + request.getContextPath();
final SimpleMailMessage email = constructResetTokenEmail(appUrl, request.getLocale(), token, user);
LOGGER.debug(email.getText());
mailSender.send(email); mailSender.send(email);
} catch (final MailAuthenticationException e) { } catch (final MailAuthenticationException e) {
LOGGER.debug("MailAuthenticationException", e); LOGGER.debug("MailAuthenticationException", e);
@ -165,7 +165,7 @@ public class RegistrationController {
} }
@RequestMapping(value = "/user/changePassword", method = RequestMethod.GET) @RequestMapping(value = "/user/changePassword", method = RequestMethod.GET)
public String changePassword(final WebRequest request, final Model model, @RequestParam("id") final long id, @RequestParam("token") final String token) { public String changePassword(final HttpServletRequest request, final Model model, @RequestParam("id") final long id, @RequestParam("token") final String token) {
final Locale locale = request.getLocale(); final Locale locale = request.getLocale();
final PasswordResetToken passToken = userService.getPasswordResetToken(token); final PasswordResetToken passToken = userService.getPasswordResetToken(token);
@ -190,7 +190,7 @@ public class RegistrationController {
@RequestMapping(value = "/user/savePassword", method = RequestMethod.POST) @RequestMapping(value = "/user/savePassword", method = RequestMethod.POST)
@PreAuthorize("hasRole('READ_PRIVILEGE')") @PreAuthorize("hasRole('READ_PRIVILEGE')")
public String savePassword(final WebRequest request, final Model model, @RequestParam("password") final String password) { public String savePassword(final HttpServletRequest request, final Model model, @RequestParam("password") final String password) {
final Locale locale = request.getLocale(); final Locale locale = request.getLocale();
final User user = (User) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); final User user = (User) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
@ -206,7 +206,7 @@ public class RegistrationController {
final String message = messages.getMessage("message.resendToken", null, locale); final String message = messages.getMessage("message.resendToken", null, locale);
final SimpleMailMessage email = new SimpleMailMessage(); final SimpleMailMessage email = new SimpleMailMessage();
email.setSubject("Resend Registration Token"); email.setSubject("Resend Registration Token");
email.setText(message + " \r\n" + "http://localhost:8080" + confirmationUrl); email.setText(message + " \r\n" + confirmationUrl);
email.setTo(user.getEmail()); email.setTo(user.getEmail());
return email; return email;
} }
@ -217,7 +217,7 @@ public class RegistrationController {
final SimpleMailMessage email = new SimpleMailMessage(); final SimpleMailMessage email = new SimpleMailMessage();
email.setTo(user.getEmail()); email.setTo(user.getEmail());
email.setSubject("Reset Password"); email.setSubject("Reset Password");
email.setText(message + " \r\n" + "http://localhost:8080" + url); email.setText(message + " \r\n" + url);
return email; return email;
} }