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 scope = params.get("scope") || "read_profile";
// 重新构建授权页面 URL
// 重新构建授权页面 URL,使用相对路径
const authParams = new URLSearchParams({
response_type: "code",
client_id: clientId || "",
@ -37,14 +37,16 @@ export async function signIn(data: Record<string, any>) {
authParams.set("state", state);
}
// 重定向到授权页面
// 重定向到授权页面,使用相对路径
redirect(`/oauth/authorize?${authParams.toString()}`);
return;
}
// 如果没有 OAuth 参数,重定向到仪表板
redirect("/dashboard");
} catch (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 searchParams = useSearchParams();
// 检查是否是 OAuth 授权页面
const isOAuthFlow =
searchParams?.has("client_id") && searchParams?.has("redirect_uri");
// 检查是否是 SSO 回调
const isSSOCallback = searchParams?.has("sso") && searchParams?.has("sig");
const signInCallback = useCallback(async () => {
if (isLoading) {
return;
}
// 检查是否是 SSO 回调
const sso = searchParams?.get("sso");
const sig = searchParams?.get("sig");
// 如果没有 SSO 参数,说明这是 OAuth 授权页面,不需要处理
if (!sso || !sig) {
if (isLoading || !isSSOCallback || isOAuthFlow) {
return;
}
setIsLoading(true);
try {
const sso = searchParams?.get("sso");
const sig = searchParams?.get("sig");
// 传递 SSO 参数
await signIn({
sso,
@ -50,31 +50,12 @@ export function UserAuthorize({
} finally {
setIsLoading(false);
}
}, [isLoading, router, searchParams]);
}, [isLoading, isSSOCallback, isOAuthFlow, router, searchParams]);
useEffect(() => {
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) {
return (
<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="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>
<CardTitle className="text-2xl font-semibold">
{isOAuthFlow ? "正在处理授权" : "正在处理登录"}
</CardTitle>
</CardHeader>
<CardContent>
<p className="text-center text-gray-500">
{isLoading ? "请稍候,我们正在处理您的登录请求" : "正在跳转..."}
{isOAuthFlow
? "请稍候,我们正在处理您的授权请求"
: isLoading
? "请稍候,我们正在处理您的登录请求"
: "正在跳转..."}
</p>
</CardContent>
</Card>