uppy/packages/@uppy/react/src/useScreenCapture.ts

import {
  createScreenCaptureController,
  type ScreenCaptureSnapshot,
} from '@uppy/components'
import { useMemo } from 'react'
import { useSyncExternalStore } from 'use-sync-external-store/shim/index.js'
import { useUppyContext } from './headless/UppyContextProvider.js'

type ScreenCaptureProps = {
  onSubmit: () => void
}

export function useScreenCapture({
  onSubmit,
}: ScreenCaptureProps): ScreenCaptureSnapshot {
  const { uppy } = useUppyContext()
  const controller = useMemo(
    () => createScreenCaptureController(uppy, onSubmit),
    [uppy, onSubmit],
  )
  const store = useSyncExternalStore(
    controller.subscribe,
    controller.getSnapshot,
    controller.getSnapshot,
  )
  return store
}