127 lines
3.5 KiB
TypeScript
127 lines
3.5 KiB
TypeScript
import { z } from 'zod'
|
|
import { ComparisonOperators, LogicalOperator } from '../../logic/condition'
|
|
import { createId } from '@paralleldrive/cuid2'
|
|
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: createId(),
|
|
},
|
|
],
|
|
filter: {
|
|
comparisons: [
|
|
{
|
|
id: createId(),
|
|
},
|
|
],
|
|
logicalOperator: LogicalOperator.AND,
|
|
},
|
|
}
|
|
|
|
export const defaultGoogleSheetsInsertOptions: GoogleSheetsInsertRowOptions = {
|
|
action: GoogleSheetsAction.INSERT_ROW,
|
|
cellsToInsert: [
|
|
{
|
|
id: createId(),
|
|
},
|
|
],
|
|
}
|
|
|
|
export const defaultGoogleSheetsUpdateOptions: GoogleSheetsUpdateRowOptions = {
|
|
action: GoogleSheetsAction.UPDATE_ROW,
|
|
cellsToUpsert: [
|
|
{
|
|
id: createId(),
|
|
},
|
|
],
|
|
}
|
|
|
|
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>
|