12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973 |
- // Copyright 2019-2021 Tauri Programme within The Commons Conservancy
- // SPDX-License-Identifier: Apache-2.0
- // SPDX-License-Identifier: MIT
- //! The [`wry`] Tauri [`Runtime`].
- use tauri_runtime::{
- monitor::Monitor,
- webview::{
- FileDropEvent, FileDropHandler, RpcRequest, WebviewRpcHandler, WindowBuilder, WindowBuilderBase,
- },
- window::{
- dpi::{LogicalPosition, LogicalSize, PhysicalPosition, PhysicalSize, Position, Size},
- DetachedWindow, PendingWindow, WindowEvent,
- },
- ClipboardManager, Dispatch, Error, GlobalShortcutManager, Icon, Params, Result, RunEvent,
- RunIteration, Runtime, RuntimeHandle, UserAttentionType,
- };
- #[cfg(feature = "menu")]
- use tauri_runtime::window::MenuEvent;
- #[cfg(feature = "system-tray")]
- use tauri_runtime::{SystemTray, SystemTrayEvent};
- #[cfg(windows)]
- use winapi::shared::windef::HWND;
- #[cfg(windows)]
- use wry::application::platform::windows::WindowBuilderExtWindows;
- #[cfg(feature = "system-tray")]
- use wry::application::system_tray::{SystemTray as WrySystemTray, SystemTrayBuilder};
- use tauri_utils::config::WindowConfig;
- use uuid::Uuid;
- use wry::{
- application::{
- accelerator::{Accelerator, AcceleratorId},
- clipboard::Clipboard,
- dpi::{
- LogicalPosition as WryLogicalPosition, LogicalSize as WryLogicalSize,
- PhysicalPosition as WryPhysicalPosition, PhysicalSize as WryPhysicalSize,
- Position as WryPosition, Size as WrySize,
- },
- event::{Event, WindowEvent as WryWindowEvent},
- event_loop::{ControlFlow, EventLoop, EventLoopProxy, EventLoopWindowTarget},
- monitor::MonitorHandle,
- platform::global_shortcut::{GlobalShortcut, ShortcutManager as WryShortcutManager},
- window::{
- Fullscreen, Icon as WindowIcon, UserAttentionType as WryUserAttentionType, Window,
- WindowBuilder as WryWindowBuilder, WindowId,
- },
- },
- webview::{
- FileDropEvent as WryFileDropEvent, RpcRequest as WryRpcRequest, RpcResponse, WebContext,
- WebView, WebViewBuilder,
- },
- };
- use std::{
- collections::HashMap,
- convert::TryFrom,
- fs::read,
- sync::{
- atomic::{AtomicBool, Ordering},
- mpsc::{channel, Sender},
- Arc, Mutex, MutexGuard,
- },
- thread::{current as current_thread, ThreadId},
- };
- #[cfg(any(feature = "menu", feature = "system-tray"))]
- mod menu;
- #[cfg(any(feature = "menu", feature = "system-tray"))]
- use menu::*;
- mod mime_type;
- use mime_type::MimeType;
- type MainTask = Arc<Mutex<Option<Box<dyn FnOnce() + Send>>>>;
- type CreateWebviewHandler =
- Box<dyn FnOnce(&EventLoopWindowTarget<Message>) -> Result<WebviewWrapper> + Send>;
- type WindowEventHandler = Box<dyn Fn(&WindowEvent) + Send>;
- type WindowEventListenersMap = Arc<Mutex<HashMap<Uuid, WindowEventHandler>>>;
- type WindowEventListeners = Arc<Mutex<HashMap<WindowId, WindowEventListenersMap>>>;
- type GlobalShortcutListeners = Arc<Mutex<HashMap<AcceleratorId, Box<dyn Fn() + Send>>>>;
- macro_rules! dispatcher_getter {
- ($self: ident, $message: expr) => {{
- if current_thread().id() == $self.context.main_thread_id
- && !$self.context.is_event_loop_running.load(Ordering::Relaxed)
- {
- panic!("This API cannot be called when the event loop is not running");
- }
- let (tx, rx) = channel();
- $self
- .context
- .proxy
- .send_event(Message::Window($self.window_id, $message(tx)))
- .map_err(|_| Error::FailedToSendMessage)?;
- rx.recv().unwrap()
- }};
- }
- macro_rules! getter {
- ($self: ident, $rx: expr, $message: expr) => {{
- if current_thread().id() == $self.context.main_thread_id
- && !$self.context.is_event_loop_running.load(Ordering::Relaxed)
- {
- panic!("This API cannot be called when the event loop is not running");
- }
- $self
- .context
- .proxy
- .send_event($message)
- .map_err(|_| Error::FailedToSendMessage)?;
- $rx.recv().unwrap()
- }};
- }
- #[derive(Clone)]
- struct EventLoopContext {
- main_thread_id: ThreadId,
- is_event_loop_running: Arc<AtomicBool>,
- proxy: EventLoopProxy<Message>,
- }
- #[derive(Debug, Clone)]
- struct GlobalShortcutWrapper(GlobalShortcut);
- unsafe impl Send for GlobalShortcutWrapper {}
- /// Wrapper around [`WryShortcutManager`].
- #[derive(Clone)]
- pub struct GlobalShortcutManagerHandle {
- context: EventLoopContext,
- shortcuts: Arc<Mutex<HashMap<String, (AcceleratorId, GlobalShortcutWrapper)>>>,
- listeners: GlobalShortcutListeners,
- }
- impl GlobalShortcutManager for GlobalShortcutManagerHandle {
- fn is_registered(&self, accelerator: &str) -> Result<bool> {
- let (tx, rx) = channel();
- Ok(getter!(
- self,
- rx,
- Message::GlobalShortcut(GlobalShortcutMessage::IsRegistered(
- accelerator.parse().expect("invalid accelerator"),
- tx
- ))
- ))
- }
- fn register<F: Fn() + Send + 'static>(&mut self, accelerator: &str, handler: F) -> Result<()> {
- let wry_accelerator: Accelerator = accelerator.parse().expect("invalid accelerator");
- let id = wry_accelerator.clone().id();
- let (tx, rx) = channel();
- let shortcut = getter!(
- self,
- rx,
- Message::GlobalShortcut(GlobalShortcutMessage::Register(wry_accelerator, tx))
- )?;
- self.listeners.lock().unwrap().insert(id, Box::new(handler));
- self
- .shortcuts
- .lock()
- .unwrap()
- .insert(accelerator.into(), (id, shortcut));
- Ok(())
- }
- fn unregister_all(&mut self) -> Result<()> {
- let (tx, rx) = channel();
- getter!(
- self,
- rx,
- Message::GlobalShortcut(GlobalShortcutMessage::UnregisterAll(tx))
- )?;
- self.listeners.lock().unwrap().clear();
- self.shortcuts.lock().unwrap().clear();
- Ok(())
- }
- fn unregister(&mut self, accelerator: &str) -> Result<()> {
- if let Some((accelerator_id, shortcut)) = self.shortcuts.lock().unwrap().remove(accelerator) {
- let (tx, rx) = channel();
- getter!(
- self,
- rx,
- Message::GlobalShortcut(GlobalShortcutMessage::Unregister(shortcut, tx))
- )?;
- self.listeners.lock().unwrap().remove(&accelerator_id);
- }
- Ok(())
- }
- }
- #[derive(Clone)]
- pub struct ClipboardManagerWrapper {
- context: EventLoopContext,
- }
- impl ClipboardManager for ClipboardManagerWrapper {
- fn read_text(&self) -> Result<Option<String>> {
- let (tx, rx) = channel();
- Ok(getter!(
- self,
- rx,
- Message::Clipboard(ClipboardMessage::ReadText(tx))
- ))
- }
- fn write_text<T: Into<String>>(&mut self, text: T) -> Result<()> {
- let (tx, rx) = channel();
- getter!(
- self,
- rx,
- Message::Clipboard(ClipboardMessage::WriteText(text.into(), tx))
- );
- Ok(())
- }
- }
- /// Wrapper around a [`wry::application::window::Icon`] that can be created from an [`Icon`].
- pub struct WryIcon(WindowIcon);
- fn icon_err<E: std::error::Error + Send + 'static>(e: E) -> Error {
- Error::InvalidIcon(Box::new(e))
- }
- impl TryFrom<Icon> for WryIcon {
- type Error = Error;
- fn try_from(icon: Icon) -> std::result::Result<Self, Self::Error> {
- let image_bytes = match icon {
- Icon::File(path) => read(path).map_err(icon_err)?,
- Icon::Raw(raw) => raw,
- _ => unimplemented!(),
- };
- let extension = infer::get(&image_bytes)
- .expect("could not determine icon extension")
- .extension();
- match extension {
- #[cfg(windows)]
- "ico" => {
- let icon_dir = ico::IconDir::read(std::io::Cursor::new(image_bytes)).map_err(icon_err)?;
- let entry = &icon_dir.entries()[0];
- let icon = WindowIcon::from_rgba(
- entry.decode().map_err(icon_err)?.rgba_data().to_vec(),
- entry.width(),
- entry.height(),
- )
- .map_err(icon_err)?;
- Ok(Self(icon))
- }
- #[cfg(target_os = "linux")]
- "png" => {
- let decoder = png::Decoder::new(std::io::Cursor::new(image_bytes));
- let (info, mut reader) = decoder.read_info().map_err(icon_err)?;
- let mut buffer = Vec::new();
- while let Ok(Some(row)) = reader.next_row() {
- buffer.extend(row);
- }
- let icon = WindowIcon::from_rgba(buffer, info.width, info.height).map_err(icon_err)?;
- Ok(Self(icon))
- }
- _ => panic!(
- "image `{}` extension not supported; please file a Tauri feature request",
- extension
- ),
- }
- }
- }
- struct WindowEventWrapper(Option<WindowEvent>);
- impl<'a> From<&WryWindowEvent<'a>> for WindowEventWrapper {
- fn from(event: &WryWindowEvent<'a>) -> Self {
- let event = match event {
- WryWindowEvent::Resized(size) => WindowEvent::Resized(PhysicalSizeWrapper(*size).into()),
- WryWindowEvent::Moved(position) => {
- WindowEvent::Moved(PhysicalPositionWrapper(*position).into())
- }
- WryWindowEvent::CloseRequested => WindowEvent::CloseRequested,
- WryWindowEvent::Destroyed => WindowEvent::Destroyed,
- WryWindowEvent::Focused(focused) => WindowEvent::Focused(*focused),
- WryWindowEvent::ScaleFactorChanged {
- scale_factor,
- new_inner_size,
- } => WindowEvent::ScaleFactorChanged {
- scale_factor: *scale_factor,
- new_inner_size: PhysicalSizeWrapper(**new_inner_size).into(),
- },
- _ => return Self(None),
- };
- Self(Some(event))
- }
- }
- pub struct MonitorHandleWrapper(MonitorHandle);
- impl From<MonitorHandleWrapper> for Monitor {
- fn from(monitor: MonitorHandleWrapper) -> Monitor {
- Self {
- name: monitor.0.name(),
- position: PhysicalPositionWrapper(monitor.0.position()).into(),
- size: PhysicalSizeWrapper(monitor.0.size()).into(),
- scale_factor: monitor.0.scale_factor(),
- }
- }
- }
- struct PhysicalPositionWrapper<T>(WryPhysicalPosition<T>);
- impl<T> From<PhysicalPositionWrapper<T>> for PhysicalPosition<T> {
- fn from(position: PhysicalPositionWrapper<T>) -> Self {
- Self {
- x: position.0.x,
- y: position.0.y,
- }
- }
- }
- impl<T> From<PhysicalPosition<T>> for PhysicalPositionWrapper<T> {
- fn from(position: PhysicalPosition<T>) -> Self {
- Self(WryPhysicalPosition {
- x: position.x,
- y: position.y,
- })
- }
- }
- struct LogicalPositionWrapper<T>(WryLogicalPosition<T>);
- impl<T> From<LogicalPosition<T>> for LogicalPositionWrapper<T> {
- fn from(position: LogicalPosition<T>) -> Self {
- Self(WryLogicalPosition {
- x: position.x,
- y: position.y,
- })
- }
- }
- struct PhysicalSizeWrapper<T>(WryPhysicalSize<T>);
- impl<T> From<PhysicalSizeWrapper<T>> for PhysicalSize<T> {
- fn from(size: PhysicalSizeWrapper<T>) -> Self {
- Self {
- width: size.0.width,
- height: size.0.height,
- }
- }
- }
- impl<T> From<PhysicalSize<T>> for PhysicalSizeWrapper<T> {
- fn from(size: PhysicalSize<T>) -> Self {
- Self(WryPhysicalSize {
- width: size.width,
- height: size.height,
- })
- }
- }
- struct LogicalSizeWrapper<T>(WryLogicalSize<T>);
- impl<T> From<LogicalSize<T>> for LogicalSizeWrapper<T> {
- fn from(size: LogicalSize<T>) -> Self {
- Self(WryLogicalSize {
- width: size.width,
- height: size.height,
- })
- }
- }
- struct SizeWrapper(WrySize);
- impl From<Size> for SizeWrapper {
- fn from(size: Size) -> Self {
- match size {
- Size::Logical(s) => Self(WrySize::Logical(LogicalSizeWrapper::from(s).0)),
- Size::Physical(s) => Self(WrySize::Physical(PhysicalSizeWrapper::from(s).0)),
- }
- }
- }
- struct PositionWrapper(WryPosition);
- impl From<Position> for PositionWrapper {
- fn from(position: Position) -> Self {
- match position {
- Position::Logical(s) => Self(WryPosition::Logical(LogicalPositionWrapper::from(s).0)),
- Position::Physical(s) => Self(WryPosition::Physical(PhysicalPositionWrapper::from(s).0)),
- }
- }
- }
- #[derive(Debug, Clone)]
- struct UserAttentionTypeWrapper(WryUserAttentionType);
- impl From<UserAttentionType> for UserAttentionTypeWrapper {
- fn from(request_type: UserAttentionType) -> UserAttentionTypeWrapper {
- let o = match request_type {
- UserAttentionType::Critical => WryUserAttentionType::Critical,
- UserAttentionType::Informational => WryUserAttentionType::Informational,
- };
- Self(o)
- }
- }
- #[derive(Debug, Clone, Default)]
- pub struct WindowBuilderWrapper {
- inner: WryWindowBuilder,
- center: bool,
- #[cfg(feature = "menu")]
- menu: Menu<u16>,
- }
- // safe since `menu_items` are read only here
- unsafe impl Send for WindowBuilderWrapper {}
- impl WindowBuilderBase for WindowBuilderWrapper {}
- impl WindowBuilder for WindowBuilderWrapper {
- fn new() -> Self {
- Default::default()
- }
- fn with_config(config: WindowConfig) -> Self {
- let mut window = WindowBuilderWrapper::new()
- .title(config.title.to_string())
- .inner_size(config.width, config.height)
- .visible(config.visible)
- .resizable(config.resizable)
- .decorations(config.decorations)
- .maximized(config.maximized)
- .fullscreen(config.fullscreen)
- .transparent(config.transparent)
- .always_on_top(config.always_on_top)
- .skip_taskbar(config.skip_taskbar);
- if let (Some(min_width), Some(min_height)) = (config.min_width, config.min_height) {
- window = window.min_inner_size(min_width, min_height);
- }
- if let (Some(max_width), Some(max_height)) = (config.max_width, config.max_height) {
- window = window.max_inner_size(max_width, max_height);
- }
- if let (Some(x), Some(y)) = (config.x, config.y) {
- window = window.position(x, y);
- }
- if config.focus {
- window = window.focus();
- }
- window
- }
- #[cfg(feature = "menu")]
- fn menu<I: MenuId>(mut self, menu: Menu<I>) -> Self {
- self.menu = convert_menu_id(Menu::new(), menu);
- self
- }
- fn center(mut self) -> Self {
- self.center = true;
- self
- }
- fn position(mut self, x: f64, y: f64) -> Self {
- self.inner = self.inner.with_position(WryLogicalPosition::new(x, y));
- self
- }
- fn inner_size(mut self, width: f64, height: f64) -> Self {
- self.inner = self
- .inner
- .with_inner_size(WryLogicalSize::new(width, height));
- self
- }
- fn min_inner_size(mut self, min_width: f64, min_height: f64) -> Self {
- self.inner = self
- .inner
- .with_min_inner_size(WryLogicalSize::new(min_width, min_height));
- self
- }
- fn max_inner_size(mut self, max_width: f64, max_height: f64) -> Self {
- self.inner = self
- .inner
- .with_max_inner_size(WryLogicalSize::new(max_width, max_height));
- self
- }
- fn resizable(mut self, resizable: bool) -> Self {
- self.inner = self.inner.with_resizable(resizable);
- self
- }
- fn title<S: Into<String>>(mut self, title: S) -> Self {
- self.inner = self.inner.with_title(title.into());
- self
- }
- fn fullscreen(mut self, fullscreen: bool) -> Self {
- self.inner = if fullscreen {
- self
- .inner
- .with_fullscreen(Some(Fullscreen::Borderless(None)))
- } else {
- self.inner.with_fullscreen(None)
- };
- self
- }
- fn focus(mut self) -> Self {
- self.inner = self.inner.with_focus();
- self
- }
- fn maximized(mut self, maximized: bool) -> Self {
- self.inner = self.inner.with_maximized(maximized);
- self
- }
- fn visible(mut self, visible: bool) -> Self {
- self.inner = self.inner.with_visible(visible);
- self
- }
- fn transparent(mut self, transparent: bool) -> Self {
- self.inner = self.inner.with_transparent(transparent);
- self
- }
- fn decorations(mut self, decorations: bool) -> Self {
- self.inner = self.inner.with_decorations(decorations);
- self
- }
- fn always_on_top(mut self, always_on_top: bool) -> Self {
- self.inner = self.inner.with_always_on_top(always_on_top);
- self
- }
- #[cfg(windows)]
- fn parent_window(mut self, parent: HWND) -> Self {
- self.inner = self.inner.with_parent_window(parent);
- self
- }
- #[cfg(windows)]
- fn owner_window(mut self, owner: HWND) -> Self {
- self.inner = self.inner.with_owner_window(owner);
- self
- }
- fn icon(mut self, icon: Icon) -> Result<Self> {
- self.inner = self
- .inner
- .with_window_icon(Some(WryIcon::try_from(icon)?.0));
- Ok(self)
- }
- fn skip_taskbar(mut self, skip: bool) -> Self {
- self.inner = self.inner.with_skip_taskbar(skip);
- self
- }
- fn has_icon(&self) -> bool {
- self.inner.window.window_icon.is_some()
- }
- #[cfg(feature = "menu")]
- fn has_menu(&self) -> bool {
- self.inner.window.window_menu.is_some()
- }
- }
- pub struct RpcRequestWrapper(WryRpcRequest);
- impl From<RpcRequestWrapper> for RpcRequest {
- fn from(request: RpcRequestWrapper) -> Self {
- Self {
- command: request.0.method,
- params: request.0.params,
- }
- }
- }
- pub struct FileDropEventWrapper(WryFileDropEvent);
- impl From<FileDropEventWrapper> for FileDropEvent {
- fn from(event: FileDropEventWrapper) -> Self {
- match event.0 {
- WryFileDropEvent::Hovered(paths) => FileDropEvent::Hovered(paths),
- WryFileDropEvent::Dropped(paths) => FileDropEvent::Dropped(paths),
- WryFileDropEvent::Cancelled => FileDropEvent::Cancelled,
- }
- }
- }
- #[cfg(windows)]
- struct Hwnd(HWND);
- #[cfg(windows)]
- unsafe impl Send for Hwnd {}
- #[derive(Debug, Clone)]
- enum WindowMessage {
- // Getters
- ScaleFactor(Sender<f64>),
- InnerPosition(Sender<Result<PhysicalPosition<i32>>>),
- OuterPosition(Sender<Result<PhysicalPosition<i32>>>),
- InnerSize(Sender<PhysicalSize<u32>>),
- OuterSize(Sender<PhysicalSize<u32>>),
- IsFullscreen(Sender<bool>),
- IsMaximized(Sender<bool>),
- IsDecorated(Sender<bool>),
- IsResizable(Sender<bool>),
- IsVisible(Sender<bool>),
- #[cfg(feature = "menu")]
- IsMenuVisible(Sender<bool>),
- CurrentMonitor(Sender<Option<MonitorHandle>>),
- PrimaryMonitor(Sender<Option<MonitorHandle>>),
- AvailableMonitors(Sender<Vec<MonitorHandle>>),
- #[cfg(windows)]
- Hwnd(Sender<Hwnd>),
- // Setters
- Center(Sender<Result<()>>),
- RequestUserAttention(Option<UserAttentionTypeWrapper>),
- SetResizable(bool),
- SetTitle(String),
- Maximize,
- Unmaximize,
- Minimize,
- Unminimize,
- #[cfg(feature = "menu")]
- ShowMenu,
- #[cfg(feature = "menu")]
- HideMenu,
- Show,
- Hide,
- Close,
- SetDecorations(bool),
- SetAlwaysOnTop(bool),
- SetSize(Size),
- SetMinSize(Option<Size>),
- SetMaxSize(Option<Size>),
- SetPosition(Position),
- SetFullscreen(bool),
- SetFocus,
- SetIcon(WindowIcon),
- SetSkipTaskbar(bool),
- DragWindow,
- #[cfg(feature = "menu")]
- UpdateMenuItem(u16, menu::MenuUpdate),
- }
- #[derive(Debug, Clone)]
- enum WebviewMessage {
- EvaluateScript(String),
- Print,
- }
- #[cfg(feature = "system-tray")]
- #[derive(Clone)]
- pub(crate) enum TrayMessage {
- UpdateItem(u16, menu::MenuUpdate),
- UpdateIcon(Icon),
- #[cfg(windows)]
- Remove,
- }
- #[derive(Clone)]
- pub(crate) enum GlobalShortcutMessage {
- IsRegistered(Accelerator, Sender<bool>),
- Register(Accelerator, Sender<Result<GlobalShortcutWrapper>>),
- Unregister(GlobalShortcutWrapper, Sender<Result<()>>),
- UnregisterAll(Sender<Result<()>>),
- }
- #[derive(Clone)]
- pub(crate) enum ClipboardMessage {
- WriteText(String, Sender<()>),
- ReadText(Sender<Option<String>>),
- }
- #[derive(Clone)]
- pub(crate) enum Message {
- Task(MainTask),
- Window(WindowId, WindowMessage),
- Webview(WindowId, WebviewMessage),
- #[cfg(feature = "system-tray")]
- Tray(TrayMessage),
- CreateWebview(Arc<Mutex<Option<CreateWebviewHandler>>>, Sender<WindowId>),
- GlobalShortcut(GlobalShortcutMessage),
- Clipboard(ClipboardMessage),
- }
- #[derive(Clone)]
- struct DispatcherContext {
- main_thread_id: ThreadId,
- is_event_loop_running: Arc<AtomicBool>,
- proxy: EventLoopProxy<Message>,
- window_event_listeners: WindowEventListeners,
- #[cfg(feature = "menu")]
- menu_event_listeners: MenuEventListeners,
- }
- /// The Tauri [`Dispatch`] for [`Wry`].
- #[derive(Clone)]
- pub struct WryDispatcher {
- window_id: WindowId,
- context: DispatcherContext,
- }
- impl Dispatch for WryDispatcher {
- type Runtime = Wry;
- type WindowBuilder = WindowBuilderWrapper;
- fn run_on_main_thread<F: FnOnce() + Send + 'static>(&self, f: F) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Task(Arc::new(Mutex::new(Some(Box::new(f))))))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn on_window_event<F: Fn(&WindowEvent) + Send + 'static>(&self, f: F) -> Uuid {
- let id = Uuid::new_v4();
- self
- .context
- .window_event_listeners
- .lock()
- .unwrap()
- .get(&self.window_id)
- .unwrap()
- .lock()
- .unwrap()
- .insert(id, Box::new(f));
- id
- }
- #[cfg(feature = "menu")]
- fn on_menu_event<F: Fn(&MenuEvent) + Send + 'static>(&self, f: F) -> Uuid {
- let id = Uuid::new_v4();
- self
- .context
- .menu_event_listeners
- .lock()
- .unwrap()
- .get(&self.window_id)
- .unwrap()
- .lock()
- .unwrap()
- .insert(id, Box::new(f));
- id
- }
- // Getters
- fn scale_factor(&self) -> Result<f64> {
- Ok(dispatcher_getter!(self, WindowMessage::ScaleFactor))
- }
- fn inner_position(&self) -> Result<PhysicalPosition<i32>> {
- dispatcher_getter!(self, WindowMessage::InnerPosition)
- }
- fn outer_position(&self) -> Result<PhysicalPosition<i32>> {
- dispatcher_getter!(self, WindowMessage::OuterPosition)
- }
- fn inner_size(&self) -> Result<PhysicalSize<u32>> {
- Ok(dispatcher_getter!(self, WindowMessage::InnerSize))
- }
- fn outer_size(&self) -> Result<PhysicalSize<u32>> {
- Ok(dispatcher_getter!(self, WindowMessage::OuterSize))
- }
- fn is_fullscreen(&self) -> Result<bool> {
- Ok(dispatcher_getter!(self, WindowMessage::IsFullscreen))
- }
- fn is_maximized(&self) -> Result<bool> {
- Ok(dispatcher_getter!(self, WindowMessage::IsMaximized))
- }
- /// Gets the window’s current decoration state.
- fn is_decorated(&self) -> Result<bool> {
- Ok(dispatcher_getter!(self, WindowMessage::IsDecorated))
- }
- /// Gets the window’s current resizable state.
- fn is_resizable(&self) -> Result<bool> {
- Ok(dispatcher_getter!(self, WindowMessage::IsResizable))
- }
- fn is_visible(&self) -> Result<bool> {
- Ok(dispatcher_getter!(self, WindowMessage::IsVisible))
- }
- #[cfg(feature = "menu")]
- fn is_menu_visible(&self) -> Result<bool> {
- Ok(dispatcher_getter!(self, WindowMessage::IsMenuVisible))
- }
- fn current_monitor(&self) -> Result<Option<Monitor>> {
- Ok(
- dispatcher_getter!(self, WindowMessage::CurrentMonitor)
- .map(|m| MonitorHandleWrapper(m).into()),
- )
- }
- fn primary_monitor(&self) -> Result<Option<Monitor>> {
- Ok(
- dispatcher_getter!(self, WindowMessage::PrimaryMonitor)
- .map(|m| MonitorHandleWrapper(m).into()),
- )
- }
- fn available_monitors(&self) -> Result<Vec<Monitor>> {
- Ok(
- dispatcher_getter!(self, WindowMessage::AvailableMonitors)
- .into_iter()
- .map(|m| MonitorHandleWrapper(m).into())
- .collect(),
- )
- }
- #[cfg(windows)]
- fn hwnd(&self) -> Result<HWND> {
- Ok(dispatcher_getter!(self, WindowMessage::Hwnd).0)
- }
- // Setters
- fn center(&self) -> Result<()> {
- dispatcher_getter!(self, WindowMessage::Center)
- }
- fn print(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Webview(self.window_id, WebviewMessage::Print))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn request_user_attention(&self, request_type: Option<UserAttentionType>) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::RequestUserAttention(request_type.map(Into::into)),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- // Creates a window by dispatching a message to the event loop.
- // Note that this must be called from a separate thread, otherwise the channel will introduce a deadlock.
- fn create_window<P: Params<Runtime = Self::Runtime>>(
- &mut self,
- pending: PendingWindow<P>,
- ) -> Result<DetachedWindow<P>> {
- let (tx, rx) = channel();
- let label = pending.label.clone();
- let context = self.context.clone();
- self
- .context
- .proxy
- .send_event(Message::CreateWebview(
- Arc::new(Mutex::new(Some(Box::new(move |event_loop| {
- create_webview(event_loop, context, pending)
- })))),
- tx,
- ))
- .map_err(|_| Error::FailedToSendMessage)?;
- let window_id = rx.recv().unwrap();
- let dispatcher = WryDispatcher {
- window_id,
- context: self.context.clone(),
- };
- Ok(DetachedWindow { label, dispatcher })
- }
- fn set_resizable(&self, resizable: bool) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetResizable(resizable),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_title<S: Into<String>>(&self, title: S) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetTitle(title.into()),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn maximize(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Maximize))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn unmaximize(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Unmaximize))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn minimize(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Minimize))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn unminimize(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Unminimize))
- .map_err(|_| Error::FailedToSendMessage)
- }
- #[cfg(feature = "menu")]
- fn show_menu(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::ShowMenu))
- .map_err(|_| Error::FailedToSendMessage)
- }
- #[cfg(feature = "menu")]
- fn hide_menu(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::HideMenu))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn show(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Show))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn hide(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Hide))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn close(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::Close))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_decorations(&self, decorations: bool) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetDecorations(decorations),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_always_on_top(&self, always_on_top: bool) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetAlwaysOnTop(always_on_top),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_size(&self, size: Size) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetSize(size),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_min_size(&self, size: Option<Size>) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetMinSize(size),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_max_size(&self, size: Option<Size>) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetMaxSize(size),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_position(&self, position: Position) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetPosition(position),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_fullscreen(&self, fullscreen: bool) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetFullscreen(fullscreen),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_focus(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::SetFocus))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_icon(&self, icon: Icon) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetIcon(WryIcon::try_from(icon)?.0),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn set_skip_taskbar(&self, skip: bool) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::SetSkipTaskbar(skip),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn start_dragging(&self) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(self.window_id, WindowMessage::DragWindow))
- .map_err(|_| Error::FailedToSendMessage)
- }
- fn eval_script<S: Into<String>>(&self, script: S) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Webview(
- self.window_id,
- WebviewMessage::EvaluateScript(script.into()),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- #[cfg(feature = "menu")]
- fn update_menu_item(&self, id: u16, update: menu::MenuUpdate) -> Result<()> {
- self
- .context
- .proxy
- .send_event(Message::Window(
- self.window_id,
- WindowMessage::UpdateMenuItem(id, update),
- ))
- .map_err(|_| Error::FailedToSendMessage)
- }
- }
- #[cfg(feature = "system-tray")]
- #[derive(Clone, Default)]
- struct TrayContext {
- tray: Arc<Mutex<Option<Arc<Mutex<WrySystemTray>>>>>,
- listeners: SystemTrayEventListeners,
- items: SystemTrayItems,
- }
- struct WebviewWrapper {
- label: String,
- inner: WebView,
- #[cfg(feature = "menu")]
- menu_items: HashMap<u16, WryCustomMenuItem>,
- #[cfg(feature = "menu")]
- is_menu_visible: AtomicBool,
- }
- /// A Tauri [`Runtime`] wrapper around wry.
- pub struct Wry {
- main_thread_id: ThreadId,
- global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
- global_shortcut_manager_handle: GlobalShortcutManagerHandle,
- clipboard_manager: Arc<Mutex<Clipboard>>,
- clipboard_manager_handle: ClipboardManagerWrapper,
- is_event_loop_running: Arc<AtomicBool>,
- event_loop: EventLoop<Message>,
- webviews: Arc<Mutex<HashMap<WindowId, WebviewWrapper>>>,
- window_event_listeners: WindowEventListeners,
- #[cfg(feature = "menu")]
- menu_event_listeners: MenuEventListeners,
- #[cfg(feature = "system-tray")]
- tray_context: TrayContext,
- }
- /// A handle to the Wry runtime.
- #[derive(Clone)]
- pub struct WryHandle {
- dispatcher_context: DispatcherContext,
- }
- impl RuntimeHandle for WryHandle {
- type Runtime = Wry;
- // Creates a window by dispatching a message to the event loop.
- // Note that this must be called from a separate thread, otherwise the channel will introduce a deadlock.
- fn create_window<P: Params<Runtime = Self::Runtime>>(
- &self,
- pending: PendingWindow<P>,
- ) -> Result<DetachedWindow<P>> {
- let (tx, rx) = channel();
- let label = pending.label.clone();
- let dispatcher_context = self.dispatcher_context.clone();
- self
- .dispatcher_context
- .proxy
- .send_event(Message::CreateWebview(
- Arc::new(Mutex::new(Some(Box::new(move |event_loop| {
- create_webview(event_loop, dispatcher_context, pending)
- })))),
- tx,
- ))
- .map_err(|_| Error::FailedToSendMessage)?;
- let window_id = rx.recv().unwrap();
- let dispatcher = WryDispatcher {
- window_id,
- context: self.dispatcher_context.clone(),
- };
- Ok(DetachedWindow { label, dispatcher })
- }
- #[cfg(all(windows, feature = "system-tray"))]
- fn remove_system_tray(&self) -> Result<()> {
- self
- .dispatcher_context
- .proxy
- .send_event(Message::Tray(TrayMessage::Remove))
- .map_err(|_| Error::FailedToSendMessage)
- }
- }
- impl Runtime for Wry {
- type Dispatcher = WryDispatcher;
- type Handle = WryHandle;
- type GlobalShortcutManager = GlobalShortcutManagerHandle;
- type ClipboardManager = ClipboardManagerWrapper;
- #[cfg(feature = "system-tray")]
- type TrayHandler = SystemTrayHandle;
- fn new() -> Result<Self> {
- let event_loop = EventLoop::<Message>::with_user_event();
- let proxy = event_loop.create_proxy();
- let main_thread_id = current_thread().id();
- let is_event_loop_running = Arc::new(AtomicBool::default());
- let event_loop_context = EventLoopContext {
- main_thread_id,
- is_event_loop_running: is_event_loop_running.clone(),
- proxy,
- };
- let global_shortcut_manager = WryShortcutManager::new(&event_loop);
- let global_shortcut_listeners = GlobalShortcutListeners::default();
- let clipboard_manager = Clipboard::new();
- let clipboard_manager_handle = ClipboardManagerWrapper {
- context: event_loop_context.clone(),
- };
- Ok(Self {
- main_thread_id,
- global_shortcut_manager: Arc::new(Mutex::new(global_shortcut_manager)),
- global_shortcut_manager_handle: GlobalShortcutManagerHandle {
- context: event_loop_context,
- shortcuts: Default::default(),
- listeners: global_shortcut_listeners,
- },
- clipboard_manager: Arc::new(Mutex::new(clipboard_manager)),
- clipboard_manager_handle,
- is_event_loop_running,
- event_loop,
- webviews: Default::default(),
- window_event_listeners: Default::default(),
- #[cfg(feature = "menu")]
- menu_event_listeners: Default::default(),
- #[cfg(feature = "system-tray")]
- tray_context: Default::default(),
- })
- }
- fn handle(&self) -> Self::Handle {
- WryHandle {
- dispatcher_context: DispatcherContext {
- main_thread_id: self.main_thread_id,
- is_event_loop_running: self.is_event_loop_running.clone(),
- proxy: self.event_loop.create_proxy(),
- window_event_listeners: self.window_event_listeners.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: self.menu_event_listeners.clone(),
- },
- }
- }
- fn global_shortcut_manager(&self) -> Self::GlobalShortcutManager {
- self.global_shortcut_manager_handle.clone()
- }
- fn clipboard_manager(&self) -> Self::ClipboardManager {
- self.clipboard_manager_handle.clone()
- }
- fn create_window<P: Params<Runtime = Self>>(
- &self,
- pending: PendingWindow<P>,
- ) -> Result<DetachedWindow<P>> {
- let label = pending.label.clone();
- let proxy = self.event_loop.create_proxy();
- let webview = create_webview(
- &self.event_loop,
- DispatcherContext {
- main_thread_id: self.main_thread_id,
- is_event_loop_running: self.is_event_loop_running.clone(),
- proxy: proxy.clone(),
- window_event_listeners: self.window_event_listeners.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: self.menu_event_listeners.clone(),
- },
- pending,
- )?;
- let dispatcher = WryDispatcher {
- window_id: webview.inner.window().id(),
- context: DispatcherContext {
- main_thread_id: self.main_thread_id,
- is_event_loop_running: self.is_event_loop_running.clone(),
- proxy,
- window_event_listeners: self.window_event_listeners.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: self.menu_event_listeners.clone(),
- },
- };
- self
- .webviews
- .lock()
- .unwrap()
- .insert(webview.inner.window().id(), webview);
- Ok(DetachedWindow { label, dispatcher })
- }
- #[cfg(feature = "system-tray")]
- fn system_tray<I: MenuId>(&self, system_tray: SystemTray<I>) -> Result<Self::TrayHandler> {
- let icon = system_tray
- .icon
- .expect("tray icon not set")
- .into_tray_icon();
- let mut items = HashMap::new();
- let tray = SystemTrayBuilder::new(
- icon,
- system_tray
- .menu
- .map(|menu| to_wry_context_menu(&mut items, menu)),
- )
- .build(&self.event_loop)
- .map_err(|e| Error::SystemTray(Box::new(e)))?;
- *self.tray_context.items.lock().unwrap() = items;
- *self.tray_context.tray.lock().unwrap() = Some(Arc::new(Mutex::new(tray)));
- Ok(SystemTrayHandle {
- proxy: self.event_loop.create_proxy(),
- })
- }
- #[cfg(feature = "system-tray")]
- fn on_system_tray_event<F: Fn(&SystemTrayEvent) + Send + 'static>(&mut self, f: F) -> Uuid {
- let id = Uuid::new_v4();
- self
- .tray_context
- .listeners
- .lock()
- .unwrap()
- .insert(id, Box::new(f));
- id
- }
- #[cfg(any(target_os = "windows", target_os = "macos"))]
- fn run_iteration<F: Fn(RunEvent) + 'static>(&mut self, callback: F) -> RunIteration {
- use wry::application::platform::run_return::EventLoopExtRunReturn;
- let webviews = self.webviews.clone();
- let window_event_listeners = self.window_event_listeners.clone();
- #[cfg(feature = "menu")]
- let menu_event_listeners = self.menu_event_listeners.clone();
- #[cfg(feature = "system-tray")]
- let tray_context = self.tray_context.clone();
- let global_shortcut_manager = self.global_shortcut_manager.clone();
- let global_shortcut_manager_handle = self.global_shortcut_manager_handle.clone();
- let clipboard_manager = self.clipboard_manager.clone();
- let mut iteration = RunIteration::default();
- self.is_event_loop_running.store(true, Ordering::Relaxed);
- self
- .event_loop
- .run_return(|event, event_loop, control_flow| {
- if let Event::MainEventsCleared = &event {
- *control_flow = ControlFlow::Exit;
- }
- iteration = handle_event_loop(
- event,
- event_loop,
- control_flow,
- EventLoopIterationContext {
- callback: &callback,
- webviews: webviews.lock().expect("poisoned webview collection"),
- window_event_listeners: &window_event_listeners,
- global_shortcut_manager: global_shortcut_manager.clone(),
- global_shortcut_manager_handle: &global_shortcut_manager_handle,
- clipboard_manager: clipboard_manager.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: &menu_event_listeners,
- #[cfg(feature = "system-tray")]
- tray_context: &tray_context,
- },
- );
- });
- self.is_event_loop_running.store(false, Ordering::Relaxed);
- iteration
- }
- fn run<F: Fn(RunEvent) + 'static>(self, callback: F) {
- self.is_event_loop_running.store(true, Ordering::Relaxed);
- let webviews = self.webviews.clone();
- let window_event_listeners = self.window_event_listeners.clone();
- #[cfg(feature = "menu")]
- let menu_event_listeners = self.menu_event_listeners.clone();
- #[cfg(feature = "system-tray")]
- let tray_context = self.tray_context;
- let global_shortcut_manager = self.global_shortcut_manager.clone();
- let global_shortcut_manager_handle = self.global_shortcut_manager_handle.clone();
- let clipboard_manager = self.clipboard_manager.clone();
- self.event_loop.run(move |event, event_loop, control_flow| {
- handle_event_loop(
- event,
- event_loop,
- control_flow,
- EventLoopIterationContext {
- callback: &callback,
- webviews: webviews.lock().expect("poisoned webview collection"),
- window_event_listeners: &window_event_listeners,
- global_shortcut_manager: global_shortcut_manager.clone(),
- global_shortcut_manager_handle: &global_shortcut_manager_handle,
- clipboard_manager: clipboard_manager.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: &menu_event_listeners,
- #[cfg(feature = "system-tray")]
- tray_context: &tray_context,
- },
- );
- })
- }
- }
- struct EventLoopIterationContext<'a> {
- callback: &'a (dyn Fn(RunEvent) + 'static),
- webviews: MutexGuard<'a, HashMap<WindowId, WebviewWrapper>>,
- window_event_listeners: &'a WindowEventListeners,
- global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
- global_shortcut_manager_handle: &'a GlobalShortcutManagerHandle,
- clipboard_manager: Arc<Mutex<Clipboard>>,
- #[cfg(feature = "menu")]
- menu_event_listeners: &'a MenuEventListeners,
- #[cfg(feature = "system-tray")]
- tray_context: &'a TrayContext,
- }
- fn handle_event_loop(
- event: Event<Message>,
- event_loop: &EventLoopWindowTarget<Message>,
- control_flow: &mut ControlFlow,
- context: EventLoopIterationContext<'_>,
- ) -> RunIteration {
- let EventLoopIterationContext {
- callback,
- mut webviews,
- window_event_listeners,
- global_shortcut_manager,
- global_shortcut_manager_handle,
- clipboard_manager,
- #[cfg(feature = "menu")]
- menu_event_listeners,
- #[cfg(feature = "system-tray")]
- tray_context,
- } = context;
- *control_flow = ControlFlow::Wait;
- match event {
- Event::GlobalShortcutEvent(accelerator_id) => {
- for (id, handler) in &*global_shortcut_manager_handle.listeners.lock().unwrap() {
- if accelerator_id == *id {
- handler();
- }
- }
- }
- #[cfg(feature = "menu")]
- Event::MenuEvent {
- window_id,
- menu_id,
- origin: MenuType::MenuBar,
- } => {
- let window_id = window_id.unwrap(); // always Some on MenuBar event
- let event = MenuEvent {
- menu_item_id: menu_id.0,
- };
- let listeners = menu_event_listeners.lock().unwrap();
- let window_menu_event_listeners = listeners.get(&window_id).cloned().unwrap_or_default();
- for handler in window_menu_event_listeners.lock().unwrap().values() {
- handler(&event);
- }
- }
- #[cfg(feature = "system-tray")]
- Event::MenuEvent {
- window_id: _,
- menu_id,
- origin: MenuType::ContextMenu,
- } => {
- let event = SystemTrayEvent::MenuItemClick(menu_id.0);
- for handler in tray_context.listeners.lock().unwrap().values() {
- handler(&event);
- }
- }
- #[cfg(feature = "system-tray")]
- Event::TrayEvent {
- bounds,
- event,
- position: _cursor_position,
- } => {
- let (position, size) = (
- PhysicalPositionWrapper(bounds.position).into(),
- PhysicalSizeWrapper(bounds.size).into(),
- );
- let event = match event {
- TrayEvent::LeftClick => SystemTrayEvent::LeftClick { position, size },
- TrayEvent::RightClick => SystemTrayEvent::RightClick { position, size },
- TrayEvent::DoubleClick => SystemTrayEvent::DoubleClick { position, size },
- };
- for handler in tray_context.listeners.lock().unwrap().values() {
- handler(&event);
- }
- }
- Event::WindowEvent { event, window_id } => {
- if let Some(event) = WindowEventWrapper::from(&event).0 {
- for handler in window_event_listeners
- .lock()
- .unwrap()
- .get(&window_id)
- .unwrap()
- .lock()
- .unwrap()
- .values()
- {
- handler(&event);
- }
- }
- match event {
- WryWindowEvent::CloseRequested => {
- on_window_close(
- callback,
- window_id,
- &mut webviews,
- control_flow,
- #[cfg(feature = "menu")]
- menu_event_listeners.clone(),
- );
- }
- WryWindowEvent::Resized(_) => {
- if let Err(e) = webviews[&window_id].inner.resize() {
- eprintln!("{}", e);
- }
- }
- _ => {}
- }
- }
- Event::UserEvent(message) => match message {
- Message::Task(task) => {
- if let Some(task) = task.lock().unwrap().take() {
- task();
- }
- }
- Message::Window(id, window_message) => {
- if let Some(webview) = webviews.get_mut(&id) {
- let window = webview.inner.window();
- match window_message {
- // Getters
- WindowMessage::ScaleFactor(tx) => tx.send(window.scale_factor()).unwrap(),
- WindowMessage::InnerPosition(tx) => tx
- .send(
- window
- .inner_position()
- .map(|p| PhysicalPositionWrapper(p).into())
- .map_err(|_| Error::FailedToSendMessage),
- )
- .unwrap(),
- WindowMessage::OuterPosition(tx) => tx
- .send(
- window
- .outer_position()
- .map(|p| PhysicalPositionWrapper(p).into())
- .map_err(|_| Error::FailedToSendMessage),
- )
- .unwrap(),
- WindowMessage::InnerSize(tx) => tx
- .send(PhysicalSizeWrapper(window.inner_size()).into())
- .unwrap(),
- WindowMessage::OuterSize(tx) => tx
- .send(PhysicalSizeWrapper(window.outer_size()).into())
- .unwrap(),
- WindowMessage::IsFullscreen(tx) => tx.send(window.fullscreen().is_some()).unwrap(),
- WindowMessage::IsMaximized(tx) => tx.send(window.is_maximized()).unwrap(),
- WindowMessage::IsDecorated(tx) => tx.send(window.is_decorated()).unwrap(),
- WindowMessage::IsResizable(tx) => tx.send(window.is_resizable()).unwrap(),
- WindowMessage::IsVisible(tx) => tx.send(window.is_visible()).unwrap(),
- #[cfg(feature = "menu")]
- WindowMessage::IsMenuVisible(tx) => tx
- .send(webview.is_menu_visible.load(Ordering::Relaxed))
- .unwrap(),
- WindowMessage::CurrentMonitor(tx) => tx.send(window.current_monitor()).unwrap(),
- WindowMessage::PrimaryMonitor(tx) => tx.send(window.primary_monitor()).unwrap(),
- WindowMessage::AvailableMonitors(tx) => {
- tx.send(window.available_monitors().collect()).unwrap()
- }
- #[cfg(windows)]
- WindowMessage::Hwnd(tx) => {
- use wry::application::platform::windows::WindowExtWindows;
- tx.send(Hwnd(window.hwnd() as HWND)).unwrap()
- }
- // Setters
- WindowMessage::Center(tx) => {
- tx.send(center_window(window)).unwrap();
- }
- WindowMessage::RequestUserAttention(request_type) => {
- window.request_user_attention(request_type.map(|r| r.0));
- }
- WindowMessage::SetResizable(resizable) => window.set_resizable(resizable),
- WindowMessage::SetTitle(title) => window.set_title(&title),
- WindowMessage::Maximize => window.set_maximized(true),
- WindowMessage::Unmaximize => window.set_maximized(false),
- WindowMessage::Minimize => window.set_minimized(true),
- WindowMessage::Unminimize => window.set_minimized(false),
- #[cfg(feature = "menu")]
- WindowMessage::ShowMenu => {
- window.show_menu();
- webview.is_menu_visible.store(true, Ordering::Relaxed);
- }
- #[cfg(feature = "menu")]
- WindowMessage::HideMenu => {
- window.hide_menu();
- webview.is_menu_visible.store(false, Ordering::Relaxed);
- }
- WindowMessage::Show => window.set_visible(true),
- WindowMessage::Hide => window.set_visible(false),
- WindowMessage::Close => {
- for handler in window_event_listeners
- .lock()
- .unwrap()
- .get(&window.id())
- .unwrap()
- .lock()
- .unwrap()
- .values()
- {
- handler(&WindowEvent::CloseRequested);
- }
- on_window_close(
- callback,
- id,
- &mut webviews,
- control_flow,
- #[cfg(feature = "menu")]
- menu_event_listeners.clone(),
- );
- }
- WindowMessage::SetDecorations(decorations) => window.set_decorations(decorations),
- WindowMessage::SetAlwaysOnTop(always_on_top) => window.set_always_on_top(always_on_top),
- WindowMessage::SetSize(size) => {
- window.set_inner_size(SizeWrapper::from(size).0);
- }
- WindowMessage::SetMinSize(size) => {
- window.set_min_inner_size(size.map(|s| SizeWrapper::from(s).0));
- }
- WindowMessage::SetMaxSize(size) => {
- window.set_max_inner_size(size.map(|s| SizeWrapper::from(s).0));
- }
- WindowMessage::SetPosition(position) => {
- window.set_outer_position(PositionWrapper::from(position).0)
- }
- WindowMessage::SetFullscreen(fullscreen) => {
- if fullscreen {
- window.set_fullscreen(Some(Fullscreen::Borderless(None)))
- } else {
- window.set_fullscreen(None)
- }
- }
- WindowMessage::SetFocus => {
- window.set_focus();
- }
- WindowMessage::SetIcon(icon) => {
- window.set_window_icon(Some(icon));
- }
- WindowMessage::SetSkipTaskbar(skip) => {
- window.set_skip_taskbar(skip);
- }
- WindowMessage::DragWindow => {
- let _ = window.drag_window();
- }
- #[cfg(feature = "menu")]
- WindowMessage::UpdateMenuItem(id, update) => {
- let item = webview
- .menu_items
- .get_mut(&id)
- .expect("menu item not found");
- match update {
- MenuUpdate::SetEnabled(enabled) => item.set_enabled(enabled),
- MenuUpdate::SetTitle(title) => item.set_title(&title),
- MenuUpdate::SetSelected(selected) => item.set_selected(selected),
- #[cfg(target_os = "macos")]
- MenuUpdate::SetNativeImage(image) => {
- item.set_native_image(NativeImageWrapper::from(image).0)
- }
- }
- }
- }
- }
- }
- Message::Webview(id, webview_message) => {
- if let Some(webview) = webviews.get_mut(&id) {
- match webview_message {
- WebviewMessage::EvaluateScript(script) => {
- if let Err(e) = webview.inner.evaluate_script(&script) {
- eprintln!("{}", e);
- }
- }
- WebviewMessage::Print => {
- let _ = webview.inner.print();
- }
- }
- }
- }
- Message::CreateWebview(handler, sender) => {
- let handler = {
- let mut lock = handler.lock().expect("poisoned create webview handler");
- std::mem::take(&mut *lock).unwrap()
- };
- match handler(event_loop) {
- Ok(webview) => {
- let window_id = webview.inner.window().id();
- webviews.insert(window_id, webview);
- sender.send(window_id).unwrap();
- }
- Err(e) => {
- eprintln!("{}", e);
- }
- }
- }
- #[cfg(feature = "system-tray")]
- Message::Tray(tray_message) => match tray_message {
- TrayMessage::UpdateItem(menu_id, update) => {
- let mut tray = tray_context.items.as_ref().lock().unwrap();
- let item = tray.get_mut(&menu_id).expect("menu item not found");
- match update {
- MenuUpdate::SetEnabled(enabled) => item.set_enabled(enabled),
- MenuUpdate::SetTitle(title) => item.set_title(&title),
- MenuUpdate::SetSelected(selected) => item.set_selected(selected),
- #[cfg(target_os = "macos")]
- MenuUpdate::SetNativeImage(image) => {
- item.set_native_image(NativeImageWrapper::from(image).0)
- }
- }
- }
- TrayMessage::UpdateIcon(icon) => {
- if let Some(tray) = &*tray_context.tray.lock().unwrap() {
- tray.lock().unwrap().set_icon(icon.into_tray_icon());
- }
- }
- #[cfg(windows)]
- TrayMessage::Remove => {
- if let Some(tray) = tray_context.tray.lock().unwrap().as_ref() {
- use wry::application::platform::windows::SystemTrayExtWindows;
- tray.lock().unwrap().remove();
- }
- }
- },
- Message::GlobalShortcut(message) => match message {
- GlobalShortcutMessage::IsRegistered(accelerator, tx) => tx
- .send(
- global_shortcut_manager
- .lock()
- .unwrap()
- .is_registered(&accelerator),
- )
- .unwrap(),
- GlobalShortcutMessage::Register(accelerator, tx) => tx
- .send(
- global_shortcut_manager
- .lock()
- .unwrap()
- .register(accelerator)
- .map(GlobalShortcutWrapper)
- .map_err(|e| Error::GlobalShortcut(Box::new(e))),
- )
- .unwrap(),
- GlobalShortcutMessage::Unregister(shortcut, tx) => tx
- .send(
- global_shortcut_manager
- .lock()
- .unwrap()
- .unregister(shortcut.0)
- .map_err(|e| Error::GlobalShortcut(Box::new(e))),
- )
- .unwrap(),
- GlobalShortcutMessage::UnregisterAll(tx) => tx
- .send(
- global_shortcut_manager
- .lock()
- .unwrap()
- .unregister_all()
- .map_err(|e| Error::GlobalShortcut(Box::new(e))),
- )
- .unwrap(),
- },
- Message::Clipboard(message) => match message {
- ClipboardMessage::WriteText(text, tx) => {
- clipboard_manager.lock().unwrap().write_text(text);
- tx.send(()).unwrap();
- }
- ClipboardMessage::ReadText(tx) => tx
- .send(clipboard_manager.lock().unwrap().read_text())
- .unwrap(),
- },
- },
- _ => (),
- }
- RunIteration {
- webview_count: webviews.len(),
- }
- }
- fn on_window_close<'a>(
- callback: &'a (dyn Fn(RunEvent) + 'static),
- window_id: WindowId,
- webviews: &mut MutexGuard<'a, HashMap<WindowId, WebviewWrapper>>,
- control_flow: &mut ControlFlow,
- #[cfg(feature = "menu")] menu_event_listeners: MenuEventListeners,
- ) {
- if let Some(webview) = webviews.remove(&window_id) {
- #[cfg(feature = "menu")]
- menu_event_listeners.lock().unwrap().remove(&window_id);
- callback(RunEvent::WindowClose(webview.label));
- }
- if webviews.is_empty() {
- *control_flow = ControlFlow::Exit;
- callback(RunEvent::Exit);
- }
- }
- fn center_window(window: &Window) -> Result<()> {
- if let Some(monitor) = window.current_monitor() {
- let screen_size = monitor.size();
- let window_size = window.inner_size();
- let x = (screen_size.width - window_size.width) / 2;
- let y = (screen_size.height - window_size.height) / 2;
- window.set_outer_position(WryPhysicalPosition::new(x, y));
- Ok(())
- } else {
- Err(Error::FailedToGetMonitor)
- }
- }
- fn create_webview<P: Params<Runtime = Wry>>(
- event_loop: &EventLoopWindowTarget<Message>,
- context: DispatcherContext,
- pending: PendingWindow<P>,
- ) -> Result<WebviewWrapper> {
- #[allow(unused_mut)]
- let PendingWindow {
- webview_attributes,
- mut window_builder,
- rpc_handler,
- file_drop_handler,
- label,
- url,
- ..
- } = pending;
- let is_window_transparent = window_builder.inner.window.transparent;
- #[cfg(feature = "menu")]
- let menu_items = {
- let mut menu_items = HashMap::new();
- let menu = to_wry_menu(&mut menu_items, window_builder.menu);
- window_builder.inner = window_builder.inner.with_menu(menu);
- menu_items
- };
- let window = window_builder.inner.build(event_loop).unwrap();
- context
- .window_event_listeners
- .lock()
- .unwrap()
- .insert(window.id(), WindowEventListenersMap::default());
- #[cfg(feature = "menu")]
- context
- .menu_event_listeners
- .lock()
- .unwrap()
- .insert(window.id(), WindowMenuEventListeners::default());
- if window_builder.center {
- let _ = center_window(&window);
- }
- let mut webview_builder = WebViewBuilder::new(window)
- .map_err(|e| Error::CreateWebview(Box::new(e)))?
- .with_url(&url)
- .unwrap() // safe to unwrap because we validate the URL beforehand
- .with_transparent(is_window_transparent);
- if let Some(handler) = rpc_handler {
- webview_builder =
- webview_builder.with_rpc_handler(create_rpc_handler(context.clone(), label.clone(), handler));
- }
- if let Some(handler) = file_drop_handler {
- webview_builder = webview_builder.with_file_drop_handler(create_file_drop_handler(
- context,
- label.clone(),
- handler,
- ));
- }
- for (scheme, protocol) in webview_attributes.uri_scheme_protocols {
- webview_builder = webview_builder.with_custom_protocol(scheme, move |_window, url| {
- protocol(url)
- .map(|data| {
- let mime_type = MimeType::parse(&data, url);
- (data, mime_type)
- })
- .map_err(|_| wry::Error::InitScriptError)
- });
- }
- let context = WebContext::new(webview_attributes.data_directory);
- webview_builder = webview_builder.with_web_context(&context);
- for script in webview_attributes.initialization_scripts {
- webview_builder = webview_builder.with_initialization_script(&script);
- }
- let webview = webview_builder
- .build()
- .map_err(|e| Error::CreateWebview(Box::new(e)))?;
- Ok(WebviewWrapper {
- label: format!("{}", label),
- inner: webview,
- #[cfg(feature = "menu")]
- menu_items,
- #[cfg(feature = "menu")]
- is_menu_visible: AtomicBool::new(true),
- })
- }
- /// Create a wry rpc handler from a tauri rpc handler.
- fn create_rpc_handler<P: Params<Runtime = Wry>>(
- context: DispatcherContext,
- label: P::Label,
- handler: WebviewRpcHandler<P>,
- ) -> Box<dyn Fn(&Window, WryRpcRequest) -> Option<RpcResponse> + 'static> {
- Box::new(move |window, request| {
- handler(
- DetachedWindow {
- dispatcher: WryDispatcher {
- window_id: window.id(),
- context: context.clone(),
- },
- label: label.clone(),
- },
- RpcRequestWrapper(request).into(),
- );
- None
- })
- }
- /// Create a wry file drop handler from a tauri file drop handler.
- fn create_file_drop_handler<P: Params<Runtime = Wry>>(
- context: DispatcherContext,
- label: P::Label,
- handler: FileDropHandler<P>,
- ) -> Box<dyn Fn(&Window, WryFileDropEvent) -> bool + 'static> {
- Box::new(move |window, event| {
- handler(
- FileDropEventWrapper(event).into(),
- DetachedWindow {
- dispatcher: WryDispatcher {
- window_id: window.id(),
- context: context.clone(),
- },
- label: label.clone(),
- },
- )
- })
- }
|