modeling-app/src/components/OpenInDesktopAppHandler.tsx

import { Transition } from '@headlessui/react'
import { VITE_KC_SITE_BASE_URL } from '@src/env'
import { useSearchParams } from 'react-router-dom'

import { ActionButton } from '@src/components/ActionButton'
import {
  ASK_TO_OPEN_QUERY_PARAM,
  ZOO_STUDIO_PROTOCOL,
} from '@src/lib/constants'
import { isDesktop } from '@src/lib/isDesktop'
import { Themes, getSystemTheme } from '@src/lib/theme'

/**
 * This component is a handler that checks if a certain query parameter
 * is present, and if so, it will show a modal asking the user if they
 * want to open the current page in the desktop app.
 */
export const OpenInDesktopAppHandler = (props: React.PropsWithChildren) => {
  const theme = getSystemTheme()
  const buttonClasses =
    'bg-transparent flex-0 hover:bg-primary/10 dark:hover:bg-primary/10'
  const pathLogomarkSvg = `${isDesktop() ? '.' : ''}/zma-logomark${
    theme === Themes.Light ? '-dark' : ''
  }.svg`
  const [searchParams, setSearchParams] = useSearchParams()
  // We also ignore this param on desktop, as it is redundant
  const hasAskToOpenParam =
    !isDesktop() && searchParams.has(ASK_TO_OPEN_QUERY_PARAM)

  /**
   * This function removes the query param to ask to open in desktop app
   * and then navigates to the same route but with our custom protocol
   * `zoo-studio:` instead of `https://${BASE_URL}`, to trigger the user's
   * desktop app to open.
   */
  function onOpenInDesktopApp() {
    const newSearchParams = new URLSearchParams(globalThis.location.search)
    newSearchParams.delete(ASK_TO_OPEN_QUERY_PARAM)
    const newURL = `${ZOO_STUDIO_PROTOCOL}://${globalThis.location.pathname.replace(
      '/',
      ''
    )}${searchParams.size > 0 ? `?${newSearchParams.toString()}` : ''}`
    globalThis.location.href = newURL
  }

  /**
   * Just remove the query param to ask to open in desktop app
   * and continue to the web app.
   */
  function continueToWebApp() {
    searchParams.delete(ASK_TO_OPEN_QUERY_PARAM)
    setSearchParams(searchParams)
  }

  return hasAskToOpenParam ? (
    <Transition
      appear
      show={true}
      as="div"
      className={
        theme +
        ` fixed inset-0 grid p-4 place-content-center ${
          theme === Themes.Dark ? '!bg-chalkboard-110 text-chalkboard-20' : ''
        }`
      }
    >
      <Transition.Child
        as="div"
        className={`max-w-3xl py-6 px-10 flex flex-col items-center gap-8 
          mx-auto border rounded-lg shadow-lg dark:bg-chalkboard-100`}
        enter="ease-out duration-300"
        enterFrom="opacity-0 scale-95"
        enterTo="opacity-100 scale-100"
        leave="ease-in duration-200"
        leaveFrom="opacity-100 scale-100"
        leaveTo="opacity-0 scale-95"
        style={{ zIndex: 10 }}
      >
        <div>
          <h1 className="text-2xl">
            Launching{' '}
            <img
              src={pathLogomarkSvg}
              className="w-48"
              alt="Zoo Design Studio"
            />
          </h1>
        </div>
        <p className="text-primary flex items-center gap-2">
          Choose where to open this link...
        </p>
        <div className="flex flex-col md:flex-row items-start justify-between gap-4 xl:gap-8">
          <div className="flex flex-col gap-2">
            <ActionButton
              Element="button"
              className={buttonClasses + ' !text-base'}
              onClick={onOpenInDesktopApp}
              iconEnd={{ icon: 'arrowRight' }}
            >
              Open in desktop app
            </ActionButton>
            <ActionButton
              Element="externalLink"
              className={
                buttonClasses +
                ' text-sm border-transparent justify-center dark:bg-transparent'
              }
              to={`${VITE_KC_SITE_BASE_URL}/modeling-app/download`}
              iconEnd={{ icon: 'link', bgClassName: '!bg-transparent' }}
            >
              Download desktop app
            </ActionButton>
          </div>
          <ActionButton
            Element="button"
            className={buttonClasses + ' -order-1 !text-base'}
            onClick={continueToWebApp}
            iconStart={{ icon: 'arrowLeft' }}
            data-testid="continue-to-web-app-button"
          >
            Continue to web app
          </ActionButton>
        </div>
      </Transition.Child>
    </Transition>
  ) : (
    props.children
  )
}