Email Confirmation Article

All Changes as of Video Nov 22

Author: egmp777

spring-security-login-and-registration/src/main/java/org/baeldung/event/OnRegistrationComplete.java

@@ -0,0 +1,34 @@
+package org.baeldung.event;
+
+import java.util.Locale;
+
+import org.baeldung.persistence.model.User;
+import org.springframework.context.ApplicationEvent;
+
+@SuppressWarnings("serial")
+public class OnRegistrationCompleteEvent extends ApplicationEvent {
+
+    private final String appUrl;
+    private final Locale locale;
+    private final User user;
+    
+    public OnRegistrationCompleteEvent(User user, Locale locale, String appUrl) {
+        super(user);
+        this.user = user;
+        this.locale = locale;
+        this.appUrl = appUrl;     
+    }
+
+    public String getAppUrl() {
+        return appUrl;
+    }
+
+    public Locale getLocale() {
+        return locale;
+    }
+
+    public User getUser() {
+        return user;
+    }
+
+}

spring-security-login-and-registration/src/main/java/org/baeldung/event/OnRegistrationCompleteEvent.java

@@ -0,0 +1,44 @@
+package org.baeldung.event.listener;
+
+import java.util.UUID;
+import org.baeldung.event.OnRegistrationCompleteEvent;
+import org.baeldung.persistence.model.User;
+import org.baeldung.persistence.service.IUserService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationListener;
+import org.springframework.context.MessageSource;
+import org.springframework.mail.SimpleMailMessage;
+import org.springframework.mail.javamail.JavaMailSender;
+import org.springframework.stereotype.Component;
+
+@Component
+public class RegistrationListener implements ApplicationListener<OnRegistrationCompleteEvent> {
+    @Autowired
+    private IUserService service;
+    
+    @Autowired
+    private MessageSource messages;
+
+    @Autowired
+    private JavaMailSender mailSender;
+
+    @Override
+    public void onApplicationEvent(OnRegistrationCompleteEvent event) {
+        this.confirmRegistration(event);
+    }
+
+    private void confirmRegistration(OnRegistrationCompleteEvent event) {
+        User user = event.getUser();
+        String token = UUID.randomUUID().toString();
+        service.addVerificationToken(user, token);
+        String recipientAddress = user.getEmail();
+        String subject = "Registration Confirmation";
+        String confirmationUrl = event.getAppUrl() + "/regitrationConfirm.html?token=" + token;
+        String message = messages.getMessage("message.regSucc", null, event.getLocale());
+        SimpleMailMessage email = new SimpleMailMessage();
+        email.setTo(recipientAddress);
+        email.setSubject(subject);
+        email.setText(message + " \r\n" + "http://localhost:8080" + confirmationUrl);
+        mailSender.send(email);
+    }
+}

spring-security-login-and-registration/src/main/java/org/baeldung/event/Registration.java

@@ -4,7 +4,8 @@
 import org.baeldung.persistence.model.User;
 
 public interface UserRepository extends JpaRepository<User, Long> {
-	public User findByEmail(String email);
+    public User findByEmail(String email);
+
+    public void delete(User user);
 
-	public void delete(User user);
 }

spring-security-login-and-registration/src/main/java/org/baeldung/event/listener/RegistrationListener.java

@@ -1,9 +1,12 @@
 package org.baeldung.persistence.dao;
 
+import org.baeldung.persistence.model.User;
 import org.baeldung.persistence.model.VerificationToken;
 import org.springframework.data.jpa.repository.JpaRepository;
 
 public interface VerificationTokenRepository extends JpaRepository<VerificationToken, Long> {
 
-	public VerificationToken findByToken(String token);
+    public VerificationToken findByToken(String token);
+
+    public VerificationToken findByUser(User user);
 }

spring-security-login-and-registration/src/main/java/org/baeldung/event/service/RegistrationService.java

@@ -11,8 +11,8 @@
 import javax.persistence.OneToOne;
 import javax.persistence.Table;
 
-@Entity(name = "role")
-@Table(name = "role")
+@Entity
+@Table
 public class Role {
 
     @Id
@@ -28,7 +28,6 @@ public class Role {
 
     public Role() {
         super();
-
     }
 
     public Role(Integer role) {

spring-security-login-and-registration/src/main/java/org/baeldung/persistence/dao/UserRepository.java

@@ -14,33 +14,27 @@
 @Table
 public class User {
 
-	@Id
-	@GeneratedValue(strategy = GenerationType.AUTO)
-	private Long id;
-	@Column(name = "firstName")
-	private String firstName;
-    @Column(name="lastName")
+    @Id
+    @GeneratedValue(strategy = GenerationType.AUTO)
+    private Long id;
+    
+    @Column(name = "firstName")
+    private String firstName;
+    
+    @Column(name = "lastName")
     private String lastName;
-    @Column(name="email")
+    
+    @Column(name = "email")
     private String email;
-    @Column(name="password")
+    
+    @Column(name = "password")
     private String password;
-   
-    @OneToOne(mappedBy="user",
-            fetch = FetchType.EAGER,
-            cascade= CascadeType.ALL)
-    private VerificationToken verificationToken;
-   
-    @OneToOne(mappedBy = "user",fetch = FetchType.EAGER, cascade = CascadeType.ALL)
-    private Role role;
-
-    public VerificationToken getVerificationToken() {
-        return verificationToken;
-    }
+    
+    @Column(name = "enabled")
+    private boolean enabled;
 
-    public void setVerificationToken(VerificationToken verificationToken) {
-        this.verificationToken = verificationToken;
-    }
+    @OneToOne(mappedBy = "user", fetch = FetchType.EAGER, cascade = CascadeType.ALL)
+    private Role role;
 
     public Long getId() {
         return id;
@@ -90,6 +84,14 @@ public void setRole(Role role) {
         this.role = role;
     }
 
+    public boolean isEnabled() {
+        return enabled;
+    }
+
+    public void setEnabled(boolean enabled) {
+        this.enabled = enabled;
+    }
+
     @Override
     public int hashCode() {
         final int prime = 31;
@@ -111,11 +113,11 @@ public boolean equals(final Object obj) {
             return false;
         return true;
     }
+
     @Override
     public String toString() {
         final StringBuilder builder = new StringBuilder();
-        builder.append("User [firstName=").append(firstName).append("]").
-            append("[lastName=").append(lastName).append("]").append("[username").append(email).append("]");
+        builder.append("User [firstName=").append(firstName).append("]").append("[lastName=").append(lastName).append("]").append("[username").append(email).append("]");
         return builder.toString();
     }
 }