From 396ab935859f4eb46b3cf7aa36e7b496b5898b72 Mon Sep 17 00:00:00 2001
From: Ivan Molodetskikh <yalterz@gmail.com>
Date: Wed, 9 Aug 2023 11:03:38 +0400
Subject: Restructure things and add tty backend

---
 src/niri.rs | 168 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 168 insertions(+)
 create mode 100644 src/niri.rs

(limited to 'src/niri.rs')

diff --git a/src/niri.rs b/src/niri.rs
new file mode 100644
index 00000000..1750ed3d
--- /dev/null
+++ b/src/niri.rs
@@ -0,0 +1,168 @@
+use std::os::unix::io::AsRawFd;
+use std::sync::Arc;
+use std::time::Duration;
+
+use smithay::desktop::space::space_render_elements;
+use smithay::desktop::{Space, Window, WindowSurfaceType};
+use smithay::input::keyboard::XkbConfig;
+use smithay::input::pointer::PointerHandle;
+use smithay::input::{Seat, SeatState};
+use smithay::output::Output;
+use smithay::reexports::calloop::generic::Generic;
+use smithay::reexports::calloop::{Interest, LoopHandle, LoopSignal, Mode, PostAction};
+use smithay::reexports::wayland_server::backend::{ClientData, ClientId, DisconnectReason};
+use smithay::reexports::wayland_server::protocol::wl_surface::WlSurface;
+use smithay::reexports::wayland_server::{Display, DisplayHandle};
+use smithay::utils::{Logical, Point};
+use smithay::wayland::compositor::{CompositorClientState, CompositorState};
+use smithay::wayland::data_device::DataDeviceState;
+use smithay::wayland::output::OutputManagerState;
+use smithay::wayland::shell::xdg::XdgShellState;
+use smithay::wayland::shm::ShmState;
+use smithay::wayland::socket::ListeningSocketSource;
+
+use crate::backend::Backend;
+use crate::LoopData;
+
+pub struct Niri {
+    pub start_time: std::time::Instant,
+    pub event_loop: LoopHandle<'static, LoopData>,
+    pub stop_signal: LoopSignal,
+    pub display_handle: DisplayHandle,
+
+    pub space: Space<Window>,
+
+    // Smithay state.
+    pub compositor_state: CompositorState,
+    pub xdg_shell_state: XdgShellState,
+    pub shm_state: ShmState,
+    pub output_manager_state: OutputManagerState,
+    pub seat_state: SeatState<Self>,
+    pub data_device_state: DataDeviceState,
+
+    pub seat: Seat<Self>,
+    pub output: Option<Output>,
+}
+
+impl Niri {
+    pub fn new(
+        event_loop: LoopHandle<'static, LoopData>,
+        stop_signal: LoopSignal,
+        display: &mut Display<Self>,
+        seat_name: String,
+    ) -> Self {
+        let start_time = std::time::Instant::now();
+
+        let display_handle = display.handle();
+
+        let compositor_state = CompositorState::new::<Self>(&display_handle);
+        let xdg_shell_state = XdgShellState::new::<Self>(&display_handle);
+        let shm_state = ShmState::new::<Self>(&display_handle, vec![]);
+        let output_manager_state = OutputManagerState::new_with_xdg_output::<Self>(&display_handle);
+        let mut seat_state = SeatState::new();
+        let data_device_state = DataDeviceState::new::<Self>(&display_handle);
+
+        let mut seat: Seat<Self> = seat_state.new_wl_seat(&display_handle, seat_name);
+        // FIXME: get Xkb and repeat interval from GNOME dconf.
+        seat.add_keyboard(XkbConfig::default(), 400, 30).unwrap();
+        seat.add_pointer();
+
+        let space = Space::default();
+
+        let socket_source = ListeningSocketSource::new_auto().unwrap();
+        let socket_name = socket_source.socket_name().to_os_string();
+        event_loop
+            .insert_source(socket_source, move |client, _, data| {
+                if let Err(err) = data
+                    .display_handle
+                    .insert_client(client, Arc::new(ClientState::default()))
+                {
+                    error!("error inserting client: {err}");
+                }
+            })
+            .unwrap();
+        std::env::set_var("WAYLAND_DISPLAY", &socket_name);
+        info!(
+            "listening on Wayland socket: {}",
+            socket_name.to_string_lossy()
+        );
+
+        let display_source = Generic::new(
+            display.backend().poll_fd().as_raw_fd(),
+            Interest::READ,
+            Mode::Level,
+        );
+        event_loop
+            .insert_source(display_source, |_, _, data| {
+                data.display.dispatch_clients(&mut data.niri).unwrap();
+                Ok(PostAction::Continue)
+            })
+            .unwrap();
+
+        Self {
+            start_time,
+            event_loop,
+            stop_signal,
+            display_handle,
+
+            space,
+
+            compositor_state,
+            xdg_shell_state,
+            shm_state,
+            output_manager_state,
+            seat_state,
+            data_device_state,
+
+            seat,
+            output: None,
+        }
+    }
+
+    pub fn surface_under_pointer(
+        &self,
+        pointer: &PointerHandle<Self>,
+    ) -> Option<(WlSurface, Point<i32, Logical>)> {
+        let pos = pointer.current_location();
+        self.space
+            .element_under(pos)
+            .and_then(|(window, location)| {
+                window
+                    .surface_under(pos - location.to_f64(), WindowSurfaceType::ALL)
+                    .map(|(s, p)| (s, p + location))
+            })
+    }
+
+    pub fn redraw(&mut self, backend: &mut dyn Backend) {
+        let elements = space_render_elements(
+            backend.renderer(),
+            [&self.space],
+            self.output.as_ref().unwrap(),
+            1.,
+        )
+        .unwrap();
+        backend.render(self, &elements);
+
+        let output = self.output.as_ref().unwrap();
+        self.space.elements().for_each(|window| {
+            window.send_frame(
+                output,
+                self.start_time.elapsed(),
+                Some(Duration::ZERO),
+                |_, _| Some(output.clone()),
+            )
+        });
+
+        self.space.refresh();
+    }
+}
+
+#[derive(Default)]
+pub struct ClientState {
+    pub compositor_state: CompositorClientState,
+}
+
+impl ClientData for ClientState {
+    fn initialized(&self, _client_id: ClientId) {}
+    fn disconnected(&self, _client_id: ClientId, _reason: DisconnectReason) {}
+}
-- 
cgit