fix(google-drive): add auto export format and Azure storage debug logging

Author: waleedlatif1

apps/sim/app/api/tools/google_drive/download/route.ts

@@ -65,10 +65,12 @@ export async function POST(request: NextRequest) {
     const {
       accessToken,
       fileId,
-      mimeType: exportMimeType,
+      mimeType: rawExportMimeType,
       fileName,
       includeRevisions,
     } = validatedData
+    const exportMimeType =
+      rawExportMimeType && rawExportMimeType !== 'auto' ? rawExportMimeType : null
     const authHeader = `Bearer ${accessToken}`
 
     logger.info(`[${requestId}] Getting file metadata from Google Drive`, { fileId })
@@ -129,7 +131,7 @@ export async function POST(request: NextRequest) {
         )
       }
 
-      const exportResponse = await secureFetchWithPinnedIP(
+      let exportResponse = await secureFetchWithPinnedIP(
         exportUrl,
         exportUrlValidation.resolvedIP!,
         { headers: { Authorization: authHeader } }
@@ -139,17 +141,67 @@ export async function POST(request: NextRequest) {
         const exportError = (await exportResponse
           .json()
           .catch(() => ({}))) as GoogleApiErrorResponse
-        logger.error(`[${requestId}] Failed to export file`, {
-          status: exportResponse.status,
-          error: exportError,
-        })
-        return NextResponse.json(
-          {
-            success: false,
-            error: exportError.error?.message || 'Failed to export Google Workspace file',
-          },
-          { status: 400 }
-        )
+        const errorMessage = exportError.error?.message || ''
+
+        const hasCustomFormat = !!exportMimeType
+        const defaultFormat = DEFAULT_EXPORT_FORMATS[fileMimeType]
+        if (
+          hasCustomFormat &&
+          defaultFormat &&
+          exportMimeType !== defaultFormat &&
+          errorMessage.toLowerCase().includes('conversion')
+        ) {
+          logger.warn(`[${requestId}] Export format not supported, falling back to default`, {
+            requestedFormat: exportFormat,
+            fallbackFormat: defaultFormat,
+            fileMimeType,
+          })
+
+          finalMimeType = defaultFormat
+          const fallbackUrl = `https://www.googleapis.com/drive/v3/files/${fileId}/export?mimeType=${encodeURIComponent(defaultFormat)}&supportsAllDrives=true`
+          const fallbackUrlValidation = await validateUrlWithDNS(fallbackUrl, 'fallbackExportUrl')
+          if (!fallbackUrlValidation.isValid) {
+            return NextResponse.json(
+              { success: false, error: fallbackUrlValidation.error },
+              { status: 400 }
+            )
+          }
+
+          exportResponse = await secureFetchWithPinnedIP(
+            fallbackUrl,
+            fallbackUrlValidation.resolvedIP!,
+            { headers: { Authorization: authHeader } }
+          )
+
+          if (!exportResponse.ok) {
+            const fallbackError = (await exportResponse
+              .json()
+              .catch(() => ({}))) as GoogleApiErrorResponse
+            logger.error(`[${requestId}] Fallback export also failed`, {
+              status: exportResponse.status,
+              error: fallbackError,
+            })
+            return NextResponse.json(
+              {
+                success: false,
+                error: fallbackError.error?.message || 'Failed to export Google Workspace file',
+              },
+              { status: 400 }
+            )
+          }
+        } else {
+          logger.error(`[${requestId}] Failed to export file`, {
+            status: exportResponse.status,
+            error: exportError,
+          })
+          return NextResponse.json(
+            {
+              success: false,
+              error: errorMessage || 'Failed to export Google Workspace file',
+            },
+            { status: 400 }
+          )
+        }
       }
 
       const arrayBuffer = await exportResponse.arrayBuffer()

apps/sim/blocks/blocks/google_drive.ts

@@ -316,6 +316,7 @@ Return ONLY the query string - no explanations, no quotes around the whole thing
       title: 'Export Format',
       type: 'dropdown',
       options: [
+        { label: 'Auto (best format for file type)', id: 'auto' },
         { label: 'Plain Text (text/plain)', id: 'text/plain' },
         { label: 'HTML (text/html)', id: 'text/html' },
         { label: 'PDF (application/pdf)', id: 'application/pdf' },
@@ -333,7 +334,8 @@ Return ONLY the query string - no explanations, no quotes around the whole thing
         },
         { label: 'CSV (text/csv)', id: 'text/csv' },
       ],
-      placeholder: 'Optional: Choose export format for Google Docs/Sheets/Slides',
+      value: 'auto',
+      placeholder: 'Export format for Google Docs/Sheets/Slides',
       condition: { field: 'operation', value: 'download' },
     },
     {
@@ -867,7 +869,7 @@ Return ONLY the message text - no subject line, no greetings/signatures, no extr
           destinationFolderId: effectiveDestinationFolderId,
           file: normalizedFile,
           pageSize: rest.pageSize ? Number.parseInt(rest.pageSize as string, 10) : undefined,
-          mimeType: mimeType,
+          mimeType: mimeType === 'auto' ? undefined : mimeType,
           type: shareType, // Map shareType to type for share tool
           starred: starredValue,
           sendNotification: sendNotificationValue,

apps/sim/lib/uploads/contexts/workspace/workspace-file-manager.ts

@@ -247,6 +247,21 @@ export async function uploadWorkspaceFile(
         continue
       }
       logger.error(`Failed to upload workspace file ${fileName}:`, error)
+      if (
+        error instanceof Error &&
+        'code' in error &&
+        (error as { code?: string }).code === 'AuthorizationFailure'
+      ) {
+        const { getStorageProvider, BLOB_CONFIG } = await import('@/lib/uploads/config')
+        logger.error('Azure storage authorization diagnosis', {
+          provider: getStorageProvider(),
+          accountName: BLOB_CONFIG.accountName || '(empty)',
+          containerName: BLOB_CONFIG.containerName || '(empty)',
+          hasAccountKey: !!BLOB_CONFIG.accountKey,
+          keyLength: BLOB_CONFIG.accountKey?.length || 0,
+          hasConnectionString: !!BLOB_CONFIG.connectionString,
+        })
+      }
       throw new Error(
         `Failed to upload file: ${error instanceof Error ? error.message : 'Unknown error'}`
       )

apps/sim/lib/uploads/providers/blob/client.ts

@@ -72,8 +72,15 @@ export async function getBlobServiceClient(): Promise<BlobServiceClientInstance>
   const { accountName, accountKey, connectionString } = BLOB_CONFIG
 
   if (connectionString) {
+    logger.info('Initializing Azure Blob client using connection string', {
+      accountName: accountName || '(from connection string)',
+    })
     _blobServiceClient = BlobServiceClient.fromConnectionString(connectionString)
   } else if (accountName && accountKey) {
+    logger.info('Initializing Azure Blob client using account name and key', {
+      accountName,
+      keyLength: accountKey.length,
+    })
     const sharedKeyCredential = new StorageSharedKeyCredential(accountName, accountKey)
     _blobServiceClient = new BlobServiceClient(
       `https://${accountName}.blob.core.windows.net`,
@@ -131,6 +138,12 @@ export async function uploadToBlob(
   const uniqueKey = shouldPreserveKey ? fileName : `${Date.now()}-${safeFileName}`
 
   const blobServiceClient = await getBlobServiceClient()
+  logger.info('Uploading to Azure Blob Storage', {
+    containerName: config.containerName,
+    key: uniqueKey,
+    contentType,
+    fileSize,
+  })
   const containerClient = blobServiceClient.getContainerClient(config.containerName)
   const blockBlobClient = containerClient.getBlockBlobClient(uniqueKey)