seat: implement input methods

src/ifs/wl_seat/text_input.rs

@@ -0,0 +1,96 @@
+use {
+    crate::ifs::{
+        wl_seat::{
+            text_input::{
+                zwp_input_method_v2::ZwpInputMethodV2, zwp_text_input_v3::ZwpTextInputV3,
+            },
+            WlSeatGlobal,
+        },
+        wl_surface::WlSurface,
+    },
+    std::rc::Rc,
+};
+
+pub mod zwp_input_method_keyboard_grab_v2;
+pub mod zwp_input_method_manager_v2;
+pub mod zwp_input_method_v2;
+pub mod zwp_text_input_manager_v3;
+pub mod zwp_text_input_v3;
+
+const MAX_TEXT_SIZE: usize = 4000;
+
+pub struct TextInputConnection {
+    pub seat: Rc<WlSeatGlobal>,
+    pub text_input: Rc<ZwpTextInputV3>,
+    pub input_method: Rc<ZwpInputMethodV2>,
+    pub surface: Rc<WlSurface>,
+}
+
+#[derive(Copy, Clone, Debug, Eq, PartialEq)]
+pub enum TextConnectReason {
+    TextInputEnabled,
+    InputMethodCreated,
+}
+
+#[derive(Copy, Clone, Debug, Eq, PartialEq)]
+pub enum TextDisconnectReason {
+    FocusLost,
+    TextInputDisabled,
+    InputMethodDestroyed,
+}
+
+impl WlSeatGlobal {
+    fn create_text_input_connection(self: &Rc<Self>, text_connect_reason: TextConnectReason) {
+        let Some(im) = self.input_method.get() else {
+            return;
+        };
+        let Some(ti) = self.text_input.get() else {
+            return;
+        };
+        let Some(surface) = self.keyboard_node.get().node_into_surface() else {
+            log::warn!("Seat has text input but keyboard node is not a surface");
+            return;
+        };
+        if surface.client.id != ti.client.id {
+            log::warn!("Seat's text input belongs to different client than the keyboard node");
+            return;
+        }
+        let con = Rc::new(TextInputConnection {
+            seat: self.clone(),
+            text_input: ti.clone(),
+            input_method: im.clone(),
+            surface: surface.clone(),
+        });
+        con.connect(text_connect_reason);
+    }
+}
+
+impl TextInputConnection {
+    fn connect(self: &Rc<Self>, reason: TextConnectReason) {
+        self.input_method.connection.set(Some(self.clone()));
+        self.text_input.connection.set(Some(self.clone()));
+        self.surface
+            .text_input_connections
+            .insert(self.seat.id, self.clone());
+
+        self.input_method.activate();
+        if reason == TextConnectReason::InputMethodCreated {
+            self.text_input.send_all_to(&self.input_method);
+            self.input_method.send_done();
+        }
+    }
+
+    pub fn disconnect(&self, reason: TextDisconnectReason) {
+        self.text_input.connection.take();
+        self.input_method.connection.take();
+        self.surface.text_input_connections.remove(&self.seat.id);
+
+        if reason != TextDisconnectReason::InputMethodDestroyed {
+            self.input_method.send_deactivate();
+            self.input_method.send_done();
+            for (_, popup) in &self.input_method.popups {
+                popup.update_visible();
+            }
+        }
+    }
+}