server/src/wayland/xdg/toplevel.rs

use super::backend::XdgBackend;
use crate::{
	nodes::{Node, items::panel::PanelItem},
	wayland::core::surface::Surface,
};
use mint::Vector2;
use parking_lot::Mutex;
use std::sync::Arc;
use std::sync::Weak;
pub use waynest::server::protocol::stable::xdg_shell::xdg_toplevel::*;
use waynest::{
	server::{Client, Dispatcher, Result},
	wire::ObjectId,
};

#[derive(Debug)]
pub struct Mapped {
	pub panel_item_node: Arc<Node>,
	pub _panel_item: Arc<PanelItem<XdgBackend>>,
}
impl Mapped {
	pub fn create(toplevel: Arc<Toplevel>, pid: Option<i32>) -> Self {
		let (panel_item_node, _panel_item) =
			PanelItem::create(Box::new(XdgBackend::new(toplevel)), pid);

		Self {
			panel_item_node,
			_panel_item,
		}
	}
}

#[derive(Debug, Clone)]
struct ToplevelData {
	parent: Option<u64>,
	app_id: Option<String>,
	title: Option<String>,
	activated: bool,
	fullscreen: bool,
	pub size: Option<Vector2<u32>>,
}
impl Default for ToplevelData {
	fn default() -> Self {
		Self {
			parent: None,
			app_id: None,
			title: None,
			activated: true,
			fullscreen: false,
			size: None,
		}
	}
}

#[derive(Debug, Dispatcher)]
pub struct Toplevel {
	pub id: ObjectId,
	wl_surface: Weak<Surface>,
	xdg_surface: Weak<super::surface::Surface>,
	pub mapped: Mutex<Option<Mapped>>,
	data: Mutex<ToplevelData>,
}
impl Toplevel {
	pub fn new(
		object_id: ObjectId,
		wl_surface: Arc<Surface>,
		xdg_surface: Arc<super::surface::Surface>,
	) -> Self {
		Toplevel {
			id: object_id,
			wl_surface: Arc::downgrade(&wl_surface),
			xdg_surface: Arc::downgrade(&xdg_surface),
			mapped: Mutex::new(None),
			data: Mutex::new(ToplevelData::default()),
		}
	}

	pub fn surface(&self) -> Arc<Surface> {
		// We can safely unwrap as the surface must exist for the lifetime of the toplevel
		self.wl_surface
			.upgrade()
			.expect("Surface was dropped before toplevel")
	}

	pub fn title(&self) -> Option<String> {
		self.data.lock().title.clone()
	}
	pub fn app_id(&self) -> Option<String> {
		self.data.lock().app_id.clone()
	}
	pub fn parent(&self) -> Option<u64> {
		self.data.lock().parent
	}

	pub fn set_size(&self, size: Option<Vector2<u32>>) {
		self.data.lock().size = size;
	}

	pub fn set_activated(&self, activated: bool) {
		self.data.lock().activated = activated;
	}

	// Helper to clamp size against constraints
	fn clamp_size(&self, size: Vector2<u32>) -> Vector2<u32> {
		let state = self.surface().current_state();
		let mut clamped = size;

		if let Some(min_size) = state.min_size {
			clamped.x = clamped.x.max(min_size.x);
			clamped.y = clamped.y.max(min_size.y);
		}
		if let Some(max_size) = state.max_size {
			clamped.x = clamped.x.min(max_size.x);
			clamped.y = clamped.y.min(max_size.y);
		}
		clamped
	}

	pub async fn reconfigure(&self, client: &mut Client) -> Result<()> {
		let data = self.data.lock().clone();

		// Use the explicitly set size, applying constraints
		let size = data.size.map(|s| self.clamp_size(s));

		let mut states = vec![
			State::TiledTop,
			State::TiledLeft,
			State::TiledRight,
			State::TiledBottom,
			if data.fullscreen {
				State::Fullscreen
			} else {
				State::Maximized
			},
		];
		if data.activated {
			states.push(State::Activated);
		}

		self.configure(
			client,
			self.id,
			size.map(|v| v.x as i32).unwrap_or(0),
			size.map(|v| v.y as i32).unwrap_or(0),
			states
				.into_iter()
				.flat_map(|x| (x as u32).to_ne_bytes())
				.collect(),
		)
		.await?;
		self.xdg_surface
			.upgrade()
			.unwrap()
			.reconfigure(client)
			.await
	}
}
impl XdgToplevel for Toplevel {
	async fn set_parent(
		&self,
		client: &mut Client,
		_sender_id: ObjectId,
		parent: Option<ObjectId>,
	) -> Result<()> {
		// Handle case where parent is specified
		if let Some(parent) = parent {
			// Per spec: parent must be another xdg_toplevel surface
			if let Some(parent_toplevel) = client.get::<Toplevel>(parent) {
				let Some(mapped) = &*parent_toplevel.mapped.lock() else {
					// Per spec: parent surfaces must be mapped before being used as a parent
					// Setting an unmapped window as parent should raise a protocol error
					// For now we just unset the parent as a fallback
					self.data.lock().parent.take();
					return Ok(());
				};

				// Per spec: store parent to ensure this surface is stacked above parent
				// and other ancestor surfaces. Used for proper window stacking order.
				self.data
					.lock()
					.parent
					.replace(mapped.panel_item_node.get_id());
			}
		} else {
			// Per spec: null parent unsets the parent, making this a top-level window
			// This allows converting child windows back to independent top-level windows
			self.data.lock().parent.take();
		}

		Ok(())
	}

	async fn set_title(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		title: String,
	) -> Result<()> {
		self.data.lock().title.replace(title);
		Ok(())
	}

	async fn set_app_id(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		app_id: String,
	) -> Result<()> {
		self.data.lock().app_id.replace(app_id);
		Ok(())
	}

	async fn show_window_menu(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		_seat: ObjectId,
		_serial: u32,
		_x: i32,
		_y: i32,
	) -> Result<()> {
		Ok(())
	}

	async fn r#move(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		_seat: ObjectId,
		_serial: u32,
	) -> Result<()> {
		Ok(())
	}

	async fn resize(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		_seat: ObjectId,
		_serial: u32,
		_edges: ResizeEdge,
	) -> Result<()> {
		Ok(())
	}

	async fn set_max_size(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		width: i32,
		height: i32,
	) -> Result<()> {
		self.surface().pending_state().pending.max_size = if width == 0 && height == 0 {
			None
		} else {
			Some([width as u32, height as u32].into())
		};
		Ok(())
	}

	async fn set_min_size(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		width: i32,
		height: i32,
	) -> Result<()> {
		self.surface().pending_state().pending.min_size = if width == 0 && height == 0 {
			None
		} else {
			Some([width as u32, height as u32].into())
		};
		Ok(())
	}

	async fn set_maximized(&self, _client: &mut Client, _sender_id: ObjectId) -> Result<()> {
		Ok(())
	}

	async fn unset_maximized(&self, _client: &mut Client, _sender_id: ObjectId) -> Result<()> {
		Ok(())
	}

	async fn set_fullscreen(
		&self,
		_client: &mut Client,
		_sender_id: ObjectId,
		_output: Option<ObjectId>,
	) -> Result<()> {
		Ok(())
	}

	async fn unset_fullscreen(&self, _client: &mut Client, _sender_id: ObjectId) -> Result<()> {
		Ok(())
	}

	async fn set_minimized(&self, _client: &mut Client, _sender_id: ObjectId) -> Result<()> {
		Ok(())
	}

	async fn destroy(&self, _client: &mut Client, _sender_id: ObjectId) -> Result<()> {
		self.mapped.lock().take();
		Ok(())
	}
}
impl Drop for Toplevel {
	fn drop(&mut self) {
		self.mapped.lock().take();
	}
}