servoshell/desktop/
app.rs

1/* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
4
5//! Application entry point, runs the event loop.
6
7use std::path::Path;
8use std::rc::Rc;
9use std::time::Instant;
10use std::{env, fs};
11
12use servo::protocol_handler::ProtocolRegistry;
13use servo::{
14    EventLoopWaker, Opts, Preferences, ServoBuilder, ServoUrl, UserContentManager, UserScript,
15};
16use url::Url;
17use winit::application::ApplicationHandler;
18use winit::event::WindowEvent;
19use winit::event_loop::{ActiveEventLoop, ControlFlow, EventLoopProxy};
20use winit::window::WindowId;
21
22use super::event_loop::AppEvent;
23use crate::desktop::event_loop::ServoShellEventLoop;
24use crate::desktop::headed_window::HeadedWindow;
25use crate::desktop::headless_window::HeadlessWindow;
26use crate::desktop::protocols;
27use crate::desktop::tracing::trace_winit_event;
28use crate::parser::get_default_url;
29use crate::prefs::ServoShellPreferences;
30use crate::running_app_state::RunningAppState;
31#[cfg(feature = "gamepad")]
32use crate::running_app_state::ServoshellGamepadDelegate;
33use crate::window::{PlatformWindow, ServoShellWindowId};
34
35pub(crate) enum AppState {
36    Initializing,
37    Running(Rc<RunningAppState>),
38    ShuttingDown,
39}
40
41pub struct App {
42    opts: Opts,
43    preferences: Preferences,
44    servoshell_preferences: ServoShellPreferences,
45    waker: Box<dyn EventLoopWaker>,
46    event_loop_proxy: Option<EventLoopProxy<AppEvent>>,
47    initial_url: ServoUrl,
48    t_start: Instant,
49    t: Instant,
50    state: AppState,
51}
52
53impl App {
54    pub fn new(
55        opts: Opts,
56        preferences: Preferences,
57        servo_shell_preferences: ServoShellPreferences,
58        event_loop: &ServoShellEventLoop,
59    ) -> Self {
60        let initial_url = get_default_url(
61            servo_shell_preferences.url.as_deref(),
62            env::current_dir().unwrap(),
63            |path| fs::metadata(path).is_ok(),
64            &servo_shell_preferences,
65        );
66
67        let t = Instant::now();
68        App {
69            opts,
70            preferences,
71            servoshell_preferences: servo_shell_preferences,
72            waker: event_loop.create_event_loop_waker(),
73            event_loop_proxy: event_loop.event_loop_proxy(),
74            initial_url,
75            t_start: t,
76            t,
77            state: AppState::Initializing,
78        }
79    }
80
81    /// Initialize Application once event loop start running.
82    pub fn init(&mut self, active_event_loop: Option<&ActiveEventLoop>) {
83        let mut protocol_registry = ProtocolRegistry::default();
84        let _ = protocol_registry.register(
85            "urlinfo",
86            protocols::urlinfo::UrlInfoProtocolHander::default(),
87        );
88        let _ =
89            protocol_registry.register("servo", protocols::servo::ServoProtocolHandler::default());
90        let _ = protocol_registry.register(
91            "resource",
92            protocols::resource::ResourceProtocolHandler::default(),
93        );
94
95        let servo_builder = ServoBuilder::default()
96            .opts(self.opts.clone())
97            .preferences(self.preferences.clone())
98            .protocol_registry(protocol_registry)
99            .event_loop_waker(self.waker.clone());
100
101        let url = self.initial_url.as_url().clone();
102        let platform_window = self.create_platform_window(url, active_event_loop);
103
104        #[cfg(feature = "webxr")]
105        let servo_builder =
106            servo_builder.webxr_registry(super::webxr::XrDiscoveryWebXrRegistry::new_boxed(
107                platform_window.clone(),
108                active_event_loop,
109                &self.preferences,
110            ));
111
112        let servo = servo_builder.build();
113        servo.setup_logging();
114
115        let user_content_manager = Rc::new(UserContentManager::new(&servo));
116        for script in load_userscripts(self.servoshell_preferences.userscripts_directory.as_deref())
117            .expect("Loading userscripts failed")
118        {
119            user_content_manager.add_script(Rc::new(script));
120        }
121
122        for user_stylesheet in &self.servoshell_preferences.user_stylesheets {
123            user_content_manager.add_stylesheet(user_stylesheet.clone());
124        }
125
126        let running_state = Rc::new(RunningAppState::new(
127            servo,
128            self.servoshell_preferences.clone(),
129            self.waker.clone(),
130            user_content_manager,
131            self.preferences.clone(),
132            #[cfg(feature = "gamepad")]
133            ServoshellGamepadDelegate::maybe_new().map(Rc::new),
134        ));
135        running_state.open_window(platform_window, self.initial_url.as_url().clone());
136
137        self.state = AppState::Running(running_state);
138    }
139
140    #[servo::servo_tracing::instrument(level = "debug", skip_all)]
141    fn create_platform_window(
142        &self,
143        url: Url,
144        active_event_loop: Option<&ActiveEventLoop>,
145    ) -> Rc<dyn PlatformWindow> {
146        assert_eq!(
147            self.servoshell_preferences.headless,
148            active_event_loop.is_none()
149        );
150
151        let Some(active_event_loop) = active_event_loop else {
152            return HeadlessWindow::new(&self.servoshell_preferences);
153        };
154
155        HeadedWindow::new(
156            &self.servoshell_preferences,
157            active_event_loop,
158            self.event_loop_proxy
159                .clone()
160                .expect("Should always have event loop proxy in headed mode."),
161            url,
162        )
163    }
164
165    pub fn pump_servo_event_loop(&mut self, active_event_loop: Option<&ActiveEventLoop>) -> bool {
166        let AppState::Running(state) = &self.state else {
167            return false;
168        };
169
170        let create_platform_window = |url: Url| self.create_platform_window(url, active_event_loop);
171        if !state.spin_event_loop(Some(&create_platform_window)) {
172            self.state = AppState::ShuttingDown;
173            return false;
174        }
175        true
176    }
177}
178
179impl ApplicationHandler<AppEvent> for App {
180    fn resumed(&mut self, event_loop: &ActiveEventLoop) {
181        self.init(Some(event_loop));
182    }
183
184    fn window_event(
185        &mut self,
186        event_loop: &ActiveEventLoop,
187        window_id: WindowId,
188        window_event: WindowEvent,
189    ) {
190        let now = Instant::now();
191        trace_winit_event!(
192            window_event,
193            "@{:?} (+{:?}) {window_event:?}",
194            now - self.t_start,
195            now - self.t
196        );
197        self.t = now;
198
199        let AppState::Running(state) = &self.state else {
200            return;
201        };
202
203        if let Some(window) = state.window(ServoShellWindowId::from(u64::from(window_id))) &&
204            let Some(headed_window) = window.platform_window().as_headed_window()
205        {
206            headed_window.handle_winit_window_event(state.clone(), window, window_event);
207        }
208
209        if !self.pump_servo_event_loop(event_loop.into()) {
210            event_loop.exit();
211        }
212        // Block until the window gets an event
213        event_loop.set_control_flow(ControlFlow::Wait);
214    }
215
216    fn user_event(&mut self, event_loop: &ActiveEventLoop, app_event: AppEvent) {
217        let AppState::Running(state) = &self.state else {
218            return;
219        };
220
221        if let Some(window) = app_event
222            .window_id()
223            .and_then(|window_id| state.window(ServoShellWindowId::from(u64::from(window_id)))) &&
224            let Some(headed_window) = window.platform_window().as_headed_window()
225        {
226            headed_window.handle_winit_app_event(state.clone(), app_event);
227        }
228
229        if !self.pump_servo_event_loop(event_loop.into()) {
230            event_loop.exit();
231        }
232
233        // Block until the window gets an event
234        event_loop.set_control_flow(ControlFlow::Wait);
235    }
236}
237
238fn load_userscripts(userscripts_directory: Option<&Path>) -> std::io::Result<Vec<UserScript>> {
239    let mut userscripts = Vec::new();
240    if let Some(userscripts_directory) = &userscripts_directory {
241        let mut files = std::fs::read_dir(userscripts_directory)?
242            .map(|e| e.map(|entry| entry.path()))
243            .collect::<Result<Vec<_>, _>>()?;
244        files.sort_unstable();
245        for file in files {
246            let script = std::fs::read_to_string(&file)?;
247            userscripts.push(UserScript::new(script, Some(file)));
248        }
249    }
250    Ok(userscripts)
251}
servoshell/desktop/app.rs

servoshell/desktop/
app.rs
