fixes #4

Author: xgp

src/main/java/io/phasetwo/keycloak/magic/MagicLink.java

@@ -114,7 +114,8 @@ public static boolean sendMagicLinkEmail(KeycloakSession session, UserModel user
     try {
       EmailTemplateProvider emailTemplateProvider =
           session.getProvider(EmailTemplateProvider.class);
-      String realmName = Strings.isNullOrEmpty(realm.getDisplayName()) ? realm.getName() : realm.getDisplayName();
+      String realmName =
+          Strings.isNullOrEmpty(realm.getDisplayName()) ? realm.getName() : realm.getDisplayName();
       List<Object> subjAttr = ImmutableList.of(realmName);
       Map<String, Object> bodyAttr = Maps.newHashMap();
       bodyAttr.put("realmName", realmName);

src/main/java/io/phasetwo/keycloak/magic/auth/MagicLinkAuthenticator.java

@@ -1,6 +1,7 @@
 package io.phasetwo.keycloak.magic.auth;
 
 import io.phasetwo.keycloak.magic.MagicLink;
+import io.phasetwo.keycloak.magic.auth.token.MagicLinkActionToken;
 import java.util.List;
 import java.util.Map;
 import java.util.OptionalInt;
@@ -9,6 +10,7 @@
 import lombok.extern.jbosslog.JBossLog;
 import org.keycloak.authentication.AuthenticationFlowContext;
 import org.keycloak.authentication.Authenticator;
+import org.keycloak.authentication.authenticators.browser.AbstractUsernameFormAuthenticator;
 import org.keycloak.authentication.authenticators.browser.UsernamePasswordForm;
 import org.keycloak.events.EventBuilder;
 import org.keycloak.forms.login.LoginFormsProvider;
@@ -17,7 +19,6 @@
 import org.keycloak.models.IdentityProviderModel;
 import org.keycloak.models.UserModel;
 import org.keycloak.services.messages.Messages;
-import org.keycloak.authentication.authenticators.browser.AbstractUsernameFormAuthenticator;
 
 @JBossLog
 public class MagicLinkAuthenticator extends UsernamePasswordForm implements Authenticator {
@@ -60,9 +61,11 @@ public void action(AuthenticationFlowContext context) {
     boolean sent = MagicLink.sendMagicLinkEmail(context.getSession(), user, link);
     log.debugf("sent email to %s? %b. Link? %s", user.getEmail(), sent, link);
 
-    //context.setUser(user);
+    // context.setUser(user);
     context.clearUser(); // just in case
-    context.setAuthNote(AbstractUsernameFormAuthenticator.ATTEMPTED_USERNAME, email);
+    context
+        .getAuthenticationSession()
+        .setAuthNote(AbstractUsernameFormAuthenticator.ATTEMPTED_USERNAME, email);
     context.challenge(context.form().createForm("view-email.ftl"));
   }