⚡ (engine) Improve engine overall robustness

2023-01-25 11:27:47 +01:00
parent ff62b922a0
commit 30baa611e5
210 changed files with 1820 additions and 1919 deletions
--- a/packages/models/features/blocks/integrations/googleSheets/schemas.ts
+++ b/packages/models/features/blocks/integrations/googleSheets/schemas.ts
@ -0,0 +1,126 @@
+import { z } from 'zod'
+import { ComparisonOperators, LogicalOperator } from '../../logic/condition'
+import cuid from 'cuid'
+import { IntegrationBlockType } from '../enums'
+import { GoogleSheetsAction } from './enums'
+import { blockBaseSchema } from '../../baseSchemas'
+
+const cellSchema = z.object({
+  column: z.string().optional(),
+  value: z.string().optional(),
+  id: z.string(),
+})
+
+const extractingCellSchema = z.object({
+  column: z.string().optional(),
+  id: z.string(),
+  variableId: z.string().optional(),
+})
+
+const googleSheetsOptionsBaseSchema = z.object({
+  credentialsId: z.string().optional(),
+  sheetId: z.string().optional(),
+  spreadsheetId: z.string().optional(),
+})
+
+const rowsFilterComparisonSchema = z.object({
+  id: z.string(),
+  column: z.string().optional(),
+  comparisonOperator: z.nativeEnum(ComparisonOperators).optional(),
+  value: z.string().optional(),
+})
+
+const googleSheetsGetOptionsSchema = googleSheetsOptionsBaseSchema.and(
+  z.object({
+    action: z.enum([GoogleSheetsAction.GET]),
+    // TODO: remove referenceCell once migrated to filtering
+    referenceCell: cellSchema.optional(),
+    filter: z.object({
+      comparisons: z.array(rowsFilterComparisonSchema),
+      logicalOperator: z.nativeEnum(LogicalOperator),
+    }),
+    cellsToExtract: z.array(extractingCellSchema),
+  })
+)
+
+const googleSheetsInsertRowOptionsSchema = googleSheetsOptionsBaseSchema.and(
+  z.object({
+    action: z.enum([GoogleSheetsAction.INSERT_ROW]),
+    cellsToInsert: z.array(cellSchema),
+  })
+)
+
+const googleSheetsUpdateRowOptionsSchema = googleSheetsOptionsBaseSchema.and(
+  z.object({
+    action: z.enum([GoogleSheetsAction.UPDATE_ROW]),
+    cellsToUpsert: z.array(cellSchema),
+    referenceCell: cellSchema.optional(),
+  })
+)
+
+export const googleSheetsOptionsSchema = googleSheetsGetOptionsSchema
+  .or(googleSheetsInsertRowOptionsSchema)
+  .or(googleSheetsUpdateRowOptionsSchema)
+  .or(googleSheetsOptionsBaseSchema)
+
+export const googleSheetsBlockSchema = blockBaseSchema.and(
+  z.object({
+    type: z.enum([IntegrationBlockType.GOOGLE_SHEETS]),
+    options: googleSheetsOptionsSchema,
+  })
+)
+
+export const defaultGoogleSheetsOptions: GoogleSheetsOptions = {}
+
+export const defaultGoogleSheetsGetOptions: GoogleSheetsGetOptions = {
+  action: GoogleSheetsAction.GET,
+  cellsToExtract: [
+    {
+      id: cuid(),
+    },
+  ],
+  filter: {
+    comparisons: [
+      {
+        id: cuid(),
+      },
+    ],
+    logicalOperator: LogicalOperator.AND,
+  },
+}
+
+export const defaultGoogleSheetsInsertOptions: GoogleSheetsInsertRowOptions = {
+  action: GoogleSheetsAction.INSERT_ROW,
+  cellsToInsert: [
+    {
+      id: cuid(),
+    },
+  ],
+}
+
+export const defaultGoogleSheetsUpdateOptions: GoogleSheetsUpdateRowOptions = {
+  action: GoogleSheetsAction.UPDATE_ROW,
+  cellsToUpsert: [
+    {
+      id: cuid(),
+    },
+  ],
+}
+
+export type GoogleSheetsBlock = z.infer<typeof googleSheetsBlockSchema>
+export type GoogleSheetsOptions = z.infer<typeof googleSheetsOptionsSchema>
+export type GoogleSheetsOptionsBase = z.infer<
+  typeof googleSheetsOptionsBaseSchema
+>
+export type GoogleSheetsGetOptions = z.infer<
+  typeof googleSheetsGetOptionsSchema
+>
+export type GoogleSheetsInsertRowOptions = z.infer<
+  typeof googleSheetsInsertRowOptionsSchema
+>
+export type GoogleSheetsUpdateRowOptions = z.infer<
+  typeof googleSheetsUpdateRowOptionsSchema
+>
+export type Cell = z.infer<typeof cellSchema>
+export type ExtractingCell = z.infer<typeof extractingCellSchema>
+export type RowsFilterComparison = z.infer<typeof rowsFilterComparisonSchema>