Customize logout flow and insert the pre-init action

Author: cslzchen

src/main/java/io/cos/cas/osf/web/flow/config/OsfCasWebflowContextConfiguration.java

@@ -1,6 +1,7 @@
 package io.cos.cas.osf.web.flow.config;
 
 import io.cos.cas.osf.web.flow.configurer.OsfCasLoginWebflowConfigurer;
+import io.cos.cas.osf.web.flow.configurer.OsfCasLogoutWebFlowConfigurer;
 
 import org.apereo.cas.configuration.CasConfigurationProperties;
 import org.apereo.cas.web.flow.CasWebflowConfigurer;
@@ -46,4 +47,16 @@ public CasWebflowConfigurer defaultWebflowConfigurer() {
         osfCasLoginWebflowConfigurer.setOrder(Ordered.HIGHEST_PRECEDENCE);
         return osfCasLoginWebflowConfigurer;
     }
+
+    @Override
+    @Bean
+    @Order(DEFAULT_WEB_FLOW_CONFIGURER_ORDER)
+    @Lazy(false)
+    public CasWebflowConfigurer defaultLogoutWebflowConfigurer() {
+        OsfCasLogoutWebFlowConfigurer osfCasLogoutWebFlowConfigurer
+                = new OsfCasLogoutWebFlowConfigurer(builder(), loginFlowRegistry(), applicationContext, casProperties);
+        osfCasLogoutWebFlowConfigurer.setLogoutFlowDefinitionRegistry(logoutFlowRegistry());
+        osfCasLogoutWebFlowConfigurer.setOrder(Ordered.HIGHEST_PRECEDENCE);
+        return osfCasLogoutWebFlowConfigurer;
+    }
 }

src/main/java/io/cos/cas/osf/web/flow/configurer/OsfCasLogoutWebFlowConfigurer.java

@@ -0,0 +1,49 @@
+package io.cos.cas.osf.web.flow.configurer;
+
+import io.cos.cas.osf.web.flow.support.OsfCasWebflowConstants;
+
+import org.apereo.cas.configuration.CasConfigurationProperties;
+import org.apereo.cas.web.flow.configurer.DefaultLogoutWebflowConfigurer;
+
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.webflow.definition.registry.FlowDefinitionRegistry;
+import org.springframework.webflow.engine.ActionList;
+import org.springframework.webflow.engine.Flow;
+import org.springframework.webflow.engine.builder.support.FlowBuilderServices;
+
+/**
+ * This is {@link OsfCasLogoutWebFlowConfigurer}.
+ *
+ * @author Longze Chen
+ * @since 20.2.0
+ */
+public class OsfCasLogoutWebFlowConfigurer extends DefaultLogoutWebflowConfigurer {
+
+    public OsfCasLogoutWebFlowConfigurer(
+            final FlowBuilderServices flowBuilderServices,
+            final FlowDefinitionRegistry flowDefinitionRegistry,
+            final ConfigurableApplicationContext applicationContext,
+            final CasConfigurationProperties casProperties
+    ) {
+        super(flowBuilderServices, flowDefinitionRegistry, applicationContext, casProperties);
+    }
+
+    @Override
+    protected void doInitialize() {
+        final Flow flow = getLogoutFlow();
+        if (flow != null) {
+            createInitialFlowActions(flow);
+        }
+        super.doInitialize();
+    }
+
+    /**
+     * Create initial flow actions similar to {@link OsfCasLoginWebflowConfigurer#createInitialFlowActions(Flow)}.
+     *
+     * @param flow the flow
+     */
+    protected void createInitialFlowActions(final Flow flow) {
+        final ActionList startActionList = flow.getStartActionList();
+        startActionList.add(createEvaluateAction(OsfCasWebflowConstants.ACTION_ID_OSF_PRE_INITIAL_FLOW_SETUP));
+    }
+}