lib.rs 80 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676
  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. http::{
  7. Request as HttpRequest, RequestParts as HttpRequestParts, Response as HttpResponse,
  8. ResponseParts as HttpResponseParts,
  9. },
  10. menu::{CustomMenuItem, Menu, MenuEntry, MenuHash, MenuId, MenuItem, MenuUpdate},
  11. monitor::Monitor,
  12. webview::{
  13. FileDropEvent, FileDropHandler, RpcRequest, WebviewRpcHandler, WindowBuilder, WindowBuilderBase,
  14. },
  15. window::{
  16. dpi::{LogicalPosition, LogicalSize, PhysicalPosition, PhysicalSize, Position, Size},
  17. DetachedWindow, PendingWindow, WindowEvent,
  18. },
  19. ClipboardManager, Dispatch, Error, ExitRequestedEventAction, GlobalShortcutManager, Icon, Result,
  20. RunEvent, RunIteration, Runtime, RuntimeHandle, UserAttentionType,
  21. };
  22. use tauri_runtime::window::MenuEvent;
  23. #[cfg(feature = "system-tray")]
  24. use tauri_runtime::{SystemTray, SystemTrayEvent};
  25. #[cfg(windows)]
  26. use webview2_com::{
  27. FocusChangedEventHandler, Windows::Win32::System::WinRT::EventRegistrationToken,
  28. };
  29. #[cfg(windows)]
  30. use windows::Win32::Foundation::HWND;
  31. #[cfg(all(feature = "system-tray", target_os = "macos"))]
  32. use wry::application::platform::macos::{SystemTrayBuilderExtMacOS, SystemTrayExtMacOS};
  33. #[cfg(target_os = "linux")]
  34. use wry::application::platform::unix::{WindowBuilderExtUnix, WindowExtUnix};
  35. #[cfg(windows)]
  36. use wry::application::platform::windows::{WindowBuilderExtWindows, WindowExtWindows};
  37. #[cfg(feature = "system-tray")]
  38. use wry::application::system_tray::{SystemTray as WrySystemTray, SystemTrayBuilder};
  39. use tauri_utils::config::WindowConfig;
  40. use uuid::Uuid;
  41. use wry::{
  42. application::{
  43. accelerator::{Accelerator, AcceleratorId},
  44. clipboard::Clipboard,
  45. dpi::{
  46. LogicalPosition as WryLogicalPosition, LogicalSize as WryLogicalSize,
  47. PhysicalPosition as WryPhysicalPosition, PhysicalSize as WryPhysicalSize,
  48. Position as WryPosition, Size as WrySize,
  49. },
  50. event::{Event, StartCause, WindowEvent as WryWindowEvent},
  51. event_loop::{ControlFlow, EventLoop, EventLoopProxy, EventLoopWindowTarget},
  52. global_shortcut::{GlobalShortcut, ShortcutManager as WryShortcutManager},
  53. menu::{
  54. CustomMenuItem as WryCustomMenuItem, MenuBar, MenuId as WryMenuId, MenuItem as WryMenuItem,
  55. MenuItemAttributes as WryMenuItemAttributes, MenuType,
  56. },
  57. monitor::MonitorHandle,
  58. window::{Fullscreen, Icon as WindowIcon, UserAttentionType as WryUserAttentionType},
  59. },
  60. http::{
  61. Request as WryHttpRequest, RequestParts as WryRequestParts, Response as WryHttpResponse,
  62. ResponseParts as WryResponseParts,
  63. },
  64. webview::{
  65. FileDropEvent as WryFileDropEvent, RpcRequest as WryRpcRequest, RpcResponse, WebContext,
  66. WebView, WebViewBuilder,
  67. },
  68. };
  69. pub use wry::application::window::{Window, WindowBuilder as WryWindowBuilder, WindowId};
  70. #[cfg(target_os = "windows")]
  71. use wry::webview::WebviewExtWindows;
  72. #[cfg(target_os = "macos")]
  73. use tauri_runtime::{menu::NativeImage, ActivationPolicy};
  74. #[cfg(target_os = "macos")]
  75. pub use wry::application::platform::macos::{
  76. ActivationPolicy as WryActivationPolicy, CustomMenuItemExtMacOS, EventLoopExtMacOS,
  77. NativeImage as WryNativeImage, WindowExtMacOS,
  78. };
  79. use std::{
  80. collections::{
  81. hash_map::Entry::{Occupied, Vacant},
  82. HashMap,
  83. },
  84. fmt,
  85. fs::read,
  86. ops::Deref,
  87. path::PathBuf,
  88. sync::{
  89. mpsc::{channel, Sender},
  90. Arc, Mutex, MutexGuard, Weak,
  91. },
  92. thread::{current as current_thread, ThreadId},
  93. };
  94. #[cfg(feature = "system-tray")]
  95. mod system_tray;
  96. #[cfg(feature = "system-tray")]
  97. use system_tray::*;
  98. type WebContextStore = Arc<Mutex<HashMap<Option<PathBuf>, WebContext>>>;
  99. // window
  100. type WindowEventHandler = Box<dyn Fn(&WindowEvent) + Send>;
  101. type WindowEventListenersMap = Arc<Mutex<HashMap<Uuid, WindowEventHandler>>>;
  102. type WindowEventListeners = Arc<Mutex<HashMap<WindowId, WindowEventListenersMap>>>;
  103. // global shortcut
  104. type GlobalShortcutListeners = Arc<Mutex<HashMap<AcceleratorId, Box<dyn Fn() + Send>>>>;
  105. // menu
  106. pub type MenuEventHandler = Box<dyn Fn(&MenuEvent) + Send>;
  107. pub type MenuEventListeners = Arc<Mutex<HashMap<WindowId, WindowMenuEventListeners>>>;
  108. pub type WindowMenuEventListeners = Arc<Mutex<HashMap<Uuid, MenuEventHandler>>>;
  109. macro_rules! getter {
  110. ($self: ident, $rx: expr, $message: expr) => {{
  111. send_user_message(&$self.context, $message)?;
  112. $rx.recv().unwrap()
  113. }};
  114. }
  115. macro_rules! window_getter {
  116. ($self: ident, $message: expr) => {{
  117. let (tx, rx) = channel();
  118. getter!($self, rx, Message::Window($self.window_id, $message(tx)))
  119. }};
  120. }
  121. fn send_user_message(context: &Context, message: Message) -> Result<()> {
  122. if current_thread().id() == context.main_thread_id {
  123. handle_user_message(
  124. &context.main_thread.window_target,
  125. message,
  126. UserMessageContext {
  127. window_event_listeners: &context.window_event_listeners,
  128. global_shortcut_manager: context.main_thread.global_shortcut_manager.clone(),
  129. clipboard_manager: context.main_thread.clipboard_manager.clone(),
  130. menu_event_listeners: &context.menu_event_listeners,
  131. windows: context.main_thread.windows.clone(),
  132. #[cfg(feature = "system-tray")]
  133. tray_context: &context.main_thread.tray_context,
  134. },
  135. &context.main_thread.web_context,
  136. );
  137. Ok(())
  138. } else {
  139. context
  140. .proxy
  141. .send_event(message)
  142. .map_err(|_| Error::FailedToSendMessage)
  143. }
  144. }
  145. #[derive(Clone)]
  146. struct Context {
  147. main_thread_id: ThreadId,
  148. proxy: EventLoopProxy<Message>,
  149. window_event_listeners: WindowEventListeners,
  150. menu_event_listeners: MenuEventListeners,
  151. main_thread: DispatcherMainThreadContext,
  152. }
  153. #[derive(Debug, Clone)]
  154. struct DispatcherMainThreadContext {
  155. window_target: EventLoopWindowTarget<Message>,
  156. web_context: WebContextStore,
  157. global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
  158. clipboard_manager: Arc<Mutex<Clipboard>>,
  159. windows: Arc<Mutex<HashMap<WindowId, WindowWrapper>>>,
  160. #[cfg(feature = "system-tray")]
  161. tray_context: TrayContext,
  162. }
  163. // the main thread context is only used on the main thread
  164. #[allow(clippy::non_send_fields_in_send_ty)]
  165. unsafe impl Send for DispatcherMainThreadContext {}
  166. impl fmt::Debug for Context {
  167. fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
  168. f.debug_struct("Context")
  169. .field("main_thread_id", &self.main_thread_id)
  170. .field("proxy", &self.proxy)
  171. .field("main_thread", &self.main_thread)
  172. .finish()
  173. }
  174. }
  175. struct HttpRequestPartsWrapper(HttpRequestParts);
  176. impl From<HttpRequestPartsWrapper> for HttpRequestParts {
  177. fn from(parts: HttpRequestPartsWrapper) -> Self {
  178. Self {
  179. method: parts.0.method,
  180. uri: parts.0.uri,
  181. headers: parts.0.headers,
  182. }
  183. }
  184. }
  185. impl From<HttpRequestParts> for HttpRequestPartsWrapper {
  186. fn from(request: HttpRequestParts) -> Self {
  187. Self(HttpRequestParts {
  188. method: request.method,
  189. uri: request.uri,
  190. headers: request.headers,
  191. })
  192. }
  193. }
  194. impl From<WryRequestParts> for HttpRequestPartsWrapper {
  195. fn from(request: WryRequestParts) -> Self {
  196. Self(HttpRequestParts {
  197. method: request.method,
  198. uri: request.uri,
  199. headers: request.headers,
  200. })
  201. }
  202. }
  203. struct HttpRequestWrapper(HttpRequest);
  204. impl From<&WryHttpRequest> for HttpRequestWrapper {
  205. fn from(req: &WryHttpRequest) -> Self {
  206. Self(HttpRequest {
  207. body: req.body.clone(),
  208. head: HttpRequestPartsWrapper::from(req.head.clone()).0,
  209. })
  210. }
  211. }
  212. // response
  213. struct HttpResponsePartsWrapper(WryResponseParts);
  214. impl From<HttpResponseParts> for HttpResponsePartsWrapper {
  215. fn from(response: HttpResponseParts) -> Self {
  216. Self(WryResponseParts {
  217. mimetype: response.mimetype,
  218. status: response.status,
  219. version: response.version,
  220. headers: response.headers,
  221. })
  222. }
  223. }
  224. struct HttpResponseWrapper(WryHttpResponse);
  225. impl From<HttpResponse> for HttpResponseWrapper {
  226. fn from(response: HttpResponse) -> Self {
  227. Self(WryHttpResponse {
  228. body: response.body,
  229. head: HttpResponsePartsWrapper::from(response.head).0,
  230. })
  231. }
  232. }
  233. pub struct MenuItemAttributesWrapper<'a>(pub WryMenuItemAttributes<'a>);
  234. impl<'a> From<&'a CustomMenuItem> for MenuItemAttributesWrapper<'a> {
  235. fn from(item: &'a CustomMenuItem) -> Self {
  236. let mut attributes = WryMenuItemAttributes::new(&item.title)
  237. .with_enabled(item.enabled)
  238. .with_selected(item.selected)
  239. .with_id(WryMenuId(item.id));
  240. if let Some(accelerator) = item.keyboard_accelerator.as_ref() {
  241. attributes = attributes.with_accelerators(&accelerator.parse().expect("invalid accelerator"));
  242. }
  243. Self(attributes)
  244. }
  245. }
  246. pub struct MenuItemWrapper(pub WryMenuItem);
  247. impl From<MenuItem> for MenuItemWrapper {
  248. fn from(item: MenuItem) -> Self {
  249. match item {
  250. MenuItem::About(v) => Self(WryMenuItem::About(v)),
  251. MenuItem::Hide => Self(WryMenuItem::Hide),
  252. MenuItem::Services => Self(WryMenuItem::Services),
  253. MenuItem::HideOthers => Self(WryMenuItem::HideOthers),
  254. MenuItem::ShowAll => Self(WryMenuItem::ShowAll),
  255. MenuItem::CloseWindow => Self(WryMenuItem::CloseWindow),
  256. MenuItem::Quit => Self(WryMenuItem::Quit),
  257. MenuItem::Copy => Self(WryMenuItem::Copy),
  258. MenuItem::Cut => Self(WryMenuItem::Cut),
  259. MenuItem::Undo => Self(WryMenuItem::Undo),
  260. MenuItem::Redo => Self(WryMenuItem::Redo),
  261. MenuItem::SelectAll => Self(WryMenuItem::SelectAll),
  262. MenuItem::Paste => Self(WryMenuItem::Paste),
  263. MenuItem::EnterFullScreen => Self(WryMenuItem::EnterFullScreen),
  264. MenuItem::Minimize => Self(WryMenuItem::Minimize),
  265. MenuItem::Zoom => Self(WryMenuItem::Zoom),
  266. MenuItem::Separator => Self(WryMenuItem::Separator),
  267. _ => unimplemented!(),
  268. }
  269. }
  270. }
  271. #[cfg(target_os = "macos")]
  272. pub struct NativeImageWrapper(pub WryNativeImage);
  273. #[cfg(target_os = "macos")]
  274. impl std::fmt::Debug for NativeImageWrapper {
  275. fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
  276. f.debug_struct("NativeImageWrapper").finish()
  277. }
  278. }
  279. #[cfg(target_os = "macos")]
  280. impl From<NativeImage> for NativeImageWrapper {
  281. fn from(image: NativeImage) -> NativeImageWrapper {
  282. let wry_image = match image {
  283. NativeImage::Add => WryNativeImage::Add,
  284. NativeImage::Advanced => WryNativeImage::Advanced,
  285. NativeImage::Bluetooth => WryNativeImage::Bluetooth,
  286. NativeImage::Bookmarks => WryNativeImage::Bookmarks,
  287. NativeImage::Caution => WryNativeImage::Caution,
  288. NativeImage::ColorPanel => WryNativeImage::ColorPanel,
  289. NativeImage::ColumnView => WryNativeImage::ColumnView,
  290. NativeImage::Computer => WryNativeImage::Computer,
  291. NativeImage::EnterFullScreen => WryNativeImage::EnterFullScreen,
  292. NativeImage::Everyone => WryNativeImage::Everyone,
  293. NativeImage::ExitFullScreen => WryNativeImage::ExitFullScreen,
  294. NativeImage::FlowView => WryNativeImage::FlowView,
  295. NativeImage::Folder => WryNativeImage::Folder,
  296. NativeImage::FolderBurnable => WryNativeImage::FolderBurnable,
  297. NativeImage::FolderSmart => WryNativeImage::FolderSmart,
  298. NativeImage::FollowLinkFreestanding => WryNativeImage::FollowLinkFreestanding,
  299. NativeImage::FontPanel => WryNativeImage::FontPanel,
  300. NativeImage::GoLeft => WryNativeImage::GoLeft,
  301. NativeImage::GoRight => WryNativeImage::GoRight,
  302. NativeImage::Home => WryNativeImage::Home,
  303. NativeImage::IChatTheater => WryNativeImage::IChatTheater,
  304. NativeImage::IconView => WryNativeImage::IconView,
  305. NativeImage::Info => WryNativeImage::Info,
  306. NativeImage::InvalidDataFreestanding => WryNativeImage::InvalidDataFreestanding,
  307. NativeImage::LeftFacingTriangle => WryNativeImage::LeftFacingTriangle,
  308. NativeImage::ListView => WryNativeImage::ListView,
  309. NativeImage::LockLocked => WryNativeImage::LockLocked,
  310. NativeImage::LockUnlocked => WryNativeImage::LockUnlocked,
  311. NativeImage::MenuMixedState => WryNativeImage::MenuMixedState,
  312. NativeImage::MenuOnState => WryNativeImage::MenuOnState,
  313. NativeImage::MobileMe => WryNativeImage::MobileMe,
  314. NativeImage::MultipleDocuments => WryNativeImage::MultipleDocuments,
  315. NativeImage::Network => WryNativeImage::Network,
  316. NativeImage::Path => WryNativeImage::Path,
  317. NativeImage::PreferencesGeneral => WryNativeImage::PreferencesGeneral,
  318. NativeImage::QuickLook => WryNativeImage::QuickLook,
  319. NativeImage::RefreshFreestanding => WryNativeImage::RefreshFreestanding,
  320. NativeImage::Refresh => WryNativeImage::Refresh,
  321. NativeImage::Remove => WryNativeImage::Remove,
  322. NativeImage::RevealFreestanding => WryNativeImage::RevealFreestanding,
  323. NativeImage::RightFacingTriangle => WryNativeImage::RightFacingTriangle,
  324. NativeImage::Share => WryNativeImage::Share,
  325. NativeImage::Slideshow => WryNativeImage::Slideshow,
  326. NativeImage::SmartBadge => WryNativeImage::SmartBadge,
  327. NativeImage::StatusAvailable => WryNativeImage::StatusAvailable,
  328. NativeImage::StatusNone => WryNativeImage::StatusNone,
  329. NativeImage::StatusPartiallyAvailable => WryNativeImage::StatusPartiallyAvailable,
  330. NativeImage::StatusUnavailable => WryNativeImage::StatusUnavailable,
  331. NativeImage::StopProgressFreestanding => WryNativeImage::StopProgressFreestanding,
  332. NativeImage::StopProgress => WryNativeImage::StopProgress,
  333. NativeImage::TrashEmpty => WryNativeImage::TrashEmpty,
  334. NativeImage::TrashFull => WryNativeImage::TrashFull,
  335. NativeImage::User => WryNativeImage::User,
  336. NativeImage::UserAccounts => WryNativeImage::UserAccounts,
  337. NativeImage::UserGroup => WryNativeImage::UserGroup,
  338. NativeImage::UserGuest => WryNativeImage::UserGuest,
  339. };
  340. Self(wry_image)
  341. }
  342. }
  343. #[derive(Debug, Clone)]
  344. pub struct GlobalShortcutWrapper(GlobalShortcut);
  345. unsafe impl Send for GlobalShortcutWrapper {}
  346. /// Wrapper around [`WryShortcutManager`].
  347. #[derive(Clone)]
  348. pub struct GlobalShortcutManagerHandle {
  349. context: Context,
  350. shortcuts: Arc<Mutex<HashMap<String, (AcceleratorId, GlobalShortcutWrapper)>>>,
  351. listeners: GlobalShortcutListeners,
  352. }
  353. impl fmt::Debug for GlobalShortcutManagerHandle {
  354. fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
  355. f.debug_struct("GlobalShortcutManagerHandle")
  356. .field("context", &self.context)
  357. .field("shortcuts", &self.shortcuts)
  358. .finish()
  359. }
  360. }
  361. impl GlobalShortcutManager for GlobalShortcutManagerHandle {
  362. fn is_registered(&self, accelerator: &str) -> Result<bool> {
  363. let (tx, rx) = channel();
  364. Ok(getter!(
  365. self,
  366. rx,
  367. Message::GlobalShortcut(GlobalShortcutMessage::IsRegistered(
  368. accelerator.parse().expect("invalid accelerator"),
  369. tx
  370. ))
  371. ))
  372. }
  373. fn register<F: Fn() + Send + 'static>(&mut self, accelerator: &str, handler: F) -> Result<()> {
  374. let wry_accelerator: Accelerator = accelerator.parse().expect("invalid accelerator");
  375. let id = wry_accelerator.clone().id();
  376. let (tx, rx) = channel();
  377. let shortcut = getter!(
  378. self,
  379. rx,
  380. Message::GlobalShortcut(GlobalShortcutMessage::Register(wry_accelerator, tx))
  381. )?;
  382. self.listeners.lock().unwrap().insert(id, Box::new(handler));
  383. self
  384. .shortcuts
  385. .lock()
  386. .unwrap()
  387. .insert(accelerator.into(), (id, shortcut));
  388. Ok(())
  389. }
  390. fn unregister_all(&mut self) -> Result<()> {
  391. let (tx, rx) = channel();
  392. getter!(
  393. self,
  394. rx,
  395. Message::GlobalShortcut(GlobalShortcutMessage::UnregisterAll(tx))
  396. )?;
  397. self.listeners.lock().unwrap().clear();
  398. self.shortcuts.lock().unwrap().clear();
  399. Ok(())
  400. }
  401. fn unregister(&mut self, accelerator: &str) -> Result<()> {
  402. if let Some((accelerator_id, shortcut)) = self.shortcuts.lock().unwrap().remove(accelerator) {
  403. let (tx, rx) = channel();
  404. getter!(
  405. self,
  406. rx,
  407. Message::GlobalShortcut(GlobalShortcutMessage::Unregister(shortcut, tx))
  408. )?;
  409. self.listeners.lock().unwrap().remove(&accelerator_id);
  410. }
  411. Ok(())
  412. }
  413. }
  414. #[derive(Debug, Clone)]
  415. pub struct ClipboardManagerWrapper {
  416. context: Context,
  417. }
  418. impl ClipboardManager for ClipboardManagerWrapper {
  419. fn read_text(&self) -> Result<Option<String>> {
  420. let (tx, rx) = channel();
  421. Ok(getter!(
  422. self,
  423. rx,
  424. Message::Clipboard(ClipboardMessage::ReadText(tx))
  425. ))
  426. }
  427. fn write_text<T: Into<String>>(&mut self, text: T) -> Result<()> {
  428. let (tx, rx) = channel();
  429. getter!(
  430. self,
  431. rx,
  432. Message::Clipboard(ClipboardMessage::WriteText(text.into(), tx))
  433. );
  434. Ok(())
  435. }
  436. }
  437. /// Wrapper around a [`wry::application::window::Icon`] that can be created from an [`Icon`].
  438. pub struct WryIcon(WindowIcon);
  439. fn icon_err<E: std::error::Error + Send + 'static>(e: E) -> Error {
  440. Error::InvalidIcon(Box::new(e))
  441. }
  442. impl TryFrom<Icon> for WryIcon {
  443. type Error = Error;
  444. fn try_from(icon: Icon) -> std::result::Result<Self, Self::Error> {
  445. let image_bytes = match icon {
  446. Icon::File(path) => read(path).map_err(icon_err)?,
  447. Icon::Raw(raw) => raw,
  448. _ => unimplemented!(),
  449. };
  450. let extension = infer::get(&image_bytes)
  451. .expect("could not determine icon extension")
  452. .extension();
  453. match extension {
  454. #[cfg(windows)]
  455. "ico" => {
  456. let icon_dir = ico::IconDir::read(std::io::Cursor::new(image_bytes)).map_err(icon_err)?;
  457. let entry = &icon_dir.entries()[0];
  458. let icon = WindowIcon::from_rgba(
  459. entry.decode().map_err(icon_err)?.rgba_data().to_vec(),
  460. entry.width(),
  461. entry.height(),
  462. )
  463. .map_err(icon_err)?;
  464. Ok(Self(icon))
  465. }
  466. #[cfg(target_os = "linux")]
  467. "png" => {
  468. let decoder = png::Decoder::new(std::io::Cursor::new(image_bytes));
  469. let (info, mut reader) = decoder.read_info().map_err(icon_err)?;
  470. let mut buffer = Vec::new();
  471. while let Ok(Some(row)) = reader.next_row() {
  472. buffer.extend(row);
  473. }
  474. let icon = WindowIcon::from_rgba(buffer, info.width, info.height).map_err(icon_err)?;
  475. Ok(Self(icon))
  476. }
  477. _ => panic!(
  478. "image `{}` extension not supported; please file a Tauri feature request",
  479. extension
  480. ),
  481. }
  482. }
  483. }
  484. struct WindowEventWrapper(Option<WindowEvent>);
  485. impl WindowEventWrapper {
  486. fn parse(webview: &WindowHandle, event: &WryWindowEvent<'_>) -> Self {
  487. match event {
  488. // resized event from tao doesn't include a reliable size on macOS
  489. // because wry replaces the NSView
  490. WryWindowEvent::Resized(_) => Self(Some(WindowEvent::Resized(
  491. PhysicalSizeWrapper(webview.inner_size()).into(),
  492. ))),
  493. e => e.into(),
  494. }
  495. }
  496. }
  497. impl<'a> From<&WryWindowEvent<'a>> for WindowEventWrapper {
  498. fn from(event: &WryWindowEvent<'a>) -> Self {
  499. let event = match event {
  500. WryWindowEvent::Resized(size) => WindowEvent::Resized(PhysicalSizeWrapper(*size).into()),
  501. WryWindowEvent::Moved(position) => {
  502. WindowEvent::Moved(PhysicalPositionWrapper(*position).into())
  503. }
  504. WryWindowEvent::CloseRequested => WindowEvent::CloseRequested,
  505. WryWindowEvent::Destroyed => WindowEvent::Destroyed,
  506. WryWindowEvent::ScaleFactorChanged {
  507. scale_factor,
  508. new_inner_size,
  509. } => WindowEvent::ScaleFactorChanged {
  510. scale_factor: *scale_factor,
  511. new_inner_size: PhysicalSizeWrapper(**new_inner_size).into(),
  512. },
  513. #[cfg(any(target_os = "linux", target_os = "macos"))]
  514. WryWindowEvent::Focused(focused) => WindowEvent::Focused(*focused),
  515. _ => return Self(None),
  516. };
  517. Self(Some(event))
  518. }
  519. }
  520. impl From<&WebviewEvent> for WindowEventWrapper {
  521. fn from(event: &WebviewEvent) -> Self {
  522. let event = match event {
  523. WebviewEvent::Focused(focused) => WindowEvent::Focused(*focused),
  524. };
  525. Self(Some(event))
  526. }
  527. }
  528. pub struct MonitorHandleWrapper(MonitorHandle);
  529. impl From<MonitorHandleWrapper> for Monitor {
  530. fn from(monitor: MonitorHandleWrapper) -> Monitor {
  531. Self {
  532. name: monitor.0.name(),
  533. position: PhysicalPositionWrapper(monitor.0.position()).into(),
  534. size: PhysicalSizeWrapper(monitor.0.size()).into(),
  535. scale_factor: monitor.0.scale_factor(),
  536. }
  537. }
  538. }
  539. struct PhysicalPositionWrapper<T>(WryPhysicalPosition<T>);
  540. impl<T> From<PhysicalPositionWrapper<T>> for PhysicalPosition<T> {
  541. fn from(position: PhysicalPositionWrapper<T>) -> Self {
  542. Self {
  543. x: position.0.x,
  544. y: position.0.y,
  545. }
  546. }
  547. }
  548. impl<T> From<PhysicalPosition<T>> for PhysicalPositionWrapper<T> {
  549. fn from(position: PhysicalPosition<T>) -> Self {
  550. Self(WryPhysicalPosition {
  551. x: position.x,
  552. y: position.y,
  553. })
  554. }
  555. }
  556. struct LogicalPositionWrapper<T>(WryLogicalPosition<T>);
  557. impl<T> From<LogicalPosition<T>> for LogicalPositionWrapper<T> {
  558. fn from(position: LogicalPosition<T>) -> Self {
  559. Self(WryLogicalPosition {
  560. x: position.x,
  561. y: position.y,
  562. })
  563. }
  564. }
  565. struct PhysicalSizeWrapper<T>(WryPhysicalSize<T>);
  566. impl<T> From<PhysicalSizeWrapper<T>> for PhysicalSize<T> {
  567. fn from(size: PhysicalSizeWrapper<T>) -> Self {
  568. Self {
  569. width: size.0.width,
  570. height: size.0.height,
  571. }
  572. }
  573. }
  574. impl<T> From<PhysicalSize<T>> for PhysicalSizeWrapper<T> {
  575. fn from(size: PhysicalSize<T>) -> Self {
  576. Self(WryPhysicalSize {
  577. width: size.width,
  578. height: size.height,
  579. })
  580. }
  581. }
  582. struct LogicalSizeWrapper<T>(WryLogicalSize<T>);
  583. impl<T> From<LogicalSize<T>> for LogicalSizeWrapper<T> {
  584. fn from(size: LogicalSize<T>) -> Self {
  585. Self(WryLogicalSize {
  586. width: size.width,
  587. height: size.height,
  588. })
  589. }
  590. }
  591. struct SizeWrapper(WrySize);
  592. impl From<Size> for SizeWrapper {
  593. fn from(size: Size) -> Self {
  594. match size {
  595. Size::Logical(s) => Self(WrySize::Logical(LogicalSizeWrapper::from(s).0)),
  596. Size::Physical(s) => Self(WrySize::Physical(PhysicalSizeWrapper::from(s).0)),
  597. }
  598. }
  599. }
  600. struct PositionWrapper(WryPosition);
  601. impl From<Position> for PositionWrapper {
  602. fn from(position: Position) -> Self {
  603. match position {
  604. Position::Logical(s) => Self(WryPosition::Logical(LogicalPositionWrapper::from(s).0)),
  605. Position::Physical(s) => Self(WryPosition::Physical(PhysicalPositionWrapper::from(s).0)),
  606. }
  607. }
  608. }
  609. #[derive(Debug, Clone)]
  610. pub struct UserAttentionTypeWrapper(WryUserAttentionType);
  611. impl From<UserAttentionType> for UserAttentionTypeWrapper {
  612. fn from(request_type: UserAttentionType) -> UserAttentionTypeWrapper {
  613. let o = match request_type {
  614. UserAttentionType::Critical => WryUserAttentionType::Critical,
  615. UserAttentionType::Informational => WryUserAttentionType::Informational,
  616. };
  617. Self(o)
  618. }
  619. }
  620. #[derive(Debug, Clone, Default)]
  621. pub struct WindowBuilderWrapper {
  622. inner: WryWindowBuilder,
  623. center: bool,
  624. menu: Option<Menu>,
  625. }
  626. // safe since `menu_items` are read only here
  627. unsafe impl Send for WindowBuilderWrapper {}
  628. impl WindowBuilderBase for WindowBuilderWrapper {}
  629. impl WindowBuilder for WindowBuilderWrapper {
  630. fn new() -> Self {
  631. Default::default()
  632. }
  633. fn with_config(config: WindowConfig) -> Self {
  634. let mut window = WindowBuilderWrapper::new()
  635. .title(config.title.to_string())
  636. .inner_size(config.width, config.height)
  637. .visible(config.visible)
  638. .resizable(config.resizable)
  639. .decorations(config.decorations)
  640. .maximized(config.maximized)
  641. .fullscreen(config.fullscreen)
  642. .transparent(config.transparent)
  643. .always_on_top(config.always_on_top)
  644. .skip_taskbar(config.skip_taskbar);
  645. if let (Some(min_width), Some(min_height)) = (config.min_width, config.min_height) {
  646. window = window.min_inner_size(min_width, min_height);
  647. }
  648. if let (Some(max_width), Some(max_height)) = (config.max_width, config.max_height) {
  649. window = window.max_inner_size(max_width, max_height);
  650. }
  651. if let (Some(x), Some(y)) = (config.x, config.y) {
  652. window = window.position(x, y);
  653. }
  654. if config.center {
  655. window = window.center();
  656. }
  657. window
  658. }
  659. fn menu(mut self, menu: Menu) -> Self {
  660. self.menu.replace(menu);
  661. self
  662. }
  663. fn center(mut self) -> Self {
  664. self.center = true;
  665. self
  666. }
  667. fn position(mut self, x: f64, y: f64) -> Self {
  668. self.inner = self.inner.with_position(WryLogicalPosition::new(x, y));
  669. self
  670. }
  671. fn inner_size(mut self, width: f64, height: f64) -> Self {
  672. self.inner = self
  673. .inner
  674. .with_inner_size(WryLogicalSize::new(width, height));
  675. self
  676. }
  677. fn min_inner_size(mut self, min_width: f64, min_height: f64) -> Self {
  678. self.inner = self
  679. .inner
  680. .with_min_inner_size(WryLogicalSize::new(min_width, min_height));
  681. self
  682. }
  683. fn max_inner_size(mut self, max_width: f64, max_height: f64) -> Self {
  684. self.inner = self
  685. .inner
  686. .with_max_inner_size(WryLogicalSize::new(max_width, max_height));
  687. self
  688. }
  689. fn resizable(mut self, resizable: bool) -> Self {
  690. self.inner = self.inner.with_resizable(resizable);
  691. self
  692. }
  693. fn title<S: Into<String>>(mut self, title: S) -> Self {
  694. self.inner = self.inner.with_title(title.into());
  695. self
  696. }
  697. fn fullscreen(mut self, fullscreen: bool) -> Self {
  698. self.inner = if fullscreen {
  699. self
  700. .inner
  701. .with_fullscreen(Some(Fullscreen::Borderless(None)))
  702. } else {
  703. self.inner.with_fullscreen(None)
  704. };
  705. self
  706. }
  707. /// Deprecated since 0.1.4 (noop)
  708. /// Windows is automatically focused when created.
  709. fn focus(self) -> Self {
  710. self
  711. }
  712. fn maximized(mut self, maximized: bool) -> Self {
  713. self.inner = self.inner.with_maximized(maximized);
  714. self
  715. }
  716. fn visible(mut self, visible: bool) -> Self {
  717. self.inner = self.inner.with_visible(visible);
  718. self
  719. }
  720. fn transparent(mut self, transparent: bool) -> Self {
  721. self.inner = self.inner.with_transparent(transparent);
  722. self
  723. }
  724. fn decorations(mut self, decorations: bool) -> Self {
  725. self.inner = self.inner.with_decorations(decorations);
  726. self
  727. }
  728. fn always_on_top(mut self, always_on_top: bool) -> Self {
  729. self.inner = self.inner.with_always_on_top(always_on_top);
  730. self
  731. }
  732. #[cfg(windows)]
  733. fn parent_window(mut self, parent: HWND) -> Self {
  734. self.inner = self.inner.with_parent_window(parent);
  735. self
  736. }
  737. #[cfg(windows)]
  738. fn owner_window(mut self, owner: HWND) -> Self {
  739. self.inner = self.inner.with_owner_window(owner);
  740. self
  741. }
  742. fn icon(mut self, icon: Icon) -> Result<Self> {
  743. self.inner = self
  744. .inner
  745. .with_window_icon(Some(WryIcon::try_from(icon)?.0));
  746. Ok(self)
  747. }
  748. #[cfg(any(target_os = "windows", target_os = "linux"))]
  749. fn skip_taskbar(mut self, skip: bool) -> Self {
  750. self.inner = self.inner.with_skip_taskbar(skip);
  751. self
  752. }
  753. #[cfg(target_os = "macos")]
  754. fn skip_taskbar(self, _skip: bool) -> Self {
  755. self
  756. }
  757. fn has_icon(&self) -> bool {
  758. self.inner.window.window_icon.is_some()
  759. }
  760. fn get_menu(&self) -> Option<&Menu> {
  761. self.menu.as_ref()
  762. }
  763. }
  764. pub struct RpcRequestWrapper(WryRpcRequest);
  765. impl From<RpcRequestWrapper> for RpcRequest {
  766. fn from(request: RpcRequestWrapper) -> Self {
  767. Self {
  768. command: request.0.method,
  769. params: request.0.params,
  770. }
  771. }
  772. }
  773. pub struct FileDropEventWrapper(WryFileDropEvent);
  774. impl From<FileDropEventWrapper> for FileDropEvent {
  775. fn from(event: FileDropEventWrapper) -> Self {
  776. match event.0 {
  777. WryFileDropEvent::Hovered(paths) => FileDropEvent::Hovered(paths),
  778. WryFileDropEvent::Dropped(paths) => FileDropEvent::Dropped(paths),
  779. // default to cancelled
  780. // FIXME(maybe): Add `FileDropEvent::Unknown` event?
  781. _ => FileDropEvent::Cancelled,
  782. }
  783. }
  784. }
  785. #[cfg(target_os = "macos")]
  786. pub struct NSWindow(*mut std::ffi::c_void);
  787. #[cfg(target_os = "macos")]
  788. #[allow(clippy::non_send_fields_in_send_ty)]
  789. unsafe impl Send for NSWindow {}
  790. #[cfg(windows)]
  791. pub struct Hwnd(HWND);
  792. #[cfg(windows)]
  793. #[allow(clippy::non_send_fields_in_send_ty)]
  794. unsafe impl Send for Hwnd {}
  795. #[cfg(any(
  796. target_os = "linux",
  797. target_os = "dragonfly",
  798. target_os = "freebsd",
  799. target_os = "netbsd",
  800. target_os = "openbsd"
  801. ))]
  802. pub struct GtkWindow(gtk::ApplicationWindow);
  803. #[cfg(any(
  804. target_os = "linux",
  805. target_os = "dragonfly",
  806. target_os = "freebsd",
  807. target_os = "netbsd",
  808. target_os = "openbsd"
  809. ))]
  810. #[allow(clippy::non_send_fields_in_send_ty)]
  811. unsafe impl Send for GtkWindow {}
  812. #[derive(Debug, Clone)]
  813. pub enum WindowMessage {
  814. // Getters
  815. ScaleFactor(Sender<f64>),
  816. InnerPosition(Sender<Result<PhysicalPosition<i32>>>),
  817. OuterPosition(Sender<Result<PhysicalPosition<i32>>>),
  818. InnerSize(Sender<PhysicalSize<u32>>),
  819. OuterSize(Sender<PhysicalSize<u32>>),
  820. IsFullscreen(Sender<bool>),
  821. IsMaximized(Sender<bool>),
  822. IsDecorated(Sender<bool>),
  823. IsResizable(Sender<bool>),
  824. IsVisible(Sender<bool>),
  825. IsMenuVisible(Sender<bool>),
  826. CurrentMonitor(Sender<Option<MonitorHandle>>),
  827. PrimaryMonitor(Sender<Option<MonitorHandle>>),
  828. AvailableMonitors(Sender<Vec<MonitorHandle>>),
  829. #[cfg(target_os = "macos")]
  830. NSWindow(Sender<NSWindow>),
  831. #[cfg(windows)]
  832. Hwnd(Sender<Hwnd>),
  833. #[cfg(any(
  834. target_os = "linux",
  835. target_os = "dragonfly",
  836. target_os = "freebsd",
  837. target_os = "netbsd",
  838. target_os = "openbsd"
  839. ))]
  840. GtkWindow(Sender<GtkWindow>),
  841. // Setters
  842. Center(Sender<Result<()>>),
  843. RequestUserAttention(Option<UserAttentionTypeWrapper>),
  844. SetResizable(bool),
  845. SetTitle(String),
  846. Maximize,
  847. Unmaximize,
  848. Minimize,
  849. Unminimize,
  850. ShowMenu,
  851. HideMenu,
  852. Show,
  853. Hide,
  854. Close,
  855. SetDecorations(bool),
  856. SetAlwaysOnTop(bool),
  857. SetSize(Size),
  858. SetMinSize(Option<Size>),
  859. SetMaxSize(Option<Size>),
  860. SetPosition(Position),
  861. SetFullscreen(bool),
  862. SetFocus,
  863. SetIcon(WindowIcon),
  864. SetSkipTaskbar(bool),
  865. DragWindow,
  866. UpdateMenuItem(u16, MenuUpdate),
  867. }
  868. #[derive(Debug, Clone)]
  869. pub enum WebviewMessage {
  870. EvaluateScript(String),
  871. #[allow(dead_code)]
  872. WebviewEvent(WebviewEvent),
  873. Print,
  874. }
  875. #[allow(dead_code)]
  876. #[derive(Debug, Clone)]
  877. pub enum WebviewEvent {
  878. Focused(bool),
  879. }
  880. #[cfg(feature = "system-tray")]
  881. #[derive(Debug, Clone)]
  882. pub enum TrayMessage {
  883. UpdateItem(u16, MenuUpdate),
  884. UpdateMenu(SystemTrayMenu),
  885. UpdateIcon(Icon),
  886. #[cfg(target_os = "macos")]
  887. UpdateIconAsTemplate(bool),
  888. }
  889. #[derive(Debug, Clone)]
  890. pub enum GlobalShortcutMessage {
  891. IsRegistered(Accelerator, Sender<bool>),
  892. Register(Accelerator, Sender<Result<GlobalShortcutWrapper>>),
  893. Unregister(GlobalShortcutWrapper, Sender<Result<()>>),
  894. UnregisterAll(Sender<Result<()>>),
  895. }
  896. #[derive(Debug, Clone)]
  897. pub enum ClipboardMessage {
  898. WriteText(String, Sender<()>),
  899. ReadText(Sender<Option<String>>),
  900. }
  901. pub enum Message {
  902. Task(Box<dyn FnOnce() + Send>),
  903. Window(WindowId, WindowMessage),
  904. Webview(WindowId, WebviewMessage),
  905. #[cfg(feature = "system-tray")]
  906. Tray(TrayMessage),
  907. CreateWebview(
  908. Box<
  909. dyn FnOnce(&EventLoopWindowTarget<Message>, &WebContextStore) -> Result<WindowWrapper> + Send,
  910. >,
  911. Sender<WindowId>,
  912. ),
  913. CreateWindow(
  914. Box<dyn FnOnce() -> (String, WryWindowBuilder) + Send>,
  915. Sender<Result<Weak<Window>>>,
  916. ),
  917. GlobalShortcut(GlobalShortcutMessage),
  918. Clipboard(ClipboardMessage),
  919. }
  920. impl Clone for Message {
  921. fn clone(&self) -> Self {
  922. match self {
  923. Self::Window(i, m) => Self::Window(*i, m.clone()),
  924. Self::Webview(i, m) => Self::Webview(*i, m.clone()),
  925. #[cfg(feature = "system-tray")]
  926. Self::Tray(m) => Self::Tray(m.clone()),
  927. Self::GlobalShortcut(m) => Self::GlobalShortcut(m.clone()),
  928. Self::Clipboard(m) => Self::Clipboard(m.clone()),
  929. _ => unimplemented!(),
  930. }
  931. }
  932. }
  933. /// The Tauri [`Dispatch`] for [`Wry`].
  934. #[derive(Debug, Clone)]
  935. pub struct WryDispatcher {
  936. window_id: WindowId,
  937. context: Context,
  938. }
  939. impl Dispatch for WryDispatcher {
  940. type Runtime = Wry;
  941. type WindowBuilder = WindowBuilderWrapper;
  942. fn run_on_main_thread<F: FnOnce() + Send + 'static>(&self, f: F) -> Result<()> {
  943. send_user_message(&self.context, Message::Task(Box::new(f)))
  944. }
  945. fn on_window_event<F: Fn(&WindowEvent) + Send + 'static>(&self, f: F) -> Uuid {
  946. let id = Uuid::new_v4();
  947. self
  948. .context
  949. .window_event_listeners
  950. .lock()
  951. .unwrap()
  952. .get(&self.window_id)
  953. .unwrap()
  954. .lock()
  955. .unwrap()
  956. .insert(id, Box::new(f));
  957. id
  958. }
  959. fn on_menu_event<F: Fn(&MenuEvent) + Send + 'static>(&self, f: F) -> Uuid {
  960. let id = Uuid::new_v4();
  961. self
  962. .context
  963. .menu_event_listeners
  964. .lock()
  965. .unwrap()
  966. .get(&self.window_id)
  967. .unwrap()
  968. .lock()
  969. .unwrap()
  970. .insert(id, Box::new(f));
  971. id
  972. }
  973. // Getters
  974. fn scale_factor(&self) -> Result<f64> {
  975. Ok(window_getter!(self, WindowMessage::ScaleFactor))
  976. }
  977. fn inner_position(&self) -> Result<PhysicalPosition<i32>> {
  978. window_getter!(self, WindowMessage::InnerPosition)
  979. }
  980. fn outer_position(&self) -> Result<PhysicalPosition<i32>> {
  981. window_getter!(self, WindowMessage::OuterPosition)
  982. }
  983. fn inner_size(&self) -> Result<PhysicalSize<u32>> {
  984. Ok(window_getter!(self, WindowMessage::InnerSize))
  985. }
  986. fn outer_size(&self) -> Result<PhysicalSize<u32>> {
  987. Ok(window_getter!(self, WindowMessage::OuterSize))
  988. }
  989. fn is_fullscreen(&self) -> Result<bool> {
  990. Ok(window_getter!(self, WindowMessage::IsFullscreen))
  991. }
  992. fn is_maximized(&self) -> Result<bool> {
  993. Ok(window_getter!(self, WindowMessage::IsMaximized))
  994. }
  995. /// Gets the window’s current decoration state.
  996. fn is_decorated(&self) -> Result<bool> {
  997. Ok(window_getter!(self, WindowMessage::IsDecorated))
  998. }
  999. /// Gets the window’s current resizable state.
  1000. fn is_resizable(&self) -> Result<bool> {
  1001. Ok(window_getter!(self, WindowMessage::IsResizable))
  1002. }
  1003. fn is_visible(&self) -> Result<bool> {
  1004. Ok(window_getter!(self, WindowMessage::IsVisible))
  1005. }
  1006. fn is_menu_visible(&self) -> Result<bool> {
  1007. Ok(window_getter!(self, WindowMessage::IsMenuVisible))
  1008. }
  1009. fn current_monitor(&self) -> Result<Option<Monitor>> {
  1010. Ok(window_getter!(self, WindowMessage::CurrentMonitor).map(|m| MonitorHandleWrapper(m).into()))
  1011. }
  1012. fn primary_monitor(&self) -> Result<Option<Monitor>> {
  1013. Ok(window_getter!(self, WindowMessage::PrimaryMonitor).map(|m| MonitorHandleWrapper(m).into()))
  1014. }
  1015. fn available_monitors(&self) -> Result<Vec<Monitor>> {
  1016. Ok(
  1017. window_getter!(self, WindowMessage::AvailableMonitors)
  1018. .into_iter()
  1019. .map(|m| MonitorHandleWrapper(m).into())
  1020. .collect(),
  1021. )
  1022. }
  1023. #[cfg(target_os = "macos")]
  1024. fn ns_window(&self) -> Result<*mut std::ffi::c_void> {
  1025. Ok(window_getter!(self, WindowMessage::NSWindow).0)
  1026. }
  1027. #[cfg(windows)]
  1028. fn hwnd(&self) -> Result<HWND> {
  1029. Ok(window_getter!(self, WindowMessage::Hwnd).0)
  1030. }
  1031. /// Returns the `ApplicatonWindow` from gtk crate that is used by this window.
  1032. #[cfg(any(
  1033. target_os = "linux",
  1034. target_os = "dragonfly",
  1035. target_os = "freebsd",
  1036. target_os = "netbsd",
  1037. target_os = "openbsd"
  1038. ))]
  1039. fn gtk_window(&self) -> Result<gtk::ApplicationWindow> {
  1040. Ok(window_getter!(self, WindowMessage::GtkWindow).0)
  1041. }
  1042. // Setters
  1043. fn center(&self) -> Result<()> {
  1044. window_getter!(self, WindowMessage::Center)
  1045. }
  1046. fn print(&self) -> Result<()> {
  1047. send_user_message(
  1048. &self.context,
  1049. Message::Webview(self.window_id, WebviewMessage::Print),
  1050. )
  1051. }
  1052. fn request_user_attention(&self, request_type: Option<UserAttentionType>) -> Result<()> {
  1053. send_user_message(
  1054. &self.context,
  1055. Message::Window(
  1056. self.window_id,
  1057. WindowMessage::RequestUserAttention(request_type.map(Into::into)),
  1058. ),
  1059. )
  1060. }
  1061. // Creates a window by dispatching a message to the event loop.
  1062. // Note that this must be called from a separate thread, otherwise the channel will introduce a deadlock.
  1063. fn create_window(
  1064. &mut self,
  1065. pending: PendingWindow<Self::Runtime>,
  1066. ) -> Result<DetachedWindow<Self::Runtime>> {
  1067. let (tx, rx) = channel();
  1068. let label = pending.label.clone();
  1069. let menu_ids = pending.menu_ids.clone();
  1070. let context = self.context.clone();
  1071. send_user_message(
  1072. &self.context,
  1073. Message::CreateWebview(
  1074. Box::new(move |event_loop, web_context| {
  1075. create_webview(event_loop, web_context, context, pending)
  1076. }),
  1077. tx,
  1078. ),
  1079. )?;
  1080. let window_id = rx.recv().unwrap();
  1081. let dispatcher = WryDispatcher {
  1082. window_id,
  1083. context: self.context.clone(),
  1084. };
  1085. Ok(DetachedWindow {
  1086. label,
  1087. dispatcher,
  1088. menu_ids,
  1089. })
  1090. }
  1091. fn set_resizable(&self, resizable: bool) -> Result<()> {
  1092. send_user_message(
  1093. &self.context,
  1094. Message::Window(self.window_id, WindowMessage::SetResizable(resizable)),
  1095. )
  1096. }
  1097. fn set_title<S: Into<String>>(&self, title: S) -> Result<()> {
  1098. send_user_message(
  1099. &self.context,
  1100. Message::Window(self.window_id, WindowMessage::SetTitle(title.into())),
  1101. )
  1102. }
  1103. fn maximize(&self) -> Result<()> {
  1104. send_user_message(
  1105. &self.context,
  1106. Message::Window(self.window_id, WindowMessage::Maximize),
  1107. )
  1108. }
  1109. fn unmaximize(&self) -> Result<()> {
  1110. send_user_message(
  1111. &self.context,
  1112. Message::Window(self.window_id, WindowMessage::Unmaximize),
  1113. )
  1114. }
  1115. fn minimize(&self) -> Result<()> {
  1116. send_user_message(
  1117. &self.context,
  1118. Message::Window(self.window_id, WindowMessage::Minimize),
  1119. )
  1120. }
  1121. fn unminimize(&self) -> Result<()> {
  1122. send_user_message(
  1123. &self.context,
  1124. Message::Window(self.window_id, WindowMessage::Unminimize),
  1125. )
  1126. }
  1127. fn show_menu(&self) -> Result<()> {
  1128. send_user_message(
  1129. &self.context,
  1130. Message::Window(self.window_id, WindowMessage::ShowMenu),
  1131. )
  1132. }
  1133. fn hide_menu(&self) -> Result<()> {
  1134. send_user_message(
  1135. &self.context,
  1136. Message::Window(self.window_id, WindowMessage::HideMenu),
  1137. )
  1138. }
  1139. fn show(&self) -> Result<()> {
  1140. send_user_message(
  1141. &self.context,
  1142. Message::Window(self.window_id, WindowMessage::Show),
  1143. )
  1144. }
  1145. fn hide(&self) -> Result<()> {
  1146. send_user_message(
  1147. &self.context,
  1148. Message::Window(self.window_id, WindowMessage::Hide),
  1149. )
  1150. }
  1151. fn close(&self) -> Result<()> {
  1152. // NOTE: close cannot use the `send_user_message` function because it accesses the event loop callback
  1153. self
  1154. .context
  1155. .proxy
  1156. .send_event(Message::Window(self.window_id, WindowMessage::Close))
  1157. .map_err(|_| Error::FailedToSendMessage)
  1158. }
  1159. fn set_decorations(&self, decorations: bool) -> Result<()> {
  1160. send_user_message(
  1161. &self.context,
  1162. Message::Window(self.window_id, WindowMessage::SetDecorations(decorations)),
  1163. )
  1164. }
  1165. fn set_always_on_top(&self, always_on_top: bool) -> Result<()> {
  1166. send_user_message(
  1167. &self.context,
  1168. Message::Window(self.window_id, WindowMessage::SetAlwaysOnTop(always_on_top)),
  1169. )
  1170. }
  1171. fn set_size(&self, size: Size) -> Result<()> {
  1172. send_user_message(
  1173. &self.context,
  1174. Message::Window(self.window_id, WindowMessage::SetSize(size)),
  1175. )
  1176. }
  1177. fn set_min_size(&self, size: Option<Size>) -> Result<()> {
  1178. send_user_message(
  1179. &self.context,
  1180. Message::Window(self.window_id, WindowMessage::SetMinSize(size)),
  1181. )
  1182. }
  1183. fn set_max_size(&self, size: Option<Size>) -> Result<()> {
  1184. send_user_message(
  1185. &self.context,
  1186. Message::Window(self.window_id, WindowMessage::SetMaxSize(size)),
  1187. )
  1188. }
  1189. fn set_position(&self, position: Position) -> Result<()> {
  1190. send_user_message(
  1191. &self.context,
  1192. Message::Window(self.window_id, WindowMessage::SetPosition(position)),
  1193. )
  1194. }
  1195. fn set_fullscreen(&self, fullscreen: bool) -> Result<()> {
  1196. send_user_message(
  1197. &self.context,
  1198. Message::Window(self.window_id, WindowMessage::SetFullscreen(fullscreen)),
  1199. )
  1200. }
  1201. fn set_focus(&self) -> Result<()> {
  1202. send_user_message(
  1203. &self.context,
  1204. Message::Window(self.window_id, WindowMessage::SetFocus),
  1205. )
  1206. }
  1207. fn set_icon(&self, icon: Icon) -> Result<()> {
  1208. send_user_message(
  1209. &self.context,
  1210. Message::Window(
  1211. self.window_id,
  1212. WindowMessage::SetIcon(WryIcon::try_from(icon)?.0),
  1213. ),
  1214. )
  1215. }
  1216. fn set_skip_taskbar(&self, skip: bool) -> Result<()> {
  1217. send_user_message(
  1218. &self.context,
  1219. Message::Window(self.window_id, WindowMessage::SetSkipTaskbar(skip)),
  1220. )
  1221. }
  1222. fn start_dragging(&self) -> Result<()> {
  1223. send_user_message(
  1224. &self.context,
  1225. Message::Window(self.window_id, WindowMessage::DragWindow),
  1226. )
  1227. }
  1228. fn eval_script<S: Into<String>>(&self, script: S) -> Result<()> {
  1229. send_user_message(
  1230. &self.context,
  1231. Message::Webview(
  1232. self.window_id,
  1233. WebviewMessage::EvaluateScript(script.into()),
  1234. ),
  1235. )
  1236. }
  1237. fn update_menu_item(&self, id: u16, update: MenuUpdate) -> Result<()> {
  1238. send_user_message(
  1239. &self.context,
  1240. Message::Window(self.window_id, WindowMessage::UpdateMenuItem(id, update)),
  1241. )
  1242. }
  1243. }
  1244. #[cfg(feature = "system-tray")]
  1245. #[derive(Clone, Default)]
  1246. struct TrayContext {
  1247. tray: Arc<Mutex<Option<Arc<Mutex<WrySystemTray>>>>>,
  1248. listeners: SystemTrayEventListeners,
  1249. items: SystemTrayItems,
  1250. }
  1251. #[cfg(feature = "system-tray")]
  1252. impl fmt::Debug for TrayContext {
  1253. fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
  1254. f.debug_struct("TrayContext")
  1255. .field("items", &self.items)
  1256. .finish()
  1257. }
  1258. }
  1259. enum WindowHandle {
  1260. Webview(WebView),
  1261. Window(Arc<Window>),
  1262. }
  1263. impl fmt::Debug for WindowHandle {
  1264. fn fmt(&self, _f: &mut fmt::Formatter<'_>) -> fmt::Result {
  1265. Ok(())
  1266. }
  1267. }
  1268. impl WindowHandle {
  1269. fn window(&self) -> &Window {
  1270. match self {
  1271. Self::Webview(w) => w.window(),
  1272. Self::Window(w) => w,
  1273. }
  1274. }
  1275. fn inner_size(&self) -> WryPhysicalSize<u32> {
  1276. match self {
  1277. WindowHandle::Window(w) => w.inner_size(),
  1278. WindowHandle::Webview(w) => w.inner_size(),
  1279. }
  1280. }
  1281. }
  1282. #[derive(Debug)]
  1283. pub struct WindowWrapper {
  1284. label: String,
  1285. inner: WindowHandle,
  1286. menu_items: Option<HashMap<u16, WryCustomMenuItem>>,
  1287. }
  1288. /// A Tauri [`Runtime`] wrapper around wry.
  1289. pub struct Wry {
  1290. main_thread_id: ThreadId,
  1291. global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
  1292. global_shortcut_manager_handle: GlobalShortcutManagerHandle,
  1293. clipboard_manager: Arc<Mutex<Clipboard>>,
  1294. clipboard_manager_handle: ClipboardManagerWrapper,
  1295. event_loop: EventLoop<Message>,
  1296. windows: Arc<Mutex<HashMap<WindowId, WindowWrapper>>>,
  1297. web_context: WebContextStore,
  1298. window_event_listeners: WindowEventListeners,
  1299. menu_event_listeners: MenuEventListeners,
  1300. #[cfg(feature = "system-tray")]
  1301. tray_context: TrayContext,
  1302. }
  1303. /// A handle to the Wry runtime.
  1304. #[derive(Debug, Clone)]
  1305. pub struct WryHandle {
  1306. context: Context,
  1307. }
  1308. impl WryHandle {
  1309. /// Creates a new tao window using a callback, and returns its window id.
  1310. pub fn create_tao_window<F: FnOnce() -> (String, WryWindowBuilder) + Send + 'static>(
  1311. &self,
  1312. f: F,
  1313. ) -> Result<Weak<Window>> {
  1314. let (tx, rx) = channel();
  1315. send_user_message(&self.context, Message::CreateWindow(Box::new(f), tx))?;
  1316. rx.recv().unwrap()
  1317. }
  1318. /// Send a message to the event loop.
  1319. pub fn send_event(&self, message: Message) -> Result<()> {
  1320. self
  1321. .context
  1322. .proxy
  1323. .send_event(message)
  1324. .map_err(|_| Error::FailedToSendMessage)?;
  1325. Ok(())
  1326. }
  1327. }
  1328. impl RuntimeHandle for WryHandle {
  1329. type Runtime = Wry;
  1330. // Creates a window by dispatching a message to the event loop.
  1331. // Note that this must be called from a separate thread, otherwise the channel will introduce a deadlock.
  1332. fn create_window(
  1333. &self,
  1334. pending: PendingWindow<Self::Runtime>,
  1335. ) -> Result<DetachedWindow<Self::Runtime>> {
  1336. let (tx, rx) = channel();
  1337. let label = pending.label.clone();
  1338. let menu_ids = pending.menu_ids.clone();
  1339. let context = self.context.clone();
  1340. send_user_message(
  1341. &self.context,
  1342. Message::CreateWebview(
  1343. Box::new(move |event_loop, web_context| {
  1344. create_webview(event_loop, web_context, context, pending)
  1345. }),
  1346. tx,
  1347. ),
  1348. )?;
  1349. let window_id = rx.recv().unwrap();
  1350. let dispatcher = WryDispatcher {
  1351. window_id,
  1352. context: self.context.clone(),
  1353. };
  1354. Ok(DetachedWindow {
  1355. label,
  1356. dispatcher,
  1357. menu_ids,
  1358. })
  1359. }
  1360. fn run_on_main_thread<F: FnOnce() + Send + 'static>(&self, f: F) -> Result<()> {
  1361. send_user_message(&self.context, Message::Task(Box::new(f)))
  1362. }
  1363. #[cfg(all(windows, feature = "system-tray"))]
  1364. /// Deprecated. (not needed anymore)
  1365. fn remove_system_tray(&self) -> Result<()> {
  1366. Ok(())
  1367. }
  1368. }
  1369. impl Runtime for Wry {
  1370. type Dispatcher = WryDispatcher;
  1371. type Handle = WryHandle;
  1372. type GlobalShortcutManager = GlobalShortcutManagerHandle;
  1373. type ClipboardManager = ClipboardManagerWrapper;
  1374. #[cfg(feature = "system-tray")]
  1375. type TrayHandler = SystemTrayHandle;
  1376. fn new() -> Result<Self> {
  1377. let event_loop = EventLoop::<Message>::with_user_event();
  1378. let proxy = event_loop.create_proxy();
  1379. let main_thread_id = current_thread().id();
  1380. let web_context = WebContextStore::default();
  1381. let global_shortcut_manager = Arc::new(Mutex::new(WryShortcutManager::new(&event_loop)));
  1382. let clipboard_manager = Arc::new(Mutex::new(Clipboard::new()));
  1383. let windows = Arc::new(Mutex::new(HashMap::default()));
  1384. let window_event_listeners = WindowEventListeners::default();
  1385. let menu_event_listeners = MenuEventListeners::default();
  1386. #[cfg(feature = "system-tray")]
  1387. let tray_context = TrayContext::default();
  1388. let event_loop_context = Context {
  1389. main_thread_id,
  1390. proxy,
  1391. window_event_listeners: window_event_listeners.clone(),
  1392. menu_event_listeners: menu_event_listeners.clone(),
  1393. main_thread: DispatcherMainThreadContext {
  1394. window_target: event_loop.deref().clone(),
  1395. web_context: web_context.clone(),
  1396. global_shortcut_manager: global_shortcut_manager.clone(),
  1397. clipboard_manager: clipboard_manager.clone(),
  1398. windows: windows.clone(),
  1399. #[cfg(feature = "system-tray")]
  1400. tray_context: tray_context.clone(),
  1401. },
  1402. };
  1403. let global_shortcut_listeners = GlobalShortcutListeners::default();
  1404. let clipboard_manager_handle = ClipboardManagerWrapper {
  1405. context: event_loop_context.clone(),
  1406. };
  1407. Ok(Self {
  1408. main_thread_id,
  1409. global_shortcut_manager,
  1410. global_shortcut_manager_handle: GlobalShortcutManagerHandle {
  1411. context: event_loop_context,
  1412. shortcuts: Default::default(),
  1413. listeners: global_shortcut_listeners,
  1414. },
  1415. clipboard_manager,
  1416. clipboard_manager_handle,
  1417. event_loop,
  1418. windows,
  1419. web_context,
  1420. window_event_listeners,
  1421. menu_event_listeners,
  1422. #[cfg(feature = "system-tray")]
  1423. tray_context,
  1424. })
  1425. }
  1426. fn handle(&self) -> Self::Handle {
  1427. WryHandle {
  1428. context: Context {
  1429. main_thread_id: self.main_thread_id,
  1430. proxy: self.event_loop.create_proxy(),
  1431. window_event_listeners: self.window_event_listeners.clone(),
  1432. menu_event_listeners: self.menu_event_listeners.clone(),
  1433. main_thread: DispatcherMainThreadContext {
  1434. window_target: self.event_loop.deref().clone(),
  1435. web_context: self.web_context.clone(),
  1436. global_shortcut_manager: self.global_shortcut_manager.clone(),
  1437. clipboard_manager: self.clipboard_manager.clone(),
  1438. windows: self.windows.clone(),
  1439. #[cfg(feature = "system-tray")]
  1440. tray_context: self.tray_context.clone(),
  1441. },
  1442. },
  1443. }
  1444. }
  1445. fn global_shortcut_manager(&self) -> Self::GlobalShortcutManager {
  1446. self.global_shortcut_manager_handle.clone()
  1447. }
  1448. fn clipboard_manager(&self) -> Self::ClipboardManager {
  1449. self.clipboard_manager_handle.clone()
  1450. }
  1451. fn create_window(&self, pending: PendingWindow<Self>) -> Result<DetachedWindow<Self>> {
  1452. let label = pending.label.clone();
  1453. let menu_ids = pending.menu_ids.clone();
  1454. let proxy = self.event_loop.create_proxy();
  1455. let webview = create_webview(
  1456. &self.event_loop,
  1457. &self.web_context,
  1458. Context {
  1459. main_thread_id: self.main_thread_id,
  1460. proxy: proxy.clone(),
  1461. window_event_listeners: self.window_event_listeners.clone(),
  1462. menu_event_listeners: self.menu_event_listeners.clone(),
  1463. main_thread: DispatcherMainThreadContext {
  1464. window_target: self.event_loop.deref().clone(),
  1465. web_context: self.web_context.clone(),
  1466. global_shortcut_manager: self.global_shortcut_manager.clone(),
  1467. clipboard_manager: self.clipboard_manager.clone(),
  1468. windows: self.windows.clone(),
  1469. #[cfg(feature = "system-tray")]
  1470. tray_context: self.tray_context.clone(),
  1471. },
  1472. },
  1473. pending,
  1474. )?;
  1475. #[cfg(target_os = "windows")]
  1476. {
  1477. let id = webview.inner.window().id();
  1478. if let WindowHandle::Webview(ref webview) = webview.inner {
  1479. if let Some(controller) = webview.controller() {
  1480. let proxy = self.event_loop.create_proxy();
  1481. let mut token = EventRegistrationToken::default();
  1482. unsafe {
  1483. controller.GotFocus(
  1484. FocusChangedEventHandler::create(Box::new(move |_, _| {
  1485. let _ = proxy.send_event(Message::Webview(
  1486. id,
  1487. WebviewMessage::WebviewEvent(WebviewEvent::Focused(true)),
  1488. ));
  1489. Ok(())
  1490. })),
  1491. &mut token,
  1492. )
  1493. }
  1494. .unwrap();
  1495. let proxy = self.event_loop.create_proxy();
  1496. unsafe {
  1497. controller.LostFocus(
  1498. FocusChangedEventHandler::create(Box::new(move |_, _| {
  1499. let _ = proxy.send_event(Message::Webview(
  1500. id,
  1501. WebviewMessage::WebviewEvent(WebviewEvent::Focused(false)),
  1502. ));
  1503. Ok(())
  1504. })),
  1505. &mut token,
  1506. )
  1507. }
  1508. .unwrap();
  1509. }
  1510. }
  1511. }
  1512. let dispatcher = WryDispatcher {
  1513. window_id: webview.inner.window().id(),
  1514. context: Context {
  1515. main_thread_id: self.main_thread_id,
  1516. proxy,
  1517. window_event_listeners: self.window_event_listeners.clone(),
  1518. menu_event_listeners: self.menu_event_listeners.clone(),
  1519. main_thread: DispatcherMainThreadContext {
  1520. window_target: self.event_loop.deref().clone(),
  1521. web_context: self.web_context.clone(),
  1522. global_shortcut_manager: self.global_shortcut_manager.clone(),
  1523. clipboard_manager: self.clipboard_manager.clone(),
  1524. windows: self.windows.clone(),
  1525. #[cfg(feature = "system-tray")]
  1526. tray_context: self.tray_context.clone(),
  1527. },
  1528. },
  1529. };
  1530. self
  1531. .windows
  1532. .lock()
  1533. .unwrap()
  1534. .insert(webview.inner.window().id(), webview);
  1535. Ok(DetachedWindow {
  1536. label,
  1537. dispatcher,
  1538. menu_ids,
  1539. })
  1540. }
  1541. #[cfg(feature = "system-tray")]
  1542. fn system_tray(&self, system_tray: SystemTray) -> Result<Self::TrayHandler> {
  1543. let icon = system_tray
  1544. .icon
  1545. .expect("tray icon not set")
  1546. .into_tray_icon();
  1547. let mut items = HashMap::new();
  1548. #[cfg(target_os = "macos")]
  1549. let tray = SystemTrayBuilder::new(
  1550. icon,
  1551. system_tray
  1552. .menu
  1553. .map(|menu| to_wry_context_menu(&mut items, menu)),
  1554. )
  1555. .with_icon_as_template(system_tray.icon_as_template)
  1556. .build(&self.event_loop)
  1557. .map_err(|e| Error::SystemTray(Box::new(e)))?;
  1558. #[cfg(not(target_os = "macos"))]
  1559. let tray = SystemTrayBuilder::new(
  1560. icon,
  1561. system_tray
  1562. .menu
  1563. .map(|menu| to_wry_context_menu(&mut items, menu)),
  1564. )
  1565. .build(&self.event_loop)
  1566. .map_err(|e| Error::SystemTray(Box::new(e)))?;
  1567. *self.tray_context.items.lock().unwrap() = items;
  1568. *self.tray_context.tray.lock().unwrap() = Some(Arc::new(Mutex::new(tray)));
  1569. Ok(SystemTrayHandle {
  1570. proxy: self.event_loop.create_proxy(),
  1571. })
  1572. }
  1573. #[cfg(feature = "system-tray")]
  1574. fn on_system_tray_event<F: Fn(&SystemTrayEvent) + Send + 'static>(&mut self, f: F) -> Uuid {
  1575. let id = Uuid::new_v4();
  1576. self
  1577. .tray_context
  1578. .listeners
  1579. .lock()
  1580. .unwrap()
  1581. .insert(id, Box::new(f));
  1582. id
  1583. }
  1584. #[cfg(target_os = "macos")]
  1585. fn set_activation_policy(&mut self, activation_policy: ActivationPolicy) {
  1586. self
  1587. .event_loop
  1588. .set_activation_policy(match activation_policy {
  1589. ActivationPolicy::Regular => WryActivationPolicy::Regular,
  1590. ActivationPolicy::Accessory => WryActivationPolicy::Accessory,
  1591. ActivationPolicy::Prohibited => WryActivationPolicy::Prohibited,
  1592. _ => unimplemented!(),
  1593. });
  1594. }
  1595. #[cfg(any(target_os = "windows", target_os = "macos"))]
  1596. fn run_iteration<F: FnMut(RunEvent) + 'static>(&mut self, mut callback: F) -> RunIteration {
  1597. use wry::application::platform::run_return::EventLoopExtRunReturn;
  1598. let windows = self.windows.clone();
  1599. let web_context = &self.web_context;
  1600. let window_event_listeners = self.window_event_listeners.clone();
  1601. let menu_event_listeners = self.menu_event_listeners.clone();
  1602. #[cfg(feature = "system-tray")]
  1603. let tray_context = self.tray_context.clone();
  1604. let global_shortcut_manager = self.global_shortcut_manager.clone();
  1605. let global_shortcut_manager_handle = self.global_shortcut_manager_handle.clone();
  1606. let clipboard_manager = self.clipboard_manager.clone();
  1607. let mut iteration = RunIteration::default();
  1608. self
  1609. .event_loop
  1610. .run_return(|event, event_loop, control_flow| {
  1611. *control_flow = ControlFlow::Wait;
  1612. if let Event::MainEventsCleared = &event {
  1613. *control_flow = ControlFlow::Exit;
  1614. }
  1615. iteration = handle_event_loop(
  1616. event,
  1617. event_loop,
  1618. control_flow,
  1619. EventLoopIterationContext {
  1620. callback: &mut callback,
  1621. windows: windows.clone(),
  1622. window_event_listeners: &window_event_listeners,
  1623. global_shortcut_manager: global_shortcut_manager.clone(),
  1624. global_shortcut_manager_handle: &global_shortcut_manager_handle,
  1625. clipboard_manager: clipboard_manager.clone(),
  1626. menu_event_listeners: &menu_event_listeners,
  1627. #[cfg(feature = "system-tray")]
  1628. tray_context: &tray_context,
  1629. },
  1630. web_context,
  1631. );
  1632. });
  1633. iteration
  1634. }
  1635. fn run<F: FnMut(RunEvent) + 'static>(self, mut callback: F) {
  1636. let windows = self.windows.clone();
  1637. let web_context = self.web_context;
  1638. let window_event_listeners = self.window_event_listeners.clone();
  1639. let menu_event_listeners = self.menu_event_listeners.clone();
  1640. #[cfg(feature = "system-tray")]
  1641. let tray_context = self.tray_context;
  1642. let global_shortcut_manager = self.global_shortcut_manager.clone();
  1643. let global_shortcut_manager_handle = self.global_shortcut_manager_handle.clone();
  1644. let clipboard_manager = self.clipboard_manager.clone();
  1645. self.event_loop.run(move |event, event_loop, control_flow| {
  1646. handle_event_loop(
  1647. event,
  1648. event_loop,
  1649. control_flow,
  1650. EventLoopIterationContext {
  1651. callback: &mut callback,
  1652. windows: windows.clone(),
  1653. window_event_listeners: &window_event_listeners,
  1654. global_shortcut_manager: global_shortcut_manager.clone(),
  1655. global_shortcut_manager_handle: &global_shortcut_manager_handle,
  1656. clipboard_manager: clipboard_manager.clone(),
  1657. menu_event_listeners: &menu_event_listeners,
  1658. #[cfg(feature = "system-tray")]
  1659. tray_context: &tray_context,
  1660. },
  1661. &web_context,
  1662. );
  1663. })
  1664. }
  1665. }
  1666. struct EventLoopIterationContext<'a> {
  1667. callback: &'a mut (dyn FnMut(RunEvent) + 'static),
  1668. windows: Arc<Mutex<HashMap<WindowId, WindowWrapper>>>,
  1669. window_event_listeners: &'a WindowEventListeners,
  1670. global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
  1671. global_shortcut_manager_handle: &'a GlobalShortcutManagerHandle,
  1672. clipboard_manager: Arc<Mutex<Clipboard>>,
  1673. menu_event_listeners: &'a MenuEventListeners,
  1674. #[cfg(feature = "system-tray")]
  1675. tray_context: &'a TrayContext,
  1676. }
  1677. struct UserMessageContext<'a> {
  1678. window_event_listeners: &'a WindowEventListeners,
  1679. global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
  1680. clipboard_manager: Arc<Mutex<Clipboard>>,
  1681. menu_event_listeners: &'a MenuEventListeners,
  1682. windows: Arc<Mutex<HashMap<WindowId, WindowWrapper>>>,
  1683. #[cfg(feature = "system-tray")]
  1684. tray_context: &'a TrayContext,
  1685. }
  1686. fn handle_user_message(
  1687. event_loop: &EventLoopWindowTarget<Message>,
  1688. message: Message,
  1689. context: UserMessageContext<'_>,
  1690. web_context: &WebContextStore,
  1691. ) -> RunIteration {
  1692. let UserMessageContext {
  1693. window_event_listeners,
  1694. menu_event_listeners,
  1695. global_shortcut_manager,
  1696. clipboard_manager,
  1697. windows,
  1698. #[cfg(feature = "system-tray")]
  1699. tray_context,
  1700. } = context;
  1701. match message {
  1702. Message::Task(task) => task(),
  1703. Message::Window(id, window_message) => {
  1704. if let Some(webview) = windows
  1705. .lock()
  1706. .expect("poisoned webview collection")
  1707. .get_mut(&id)
  1708. {
  1709. let window = webview.inner.window();
  1710. match window_message {
  1711. // Getters
  1712. WindowMessage::ScaleFactor(tx) => tx.send(window.scale_factor()).unwrap(),
  1713. WindowMessage::InnerPosition(tx) => tx
  1714. .send(
  1715. window
  1716. .inner_position()
  1717. .map(|p| PhysicalPositionWrapper(p).into())
  1718. .map_err(|_| Error::FailedToSendMessage),
  1719. )
  1720. .unwrap(),
  1721. WindowMessage::OuterPosition(tx) => tx
  1722. .send(
  1723. window
  1724. .outer_position()
  1725. .map(|p| PhysicalPositionWrapper(p).into())
  1726. .map_err(|_| Error::FailedToSendMessage),
  1727. )
  1728. .unwrap(),
  1729. WindowMessage::InnerSize(tx) => tx
  1730. .send(PhysicalSizeWrapper(webview.inner.inner_size()).into())
  1731. .unwrap(),
  1732. WindowMessage::OuterSize(tx) => tx
  1733. .send(PhysicalSizeWrapper(window.outer_size()).into())
  1734. .unwrap(),
  1735. WindowMessage::IsFullscreen(tx) => tx.send(window.fullscreen().is_some()).unwrap(),
  1736. WindowMessage::IsMaximized(tx) => tx.send(window.is_maximized()).unwrap(),
  1737. WindowMessage::IsDecorated(tx) => tx.send(window.is_decorated()).unwrap(),
  1738. WindowMessage::IsResizable(tx) => tx.send(window.is_resizable()).unwrap(),
  1739. WindowMessage::IsVisible(tx) => tx.send(window.is_visible()).unwrap(),
  1740. WindowMessage::IsMenuVisible(tx) => tx.send(window.is_menu_visible()).unwrap(),
  1741. WindowMessage::CurrentMonitor(tx) => tx.send(window.current_monitor()).unwrap(),
  1742. WindowMessage::PrimaryMonitor(tx) => tx.send(window.primary_monitor()).unwrap(),
  1743. WindowMessage::AvailableMonitors(tx) => {
  1744. tx.send(window.available_monitors().collect()).unwrap()
  1745. }
  1746. #[cfg(target_os = "macos")]
  1747. WindowMessage::NSWindow(tx) => tx.send(NSWindow(window.ns_window())).unwrap(),
  1748. #[cfg(windows)]
  1749. WindowMessage::Hwnd(tx) => tx.send(Hwnd(HWND(window.hwnd() as _))).unwrap(),
  1750. #[cfg(any(
  1751. target_os = "linux",
  1752. target_os = "dragonfly",
  1753. target_os = "freebsd",
  1754. target_os = "netbsd",
  1755. target_os = "openbsd"
  1756. ))]
  1757. WindowMessage::GtkWindow(tx) => tx.send(GtkWindow(window.gtk_window().clone())).unwrap(),
  1758. // Setters
  1759. WindowMessage::Center(tx) => {
  1760. tx.send(center_window(window, webview.inner.inner_size()))
  1761. .unwrap();
  1762. }
  1763. WindowMessage::RequestUserAttention(request_type) => {
  1764. window.request_user_attention(request_type.map(|r| r.0));
  1765. }
  1766. WindowMessage::SetResizable(resizable) => window.set_resizable(resizable),
  1767. WindowMessage::SetTitle(title) => window.set_title(&title),
  1768. WindowMessage::Maximize => window.set_maximized(true),
  1769. WindowMessage::Unmaximize => window.set_maximized(false),
  1770. WindowMessage::Minimize => window.set_minimized(true),
  1771. WindowMessage::Unminimize => window.set_minimized(false),
  1772. WindowMessage::ShowMenu => window.show_menu(),
  1773. WindowMessage::HideMenu => window.hide_menu(),
  1774. WindowMessage::Show => window.set_visible(true),
  1775. WindowMessage::Hide => window.set_visible(false),
  1776. WindowMessage::Close => panic!("cannot handle `WindowMessage::Close` on the main thread"),
  1777. WindowMessage::SetDecorations(decorations) => window.set_decorations(decorations),
  1778. WindowMessage::SetAlwaysOnTop(always_on_top) => window.set_always_on_top(always_on_top),
  1779. WindowMessage::SetSize(size) => {
  1780. window.set_inner_size(SizeWrapper::from(size).0);
  1781. }
  1782. WindowMessage::SetMinSize(size) => {
  1783. window.set_min_inner_size(size.map(|s| SizeWrapper::from(s).0));
  1784. }
  1785. WindowMessage::SetMaxSize(size) => {
  1786. window.set_max_inner_size(size.map(|s| SizeWrapper::from(s).0));
  1787. }
  1788. WindowMessage::SetPosition(position) => {
  1789. window.set_outer_position(PositionWrapper::from(position).0)
  1790. }
  1791. WindowMessage::SetFullscreen(fullscreen) => {
  1792. if fullscreen {
  1793. window.set_fullscreen(Some(Fullscreen::Borderless(None)))
  1794. } else {
  1795. window.set_fullscreen(None)
  1796. }
  1797. }
  1798. WindowMessage::SetFocus => {
  1799. window.set_focus();
  1800. }
  1801. WindowMessage::SetIcon(icon) => {
  1802. window.set_window_icon(Some(icon));
  1803. }
  1804. WindowMessage::SetSkipTaskbar(_skip) => {
  1805. #[cfg(any(target_os = "windows", target_os = "linux"))]
  1806. window.set_skip_taskbar(_skip);
  1807. }
  1808. WindowMessage::DragWindow => {
  1809. let _ = window.drag_window();
  1810. }
  1811. WindowMessage::UpdateMenuItem(id, update) => {
  1812. if let Some(menu_items) = webview.menu_items.as_mut() {
  1813. let item = menu_items.get_mut(&id).expect("menu item not found");
  1814. match update {
  1815. MenuUpdate::SetEnabled(enabled) => item.set_enabled(enabled),
  1816. MenuUpdate::SetTitle(title) => item.set_title(&title),
  1817. MenuUpdate::SetSelected(selected) => item.set_selected(selected),
  1818. #[cfg(target_os = "macos")]
  1819. MenuUpdate::SetNativeImage(image) => {
  1820. item.set_native_image(NativeImageWrapper::from(image).0)
  1821. }
  1822. }
  1823. }
  1824. }
  1825. }
  1826. }
  1827. }
  1828. Message::Webview(id, webview_message) => match webview_message {
  1829. WebviewMessage::EvaluateScript(script) => {
  1830. if let Some(WindowHandle::Webview(webview)) = windows
  1831. .lock()
  1832. .expect("poisoned webview collection")
  1833. .get(&id)
  1834. .map(|w| &w.inner)
  1835. {
  1836. if let Err(e) = webview.evaluate_script(&script) {
  1837. eprintln!("{}", e);
  1838. }
  1839. }
  1840. }
  1841. WebviewMessage::Print => {
  1842. if let Some(WindowHandle::Webview(webview)) = windows
  1843. .lock()
  1844. .expect("poisoned webview collection")
  1845. .get(&id)
  1846. .map(|w| &w.inner)
  1847. {
  1848. let _ = webview.print();
  1849. }
  1850. }
  1851. WebviewMessage::WebviewEvent(event) => {
  1852. if let Some(event) = WindowEventWrapper::from(&event).0 {
  1853. for handler in window_event_listeners
  1854. .lock()
  1855. .unwrap()
  1856. .get(&id)
  1857. .unwrap()
  1858. .lock()
  1859. .unwrap()
  1860. .values()
  1861. {
  1862. handler(&event);
  1863. }
  1864. }
  1865. }
  1866. },
  1867. Message::CreateWebview(handler, sender) => match handler(event_loop, web_context) {
  1868. Ok(webview) => {
  1869. let window_id = webview.inner.window().id();
  1870. windows
  1871. .lock()
  1872. .expect("poisoned webview collection")
  1873. .insert(window_id, webview);
  1874. sender.send(window_id).unwrap();
  1875. }
  1876. Err(e) => {
  1877. eprintln!("{}", e);
  1878. }
  1879. },
  1880. Message::CreateWindow(handler, sender) => {
  1881. let (label, builder) = handler();
  1882. if let Ok(window) = builder.build(event_loop) {
  1883. let window_id = window.id();
  1884. window_event_listeners
  1885. .lock()
  1886. .unwrap()
  1887. .insert(window.id(), WindowEventListenersMap::default());
  1888. menu_event_listeners
  1889. .lock()
  1890. .unwrap()
  1891. .insert(window.id(), WindowMenuEventListeners::default());
  1892. let w = Arc::new(window);
  1893. windows.lock().expect("poisoned webview collection").insert(
  1894. window_id,
  1895. WindowWrapper {
  1896. label,
  1897. inner: WindowHandle::Window(w.clone()),
  1898. menu_items: Default::default(),
  1899. },
  1900. );
  1901. sender.send(Ok(Arc::downgrade(&w))).unwrap();
  1902. } else {
  1903. sender.send(Err(Error::CreateWindow)).unwrap();
  1904. }
  1905. }
  1906. #[cfg(feature = "system-tray")]
  1907. Message::Tray(tray_message) => match tray_message {
  1908. TrayMessage::UpdateItem(menu_id, update) => {
  1909. let mut tray = tray_context.items.as_ref().lock().unwrap();
  1910. let item = tray.get_mut(&menu_id).expect("menu item not found");
  1911. match update {
  1912. MenuUpdate::SetEnabled(enabled) => item.set_enabled(enabled),
  1913. MenuUpdate::SetTitle(title) => item.set_title(&title),
  1914. MenuUpdate::SetSelected(selected) => item.set_selected(selected),
  1915. #[cfg(target_os = "macos")]
  1916. MenuUpdate::SetNativeImage(image) => {
  1917. item.set_native_image(NativeImageWrapper::from(image).0)
  1918. }
  1919. }
  1920. }
  1921. TrayMessage::UpdateMenu(menu) => {
  1922. if let Some(tray) = &*tray_context.tray.lock().unwrap() {
  1923. let mut items = HashMap::new();
  1924. tray
  1925. .lock()
  1926. .unwrap()
  1927. .set_menu(&to_wry_context_menu(&mut items, menu));
  1928. *tray_context.items.lock().unwrap() = items;
  1929. }
  1930. }
  1931. TrayMessage::UpdateIcon(icon) => {
  1932. if let Some(tray) = &*tray_context.tray.lock().unwrap() {
  1933. tray.lock().unwrap().set_icon(icon.into_tray_icon());
  1934. }
  1935. }
  1936. #[cfg(target_os = "macos")]
  1937. TrayMessage::UpdateIconAsTemplate(is_template) => {
  1938. if let Some(tray) = &*tray_context.tray.lock().unwrap() {
  1939. tray.lock().unwrap().set_icon_as_template(is_template);
  1940. }
  1941. }
  1942. },
  1943. Message::GlobalShortcut(message) => match message {
  1944. GlobalShortcutMessage::IsRegistered(accelerator, tx) => tx
  1945. .send(
  1946. global_shortcut_manager
  1947. .lock()
  1948. .unwrap()
  1949. .is_registered(&accelerator),
  1950. )
  1951. .unwrap(),
  1952. GlobalShortcutMessage::Register(accelerator, tx) => tx
  1953. .send(
  1954. global_shortcut_manager
  1955. .lock()
  1956. .unwrap()
  1957. .register(accelerator)
  1958. .map(GlobalShortcutWrapper)
  1959. .map_err(|e| Error::GlobalShortcut(Box::new(e))),
  1960. )
  1961. .unwrap(),
  1962. GlobalShortcutMessage::Unregister(shortcut, tx) => tx
  1963. .send(
  1964. global_shortcut_manager
  1965. .lock()
  1966. .unwrap()
  1967. .unregister(shortcut.0)
  1968. .map_err(|e| Error::GlobalShortcut(Box::new(e))),
  1969. )
  1970. .unwrap(),
  1971. GlobalShortcutMessage::UnregisterAll(tx) => tx
  1972. .send(
  1973. global_shortcut_manager
  1974. .lock()
  1975. .unwrap()
  1976. .unregister_all()
  1977. .map_err(|e| Error::GlobalShortcut(Box::new(e))),
  1978. )
  1979. .unwrap(),
  1980. },
  1981. Message::Clipboard(message) => match message {
  1982. ClipboardMessage::WriteText(text, tx) => {
  1983. clipboard_manager.lock().unwrap().write_text(text);
  1984. tx.send(()).unwrap();
  1985. }
  1986. ClipboardMessage::ReadText(tx) => tx
  1987. .send(clipboard_manager.lock().unwrap().read_text())
  1988. .unwrap(),
  1989. },
  1990. }
  1991. let it = RunIteration {
  1992. window_count: windows.lock().expect("poisoned webview collection").len(),
  1993. };
  1994. it
  1995. }
  1996. fn handle_event_loop(
  1997. event: Event<'_, Message>,
  1998. event_loop: &EventLoopWindowTarget<Message>,
  1999. control_flow: &mut ControlFlow,
  2000. context: EventLoopIterationContext<'_>,
  2001. web_context: &WebContextStore,
  2002. ) -> RunIteration {
  2003. let EventLoopIterationContext {
  2004. callback,
  2005. windows,
  2006. window_event_listeners,
  2007. global_shortcut_manager,
  2008. global_shortcut_manager_handle,
  2009. clipboard_manager,
  2010. menu_event_listeners,
  2011. #[cfg(feature = "system-tray")]
  2012. tray_context,
  2013. } = context;
  2014. if *control_flow == ControlFlow::Exit {
  2015. return RunIteration {
  2016. window_count: windows.lock().expect("poisoned webview collection").len(),
  2017. };
  2018. }
  2019. *control_flow = ControlFlow::Wait;
  2020. match event {
  2021. Event::NewEvents(StartCause::Init) => {
  2022. callback(RunEvent::Ready);
  2023. }
  2024. Event::NewEvents(StartCause::Poll) => {
  2025. callback(RunEvent::Resumed);
  2026. }
  2027. Event::MainEventsCleared => {
  2028. callback(RunEvent::MainEventsCleared);
  2029. }
  2030. Event::GlobalShortcutEvent(accelerator_id) => {
  2031. for (id, handler) in &*global_shortcut_manager_handle.listeners.lock().unwrap() {
  2032. if accelerator_id == *id {
  2033. handler();
  2034. }
  2035. }
  2036. }
  2037. Event::MenuEvent {
  2038. window_id,
  2039. menu_id,
  2040. origin: MenuType::MenuBar,
  2041. ..
  2042. } => {
  2043. let window_id = window_id.unwrap(); // always Some on MenuBar event
  2044. let event = MenuEvent {
  2045. menu_item_id: menu_id.0,
  2046. };
  2047. let listeners = menu_event_listeners.lock().unwrap();
  2048. let window_menu_event_listeners = listeners.get(&window_id).cloned().unwrap_or_default();
  2049. for handler in window_menu_event_listeners.lock().unwrap().values() {
  2050. handler(&event);
  2051. }
  2052. }
  2053. #[cfg(feature = "system-tray")]
  2054. Event::MenuEvent {
  2055. window_id: _,
  2056. menu_id,
  2057. origin: MenuType::ContextMenu,
  2058. ..
  2059. } => {
  2060. let event = SystemTrayEvent::MenuItemClick(menu_id.0);
  2061. for handler in tray_context.listeners.lock().unwrap().values() {
  2062. handler(&event);
  2063. }
  2064. }
  2065. #[cfg(feature = "system-tray")]
  2066. Event::TrayEvent {
  2067. bounds,
  2068. event,
  2069. position: _cursor_position,
  2070. ..
  2071. } => {
  2072. let (position, size) = (
  2073. PhysicalPositionWrapper(bounds.position).into(),
  2074. PhysicalSizeWrapper(bounds.size).into(),
  2075. );
  2076. let event = match event {
  2077. TrayEvent::RightClick => SystemTrayEvent::RightClick { position, size },
  2078. TrayEvent::DoubleClick => SystemTrayEvent::DoubleClick { position, size },
  2079. // default to left click
  2080. _ => SystemTrayEvent::LeftClick { position, size },
  2081. };
  2082. for handler in tray_context.listeners.lock().unwrap().values() {
  2083. handler(&event);
  2084. }
  2085. }
  2086. Event::WindowEvent {
  2087. event, window_id, ..
  2088. } => {
  2089. // NOTE(amrbashir): we handle this event here instead of `match` statement below because
  2090. // we want to focus the webview as soon as possible, especially on windows.
  2091. if event == WryWindowEvent::Focused(true) {
  2092. if let Some(WindowHandle::Webview(webview)) = windows
  2093. .lock()
  2094. .expect("poisoned webview collection")
  2095. .get(&window_id)
  2096. .map(|w| &w.inner)
  2097. {
  2098. webview.focus();
  2099. }
  2100. }
  2101. {
  2102. let windows_lock = windows.lock().expect("poisoned webview collection");
  2103. if let Some(window_handle) = windows_lock.get(&window_id).map(|w| &w.inner) {
  2104. if let Some(event) = WindowEventWrapper::parse(window_handle, &event).0 {
  2105. drop(windows_lock);
  2106. for handler in window_event_listeners
  2107. .lock()
  2108. .unwrap()
  2109. .get(&window_id)
  2110. .unwrap()
  2111. .lock()
  2112. .unwrap()
  2113. .values()
  2114. {
  2115. handler(&event);
  2116. }
  2117. }
  2118. }
  2119. }
  2120. match event {
  2121. WryWindowEvent::CloseRequested => {
  2122. let (tx, rx) = channel();
  2123. let windows_guard = windows.lock().expect("poisoned webview collection");
  2124. if let Some(w) = windows_guard.get(&window_id) {
  2125. let label = w.label.clone();
  2126. drop(windows_guard);
  2127. callback(RunEvent::CloseRequested {
  2128. label,
  2129. signal_tx: tx,
  2130. });
  2131. if let Ok(true) = rx.try_recv() {
  2132. } else {
  2133. on_window_close(
  2134. callback,
  2135. window_id,
  2136. windows.lock().expect("poisoned webview collection"),
  2137. control_flow,
  2138. #[cfg(target_os = "linux")]
  2139. window_event_listeners,
  2140. menu_event_listeners.clone(),
  2141. );
  2142. }
  2143. }
  2144. }
  2145. WryWindowEvent::Resized(_) => {
  2146. if let Some(WindowHandle::Webview(webview)) = windows
  2147. .lock()
  2148. .expect("poisoned webview collection")
  2149. .get(&window_id)
  2150. .map(|w| &w.inner)
  2151. {
  2152. if let Err(e) = webview.resize() {
  2153. eprintln!("{}", e);
  2154. }
  2155. }
  2156. }
  2157. _ => {}
  2158. }
  2159. }
  2160. Event::UserEvent(message) => {
  2161. if let Message::Window(id, WindowMessage::Close) = message {
  2162. on_window_close(
  2163. callback,
  2164. id,
  2165. windows.lock().expect("poisoned webview collection"),
  2166. control_flow,
  2167. #[cfg(target_os = "linux")]
  2168. window_event_listeners,
  2169. menu_event_listeners.clone(),
  2170. );
  2171. } else {
  2172. return handle_user_message(
  2173. event_loop,
  2174. message,
  2175. UserMessageContext {
  2176. window_event_listeners,
  2177. global_shortcut_manager,
  2178. clipboard_manager,
  2179. menu_event_listeners,
  2180. windows,
  2181. #[cfg(feature = "system-tray")]
  2182. tray_context,
  2183. },
  2184. web_context,
  2185. );
  2186. }
  2187. }
  2188. _ => (),
  2189. }
  2190. let it = RunIteration {
  2191. window_count: windows.lock().expect("poisoned webview collection").len(),
  2192. };
  2193. it
  2194. }
  2195. fn on_window_close<'a>(
  2196. callback: &'a mut (dyn FnMut(RunEvent) + 'static),
  2197. window_id: WindowId,
  2198. mut windows: MutexGuard<'a, HashMap<WindowId, WindowWrapper>>,
  2199. control_flow: &mut ControlFlow,
  2200. #[cfg(target_os = "linux")] window_event_listeners: &WindowEventListeners,
  2201. menu_event_listeners: MenuEventListeners,
  2202. ) {
  2203. if let Some(webview) = windows.remove(&window_id) {
  2204. let is_empty = windows.is_empty();
  2205. drop(windows);
  2206. menu_event_listeners.lock().unwrap().remove(&window_id);
  2207. callback(RunEvent::WindowClose(webview.label.clone()));
  2208. if is_empty {
  2209. let (tx, rx) = channel();
  2210. callback(RunEvent::ExitRequested {
  2211. window_label: webview.label,
  2212. tx,
  2213. });
  2214. let recv = rx.try_recv();
  2215. let should_prevent = matches!(recv, Ok(ExitRequestedEventAction::Prevent));
  2216. if !should_prevent {
  2217. *control_flow = ControlFlow::Exit;
  2218. callback(RunEvent::Exit);
  2219. }
  2220. }
  2221. }
  2222. // TODO: tao does not fire the destroyed event properly
  2223. #[cfg(target_os = "linux")]
  2224. {
  2225. for handler in window_event_listeners
  2226. .lock()
  2227. .unwrap()
  2228. .get(&window_id)
  2229. .unwrap()
  2230. .lock()
  2231. .unwrap()
  2232. .values()
  2233. {
  2234. handler(&WindowEvent::Destroyed);
  2235. }
  2236. }
  2237. }
  2238. fn center_window(window: &Window, window_size: WryPhysicalSize<u32>) -> Result<()> {
  2239. if let Some(monitor) = window.current_monitor() {
  2240. let screen_size = monitor.size();
  2241. let x = (screen_size.width - window_size.width) / 2;
  2242. let y = (screen_size.height - window_size.height) / 2;
  2243. window.set_outer_position(WryPhysicalPosition::new(x, y));
  2244. Ok(())
  2245. } else {
  2246. Err(Error::FailedToGetMonitor)
  2247. }
  2248. }
  2249. fn to_wry_menu(
  2250. custom_menu_items: &mut HashMap<MenuHash, WryCustomMenuItem>,
  2251. menu: Menu,
  2252. ) -> MenuBar {
  2253. let mut wry_menu = MenuBar::new();
  2254. for item in menu.items {
  2255. match item {
  2256. MenuEntry::CustomItem(c) => {
  2257. let mut attributes = MenuItemAttributesWrapper::from(&c).0;
  2258. attributes = attributes.with_id(WryMenuId(c.id));
  2259. #[allow(unused_mut)]
  2260. let mut item = wry_menu.add_item(attributes);
  2261. #[cfg(target_os = "macos")]
  2262. if let Some(native_image) = c.native_image {
  2263. item.set_native_image(NativeImageWrapper::from(native_image).0);
  2264. }
  2265. custom_menu_items.insert(c.id, item);
  2266. }
  2267. MenuEntry::NativeItem(i) => {
  2268. wry_menu.add_native_item(MenuItemWrapper::from(i).0);
  2269. }
  2270. MenuEntry::Submenu(submenu) => {
  2271. wry_menu.add_submenu(
  2272. &submenu.title,
  2273. submenu.enabled,
  2274. to_wry_menu(custom_menu_items, submenu.inner),
  2275. );
  2276. }
  2277. }
  2278. }
  2279. wry_menu
  2280. }
  2281. fn create_webview(
  2282. event_loop: &EventLoopWindowTarget<Message>,
  2283. web_context: &WebContextStore,
  2284. context: Context,
  2285. pending: PendingWindow<Wry>,
  2286. ) -> Result<WindowWrapper> {
  2287. #[allow(unused_mut)]
  2288. let PendingWindow {
  2289. webview_attributes,
  2290. uri_scheme_protocols,
  2291. mut window_builder,
  2292. rpc_handler,
  2293. file_drop_handler,
  2294. label,
  2295. url,
  2296. menu_ids,
  2297. ..
  2298. } = pending;
  2299. let is_window_transparent = window_builder.inner.window.transparent;
  2300. let menu_items = if let Some(menu) = window_builder.menu {
  2301. let mut menu_items = HashMap::new();
  2302. let menu = to_wry_menu(&mut menu_items, menu);
  2303. window_builder.inner = window_builder.inner.with_menu(menu);
  2304. Some(menu_items)
  2305. } else {
  2306. None
  2307. };
  2308. let window = window_builder.inner.build(event_loop).unwrap();
  2309. context
  2310. .window_event_listeners
  2311. .lock()
  2312. .unwrap()
  2313. .insert(window.id(), WindowEventListenersMap::default());
  2314. context
  2315. .menu_event_listeners
  2316. .lock()
  2317. .unwrap()
  2318. .insert(window.id(), WindowMenuEventListeners::default());
  2319. if window_builder.center {
  2320. let _ = center_window(&window, window.inner_size());
  2321. }
  2322. let mut webview_builder = WebViewBuilder::new(window)
  2323. .map_err(|e| Error::CreateWebview(Box::new(e)))?
  2324. .with_url(&url)
  2325. .unwrap() // safe to unwrap because we validate the URL beforehand
  2326. .with_transparent(is_window_transparent);
  2327. if let Some(handler) = rpc_handler {
  2328. webview_builder = webview_builder.with_rpc_handler(create_rpc_handler(
  2329. context.clone(),
  2330. label.clone(),
  2331. menu_ids.clone(),
  2332. handler,
  2333. ));
  2334. }
  2335. if let Some(handler) = file_drop_handler {
  2336. webview_builder = webview_builder.with_file_drop_handler(create_file_drop_handler(
  2337. context,
  2338. label.clone(),
  2339. menu_ids,
  2340. handler,
  2341. ));
  2342. }
  2343. for (scheme, protocol) in uri_scheme_protocols {
  2344. webview_builder = webview_builder.with_custom_protocol(scheme, move |wry_request| {
  2345. protocol(&HttpRequestWrapper::from(wry_request).0)
  2346. .map(|tauri_response| HttpResponseWrapper::from(tauri_response).0)
  2347. .map_err(|_| wry::Error::InitScriptError)
  2348. });
  2349. }
  2350. for script in webview_attributes.initialization_scripts {
  2351. webview_builder = webview_builder.with_initialization_script(&script);
  2352. }
  2353. let mut web_context = web_context.lock().expect("poisoned WebContext store");
  2354. let is_first_context = web_context.is_empty();
  2355. let automation_enabled = std::env::var("TAURI_AUTOMATION").as_deref() == Ok("true");
  2356. let web_context = match web_context.entry(
  2357. // force a unique WebContext when automation is false;
  2358. // the context must be stored on the HashMap because it must outlive the WebView on macOS
  2359. if automation_enabled {
  2360. webview_attributes.data_directory.clone()
  2361. } else {
  2362. // random unique key
  2363. Some(Uuid::new_v4().to_hyphenated().to_string().into())
  2364. },
  2365. ) {
  2366. Occupied(occupied) => occupied.into_mut(),
  2367. Vacant(vacant) => {
  2368. let mut web_context = WebContext::new(webview_attributes.data_directory);
  2369. web_context.set_allows_automation(if automation_enabled {
  2370. is_first_context
  2371. } else {
  2372. false
  2373. });
  2374. vacant.insert(web_context)
  2375. }
  2376. };
  2377. let webview = webview_builder
  2378. .with_web_context(web_context)
  2379. .build()
  2380. .map_err(|e| Error::CreateWebview(Box::new(e)))?;
  2381. Ok(WindowWrapper {
  2382. label,
  2383. inner: WindowHandle::Webview(webview),
  2384. menu_items,
  2385. })
  2386. }
  2387. /// Create a wry rpc handler from a tauri rpc handler.
  2388. fn create_rpc_handler(
  2389. context: Context,
  2390. label: String,
  2391. menu_ids: HashMap<MenuHash, MenuId>,
  2392. handler: WebviewRpcHandler<Wry>,
  2393. ) -> Box<dyn Fn(&Window, WryRpcRequest) -> Option<RpcResponse> + 'static> {
  2394. Box::new(move |window, request| {
  2395. handler(
  2396. DetachedWindow {
  2397. dispatcher: WryDispatcher {
  2398. window_id: window.id(),
  2399. context: context.clone(),
  2400. },
  2401. label: label.clone(),
  2402. menu_ids: menu_ids.clone(),
  2403. },
  2404. RpcRequestWrapper(request).into(),
  2405. );
  2406. None
  2407. })
  2408. }
  2409. /// Create a wry file drop handler from a tauri file drop handler.
  2410. fn create_file_drop_handler(
  2411. context: Context,
  2412. label: String,
  2413. menu_ids: HashMap<MenuHash, MenuId>,
  2414. handler: FileDropHandler<Wry>,
  2415. ) -> Box<dyn Fn(&Window, WryFileDropEvent) -> bool + 'static> {
  2416. Box::new(move |window, event| {
  2417. handler(
  2418. FileDropEventWrapper(event).into(),
  2419. DetachedWindow {
  2420. dispatcher: WryDispatcher {
  2421. window_id: window.id(),
  2422. context: context.clone(),
  2423. },
  2424. label: label.clone(),
  2425. menu_ids: menu_ids.clone(),
  2426. },
  2427. )
  2428. })
  2429. }