For those in need. Here is a class that you can use to re-capture the cursor even in the editor automatically (though use with caution):
https://gist.github.com/kamgam/856cdfc03ae14b11d18bf64b3f3fc9e6

#if UNITY_EDITOR
using UnityEditor;
using System.Reflection;

/// <summary>
/// Allows the game code for a short time to override the cursor lock behaviour even in the editor.
/// Usage:
/// // Call it like this after some input that should trigger capturing the cursor again.
/// #if UNITY_EDITOR
/// ForceGameViewCursorLock.AllowCursorLockOverrideForNSec(0.1f);
/// #endif
/// </summary>
[InitializeOnLoad]
public static class ForceGameViewCursorLock
{
private static double s_overrideUntil;

private static EditorWindow s_gameView;
public static MethodInfo s_allowCursorLockMethod;

static ForceGameViewCursorLock()
{
System.Type gameViewType = typeof(Editor).Assembly.GetType("UnityEditor.GameView");

EditorApplication.update += onEditorUpdate;
}

public static void AllowCursorLockOverrideForNSec(double seconds)
{
s_overrideUntil = EditorApplication.timeSinceStartup + seconds;
}

private static void onEditorUpdate()
{
bool isAllowed = EditorApplication.timeSinceStartup < s_overrideUntil;

if (!isAllowed || !EditorApplication.isPlaying || EditorApplication.isPaused)
return;

// Cache reflections
if (s_gameView == null &&
EditorWindow.focusedWindow != null &&
EditorWindow.focusedWindow.GetType().Name == "GameView")
{
s_gameView = EditorWindow.focusedWindow;
s_allowCursorLockMethod = EditorWindow.focusedWindow.GetType().GetMethod("AllowCursorLockAndHide", BindingFlags.Instance | BindingFlags.NonPublic | BindingFlags.Public);
}

// Enable
if (EditorWindow.focusedWindow == s_gameView && s_gameView != null && s_allowCursorLockMethod != null)
{
UnityEngine.Debug.Log("override");
s_allowCursorLockMethod?.Invoke(EditorWindow.focusedWindow, new object[] { true });
}
}
}
#endif