From 23f0581ad5e867ad88ee092191f5ec1406b3cf40 Mon Sep 17 00:00:00 2001
From: Kevin Foong <55353265+kevin9foong@users.noreply.github.com>
Date: Tue, 24 Dec 2024 10:45:40 +0800
Subject: [PATCH] feat: add accept deny modal and created field ids from mfb

---
 .../assistance/AssistanceService.ts           |   2 +-
 .../admin-form/assistance/mutations.ts        |   7 +-
 .../BuilderAndDesignContent/BuilderFields.tsx |   9 +
 .../FieldRow/FieldRowContainer.tsx            |  31 +++-
 .../MagicFormBuilderContainer.tsx             | 168 +++++++++++++-----
 .../useMagicFormBuilderStore.tsx              |  22 +++
 .../admin-form.assistance.controller.ts       |   5 +-
 .../admin-form.assistance.service.ts          |  10 +-
 8 files changed, 200 insertions(+), 54 deletions(-)
 create mode 100644 frontend/src/features/admin-form/create/builder-and-design/useMagicFormBuilderStore.tsx

diff --git a/frontend/src/features/admin-form/assistance/AssistanceService.ts b/frontend/src/features/admin-form/assistance/AssistanceService.ts
index de49afca22..68b1b24abd 100644
--- a/frontend/src/features/admin-form/assistance/AssistanceService.ts
+++ b/frontend/src/features/admin-form/assistance/AssistanceService.ts
@@ -9,7 +9,7 @@ export const makeTextPrompt = ({
   formId: string
   prompt: string
 }) => {
-  return ApiService.post<undefined>(
+  return ApiService.post<{ message: string; createdFieldIds?: string[] }>(
     `${ADMIN_FORM_ENDPOINT}/${formId}/assistance/text-prompt`,
     { prompt },
   ).then(({ data }) => data)
diff --git a/frontend/src/features/admin-form/assistance/mutations.ts b/frontend/src/features/admin-form/assistance/mutations.ts
index d5c87cfd2b..823cff5048 100644
--- a/frontend/src/features/admin-form/assistance/mutations.ts
+++ b/frontend/src/features/admin-form/assistance/mutations.ts
@@ -5,6 +5,7 @@ import { makeTextPrompt } from '~features/admin-form/assistance/AssistanceServic
 
 import { useToast } from '../../../hooks/useToast'
 import { adminFormKeys } from '../common/queries'
+import { useMagicFormBuilderStore } from '../create/builder-and-design/useMagicFormBuilderStore'
 
 export const useAssistanceMutations = () => {
   const { formId } = useParams()
@@ -19,7 +20,11 @@ export const useAssistanceMutations = () => {
   const useMakeTextPromptMutation = useMutation(
     (prompt: string) => makeTextPrompt({ formId, prompt }),
     {
-      onSuccess: () => {
+      onSuccess: (data) => {
+        const { createdFieldIds } = data
+        useMagicFormBuilderStore.setState({
+          recentlyCreatedFieldIds: new Set(createdFieldIds),
+        })
         queryClient.invalidateQueries(adminFormKeys.id(formId))
         toast.closeAll()
         toast({ description: 'Form generated successfully', status: 'success' })
diff --git a/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/BuilderFields.tsx b/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/BuilderFields.tsx
index 5c87f5cf8c..e2c0eeddd8 100644
--- a/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/BuilderFields.tsx
+++ b/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/BuilderFields.tsx
@@ -11,6 +11,10 @@ import {
   stateDataSelector,
   useFieldBuilderStore,
 } from '../useFieldBuilderStore'
+import {
+  recentlyCreatedFieldIdsSelector,
+  useMagicFormBuilderStore,
+} from '../useMagicFormBuilderStore'
 import { useDesignColorTheme } from '../utils/useDesignColorTheme'
 
 import FieldRow from './FieldRow'
@@ -44,6 +48,10 @@ export const BuilderFields = ({
 
   const isDirty = useDirtyFieldStore(isDirtySelector)
 
+  const mfbRecentlyCreatedFieldIds = useMagicFormBuilderStore(
+    recentlyCreatedFieldIdsSelector,
+  )
+
   return (
     <>
       {fieldsWithQuestionNos.map((f, i) => {
@@ -64,6 +72,7 @@ export const BuilderFields = ({
             handleBuilderClick={handleBuilderClick}
             isDirty={isDirty}
             colorTheme={colorTheme}
+            isHighlighted={mfbRecentlyCreatedFieldIds.has(f._id)}
             {...activeFieldExtraProps}
           />
         )
diff --git a/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/FieldRow/FieldRowContainer.tsx b/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/FieldRow/FieldRowContainer.tsx
index 55636375cc..3de64c686f 100644
--- a/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/FieldRow/FieldRowContainer.tsx
+++ b/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignContent/FieldRow/FieldRowContainer.tsx
@@ -5,6 +5,7 @@ import { BiCog, BiDuplicate, BiGridHorizontal, BiTrash } from 'react-icons/bi'
 import { useIsMutating } from 'react-query'
 import {
   Box,
+  BoxProps,
   ButtonGroup,
   chakra,
   Collapse,
@@ -94,6 +95,30 @@ export interface FieldRowContainerProps {
   colorTheme?: FormColorTheme
   // handleBuilderClick is passed down to prevent unnecessary re-renders from useContext
   handleBuilderClick: CreatePageSidebarContextProps['handleBuilderClick']
+  isHighlighted?: boolean
+}
+
+/**
+ * Used for highlighting fields that were created by Magic Form Builder.
+ */
+const HighlightableBox = ({
+  children,
+  isHighlighted,
+  ...props
+}: {
+  children: React.ReactNode
+  isHighlighted?: boolean
+} & BoxProps) => {
+  return (
+    <Box
+      borderColor="danger.300"
+      borderRadius="0.25rem"
+      borderWidth={isHighlighted ? '0.125rem' : '0'}
+      {...props}
+    >
+      {children}
+    </Box>
+  )
 }
 
 const FieldRowContainer = ({
@@ -106,6 +131,7 @@ const FieldRowContainer = ({
   isDirty,
   colorTheme,
   handleBuilderClick,
+  isHighlighted,
 }: FieldRowContainerProps): JSX.Element => {
   const isMobile = useIsMobile()
   const numFormFieldMutations = useIsMutating(adminFormKeys.base)
@@ -285,7 +311,8 @@ const FieldRowContainer = ({
                   )}
                 </chakra.button>
               </Fade>
-              <Box
+              <HighlightableBox
+                isHighlighted={isHighlighted}
                 px={{ base: '0.75rem', md: '1.5rem' }}
                 pb={{ base: '0.75rem', md: '1.5rem' }}
                 w="100%"
@@ -300,7 +327,7 @@ const FieldRowContainer = ({
                     showMyInfoBadge={isMyInfoField}
                   />
                 </FormProvider>
-              </Box>
+              </HighlightableBox>
               <Collapse in={isActive} style={{ width: '100%' }}>
                 {isActive && (
                   <FieldButtonGroup
diff --git a/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignDrawer/FieldListDrawer/MagicFormBuilderContainer.tsx b/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignDrawer/FieldListDrawer/MagicFormBuilderContainer.tsx
index fb306cbca2..ac6d898624 100644
--- a/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignDrawer/FieldListDrawer/MagicFormBuilderContainer.tsx
+++ b/frontend/src/features/admin-form/create/builder-and-design/BuilderAndDesignDrawer/FieldListDrawer/MagicFormBuilderContainer.tsx
@@ -1,6 +1,6 @@
 import { useState } from 'react'
 import { useForm } from 'react-hook-form'
-import { BiSolidMagicWand } from 'react-icons/bi'
+import { BiSolidMagicWand, BiTrash } from 'react-icons/bi'
 import { HiSparkles } from 'react-icons/hi2'
 import {
   Button,
@@ -20,11 +20,15 @@ import {
   Tooltip,
 } from '@chakra-ui/react'
 
+import { BxCheck } from '~/assets/icons'
+
 import { useIsMobile } from '~hooks/useIsMobile'
 import { FormErrorMessage } from '~components/FormControl/FormErrorMessage/FormErrorMessage'
 
 import { useAssistanceMutations } from '~features/admin-form/assistance/mutations'
 
+import { useMagicFormBuilderStore } from '../../useMagicFormBuilderStore'
+
 const GENERATE_FORM_PLACEHOLDER =
   'Describe form, fields and sections to create...'
 
@@ -61,16 +65,12 @@ const MagicFormBuilderButton = ({
   )
 }
 
-interface TextPromptInputs {
-  prompt: string
-}
-
-const MagicFormBuilderPopover = ({
-  isOpen,
-  setIsOpen,
+const MagicFormBuilderCreateFormPrompt = ({
+  onSettled,
+  onClose,
 }: {
-  isOpen: boolean
-  setIsOpen: (isOpen: boolean) => void
+  onSettled: () => void
+  onClose: () => void
 }) => {
   const {
     register,
@@ -82,50 +82,130 @@ const MagicFormBuilderPopover = ({
 
   const onSubmit = async ({ prompt }: TextPromptInputs) => {
     useMakeTextPromptMutation.mutate(prompt, {
-      onSettled: () => setIsOpen(false),
+      onSettled: onSettled,
     })
   }
 
   return (
-    <Popover placement="right" isOpen={isOpen}>
+    <form onSubmit={handleSubmit(onSubmit)}>
+      <PopoverHeader>
+        <Text textStyle="h4">Generate form</Text>
+      </PopoverHeader>
+      <PopoverCloseButton onClick={onClose} />
+      <PopoverBody>
+        <FormControl isRequired isInvalid={false}>
+          <Textarea
+            placeholder={GENERATE_FORM_PLACEHOLDER}
+            {...register('prompt', {
+              required: 'Please enter a prompt.',
+              maxLength: {
+                value: 30,
+                message: 'Please enter at most 500 characters.',
+              },
+            })}
+          />
+          <FormErrorMessage>{errors.prompt?.message}</FormErrorMessage>
+        </FormControl>
+      </PopoverBody>
+      <PopoverFooter>
+        <Flex justifyContent="flex-end">
+          <Button
+            leftIcon={<HiSparkles fontSize="1.5rem" />}
+            type="submit"
+            isLoading={useMakeTextPromptMutation.isLoading}
+          >
+            Generate
+          </Button>
+        </Flex>
+      </PopoverFooter>
+    </form>
+  )
+}
+
+const MagicFormBuilderAcceptDeny = ({
+  onAccept,
+  onDeny,
+  onClose,
+}: {
+  onAccept: () => void
+  onDeny: () => void
+  onClose: () => void
+}) => {
+  return (
+    <>
+      <PopoverHeader>
+        <Text textStyle="h4">Keep these changes?</Text>
+      </PopoverHeader>
+      <PopoverCloseButton onClick={onClose} />
+      <PopoverBody>
+        <Flex direction="column" gap="0.25rem">
+          <Button
+            leftIcon={<BxCheck />}
+            variant="solid"
+            onClick={onAccept}
+            colorScheme="success"
+          >
+            Accept
+          </Button>
+          <Button
+            leftIcon={<BiTrash />}
+            variant="solid"
+            onClick={onDeny}
+            colorScheme="danger"
+          >
+            Deny
+          </Button>
+        </Flex>
+      </PopoverBody>
+    </>
+  )
+}
+
+interface TextPromptInputs {
+  prompt: string
+}
+
+const MagicFormBuilderPopover = ({
+  isOpen,
+  setIsOpen,
+}: {
+  isOpen: boolean
+  setIsOpen: (isOpen: boolean) => void
+}) => {
+  const [isAcceptDenyOpen, setIsAcceptDenyOpen] = useState(false)
+  const clearRecentlyCreatedFieldIds = useMagicFormBuilderStore(
+    (state) => state.clearRecentlyCreatedFieldIds,
+  )
+
+  const onClickDefaults = () => {
+    clearRecentlyCreatedFieldIds()
+    setIsOpen(false)
+    setTimeout(() => setIsAcceptDenyOpen(false), 100) // delay to allow popover to close before updating state
+  }
+
+  return (
+    <Popover isLazy placement="right" isOpen={isOpen}>
       <PopoverAnchor>
         <MagicFormBuilderButton onClick={() => setIsOpen(!isOpen)} />
       </PopoverAnchor>
       <Portal>
         {/* TODO: (MFBv1.1) Fix the position of the popover. */}
         <PopoverContent bg="white" top="15vh" left="40vw">
-          <form onSubmit={handleSubmit(onSubmit)}>
-            <PopoverHeader>
-              <Text textStyle="h4">Generate form</Text>
-            </PopoverHeader>
-            <PopoverCloseButton onClick={() => setIsOpen(false)} />
-            <PopoverBody>
-              <FormControl isRequired isInvalid={false}>
-                <Textarea
-                  placeholder={GENERATE_FORM_PLACEHOLDER}
-                  {...register('prompt', {
-                    required: 'Please enter a prompt.',
-                    maxLength: {
-                      value: 30,
-                      message: 'Please enter at most 500 characters.',
-                    },
-                  })}
-                />
-                <FormErrorMessage>{errors.prompt?.message}</FormErrorMessage>
-              </FormControl>
-            </PopoverBody>
-            <PopoverFooter>
-              <Flex justifyContent="flex-end">
-                <Button
-                  leftIcon={<HiSparkles fontSize="1.5rem" />}
-                  type="submit"
-                  isLoading={useMakeTextPromptMutation.isLoading}
-                >
-                  Generate
-                </Button>
-              </Flex>
-            </PopoverFooter>
-          </form>
+          {!isAcceptDenyOpen ? (
+            <MagicFormBuilderCreateFormPrompt
+              onClose={() => setIsOpen(false)}
+              onSettled={() => setIsAcceptDenyOpen(true)}
+            />
+          ) : (
+            <MagicFormBuilderAcceptDeny
+              onAccept={onClickDefaults}
+              onDeny={() => {
+                // trigger deletion of fields with field ids in recentlyCreatedFieldIds
+                onClickDefaults()
+              }}
+              onClose={() => setIsOpen(false)}
+            />
+          )}
         </PopoverContent>
       </Portal>
     </Popover>
diff --git a/frontend/src/features/admin-form/create/builder-and-design/useMagicFormBuilderStore.tsx b/frontend/src/features/admin-form/create/builder-and-design/useMagicFormBuilderStore.tsx
new file mode 100644
index 0000000000..16e58230ba
--- /dev/null
+++ b/frontend/src/features/admin-form/create/builder-and-design/useMagicFormBuilderStore.tsx
@@ -0,0 +1,22 @@
+import create from 'zustand'
+
+export type MagicFormBuilderStore = {
+  recentlyCreatedFieldIds: Set<string>
+  setRecentlyCreatedFieldIds: (fieldIds: Set<string>) => void
+  clearRecentlyCreatedFieldIds: () => void
+}
+
+export const useMagicFormBuilderStore = create<MagicFormBuilderStore>(
+  (set) => ({
+    recentlyCreatedFieldIds: new Set(),
+    setRecentlyCreatedFieldIds: (fieldIds) =>
+      set({ recentlyCreatedFieldIds: fieldIds }),
+    clearRecentlyCreatedFieldIds: () =>
+      set({ recentlyCreatedFieldIds: new Set() }),
+  }),
+)
+
+export const recentlyCreatedFieldIdsSelector = (
+  state: MagicFormBuilderStore,
+): MagicFormBuilderStore['recentlyCreatedFieldIds'] =>
+  state.recentlyCreatedFieldIds
diff --git a/src/app/modules/form/admin-form/admin-form.assistance.controller.ts b/src/app/modules/form/admin-form/admin-form.assistance.controller.ts
index 15f8788cf0..c167eea984 100644
--- a/src/app/modules/form/admin-form/admin-form.assistance.controller.ts
+++ b/src/app/modules/form/admin-form/admin-form.assistance.controller.ts
@@ -32,7 +32,7 @@ interface ITextPrompt {
 
 const _handleTextPrompt: ControllerHandler<
   { formId: string },
-  { message: string },
+  { message: string; createdFieldIds?: string[] },
   ITextPrompt
 > = async (req, res) => {
   const { formId } = req.params
@@ -57,9 +57,10 @@ const _handleTextPrompt: ControllerHandler<
           userPrompt: req.body.prompt,
         }),
       )
-      .map(() =>
+      .map((createdFieldIds) =>
         res.status(StatusCodes.OK).json({
           message: 'Created form fields using text prompt successfully.',
+          createdFieldIds: createdFieldIds.map((field) => field._id.toString()),
         }),
       )
       .mapErr((error) => {
diff --git a/src/app/modules/form/admin-form/admin-form.assistance.service.ts b/src/app/modules/form/admin-form/admin-form.assistance.service.ts
index 2404087484..366155a310 100644
--- a/src/app/modules/form/admin-form/admin-form.assistance.service.ts
+++ b/src/app/modules/form/admin-form/admin-form.assistance.service.ts
@@ -12,7 +12,7 @@ import {
   RadioFieldBase,
   TableFieldBase,
 } from '../../../../../shared/types'
-import { IPopulatedForm } from '../../../../types'
+import { FormFieldSchema, IPopulatedForm } from '../../../../types'
 import { createLoggerWithLabel } from '../../../config/logger'
 import { PossibleDatabaseError } from '../../core/core.errors'
 import { FormNotFoundError } from '../form.errors'
@@ -274,7 +274,7 @@ export const createFormFieldsUsingTextPrompt = ({
   form: IPopulatedForm
   userPrompt: string
 }): ResultAsync<
-  undefined,
+  FormFieldSchema['_id'][],
   | ModelResponseFailureError
   | ModelResponseInvalidSchemaFormatError
   | ModelResponseInvalidSyntaxError
@@ -333,6 +333,8 @@ export const createFormFieldsUsingTextPrompt = ({
       )
       return createFormFields({ form, newFields: formFieldsToCreate, to: 0 })
     })
-    .andThen(() => updateFormMetadata(form, { ...form.metadata, mfb: true }))
-    .map(() => undefined)
+    .map((updatedFields) => {
+      updateFormMetadata(form, { ...form.metadata, mfb: true })
+      return updatedFields.map((field) => field._id)
+    })
 }