Merge pull request #1245 from stormpath/1242_idsite_prepost_handlers

1242 idsite pre and post handlers

Author: mrioan

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/config/filter/IDSiteFilterFactory.java renamed to extensions/servlet/src/main/java/com/stormpath/sdk/servlet/config/filter/AbstractIDSiteFilterFactory.java

@@ -26,15 +26,13 @@
 /**
  * @since 1.0.0
  */
-public abstract class IDSiteFilterFactory extends ControllerFilterFactory<IdSiteController> {
+public abstract class AbstractIDSiteFilterFactory<T extends IdSiteController> extends ControllerFilterFactory<T> {
 
     @Override
-    protected IdSiteController newController() {
-        return new IdSiteController();
-    }
+    protected abstract T newController();
 
     @Override
-    protected void configure(IdSiteController controller, Config config) throws Exception {
+    protected void configure(T controller, Config config) throws Exception {
 
         SubdomainResolver subdomainResolver = new SubdomainResolver();
         subdomainResolver.setBaseDomainName(config.get("stormpath.web.application.domain"));
@@ -58,5 +56,5 @@ protected void configure(IdSiteController controller, Config config) throws Exce
         doConfigure(controller, config);
     }
 
-    public abstract void doConfigure(IdSiteController controller, Config config);
+    public abstract void doConfigure(T controller, Config config);
 }

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/config/filter/IDSiteForgotFilterFactory.java

@@ -21,7 +21,7 @@
 /**
  * @since 1.0.0
  */
-public class IDSiteForgotFilterFactory extends IDSiteFilterFactory {
+public class IDSiteForgotFilterFactory extends AbstractIDSiteFilterFactory<IdSiteController> {
 
     @Override
     protected IdSiteController newController() {

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/config/filter/IDSiteLoginFilterFactory.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2016 Stormpath, Inc.
+ * Copyright 2017 Stormpath, Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -17,19 +17,21 @@
 
 import com.stormpath.sdk.servlet.config.Config;
 import com.stormpath.sdk.servlet.mvc.IdSiteController;
+import com.stormpath.sdk.servlet.mvc.IdSiteLoginController;
 
 /**
- * @since 1.0.0
+ * @since 1.5.0
  */
-public class IDSiteLoginFilterFactory extends IDSiteFilterFactory {
+public class IDSiteLoginFilterFactory extends AbstractIDSiteFilterFactory<IdSiteLoginController> {
 
     @Override
-    protected IdSiteController newController() {
-        return new IdSiteController();
+    protected IdSiteLoginController newController() {
+        return new IdSiteLoginController();
     }
 
-    public void doConfigure(IdSiteController controller, Config config) {
+    public void doConfigure(IdSiteLoginController controller, Config config) {
         controller.setIdSiteUri(config.get("stormpath.web.idSite.loginUri"));
         controller.setNextUri(config.get("stormpath.web.login.nextUri"));
+        controller.setPreLoginHandler(config.getLoginPreHandler());
     }
 }

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/config/filter/IDSiteRegisterFilterFactory.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2016 Stormpath, Inc.
+ * Copyright 2017 Stormpath, Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,22 +16,23 @@
 package com.stormpath.sdk.servlet.config.filter;
 
 import com.stormpath.sdk.servlet.config.Config;
-import com.stormpath.sdk.servlet.mvc.IdSiteController;
+import com.stormpath.sdk.servlet.mvc.IdSiteRegisterController;
 
 /**
- * @since 1.0.0
+ * @since 1.5.0
  */
-public class IDSiteRegisterFilterFactory extends IDSiteFilterFactory {
+public class IDSiteRegisterFilterFactory extends AbstractIDSiteFilterFactory<IdSiteRegisterController> {
 
     @Override
-    protected IdSiteController newController() {
-        return new IdSiteController();
+    protected IdSiteRegisterController newController() {
+        return new IdSiteRegisterController();
     }
 
     @Override
-    public void doConfigure(IdSiteController controller, Config config) {
+    public void doConfigure(IdSiteRegisterController controller, Config config) {
         controller.setIdSiteUri(config.get("stormpath.web.idSite.registerUri"));
         controller.setNextUri(config.get("stormpath.web.register.nextUri"));
+        controller.setPreRegisterHandler(config.getRegisterPreHandler());
     }
 
 }

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/config/filter/IDSiteResultFilterFactory.java

@@ -31,6 +31,8 @@ protected IdSiteResultController newController() {
     @Override
     public void doConfigure(IdSiteResultController controller, Config config) {
         controller.setRegisterNextUri(config.getRegisterConfig().getNextUri());
+        controller.setPostRegisterHandler(config.getRegisterPostHandler());
+        controller.setPostLoginHandler(config.getLoginPostHandler());
     }
 
 }

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/mvc/CallbackController.java

@@ -19,7 +19,6 @@
 import com.stormpath.sdk.authc.AuthenticationResult;
 import com.stormpath.sdk.idsite.LogoutResult;
 import com.stormpath.sdk.lang.Assert;
-import com.stormpath.sdk.lang.Strings;
 import com.stormpath.sdk.saml.SamlResultListener;
 import com.stormpath.sdk.servlet.application.ApplicationResolver;
 import com.stormpath.sdk.servlet.authc.impl.DefaultSuccessfulAuthenticationRequestEvent;
@@ -43,6 +42,8 @@ public abstract class CallbackController extends AbstractController {
     protected Saver<AuthenticationResult> authenticationResultSaver;
     protected Publisher<RequestEvent> eventPublisher;
     protected List<SamlResultListener> samlResultListeners = new ArrayList<SamlResultListener>();
+    protected WebHandler postLoginHandler;
+    protected WebHandler postRegisterHandler;
 
     public void setLoginNextUri(String loginNextUri) {
         this.loginNextUri = loginNextUri;
@@ -106,6 +107,10 @@ protected ViewModel onAuthentication(HttpServletRequest request,
 
         publish(new DefaultSuccessfulAuthenticationRequestEvent(request, response, null, authcResult));
 
+        if (postLoginHandler != null && !postLoginHandler.handle(request, response, result.getAccount())) {
+            return null;
+        }
+
         return new DefaultViewModel(loginNextUri).setRedirect(true);
     }
 

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/mvc/IdSiteLoginController.java

@@ -0,0 +1,43 @@
+/*
+ * Copyright 2017 Stormpath, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.stormpath.sdk.servlet.mvc;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+/**
+ * @since 1.5.0
+ */
+public class IdSiteLoginController extends IdSiteController {
+
+    protected WebHandler preLoginHandler;
+
+    public void setPreLoginHandler(WebHandler preLoginHandler) {
+        this.preLoginHandler = preLoginHandler;
+    }
+
+    @Override
+    protected ViewModel doGet(HttpServletRequest request, HttpServletResponse response) throws Exception {
+
+        if (preLoginHandler != null) {
+            if (!preLoginHandler.handle(request, response, null)) {
+                return null;
+            }
+        }
+
+        return super.doGet(request, response);
+    }
+}

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/mvc/IdSiteRegisterController.java

@@ -0,0 +1,43 @@
+/*
+ * Copyright 2017 Stormpath, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.stormpath.sdk.servlet.mvc;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+/**
+ * @since 1.5.0
+ */
+public class IdSiteRegisterController extends IdSiteController {
+
+    protected WebHandler preRegisterHandler;
+
+    public void setPreRegisterHandler(WebHandler preRegisterHandler) {
+        this.preRegisterHandler = preRegisterHandler;
+    }
+
+    @Override
+    protected ViewModel doGet(HttpServletRequest request, HttpServletResponse response) throws Exception {
+
+        if (preRegisterHandler != null) {
+            if (!preRegisterHandler.handle(request, response, null)) {
+                return null;
+            }
+        }
+
+        return super.doGet(request, response);
+    }
+}

extensions/servlet/src/main/java/com/stormpath/sdk/servlet/mvc/IdSiteResultController.java

@@ -20,6 +20,7 @@
 import com.stormpath.sdk.application.Application;
 import com.stormpath.sdk.authc.AuthenticationResult;
 import com.stormpath.sdk.client.Client;
+import com.stormpath.sdk.idsite.AccountResult;
 import com.stormpath.sdk.idsite.IdSiteCallbackHandler;
 import com.stormpath.sdk.idsite.IdSiteResultListener;
 import com.stormpath.sdk.idsite.LogoutResult;
@@ -106,7 +107,7 @@ public void onLogout(LogoutResult result) {
             idSiteCallbackHandler.addResultListener(resultListener);
         }
 
-        idSiteCallbackHandler.getAccountResult();
+        AccountResult accountResult = idSiteCallbackHandler.getAccountResult();
 
         return viewModel[0];
     }
@@ -126,7 +127,12 @@ private ViewModel onRegistration(final HttpServletRequest request, final HttpSer
             AuthenticationResult authcResult = new TransientAuthenticationResult(account);
             saveResult(request, response, authcResult);
         }
-        // else - do we need to do anything else?
+
+        if (postRegisterHandler != null) {
+            if (!postRegisterHandler.handle(request, response, account)) {
+                return null;
+            }
+        }
 
         //just redirect to post-register view:
         return new DefaultViewModel(registerNextUri).setRedirect(true);
@@ -136,4 +142,12 @@ private RegisteredAccountRequestEvent createRegisteredEvent(HttpServletRequest r
                                                                 HttpServletResponse response, Account account) {
         return new DefaultRegisteredAccountRequestEvent(request, response, account);
     }
+
+    public void setPostRegisterHandler(WebHandler postRegisterHandler) {
+        this.postRegisterHandler = postRegisterHandler;
+    }
+
+    public void setPostLoginHandler(WebHandler postLoginHandler) {
+        this.postLoginHandler = postLoginHandler;
+    }
 }

extensions/spring/boot/stormpath-webmvc-spring-boot-starter/src/test/groovy/com/stormpath/spring/boot/autoconfigure/StormpathWebSecurityAutoConfigurationIT.groovy

@@ -17,27 +17,62 @@ package com.stormpath.spring.boot.autoconfigure
 
 import autoconfigure.StormpathWebSecurityAutoConfigurationTestApplication
 import com.stormpath.sdk.servlet.csrf.CsrfTokenManager
+import com.stormpath.sdk.servlet.filter.StormpathFilter
+import com.stormpath.sdk.servlet.mvc.Controller
+import com.stormpath.sdk.servlet.mvc.IdSiteLoginController
+import com.stormpath.sdk.servlet.mvc.LoginController
 import com.stormpath.spring.config.TwoAppTenantStormpathTestConfiguration
 import org.springframework.beans.factory.annotation.Autowired
+import org.springframework.beans.factory.annotation.Qualifier
+import org.springframework.beans.factory.annotation.Value
 import org.springframework.boot.test.context.SpringBootTest
+import org.springframework.test.context.TestPropertySource
 import org.springframework.test.context.testng.AbstractTestNGSpringContextTests
 import org.springframework.test.context.web.WebAppConfiguration
 import org.testng.annotations.Test
 
 import static org.testng.Assert.assertEquals
+import static org.testng.Assert.assertNotNull
+import static org.testng.Assert.assertTrue
 
 /**
  * @since 1.0.RC5.2
  */
 @SpringBootTest(classes = [StormpathWebSecurityAutoConfigurationTestApplication.class, TwoAppTenantStormpathTestConfiguration.class])
 @WebAppConfiguration
+@TestPropertySource(locations = "classpath:it.application.properties")
 class StormpathWebSecurityAutoConfigurationIT extends AbstractTestNGSpringContextTests {
 
     @Autowired
     CsrfTokenManager csrfTokenManager
 
+    @Autowired
+    @Qualifier("stormpathLoginController")
+    Controller login
+
+    @Autowired
+    @Qualifier("stormpathRegisterController")
+    Controller register
+
+    @Autowired
+    @Qualifier("stormpathIdSiteResultController")
+    Controller idSiteResultController
+
+    @Value("#{ @environment['stormpath.web.idSite.enabled'] ?: false }")
+    protected boolean idSiteEnabled;
+
     @Test
     void testCsrfTokenManager() {
         assertEquals csrfTokenManager.tokenName, 'csrfToken'
     }
+
+    //asserts https://github.com/stormpath/stormpath-sdk-java/issues/1242
+    @Test
+    void assertPrePostHandlersArePresentWhenIDSiteIsEnabled() {
+        assertTrue idSiteEnabled
+        assertTrue(register.preRegisterHandler instanceof StormpathWebSecurityAutoConfigurationTestApplication.CustomRegisterPreHandler)
+        assertTrue(idSiteResultController.postRegisterHandler instanceof StormpathWebSecurityAutoConfigurationTestApplication.CustomRegisterPostHandler)
+        assertTrue(login.preLoginHandler instanceof StormpathWebSecurityAutoConfigurationTestApplication.CustomLoginPreHandler)
+        assertTrue(idSiteResultController.postLoginHandler instanceof StormpathWebSecurityAutoConfigurationTestApplication.CustomLoginPostHandler)
+    }
 }