[fix] focus events (#2013)

Somehow the previous focus PR snuck through with broken tests.

### Change Type

- [x] `patch` — Bug fix

packages/editor/src/lib/hooks/useFocusEvents.ts

@@ -1,12 +1,12 @@
 import { useLayoutEffect } from 'react'
+import { useContainer } from './useContainer'
 import { useEditor } from './useEditor'
 
 /** @internal */
 export function useFocusEvents(autoFocus: boolean) {
 	const editor = useEditor()
+	const container = useContainer()
 	useLayoutEffect(() => {
-		const container = editor.getContainer()
-
 		function handleFocus() {
 			if (!editor.instanceState.isFocused) {
 				editor.updateInstanceState({ isFocused: true })
@@ -14,16 +14,18 @@ export function useFocusEvents(autoFocus: boolean) {
 		}
 
 		container.addEventListener('focus', handleFocus)
+		container.addEventListener('pointerdown', handleFocus)
 
 		if (autoFocus && !editor.instanceState.isFocused) {
 			editor.updateInstanceState({ isFocused: true })
-			editor.getContainer().focus()
+			container.focus()
 		} else if (editor.instanceState.isFocused) {
 			editor.updateInstanceState({ isFocused: false })
 		}
 
 		return () => {
 			container.removeEventListener('focus', handleFocus)
+			container.removeEventListener('pointerdown', handleFocus)
 		}
-	}, [editor, autoFocus])
+	}, [editor, container, autoFocus])
 }