Staging

apps/mail/actions/ai-reply.ts

@@ -1,8 +1,9 @@
 'use server';
 
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
+import { generateCompletions } from '@/lib/groq';
 import { headers } from 'next/headers';
 import { auth } from '@/lib/auth';
-import { generateCompletions } from '@/lib/groq';
 
 // Generates an AI response for an email reply based on the thread content
 export async function generateAIResponse(
@@ -13,15 +14,15 @@ export async function generateAIResponse(
   const session = await auth.api.getSession({ headers: headersList });
 
   if (!session?.user) {
-    throw new Error('Unauthorized');
+    return throwUnauthorizedGracefully();
   }
 
   if (!process.env.GROQ_API_KEY) {
     throw new Error('Groq API key is not configured');
   }
 
   // Use a more aggressive content reduction approach
-  const processedContent = threadContent
+  const processedContent = threadContent;
 
   // Create the system message
   const systemPrompt = `You are an email assistant helping ${session.user.name} write professional and concise email replies.
@@ -44,7 +45,7 @@ export async function generateAIResponse(
   - Double space paragraphs (2 newlines)
   - Add two spaces below the sign-off
   - End with the name: ${session.user.name}`;
-  
+
   // Create the user message - keep it shorter
   const prompt = `
   Here's the context of the email thread (some parts may be summarized or truncated due to length):
@@ -82,7 +83,7 @@ export async function generateAIResponse(
       systemPrompt: process.env.AI_SYSTEM_PROMPT || systemPrompt,
       prompt,
       temperature: 0.6,
-      userName: session.user.name
+      userName: session.user.name,
     });
 
     return completion;

apps/mail/actions/ai.ts

@@ -1,6 +1,7 @@
 // The brain.ts file in /actions should replace this file once ready.
 'use server';
 
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
 import { generateEmailContent } from '@/lib/ai';
 import { headers } from 'next/headers';
 import { JSONContent } from 'novel';
@@ -37,9 +38,9 @@ export async function generateAIEmailContent({
     const session = await auth.api.getSession({ headers: headersList });
 
     if (!session?.user) {
-      throw new Error('Unauthorized');
+      return throwUnauthorizedGracefully();
     }
-    
+
     const responses = await generateEmailContent(
       prompt,
       currentContent,
@@ -58,7 +59,7 @@ export async function generateAIEmailContent({
     }
 
     const emailResponses = responses.filter((r) => r.type === 'email');
-    
+
     const cleanedContent = emailResponses
       .map((r) => r.content)
       .join('\n\n')

apps/mail/actions/brain.ts

@@ -1,24 +1,20 @@
-'use server'
-import { auth } from "@/lib/auth";
-import { db } from "@zero/db";
-import { connection } from "@zero/db/schema";
-import { headers } from "next/headers";
-import { and, eq } from "drizzle-orm";
-import axios from "axios";
-import { getActiveConnection } from "./utils";
+'use server';
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
+import { getActiveConnection } from './utils';
+import axios from 'axios';
 
 export const EnableBrain = async () => {
-    if (!process.env.BRAIN_URL) {
-        throw new Error('Brain URL not found');
-    }
+  if (!process.env.BRAIN_URL) {
+    return null;
+  }
 
-    const connection = await getActiveConnection()
+  const connection = await getActiveConnection();
 
-    if (!connection?.accessToken || !connection.refreshToken) {
-        throw new Error("Unauthorized, reconnect");
-    }
+  if (!connection?.accessToken || !connection.refreshToken) {
+    return throwUnauthorizedGracefully();
+  }
 
-    return await axios.put(process.env.BRAIN_URL + `/subscribe/${connection.providerId}`, {
-        connectionId: connection.id,
-    })
-}
+  return await axios.put(process.env.BRAIN_URL + `/subscribe/${connection.providerId}`, {
+    connectionId: connection.id,
+  });
+};

apps/mail/actions/connections.ts

@@ -1,6 +1,6 @@
 'use server';
 
-import { getAuthenticatedUserId } from '@/app/api/utils';
+import { getAuthenticatedUserId, throwUnauthorizedGracefully } from '@/app/api/utils';
 import { connection, user } from '@zero/db/schema';
 import { type IConnection } from '@/types';
 import { headers } from 'next/headers';
@@ -14,13 +14,13 @@ export async function deleteConnection(connectionId: string) {
     const session = await auth.api.getSession({ headers: headersList });
 
     if (!session) {
-      throw new Error('Unauthorized, reconnect');
+      return throwUnauthorizedGracefully();
     }
 
     const userId = session?.user?.id;
 
     if (!userId) {
-      throw new Error('Unauthorized, reconnect');
+      return throwUnauthorizedGracefully();
     }
 
     await db
@@ -46,13 +46,13 @@ export async function putConnection(connectionId: string) {
     const session = await auth.api.getSession({ headers: headersList });
 
     if (!session) {
-      throw new Error('Unauthorized, reconnect');
+      return throwUnauthorizedGracefully();
     }
 
     const userId = session?.user?.id;
 
     if (!userId) {
-      throw new Error('Unauthorized, reconnect');
+      return throwUnauthorizedGracefully();
     }
 
     const [foundConnection] = await db

apps/mail/actions/drafts.ts

@@ -1,5 +1,6 @@
-"use server";
-import { getActiveDriver } from "./utils";
+'use server';
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
+import { getActiveDriver } from './utils';
 
 export const getDrafts = async ({
   q,
@@ -14,8 +15,9 @@ export const getDrafts = async ({
     const driver = await getActiveDriver();
     return await driver.listDrafts(q, max, pageToken);
   } catch (error) {
-    console.error("Error getting threads:", error);
-    throw error;
+    console.error('Error getting threads:', error);
+    await throwUnauthorizedGracefully();
+    // throw error;
   }
 };
 
@@ -27,23 +29,23 @@ export const createDraft = async (data: any) => {
 
     return { success: true, id: res.id };
   } catch (error) {
-    console.error("Error creating draft:", error);
+    console.error('Error creating draft:', error);
     return { success: false, error: String(error) };
   }
 };
 
 export const getDraft = async (id: string) => {
   if (!id) {
-    throw new Error("Missing draft ID");
+    throw new Error('Missing draft ID');
   }
 
-  console.log("getting email:", id);
+  console.log('getting email:', id);
 
   try {
     const driver = await getActiveDriver();
     return await driver.getDraft(id);
   } catch (error) {
-    console.error("Error getting draft:", error);
+    console.error('Error getting draft:', error);
     throw error;
   }
 };

apps/mail/actions/getSummary.ts

@@ -1,4 +1,5 @@
 'use server';
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
 import { connection, summary } from '@zero/db/schema';
 import { headers } from 'next/headers';
 import { and, eq } from 'drizzle-orm';
@@ -11,7 +12,7 @@ export const GetSummary = async (threadId: string) => {
   const session = await auth.api.getSession({ headers: headersList });
 
   if (!session || !session.connectionId) {
-    throw new Error('Unauthorized, reconnect');
+    return throwUnauthorizedGracefully();
   }
 
   const [_connection] = await db
@@ -20,23 +21,21 @@ export const GetSummary = async (threadId: string) => {
     .where(and(eq(connection.userId, session.user.id), eq(connection.id, session.connectionId)));
 
   if (!_connection) {
-    throw new Error('Unauthorized, reconnect');
+    return throwUnauthorizedGracefully();
   }
 
   try {
     if (!process.env.BRAIN_URL || process.env.NODE_ENV !== 'production') {
-      throw new Error('Brain URL not found');
+      return null;
     }
 
     const response = await axios
       .get(process.env.BRAIN_URL + `/brain/message/thread/${threadId}`)
       .then((e) => e.data);
 
-    console.log('Summary:', response);
-
     return response?.content ?? null;
   } catch (error) {
-    console.error('Error getting summary:', error);
+    // console.error('Error getting summary:', error);
     return null;
   }
 };

apps/mail/actions/mail.ts

@@ -1,5 +1,6 @@
 'use server';
 import { deleteActiveConnection, FatalErrors, getActiveDriver } from './utils';
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
 import { IGetThreadResponse } from '@/app/api/driver/types';
 import { ParsedMessage } from '@/types';
 
@@ -26,7 +27,9 @@ export const getMails = async ({
   } catch (error) {
     if (FatalErrors.includes((error as Error).message)) await deleteActiveConnection();
     console.error('Error getting threads:', error);
-    throw error;
+    // throw error;
+    await throwUnauthorizedGracefully();
+    return [];
   }
 };
 

apps/mail/actions/send.ts

@@ -1,8 +1,9 @@
-"use server";
+'use server';
 
-import { createDriver } from "@/app/api/driver";
-import { Sender } from "@/types";
-import { getActiveConnection } from "./utils";
+import { throwUnauthorizedGracefully } from '@/app/api/utils';
+import { createDriver } from '@/app/api/driver';
+import { getActiveConnection } from './utils';
+import { Sender } from '@/types';
 
 export async function sendEmail({
   to,
@@ -12,7 +13,7 @@ export async function sendEmail({
   bcc,
   cc,
   headers: additionalHeaders = {},
-  threadId
+  threadId,
 }: {
   to: Sender[];
   subject: string;
@@ -21,25 +22,23 @@ export async function sendEmail({
   headers?: Record<string, string>;
   cc?: Sender[];
   bcc?: Sender[];
-  threadId?: string
+  threadId?: string;
 }) {
   if (!to || !subject || !message) {
-    throw new Error("Missing required fields");
+    throw new Error('Missing required fields');
   }
 
-  console.log(additionalHeaders)
-
-  const connection = await getActiveConnection()
+  const connection = await getActiveConnection();
 
   if (!connection?.accessToken || !connection.refreshToken) {
-    throw new Error("Unauthorized, reconnect");
+    return throwUnauthorizedGracefully();
   }
 
   const driver = await createDriver(connection.providerId, {
     auth: {
       access_token: connection.accessToken,
       refresh_token: connection.refreshToken,
-      email: connection.email
+      email: connection.email,
     },
   });
 
@@ -51,7 +50,7 @@ export async function sendEmail({
     headers: additionalHeaders,
     cc,
     bcc,
-    threadId
+    threadId,
   });
 
   return { success: true };