refactor: Improve OAuth and SSO authorization flow with enhanced loading and error handling

This commit is contained in:
wood chen 2025-02-21 18:27:55 +08:00
parent 86adf32db8
commit 2f848af557
2 changed files with 24 additions and 35 deletions

View File

@ -26,7 +26,7 @@ export async function signIn(data: Record<string, any>) {
const state = params.get("state"); const state = params.get("state");
const scope = params.get("scope") || "read_profile"; const scope = params.get("scope") || "read_profile";
// 重新构建授权页面 URL // 重新构建授权页面 URL,使用相对路径
const authParams = new URLSearchParams({ const authParams = new URLSearchParams({
response_type: "code", response_type: "code",
client_id: clientId || "", client_id: clientId || "",
@ -37,14 +37,16 @@ export async function signIn(data: Record<string, any>) {
authParams.set("state", state); authParams.set("state", state);
} }
// 重定向到授权页面 // 重定向到授权页面,使用相对路径
redirect(`/oauth/authorize?${authParams.toString()}`); redirect(`/oauth/authorize?${authParams.toString()}`);
return;
} }
// 如果没有 OAuth 参数,重定向到仪表板 // 如果没有 OAuth 参数,重定向到仪表板
redirect("/dashboard"); redirect("/dashboard");
} catch (error) { } catch (error) {
console.error("登录失败:", error); console.error("登录失败:", error);
redirect("/sign-in?error=AuthenticationError"); // 添加时间戳防止循环
redirect(`/sign-in?error=AuthenticationError&t=${Date.now()}`);
} }
} }

View File

@ -20,22 +20,22 @@ export function UserAuthorize({
const router = useRouter(); const router = useRouter();
const searchParams = useSearchParams(); const searchParams = useSearchParams();
const signInCallback = useCallback(async () => { // 检查是否是 OAuth 授权页面
if (isLoading) { const isOAuthFlow =
return; searchParams?.has("client_id") && searchParams?.has("redirect_uri");
}
// 检查是否是 SSO 回调 // 检查是否是 SSO 回调
const sso = searchParams?.get("sso"); const isSSOCallback = searchParams?.has("sso") && searchParams?.has("sig");
const sig = searchParams?.get("sig");
// 如果没有 SSO 参数,说明这是 OAuth 授权页面,不需要处理 const signInCallback = useCallback(async () => {
if (!sso || !sig) { if (isLoading || !isSSOCallback || isOAuthFlow) {
return; return;
} }
setIsLoading(true); setIsLoading(true);
try { try {
const sso = searchParams?.get("sso");
const sig = searchParams?.get("sig");
// 传递 SSO 参数 // 传递 SSO 参数
await signIn({ await signIn({
sso, sso,
@ -50,31 +50,12 @@ export function UserAuthorize({
} finally { } finally {
setIsLoading(false); setIsLoading(false);
} }
}, [isLoading, router, searchParams]); }, [isLoading, isSSOCallback, isOAuthFlow, router, searchParams]);
useEffect(() => { useEffect(() => {
signInCallback(); signInCallback();
}, [signInCallback]); }, [signInCallback]);
// 如果没有 SSO 参数,显示正在处理授权的界面
if (!searchParams?.get("sso")) {
return (
<Card className="w-full">
<CardHeader className="space-y-4 text-center">
<div className="mx-auto flex h-16 w-16 items-center justify-center rounded-full bg-blue-100">
<div className="h-8 w-8 animate-spin rounded-full border-4 border-blue-600 border-t-transparent"></div>
</div>
<CardTitle className="text-2xl font-semibold"></CardTitle>
</CardHeader>
<CardContent>
<p className="text-center text-gray-500">
</p>
</CardContent>
</Card>
);
}
if (error) { if (error) {
return ( return (
<Card className="w-full"> <Card className="w-full">
@ -103,11 +84,17 @@ export function UserAuthorize({
<div className="mx-auto flex h-16 w-16 items-center justify-center rounded-full bg-blue-100"> <div className="mx-auto flex h-16 w-16 items-center justify-center rounded-full bg-blue-100">
<div className="h-8 w-8 animate-spin rounded-full border-4 border-blue-600 border-t-transparent"></div> <div className="h-8 w-8 animate-spin rounded-full border-4 border-blue-600 border-t-transparent"></div>
</div> </div>
<CardTitle className="text-2xl font-semibold"></CardTitle> <CardTitle className="text-2xl font-semibold">
{isOAuthFlow ? "正在处理授权" : "正在处理登录"}
</CardTitle>
</CardHeader> </CardHeader>
<CardContent> <CardContent>
<p className="text-center text-gray-500"> <p className="text-center text-gray-500">
{isLoading ? "请稍候,我们正在处理您的登录请求" : "正在跳转..."} {isOAuthFlow
? "请稍候,我们正在处理您的授权请求"
: isLoading
? "请稍候,我们正在处理您的登录请求"
: "正在跳转..."}
</p> </p>
</CardContent> </CardContent>
</Card> </Card>