@divyacon/piece-google-drive 0.0.0-stage → 0.1.0

package/src/lib/triggers/new-file.ts

@@ -0,0 +1,119 @@
+import {
+  AppConnectionValueForAuthProperty,
+  Property,
+  createTrigger,
+} from '@agnocon/pieces-framework';
+import { TriggerStrategy } from '@agnocon/pieces-framework';
+import {
+  DedupeStrategy,
+  Polling,
+  pollingHelper,
+} from '@agnocon/pieces-common';
+
+import dayjs from 'dayjs';
+import { googleDriveAuth } from '../auth';
+import { common } from '../common';
+import { downloadFileFromDrive } from '../common/get-file-content';
+import { newFileTriggerOutputSchema } from '../output-schemas';
+
+const polling: Polling<
+  AppConnectionValueForAuthProperty<typeof googleDriveAuth>,
+  { parentFolder?: any; include_team_drives?: boolean }
+> = {
+  strategy: DedupeStrategy.TIMEBASED,
+  items: async ({ auth, propsValue, lastFetchEpochMS }) => {
+    const currentValues =
+      (await common.getFiles(auth, {
+        parent: propsValue.parentFolder,
+        createdTime: lastFetchEpochMS,
+        includeTeamDrive: propsValue.include_team_drives,
+      })) ?? [];
+    const items = currentValues.map((item: any) => ({
+      epochMilliSeconds: dayjs(item.createdTime).valueOf(),
+      data: item,
+    }));
+    return items;
+  },
+};
+
+export const newFile = createTrigger({
+  auth: googleDriveAuth,
+  name: 'new_file',
+  displayName: 'New File',
+  description: 'Trigger when a new file is uploaded.',
+  aiMetadata: {
+    description: 'Fires when a new file appears in Google Drive, optionally scoped to a specific parent folder. Each event represents one newly created file and its metadata, with optional inclusion of the file content.',
+  },
+  props: {
+    parentFolder: common.properties.parentFolder,
+    include_team_drives: common.properties.include_team_drives,
+    include_file_content: Property.Checkbox({
+      displayName: 'Include File Content',
+      description: 'Include the file content in the output. This will increase the time taken to fetch the files and might cause issues with large files.',
+      required: false,
+      defaultValue: false
+    }),
+  },
+  outputSchema: newFileTriggerOutputSchema,
+  type: TriggerStrategy.POLLING,
+  onEnable: async (context) => {
+    await pollingHelper.onEnable(polling, {
+      auth: context.auth,
+      store: context.store,
+      propsValue: context.propsValue,
+    });
+  },
+  onDisable: async (context) => {
+    await pollingHelper.onDisable(polling, {
+      auth: context.auth,
+      store: context.store,
+      propsValue: context.propsValue,
+    });
+  },
+  run: async (context) => {
+    const newFiles = await pollingHelper.poll(polling, {
+      auth: context.auth,
+      store: context.store,
+      propsValue: context.propsValue,
+      files: context.files,
+    });
+
+    return await handleFileContent(newFiles, context)
+  },
+  test: async (context) => {
+    const newFiles = await pollingHelper.test(polling, {
+      auth: context.auth,
+      store: context.store,
+      propsValue: context.propsValue,
+      files: context.files,
+    });
+
+    return await handleFileContent(newFiles, context)
+  },
+
+  sampleData: {
+    kind: 'drive#file',
+    mimeType: 'image/png',
+    id: '1dpv4-sKJfKRwI9qx1vWqQhEGEn3EpbI5',
+    name: 'google-drive.png',
+    link: 'https://cdn.activepieces.com/pieces/google-drive.png'
+  },
+});
+
+async function handleFileContent(newFiles: unknown[], context: any) {
+  const newFilesObj = JSON.parse(JSON.stringify(newFiles))
+
+  if (context.propsValue.include_file_content) {
+    const fileContentPromises: Promise<string>[] = []
+    for (const file of newFilesObj) {
+      fileContentPromises.push(downloadFileFromDrive(context.auth, context.files, file["id"], file["name"]));
+    }
+
+    const filesContent = await Promise.all(fileContentPromises)
+
+    for (let i = 0; i < newFilesObj.length; i++) {
+      newFilesObj[i].content = filesContent[i]
+    }
+  }
+  return newFilesObj
+}

package/src/lib/triggers/new-folder.ts

@@ -0,0 +1,78 @@
+import {
+  AppConnectionValueForAuthProperty,
+  createTrigger,
+} from '@agnocon/pieces-framework';
+import { TriggerStrategy } from '@agnocon/pieces-framework';
+import {
+  DedupeStrategy,
+  Polling,
+  pollingHelper,
+} from '@agnocon/pieces-common';
+
+import dayjs from 'dayjs';
+import { googleDriveAuth } from '../auth';
+import { common } from '../common';
+import { newFolderTriggerOutputSchema } from '../output-schemas';
+
+const polling: Polling<
+  AppConnectionValueForAuthProperty<typeof googleDriveAuth>,
+  { parentFolder?: any,include_team_drives?:boolean }
+> = {
+  strategy: DedupeStrategy.TIMEBASED,
+  items: async ({ auth, propsValue, lastFetchEpochMS }) => {
+    const currentValues =
+      (await common.getFolders(auth, {
+        parent: propsValue.parentFolder,
+        createdTime: lastFetchEpochMS,
+        includeTeamDrive:propsValue.include_team_drives
+      })) ?? [];
+    const items = currentValues.map((item: any) => ({
+      epochMilliSeconds: dayjs(item.createdTime).valueOf(),
+      data: item,
+    }));
+    return items;
+  },
+};
+
+export const newFolder = createTrigger({
+  auth: googleDriveAuth,
+  name: 'new_folder',
+  displayName: 'New Folder',
+  description: 'Trigger when a new folder is created or uploaded.',
+  aiMetadata: {
+    description: 'Fires when a new folder is created in Google Drive, optionally scoped to a specific parent folder. Each event represents one newly created folder and its metadata.',
+  },
+  props: {
+    parentFolder: common.properties.parentFolder,
+    include_team_drives: common.properties.include_team_drives,
+  },
+  outputSchema: newFolderTriggerOutputSchema,
+  type: TriggerStrategy.POLLING,
+  onEnable: async (context) => {
+    await pollingHelper.onEnable(polling, {
+      auth: context.auth,
+      store: context.store,
+      propsValue: context.propsValue,
+    });
+  },
+  onDisable: async (context) => {
+    await pollingHelper.onDisable(polling, {
+      auth: context.auth,
+      store: context.store,
+      propsValue: context.propsValue,
+    });
+  },
+  run: async (context) => {
+    return await pollingHelper.poll(polling, context);
+  },
+  test: async (context) => {
+    return await pollingHelper.test(polling, context);
+  },
+
+  sampleData: {
+    kind: 'drive#file',
+    mimeType: 'application/vnd.google-apps.folder',
+    id: '1aMEtTqIYn5651wdK7WLxaK_SDim4mvXW',
+    name: 'New Folder WOOOO',
+  },
+});

package/README.md

@@ -1,3 +0,0 @@
-# Temporary Holding Version
-
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.