demo auth redirect issue fixed and limit API generation daily

Author: rasenga223

public/images/create-beautiful-api.png

@@ -4,7 +4,7 @@ import { useState, ChangeEvent, FormEvent, useEffect } from 'react';
 import { useRouter } from 'next/navigation';
 import Link from 'next/link';
 import { generateDocumentation, AVAILABLE_MODELS } from '@/lib/openrouter';
-import { createProject, saveMdxFiles, updateProjectStatus, saveAiJob, updateAiJob } from '@/lib/docService';
+import { createProject, saveMdxFiles, updateProjectStatus, saveAiJob, updateAiJob, getUserDailyAiJobCount } from '@/lib/docService';
 import { getUser } from '@/lib/supabase';
 import FileUploader from '@/components/api-doc-generator/FileUploader';
 
@@ -27,19 +27,23 @@ export default function GenerateDocumentationPage() {
   const [isLoading, setIsLoading] = useState(false);
   const [error, setError] = useState<string | null>(null);
   const [userId, setUserId] = useState<string | null>(null);
+  const [aiJobCount, setAiJobCount] = useState<number>(0);
+  const AI_JOB_DAILY_LIMIT = 7;
 
-  // Get the current user
+  // Get the current user and AI job count
   useEffect(() => {
-    async function fetchUser() {
+    async function fetchUserAndJobCount() {
       const user = await getUser();
       if (user) {
         setUserId(user.id);
+        const count = await getUserDailyAiJobCount(user.id);
+        setAiJobCount(count);
       } else {
         // Redirect to login if user is not authenticated
         router.push('/login');
       }
     }
-    fetchUser();
+    fetchUserAndJobCount();
   }, [router]);
 
   /**
@@ -96,6 +100,11 @@ export default function GenerateDocumentationPage() {
       setError('You must be logged in to generate documentation');
       return;
     }
+
+    if (aiJobCount >= AI_JOB_DAILY_LIMIT) {
+      setError(`You have reached the limit of ${AI_JOB_DAILY_LIMIT} AI generations per day. Please try again tomorrow.`);
+      return;
+    }
 
     setIsLoading(true);
     setError(null);
@@ -186,6 +195,9 @@ export default function GenerateDocumentationPage() {
       sessionStorage.setItem('generatedDocs', JSON.stringify(formattedFiles));
       console.log('💾 Saved formatted docs to sessionStorage');
 
+      // Update the AI job count after successful generation
+      setAiJobCount(prevCount => prevCount + 1);
+      
       // Navigate directly to the editor instead of preview
       router.push('/editor');
     } catch (err) {
@@ -500,9 +512,9 @@ export default function GenerateDocumentationPage() {
               <div className="flex justify-end">
                 <button
                   type="submit"
-                  disabled={isLoading || !apiSpec.trim() || !userId}
+                  disabled={isLoading || !apiSpec.trim() || !userId || aiJobCount >= AI_JOB_DAILY_LIMIT}
                   className={`inline-flex items-center justify-center px-8 py-3 text-base font-medium rounded-lg transition-all duration-200 ${
-                    isLoading || !apiSpec.trim() || !userId
+                    isLoading || !apiSpec.trim() || !userId || aiJobCount >= AI_JOB_DAILY_LIMIT
                       ? 'cursor-not-allowed bg-muted text-muted-foreground'
                       : 'bg-primary text-primary-foreground hover:bg-primary/90 shadow-lg shadow-primary/20 hover:shadow-xl hover:shadow-primary/30'
                   }`}
@@ -533,7 +545,12 @@ export default function GenerateDocumentationPage() {
                     </>
                   ) : (
                     <>
-                      <span>Generate Documentation</span>
+                      <span>
+                        {aiJobCount >= AI_JOB_DAILY_LIMIT 
+                          ? `Daily Limit Reached (${aiJobCount}/${AI_JOB_DAILY_LIMIT})`
+                          : `Generate Documentation (${aiJobCount}/${AI_JOB_DAILY_LIMIT})`
+                        }
+                      </span>
                       <svg
                         className="w-5 h-5 ml-2"
                         fill="none"

src/app/(dashboard)/api-doc-generator/generate/page.tsx

@@ -28,7 +28,7 @@ export const metadata: Metadata = {
       "One-click API documentation generator with AI enhancement and custom domain support",
     images: [
       {
-        url: "/og-image.png",
+        url: "/images/create-beautiful-api.png",
         width: 1200,
         height: 630,
         alt: "Clario API Documentation Generator",
@@ -40,7 +40,7 @@ export const metadata: Metadata = {
     title: "Clario | API Documentation Generator",
     description:
       "Transform your API spec into a clean, branded, shareable documentation site instantly",
-    images: ["/og-image.png"],
+    images: ["/images/create-beautiful-api.png"],
   },
   icons: {
     icon: ["/favicon_io/favicon.ico?v=4"],

src/app/layout.tsx

@@ -572,4 +572,37 @@ export async function updateMdxFileOrder(
     console.error('Error updating MDX file order:', error);
     return false;
   }
+}
+
+/**
+ * Gets the count of AI jobs created by a user in the last 24 hours
+ * 
+ * @param userId - User ID
+ * @returns Number of AI jobs created today
+ */
+export async function getUserDailyAiJobCount(userId: string): Promise<number> {
+  const oneDayAgo = new Date();
+  oneDayAgo.setDate(oneDayAgo.getDate() - 1);
+  
+  // First get the user's project IDs
+  const { data: projectIds } = await supabase
+    .from('projects')
+    .select('id')
+    .eq('user_id', userId);
+    
+  if (!projectIds) return 0;
+  
+  // Then count AI jobs for these projects
+  const { count, error } = await supabase
+    .from('ai_jobs')
+    .select('*', { count: 'exact', head: true })
+    .in('project_id', projectIds.map(p => p.id))
+    .gte('started_at', oneDayAgo.toISOString());
+    
+  if (error) {
+    console.error('Error getting AI job count:', error);
+    return 0;
+  }
+  
+  return count || 0;
 } 

src/lib/docService.ts

@@ -150,9 +150,12 @@ export function AuthProvider({ children }: { children: ReactNode }) {
       // Create demo user in public table if needed
       if (data.user) {
         await ensureUserInPublicTable(data.user as UserData);
+        // Set the user state immediately after successful login
+        setUser(data.user as UserData);
+        // Wait a brief moment to ensure auth state is properly updated
+        await new Promise(resolve => setTimeout(resolve, 100));
+        router.push('/dashboard');
       }
-      
-      router.push('/dashboard');
     } catch (error) {
       console.error("Error signing in with demo account:", error);
       toast.error("Failed to sign in with demo account");