bot/apps/builder/components/shared/modals/UpgradeModal/UpgradeModal.tsx

import { useEffect, useState } from 'react'
import {
  Heading,
  Modal,
  ModalBody,
  Text,
  ModalContent,
  ModalFooter,
  ModalOverlay,
  Stack,
  ListItem,
  UnorderedList,
  ListIcon,
  chakra,
  Tooltip,
  ListProps,
  Button,
  HStack,
} from '@chakra-ui/react'
import { pay } from 'services/stripe'
import { useUser } from 'contexts/UserContext'
import { Plan } from 'db'
import { useWorkspace } from 'contexts/WorkspaceContext'
import { TypebotLogo } from 'assets/logos'
import { CheckIcon } from 'assets/icons'
import { toTitleCase } from 'utils'
import { useToast } from 'components/shared/hooks/useToast'
import { Info } from 'components/shared/Info'

export enum LimitReached {
  BRAND = 'remove branding',
  CUSTOM_DOMAIN = 'add custom domain',
  FOLDER = 'create folders',
  FILE_INPUT = 'use file input blocks',
}

type UpgradeModalProps = {
  type?: LimitReached
  isOpen: boolean
  onClose: () => void
  plan?: Plan
}

export const UpgradeModal = ({
  onClose,
  isOpen,
  type,
  plan = Plan.PRO,
}: UpgradeModalProps) => {
  const { user } = useUser()
  const { workspace, refreshWorkspace } = useWorkspace()
  const [payLoading, setPayLoading] = useState(false)
  const [currency, setCurrency] = useState<'usd' | 'eur'>('usd')
  const { showToast } = useToast()

  useEffect(() => {
    setCurrency(
      navigator.languages.find((l) => l.includes('fr')) ? 'eur' : 'usd'
    )
  }, [])

  const handlePayClick = async () => {
    if (!user || !workspace) return
    setPayLoading(true)
    const response = await pay({
      customerId: workspace.stripeId ?? undefined,
      user,
      currency,
      plan: plan === Plan.TEAM ? 'team' : 'pro',
      workspaceId: workspace.id,
    })
    setPayLoading(false)
    if (response?.newPlan) {
      refreshWorkspace({ plan: response.newPlan })
      showToast({
        status: 'success',
        title: 'Upgrade success!',
        description: `Workspace successfully upgraded to ${toTitleCase(
          response.newPlan
        )} plan 🎉`,
      })
    }
  }

  return (
    <Modal isOpen={isOpen} onClose={onClose}>
      <ModalOverlay />
      <ModalContent>
        <ModalBody as={Stack} pt="10">
          {plan === Plan.PRO ? (
            <PersonalProPlanContent currency={currency} type={type} />
          ) : (
            <TeamPlanContent currency={currency} type={type} />
          )}
        </ModalBody>

        <ModalFooter>
          <HStack>
            <Button colorScheme="gray" onClick={onClose}>
              Cancel
            </Button>
            <Button
              onClick={handlePayClick}
              isLoading={payLoading}
              colorScheme="blue"
            >
              Upgrade
            </Button>
          </HStack>
        </ModalFooter>
      </ModalContent>
    </Modal>
  )
}

const PersonalProPlanContent = ({
  currency,
  type,
}: {
  currency: 'eur' | 'usd'
  type?: LimitReached
}) => {
  return (
    <Stack spacing="4">
      <Info>You need to upgrade your plan in order to {type}</Info>
      <TypebotLogo boxSize="30px" />
      <Heading fontSize="2xl">
        Upgrade to <chakra.span color="orange.400">Personal Pro</chakra.span>{' '}
        plan
      </Heading>
      <Text>For solo creators who want to do even more.</Text>
      <Heading>
        {currency === 'eur' ? '39€' : '$39'}
        <chakra.span fontSize="md">/ month</chakra.span>
      </Heading>
      <Text fontWeight="bold">Everything in Personal, plus:</Text>
      <FeatureList
        features={[
          'Branding removed',
          'View incomplete submissions',
          'In-depth drop off analytics',
          'Unlimited custom domains',
          'Organize typebots in folders',
          'Unlimited uploads',
        ]}
      />
    </Stack>
  )
}

const TeamPlanContent = ({
  currency,
  type,
}: {
  currency: 'eur' | 'usd'
  type?: LimitReached
}) => {
  return (
    <Stack spacing="4">
      <Info>You need to upgrade your plan in order to {type}</Info>
      <TypebotLogo boxSize="30px" />
      <Heading fontSize="2xl">
        Upgrade to <chakra.span color="purple.400">Team</chakra.span> plan
      </Heading>
      <Text>For teams to build typebots together in one spot.</Text>
      <Heading>
        {currency === 'eur' ? '99€' : '$99'}
        <chakra.span fontSize="md">/ month</chakra.span>
      </Heading>
      <Text fontWeight="bold">
        <Tooltip
          label={
            <FeatureList
              features={[
                'Branding removed',
                'View incomplete submissions',
                'In-depth drop off analytics',
                'Custom domains',
                'Organize typebots in folders',
                'Unlimited uploads',
              ]}
              spacing="0"
            />
          }
          hasArrow
          placement="top"
        >
          <chakra.span textDecoration="underline" cursor="pointer">
            Everything in Pro
          </chakra.span>
        </Tooltip>
        , plus:
      </Text>
      <FeatureList
        features={[
          'Unlimited team members',
          'Collaborative workspace',
          'Custom roles',
        ]}
      />
    </Stack>
  )
}

const FeatureList = ({
  features,
  ...props
}: { features: string[] } & ListProps) => (
  <UnorderedList listStyleType="none" spacing={2} {...props}>
    {features.map((feat) => (
      <ListItem key={feat}>
        <ListIcon as={CheckIcon} />
        {feat}
      </ListItem>
    ))}
  </UnorderedList>
)