expanded metadata fields for google drive

Author: aadamgough

apps/sim/tools/google_drive/download.ts

@@ -1,14 +1,115 @@
 import { createLogger } from '@sim/logger'
-import type { GoogleDriveDownloadResponse, GoogleDriveToolParams } from '@/tools/google_drive/types'
+import type {
+  GoogleDriveDownloadResponse,
+  GoogleDriveFile,
+  GoogleDriveRevision,
+  GoogleDriveToolParams,
+} from '@/tools/google_drive/types'
 import { DEFAULT_EXPORT_FORMATS, GOOGLE_WORKSPACE_MIME_TYPES } from '@/tools/google_drive/utils'
 import type { ToolConfig } from '@/tools/types'
 
 const logger = createLogger('GoogleDriveDownloadTool')
 
+// All available file metadata fields from Google Drive API v3
+const ALL_FILE_FIELDS = [
+  // Basic Info
+  'id',
+  'name',
+  'mimeType',
+  'kind',
+  'description',
+  'originalFilename',
+  'fullFileExtension',
+  'fileExtension',
+  // Ownership & Sharing
+  'owners',
+  'permissions',
+  'permissionIds',
+  'shared',
+  'ownedByMe',
+  'writersCanShare',
+  'viewersCanCopyContent',
+  'copyRequiresWriterPermission',
+  'sharingUser',
+  // Labels/Tags
+  'starred',
+  'trashed',
+  'explicitlyTrashed',
+  'properties',
+  'appProperties',
+  'folderColorRgb',
+  // Timestamps
+  'createdTime',
+  'modifiedTime',
+  'modifiedByMeTime',
+  'viewedByMeTime',
+  'sharedWithMeTime',
+  'trashedTime',
+  // User Info
+  'lastModifyingUser',
+  'trashingUser',
+  'viewedByMe',
+  'modifiedByMe',
+  // Links
+  'webViewLink',
+  'webContentLink',
+  'iconLink',
+  'thumbnailLink',
+  'exportLinks',
+  // Size & Storage
+  'size',
+  'quotaBytesUsed',
+  // Checksums
+  'md5Checksum',
+  'sha1Checksum',
+  'sha256Checksum',
+  // Hierarchy & Location
+  'parents',
+  'spaces',
+  'driveId',
+  'teamDriveId',
+  // Capabilities
+  'capabilities',
+  // Versions
+  'version',
+  'headRevisionId',
+  // Media Metadata
+  'hasThumbnail',
+  'thumbnailVersion',
+  'imageMediaMetadata',
+  'videoMediaMetadata',
+  'contentHints',
+  // Other
+  'isAppAuthorized',
+  'contentRestrictions',
+  'resourceKey',
+  'shortcutDetails',
+  'linkShareMetadata',
+].join(',')
+
+// All revision fields
+const ALL_REVISION_FIELDS = [
+  'id',
+  'mimeType',
+  'modifiedTime',
+  'keepForever',
+  'published',
+  'publishAuto',
+  'publishedLink',
+  'publishedOutsideDomain',
+  'lastModifyingUser',
+  'originalFilename',
+  'md5Checksum',
+  'size',
+  'exportLinks',
+  'kind',
+].join(',')
+
 export const downloadTool: ToolConfig<GoogleDriveToolParams, GoogleDriveDownloadResponse> = {
   id: 'google_drive_download',
   name: 'Download File from Google Drive',
-  description: 'Download a file from Google Drive (exports Google Workspace files automatically)',
+  description:
+    'Download a file from Google Drive with complete metadata (exports Google Workspace files automatically)',
   version: '1.0',
 
   oauth: {
@@ -41,11 +142,17 @@ export const downloadTool: ToolConfig<GoogleDriveToolParams, GoogleDriveDownload
       visibility: 'user-only',
       description: 'Optional filename override',
     },
+    includeRevisions: {
+      type: 'boolean',
+      required: false,
+      visibility: 'user-or-llm',
+      description: 'Whether to include revision history in the metadata (default: true)',
+    },
   },
 
   request: {
     url: (params) =>
-      `https://www.googleapis.com/drive/v3/files/${params.fileId}?fields=id,name,mimeType&supportsAllDrives=true`,
+      `https://www.googleapis.com/drive/v3/files/${params.fileId}?fields=${ALL_FILE_FIELDS}&supportsAllDrives=true`,
     method: 'GET',
     headers: (params) => ({
       Authorization: `Bearer ${params.accessToken}`,
@@ -64,7 +171,7 @@ export const downloadTool: ToolConfig<GoogleDriveToolParams, GoogleDriveDownload
         throw new Error(errorDetails.error?.message || 'Failed to get file metadata')
       }
 
-      const metadata = await response.json()
+      const metadata: GoogleDriveFile = await response.json()
       const fileId = metadata.id
       const mimeType = metadata.mimeType
       const authHeader = `Bearer ${params?.accessToken || ''}`
@@ -132,13 +239,49 @@ export const downloadTool: ToolConfig<GoogleDriveToolParams, GoogleDriveDownload
         fileBuffer = Buffer.from(arrayBuffer)
       }
 
+      // Fetch revisions if requested (default: true)
+      const includeRevisions = params?.includeRevisions !== false
+      if (includeRevisions) {
+        try {
+          const revisionsResponse = await fetch(
+            `https://www.googleapis.com/drive/v3/files/${fileId}/revisions?fields=revisions(${ALL_REVISION_FIELDS})&pageSize=100`,
+            {
+              headers: {
+                Authorization: authHeader,
+              },
+            }
+          )
+
+          if (revisionsResponse.ok) {
+            const revisionsData = await revisionsResponse.json()
+            metadata.revisions = revisionsData.revisions as GoogleDriveRevision[]
+            logger.info('Fetched file revisions', {
+              fileId,
+              revisionCount: metadata.revisions?.length || 0,
+            })
+          } else {
+            logger.warn('Failed to fetch revisions, continuing without them', {
+              status: revisionsResponse.status,
+              statusText: revisionsResponse.statusText,
+            })
+          }
+        } catch (revisionError: any) {
+          logger.warn('Error fetching revisions, continuing without them', {
+            error: revisionError.message,
+          })
+        }
+      }
+
       const resolvedName = params?.fileName || metadata.name || 'download'
 
       logger.info('File downloaded successfully', {
         fileId,
         name: resolvedName,
         size: fileBuffer.length,
         mimeType: finalMimeType,
+        hasOwners: !!metadata.owners?.length,
+        hasPermissions: !!metadata.permissions?.length,
+        hasRevisions: !!metadata.revisions?.length,
       })
 
       return {
@@ -150,6 +293,7 @@ export const downloadTool: ToolConfig<GoogleDriveToolParams, GoogleDriveDownload
             data: fileBuffer,
             size: fileBuffer.length,
           },
+          metadata,
         },
       }
     } catch (error: any) {
@@ -163,5 +307,10 @@ export const downloadTool: ToolConfig<GoogleDriveToolParams, GoogleDriveDownload
 
   outputs: {
     file: { type: 'file', description: 'Downloaded file stored in execution files' },
+    metadata: {
+      type: 'json',
+      description:
+        'Complete file metadata including ownership, sharing, permissions, labels, checksums, capabilities, and revision history',
+    },
   },
 }