feat(integration): ✨ Add Google Sheets integration

2022-01-18 18:25:18 +01:00
parent 2814a352b2
commit f49b5143cf
67 changed files with 2560 additions and 391 deletions
--- a/packages/bot-engine/src/services/integration.ts
+++ b/packages/bot-engine/src/services/integration.ts
@ -0,0 +1,121 @@
+import {
+  IntegrationStep,
+  IntegrationStepType,
+  GoogleSheetsStep,
+  GoogleSheetsAction,
+  GoogleSheetsInsertRowOptions,
+  Variable,
+  Table,
+  GoogleSheetsUpdateRowOptions,
+  Cell,
+  GoogleSheetsGetOptions,
+} from 'models'
+import { stringify } from 'qs'
+import { sendRequest } from 'utils'
+import { parseVariables } from './variable'
+
+export const executeIntegration = (
+  step: IntegrationStep,
+  variables: Table<Variable>,
+  updateVariableValue: (variableId: string, value: string) => void
+) => {
+  switch (step.type) {
+    case IntegrationStepType.GOOGLE_SHEETS:
+      return executeGoogleSheetIntegration(step, variables, updateVariableValue)
+  }
+}
+
+const executeGoogleSheetIntegration = async (
+  step: GoogleSheetsStep,
+  variables: Table<Variable>,
+  updateVariableValue: (variableId: string, value: string) => void
+) => {
+  if (!step.options) return step.target
+  switch (step.options?.action) {
+    case GoogleSheetsAction.INSERT_ROW:
+      await insertRowInGoogleSheets(step.options, variables)
+      break
+    case GoogleSheetsAction.UPDATE_ROW:
+      await updateRowInGoogleSheets(step.options, variables)
+      break
+    case GoogleSheetsAction.GET:
+      await getRowFromGoogleSheets(step.options, variables, updateVariableValue)
+      break
+  }
+  return step.target
+}
+
+const insertRowInGoogleSheets = async (
+  options: GoogleSheetsInsertRowOptions,
+  variables: Table<Variable>
+) => {
+  if (!options.cellsToInsert) return
+  return sendRequest({
+    url: `http://localhost:3001/api/integrations/google-sheets/spreadsheets/${options.spreadsheetId}/sheets/${options.sheetId}`,
+    method: 'POST',
+    body: {
+      credentialsId: options.credentialsId,
+      values: parseCellValues(options.cellsToInsert, variables),
+    },
+  })
+}
+
+const updateRowInGoogleSheets = async (
+  options: GoogleSheetsUpdateRowOptions,
+  variables: Table<Variable>
+) => {
+  if (!options.cellsToUpsert || !options.referenceCell) return
+  return sendRequest({
+    url: `http://localhost:3001/api/integrations/google-sheets/spreadsheets/${options.spreadsheetId}/sheets/${options.sheetId}`,
+    method: 'PATCH',
+    body: {
+      credentialsId: options.credentialsId,
+      values: parseCellValues(options.cellsToUpsert, variables),
+      referenceCell: {
+        column: options.referenceCell.column,
+        value: parseVariables(options.referenceCell.value ?? '', variables),
+      },
+    },
+  })
+}
+
+const getRowFromGoogleSheets = async (
+  options: GoogleSheetsGetOptions,
+  variables: Table<Variable>,
+  updateVariableValue: (variableId: string, value: string) => void
+) => {
+  if (!options.referenceCell || !options.cellsToExtract) return
+  const queryParams = stringify(
+    {
+      credentialsId: options.credentialsId,
+      referenceCell: {
+        column: options.referenceCell.column,
+        value: parseVariables(options.referenceCell.value ?? '', variables),
+      },
+      columns: options.cellsToExtract.allIds.map(
+        (id) => options.cellsToExtract?.byId[id].column
+      ),
+    },
+    { indices: false }
+  )
+  const { data } = await sendRequest<{ [key: string]: string }>({
+    url: `http://localhost:3001/api/integrations/google-sheets/spreadsheets/${options.spreadsheetId}/sheets/${options.sheetId}?${queryParams}`,
+    method: 'GET',
+  })
+  if (!data) return
+  options.cellsToExtract.allIds.forEach((cellId) => {
+    const cell = options.cellsToExtract?.byId[cellId]
+    if (!cell) return
+    updateVariableValue(cell.variableId ?? '', data[cell.column ?? ''])
+  })
+}
+const parseCellValues = (
+  cells: Table<Cell>,
+  variables: Table<Variable>
+): { [key: string]: string } =>
+  cells.allIds.reduce((row, id) => {
+    const cell = cells.byId[id]
+    return !cell.column || !cell.value
+      ? row
+      : { ...row, [cell.column]: parseVariables(cell.value, variables) }
+  }, {})