lib.rs 45 KB

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