overleaf/services/web/frontend/js/features/source-editor/components/math-preview-tooltip.tsx

import { useProjectSettingsContext } from '@/features/editor-left-menu/context/project-settings-context'
import {
  Dropdown,
  DropdownMenu,
  DropdownToggle,
} from '@/features/ui/components/bootstrap-5/dropdown-menu'
import OLButton from '@/features/ui/components/ol/ol-button'
import OLModal, {
  OLModalBody,
  OLModalFooter,
  OLModalHeader,
  OLModalTitle,
} from '@/features/ui/components/ol/ol-modal'
import MaterialIcon from '@/shared/components/material-icon'
import useEventListener from '@/shared/hooks/use-event-listener'
import { FC, useCallback, useEffect, useRef, useState } from 'react'
import { Trans, useTranslation } from 'react-i18next'
import {
  useCodeMirrorStateContext,
  useCodeMirrorViewContext,
} from './codemirror-context'
import { mathPreviewStateField } from '../extensions/math-preview'
import { getTooltip } from '@codemirror/view'
import ReactDOM from 'react-dom'
import OLDropdownMenuItem from '@/features/ui/components/ol/ol-dropdown-menu-item'
import BootstrapVersionSwitcher from '@/features/ui/components/bootstrap-5/bootstrap-version-switcher'
import ControlledDropdown from '@/shared/components/controlled-dropdown'
import {
  Dropdown as BS3Dropdown,
  MenuItem as BS3MenuItem,
} from 'react-bootstrap'
import { useSplitTestContext } from '@/shared/context/split-test-context'

const MathPreviewTooltipContainer: FC = () => {
  const state = useCodeMirrorStateContext()
  const view = useCodeMirrorViewContext()

  const mathPreviewState = state.field(mathPreviewStateField, false)

  if (!mathPreviewState) {
    return null
  }

  const { tooltip, mathContent } = mathPreviewState

  if (!tooltip || !mathContent) {
    return null
  }

  const tooltipView = getTooltip(view, tooltip)

  if (!tooltipView) {
    return null
  }

  return ReactDOM.createPortal(
    <MathPreviewTooltip mathContent={mathContent} />,
    tooltipView.dom
  )
}

const MathPreviewTooltip: FC<{ mathContent: HTMLDivElement }> = ({
  mathContent,
}) => {
  const { t } = useTranslation()

  const [showDisableModal, setShowDisableModal] = useState(false)
  const { setMathPreview } = useProjectSettingsContext()
  const openDisableModal = useCallback(() => setShowDisableModal(true), [])
  const closeDisableModal = useCallback(() => setShowDisableModal(false), [])

  const onHide = useCallback(() => {
    window.dispatchEvent(new Event('editor:hideMathTooltip'))
  }, [])

  const mathRef = useRef<HTMLSpanElement>(null)

  const keyDownListener = useCallback(
    (event: KeyboardEvent) => {
      if (event.key === 'Escape') {
        onHide()
      }
    },
    [onHide]
  )

  useEventListener('keydown', keyDownListener)

  useEffect(() => {
    if (mathRef.current) {
      mathRef.current.replaceChildren(mathContent)
    }
  }, [mathContent])

  return (
    <>
      <div className="ol-cm-math-tooltip">
        <span ref={mathRef} />
        <CustomSplitTestBadge />

        <BootstrapVersionSwitcher
          bs5={
            <Dropdown align="end">
              <DropdownToggle
                id="some-id"
                className="math-tooltip-options-toggle"
                variant="secondary"
                size="sm"
              >
                <MaterialIcon
                  type="more_vert"
                  accessibilityLabel={t('more_options')}
                />
              </DropdownToggle>
              <DropdownMenu flip={false}>
                <li>
                  <OLDropdownMenuItem
                    onClick={onHide}
                    description={t('temporarily_hides_the_preview')}
                    trailingIcon={
                      <span className="math-tooltip-options-keyboard-shortcut">
                        Esc
                      </span>
                    }
                  >
                    {t('hide')}
                  </OLDropdownMenuItem>
                </li>
                <li>
                  <OLDropdownMenuItem
                    onClick={openDisableModal}
                    description={t('permanently_disables_the_preview')}
                  >
                    {t('disable')}
                  </OLDropdownMenuItem>
                </li>
              </DropdownMenu>
            </Dropdown>
          }
          bs3={
            <ControlledDropdown id="math-preview-tooltip-options" pullRight>
              <BS3Dropdown.Toggle
                noCaret
                bsSize="small"
                bsStyle={null}
                className="math-tooltip-options-toggle"
              >
                <MaterialIcon
                  type="more_vert"
                  accessibilityLabel={t('more_options')}
                />
              </BS3Dropdown.Toggle>
              <BS3Dropdown.Menu className="math-preview-tooltip-menu">
                <BS3MenuItem
                  className="math-preview-tooltip-option"
                  onClick={onHide}
                >
                  <div className="math-preview-tooltip-option-content">
                    <div className="math-preview-tooltip-option-label">
                      {t('hide')}
                    </div>
                    <div className="math-preview-tooltip-option-description">
                      {t('temporarily_hides_the_preview')}
                    </div>
                  </div>
                  <div className="math-preview-tooltip-option-shortcut">
                    Esc
                  </div>
                </BS3MenuItem>
                <BS3MenuItem
                  className="math-preview-tooltip-option"
                  onClick={openDisableModal}
                >
                  <div className="math-preview-tooltip-option-content">
                    <div className="math-preview-tooltip-option-label">
                      {t('disable')}
                    </div>
                    <div className="math-preview-tooltip-option-description">
                      {t('permanently_disables_the_preview')}
                    </div>
                  </div>
                </BS3MenuItem>
              </BS3Dropdown.Menu>
            </ControlledDropdown>
          }
        />
      </div>

      {showDisableModal && (
        <OLModal show onHide={closeDisableModal}>
          <OLModalHeader>
            <OLModalTitle>{t('disable_equation_preview')}</OLModalTitle>
          </OLModalHeader>

          <OLModalBody>
            {t('disable_equation_preview_confirm')}
            <br />
            <Trans
              i18nKey="disable_equation_preview_enable"
              components={{ b: <strong /> }}
            />
          </OLModalBody>

          <OLModalFooter>
            <OLButton variant="secondary" onClick={closeDisableModal}>
              {t('cancel')}
            </OLButton>
            <OLButton variant="danger" onClick={() => setMathPreview(false)}>
              {t('disable')}
            </OLButton>
          </OLModalFooter>
        </OLModal>
      )}
    </>
  )
}

// Temporary SplitTest badge for just this component that always shows
// as in beta and doesn't have a tooltip
const CustomSplitTestBadge: FC = () => {
  const { splitTestVariants } = useSplitTestContext()

  const variant = splitTestVariants['math-preview']
  if (!variant || variant !== 'enabled') {
    return null
  }

  return (
    <a
      href="https://forms.gle/uDVGSqnBBJQ8fUPt5"
      target="_blank"
      rel="noopener noreferrer"
    >
      <BootstrapVersionSwitcher
        bs3={<span className="badge beta-badge" />}
        bs5={<MaterialIcon type="info" className="align-middle beta-badge" />}
      />
    </a>
  )
}

export default MathPreviewTooltipContainer