Google Drive

GOOGLE_DRIVE_CLIENT_ID=123456789-abc.apps.googleusercontent.com
GOOGLE_DRIVE_CLIENT_SECRET=GOCSPX-abcdefghijklmnop

integrations:
  google-drive:
    enabled: true
    credentials:
      client_id: ${GOOGLE_DRIVE_CLIENT_ID}
      client_secret: ${GOOGLE_DRIVE_CLIENT_SECRET}
    settings:
      # OAuth redirect URI (must match Google Cloud Console)
      redirect_uri: http://localhost:3000/auth/google/callback
      # Scopes to request during authorization
      scopes:
        - https://www.googleapis.com/auth/drive
        - https://www.googleapis.com/auth/drive.file
      # Root folder ID to scope operations (optional, defaults to entire drive)
      root_folder_id: ""
      # Enable push notifications for real-time file change events
      watch_changes: true
      # Webhook URL for change notifications (required if watch_changes is true)
      webhook_url: https://your-domain.com/webhooks/google-drive
      # Maximum file upload size in MB (default 100)
      max_upload_size_mb: 100

drive.on('file:created', async (event) => {
  console.log(`New file: ${event.name} (by ${event.createdBy})`)
})

drive.on('file:modified', async (event) => {
  console.log(`File updated: ${event.name} — version ${event.version}`)
})

drive.on('file:deleted', async (event) => {
  console.log(`File trashed: ${event.name}`)
})

drive.on('file:shared', async (event) => {
  console.log(`${event.name} shared with ${event.sharedWith} as ${event.role}`)
})

import { Cognest } from '@cognest/sdk'
import { readFile } from 'fs/promises'

const cognest = new Cognest()
const drive = cognest.integration('google-drive')
const slack = cognest.integration('slack')

const INBOX_FOLDER = '1AbCdEfGhIjKlMnOpQrStUvWxYz' // "Inbox" folder ID
const ARCHIVE_FOLDER = '0ZyXwVuTsRqPoNmLkJiHgFeDcBa' // "Archive" folder ID

// Auto-organize: move files from inbox to date-based folders
drive.on('file:created', async (event) => {
  if (event.folderId !== INBOX_FOLDER) return

  const today = new Date()
  const folderName = `${today.getFullYear()}-${String(today.getMonth() + 1).padStart(2, '0')}`

  // Find or create the monthly folder
  const existing = await drive.searchFiles(
    `name = '${folderName}' and '${ARCHIVE_FOLDER}' in parents and mimeType = 'application/vnd.google-apps.folder'`
  )

  let targetFolderId: string
  if (existing.length > 0) {
    targetFolderId = existing[0].id
  } else {
    const folder = await drive.createFolder({
      name: folderName,
      parentId: ARCHIVE_FOLDER,
      description: `Auto-organized files for ${folderName}`,
    })
    targetFolderId = folder.id
  }

  // Move the file (upload a copy and delete original)
  const content = await drive.downloadFile(event.fileId)
  await drive.uploadFile({
    name: event.name,
    content,
    mimeType: event.mimeType,
    folderId: targetFolderId,
  })
  await drive.deleteFile(event.fileId)

  console.log(`Organized ${event.name} into ${folderName}/`)
})

// Notify Slack when important files are shared externally
drive.on('file:shared', async (event) => {
  // Only alert if shared with someone outside the organization
  if (!event.sharedWith.endsWith('@your-company.com')) {
    await slack.sendMessage({
      channel: '#security-alerts',
      text: `External share detected: "${event.name}" shared with ${event.sharedWith} (${event.role}) by ${event.sharedBy}`,
    })
  }
})

// Daily backup report
cognest.skill('drive-report', {
  schedule: '0 18 * * 1-5', // 6 PM weekdays
  handler: async () => {
    const recentFiles = await drive.listFiles({
      pageSize: 50,
      orderBy: 'modifiedTime desc',
    })

    const todayStart = new Date()
    todayStart.setHours(0, 0, 0, 0)

    const modifiedToday = recentFiles.filter(
      (f) => new Date(f.modifiedTime) >= todayStart
    )

    await slack.sendMessage({
      channel: '#team-updates',
      text: [
        ':file_folder: *Daily Drive Summary*',
        `Files modified today: ${modifiedToday.length}`,
        ...modifiedToday.slice(0, 10).map(
          (f) => `  - ${f.name} (last edited by ${f.lastModifyingUser?.displayName ?? 'unknown'})`
        ),
        modifiedToday.length > 10 ? `  ...and ${modifiedToday.length - 10} more` : '',
      ].filter(Boolean).join('\n'),
    })
  },
})

// Skill: Upload a local file to a specific Drive folder
cognest.skill('upload-to-drive', {
  handler: async ({ params }) => {
    const { filePath, folderId, shareWith } = params

    const content = await readFile(filePath)
    const name = filePath.split('/').pop() ?? 'upload'

    const file = await drive.uploadFile({
      name,
      content,
      mimeType: 'application/octet-stream',
      folderId,
    })

    // Optionally share with a collaborator
    if (shareWith) {
      await drive.shareFile(file.id, {
        email: shareWith,
        role: 'writer',
        type: 'user',
        sendNotification: true,
      })
    }

    return { fileId: file.id, url: file.webViewLink }
  },
})

await cognest.start()
console.log('Google Drive automation is running...')