| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001 |
- // 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::{
- hash_map::Entry::{Occupied, Vacant},
- HashMap,
- },
- convert::TryFrom,
- fs::read,
- path::PathBuf,
- 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 WebContextStore = Mutex<HashMap<Option<PathBuf>, WebContext>>;
- type CreateWebviewHandler = Box<
- dyn FnOnce(&EventLoopWindowTarget<Message>, &WebContextStore) -> 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: 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.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.clear();
- Ok(())
- }
- fn unregister(&mut self, accelerator: &str) -> Result<()> {
- if let Some((accelerator_id, shortcut)) = self.shortcuts.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(*mut std::ffi::c_void);
- #[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<*mut std::ffi::c_void> {
- 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, web_context| create_webview(event_loop, web_context, 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>,
- web_context: WebContextStore,
- 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, web_context| {
- create_webview(event_loop, web_context, 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,
- web_context: Default::default(),
- 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,
- &self.web_context,
- 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 web_context = &self.web_context;
- 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.clone(),
- global_shortcut_manager: global_shortcut_manager.clone(),
- global_shortcut_manager_handle: global_shortcut_manager_handle.clone(),
- clipboard_manager: clipboard_manager.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: menu_event_listeners.clone(),
- #[cfg(feature = "system-tray")]
- tray_context: tray_context.clone(),
- },
- web_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 web_context = self.web_context;
- 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.clone(),
- global_shortcut_manager: global_shortcut_manager.clone(),
- global_shortcut_manager_handle: global_shortcut_manager_handle.clone(),
- clipboard_manager: clipboard_manager.clone(),
- #[cfg(feature = "menu")]
- menu_event_listeners: menu_event_listeners.clone(),
- #[cfg(feature = "system-tray")]
- tray_context: tray_context.clone(),
- },
- &web_context,
- );
- })
- }
- }
- struct EventLoopIterationContext<'a> {
- callback: &'a (dyn Fn(RunEvent) + 'static),
- webviews: MutexGuard<'a, HashMap<WindowId, WebviewWrapper>>,
- window_event_listeners: WindowEventListeners,
- global_shortcut_manager: Arc<Mutex<WryShortcutManager>>,
- global_shortcut_manager_handle: GlobalShortcutManagerHandle,
- clipboard_manager: Arc<Mutex<Clipboard>>,
- #[cfg(feature = "menu")]
- menu_event_listeners: MenuEventListeners,
- #[cfg(feature = "system-tray")]
- tray_context: TrayContext,
- }
- fn handle_event_loop(
- event: Event<Message>,
- event_loop: &EventLoopWindowTarget<Message>,
- control_flow: &mut ControlFlow,
- context: EventLoopIterationContext<'_>,
- web_context: &WebContextStore,
- ) -> 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())).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, web_context) {
- 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>,
- web_context: &WebContextStore,
- 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)
- });
- }
- for script in webview_attributes.initialization_scripts {
- webview_builder = webview_builder.with_initialization_script(&script);
- }
- let mut web_context = web_context.lock().expect("poisoned WebContext store");
- let is_first_context = web_context.is_empty();
- let web_context = match web_context.entry(webview_attributes.data_directory) {
- Occupied(occupied) => occupied.into_mut(),
- Vacant(vacant) => {
- let mut web_context = WebContext::new(vacant.key().clone());
- web_context.set_allows_automation(match std::env::var("TAURI_AUTOMATION").as_deref() {
- Ok("true") => is_first_context,
- _ => false,
- });
- vacant.insert(web_context)
- }
- };
- let webview = webview_builder
- .with_web_context(web_context)
- .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(),
- },
- )
- })
- }
|