bot/apps/builder/src/features/theme/components/SaveThemeModal.tsx

import { TextInput } from '@/components/inputs'
import { useToast } from '@/hooks/useToast'
import { trpc } from '@/lib/trpc'
import {
  Button,
  HStack,
  Modal,
  ModalBody,
  ModalCloseButton,
  ModalContent,
  ModalFooter,
  ModalHeader,
  ModalOverlay,
} from '@chakra-ui/react'
import { createId } from '@paralleldrive/cuid2'
import { ThemeTemplate } from '@typebot.io/schemas'
import { FormEvent, useRef, useState } from 'react'

type Props = {
  workspaceId: string
  isOpen: boolean
  onClose: (template?: Pick<ThemeTemplate, 'id' | 'theme'>) => void
  selectedTemplate: Pick<ThemeTemplate, 'id' | 'name'> | undefined
  theme: ThemeTemplate['theme']
}

export const SaveThemeModal = ({
  workspaceId,
  isOpen,
  onClose,
  selectedTemplate,
  theme,
}: Props) => {
  const { showToast } = useToast()
  const [isSaving, setIsSaving] = useState(false)
  const inputRef = useRef<HTMLInputElement>(null)
  const {
    theme: {
      listThemeTemplates: { refetch: refetchThemeTemplates },
    },
  } = trpc.useContext()
  const { mutate } = trpc.theme.saveThemeTemplate.useMutation({
    onMutate: () => setIsSaving(true),
    onSettled: () => setIsSaving(false),
    onSuccess: ({ themeTemplate }) => {
      refetchThemeTemplates()
      onClose(themeTemplate)
    },
    onError: (error) => {
      showToast({
        description: error.message,
      })
    },
  })

  const updateExistingTemplate = (e: FormEvent) => {
    e.preventDefault()
    const newName = inputRef.current?.value
    if (!newName) return
    mutate({
      name: newName,
      theme,
      workspaceId,
      themeTemplateId: selectedTemplate?.id ?? createId(),
    })
  }

  const saveNewTemplate = () => {
    const newName = inputRef.current?.value
    if (!newName) return
    mutate({
      name: newName,
      theme,
      workspaceId,
      themeTemplateId: createId(),
    })
  }

  return (
    <Modal isOpen={isOpen} onClose={onClose} initialFocusRef={inputRef}>
      <ModalOverlay />
      <ModalContent as="form" onSubmit={updateExistingTemplate}>
        <ModalHeader>Save theme</ModalHeader>
        <ModalCloseButton />
        <ModalBody>
          <TextInput
            ref={inputRef}
            label="Name:"
            defaultValue={selectedTemplate?.name}
            withVariableButton={false}
            placeholder="My template"
            isRequired
          />
        </ModalBody>

        <ModalFooter as={HStack}>
          {selectedTemplate?.id && (
            <Button isLoading={isSaving} onClick={saveNewTemplate}>
              Save as new template
            </Button>
          )}
          <Button type="submit" colorScheme="blue" isLoading={isSaving}>
            {selectedTemplate?.id ? 'Update' : 'Save'}
          </Button>
        </ModalFooter>
      </ModalContent>
    </Modal>
  )
}