cleanup: UI

Author: izadoesdev

apps/dashboard/app/(auth)/login/page.tsx

@@ -33,46 +33,28 @@ function LoginPage() {
 		setLastUsed(localStorage.getItem('lastUsedLogin'));
 	}, []);
 
-	const handlePostAuthCallback = () => {
-		const callbackUrl = searchParams.get('callback');
-		if (callbackUrl) {
-			router.push(callbackUrl);
-		}
-	};
-
-	const handleGoogleLogin = () => {
+	const handleSocialLogin = (provider: 'github' | 'google') => {
 		setIsLoading(true);
-		signIn.social({
-			provider: 'google',
-			callbackURL: defaultCallbackUrl,
-			newUserCallbackURL: '/onboarding',
-			fetchOptions: {
-				onSuccess: () => {
-					localStorage.setItem('lastUsedLogin', 'google');
-					handlePostAuthCallback();
-				},
-				onError: () => {
-					setIsLoading(false);
-					toast.error('Google login failed. Please try again.');
-				},
-			},
-		});
-	};
 
-	const handleGithubLogin = () => {
-		setIsLoading(true);
+		const callbackUrl = searchParams.get('callback');
+		const finalCallbackUrl = callbackUrl || defaultCallbackUrl;
+
 		signIn.social({
-			provider: 'github',
-			callbackURL: defaultCallbackUrl,
+			provider,
+			callbackURL: finalCallbackUrl,
 			newUserCallbackURL: '/onboarding',
 			fetchOptions: {
 				onSuccess: () => {
-					localStorage.setItem('lastUsedLogin', 'github');
-					handlePostAuthCallback();
+					localStorage.setItem('lastUsedLogin', provider);
+					if (callbackUrl) {
+						router.push(callbackUrl);
+					}
 				},
 				onError: () => {
 					setIsLoading(false);
-					toast.error('GitHub login failed. Please try again.');
+					toast.error(
+						`${provider === 'github' ? 'GitHub' : 'Google'} login failed. Please try again.`
+					);
 				},
 			},
 		});
@@ -94,7 +76,10 @@ function LoginPage() {
 			fetchOptions: {
 				onSuccess: () => {
 					localStorage.setItem('lastUsedLogin', 'email');
-					handlePostAuthCallback();
+					const callbackUrl = searchParams.get('callback');
+					if (callbackUrl) {
+						router.push(callbackUrl);
+					}
 				},
 				onError: (error) => {
 					setIsLoading(false);
@@ -135,7 +120,7 @@ function LoginPage() {
 							<Button
 								className="relative flex h-11 w-full cursor-pointer items-center justify-center transition-all duration-200 hover:bg-primary/5"
 								disabled={isLoading}
-								onClick={handleGithubLogin}
+								onClick={() => handleSocialLogin('github')}
 								type="button"
 								variant="outline"
 							>
@@ -150,7 +135,7 @@ function LoginPage() {
 							<Button
 								className="relative flex h-11 w-full cursor-pointer items-center justify-center transition-all duration-200 hover:bg-primary/5"
 								disabled={isLoading}
-								onClick={handleGoogleLogin}
+								onClick={() => handleSocialLogin('google')}
 								type="button"
 								variant="outline"
 							>

apps/dashboard/app/(auth)/register/page.tsx

@@ -54,9 +54,15 @@ function RegisterPageContent() {
 		setFormData((prev) => ({ ...prev, [name]: value }));
 	};
 
-	const handlePostAuthCallback = () => {
+	const handleAuthSuccess = () => {
 		if (callbackUrl) {
+			toast.success('Account created! Completing integration...');
 			router.push(callbackUrl);
+		} else if (selectedPlan) {
+			localStorage.setItem('pendingPlanSelection', selectedPlan);
+			router.push(`/billing?tab=plans&plan=${selectedPlan}`);
+		} else {
+			router.push('/websites');
 		}
 	};
 
@@ -85,28 +91,18 @@ function RegisterPageContent() {
 			password: formData.password,
 			name: formData.name,
 			fetchOptions: {
-				onSuccess: (ctx) => {
-					const authToken = ctx.response.headers.get('set-auth-token');
-					if (authToken) {
-						localStorage.setItem('authToken', authToken);
-					}
-
+				onSuccess: () => {
 					if (callbackUrl) {
-						toast.success('Account created! Completing integration...');
-						handlePostAuthCallback();
-						return;
-					}
-
-					toast.success(
-						'Account created! Please check your email to verify your account.'
-					);
-					setRegistrationStep('verification-needed');
-
-					// Store plan selection for post-verification redirect
-					if (selectedPlan) {
-						localStorage.setItem('pendingPlanSelection', selectedPlan);
+						handleAuthSuccess();
+					} else {
+						toast.success(
+							'Account created! Please check your email to verify your account.'
+						);
+						setRegistrationStep('verification-needed');
+						if (selectedPlan) {
+							localStorage.setItem('pendingPlanSelection', selectedPlan);
+						}
 					}
-					// router.push(`/verify?email=${encodeURIComponent(formData.email)}`);
 				},
 			},
 		});
@@ -142,33 +138,27 @@ function RegisterPageContent() {
 	const handleSocialLogin = async (provider: 'github' | 'google') => {
 		setIsLoading(true);
 
-		await authClient.signIn.social({
-			provider,
-			callbackURL: callbackUrl || '/websites',
-			fetchOptions: {
-				onSuccess: (ctx) => {
-					const authToken = ctx.response.headers.get('set-auth-token');
-					if (authToken) {
-						localStorage.setItem('authToken', authToken);
-					}
-
-					toast.success('Registration successful!');
-
-					// Redirect to billing with plan selection if plan was specified
-					if (selectedPlan) {
-						localStorage.setItem('pendingPlanSelection', selectedPlan);
-						router.push(`/billing?tab=plans&plan=${selectedPlan}`);
-					} else {
-						router.push('/websites');
-					}
-				},
-				onError: () => {
-					toast.error('Login failed. Please try again.');
+		try {
+			await authClient.signIn.social({
+				provider,
+				callbackURL: callbackUrl || '/websites',
+				fetchOptions: {
+					onSuccess: () => {
+						toast.success('Registration successful!');
+						handleAuthSuccess();
+					},
+					onError: () => {
+						toast.error(
+							`${provider === 'github' ? 'GitHub' : 'Google'} login failed. Please try again.`
+						);
+						setIsLoading(false);
+					},
 				},
-			},
-		});
-
-		setIsLoading(false);
+			});
+		} catch (error) {
+			toast.error('Login failed. Please try again.');
+			setIsLoading(false);
+		}
 	};
 
 	// Render header content based on current registration step