feat: add Google Drive integration via Composio

app/api/connectedAccounts/refresh/route.ts → app/api/connectedAccounts/googleDrive/refresh/route.ts

@@ -1,11 +1,9 @@
 import { NextRequest, NextResponse } from "next/server";
-import refreshConnectedAccount from "@/lib/composio/googleSheets/refreshConnectedAccount";
+import { refreshConnectedAccount } from "@/lib/composio/refreshConnectedAccount";
 import { getCorsHeaders } from "@/lib/networking/getCorsHeaders";
 
 /**
  * OPTIONS handler for CORS preflight requests.
- *
- * @returns A NextResponse with CORS headers.
  */
 export async function OPTIONS() {
   return new NextResponse(null, {
@@ -15,10 +13,7 @@ export async function OPTIONS() {
 }
 
 /**
- * POST handler for refreshing a connected account.
- *
- * @param request - The request object.
- * @returns A NextResponse with the refreshed connected account.
+ * POST handler for refreshing a Google Drive connected account.
  */
 export async function POST(request: NextRequest) {
   try {
@@ -28,34 +23,28 @@ export async function POST(request: NextRequest) {
     if (!accountId) {
       return NextResponse.json(
         { error: "accountId is required" },
-        {
-          status: 400,
-          headers: getCorsHeaders(),
-        },
+        { status: 400, headers: getCorsHeaders() },
       );
     }
 
-    const response = await refreshConnectedAccount(accountId, redirectUrl);
+    const response = await refreshConnectedAccount(
+      "GOOGLE_DRIVE",
+      accountId,
+      redirectUrl,
+    );
 
     return NextResponse.json(
       { message: "Connected account refreshed successfully", ...response },
-      {
-        status: 200,
-        headers: getCorsHeaders(),
-      },
+      { status: 200, headers: getCorsHeaders() },
     );
   } catch (error) {
-    console.error("Error refreshing connected account:", error);
-
-    const errorMessage = error instanceof Error ? error.message : "Internal server error";
+    const errorMessage =
+      error instanceof Error ? error.message : "Internal server error";
     const statusCode = errorMessage.includes("not found") ? 404 : 500;
 
     return NextResponse.json(
       { error: errorMessage },
-      {
-        status: statusCode,
-        headers: getCorsHeaders(),
-      },
+      { status: statusCode, headers: getCorsHeaders() },
     );
   }
 }

app/api/connectedAccounts/googleSheets/refresh/route.ts

@@ -0,0 +1,50 @@
+import { NextRequest, NextResponse } from "next/server";
+import { refreshConnectedAccount } from "@/lib/composio/refreshConnectedAccount";
+import { getCorsHeaders } from "@/lib/networking/getCorsHeaders";
+
+/**
+ * OPTIONS handler for CORS preflight requests.
+ */
+export async function OPTIONS() {
+  return new NextResponse(null, {
+    status: 200,
+    headers: getCorsHeaders(),
+  });
+}
+
+/**
+ * POST handler for refreshing a Google Sheets connected account.
+ */
+export async function POST(request: NextRequest) {
+  try {
+    const body = await request.json();
+    const { accountId, redirectUrl } = body;
+
+    if (!accountId) {
+      return NextResponse.json(
+        { error: "accountId is required" },
+        { status: 400, headers: getCorsHeaders() },
+      );
+    }
+
+    const response = await refreshConnectedAccount(
+      "GOOGLE_SHEETS",
+      accountId,
+      redirectUrl,
+    );
+
+    return NextResponse.json(
+      { message: "Connected account refreshed successfully", ...response },
+      { status: 200, headers: getCorsHeaders() },
+    );
+  } catch (error) {
+    const errorMessage =
+      error instanceof Error ? error.message : "Internal server error";
+    const statusCode = errorMessage.includes("not found") ? 404 : 500;
+
+    return NextResponse.json(
+      { error: errorMessage },
+      { status: statusCode, headers: getCorsHeaders() },
+    );
+  }
+}

lib/composio/authenticateToolkit.ts

@@ -0,0 +1,28 @@
+import { CreateConnectedAccountOptions } from "@composio/core";
+import { getComposioClient } from "./client";
+import { ComposioToolkitKey, getToolkitAuthConfigId } from "./toolkits";
+
+/**
+ * Initiate OAuth authentication for a Composio toolkit.
+ *
+ * @param toolkitKey - The toolkit to authenticate (e.g., "GOOGLE_SHEETS", "GOOGLE_DRIVE")
+ * @param userId - The user ID to authenticate
+ * @param options - Optional callback URL and other options
+ * @returns The connection request from Composio
+ */
+export async function authenticateToolkit(
+  toolkitKey: ComposioToolkitKey,
+  userId: string,
+  options?: CreateConnectedAccountOptions,
+) {
+  const authConfigId = getToolkitAuthConfigId(toolkitKey);
+  const composio = getComposioClient();
+
+  const connectionRequest = await composio.connectedAccounts.initiate(
+    userId,
+    authConfigId,
+    options,
+  );
+
+  return connectionRequest;
+}

lib/composio/getConnectedAccount.ts

@@ -0,0 +1,37 @@
+import { CreateConnectedAccountOptions } from "@composio/core";
+import { getComposioClient } from "./client";
+import { authenticateToolkit } from "./authenticateToolkit";
+import { ComposioToolkitKey, getToolkitConfig } from "./toolkits";
+
+/**
+ * Get a user's connected account for a Composio toolkit.
+ * If no connection exists, initiates the authentication flow.
+ *
+ * @param toolkitKey - The toolkit to check (e.g., "GOOGLE_SHEETS", "GOOGLE_DRIVE")
+ * @param accountId - The user's account ID
+ * @param options - Optional callback URL and other options
+ * @returns The user's connected accounts for this toolkit
+ */
+export async function getConnectedAccount(
+  toolkitKey: ComposioToolkitKey,
+  accountId: string,
+  options?: CreateConnectedAccountOptions,
+) {
+  const config = getToolkitConfig(toolkitKey);
+  const composio = getComposioClient();
+
+  let userAccounts = await composio.connectedAccounts.list({
+    userIds: [accountId],
+    toolkitSlugs: [config.slug],
+  });
+
+  if (userAccounts.items.length === 0) {
+    await authenticateToolkit(toolkitKey, accountId, options);
+    userAccounts = await composio.connectedAccounts.list({
+      userIds: [accountId],
+      toolkitSlugs: [config.slug],
+    });
+  }
+
+  return userAccounts;
+}

lib/composio/refreshConnectedAccount.ts

@@ -0,0 +1,72 @@
+import { getConnectedAccount } from "./getConnectedAccount";
+import { getComposioApiKey } from "./getComposioApiKey";
+import { ComposioToolkitKey, getToolkitConfig } from "./toolkits";
+
+export interface ConnectedAccountRefreshResponse {
+  id: string;
+  redirect_url: string | null;
+  status:
+    | "INITIALIZING"
+    | "INITIATED"
+    | "ACTIVE"
+    | "FAILED"
+    | "EXPIRED"
+    | "INACTIVE";
+}
+
+const COMPOSIO_API_BASE_URL = "https://backend.composio.dev";
+
+/**
+ * Refresh a connected account for a Composio toolkit.
+ *
+ * @param toolkitKey - The toolkit to refresh (e.g., "GOOGLE_SHEETS", "GOOGLE_DRIVE")
+ * @param accountId - The user's account ID
+ * @param redirectUrl - The URL to redirect to after OAuth
+ * @returns The refresh response with OAuth redirect URL
+ */
+export async function refreshConnectedAccount(
+  toolkitKey: ComposioToolkitKey,
+  accountId: string,
+  redirectUrl?: string,
+): Promise<ConnectedAccountRefreshResponse | null> {
+  const apiKey = getComposioApiKey();
+  const config = getToolkitConfig(toolkitKey);
+  const accounts = await getConnectedAccount(toolkitKey, accountId);
+
+  if (!accounts.items || accounts.items.length === 0) {
+    throw new Error(`${config.name} connected account not found`);
+  }
+
+  const connectedAccountId = accounts.items[0].id;
+
+  if (!connectedAccountId) {
+    throw new Error(`${config.name} connected account ID not found`);
+  }
+
+  const url = new URL(
+    `${COMPOSIO_API_BASE_URL}/api/v3/connected_accounts/${connectedAccountId}/refresh`,
+  );
+
+  const body: { redirect_url?: string } = {};
+  if (redirectUrl) {
+    body.redirect_url = redirectUrl;
+  }
+
+  const response = await fetch(url.toString(), {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      "x-api-key": apiKey,
+    },
+    body: JSON.stringify(body),
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(
+      `Failed to refresh ${config.name} connected account: ${response.status} ${errorText}`,
+    );
+  }
+
+  return await response.json();
+}