From 1cb30e3f26e9c78f286a4baa968e1501301a4ed1 Mon Sep 17 00:00:00 2001
From: Baptiste Arnaud <contact@baptiste-arnaud.fr>
Date: Thu, 28 Sep 2023 16:30:54 +0200
Subject: [PATCH] :children_crossing: (pictureChoice) Allow dynamic picture
 choice with string variables

---
 .../components/GoogleSheetsNodeContent.tsx    | 36 ++++++++++++++-----
 .../components/GoogleSheetsSettings.tsx       | 14 ++++----
 .../nodes/block/BlockNodeContent.tsx          |  6 +---
 ...njectVariableValuesInPictureChoiceBlock.ts | 35 +++++++++++-------
 4 files changed, 59 insertions(+), 32 deletions(-)

diff --git a/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsNodeContent.tsx b/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsNodeContent.tsx
index b0d3006df4..f1009e7a8c 100644
--- a/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsNodeContent.tsx
+++ b/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsNodeContent.tsx
@@ -1,13 +1,33 @@
 import React from 'react'
-import { Text } from '@chakra-ui/react'
-import { GoogleSheetsAction } from '@typebot.io/schemas'
+import { Stack, Text } from '@chakra-ui/react'
+import { GoogleSheetsAction, GoogleSheetsOptions } from '@typebot.io/schemas'
+import { useTypebot } from '@/features/editor/providers/TypebotProvider'
+import { SetVariableLabel } from '@/components/SetVariableLabel'
 
 type Props = {
-  action?: GoogleSheetsAction
+  options?: GoogleSheetsOptions
 }
 
-export const GoogleSheetsNodeContent = ({ action }: Props) => (
-  <Text color={action ? 'currentcolor' : 'gray.500'} noOfLines={1}>
-    {action ?? 'Configure...'}
-  </Text>
-)
+export const GoogleSheetsNodeContent = ({ options }: Props) => {
+  const { typebot } = useTypebot()
+  return (
+    <Stack>
+      <Text color={options?.action ? 'currentcolor' : 'gray.500'} noOfLines={1}>
+        {options?.action ?? 'Configure...'}
+      </Text>
+      {typebot &&
+        options?.action === GoogleSheetsAction.GET &&
+        options?.cellsToExtract
+          .map((mapping) => mapping.variableId)
+          .map((variableId, idx) =>
+            variableId ? (
+              <SetVariableLabel
+                key={variableId + idx}
+                variables={typebot.variables}
+                variableId={variableId}
+              />
+            ) : null
+          )}
+    </Stack>
+  )
+}
diff --git a/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsSettings.tsx b/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsSettings.tsx
index cf69b3c1ce..1c089429f9 100644
--- a/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsSettings.tsx
+++ b/apps/builder/src/features/blocks/integrations/googleSheets/components/GoogleSheetsSettings.tsx
@@ -304,12 +304,17 @@ const ActionOptions = ({
                 <AccordionItem>
                   <AccordionButton>
                     <Text w="full" textAlign="left">
-                      Rows to filter
+                      Select row(s)
                     </Text>
                     <AccordionIcon />
                   </AccordionButton>
 
-                  <AccordionPanel pt="4">
+                  <AccordionPanel pt="4" as={Stack}>
+                    <DropdownList
+                      items={totalRowsToExtractOptions}
+                      currentItem={options.totalRowsToExtract ?? 'All'}
+                      onItemSelect={updateTotalRowsToExtract}
+                    />
                     <RowsFilterTableList
                       columns={sheet?.columns ?? []}
                       filter={options.filter}
@@ -317,11 +322,6 @@ const ActionOptions = ({
                     />
                   </AccordionPanel>
                 </AccordionItem>
-                <DropdownList
-                  items={totalRowsToExtractOptions}
-                  currentItem={options.totalRowsToExtract ?? 'All'}
-                  onItemSelect={updateTotalRowsToExtract}
-                />
               </>
             )}
 
diff --git a/apps/builder/src/features/graph/components/nodes/block/BlockNodeContent.tsx b/apps/builder/src/features/graph/components/nodes/block/BlockNodeContent.tsx
index 60c3edb88e..753391198f 100644
--- a/apps/builder/src/features/graph/components/nodes/block/BlockNodeContent.tsx
+++ b/apps/builder/src/features/graph/components/nodes/block/BlockNodeContent.tsx
@@ -159,11 +159,7 @@ export const BlockNodeContent = ({ block, indices }: Props): JSX.Element => {
     case LogicBlockType.CONDITION:
       return <ItemNodesList block={block} indices={indices} />
     case IntegrationBlockType.GOOGLE_SHEETS: {
-      return (
-        <GoogleSheetsNodeContent
-          action={'action' in block.options ? block.options.action : undefined}
-        />
-      )
+      return <GoogleSheetsNodeContent options={block.options} />
     }
     case IntegrationBlockType.GOOGLE_ANALYTICS: {
       return <GoogleAnalyticsNodeBody action={block.options?.action} />
diff --git a/packages/bot-engine/blocks/inputs/pictureChoice/injectVariableValuesInPictureChoiceBlock.ts b/packages/bot-engine/blocks/inputs/pictureChoice/injectVariableValuesInPictureChoiceBlock.ts
index 8d29496eaf..695310137d 100644
--- a/packages/bot-engine/blocks/inputs/pictureChoice/injectVariableValuesInPictureChoiceBlock.ts
+++ b/packages/bot-engine/blocks/inputs/pictureChoice/injectVariableValuesInPictureChoiceBlock.ts
@@ -20,8 +20,7 @@ export const injectVariableValuesInPictureChoiceBlock =
           variable.id === block.options.dynamicItems?.pictureSrcsVariableId &&
           isDefined(variable.value)
       ) as VariableWithValue | undefined
-      if (!pictureSrcsVariable || typeof pictureSrcsVariable.value === 'string')
-        return block
+      if (!pictureSrcsVariable) return block
       const titlesVariable = block.options.dynamicItems.titlesVariableId
         ? (variables.find(
             (variable) =>
@@ -29,6 +28,10 @@ export const injectVariableValuesInPictureChoiceBlock =
               isDefined(variable.value)
           ) as VariableWithValue | undefined)
         : undefined
+      const titlesVariableValues =
+        typeof titlesVariable?.value === 'string'
+          ? [titlesVariable.value]
+          : titlesVariable?.value
       const descriptionsVariable = block.options.dynamicItems
         .descriptionsVariableId
         ? (variables.find(
@@ -38,18 +41,26 @@ export const injectVariableValuesInPictureChoiceBlock =
               isDefined(variable.value)
           ) as VariableWithValue | undefined)
         : undefined
+      const descriptionsVariableValues =
+        typeof descriptionsVariable?.value === 'string'
+          ? [descriptionsVariable.value]
+          : descriptionsVariable?.value
+
+      const variableValues =
+        typeof pictureSrcsVariable.value === 'string'
+          ? [pictureSrcsVariable.value]
+          : pictureSrcsVariable.value
+
       return {
         ...block,
-        items: pictureSrcsVariable.value
-          .filter(isDefined)
-          .map((pictureSrc, idx) => ({
-            id: idx.toString(),
-            type: ItemType.PICTURE_CHOICE,
-            blockId: block.id,
-            pictureSrc,
-            title: titlesVariable?.value?.[idx] ?? '',
-            description: descriptionsVariable?.value?.[idx] ?? '',
-          })),
+        items: variableValues.filter(isDefined).map((pictureSrc, idx) => ({
+          id: idx.toString(),
+          type: ItemType.PICTURE_CHOICE,
+          blockId: block.id,
+          pictureSrc,
+          title: titlesVariableValues?.[idx] ?? '',
+          description: descriptionsVariableValues?.[idx] ?? '',
+        })),
       }
     }
     return deepParseVariables(variables)(