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!("The number of workspaces needs to be less than or equal to the number of tags available. No more workspaces will be added.");
27        }
28        new_workspace.load_config(&self.config);
29
30        // Make sure there are enough tags for this new screen.
31        let next_id = if tag_len > tag_index {
32            tag_index + 1
33        } else {
34            // Add a new tag for the workspace.
35            self.state.tags.add_new_unlabeled()
36        };
37
38        self.state.focus_workspace(&new_workspace); // focus_workspace is called.
39        self.state.focus_tag(&next_id);
40        new_workspace.show_tag(&next_id);
41        self.state.workspaces.push(new_workspace.clone());
42        self.state.screens.push(screen);
43        self.state.focus_workspace(&new_workspace); // focus_workspace is called again.
44        false
45    }
46}
47
48#[cfg(test)]
49mod tests {
50    use super::*;
51    use crate::Manager;
52
53    #[test]
54    fn creating_two_screens_should_tag_them_with_first_and_second_tags() {
55        let mut manager = Manager::new_test(vec!["1".to_string(), "2".to_string()]);
56        manager.screen_create_handler(Screen::default());
57        manager.screen_create_handler(Screen::default());
58        assert!(manager.state.workspaces[0].has_tag(&1));
59        assert!(manager.state.workspaces[1].has_tag(&2));
60    }
61
62    #[test]
63    fn should_be_able_to_add_screens_with_preexisting_tags() {
64        let mut manager = Manager::new_test(vec![
65            "web".to_string(),
66            "console".to_string(),
67            "code".to_string(),
68        ]);
69        manager.screen_create_handler(Screen::default());
70        manager.screen_create_handler(Screen::default());
71        assert!(manager.state.workspaces[0].has_tag(&1));
72        assert!(manager.state.workspaces[1].has_tag(&2));
73    }
74
75    #[test]
76    fn creating_more_screens_than_tags_should_automatically_create_new_tags() {
77        let mut manager = Manager::new_test(vec!["web".to_string(), "console".to_string()]);
78
79        // there should be 2 tags in the beginning
80        assert_eq!(manager.state.tags.len_normal(), 2);
81
82        manager.screen_create_handler(Screen::default());
83        manager.screen_create_handler(Screen::default());
84        manager.screen_create_handler(Screen::default());
85        manager.screen_create_handler(Screen::default());
86
87        // there should be 4 workspaces
88        assert_eq!(manager.state.workspaces.len(), 4);
89
90        // there should be 4 tags after creating 4 screens/workspaces
91        assert_eq!(manager.state.tags.len_normal(), 4);
92
93        // workspaces should have tags in order
94        assert!(manager.state.workspaces[0].has_tag(&1));
95        assert!(manager.state.workspaces[1].has_tag(&2));
96        assert!(manager.state.workspaces[2].has_tag(&3));
97        assert!(manager.state.workspaces[3].has_tag(&4));
98    }
99}
leftwm_core/handlers/screen_create_handler.rs

leftwm_core/handlers/
screen_create_handler.rs
