From a8353b654b1fc65a73261f8f3825bfc2204be7c1 Mon Sep 17 00:00:00 2001
From: thinkgem <thinkgem@163.com>
Date: Sat, 22 Jun 2024 09:05:53 +0800
Subject: [PATCH] update sso api success url

---
 .../java/com/jeesite/common/shiro/filter/FormFilter.java     | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/core/src/main/java/com/jeesite/common/shiro/filter/FormFilter.java b/modules/core/src/main/java/com/jeesite/common/shiro/filter/FormFilter.java
index 74f6b31d..9375c9cf 100644
--- a/modules/core/src/main/java/com/jeesite/common/shiro/filter/FormFilter.java
+++ b/modules/core/src/main/java/com/jeesite/common/shiro/filter/FormFilter.java
@@ -302,6 +302,11 @@ public class FormFilter extends org.apache.shiro.web.filter.authc.FormAuthentica
 		String successUrl = getSuccessUrl(); // shiro.successUrl in application.yml
 		if (StringUtils.contains((request).getRequestURI(), "/oauth2/callback/")) {
 			successUrl = Global.getConfig("oauth2.successUrl", successUrl);
+		} else if (StringUtils.contains((request).getRequestURI(), "/sso")) {
+			String ssoSuccessUrl = (String)request.getAttribute("__url");
+			if (StringUtils.isNotBlank(ssoSuccessUrl)) {
+				successUrl = ssoSuccessUrl;
+			}
 		}
 		ServletUtils.redirectUrl(request, (HttpServletResponse)response, successUrl);
 		return false;