lib.rs 58 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001
  1. // Copyright 2019-2021 Tauri Programme within The Commons Conservancy
  2. // SPDX-License-Identifier: Apache-2.0
  3. // SPDX-License-Identifier: MIT
  4. //! The [`wry`] Tauri [`Runtime`].
  5. use tauri_runtime::{
  6. monitor::Monitor,
  7. webview::{
  8. FileDropEvent, FileDropHandler, RpcRequest, WebviewRpcHandler, WindowBuilder, WindowBuilderBase,
  9. },
  10. window::{
  11. dpi::{LogicalPosition, LogicalSize, PhysicalPosition, PhysicalSize, Position, Size},
  12. DetachedWindow, PendingWindow, WindowEvent,
  13. },
  14. ClipboardManager, Dispatch, Error, GlobalShortcutManager, Icon, Params, Result, RunEvent,
  15. RunIteration, Runtime, RuntimeHandle, UserAttentionType,
  16. };
  17. #[cfg(feature = "menu")]
  18. use tauri_runtime::window::MenuEvent;
  19. #[cfg(feature = "system-tray")]
  20. use tauri_runtime::{SystemTray, SystemTrayEvent};
  21. #[cfg(windows)]
  22. use winapi::shared::windef::HWND;
  23. #[cfg(windows)]
  24. use wry::application::platform::windows::WindowBuilderExtWindows;
  25. #[cfg(feature = "system-tray")]
  26. use wry::application::system_tray::{SystemTray as WrySystemTray, SystemTrayBuilder};
  27. use tauri_utils::config::WindowConfig;
  28. use uuid::Uuid;
  29. use wry::{
  30. application::{
  31. accelerator::{Accelerator, AcceleratorId},
  32. clipboard::Clipboard,
  33. dpi::{
  34. LogicalPosition as WryLogicalPosition, LogicalSize as WryLogicalSize,
  35. PhysicalPosition as WryPhysicalPosition, PhysicalSize as WryPhysicalSize,
  36. Position as WryPosition, Size as WrySize,
  37. },
  38. event::{Event, WindowEvent as WryWindowEvent},
  39. event_loop::{ControlFlow, EventLoop, EventLoopProxy, EventLoopWindowTarget},
  40. monitor::MonitorHandle,
  41. platform::global_shortcut::{GlobalShortcut, ShortcutManager as WryShortcutManager},
  42. window::{
  43. Fullscreen, Icon as WindowIcon, UserAttentionType as WryUserAttentionType, Window,
  44. WindowBuilder as WryWindowBuilder, WindowId,
  45. },
  46. },
  47. webview::{
  48. FileDropEvent as WryFileDropEvent, RpcRequest as WryRpcRequest, RpcResponse, WebContext,
  49. WebView, WebViewBuilder,
  50. },
  51. };
  52. use std::{
  53. collections::{
  54. hash_map::Entry::{Occupied, Vacant},
  55. HashMap,
  56. },
  57. convert::TryFrom,
  58. fs::read,
  59. path::PathBuf,
  60. sync::{
  61. atomic::{AtomicBool, Ordering},
  62. mpsc::{channel, Sender},
  63. Arc, Mutex, MutexGuard,
  64. },
  65. thread::{current as current_thread, ThreadId},
  66. };
  67. #[cfg(any(feature = "menu", feature = "system-tray"))]
  68. mod menu;
  69. #[cfg(any(feature = "menu", feature = "system-tray"))]
  70. use menu::*;
  71. mod mime_type;
  72. use mime_type::MimeType;
  73. type MainTask = Arc<Mutex<Option<Box<dyn FnOnce() + Send>>>>;
  74. type WebContextStore = Mutex<HashMap<Option<PathBuf>, WebContext>>;
  75. type CreateWebviewHandler = Box<
  76. dyn FnOnce(&EventLoopWindowTarget<Message>, &WebContextStore) -> Result<WebviewWrapper> + Send,
  77. >;
  78. type WindowEventHandler = Box<dyn Fn(&WindowEvent) + Send>;
  79. type WindowEventListenersMap = Arc<Mutex<HashMap<Uuid, WindowEventHandler>>>;
  80. type WindowEventListeners = Arc<Mutex<HashMap<WindowId, WindowEventListenersMap>>>;
  81. type GlobalShortcutListeners = Arc<Mutex<HashMap<AcceleratorId, Box<dyn Fn() + Send>>>>;
  82. macro_rules! dispatcher_getter {
  83. ($self: ident, $message: expr) => {{
  84. if current_thread().id() == $self.context.main_thread_id
  85. && !$self.context.is_event_loop_running.load(Ordering::Relaxed)
  86. {
  87. panic!("This API cannot be called when the event loop is not running");
  88. }
  89. let (tx, rx) = channel();
  90. $self
  91. .context
  92. .proxy
  93. .send_event(Message::Window($self.window_id, $message(tx)))
  94. .map_err(|_| Error::FailedToSendMessage)?;
  95. rx.recv().unwrap()
  96. }};
  97. }
  98. macro_rules! getter {
  99. ($self: ident, $rx: expr, $message: expr) => {{
  100. if current_thread().id() == $self.context.main_thread_id
  101. && !$self.context.is_event_loop_running.load(Ordering::Relaxed)
  102. {
  103. panic!("This API cannot be called when the event loop is not running");
  104. }
  105. $self
  106. .context
  107. .proxy
  108. .send_event($message)
  109. .map_err(|_| Error::FailedToSendMessage)?;
  110. $rx.recv().unwrap()
  111. }};
  112. }
  113. #[derive(Clone)]
  114. struct EventLoopContext {
  115. main_thread_id: ThreadId,
  116. is_event_loop_running: Arc<AtomicBool>,
  117. proxy: EventLoopProxy<Message>,
  118. }
  119. #[derive(Debug, Clone)]
  120. struct GlobalShortcutWrapper(GlobalShortcut);
  121. unsafe impl Send for GlobalShortcutWrapper {}
  122. /// Wrapper around [`WryShortcutManager`].
  123. #[derive(Clone)]
  124. pub struct GlobalShortcutManagerHandle {
  125. context: EventLoopContext,
  126. shortcuts: HashMap<String, (AcceleratorId, GlobalShortcutWrapper)>,
  127. listeners: GlobalShortcutListeners,
  128. }
  129. impl GlobalShortcutManager for GlobalShortcutManagerHandle {
  130. fn is_registered(&self, accelerator: &str) -> Result<bool> {
  131. let (tx, rx) = channel();
  132. Ok(getter!(
  133. self,
  134. rx,
  135. Message::GlobalShortcut(GlobalShortcutMessage::IsRegistered(
  136. accelerator.parse().expect("invalid accelerator"),
  137. tx
  138. ))
  139. ))
  140. }
  141. fn register<F: Fn() + Send + 'static>(&mut self, accelerator: &str, handler: F) -> Result<()> {
  142. let wry_accelerator: Accelerator = accelerator.parse().expect("invalid accelerator");
  143. let id = wry_accelerator.clone().id();
  144. let (tx, rx) = channel();
  145. let shortcut = getter!(
  146. self,
  147. rx,
  148. Message::GlobalShortcut(GlobalShortcutMessage::Register(wry_accelerator, tx))
  149. )?;
  150. self.listeners.lock().unwrap().insert(id, Box::new(handler));
  151. self.shortcuts.insert(accelerator.into(), (id, shortcut));
  152. Ok(())
  153. }
  154. fn unregister_all(&mut self) -> Result<()> {
  155. let (tx, rx) = channel();
  156. getter!(
  157. self,
  158. rx,
  159. Message::GlobalShortcut(GlobalShortcutMessage::UnregisterAll(tx))
  160. )?;
  161. self.listeners.lock().unwrap().clear();
  162. self.shortcuts.clear();
  163. Ok(())
  164. }
  165. fn unregister(&mut self, accelerator: &str) -> Result<()> {
  166. if let Some((accelerator_id, shortcut)) = self.shortcuts.remove(accelerator) {
  167. let (tx, rx) = channel();
  168. getter!(
  169. self,
  170. rx,
  171. Message::GlobalShortcut(GlobalShortcutMessage::Unregister(shortcut, tx))
  172. )?;
  173. self.listeners.lock().unwrap().remove(&accelerator_id);
  174. }
  175. Ok(())
  176. }
  177. }
  178. #[derive(Clone)]
  179. pub struct ClipboardManagerWrapper {
  180. context: EventLoopContext,
  181. }
  182. impl ClipboardManager for ClipboardManagerWrapper {
  183. fn read_text(&self) -> Result<Option<String>> {
  184. let (tx, rx) = channel();
  185. Ok(getter!(
  186. self,
  187. rx,
  188. Message::Clipboard(ClipboardMessage::ReadText(tx))
  189. ))
  190. }
  191. fn write_text<T: Into<String>>(&mut self, text: T) -> Result<()> {
  192. let (tx, rx) = channel();
  193. getter!(
  194. self,
  195. rx,
  196. Message::Clipboard(ClipboardMessage::WriteText(text.into(), tx))
  197. );
  198. Ok(())
  199. }
  200. }
  201. /// Wrapper around a [`wry::application::window::Icon`] that can be created from an [`Icon`].
  202. pub struct WryIcon(WindowIcon);
  203. fn icon_err<E: std::error::Error + Send + 'static>(e: E) -> Error {
  204. Error::InvalidIcon(Box::new(e))
  205. }
  206. impl TryFrom<Icon> for WryIcon {
  207. type Error = Error;
  208. fn try_from(icon: Icon) -> std::result::Result<Self, Self::Error> {
  209. let image_bytes = match icon {
  210. Icon::File(path) => read(path).map_err(icon_err)?,
  211. Icon::Raw(raw) => raw,
  212. _ => unimplemented!(),
  213. };
  214. let extension = infer::get(&image_bytes)
  215. .expect("could not determine icon extension")
  216. .extension();
  217. match extension {
  218. #[cfg(windows)]
  219. "ico" => {
  220. let icon_dir = ico::IconDir::read(std::io::Cursor::new(image_bytes)).map_err(icon_err)?;
  221. let entry = &icon_dir.entries()[0];
  222. let icon = WindowIcon::from_rgba(
  223. entry.decode().map_err(icon_err)?.rgba_data().to_vec(),
  224. entry.width(),
  225. entry.height(),
  226. )
  227. .map_err(icon_err)?;
  228. Ok(Self(icon))
  229. }
  230. #[cfg(target_os = "linux")]
  231. "png" => {
  232. let decoder = png::Decoder::new(std::io::Cursor::new(image_bytes));
  233. let (info, mut reader) = decoder.read_info().map_err(icon_err)?;
  234. let mut buffer = Vec::new();
  235. while let Ok(Some(row)) = reader.next_row() {
  236. buffer.extend(row);
  237. }
  238. let icon = WindowIcon::from_rgba(buffer, info.width, info.height).map_err(icon_err)?;
  239. Ok(Self(icon))
  240. }
  241. _ => panic!(
  242. "image `{}` extension not supported; please file a Tauri feature request",
  243. extension
  244. ),
  245. }
  246. }
  247. }
  248. struct WindowEventWrapper(Option<WindowEvent>);
  249. impl<'a> From<&WryWindowEvent<'a>> for WindowEventWrapper {
  250. fn from(event: &WryWindowEvent<'a>) -> Self {
  251. let event = match event {
  252. WryWindowEvent::Resized(size) => WindowEvent::Resized(PhysicalSizeWrapper(*size).into()),
  253. WryWindowEvent::Moved(position) => {
  254. WindowEvent::Moved(PhysicalPositionWrapper(*position).into())
  255. }
  256. WryWindowEvent::CloseRequested => WindowEvent::CloseRequested,
  257. WryWindowEvent::Destroyed => WindowEvent::Destroyed,
  258. WryWindowEvent::Focused(focused) => WindowEvent::Focused(*focused),
  259. WryWindowEvent::ScaleFactorChanged {
  260. scale_factor,
  261. new_inner_size,
  262. } => WindowEvent::ScaleFactorChanged {
  263. scale_factor: *scale_factor,
  264. new_inner_size: PhysicalSizeWrapper(**new_inner_size).into(),
  265. },
  266. _ => return Self(None),
  267. };
  268. Self(Some(event))
  269. }
  270. }
  271. pub struct MonitorHandleWrapper(MonitorHandle);
  272. impl From<MonitorHandleWrapper> for Monitor {
  273. fn from(monitor: MonitorHandleWrapper) -> Monitor {
  274. Self {
  275. name: monitor.0.name(),
  276. position: PhysicalPositionWrapper(monitor.0.position()).into(),
  277. size: PhysicalSizeWrapper(monitor.0.size()).into(),
  278. scale_factor: monitor.0.scale_factor(),
  279. }
  280. }
  281. }
  282. struct PhysicalPositionWrapper<T>(WryPhysicalPosition<T>);
  283. impl<T> From<PhysicalPositionWrapper<T>> for PhysicalPosition<T> {
  284. fn from(position: PhysicalPositionWrapper<T>) -> Self {
  285. Self {
  286. x: position.0.x,
  287. y: position.0.y,
  288. }
  289. }
  290. }
  291. impl<T> From<PhysicalPosition<T>> for PhysicalPositionWrapper<T> {
  292. fn from(position: PhysicalPosition<T>) -> Self {
  293. Self(WryPhysicalPosition {
  294. x: position.x,
  295. y: position.y,
  296. })
  297. }
  298. }
  299. struct LogicalPositionWrapper<T>(WryLogicalPosition<T>);
  300. impl<T> From<LogicalPosition<T>> for LogicalPositionWrapper<T> {
  301. fn from(position: LogicalPosition<T>) -> Self {
  302. Self(WryLogicalPosition {
  303. x: position.x,
  304. y: position.y,
  305. })
  306. }
  307. }
  308. struct PhysicalSizeWrapper<T>(WryPhysicalSize<T>);
  309. impl<T> From<PhysicalSizeWrapper<T>> for PhysicalSize<T> {
  310. fn from(size: PhysicalSizeWrapper<T>) -> Self {
  311. Self {
  312. width: size.0.width,
  313. height: size.0.height,
  314. }
  315. }
  316. }
  317. impl<T> From<PhysicalSize<T>> for PhysicalSizeWrapper<T> {
  318. fn from(size: PhysicalSize<T>) -> Self {
  319. Self(WryPhysicalSize {
  320. width: size.width,
  321. height: size.height,
  322. })
  323. }
  324. }
  325. struct LogicalSizeWrapper<T>(WryLogicalSize<T>);
  326. impl<T> From<LogicalSize<T>> for LogicalSizeWrapper<T> {
  327. fn from(size: LogicalSize<T>) -> Self {
  328. Self(WryLogicalSize {
  329. width: size.width,
  330. height: size.height,
  331. })
  332. }
  333. }
  334. struct SizeWrapper(WrySize);
  335. impl From<Size> for SizeWrapper {
  336. fn from(size: Size) -> Self {
  337. match size {
  338. Size::Logical(s) => Self(WrySize::Logical(LogicalSizeWrapper::from(s).0)),
  339. Size::Physical(s) => Self(WrySize::Physical(PhysicalSizeWrapper::from(s).0)),
  340. }
  341. }
  342. }
  343. struct PositionWrapper(WryPosition);
  344. impl From<Position> for PositionWrapper {
  345. fn from(position: Position) -> Self {
  346. match position {
  347. Position::Logical(s) => Self(WryPosition::Logical(LogicalPositionWrapper::from(s).0)),
  348. Position::Physical(s) => Self(WryPosition::Physical(PhysicalPositionWrapper::from(s).0)),
  349. }
  350. }
  351. }
  352. #[derive(Debug, Clone)]
  353. struct UserAttentionTypeWrapper(WryUserAttentionType);
  354. impl From<UserAttentionType> for UserAttentionTypeWrapper {
  355. fn from(request_type: UserAttentionType) -> UserAttentionTypeWrapper {
  356. let o = match request_type {
  357. UserAttentionType::Critical => WryUserAttentionType::Critical,
  358. UserAttentionType::Informational => WryUserAttentionType::Informational,
  359. };
  360. Self(o)
  361. }
  362. }
  363. #[derive(Debug, Clone, Default)]
  364. pub struct WindowBuilderWrapper {
  365. inner: WryWindowBuilder,
  366. center: bool,
  367. #[cfg(feature = "menu")]
  368. menu: Menu<u16>,
  369. }
  370. // safe since `menu_items` are read only here
  371. unsafe impl Send for WindowBuilderWrapper {}
  372. impl WindowBuilderBase for WindowBuilderWrapper {}
  373. impl WindowBuilder for WindowBuilderWrapper {
  374. fn new() -> Self {
  375. Default::default()
  376. }
  377. fn with_config(config: WindowConfig) -> Self {
  378. let mut window = WindowBuilderWrapper::new()
  379. .title(config.title.to_string())
  380. .inner_size(config.width, config.height)
  381. .visible(config.visible)
  382. .resizable(config.resizable)
  383. .decorations(config.decorations)
  384. .maximized(config.maximized)
  385. .fullscreen(config.fullscreen)
  386. .transparent(config.transparent)
  387. .always_on_top(config.always_on_top)
  388. .skip_taskbar(config.skip_taskbar);
  389. if let (Some(min_width), Some(min_height)) = (config.min_width, config.min_height) {
  390. window = window.min_inner_size(min_width, min_height);
  391. }
  392. if let (Some(max_width), Some(max_height)) = (config.max_width, config.max_height) {
  393. window = window.max_inner_size(max_width, max_height);
  394. }
  395. if let (Some(x), Some(y)) = (config.x, config.y) {
  396. window = window.position(x, y);
  397. }
  398. if config.focus {
  399. window = window.focus();
  400. }
  401. window
  402. }
  403. #[cfg(feature = "menu")]
  404. fn menu<I: MenuId>(mut self, menu: Menu<I>) -> Self {
  405. self.menu = convert_menu_id(Menu::new(), menu);
  406. self
  407. }
  408. fn center(mut self) -> Self {
  409. self.center = true;
  410. self
  411. }
  412. fn position(mut self, x: f64, y: f64) -> Self {
  413. self.inner = self.inner.with_position(WryLogicalPosition::new(x, y));
  414. self
  415. }
  416. fn inner_size(mut self, width: f64, height: f64) -> Self {
  417. self.inner = self
  418. .inner
  419. .with_inner_size(WryLogicalSize::new(width, height));
  420. self
  421. }
  422. fn min_inner_size(mut self, min_width: f64, min_height: f64) -> Self {
  423. self.inner = self
  424. .inner
  425. .with_min_inner_size(WryLogicalSize::new(min_width, min_height));
  426. self
  427. }
  428. fn max_inner_size(mut self, max_width: f64, max_height: f64) -> Self {
  429. self.inner = self
  430. .inner
  431. .with_max_inner_size(WryLogicalSize::new(max_width, max_height));
  432. self
  433. }
  434. fn resizable(mut self, resizable: bool) -> Self {
  435. self.inner = self.inner.with_resizable(resizable);
  436. self
  437. }
  438. fn title<S: Into<String>>(mut self, title: S) -> Self {
  439. self.inner = self.inner.with_title(title.into());
  440. self
  441. }
  442. fn fullscreen(mut self, fullscreen: bool) -> Self {
  443. self.inner = if fullscreen {
  444. self
  445. .inner
  446. .with_fullscreen(Some(Fullscreen::Borderless(None)))
  447. } else {
  448. self.inner.with_fullscreen(None)
  449. };
  450. self
  451. }
  452. fn focus(mut self) -> Self {
  453. self.inner = self.inner.with_focus();
  454. self
  455. }
  456. fn maximized(mut self, maximized: bool) -> Self {
  457. self.inner = self.inner.with_maximized(maximized);
  458. self
  459. }
  460. fn visible(mut self, visible: bool) -> Self {
  461. self.inner = self.inner.with_visible(visible);
  462. self
  463. }
  464. fn transparent(mut self, transparent: bool) -> Self {
  465. self.inner = self.inner.with_transparent(transparent);
  466. self
  467. }
  468. fn decorations(mut self, decorations: bool) -> Self {
  469. self.inner = self.inner.with_decorations(decorations);
  470. self
  471. }
  472. fn always_on_top(mut self, always_on_top: bool) -> Self {
  473. self.inner = self.inner.with_always_on_top(always_on_top);
  474. self
  475. }
  476. #[cfg(windows)]
  477. fn parent_window(mut self, parent: HWND) -> Self {
  478. self.inner = self.inner.with_parent_window(parent);
  479. self
  480. }
  481. #[cfg(windows)]
  482. fn owner_window(mut self, owner: HWND) -> Self {
  483. self.inner = self.inner.with_owner_window(owner);
  484. self
  485. }
  486. fn icon(mut self, icon: Icon) -> Result<Self> {
  487. self.inner = self
  488. .inner
  489. .with_window_icon(Some(WryIcon::try_from(icon)?.0));
  490. Ok(self)
  491. }
  492. fn skip_taskbar(mut self, skip: bool) -> Self {
  493. self.inner = self.inner.with_skip_taskbar(skip);
  494. self
  495. }
  496. fn has_icon(&self) -> bool {
  497. self.inner.window.window_icon.is_some()
  498. }
  499. #[cfg(feature = "menu")]
  500. fn has_menu(&self) -> bool {
  501. self.inner.window.window_menu.is_some()
  502. }
  503. }
  504. pub struct RpcRequestWrapper(WryRpcRequest);
  505. impl From<RpcRequestWrapper> for RpcRequest {
  506. fn from(request: RpcRequestWrapper) -> Self {
  507. Self {
  508. command: request.0.method,
  509. params: request.0.params,
  510. }
  511. }
  512. }
  513. pub struct FileDropEventWrapper(WryFileDropEvent);
  514. impl From<FileDropEventWrapper> for FileDropEvent {
  515. fn from(event: FileDropEventWrapper) -> Self {
  516. match event.0 {
  517. WryFileDropEvent::Hovered(paths) => FileDropEvent::Hovered(paths),
  518. WryFileDropEvent::Dropped(paths) => FileDropEvent::Dropped(paths),
  519. WryFileDropEvent::Cancelled => FileDropEvent::Cancelled,
  520. }
  521. }
  522. }
  523. #[cfg(windows)]
  524. struct Hwnd(*mut std::ffi::c_void);
  525. #[cfg(windows)]
  526. unsafe impl Send for Hwnd {}
  527. #[derive(Debug, Clone)]
  528. enum WindowMessage {
  529. // Getters
  530. ScaleFactor(Sender<f64>),
  531. InnerPosition(Sender<Result<PhysicalPosition<i32>>>),
  532. OuterPosition(Sender<Result<PhysicalPosition<i32>>>),
  533. InnerSize(Sender<PhysicalSize<u32>>),
  534. OuterSize(Sender<PhysicalSize<u32>>),
  535. IsFullscreen(Sender<bool>),
  536. IsMaximized(Sender<bool>),
  537. IsDecorated(Sender<bool>),
  538. IsResizable(Sender<bool>),
  539. IsVisible(Sender<bool>),
  540. #[cfg(feature = "menu")]
  541. IsMenuVisible(Sender<bool>),
  542. CurrentMonitor(Sender<Option<MonitorHandle>>),
  543. PrimaryMonitor(Sender<Option<MonitorHandle>>),
  544. AvailableMonitors(Sender<Vec<MonitorHandle>>),
  545. #[cfg(windows)]
  546. Hwnd(Sender<Hwnd>),
  547. // Setters
  548. Center(Sender<Result<()>>),
  549. RequestUserAttention(Option<UserAttentionTypeWrapper>),
  550. SetResizable(bool),
  551. SetTitle(String),
  552. Maximize,
  553. Unmaximize,
  554. Minimize,
  555. Unminimize,
  556. #[cfg(feature = "menu")]
  557. ShowMenu,
  558. #[cfg(feature = "menu")]
  559. HideMenu,
  560. Show,
  561. Hide,
  562. Close,
  563. SetDecorations(bool),
  564. SetAlwaysOnTop(bool),
  565. SetSize(Size),
  566. SetMinSize(Option<Size>),
  567. SetMaxSize(Option<Size>),
  568. SetPosition(Position),
  569. SetFullscreen(bool),
  570. SetFocus,
  571. SetIcon(WindowIcon),
  572. SetSkipTaskbar(bool),
  573. DragWindow,
  574. #[cfg(feature = "menu")]
  575. UpdateMenuItem(u16, menu::MenuUpdate),
  576. }
  577. #[derive(Debug, Clone)]
  578. enum WebviewMessage {
  579. EvaluateScript(String),
  580. Print,
  581. }
  582. #[cfg(feature = "system-tray")]
  583. #[derive(Clone)]
  584. pub(crate) enum TrayMessage {
  585. UpdateItem(u16, menu::MenuUpdate),
  586. UpdateIcon(Icon),
  587. #[cfg(windows)]
  588. Remove,
  589. }
  590. #[derive(Clone)]
  591. pub(crate) enum GlobalShortcutMessage {
  592. IsRegistered(Accelerator, Sender<bool>),
  593. Register(Accelerator, Sender<Result<GlobalShortcutWrapper>>),
  594. Unregister(GlobalShortcutWrapper, Sender<Result<()>>),
  595. UnregisterAll(Sender<Result<()>>),
  596. }
  597. #[derive(Clone)]
  598. pub(crate) enum ClipboardMessage {
  599. WriteText(String, Sender<()>),
  600. ReadText(Sender<Option<String>>),
  601. }
  602. #[derive(Clone)]
  603. pub(crate) enum Message {
  604. Task(MainTask),
  605. Window(WindowId, WindowMessage),
  606. Webview(WindowId, WebviewMessage),
  607. #[cfg(feature = "system-tray")]
  608. Tray(TrayMessage),
  609. CreateWebview(Arc<Mutex<Option<CreateWebviewHandler>>>, Sender<WindowId>),
  610. GlobalShortcut(GlobalShortcutMessage),
  611. Clipboard(ClipboardMessage),
  612. }
  613. #[derive(Clone)]
  614. struct DispatcherContext {
  615. main_thread_id: ThreadId,
  616. is_event_loop_running: Arc<AtomicBool>,
  617. proxy: EventLoopProxy<Message>,
  618. window_event_listeners: WindowEventListeners,
  619. #[cfg(feature = "menu")]
  620. menu_event_listeners: MenuEventListeners,
  621. }
  622. /// The Tauri [`Dispatch`] for [`Wry`].
  623. #[derive(Clone)]
  624. pub struct WryDispatcher {
  625. window_id: WindowId,
  626. context: DispatcherContext,
  627. }
  628. impl Dispatch for WryDispatcher {
  629. type Runtime = Wry;
  630. type WindowBuilder = WindowBuilderWrapper;
  631. fn run_on_main_thread<F: FnOnce() + Send + 'static>(&self, f: F) -> Result<()> {
  632. self
  633. .context
  634. .proxy
  635. .send_event(Message::Task(Arc::new(Mutex::new(Some(Box::new(f))))))
  636. .map_err(|_| Error::FailedToSendMessage)
  637. }
  638. fn on_window_event<F: Fn(&WindowEvent) + Send + 'static>(&self, f: F) -> Uuid {
  639. let id = Uuid::new_v4();
  640. self
  641. .context
  642. .window_event_listeners
  643. .lock()
  644. .unwrap()
  645. .get(&self.window_id)
  646. .unwrap()
  647. .lock()
  648. .unwrap()
  649. .insert(id, Box::new(f));
  650. id
  651. }
  652. #[cfg(feature = "menu")]
  653. fn on_menu_event<F: Fn(&MenuEvent) + Send + 'static>(&self, f: F) -> Uuid {
  654. let id = Uuid::new_v4();
  655. self
  656. .context
  657. .menu_event_listeners
  658. .lock()
  659. .unwrap()
  660. .get(&self.window_id)
  661. .unwrap()
  662. .lock()
  663. .unwrap()
  664. .insert(id, Box::new(f));
  665. id
  666. }
  667. // Getters
  668. fn scale_factor(&self) -> Result<f64> {
  669. Ok(dispatcher_getter!(self, WindowMessage::ScaleFactor))
  670. }
  671. fn inner_position(&self) -> Result<PhysicalPosition<i32>> {
  672. dispatcher_getter!(self, WindowMessage::InnerPosition)
  673. }
  674. fn outer_position(&self) -> Result<PhysicalPosition<i32>> {
  675. dispatcher_getter!(self, WindowMessage::OuterPosition)
  676. }
  677. fn inner_size(&self) -> Result<PhysicalSize<u32>> {
  678. Ok(dispatcher_getter!(self, WindowMessage::InnerSize))
  679. }
  680. fn outer_size(&self) -> Result<PhysicalSize<u32>> {
  681. Ok(dispatcher_getter!(self, WindowMessage::OuterSize))
  682. }
  683. fn is_fullscreen(&self) -> Result<bool> {
  684. Ok(dispatcher_getter!(self, WindowMessage::IsFullscreen))
  685. }
  686. fn is_maximized(&self) -> Result<bool> {
  687. Ok(dispatcher_getter!(self, WindowMessage::IsMaximized))
  688. }
  689. /// Gets the window’s current decoration state.
  690. fn is_decorated(&self) -> Result<bool> {
  691. Ok(dispatcher_getter!(self, WindowMessage::IsDecorated))
  692. }
  693. /// Gets the window’s current resizable state.
  694. fn is_resizable(&self) -> Result<bool> {
  695. Ok(dispatcher_getter!(self, WindowMessage::IsResizable))
  696. }
  697. fn is_visible(&self) -> Result<bool> {
  698. Ok(dispatcher_getter!(self, WindowMessage::IsVisible))
  699. }
  700. #[cfg(feature = "menu")]
  701. fn is_menu_visible(&self) -> Result<bool> {
  702. Ok(dispatcher_getter!(self, WindowMessage::IsMenuVisible))
  703. }
  704. fn current_monitor(&self) -> Result<Option<Monitor>> {
  705. Ok(
  706. dispatcher_getter!(self, WindowMessage::CurrentMonitor)
  707. .map(|m| MonitorHandleWrapper(m).into()),
  708. )
  709. }
  710. fn primary_monitor(&self) -> Result<Option<Monitor>> {
  711. Ok(
  712. dispatcher_getter!(self, WindowMessage::PrimaryMonitor)
  713. .map(|m| MonitorHandleWrapper(m).into()),
  714. )
  715. }
  716. fn available_monitors(&self) -> Result<Vec<Monitor>> {
  717. Ok(
  718. dispatcher_getter!(self, WindowMessage::AvailableMonitors)
  719. .into_iter()
  720. .map(|m| MonitorHandleWrapper(m).into())
  721. .collect(),
  722. )
  723. }
  724. #[cfg(windows)]
  725. fn hwnd(&self) -> Result<*mut std::ffi::c_void> {
  726. Ok(dispatcher_getter!(self, WindowMessage::Hwnd).0)
  727. }
  728. // Setters
  729. fn center(&self) -> Result<()> {
  730. dispatcher_getter!(self, WindowMessage::Center)
  731. }
  732. fn print(&self) -> Result<()> {
  733. self
  734. .context
  735. .proxy
  736. .send_event(Message::Webview(self.window_id, WebviewMessage::Print))
  737. .map_err(|_| Error::FailedToSendMessage)
  738. }
  739. fn request_user_attention(&self, request_type: Option<UserAttentionType>) -> Result<()> {
  740. self
  741. .context
  742. .proxy
  743. .send_event(Message::Window(
  744. self.window_id,
  745. WindowMessage::RequestUserAttention(request_type.map(Into::into)),
  746. ))
  747. .map_err(|_| Error::FailedToSendMessage)
  748. }
  749. // Creates a window by dispatching a message to the event loop.
  750. // Note that this must be called from a separate thread, otherwise the channel will introduce a deadlock.
  751. fn create_window<P: Params<Runtime = Self::Runtime>>(
  752. &mut self,
  753. pending: PendingWindow<P>,
  754. ) -> Result<DetachedWindow<P>> {
  755. let (tx, rx) = channel();
  756. let label = pending.label.clone();
  757. let context = self.context.clone();
  758. self
  759. .context
  760. .proxy
  761. .send_event(Message::CreateWebview(
  762. Arc::new(Mutex::new(Some(Box::new(
  763. move |event_loop, web_context| create_webview(event_loop, web_context, context, pending),
  764. )))),
  765. tx,
  766. ))
  767. .map_err(|_| Error::FailedToSendMessage)?;
  768. let window_id = rx.recv().unwrap();
  769. let dispatcher = WryDispatcher {
  770. window_id,
  771. context: self.context.clone(),
  772. };
  773. Ok(DetachedWindow { label, dispatcher })
  774. }
  775. fn set_resizable(&self, resizable: bool) -> Result<()> {
  776. self
  777. .context
  778. .proxy
  779. .send_event(Message::Window(
  780. self.window_id,
  781. WindowMessage::SetResizable(resizable),
  782. ))
  783. .map_err(|_| Error::FailedToSendMessage)
  784. }
  785. fn set_title<S: Into<String>>(&self, title: S) -> Result<()> {
  786. self
  787. .context
  788. .proxy
  789. .send_event(Message::Window(
  790. self.window_id,
  791. WindowMessage::SetTitle(title.into()),
  792. ))
  793. .map_err(|_| Error::FailedToSendMessage)
  794. }
  795. fn maximize(&self) -> Result<()> {
  796. self
  797. .context
  798. .proxy
  799. .send_event(Message::Window(self.window_id, WindowMessage::Maximize))
  800. .map_err(|_| Error::FailedToSendMessage)
  801. }
  802. fn unmaximize(&self) -> Result<()> {
  803. self
  804. .context
  805. .proxy
  806. .send_event(Message::Window(self.window_id, WindowMessage::Unmaximize))
  807. .map_err(|_| Error::FailedToSendMessage)
  808. }
  809. fn minimize(&self) -> Result<()> {
  810. self
  811. .context
  812. .proxy
  813. .send_event(Message::Window(self.window_id, WindowMessage::Minimize))
  814. .map_err(|_| Error::FailedToSendMessage)
  815. }
  816. fn unminimize(&self) -> Result<()> {
  817. self
  818. .context
  819. .proxy
  820. .send_event(Message::Window(self.window_id, WindowMessage::Unminimize))
  821. .map_err(|_| Error::FailedToSendMessage)
  822. }
  823. #[cfg(feature = "menu")]
  824. fn show_menu(&self) -> Result<()> {
  825. self
  826. .context
  827. .proxy
  828. .send_event(Message::Window(self.window_id, WindowMessage::ShowMenu))
  829. .map_err(|_| Error::FailedToSendMessage)
  830. }
  831. #[cfg(feature = "menu")]
  832. fn hide_menu(&self) -> Result<()> {
  833. self
  834. .context
  835. .proxy
  836. .send_event(Message::Window(self.window_id, WindowMessage::HideMenu))
  837. .map_err(|_| Error::FailedToSendMessage)
  838. }
  839. fn show(&self) -> Result<()> {
  840. self
  841. .context
  842. .proxy
  843. .send_event(Message::Window(self.window_id, WindowMessage::Show))
  844. .map_err(|_| Error::FailedToSendMessage)
  845. }
  846. fn hide(&self) -> Result<()> {
  847. self
  848. .context
  849. .proxy
  850. .send_event(Message::Window(self.window_id, WindowMessage::Hide))
  851. .map_err(|_| Error::FailedToSendMessage)
  852. }
  853. fn close(&self) -> Result<()> {
  854. self
  855. .context
  856. .proxy
  857. .send_event(Message::Window(self.window_id, WindowMessage::Close))
  858. .map_err(|_| Error::FailedToSendMessage)
  859. }
  860. fn set_decorations(&self, decorations: bool) -> Result<()> {
  861. self
  862. .context
  863. .proxy
  864. .send_event(Message::Window(
  865. self.window_id,
  866. WindowMessage::SetDecorations(decorations),
  867. ))
  868. .map_err(|_| Error::FailedToSendMessage)
  869. }
  870. fn set_always_on_top(&self, always_on_top: bool) -> Result<()> {
  871. self
  872. .context
  873. .proxy
  874. .send_event(Message::Window(
  875. self.window_id,
  876. WindowMessage::SetAlwaysOnTop(always_on_top),
  877. ))
  878. .map_err(|_| Error::FailedToSendMessage)
  879. }
  880. fn set_size(&self, size: Size) -> Result<()> {
  881. self
  882. .context
  883. .proxy
  884. .send_event(Message::Window(
  885. self.window_id,
  886. WindowMessage::SetSize(size),
  887. ))
  888. .map_err(|_| Error::FailedToSendMessage)
  889. }
  890. fn set_min_size(&self, size: Option<Size>) -> Result<()> {
  891. self
  892. .context
  893. .proxy
  894. .send_event(Message::Window(
  895. self.window_id,
  896. WindowMessage::SetMinSize(size),
  897. ))
  898. .map_err(|_| Error::FailedToSendMessage)
  899. }
  900. fn set_max_size(&self, size: Option<Size>) -> Result<()> {
  901. self
  902. .context
  903. .proxy
  904. .send_event(Message::Window(
  905. self.window_id,
  906. WindowMessage::SetMaxSize(size),
  907. ))
  908. .map_err(|_| Error::FailedToSendMessage)
  909. }
  910. fn set_position(&self, position: Position) -> Result<()> {
  911. self
  912. .context
  913. .proxy
  914. .send_event(Message::Window(
  915. self.window_id,
  916. WindowMessage::SetPosition(position),
  917. ))
  918. .map_err(|_| Error::FailedToSendMessage)
  919. }
  920. fn set_fullscreen(&self, fullscreen: bool) -> Result<()> {
  921. self
  922. .context
  923. .proxy
  924. .send_event(Message::Window(
  925. self.window_id,
  926. WindowMessage::SetFullscreen(fullscreen),
  927. ))
  928. .map_err(|_| Error::FailedToSendMessage)
  929. }
  930. fn set_focus(&self) -> Result<()> {
  931. self
  932. .context
  933. .proxy
  934. .send_event(Message::Window(self.window_id, WindowMessage::SetFocus))
  935. .map_err(|_| Error::FailedToSendMessage)
  936. }
  937. fn set_icon(&self, icon: Icon) -> Result<()> {
  938. self
  939. .context
  940. .proxy
  941. .send_event(Message::Window(
  942. self.window_id,
  943. WindowMessage::SetIcon(WryIcon::try_from(icon)?.0),
  944. ))
  945. .map_err(|_| Error::FailedToSendMessage)
  946. }
  947. fn set_skip_taskbar(&self, skip: bool) -> Result<()> {
  948. self
  949. .context
  950. .proxy
  951. .send_event(Message::Window(
  952. self.window_id,
  953. WindowMessage::SetSkipTaskbar(skip),
  954. ))
  955. .map_err(|_| Error::FailedToSendMessage)
  956. }
  957. fn start_dragging(&self) -> Result<()> {
  958. self
  959. .context
  960. .proxy
  961. .send_event(Message::Window(self.window_id, WindowMessage::DragWindow))
  962. .map_err(|_| Error::FailedToSendMessage)
  963. }
  964. fn eval_script<S: Into<String>>(&self, script: S) -> Result<()> {
  965. self
  966. .context
  967. .proxy
  968. .send_event(Message::Webview(
  969. self.window_id,
  970. WebviewMessage::EvaluateScript(script.into()),
  971. ))
  972. .map_err(|_| Error::FailedToSendMessage)
  973. }
  974. #[cfg(feature = "menu")]
  975. fn update_menu_item(&self, id: u16, update: menu::MenuUpdate) -> Result<()> {
  976. self
  977. .context
  978. .proxy
  979. .send_event(Message::Window(
  980. self.window_id,
  981. WindowMessage::UpdateMenuItem(id, update),
  982. ))
  983. .map_err(|_| Error::FailedToSendMessage)
  984. }
  985. }
  986. #[cfg(feature = "system-tray")]
  987. #[derive(Clone, Default)]
  988. struct TrayContext {
  989. tray: Arc<Mutex<Option<Arc<Mutex<WrySystemTray>>>>>,
  990. listeners: SystemTrayEventListeners,
  991. items: SystemTrayItems,
  992. }
  993. struct WebviewWrapper {
  994. label: String,
  995. inner: WebView,
  996. #[cfg(feature = "menu")]
  997. menu_items: HashMap<u16, WryCustomMenuItem>,
  998. #[cfg(feature = "menu")]
  999. is_menu_visible: AtomicBool,
  1000. }
  1001. /// A Tauri [`Runtime`] wrapper around wry.
  1002. pub struct Wry {
  1003. main_thread_id: ThreadId,
  1004. global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
  1005. global_shortcut_manager_handle: GlobalShortcutManagerHandle,
  1006. clipboard_manager: Arc<Mutex<Clipboard>>,
  1007. clipboard_manager_handle: ClipboardManagerWrapper,
  1008. is_event_loop_running: Arc<AtomicBool>,
  1009. event_loop: EventLoop<Message>,
  1010. web_context: WebContextStore,
  1011. webviews: Arc<Mutex<HashMap<WindowId, WebviewWrapper>>>,
  1012. window_event_listeners: WindowEventListeners,
  1013. #[cfg(feature = "menu")]
  1014. menu_event_listeners: MenuEventListeners,
  1015. #[cfg(feature = "system-tray")]
  1016. tray_context: TrayContext,
  1017. }
  1018. /// A handle to the Wry runtime.
  1019. #[derive(Clone)]
  1020. pub struct WryHandle {
  1021. dispatcher_context: DispatcherContext,
  1022. }
  1023. impl RuntimeHandle for WryHandle {
  1024. type Runtime = Wry;
  1025. // Creates a window by dispatching a message to the event loop.
  1026. // Note that this must be called from a separate thread, otherwise the channel will introduce a deadlock.
  1027. fn create_window<P: Params<Runtime = Self::Runtime>>(
  1028. &self,
  1029. pending: PendingWindow<P>,
  1030. ) -> Result<DetachedWindow<P>> {
  1031. let (tx, rx) = channel();
  1032. let label = pending.label.clone();
  1033. let dispatcher_context = self.dispatcher_context.clone();
  1034. self
  1035. .dispatcher_context
  1036. .proxy
  1037. .send_event(Message::CreateWebview(
  1038. Arc::new(Mutex::new(Some(Box::new(
  1039. move |event_loop, web_context| {
  1040. create_webview(event_loop, web_context, dispatcher_context, pending)
  1041. },
  1042. )))),
  1043. tx,
  1044. ))
  1045. .map_err(|_| Error::FailedToSendMessage)?;
  1046. let window_id = rx.recv().unwrap();
  1047. let dispatcher = WryDispatcher {
  1048. window_id,
  1049. context: self.dispatcher_context.clone(),
  1050. };
  1051. Ok(DetachedWindow { label, dispatcher })
  1052. }
  1053. #[cfg(all(windows, feature = "system-tray"))]
  1054. fn remove_system_tray(&self) -> Result<()> {
  1055. self
  1056. .dispatcher_context
  1057. .proxy
  1058. .send_event(Message::Tray(TrayMessage::Remove))
  1059. .map_err(|_| Error::FailedToSendMessage)
  1060. }
  1061. }
  1062. impl Runtime for Wry {
  1063. type Dispatcher = WryDispatcher;
  1064. type Handle = WryHandle;
  1065. type GlobalShortcutManager = GlobalShortcutManagerHandle;
  1066. type ClipboardManager = ClipboardManagerWrapper;
  1067. #[cfg(feature = "system-tray")]
  1068. type TrayHandler = SystemTrayHandle;
  1069. fn new() -> Result<Self> {
  1070. let event_loop = EventLoop::<Message>::with_user_event();
  1071. let proxy = event_loop.create_proxy();
  1072. let main_thread_id = current_thread().id();
  1073. let is_event_loop_running = Arc::new(AtomicBool::default());
  1074. let event_loop_context = EventLoopContext {
  1075. main_thread_id,
  1076. is_event_loop_running: is_event_loop_running.clone(),
  1077. proxy,
  1078. };
  1079. let global_shortcut_manager = WryShortcutManager::new(&event_loop);
  1080. let global_shortcut_listeners = GlobalShortcutListeners::default();
  1081. let clipboard_manager = Clipboard::new();
  1082. let clipboard_manager_handle = ClipboardManagerWrapper {
  1083. context: event_loop_context.clone(),
  1084. };
  1085. Ok(Self {
  1086. main_thread_id,
  1087. global_shortcut_manager: Arc::new(Mutex::new(global_shortcut_manager)),
  1088. global_shortcut_manager_handle: GlobalShortcutManagerHandle {
  1089. context: event_loop_context,
  1090. shortcuts: Default::default(),
  1091. listeners: global_shortcut_listeners,
  1092. },
  1093. clipboard_manager: Arc::new(Mutex::new(clipboard_manager)),
  1094. clipboard_manager_handle,
  1095. is_event_loop_running,
  1096. event_loop,
  1097. web_context: Default::default(),
  1098. webviews: Default::default(),
  1099. window_event_listeners: Default::default(),
  1100. #[cfg(feature = "menu")]
  1101. menu_event_listeners: Default::default(),
  1102. #[cfg(feature = "system-tray")]
  1103. tray_context: Default::default(),
  1104. })
  1105. }
  1106. fn handle(&self) -> Self::Handle {
  1107. WryHandle {
  1108. dispatcher_context: DispatcherContext {
  1109. main_thread_id: self.main_thread_id,
  1110. is_event_loop_running: self.is_event_loop_running.clone(),
  1111. proxy: self.event_loop.create_proxy(),
  1112. window_event_listeners: self.window_event_listeners.clone(),
  1113. #[cfg(feature = "menu")]
  1114. menu_event_listeners: self.menu_event_listeners.clone(),
  1115. },
  1116. }
  1117. }
  1118. fn global_shortcut_manager(&self) -> Self::GlobalShortcutManager {
  1119. self.global_shortcut_manager_handle.clone()
  1120. }
  1121. fn clipboard_manager(&self) -> Self::ClipboardManager {
  1122. self.clipboard_manager_handle.clone()
  1123. }
  1124. fn create_window<P: Params<Runtime = Self>>(
  1125. &self,
  1126. pending: PendingWindow<P>,
  1127. ) -> Result<DetachedWindow<P>> {
  1128. let label = pending.label.clone();
  1129. let proxy = self.event_loop.create_proxy();
  1130. let webview = create_webview(
  1131. &self.event_loop,
  1132. &self.web_context,
  1133. DispatcherContext {
  1134. main_thread_id: self.main_thread_id,
  1135. is_event_loop_running: self.is_event_loop_running.clone(),
  1136. proxy: proxy.clone(),
  1137. window_event_listeners: self.window_event_listeners.clone(),
  1138. #[cfg(feature = "menu")]
  1139. menu_event_listeners: self.menu_event_listeners.clone(),
  1140. },
  1141. pending,
  1142. )?;
  1143. let dispatcher = WryDispatcher {
  1144. window_id: webview.inner.window().id(),
  1145. context: DispatcherContext {
  1146. main_thread_id: self.main_thread_id,
  1147. is_event_loop_running: self.is_event_loop_running.clone(),
  1148. proxy,
  1149. window_event_listeners: self.window_event_listeners.clone(),
  1150. #[cfg(feature = "menu")]
  1151. menu_event_listeners: self.menu_event_listeners.clone(),
  1152. },
  1153. };
  1154. self
  1155. .webviews
  1156. .lock()
  1157. .unwrap()
  1158. .insert(webview.inner.window().id(), webview);
  1159. Ok(DetachedWindow { label, dispatcher })
  1160. }
  1161. #[cfg(feature = "system-tray")]
  1162. fn system_tray<I: MenuId>(&self, system_tray: SystemTray<I>) -> Result<Self::TrayHandler> {
  1163. let icon = system_tray
  1164. .icon
  1165. .expect("tray icon not set")
  1166. .into_tray_icon();
  1167. let mut items = HashMap::new();
  1168. let tray = SystemTrayBuilder::new(
  1169. icon,
  1170. system_tray
  1171. .menu
  1172. .map(|menu| to_wry_context_menu(&mut items, menu)),
  1173. )
  1174. .build(&self.event_loop)
  1175. .map_err(|e| Error::SystemTray(Box::new(e)))?;
  1176. *self.tray_context.items.lock().unwrap() = items;
  1177. *self.tray_context.tray.lock().unwrap() = Some(Arc::new(Mutex::new(tray)));
  1178. Ok(SystemTrayHandle {
  1179. proxy: self.event_loop.create_proxy(),
  1180. })
  1181. }
  1182. #[cfg(feature = "system-tray")]
  1183. fn on_system_tray_event<F: Fn(&SystemTrayEvent) + Send + 'static>(&mut self, f: F) -> Uuid {
  1184. let id = Uuid::new_v4();
  1185. self
  1186. .tray_context
  1187. .listeners
  1188. .lock()
  1189. .unwrap()
  1190. .insert(id, Box::new(f));
  1191. id
  1192. }
  1193. #[cfg(any(target_os = "windows", target_os = "macos"))]
  1194. fn run_iteration<F: Fn(RunEvent) + 'static>(&mut self, callback: F) -> RunIteration {
  1195. use wry::application::platform::run_return::EventLoopExtRunReturn;
  1196. let webviews = self.webviews.clone();
  1197. let web_context = &self.web_context;
  1198. let window_event_listeners = self.window_event_listeners.clone();
  1199. #[cfg(feature = "menu")]
  1200. let menu_event_listeners = self.menu_event_listeners.clone();
  1201. #[cfg(feature = "system-tray")]
  1202. let tray_context = self.tray_context.clone();
  1203. let global_shortcut_manager = self.global_shortcut_manager.clone();
  1204. let global_shortcut_manager_handle = self.global_shortcut_manager_handle.clone();
  1205. let clipboard_manager = self.clipboard_manager.clone();
  1206. let mut iteration = RunIteration::default();
  1207. self.is_event_loop_running.store(true, Ordering::Relaxed);
  1208. self
  1209. .event_loop
  1210. .run_return(|event, event_loop, control_flow| {
  1211. if let Event::MainEventsCleared = &event {
  1212. *control_flow = ControlFlow::Exit;
  1213. }
  1214. iteration = handle_event_loop(
  1215. event,
  1216. event_loop,
  1217. control_flow,
  1218. EventLoopIterationContext {
  1219. callback: &callback,
  1220. webviews: webviews.lock().expect("poisoned webview collection"),
  1221. window_event_listeners: window_event_listeners.clone(),
  1222. global_shortcut_manager: global_shortcut_manager.clone(),
  1223. global_shortcut_manager_handle: global_shortcut_manager_handle.clone(),
  1224. clipboard_manager: clipboard_manager.clone(),
  1225. #[cfg(feature = "menu")]
  1226. menu_event_listeners: menu_event_listeners.clone(),
  1227. #[cfg(feature = "system-tray")]
  1228. tray_context: tray_context.clone(),
  1229. },
  1230. web_context,
  1231. );
  1232. });
  1233. self.is_event_loop_running.store(false, Ordering::Relaxed);
  1234. iteration
  1235. }
  1236. fn run<F: Fn(RunEvent) + 'static>(self, callback: F) {
  1237. self.is_event_loop_running.store(true, Ordering::Relaxed);
  1238. let webviews = self.webviews.clone();
  1239. let web_context = self.web_context;
  1240. let window_event_listeners = self.window_event_listeners.clone();
  1241. #[cfg(feature = "menu")]
  1242. let menu_event_listeners = self.menu_event_listeners.clone();
  1243. #[cfg(feature = "system-tray")]
  1244. let tray_context = self.tray_context;
  1245. let global_shortcut_manager = self.global_shortcut_manager.clone();
  1246. let global_shortcut_manager_handle = self.global_shortcut_manager_handle.clone();
  1247. let clipboard_manager = self.clipboard_manager.clone();
  1248. self.event_loop.run(move |event, event_loop, control_flow| {
  1249. handle_event_loop(
  1250. event,
  1251. event_loop,
  1252. control_flow,
  1253. EventLoopIterationContext {
  1254. callback: &callback,
  1255. webviews: webviews.lock().expect("poisoned webview collection"),
  1256. window_event_listeners: window_event_listeners.clone(),
  1257. global_shortcut_manager: global_shortcut_manager.clone(),
  1258. global_shortcut_manager_handle: global_shortcut_manager_handle.clone(),
  1259. clipboard_manager: clipboard_manager.clone(),
  1260. #[cfg(feature = "menu")]
  1261. menu_event_listeners: menu_event_listeners.clone(),
  1262. #[cfg(feature = "system-tray")]
  1263. tray_context: tray_context.clone(),
  1264. },
  1265. &web_context,
  1266. );
  1267. })
  1268. }
  1269. }
  1270. struct EventLoopIterationContext<'a> {
  1271. callback: &'a (dyn Fn(RunEvent) + 'static),
  1272. webviews: MutexGuard<'a, HashMap<WindowId, WebviewWrapper>>,
  1273. window_event_listeners: WindowEventListeners,
  1274. global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
  1275. global_shortcut_manager_handle: GlobalShortcutManagerHandle,
  1276. clipboard_manager: Arc<Mutex<Clipboard>>,
  1277. #[cfg(feature = "menu")]
  1278. menu_event_listeners: MenuEventListeners,
  1279. #[cfg(feature = "system-tray")]
  1280. tray_context: TrayContext,
  1281. }
  1282. fn handle_event_loop(
  1283. event: Event<Message>,
  1284. event_loop: &EventLoopWindowTarget<Message>,
  1285. control_flow: &mut ControlFlow,
  1286. context: EventLoopIterationContext<'_>,
  1287. web_context: &WebContextStore,
  1288. ) -> RunIteration {
  1289. let EventLoopIterationContext {
  1290. callback,
  1291. mut webviews,
  1292. window_event_listeners,
  1293. global_shortcut_manager,
  1294. global_shortcut_manager_handle,
  1295. clipboard_manager,
  1296. #[cfg(feature = "menu")]
  1297. menu_event_listeners,
  1298. #[cfg(feature = "system-tray")]
  1299. tray_context,
  1300. } = context;
  1301. *control_flow = ControlFlow::Wait;
  1302. match event {
  1303. Event::GlobalShortcutEvent(accelerator_id) => {
  1304. for (id, handler) in &*global_shortcut_manager_handle.listeners.lock().unwrap() {
  1305. if accelerator_id == *id {
  1306. handler();
  1307. }
  1308. }
  1309. }
  1310. #[cfg(feature = "menu")]
  1311. Event::MenuEvent {
  1312. window_id,
  1313. menu_id,
  1314. origin: MenuType::MenuBar,
  1315. } => {
  1316. let window_id = window_id.unwrap(); // always Some on MenuBar event
  1317. let event = MenuEvent {
  1318. menu_item_id: menu_id.0,
  1319. };
  1320. let listeners = menu_event_listeners.lock().unwrap();
  1321. let window_menu_event_listeners = listeners.get(&window_id).cloned().unwrap_or_default();
  1322. for handler in window_menu_event_listeners.lock().unwrap().values() {
  1323. handler(&event);
  1324. }
  1325. }
  1326. #[cfg(feature = "system-tray")]
  1327. Event::MenuEvent {
  1328. window_id: _,
  1329. menu_id,
  1330. origin: MenuType::ContextMenu,
  1331. } => {
  1332. let event = SystemTrayEvent::MenuItemClick(menu_id.0);
  1333. for handler in tray_context.listeners.lock().unwrap().values() {
  1334. handler(&event);
  1335. }
  1336. }
  1337. #[cfg(feature = "system-tray")]
  1338. Event::TrayEvent {
  1339. bounds,
  1340. event,
  1341. position: _cursor_position,
  1342. } => {
  1343. let (position, size) = (
  1344. PhysicalPositionWrapper(bounds.position).into(),
  1345. PhysicalSizeWrapper(bounds.size).into(),
  1346. );
  1347. let event = match event {
  1348. TrayEvent::LeftClick => SystemTrayEvent::LeftClick { position, size },
  1349. TrayEvent::RightClick => SystemTrayEvent::RightClick { position, size },
  1350. TrayEvent::DoubleClick => SystemTrayEvent::DoubleClick { position, size },
  1351. };
  1352. for handler in tray_context.listeners.lock().unwrap().values() {
  1353. handler(&event);
  1354. }
  1355. }
  1356. Event::WindowEvent { event, window_id } => {
  1357. if let Some(event) = WindowEventWrapper::from(&event).0 {
  1358. for handler in window_event_listeners
  1359. .lock()
  1360. .unwrap()
  1361. .get(&window_id)
  1362. .unwrap()
  1363. .lock()
  1364. .unwrap()
  1365. .values()
  1366. {
  1367. handler(&event);
  1368. }
  1369. }
  1370. match event {
  1371. WryWindowEvent::CloseRequested => {
  1372. on_window_close(
  1373. callback,
  1374. window_id,
  1375. &mut webviews,
  1376. control_flow,
  1377. #[cfg(feature = "menu")]
  1378. menu_event_listeners.clone(),
  1379. );
  1380. }
  1381. WryWindowEvent::Resized(_) => {
  1382. if let Err(e) = webviews[&window_id].inner.resize() {
  1383. eprintln!("{}", e);
  1384. }
  1385. }
  1386. _ => {}
  1387. }
  1388. }
  1389. Event::UserEvent(message) => match message {
  1390. Message::Task(task) => {
  1391. if let Some(task) = task.lock().unwrap().take() {
  1392. task();
  1393. }
  1394. }
  1395. Message::Window(id, window_message) => {
  1396. if let Some(webview) = webviews.get_mut(&id) {
  1397. let window = webview.inner.window();
  1398. match window_message {
  1399. // Getters
  1400. WindowMessage::ScaleFactor(tx) => tx.send(window.scale_factor()).unwrap(),
  1401. WindowMessage::InnerPosition(tx) => tx
  1402. .send(
  1403. window
  1404. .inner_position()
  1405. .map(|p| PhysicalPositionWrapper(p).into())
  1406. .map_err(|_| Error::FailedToSendMessage),
  1407. )
  1408. .unwrap(),
  1409. WindowMessage::OuterPosition(tx) => tx
  1410. .send(
  1411. window
  1412. .outer_position()
  1413. .map(|p| PhysicalPositionWrapper(p).into())
  1414. .map_err(|_| Error::FailedToSendMessage),
  1415. )
  1416. .unwrap(),
  1417. WindowMessage::InnerSize(tx) => tx
  1418. .send(PhysicalSizeWrapper(window.inner_size()).into())
  1419. .unwrap(),
  1420. WindowMessage::OuterSize(tx) => tx
  1421. .send(PhysicalSizeWrapper(window.outer_size()).into())
  1422. .unwrap(),
  1423. WindowMessage::IsFullscreen(tx) => tx.send(window.fullscreen().is_some()).unwrap(),
  1424. WindowMessage::IsMaximized(tx) => tx.send(window.is_maximized()).unwrap(),
  1425. WindowMessage::IsDecorated(tx) => tx.send(window.is_decorated()).unwrap(),
  1426. WindowMessage::IsResizable(tx) => tx.send(window.is_resizable()).unwrap(),
  1427. WindowMessage::IsVisible(tx) => tx.send(window.is_visible()).unwrap(),
  1428. #[cfg(feature = "menu")]
  1429. WindowMessage::IsMenuVisible(tx) => tx
  1430. .send(webview.is_menu_visible.load(Ordering::Relaxed))
  1431. .unwrap(),
  1432. WindowMessage::CurrentMonitor(tx) => tx.send(window.current_monitor()).unwrap(),
  1433. WindowMessage::PrimaryMonitor(tx) => tx.send(window.primary_monitor()).unwrap(),
  1434. WindowMessage::AvailableMonitors(tx) => {
  1435. tx.send(window.available_monitors().collect()).unwrap()
  1436. }
  1437. #[cfg(windows)]
  1438. WindowMessage::Hwnd(tx) => {
  1439. use wry::application::platform::windows::WindowExtWindows;
  1440. tx.send(Hwnd(window.hwnd())).unwrap()
  1441. }
  1442. // Setters
  1443. WindowMessage::Center(tx) => {
  1444. tx.send(center_window(window)).unwrap();
  1445. }
  1446. WindowMessage::RequestUserAttention(request_type) => {
  1447. window.request_user_attention(request_type.map(|r| r.0));
  1448. }
  1449. WindowMessage::SetResizable(resizable) => window.set_resizable(resizable),
  1450. WindowMessage::SetTitle(title) => window.set_title(&title),
  1451. WindowMessage::Maximize => window.set_maximized(true),
  1452. WindowMessage::Unmaximize => window.set_maximized(false),
  1453. WindowMessage::Minimize => window.set_minimized(true),
  1454. WindowMessage::Unminimize => window.set_minimized(false),
  1455. #[cfg(feature = "menu")]
  1456. WindowMessage::ShowMenu => {
  1457. window.show_menu();
  1458. webview.is_menu_visible.store(true, Ordering::Relaxed);
  1459. }
  1460. #[cfg(feature = "menu")]
  1461. WindowMessage::HideMenu => {
  1462. window.hide_menu();
  1463. webview.is_menu_visible.store(false, Ordering::Relaxed);
  1464. }
  1465. WindowMessage::Show => window.set_visible(true),
  1466. WindowMessage::Hide => window.set_visible(false),
  1467. WindowMessage::Close => {
  1468. for handler in window_event_listeners
  1469. .lock()
  1470. .unwrap()
  1471. .get(&window.id())
  1472. .unwrap()
  1473. .lock()
  1474. .unwrap()
  1475. .values()
  1476. {
  1477. handler(&WindowEvent::CloseRequested);
  1478. }
  1479. on_window_close(
  1480. callback,
  1481. id,
  1482. &mut webviews,
  1483. control_flow,
  1484. #[cfg(feature = "menu")]
  1485. menu_event_listeners.clone(),
  1486. );
  1487. }
  1488. WindowMessage::SetDecorations(decorations) => window.set_decorations(decorations),
  1489. WindowMessage::SetAlwaysOnTop(always_on_top) => window.set_always_on_top(always_on_top),
  1490. WindowMessage::SetSize(size) => {
  1491. window.set_inner_size(SizeWrapper::from(size).0);
  1492. }
  1493. WindowMessage::SetMinSize(size) => {
  1494. window.set_min_inner_size(size.map(|s| SizeWrapper::from(s).0));
  1495. }
  1496. WindowMessage::SetMaxSize(size) => {
  1497. window.set_max_inner_size(size.map(|s| SizeWrapper::from(s).0));
  1498. }
  1499. WindowMessage::SetPosition(position) => {
  1500. window.set_outer_position(PositionWrapper::from(position).0)
  1501. }
  1502. WindowMessage::SetFullscreen(fullscreen) => {
  1503. if fullscreen {
  1504. window.set_fullscreen(Some(Fullscreen::Borderless(None)))
  1505. } else {
  1506. window.set_fullscreen(None)
  1507. }
  1508. }
  1509. WindowMessage::SetFocus => {
  1510. window.set_focus();
  1511. }
  1512. WindowMessage::SetIcon(icon) => {
  1513. window.set_window_icon(Some(icon));
  1514. }
  1515. WindowMessage::SetSkipTaskbar(skip) => {
  1516. window.set_skip_taskbar(skip);
  1517. }
  1518. WindowMessage::DragWindow => {
  1519. let _ = window.drag_window();
  1520. }
  1521. #[cfg(feature = "menu")]
  1522. WindowMessage::UpdateMenuItem(id, update) => {
  1523. let item = webview
  1524. .menu_items
  1525. .get_mut(&id)
  1526. .expect("menu item not found");
  1527. match update {
  1528. MenuUpdate::SetEnabled(enabled) => item.set_enabled(enabled),
  1529. MenuUpdate::SetTitle(title) => item.set_title(&title),
  1530. MenuUpdate::SetSelected(selected) => item.set_selected(selected),
  1531. #[cfg(target_os = "macos")]
  1532. MenuUpdate::SetNativeImage(image) => {
  1533. item.set_native_image(NativeImageWrapper::from(image).0)
  1534. }
  1535. }
  1536. }
  1537. }
  1538. }
  1539. }
  1540. Message::Webview(id, webview_message) => {
  1541. if let Some(webview) = webviews.get_mut(&id) {
  1542. match webview_message {
  1543. WebviewMessage::EvaluateScript(script) => {
  1544. if let Err(e) = webview.inner.evaluate_script(&script) {
  1545. eprintln!("{}", e);
  1546. }
  1547. }
  1548. WebviewMessage::Print => {
  1549. let _ = webview.inner.print();
  1550. }
  1551. }
  1552. }
  1553. }
  1554. Message::CreateWebview(handler, sender) => {
  1555. let handler = {
  1556. let mut lock = handler.lock().expect("poisoned create webview handler");
  1557. std::mem::take(&mut *lock).unwrap()
  1558. };
  1559. match handler(event_loop, web_context) {
  1560. Ok(webview) => {
  1561. let window_id = webview.inner.window().id();
  1562. webviews.insert(window_id, webview);
  1563. sender.send(window_id).unwrap();
  1564. }
  1565. Err(e) => {
  1566. eprintln!("{}", e);
  1567. }
  1568. }
  1569. }
  1570. #[cfg(feature = "system-tray")]
  1571. Message::Tray(tray_message) => match tray_message {
  1572. TrayMessage::UpdateItem(menu_id, update) => {
  1573. let mut tray = tray_context.items.as_ref().lock().unwrap();
  1574. let item = tray.get_mut(&menu_id).expect("menu item not found");
  1575. match update {
  1576. MenuUpdate::SetEnabled(enabled) => item.set_enabled(enabled),
  1577. MenuUpdate::SetTitle(title) => item.set_title(&title),
  1578. MenuUpdate::SetSelected(selected) => item.set_selected(selected),
  1579. #[cfg(target_os = "macos")]
  1580. MenuUpdate::SetNativeImage(image) => {
  1581. item.set_native_image(NativeImageWrapper::from(image).0)
  1582. }
  1583. }
  1584. }
  1585. TrayMessage::UpdateIcon(icon) => {
  1586. if let Some(tray) = &*tray_context.tray.lock().unwrap() {
  1587. tray.lock().unwrap().set_icon(icon.into_tray_icon());
  1588. }
  1589. }
  1590. #[cfg(windows)]
  1591. TrayMessage::Remove => {
  1592. if let Some(tray) = tray_context.tray.lock().unwrap().as_ref() {
  1593. use wry::application::platform::windows::SystemTrayExtWindows;
  1594. tray.lock().unwrap().remove();
  1595. }
  1596. }
  1597. },
  1598. Message::GlobalShortcut(message) => match message {
  1599. GlobalShortcutMessage::IsRegistered(accelerator, tx) => tx
  1600. .send(
  1601. global_shortcut_manager
  1602. .lock()
  1603. .unwrap()
  1604. .is_registered(&accelerator),
  1605. )
  1606. .unwrap(),
  1607. GlobalShortcutMessage::Register(accelerator, tx) => tx
  1608. .send(
  1609. global_shortcut_manager
  1610. .lock()
  1611. .unwrap()
  1612. .register(accelerator)
  1613. .map(GlobalShortcutWrapper)
  1614. .map_err(|e| Error::GlobalShortcut(Box::new(e))),
  1615. )
  1616. .unwrap(),
  1617. GlobalShortcutMessage::Unregister(shortcut, tx) => tx
  1618. .send(
  1619. global_shortcut_manager
  1620. .lock()
  1621. .unwrap()
  1622. .unregister(shortcut.0)
  1623. .map_err(|e| Error::GlobalShortcut(Box::new(e))),
  1624. )
  1625. .unwrap(),
  1626. GlobalShortcutMessage::UnregisterAll(tx) => tx
  1627. .send(
  1628. global_shortcut_manager
  1629. .lock()
  1630. .unwrap()
  1631. .unregister_all()
  1632. .map_err(|e| Error::GlobalShortcut(Box::new(e))),
  1633. )
  1634. .unwrap(),
  1635. },
  1636. Message::Clipboard(message) => match message {
  1637. ClipboardMessage::WriteText(text, tx) => {
  1638. clipboard_manager.lock().unwrap().write_text(text);
  1639. tx.send(()).unwrap();
  1640. }
  1641. ClipboardMessage::ReadText(tx) => tx
  1642. .send(clipboard_manager.lock().unwrap().read_text())
  1643. .unwrap(),
  1644. },
  1645. },
  1646. _ => (),
  1647. }
  1648. RunIteration {
  1649. webview_count: webviews.len(),
  1650. }
  1651. }
  1652. fn on_window_close<'a>(
  1653. callback: &'a (dyn Fn(RunEvent) + 'static),
  1654. window_id: WindowId,
  1655. webviews: &mut MutexGuard<'a, HashMap<WindowId, WebviewWrapper>>,
  1656. control_flow: &mut ControlFlow,
  1657. #[cfg(feature = "menu")] menu_event_listeners: MenuEventListeners,
  1658. ) {
  1659. if let Some(webview) = webviews.remove(&window_id) {
  1660. #[cfg(feature = "menu")]
  1661. menu_event_listeners.lock().unwrap().remove(&window_id);
  1662. callback(RunEvent::WindowClose(webview.label));
  1663. }
  1664. if webviews.is_empty() {
  1665. *control_flow = ControlFlow::Exit;
  1666. callback(RunEvent::Exit);
  1667. }
  1668. }
  1669. fn center_window(window: &Window) -> Result<()> {
  1670. if let Some(monitor) = window.current_monitor() {
  1671. let screen_size = monitor.size();
  1672. let window_size = window.inner_size();
  1673. let x = (screen_size.width - window_size.width) / 2;
  1674. let y = (screen_size.height - window_size.height) / 2;
  1675. window.set_outer_position(WryPhysicalPosition::new(x, y));
  1676. Ok(())
  1677. } else {
  1678. Err(Error::FailedToGetMonitor)
  1679. }
  1680. }
  1681. fn create_webview<P: Params<Runtime = Wry>>(
  1682. event_loop: &EventLoopWindowTarget<Message>,
  1683. web_context: &WebContextStore,
  1684. context: DispatcherContext,
  1685. pending: PendingWindow<P>,
  1686. ) -> Result<WebviewWrapper> {
  1687. #[allow(unused_mut)]
  1688. let PendingWindow {
  1689. webview_attributes,
  1690. mut window_builder,
  1691. rpc_handler,
  1692. file_drop_handler,
  1693. label,
  1694. url,
  1695. ..
  1696. } = pending;
  1697. let is_window_transparent = window_builder.inner.window.transparent;
  1698. #[cfg(feature = "menu")]
  1699. let menu_items = {
  1700. let mut menu_items = HashMap::new();
  1701. let menu = to_wry_menu(&mut menu_items, window_builder.menu);
  1702. window_builder.inner = window_builder.inner.with_menu(menu);
  1703. menu_items
  1704. };
  1705. let window = window_builder.inner.build(event_loop).unwrap();
  1706. context
  1707. .window_event_listeners
  1708. .lock()
  1709. .unwrap()
  1710. .insert(window.id(), WindowEventListenersMap::default());
  1711. #[cfg(feature = "menu")]
  1712. context
  1713. .menu_event_listeners
  1714. .lock()
  1715. .unwrap()
  1716. .insert(window.id(), WindowMenuEventListeners::default());
  1717. if window_builder.center {
  1718. let _ = center_window(&window);
  1719. }
  1720. let mut webview_builder = WebViewBuilder::new(window)
  1721. .map_err(|e| Error::CreateWebview(Box::new(e)))?
  1722. .with_url(&url)
  1723. .unwrap() // safe to unwrap because we validate the URL beforehand
  1724. .with_transparent(is_window_transparent);
  1725. if let Some(handler) = rpc_handler {
  1726. webview_builder =
  1727. webview_builder.with_rpc_handler(create_rpc_handler(context.clone(), label.clone(), handler));
  1728. }
  1729. if let Some(handler) = file_drop_handler {
  1730. webview_builder = webview_builder.with_file_drop_handler(create_file_drop_handler(
  1731. context,
  1732. label.clone(),
  1733. handler,
  1734. ));
  1735. }
  1736. for (scheme, protocol) in webview_attributes.uri_scheme_protocols {
  1737. webview_builder = webview_builder.with_custom_protocol(scheme, move |_window, url| {
  1738. protocol(url)
  1739. .map(|data| {
  1740. let mime_type = MimeType::parse(&data, url);
  1741. (data, mime_type)
  1742. })
  1743. .map_err(|_| wry::Error::InitScriptError)
  1744. });
  1745. }
  1746. for script in webview_attributes.initialization_scripts {
  1747. webview_builder = webview_builder.with_initialization_script(&script);
  1748. }
  1749. let mut web_context = web_context.lock().expect("poisoned WebContext store");
  1750. let is_first_context = web_context.is_empty();
  1751. let web_context = match web_context.entry(webview_attributes.data_directory) {
  1752. Occupied(occupied) => occupied.into_mut(),
  1753. Vacant(vacant) => {
  1754. let mut web_context = WebContext::new(vacant.key().clone());
  1755. web_context.set_allows_automation(match std::env::var("TAURI_AUTOMATION").as_deref() {
  1756. Ok("true") => is_first_context,
  1757. _ => false,
  1758. });
  1759. vacant.insert(web_context)
  1760. }
  1761. };
  1762. let webview = webview_builder
  1763. .with_web_context(web_context)
  1764. .build()
  1765. .map_err(|e| Error::CreateWebview(Box::new(e)))?;
  1766. Ok(WebviewWrapper {
  1767. label: format!("{}", label),
  1768. inner: webview,
  1769. #[cfg(feature = "menu")]
  1770. menu_items,
  1771. #[cfg(feature = "menu")]
  1772. is_menu_visible: AtomicBool::new(true),
  1773. })
  1774. }
  1775. /// Create a wry rpc handler from a tauri rpc handler.
  1776. fn create_rpc_handler<P: Params<Runtime = Wry>>(
  1777. context: DispatcherContext,
  1778. label: P::Label,
  1779. handler: WebviewRpcHandler<P>,
  1780. ) -> Box<dyn Fn(&Window, WryRpcRequest) -> Option<RpcResponse> + 'static> {
  1781. Box::new(move |window, request| {
  1782. handler(
  1783. DetachedWindow {
  1784. dispatcher: WryDispatcher {
  1785. window_id: window.id(),
  1786. context: context.clone(),
  1787. },
  1788. label: label.clone(),
  1789. },
  1790. RpcRequestWrapper(request).into(),
  1791. );
  1792. None
  1793. })
  1794. }
  1795. /// Create a wry file drop handler from a tauri file drop handler.
  1796. fn create_file_drop_handler<P: Params<Runtime = Wry>>(
  1797. context: DispatcherContext,
  1798. label: P::Label,
  1799. handler: FileDropHandler<P>,
  1800. ) -> Box<dyn Fn(&Window, WryFileDropEvent) -> bool + 'static> {
  1801. Box::new(move |window, event| {
  1802. handler(
  1803. FileDropEventWrapper(event).into(),
  1804. DetachedWindow {
  1805. dispatcher: WryDispatcher {
  1806. window_id: window.id(),
  1807. context: context.clone(),
  1808. },
  1809. label: label.clone(),
  1810. },
  1811. )
  1812. })
  1813. }