fix(wayland): fix ext screens resetting and causing protocol error

Could not reproduce this personally but I am told this fixes it.

src/wayland/session_lock/session_lock.cpp

@@ -68,7 +68,10 @@ bool LockWindowExtension::attach(QWindow* window, SessionLockManager* manager) {
 	if (current != nullptr) {
 		current->surface->setExtension(this);
 	} else {
+		// Qt appears to be resetting the window's screen on creation on some systems. This works around it.
+		auto* screen = window->screen();
 		window->create();
+		window->setScreen(screen);
 
 		waylandWindow = dynamic_cast<QtWaylandClient::QWaylandWindow*>(window->handle());
 		if (waylandWindow == nullptr) {

src/wayland/wlr_layershell/window.cpp

@@ -54,7 +54,10 @@ bool LayershellWindowExtension::attach(QWindow* window) {
 	}
 
 	if (!hasSurface) {
+		// Qt appears to be resetting the window's screen on creation on some systems. This works around it.
+		auto* screen = window->screen();
 		window->create();
+		window->setScreen(screen);
 
 		auto* waylandWindow = dynamic_cast<QtWaylandClient::QWaylandWindow*>(window->handle());
 		if (waylandWindow == nullptr) {