useFocusReturn

See source on Github 

Records currently focused element and returns focus to it when when is set to false.

• @param when - When true, focus is recorded. When false, focus is returned.
• @param options - Configurable options
• @returns void

Example

const [isDialogOpen, setIsDialogOpen] = useState(false)
 
useFocusReturn(isDialogOpen)
 
return (
  <Dialog open={isDialogOpen} onClose={setIsDialogOpen} />
)