leftwm_core/handlers/
screen_create_handler.rs

1use super::{Manager, Screen, Workspace};
2use crate::config::Config;
3use crate::display_servers::DisplayServer;
4use crate::models::Handle;
5
6impl<H: Handle, C: Config, SERVER: DisplayServer<H>> Manager<H, C, SERVER> {
7    /// `screen_create_handler` is called when the display server sends a
8    /// `DisplayEvent::ScreenCreate(screen)` event. This happens at initialization.
9    ///
10    /// Returns `true` if changes need to be rendered.
11    pub fn screen_create_handler(&mut self, screen: Screen<H>) -> bool {
12        tracing::trace!("Screen create: {:?}", screen);
13
14        let tag_index = self.state.workspaces.len();
15        let tag_len = self.state.tags.len_normal();
16
17        // Only used in tests, where there are multiple screens being created by `Screen::default()`
18        // The screen passed to this function should normally already have it's id given in the config serialization.
19        let workspace_id = match screen.id {
20            None => self.state.workspaces.last().map_or(0, |ws| ws.id) + 1,
21            Some(set_id) => set_id,
22        };
23
24        let mut new_workspace = Workspace::new(screen.bbox, workspace_id);
25        if self.state.workspaces.len() >= tag_len {
26            tracing::warn!(
27                "The number of workspaces needs to be less than or equal to the number of tags available. No more workspaces will be added."
28            );
29        }
30        new_workspace.load_config(&self.config);
31
32        // Make sure there are enough tags for this new screen.
33        let next_id = if tag_len > tag_index {
34            tag_index + 1
35        } else {
36            // Add a new tag for the workspace.
37            self.state.tags.add_new_unlabeled()
38        };
39
40        self.state.focus_workspace(&new_workspace); // focus_workspace is called.
41        self.state.focus_tag(&next_id);
42        new_workspace.show_tag(&next_id);
43        self.state.workspaces.push(new_workspace.clone());
44        self.state.screens.push(screen);
45        self.state.focus_workspace(&new_workspace); // focus_workspace is called again.
46        false
47    }
48}
49
50#[cfg(test)]
51mod tests {
52    use super::*;
53    use crate::Manager;
54
55    #[test]
56    fn creating_two_screens_should_tag_them_with_first_and_second_tags() {
57        let mut manager = Manager::new_test(vec!["1".to_string(), "2".to_string()]);
58        manager.screen_create_handler(Screen::default());
59        manager.screen_create_handler(Screen::default());
60        assert!(manager.state.workspaces[0].has_tag(&1));
61        assert!(manager.state.workspaces[1].has_tag(&2));
62    }
63
64    #[test]
65    fn should_be_able_to_add_screens_with_preexisting_tags() {
66        let mut manager = Manager::new_test(vec![
67            "web".to_string(),
68            "console".to_string(),
69            "code".to_string(),
70        ]);
71        manager.screen_create_handler(Screen::default());
72        manager.screen_create_handler(Screen::default());
73        assert!(manager.state.workspaces[0].has_tag(&1));
74        assert!(manager.state.workspaces[1].has_tag(&2));
75    }
76
77    #[test]
78    fn creating_more_screens_than_tags_should_automatically_create_new_tags() {
79        let mut manager = Manager::new_test(vec!["web".to_string(), "console".to_string()]);
80
81        // there should be 2 tags in the beginning
82        assert_eq!(manager.state.tags.len_normal(), 2);
83
84        manager.screen_create_handler(Screen::default());
85        manager.screen_create_handler(Screen::default());
86        manager.screen_create_handler(Screen::default());
87        manager.screen_create_handler(Screen::default());
88
89        // there should be 4 workspaces
90        assert_eq!(manager.state.workspaces.len(), 4);
91
92        // there should be 4 tags after creating 4 screens/workspaces
93        assert_eq!(manager.state.tags.len_normal(), 4);
94
95        // workspaces should have tags in order
96        assert!(manager.state.workspaces[0].has_tag(&1));
97        assert!(manager.state.workspaces[1].has_tag(&2));
98        assert!(manager.state.workspaces[2].has_tag(&3));
99        assert!(manager.state.workspaces[3].has_tag(&4));
100    }
101}
leftwm_core/handlers/screen_create_handler.rs

leftwm_core/handlers/
screen_create_handler.rs
