Convert use_eval to use send/recv system (#1080)

* progress: reworked

don't run this, it'll kill your web browser

* feat: use_eval but with comms

* revision: async recv & recv_sync

* revision: use_eval

* revision: standard eval interface

* revision: use serde_json::Value instead of JsValue

* revision: docs

* revision: error message

* create: desktop eval (wip)

* fix: desktop eval

* revision: wrap use_eval in Rc<RefCell<_>>

* fix: fmt, clippy

* fix: desktop tests

* revision: change to channel system

- fixes clippy errors
- fixes playwright tests

* fix: tests

* fix: eval example

* fix: fmt

* fix: tests, desktop stuff

* fix: tests

* feat: drop handler

* fix: tests

* fix: rustfmt

* revision: web promise/callback system

* fix: recv error

* revision: IntoFuture, functionation

* fix: ci

* revision: playwright web

* remove: unescessary code

* remove dioxus-html from public examples

* prototype-patch

* fix web eval

* fix: rustfmt

* fix: CI

* make use_eval more efficient

* implement eval for liveview as well

* fix playwright tests

* fix clippy

* more clippy fixes

* fix clippy

* fix stack overflow

* fix desktop mock

* fix clippy

---------

Co-authored-by: Evan Almloff <evanalmloff@gmail.com>
This commit is contained in:
Miles MurgawandEvan Almloff authored and GitHub committed 2023-07-21 17:36:25 -05:00
1 parent b36a7a3993
commit 6210c6fefe
31 files changed
+1025 -410

No files matched your search

+28 -23
View File
@@ -5,29 +5,34 @@ fn main() {
} }
fn app(cx: Scope) -> Element { fn app(cx: Scope) -> Element {
let eval = dioxus_desktop::use_eval(cx); let eval_provider = use_eval(cx);
let script = use_state(cx, String::new);
let output = use_state(cx, String::new);
cx.render(rsx! { let future = use_future(cx, (), |_| {
div { to_owned![eval_provider];
p { "Output: {output}" } async move {
input { let eval = eval_provider(
placeholder: "Enter an expression", r#"
value: "{script}", dioxus.send("Hi from JS!");
oninput: move |e| script.set(e.value.clone()), let msg = await dioxus.recv();
} console.log(msg);
button { return "hello world";
onclick: move |_| { "#,
to_owned![script, eval, output]; )
cx.spawn(async move { .unwrap();
if let Ok(res) = eval(script.to_string()).await {
output.set(res.to_string()); eval.send("Hi from Rust!".into()).unwrap();
} let res = eval.recv().await.unwrap();
}); println!("{:?}", eval.await);
}, res
"Execute"
}
} }
}) });
match future.value() {
Some(v) => cx.render(rsx!(
p { "{v}" }
)),
_ => cx.render(rsx!(
p { "hello" }
)),
}
} }
+1
View File
@@ -36,6 +36,7 @@ slab = { workspace = true }
futures-util = { workspace = true } futures-util = { workspace = true }
urlencoding = "2.1.2" urlencoding = "2.1.2"
async-trait = "0.1.68"
[target.'cfg(any(target_os = "windows",target_os = "macos",target_os = "linux",target_os = "dragonfly", target_os = "freebsd", target_os = "netbsd", target_os = "openbsd"))'.dependencies] [target.'cfg(any(target_os = "windows",target_os = "macos",target_os = "linux",target_os = "dragonfly", target_os = "freebsd", target_os = "netbsd", target_os = "openbsd"))'.dependencies]
+103 -100
View File
@@ -29,21 +29,24 @@ pub fn main() {
} }
fn mock_event(cx: &ScopeState, id: &'static str, value: &'static str) { fn mock_event(cx: &ScopeState, id: &'static str, value: &'static str) {
use_effect(cx, (), |_| { let eval_provider = use_eval(cx).clone();
let desktop_context: DesktopContext = cx.consume_context().unwrap();
async move { use_effect(cx, (), move |_| async move {
tokio::time::sleep(std::time::Duration::from_millis(100)).await; tokio::time::sleep(std::time::Duration::from_millis(100)).await;
desktop_context.eval(&format!( let js = format!(
r#"let element = document.getElementById('{}'); r#"
//console.log("ran");
// Dispatch a synthetic event // Dispatch a synthetic event
const event = {}; let event = {};
let element = document.getElementById('{}');
console.log(element, event); console.log(element, event);
element.dispatchEvent(event); element.dispatchEvent(event);
"#, "#,
id, value value, id
)); );
}
}); eval_provider(&js).unwrap();
})
} }
#[allow(deprecated)] #[allow(deprecated)]
@@ -56,149 +59,149 @@ fn app(cx: Scope) -> Element {
cx, cx,
"button", "button",
r#"new MouseEvent("click", { r#"new MouseEvent("click", {
view: window, view: window,
bubbles: true, bubbles: true,
cancelable: true, cancelable: true,
button: 0, button: 0,
})"#, })"#,
); );
// mouse_move_div // mouse_move_div
mock_event( mock_event(
cx, cx,
"mouse_move_div", "mouse_move_div",
r#"new MouseEvent("mousemove", { r#"new MouseEvent("mousemove", {
view: window, view: window,
bubbles: true, bubbles: true,
cancelable: true, cancelable: true,
buttons: 2, buttons: 2,
})"#, })"#,
); );
// mouse_click_div // mouse_click_div
mock_event( mock_event(
cx, cx,
"mouse_click_div", "mouse_click_div",
r#"new MouseEvent("click", { r#"new MouseEvent("click", {
view: window, view: window,
bubbles: true, bubbles: true,
cancelable: true, cancelable: true,
buttons: 2, buttons: 2,
button: 2, button: 2,
})"#, })"#,
); );
// mouse_dblclick_div // mouse_dblclick_div
mock_event( mock_event(
cx, cx,
"mouse_dblclick_div", "mouse_dblclick_div",
r#"new MouseEvent("dblclick", { r#"new MouseEvent("dblclick", {
view: window, view: window,
bubbles: true, bubbles: true,
cancelable: true, cancelable: true,
buttons: 1|2, buttons: 1|2,
button: 2, button: 2,
})"#, })"#,
); );
// mouse_down_div // mouse_down_div
mock_event( mock_event(
cx, cx,
"mouse_down_div", "mouse_down_div",
r#"new MouseEvent("mousedown", { r#"new MouseEvent("mousedown", {
view: window, view: window,
bubbles: true, bubbles: true,
cancelable: true, cancelable: true,
buttons: 2, buttons: 2,
button: 2, button: 2,
})"#, })"#,
); );
// mouse_up_div // mouse_up_div
mock_event( mock_event(
cx, cx,
"mouse_up_div", "mouse_up_div",
r#"new MouseEvent("mouseup", { r#"new MouseEvent("mouseup", {
view: window, view: window,
bubbles: true, bubbles: true,
cancelable: true, cancelable: true,
buttons: 0, buttons: 0,
button: 0, button: 0,
})"#, })"#,
); );
// wheel_div // wheel_div
mock_event( mock_event(
cx, cx,
"wheel_div", "wheel_div",
r#"new WheelEvent("wheel", { r#"new WheelEvent("wheel", {
view: window, view: window,
deltaX: 1.0, deltaX: 1.0,
deltaY: 2.0, deltaY: 2.0,
deltaZ: 3.0, deltaZ: 3.0,
deltaMode: 0x00, deltaMode: 0x00,
bubbles: true, bubbles: true,
})"#, })"#,
); );
// key_down_div // key_down_div
mock_event( mock_event(
cx, cx,
"key_down_div", "key_down_div",
r#"new KeyboardEvent("keydown", { r#"new KeyboardEvent("keydown", {
key: "a", key: "a",
code: "KeyA", code: "KeyA",
location: 0, location: 0,
repeat: true, repeat: true,
keyCode: 65, keyCode: 65,
charCode: 97, charCode: 97,
char: "a", char: "a",
charCode: 0, charCode: 0,
altKey: false, altKey: false,
ctrlKey: false, ctrlKey: false,
metaKey: false, metaKey: false,
shiftKey: false, shiftKey: false,
isComposing: false, isComposing: false,
which: 65, which: 65,
bubbles: true, bubbles: true,
})"#, })"#,
); );
// key_up_div // key_up_div
mock_event( mock_event(
cx, cx,
"key_up_div", "key_up_div",
r#"new KeyboardEvent("keyup", { r#"new KeyboardEvent("keyup", {
key: "a", key: "a",
code: "KeyA", code: "KeyA",
location: 0, location: 0,
repeat: false, repeat: false,
keyCode: 65, keyCode: 65,
charCode: 97, charCode: 97,
char: "a", char: "a",
charCode: 0, charCode: 0,
altKey: false, altKey: false,
ctrlKey: false, ctrlKey: false,
metaKey: false, metaKey: false,
shiftKey: false, shiftKey: false,
isComposing: false, isComposing: false,
which: 65, which: 65,
bubbles: true, bubbles: true,
})"#, })"#,
); );
// key_press_div // key_press_div
mock_event( mock_event(
cx, cx,
"key_press_div", "key_press_div",
r#"new KeyboardEvent("keypress", { r#"new KeyboardEvent("keypress", {
key: "a", key: "a",
code: "KeyA", code: "KeyA",
location: 0, location: 0,
repeat: false, repeat: false,
keyCode: 65, keyCode: 65,
charCode: 97, charCode: 97,
char: "a", char: "a",
charCode: 0, charCode: 0,
altKey: false, altKey: false,
ctrlKey: false, ctrlKey: false,
metaKey: false, metaKey: false,
shiftKey: false, shiftKey: false,
isComposing: false, isComposing: false,
which: 65, which: 65,
bubbles: true, bubbles: true,
})"#, })"#,
); );
// focus_in_div // focus_in_div
mock_event( mock_event(
+16 -14
View File
@@ -27,20 +27,20 @@ fn main() {
} }
fn use_inner_html(cx: &ScopeState, id: &'static str) -> Option<String> { fn use_inner_html(cx: &ScopeState, id: &'static str) -> Option<String> {
let eval_provider = use_eval(cx);
let value: &UseRef<Option<String>> = use_ref(cx, || None); let value: &UseRef<Option<String>> = use_ref(cx, || None);
use_effect(cx, (), |_| { use_effect(cx, (), |_| {
to_owned![value]; to_owned![value, eval_provider];
let desktop_context: DesktopContext = cx.consume_context().unwrap();
async move { async move {
tokio::time::sleep(std::time::Duration::from_millis(100)).await; tokio::time::sleep(std::time::Duration::from_millis(100)).await;
let html = desktop_context let html = eval_provider(&format!(
.eval(&format!( r#"let element = document.getElementById('{}');
r#"let element = document.getElementById('{}'); return element.innerHTML"#,
return element.innerHTML;"#, id
id ))
)) .unwrap();
.await; if let Ok(serde_json::Value::String(html)) = html.await {
if let Ok(serde_json::Value::String(html)) = html {
println!("html: {}", html); println!("html: {}", html);
value.set(Some(html)); value.set(Some(html));
} }
@@ -53,13 +53,15 @@ const EXPECTED_HTML: &str = r#"<div id="5" style="width: 100px; height: 100px; c
fn check_html_renders(cx: Scope) -> Element { fn check_html_renders(cx: Scope) -> Element {
let inner_html = use_inner_html(cx, "main_div"); let inner_html = use_inner_html(cx, "main_div");
let desktop_context: DesktopContext = cx.consume_context().unwrap(); let desktop_context: DesktopContext = cx.consume_context().unwrap();
if let Some(raw_html) = inner_html.as_deref() { if let Some(raw_html) = inner_html {
let fragment = scraper::Html::parse_fragment(raw_html); println!("{}", raw_html);
println!("fragment: {:?}", fragment.html()); let fragment = scraper::Html::parse_fragment(&raw_html);
println!("fragment: {}", fragment.html());
let expected = scraper::Html::parse_fragment(EXPECTED_HTML); let expected = scraper::Html::parse_fragment(EXPECTED_HTML);
println!("fragment: {:?}", expected.html()); println!("expected: {}", expected.html());
if fragment == expected { if fragment == expected {
println!("html matches"); println!("html matches");
desktop_context.close(); desktop_context.close();
-9
View File
@@ -3,7 +3,6 @@ use std::rc::Rc;
use std::rc::Weak; use std::rc::Weak;
use crate::create_new_window; use crate::create_new_window;
use crate::eval::EvalResult;
use crate::events::IpcMessage; use crate::events::IpcMessage;
use crate::query::QueryEngine; use crate::query::QueryEngine;
use crate::shortcut::ShortcutId; use crate::shortcut::ShortcutId;
@@ -213,14 +212,6 @@ impl DesktopService {
log::warn!("Devtools are disabled in release builds"); log::warn!("Devtools are disabled in release builds");
} }
/// Evaluate a javascript expression
pub fn eval(&self, code: &str) -> EvalResult {
// the query id lets us keep track of the eval result and send it back to the main thread
let query = self.query.new_query(code, &self.webview);
EvalResult::new(query)
}
/// Create a wry event handler that listens for wry events. /// Create a wry event handler that listens for wry events.
/// This event handler is scoped to the currently active window and will only recieve events that are either global or related to the current window. /// This event handler is scoped to the currently active window and will only recieve events that are either global or related to the current window.
/// ///
+3 -3
View File
@@ -34,7 +34,7 @@ impl RenderedElementBacking for DesktopElement {
let fut = self let fut = self
.query .query
.new_query::<Option<Rect<f64, f64>>>(&script, &self.webview.webview) .new_query::<Option<Rect<f64, f64>>>(&script, self.webview.clone())
.resolve(); .resolve();
Box::pin(async move { Box::pin(async move {
match fut.await { match fut.await {
@@ -61,7 +61,7 @@ impl RenderedElementBacking for DesktopElement {
let fut = self let fut = self
.query .query
.new_query::<bool>(&script, &self.webview.webview) .new_query::<bool>(&script, self.webview.clone())
.resolve(); .resolve();
Box::pin(async move { Box::pin(async move {
match fut.await { match fut.await {
@@ -87,7 +87,7 @@ impl RenderedElementBacking for DesktopElement {
let fut = self let fut = self
.query .query
.new_query::<bool>(&script, &self.webview.webview) .new_query::<bool>(&script, self.webview.clone())
.resolve(); .resolve();
Box::pin(async move { Box::pin(async move {
+56 -27
View File
@@ -1,41 +1,70 @@
use std::rc::Rc; #![allow(clippy::await_holding_refcell_ref)]
use async_trait::async_trait;
use crate::query::Query;
use crate::query::QueryError;
use crate::use_window;
use dioxus_core::ScopeState; use dioxus_core::ScopeState;
use std::future::Future; use dioxus_html::prelude::{EvalError, EvalProvider, Evaluator};
use std::future::IntoFuture; use std::{cell::RefCell, rc::Rc};
use std::pin::Pin;
/// A future that resolves to the result of a JavaScript evaluation. use crate::{query::Query, DesktopContext};
pub struct EvalResult {
pub(crate) query: Query<serde_json::Value>, /// Provides the DesktopEvalProvider through [`cx.provide_context`].
pub fn init_eval(cx: &ScopeState) {
let desktop_ctx = cx.consume_context::<DesktopContext>().unwrap();
let provider: Rc<dyn EvalProvider> = Rc::new(DesktopEvalProvider { desktop_ctx });
cx.provide_context(provider);
} }
impl EvalResult { /// Reprents the desktop-target's provider of evaluators.
pub(crate) fn new(query: Query<serde_json::Value>) -> Self { pub struct DesktopEvalProvider {
Self { query } desktop_ctx: DesktopContext,
}
impl EvalProvider for DesktopEvalProvider {
fn new_evaluator(&self, js: String) -> Result<Rc<dyn Evaluator>, EvalError> {
Ok(Rc::new(DesktopEvaluator::new(self.desktop_ctx.clone(), js)))
} }
} }
impl IntoFuture for EvalResult { /// Reprents a desktop-target's JavaScript evaluator.
type Output = Result<serde_json::Value, QueryError>; pub struct DesktopEvaluator {
query: Rc<RefCell<Query<serde_json::Value>>>,
}
type IntoFuture = Pin<Box<dyn Future<Output = Result<serde_json::Value, QueryError>>>>; impl DesktopEvaluator {
/// Creates a new evaluator for desktop-based targets.
pub fn new(desktop_ctx: DesktopContext, js: String) -> Self {
let ctx = desktop_ctx.clone();
let query = desktop_ctx.query.new_query(&js, ctx);
fn into_future(self) -> Self::IntoFuture { Self {
Box::pin(self.query.resolve()) query: Rc::new(RefCell::new(query)),
as Pin<Box<dyn Future<Output = Result<serde_json::Value, QueryError>>>> }
} }
} }
/// Get a closure that executes any JavaScript in the WebView context. #[async_trait(?Send)]
pub fn use_eval(cx: &ScopeState) -> &Rc<dyn Fn(String) -> EvalResult> { impl Evaluator for DesktopEvaluator {
let desktop = use_window(cx); async fn join(&self) -> Result<serde_json::Value, EvalError> {
&*cx.use_hook(|| { self.query
let desktop = desktop.clone(); .borrow_mut()
.result()
.await
.map_err(|e| EvalError::Communication(e.to_string()))
}
Rc::new(move |script: String| desktop.eval(&script)) as Rc<dyn Fn(String) -> EvalResult> /// Sends a message to the evaluated JavaScript.
}) fn send(&self, data: serde_json::Value) -> Result<(), EvalError> {
if let Err(e) = self.query.borrow_mut().send(data) {
return Err(EvalError::Communication(e.to_string()));
}
Ok(())
}
/// Gets an UnboundedReceiver to receive messages from the evaluated JavaScript.
async fn recv(&self) -> Result<serde_json::Value, EvalError> {
self.query
.borrow_mut()
.recv()
.await
.map_err(|e| EvalError::Communication(e.to_string()))
}
} }
+10 -4
View File
@@ -30,7 +30,7 @@ use dioxus_core::*;
use dioxus_html::MountedData; use dioxus_html::MountedData;
use dioxus_html::{native_bind::NativeFileEngine, FormData, HtmlEvent}; use dioxus_html::{native_bind::NativeFileEngine, FormData, HtmlEvent};
use element::DesktopElement; use element::DesktopElement;
pub use eval::{use_eval, EvalResult}; use eval::init_eval;
use futures_util::{pin_mut, FutureExt}; use futures_util::{pin_mut, FutureExt};
use shortcut::ShortcutRegistry; use shortcut::ShortcutRegistry;
pub use shortcut::{use_global_shortcut, ShortcutHandle, ShortcutId, ShortcutRegistryError}; pub use shortcut::{use_global_shortcut, ShortcutHandle, ShortcutId, ShortcutRegistryError};
@@ -204,15 +204,17 @@ pub fn launch_with_props<P: 'static>(root: Component<P>, props: P, cfg: Config)
}, },
Event::NewEvents(StartCause::Init) => { Event::NewEvents(StartCause::Init) => {
//
let props = props.take().unwrap(); let props = props.take().unwrap();
let cfg = cfg.take().unwrap(); let cfg = cfg.take().unwrap();
// Create a dom
let dom = VirtualDom::new_with_props(root, props);
let handler = create_new_window( let handler = create_new_window(
cfg, cfg,
event_loop, event_loop,
&proxy, &proxy,
VirtualDom::new_with_props(root, props), dom,
&queue, &queue,
&event_handlers, &event_handlers,
shortcut_manager.clone(), shortcut_manager.clone(),
@@ -389,7 +391,11 @@ fn create_new_window(
shortcut_manager, shortcut_manager,
)); ));
dom.base_scope().provide_context(desktop_context.clone()); let cx = dom.base_scope();
cx.provide_context(desktop_context.clone());
// Init eval
init_eval(cx);
WebviewHandler { WebviewHandler {
// We want to poll the virtualdom and the event loop at the same time, so the waker will be connected to both // We want to poll the virtualdom and the event loop at the same time, so the waker will be connected to both
+196 -47
View File
@@ -1,52 +1,127 @@
use std::{cell::RefCell, rc::Rc}; use std::{cell::RefCell, rc::Rc};
use crate::DesktopContext;
use serde::{de::DeserializeOwned, Deserialize}; use serde::{de::DeserializeOwned, Deserialize};
use serde_json::Value; use serde_json::Value;
use slab::Slab; use slab::Slab;
use thiserror::Error; use thiserror::Error;
use tokio::sync::broadcast::error::RecvError; use tokio::sync::broadcast::error::RecvError;
use wry::webview::WebView;
const DIOXUS_CODE: &str = r#"
let dioxus = {
recv: function () {
return new Promise((resolve, _reject) => {
// Ever 50 ms check for new data
let timeout = setTimeout(() => {
let __msg = null;
while (true) {
let __data = _message_queue.shift();
if (__data) {
__msg = __data;
break;
}
}
clearTimeout(timeout);
resolve(__msg);
}, 50);
});
},
send: function (value) {
window.ipc.postMessage(
JSON.stringify({
"method":"query",
"params": {
"id": _request_id,
"data": value,
"returned_value": false
}
})
);
}
}"#;
/// Tracks what query ids are currently active /// Tracks what query ids are currently active
#[derive(Default, Clone)]
struct SharedSlab { struct SharedSlab<T = ()> {
slab: Rc<RefCell<Slab<()>>>, slab: Rc<RefCell<Slab<T>>>,
} }
/// Handles sending and receiving arbitrary queries from the webview. Queries can be resolved non-sequentially, so we use ids to track them. impl<T> Clone for SharedSlab<T> {
#[derive(Clone)] fn clone(&self) -> Self {
pub(crate) struct QueryEngine {
sender: Rc<tokio::sync::broadcast::Sender<QueryResult>>,
active_requests: SharedSlab,
}
impl Default for QueryEngine {
fn default() -> Self {
let (sender, _) = tokio::sync::broadcast::channel(1000);
Self { Self {
sender: Rc::new(sender), slab: self.slab.clone(),
active_requests: SharedSlab::default(),
} }
} }
} }
impl<T> Default for SharedSlab<T> {
fn default() -> Self {
SharedSlab {
slab: Rc::new(RefCell::new(Slab::new())),
}
}
}
struct QueryEntry {
channel_sender: tokio::sync::mpsc::UnboundedSender<Value>,
return_sender: Option<tokio::sync::oneshot::Sender<Value>>,
}
const QUEUE_NAME: &str = "__msg_queues";
/// Handles sending and receiving arbitrary queries from the webview. Queries can be resolved non-sequentially, so we use ids to track them.
#[derive(Clone, Default)]
pub(crate) struct QueryEngine {
active_requests: SharedSlab<QueryEntry>,
}
impl QueryEngine { impl QueryEngine {
/// Creates a new query and returns a handle to it. The query will be resolved when the webview returns a result with the same id. /// Creates a new query and returns a handle to it. The query will be resolved when the webview returns a result with the same id.
pub fn new_query<V: DeserializeOwned>(&self, script: &str, webview: &WebView) -> Query<V> { pub fn new_query<V: DeserializeOwned>(
let request_id = self.active_requests.slab.borrow_mut().insert(()); &self,
script: &str,
context: DesktopContext,
) -> Query<V> {
let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
let (return_tx, return_rx) = tokio::sync::oneshot::channel();
let request_id = self.active_requests.slab.borrow_mut().insert(QueryEntry {
channel_sender: tx,
return_sender: Some(return_tx),
});
// start the query // start the query
// We embed the return of the eval in a function so we can send it back to the main thread // We embed the return of the eval in a function so we can send it back to the main thread
if let Err(err) = webview.evaluate_script(&format!( if let Err(err) = context.webview.evaluate_script(&format!(
r#"window.ipc.postMessage( r#"(function(){{
JSON.stringify({{ (async (resolve, _reject) => {{
"method":"query", {DIOXUS_CODE}
"params": {{ if (!window.{QUEUE_NAME}) {{
"id": {request_id}, window.{QUEUE_NAME} = [];
"data": (function(){{{script}}})()
}} }}
let _request_id = {request_id};
if (!window.{QUEUE_NAME}[{request_id}]) {{
window.{QUEUE_NAME}[{request_id}] = [];
}}
let _message_queue = window.{QUEUE_NAME}[{request_id}];
{script}
}})().then((result)=>{{
let returned_value = {{
"method":"query",
"params": {{
"id": {request_id},
"data": result,
"returned_value": true
}}
}};
window.ipc.postMessage(
JSON.stringify(returned_value)
);
}}) }})
);"# }})();"#
)) { )) {
log::warn!("Query error: {err}"); log::warn!("Query error: {err}");
} }
@@ -54,57 +129,131 @@ impl QueryEngine {
Query { Query {
slab: self.active_requests.clone(), slab: self.active_requests.clone(),
id: request_id, id: request_id,
reciever: self.sender.subscribe(), receiver: rx,
return_receiver: Some(return_rx),
desktop: context,
phantom: std::marker::PhantomData, phantom: std::marker::PhantomData,
} }
} }
/// Send a query result /// Send a query channel message to the correct query
pub fn send(&self, data: QueryResult) { pub fn send(&self, data: QueryResult) {
let _ = self.sender.send(data); let QueryResult {
id,
data,
returned_value,
} = data;
let mut slab = self.active_requests.slab.borrow_mut();
if let Some(entry) = slab.get_mut(id) {
if returned_value {
if let Some(sender) = entry.return_sender.take() {
let _ = sender.send(data);
}
} else {
let _ = entry.channel_sender.send(data);
}
}
} }
} }
pub(crate) struct Query<V: DeserializeOwned> { pub(crate) struct Query<V: DeserializeOwned> {
slab: SharedSlab, desktop: DesktopContext,
slab: SharedSlab<QueryEntry>,
receiver: tokio::sync::mpsc::UnboundedReceiver<Value>,
return_receiver: Option<tokio::sync::oneshot::Receiver<Value>>,
id: usize, id: usize,
reciever: tokio::sync::broadcast::Receiver<QueryResult>,
phantom: std::marker::PhantomData<V>, phantom: std::marker::PhantomData<V>,
} }
impl<V: DeserializeOwned> Query<V> { impl<V: DeserializeOwned> Query<V> {
/// Resolve the query /// Resolve the query
pub async fn resolve(mut self) -> Result<V, QueryError> { pub async fn resolve(mut self) -> Result<V, QueryError> {
let result = loop { match self.receiver.recv().await {
match self.reciever.recv().await { Some(result) => V::deserialize(result).map_err(QueryError::Deserialize),
Ok(result) => { None => Err(QueryError::Recv(RecvError::Closed)),
if result.id == self.id { }
break V::deserialize(result.data).map_err(QueryError::DeserializeError); }
}
}
Err(err) => {
break Err(QueryError::RecvError(err));
}
}
};
// Remove the query from the slab /// Send a message to the query
pub fn send<S: ToString>(&self, message: S) -> Result<(), QueryError> {
let queue_id = self.id;
let data = message.to_string();
let script = format!(
r#"
if (!window.{QUEUE_NAME}) {{
window.{QUEUE_NAME} = [];
}}
if (!window.{QUEUE_NAME}[{queue_id}]) {{
window.{QUEUE_NAME}[{queue_id}] = [];
}}
window.{QUEUE_NAME}[{queue_id}].push({data});
"#
);
self.desktop
.webview
.evaluate_script(&script)
.map_err(|e| QueryError::Send(e.to_string()))?;
Ok(())
}
/// Receive a message from the query
pub async fn recv(&mut self) -> Result<Value, QueryError> {
self.receiver
.recv()
.await
.ok_or(QueryError::Recv(RecvError::Closed))
}
/// Receive the result of the query
pub async fn result(&mut self) -> Result<Value, QueryError> {
match self.return_receiver.take() {
Some(receiver) => receiver
.await
.map_err(|_| QueryError::Recv(RecvError::Closed)),
None => Err(QueryError::Finished),
}
}
}
impl<V: DeserializeOwned> Drop for Query<V> {
fn drop(&mut self) {
self.slab.slab.borrow_mut().remove(self.id); self.slab.slab.borrow_mut().remove(self.id);
let queue_id = self.id;
result _ = self.desktop.webview.evaluate_script(&format!(
r#"
if (!window.{QUEUE_NAME}) {{
window.{QUEUE_NAME} = [];
}}
if (window.{QUEUE_NAME}[{queue_id}]) {{
window.{QUEUE_NAME}[{queue_id}] = [];
}}
"#
));
} }
} }
#[derive(Error, Debug)] #[derive(Error, Debug)]
pub enum QueryError { pub enum QueryError {
#[error("Error receiving query result: {0}")] #[error("Error receiving query result: {0}")]
RecvError(RecvError), Recv(RecvError),
#[error("Error sending message to query: {0}")]
Send(String),
#[error("Error deserializing query result: {0}")] #[error("Error deserializing query result: {0}")]
DeserializeError(serde_json::Error), Deserialize(serde_json::Error),
#[error("Query has already been resolved")]
Finished,
} }
#[derive(Clone, Debug, Deserialize)] #[derive(Clone, Debug, Deserialize)]
pub(crate) struct QueryResult { pub(crate) struct QueryResult {
id: usize, id: usize,
data: Value, data: Value,
#[serde(default)]
returned_value: bool,
} }
+1 -5
View File
@@ -78,11 +78,7 @@ where
<R as std::str::FromStr>::Err: std::fmt::Display, <R as std::str::FromStr>::Err: std::fmt::Display,
{ {
fn clone(&self) -> Self { fn clone(&self) -> Self {
Self { *self
failure_external_navigation: self.failure_external_navigation,
scroll_restoration: self.scroll_restoration,
phantom: std::marker::PhantomData,
}
} }
} }
+1
View File
@@ -274,6 +274,7 @@ impl<T: Clone> UseState<T> {
/// *val.make_mut() += 1; /// *val.make_mut() += 1;
/// ``` /// ```
#[must_use] #[must_use]
#[allow(clippy::missing_panics_doc)]
pub fn make_mut(&self) -> RefMut<T> { pub fn make_mut(&self) -> RefMut<T> {
let mut slot = self.slot.borrow_mut(); let mut slot = self.slot.borrow_mut();
+4
View File
@@ -21,6 +21,9 @@ keyboard-types = "0.6.2"
async-trait = "0.1.58" async-trait = "0.1.58"
serde-value = "0.7.0" serde-value = "0.7.0"
tokio = { workspace = true, features = ["fs", "io-util"], optional = true } tokio = { workspace = true, features = ["fs", "io-util"], optional = true }
rfd = { version = "0.11.3", optional = true }
async-channel = "1.8.0"
serde_json = { version = "1", optional = true }
[dependencies.web-sys] [dependencies.web-sys]
optional = true optional = true
@@ -54,6 +57,7 @@ default = ["serialize"]
serialize = [ serialize = [
"serde", "serde",
"serde_repr", "serde_repr",
"serde_json",
"euclid/serde", "euclid/serde",
"keyboard-types/serde", "keyboard-types/serde",
"dioxus-core/serialize", "dioxus-core/serialize",
+96
View File
@@ -0,0 +1,96 @@
#![allow(clippy::await_holding_refcell_ref)]
use async_trait::async_trait;
use dioxus_core::ScopeState;
use std::future::{Future, IntoFuture};
use std::pin::Pin;
use std::rc::Rc;
/// A struct that implements EvalProvider is sent through [`ScopeState`]'s provide_context function
/// so that [`use_eval`] can provide a platform agnostic interface for evaluating JavaScript code.
pub trait EvalProvider {
fn new_evaluator(&self, js: String) -> Result<Rc<dyn Evaluator>, EvalError>;
}
/// The platform's evaluator.
#[async_trait(?Send)]
pub trait Evaluator {
/// Sends a message to the evaluated JavaScript.
fn send(&self, data: serde_json::Value) -> Result<(), EvalError>;
/// Receive any queued messages from the evaluated JavaScript.
async fn recv(&self) -> Result<serde_json::Value, EvalError>;
/// Gets the return value of the JavaScript
async fn join(&self) -> Result<serde_json::Value, EvalError>;
}
type EvalCreator = Rc<dyn Fn(&str) -> Result<UseEval, EvalError>>;
/// Get a struct that can execute any JavaScript.
///
/// # Safety
///
/// Please be very careful with this function. A script with too many dynamic
/// parts is practically asking for a hacker to find an XSS vulnerability in
/// it. **This applies especially to web targets, where the JavaScript context
/// has access to most, if not all of your application data.**
pub fn use_eval(cx: &ScopeState) -> &EvalCreator {
&*cx.use_hook(|| {
let eval_provider = cx
.consume_context::<Rc<dyn EvalProvider>>()
.expect("evaluator not provided");
Rc::new(move |script: &str| {
eval_provider
.new_evaluator(script.to_string())
.map(|evaluator| UseEval::new(evaluator))
}) as Rc<dyn Fn(&str) -> Result<UseEval, EvalError>>
})
}
/// A wrapper around the target platform's evaluator.
#[derive(Clone)]
pub struct UseEval {
evaluator: Rc<dyn Evaluator + 'static>,
}
impl UseEval {
/// Creates a new UseEval
pub fn new(evaluator: Rc<dyn Evaluator + 'static>) -> Self {
Self { evaluator }
}
/// Sends a [`serde_json::Value`] to the evaluated JavaScript.
pub fn send(&self, data: serde_json::Value) -> Result<(), EvalError> {
self.evaluator.send(data)
}
/// Gets an UnboundedReceiver to receive messages from the evaluated JavaScript.
pub async fn recv(&self) -> Result<serde_json::Value, EvalError> {
self.evaluator.recv().await
}
/// Gets the return value of the evaluated JavaScript.
pub async fn join(self) -> Result<serde_json::Value, EvalError> {
self.evaluator.join().await
}
}
impl IntoFuture for UseEval {
type Output = Result<serde_json::Value, EvalError>;
type IntoFuture = Pin<Box<dyn Future<Output = Self::Output>>>;
fn into_future(self) -> Self::IntoFuture {
Box::pin(self.join())
}
}
/// Represents an error when evaluating JavaScript
#[derive(Debug)]
pub enum EvalError {
/// The provided JavaScript has already been ran.
Finished,
/// The provided JavaScript is not valid and can't be ran.
InvalidJs(String),
/// Represents an error communicating between JavaScript and Rust.
Communication(String),
}
+3
View File
@@ -37,6 +37,9 @@ pub use events::*;
pub use global_attributes::*; pub use global_attributes::*;
pub use render_template::*; pub use render_template::*;
mod eval;
pub mod prelude { pub mod prelude {
pub use crate::eval::*;
pub use crate::events::*; pub use crate::events::*;
} }
+1
View File
@@ -35,6 +35,7 @@ axum = { version = "0.6.1", optional = true, features = ["ws"] }
# salvo # salvo
salvo = { version = "0.44.1", optional = true, features = ["ws"] } salvo = { version = "0.44.1", optional = true, features = ["ws"] }
once_cell = "1.17.1" once_cell = "1.17.1"
async-trait = "0.1.71"
# actix is ... complicated? # actix is ... complicated?
# actix-files = { version = "0.6.2", optional = true } # actix-files = { version = "0.6.2", optional = true }
+5 -17
View File
@@ -1,23 +1,17 @@
use dioxus_core::ElementId; use dioxus_core::ElementId;
use dioxus_html::{geometry::euclid::Rect, MountedResult, RenderedElementBacking}; use dioxus_html::{geometry::euclid::Rect, MountedResult, RenderedElementBacking};
use tokio::sync::mpsc::UnboundedSender;
use crate::query::QueryEngine; use crate::query::QueryEngine;
/// A mounted element passed to onmounted events /// A mounted element passed to onmounted events
pub struct LiveviewElement { pub struct LiveviewElement {
id: ElementId, id: ElementId,
query_tx: UnboundedSender<String>,
query: QueryEngine, query: QueryEngine,
} }
impl LiveviewElement { impl LiveviewElement {
pub(crate) fn new(id: ElementId, tx: UnboundedSender<String>, query: QueryEngine) -> Self { pub(crate) fn new(id: ElementId, query: QueryEngine) -> Self {
Self { Self { id, query }
id,
query_tx: tx,
query,
}
} }
} }
@@ -39,7 +33,7 @@ impl RenderedElementBacking for LiveviewElement {
let fut = self let fut = self
.query .query
.new_query::<Option<Rect<f64, f64>>>(&script, &self.query_tx) .new_query::<Option<Rect<f64, f64>>>(&script)
.resolve(); .resolve();
Box::pin(async move { Box::pin(async move {
match fut.await { match fut.await {
@@ -64,10 +58,7 @@ impl RenderedElementBacking for LiveviewElement {
serde_json::to_string(&behavior).expect("Failed to serialize ScrollBehavior") serde_json::to_string(&behavior).expect("Failed to serialize ScrollBehavior")
); );
let fut = self let fut = self.query.new_query::<bool>(&script).resolve();
.query
.new_query::<bool>(&script, &self.query_tx)
.resolve();
Box::pin(async move { Box::pin(async move {
match fut.await { match fut.await {
Ok(true) => Ok(()), Ok(true) => Ok(()),
@@ -90,10 +81,7 @@ impl RenderedElementBacking for LiveviewElement {
self.id.0, focus self.id.0, focus
); );
let fut = self let fut = self.query.new_query::<bool>(&script).resolve();
.query
.new_query::<bool>(&script, &self.query_tx)
.resolve();
Box::pin(async move { Box::pin(async move {
match fut.await { match fut.await {
+75
View File
@@ -0,0 +1,75 @@
#![allow(clippy::await_holding_refcell_ref)]
use async_trait::async_trait;
use dioxus_core::ScopeState;
use dioxus_html::prelude::{EvalError, EvalProvider, Evaluator};
use std::{cell::RefCell, rc::Rc};
use crate::query::{Query, QueryEngine};
/// Provides the DesktopEvalProvider through [`cx.provide_context`].
pub fn init_eval(cx: &ScopeState) {
let query = cx.consume_context::<QueryEngine>().unwrap();
let provider: Rc<dyn EvalProvider> = Rc::new(DesktopEvalProvider { query });
cx.provide_context(provider);
}
/// Reprents the desktop-target's provider of evaluators.
pub struct DesktopEvalProvider {
query: QueryEngine,
}
impl EvalProvider for DesktopEvalProvider {
fn new_evaluator(&self, js: String) -> Result<Rc<dyn Evaluator>, EvalError> {
Ok(Rc::new(DesktopEvaluator::new(self.query.clone(), js)))
}
}
/// Reprents a desktop-target's JavaScript evaluator.
pub(crate) struct DesktopEvaluator {
query: Rc<RefCell<Query<serde_json::Value>>>,
}
impl DesktopEvaluator {
/// Creates a new evaluator for desktop-based targets.
pub fn new(query: QueryEngine, js: String) -> Self {
let query = query.new_query(&js);
Self {
query: Rc::new(RefCell::new(query)),
}
}
}
#[async_trait(?Send)]
impl Evaluator for DesktopEvaluator {
/// # Panics
/// This will panic if the query is currently being awaited.
async fn join(&self) -> Result<serde_json::Value, EvalError> {
self.query
.borrow_mut()
.result()
.await
.map_err(|e| EvalError::Communication(e.to_string()))
}
/// Sends a message to the evaluated JavaScript.
fn send(&self, data: serde_json::Value) -> Result<(), EvalError> {
if let Err(e) = self.query.borrow_mut().send(data) {
return Err(EvalError::Communication(e.to_string()));
}
Ok(())
}
/// Gets an UnboundedReceiver to receive messages from the evaluated JavaScript.
///
/// # Panics
/// This will panic if the query is currently being awaited.
async fn recv(&self) -> Result<serde_json::Value, EvalError> {
self.query
.borrow_mut()
.recv()
.await
.map_err(|e| EvalError::Communication(e.to_string()))
}
}
+1
View File
@@ -23,6 +23,7 @@ pub mod pool;
mod query; mod query;
use futures_util::{SinkExt, StreamExt}; use futures_util::{SinkExt, StreamExt};
pub use pool::*; pub use pool::*;
mod eval;
pub trait WebsocketTx: SinkExt<String, Error = LiveViewError> {} pub trait WebsocketTx: SinkExt<String, Error = LiveViewError> {}
impl<T> WebsocketTx for T where T: SinkExt<String, Error = LiveViewError> {} impl<T> WebsocketTx for T where T: SinkExt<String, Error = LiveViewError> {}
+8 -5
View File
@@ -1,5 +1,6 @@
use crate::{ use crate::{
element::LiveviewElement, element::LiveviewElement,
eval::init_eval,
query::{QueryEngine, QueryResult}, query::{QueryEngine, QueryResult},
LiveViewError, LiveViewError,
}; };
@@ -119,6 +120,12 @@ pub async fn run(mut vdom: VirtualDom, ws: impl LiveViewSocket) -> Result<(), Li
rx rx
}; };
// Create the a proxy for query engine
let (query_tx, mut query_rx) = tokio::sync::mpsc::unbounded_channel();
let query_engine = QueryEngine::new(query_tx);
vdom.base_scope().provide_context(query_engine.clone());
init_eval(vdom.base_scope());
// todo: use an efficient binary packed format for this // todo: use an efficient binary packed format for this
let edits = serde_json::to_string(&ClientUpdate::Edits(vdom.rebuild())).unwrap(); let edits = serde_json::to_string(&ClientUpdate::Edits(vdom.rebuild())).unwrap();
@@ -128,10 +135,6 @@ pub async fn run(mut vdom: VirtualDom, ws: impl LiveViewSocket) -> Result<(), Li
// send the initial render to the client // send the initial render to the client
ws.send(edits.into_bytes()).await?; ws.send(edits.into_bytes()).await?;
// Create the a proxy for query engine
let (query_tx, mut query_rx) = tokio::sync::mpsc::unbounded_channel();
let query_engine = QueryEngine::default();
// desktop uses this wrapper struct thing around the actual event itself // desktop uses this wrapper struct thing around the actual event itself
// this is sorta driven by tao/wry // this is sorta driven by tao/wry
#[derive(serde::Deserialize, Debug)] #[derive(serde::Deserialize, Debug)]
@@ -165,7 +168,7 @@ pub async fn run(mut vdom: VirtualDom, ws: impl LiveViewSocket) -> Result<(), Li
IpcMessage::Event(evt) => { IpcMessage::Event(evt) => {
// Intercept the mounted event and insert a custom element type // Intercept the mounted event and insert a custom element type
if let EventData::Mounted = &evt.data { if let EventData::Mounted = &evt.data {
let element = LiveviewElement::new(evt.element, query_tx.clone(), query_engine.clone()); let element = LiveviewElement::new(evt.element, query_engine.clone());
vdom.handle_event( vdom.handle_event(
&evt.name, &evt.name,
Rc::new(MountedData::new(element)), Rc::new(MountedData::new(element)),
+204 -53
View File
@@ -4,110 +4,261 @@ use serde::{de::DeserializeOwned, Deserialize};
use serde_json::Value; use serde_json::Value;
use slab::Slab; use slab::Slab;
use thiserror::Error; use thiserror::Error;
use tokio::sync::{broadcast::error::RecvError, mpsc::UnboundedSender}; use tokio::sync::broadcast::error::RecvError;
const DIOXUS_CODE: &str = r#"
let dioxus = {
recv: function () {
return new Promise((resolve, _reject) => {
// Ever 50 ms check for new data
let timeout = setTimeout(() => {
let __msg = null;
while (true) {
let __data = _message_queue.shift();
if (__data) {
__msg = __data;
break;
}
}
clearTimeout(timeout);
resolve(__msg);
}, 50);
});
},
send: function (value) {
window.ipc.postMessage(
JSON.stringify({
"method":"query",
"params": {
"id": _request_id,
"data": value,
"returned_value": false
}
})
);
}
}"#;
/// Tracks what query ids are currently active /// Tracks what query ids are currently active
#[derive(Default, Clone)]
struct SharedSlab { struct SharedSlab<T = ()> {
slab: Rc<RefCell<Slab<()>>>, slab: Rc<RefCell<Slab<T>>>,
} }
/// Handles sending and receiving arbitrary queries from the webview. Queries can be resolved non-sequentially, so we use ids to track them. impl<T> Clone for SharedSlab<T> {
#[derive(Clone)] fn clone(&self) -> Self {
pub(crate) struct QueryEngine {
sender: Rc<tokio::sync::broadcast::Sender<QueryResult>>,
active_requests: SharedSlab,
}
impl Default for QueryEngine {
fn default() -> Self {
let (sender, _) = tokio::sync::broadcast::channel(8);
Self { Self {
sender: Rc::new(sender), slab: self.slab.clone(),
active_requests: SharedSlab::default(),
} }
} }
} }
impl<T> Default for SharedSlab<T> {
fn default() -> Self {
SharedSlab {
slab: Rc::new(RefCell::new(Slab::new())),
}
}
}
struct QueryEntry {
channel_sender: tokio::sync::mpsc::UnboundedSender<Value>,
return_sender: Option<tokio::sync::oneshot::Sender<Value>>,
}
const QUEUE_NAME: &str = "__msg_queues";
/// Handles sending and receiving arbitrary queries from the webview. Queries can be resolved non-sequentially, so we use ids to track them.
#[derive(Clone)]
pub(crate) struct QueryEngine {
active_requests: SharedSlab<QueryEntry>,
query_tx: tokio::sync::mpsc::UnboundedSender<String>,
}
impl QueryEngine { impl QueryEngine {
pub(crate) fn new(query_tx: tokio::sync::mpsc::UnboundedSender<String>) -> Self {
Self {
active_requests: Default::default(),
query_tx,
}
}
/// Creates a new query and returns a handle to it. The query will be resolved when the webview returns a result with the same id. /// Creates a new query and returns a handle to it. The query will be resolved when the webview returns a result with the same id.
pub fn new_query<V: DeserializeOwned>( pub fn new_query<V: DeserializeOwned>(&self, script: &str) -> Query<V> {
&self, let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
script: &str, let (return_tx, return_rx) = tokio::sync::oneshot::channel();
tx: &UnboundedSender<String>, let request_id = self.active_requests.slab.borrow_mut().insert(QueryEntry {
) -> Query<V> { channel_sender: tx,
let request_id = self.active_requests.slab.borrow_mut().insert(()); return_sender: Some(return_tx),
});
// start the query // start the query
// We embed the return of the eval in a function so we can send it back to the main thread // We embed the return of the eval in a function so we can send it back to the main thread
if let Err(err) = tx.send(format!( if let Err(err) = self.query_tx.send(format!(
r#"window.ipc.postMessage( r#"(function(){{
JSON.stringify({{ (async (resolve, _reject) => {{
"method":"query", {DIOXUS_CODE}
"params": {{ if (!window.{QUEUE_NAME}) {{
"id": {request_id}, window.{QUEUE_NAME} = [];
"data": (function(){{{script}}})()
}} }}
let _request_id = {request_id};
if (!window.{QUEUE_NAME}[{request_id}]) {{
window.{QUEUE_NAME}[{request_id}] = [];
}}
let _message_queue = window.{QUEUE_NAME}[{request_id}];
{script}
}})().then((result)=>{{
let returned_value = {{
"method":"query",
"params": {{
"id": {request_id},
"data": result,
"returned_value": true
}}
}};
window.ipc.postMessage(
JSON.stringify(returned_value)
);
}}) }})
);"# }})();"#
)) { )) {
log::warn!("Query error: {err}"); log::warn!("Query error: {err}");
} }
Query { Query {
slab: self.active_requests.clone(), query_engine: self.clone(),
id: request_id, id: request_id,
reciever: self.sender.subscribe(), receiver: rx,
return_receiver: Some(return_rx),
phantom: std::marker::PhantomData, phantom: std::marker::PhantomData,
} }
} }
/// Send a query result /// Send a query channel message to the correct query
pub fn send(&self, data: QueryResult) { pub fn send(&self, data: QueryResult) {
let _ = self.sender.send(data); let QueryResult {
id,
data,
returned_value,
} = data;
let mut slab = self.active_requests.slab.borrow_mut();
if let Some(entry) = slab.get_mut(id) {
if returned_value {
if let Some(sender) = entry.return_sender.take() {
let _ = sender.send(data);
}
} else {
let _ = entry.channel_sender.send(data);
}
}
} }
} }
pub(crate) struct Query<V: DeserializeOwned> { pub(crate) struct Query<V: DeserializeOwned> {
slab: SharedSlab, query_engine: QueryEngine,
pub receiver: tokio::sync::mpsc::UnboundedReceiver<Value>,
pub return_receiver: Option<tokio::sync::oneshot::Receiver<Value>>,
id: usize, id: usize,
reciever: tokio::sync::broadcast::Receiver<QueryResult>,
phantom: std::marker::PhantomData<V>, phantom: std::marker::PhantomData<V>,
} }
impl<V: DeserializeOwned> Query<V> { impl<V: DeserializeOwned> Query<V> {
/// Resolve the query /// Resolve the query
pub async fn resolve(mut self) -> Result<V, QueryError> { pub async fn resolve(mut self) -> Result<V, QueryError> {
let result = loop { match self.receiver.recv().await {
match self.reciever.recv().await { Some(result) => V::deserialize(result).map_err(QueryError::Deserialize),
Ok(result) => { None => Err(QueryError::Recv(RecvError::Closed)),
if result.id == self.id { }
break V::deserialize(result.data).map_err(QueryError::DeserializeError); }
}
}
Err(err) => {
break Err(QueryError::RecvError(err));
}
}
};
// Remove the query from the slab /// Send a message to the query
self.slab.slab.borrow_mut().remove(self.id); pub fn send<S: ToString>(&self, message: S) -> Result<(), QueryError> {
let queue_id = self.id;
result let data = message.to_string();
let script = format!(
r#"
if (!window.{QUEUE_NAME}) {{
window.{QUEUE_NAME} = [];
}}
if (!window.{QUEUE_NAME}[{queue_id}]) {{
window.{QUEUE_NAME}[{queue_id}] = [];
}}
window.{QUEUE_NAME}[{queue_id}].push({data});
"#
);
self.query_engine
.query_tx
.send(script)
.map_err(|e| QueryError::Send(e.to_string()))?;
Ok(())
}
/// Receive a message from the query
pub async fn recv(&mut self) -> Result<Value, QueryError> {
self.receiver
.recv()
.await
.ok_or(QueryError::Recv(RecvError::Closed))
}
/// Receive the result of the query
pub async fn result(&mut self) -> Result<Value, QueryError> {
match self.return_receiver.take() {
Some(receiver) => receiver
.await
.map_err(|_| QueryError::Recv(RecvError::Closed)),
None => Err(QueryError::Finished),
}
}
}
impl<V: DeserializeOwned> Drop for Query<V> {
fn drop(&mut self) {
self.query_engine
.active_requests
.slab
.borrow_mut()
.remove(self.id);
let queue_id = self.id;
_ = self.query_engine.query_tx.send(format!(
r#"
if (!window.{QUEUE_NAME}) {{
window.{QUEUE_NAME} = [];
}}
if (window.{QUEUE_NAME}[{queue_id}]) {{
window.{QUEUE_NAME}[{queue_id}] = [];
}}
"#
));
} }
} }
#[derive(Error, Debug)] #[derive(Error, Debug)]
pub enum QueryError { pub enum QueryError {
#[error("Error receiving query result: {0}")] #[error("Error receiving query result: {0}")]
RecvError(RecvError), Recv(RecvError),
#[error("Error sending message to query: {0}")]
Send(String),
#[error("Error deserializing query result: {0}")] #[error("Error deserializing query result: {0}")]
DeserializeError(serde_json::Error), Deserialize(serde_json::Error),
#[error("Query has already been resolved")]
Finished,
} }
#[derive(Clone, Debug, Deserialize)] #[derive(Clone, Debug, Deserialize)]
pub(crate) struct QueryResult { pub(crate) struct QueryResult {
id: usize, id: usize,
data: Value, data: Value,
#[serde(default)]
returned_value: bool,
} }
+1 -4
View File
@@ -691,10 +691,7 @@ pub struct NodeRef<'a, V: FromAnyValue + Send + Sync = ()> {
impl<'a, V: FromAnyValue + Send + Sync> Clone for NodeRef<'a, V> { impl<'a, V: FromAnyValue + Send + Sync> Clone for NodeRef<'a, V> {
fn clone(&self) -> Self { fn clone(&self) -> Self {
Self { *self
id: self.id,
dom: self.dom,
}
} }
} }
+12 -12
View File
@@ -39,23 +39,23 @@ impl FocusLevel {
impl PartialOrd for FocusLevel { impl PartialOrd for FocusLevel {
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> { fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
match (self, other) { Some(self.cmp(other))
(FocusLevel::Unfocusable, FocusLevel::Unfocusable) => Some(std::cmp::Ordering::Equal),
(FocusLevel::Unfocusable, FocusLevel::Focusable) => Some(std::cmp::Ordering::Less),
(FocusLevel::Unfocusable, FocusLevel::Ordered(_)) => Some(std::cmp::Ordering::Less),
(FocusLevel::Focusable, FocusLevel::Unfocusable) => Some(std::cmp::Ordering::Greater),
(FocusLevel::Focusable, FocusLevel::Focusable) => Some(std::cmp::Ordering::Equal),
(FocusLevel::Focusable, FocusLevel::Ordered(_)) => Some(std::cmp::Ordering::Greater),
(FocusLevel::Ordered(_), FocusLevel::Unfocusable) => Some(std::cmp::Ordering::Greater),
(FocusLevel::Ordered(_), FocusLevel::Focusable) => Some(std::cmp::Ordering::Less),
(FocusLevel::Ordered(a), FocusLevel::Ordered(b)) => a.partial_cmp(b),
}
} }
} }
impl Ord for FocusLevel { impl Ord for FocusLevel {
fn cmp(&self, other: &Self) -> std::cmp::Ordering { fn cmp(&self, other: &Self) -> std::cmp::Ordering {
self.partial_cmp(other).unwrap() match (self, other) {
(FocusLevel::Unfocusable, FocusLevel::Unfocusable) => std::cmp::Ordering::Equal,
(FocusLevel::Unfocusable, FocusLevel::Focusable) => std::cmp::Ordering::Less,
(FocusLevel::Unfocusable, FocusLevel::Ordered(_)) => std::cmp::Ordering::Less,
(FocusLevel::Focusable, FocusLevel::Unfocusable) => std::cmp::Ordering::Greater,
(FocusLevel::Focusable, FocusLevel::Focusable) => std::cmp::Ordering::Equal,
(FocusLevel::Focusable, FocusLevel::Ordered(_)) => std::cmp::Ordering::Greater,
(FocusLevel::Ordered(_), FocusLevel::Unfocusable) => std::cmp::Ordering::Greater,
(FocusLevel::Ordered(_), FocusLevel::Focusable) => std::cmp::Ordering::Less,
(FocusLevel::Ordered(a), FocusLevel::Ordered(b)) => a.cmp(b),
}
} }
} }
+4 -4
View File
@@ -89,7 +89,7 @@ impl Button {
} }
} }
fn switch(&mut self, ctx: &mut WidgetContext, node: NodeMut) { fn switch(&mut self, ctx: &WidgetContext, node: NodeMut) {
let data = FormData { let data = FormData {
value: self.value.to_string(), value: self.value.to_string(),
values: HashMap::new(), values: HashMap::new(),
@@ -185,7 +185,7 @@ impl RinkWidget for Button {
event: &crate::Event, event: &crate::Event,
mut node: dioxus_native_core::real_dom::NodeMut, mut node: dioxus_native_core::real_dom::NodeMut,
) { ) {
let mut ctx: WidgetContext = { let ctx: WidgetContext = {
node.real_dom_mut() node.real_dom_mut()
.raw_world_mut() .raw_world_mut()
.borrow::<UniqueView<WidgetContext>>() .borrow::<UniqueView<WidgetContext>>()
@@ -194,7 +194,7 @@ impl RinkWidget for Button {
}; };
match event.name { match event.name {
"click" => self.switch(&mut ctx, node), "click" => self.switch(&ctx, node),
"keydown" => { "keydown" => {
if let crate::EventData::Keyboard(data) = &event.data { if let crate::EventData::Keyboard(data) = &event.data {
if !data.is_auto_repeating() if !data.is_auto_repeating()
@@ -204,7 +204,7 @@ impl RinkWidget for Button {
_ => false, _ => false,
} }
{ {
self.switch(&mut ctx, node); self.switch(&ctx, node);
} }
} }
} }
+1 -1
View File
@@ -219,7 +219,7 @@ impl<'a> RouteTree<'a> {
Some(id) => { Some(id) => {
// If it exists, add the route to the children of the segment // If it exists, add the route to the children of the segment
let new_children = self.construct(vec![route]); let new_children = self.construct(vec![route]);
self.children_mut(id).extend(new_children.into_iter()); self.children_mut(id).extend(new_children);
} }
None => { None => {
// If it doesn't exist, add the route as a new segment // If it doesn't exist, add the route as a new segment
+2 -1
View File
@@ -31,8 +31,9 @@ smallstr = "0.2.0"
futures-channel = { workspace = true } futures-channel = { workspace = true }
serde_json = { version = "1.0" } serde_json = { version = "1.0" }
serde = { version = "1.0" } serde = { version = "1.0" }
serde-wasm-bindgen = "0.4.5" serde-wasm-bindgen = "0.5.0"
async-trait = "0.1.58" async-trait = "0.1.58"
async-channel = "1.8.0"
[dependencies.web-sys] [dependencies.web-sys]
version = "0.3.56" version = "0.3.56"
+41
View File
@@ -0,0 +1,41 @@
export class Dioxus {
constructor(sendCallback, returnCallback) {
this.sendCallback = sendCallback;
this.returnCallback = returnCallback;
this.promiseResolve = null;
this.received = [];
}
// Receive message from Rust
recv() {
return new Promise((resolve, _reject) => {
// If data already exists, resolve immediately
let data = this.received.shift();
if (data) {
resolve(data);
return;
}
// Otherwise set a resolve callback
this.promiseResolve = resolve;
});
}
// Send message to rust.
send(data) {
this.sendCallback(data);
}
// Internal rust send
rustSend(data) {
// If a promise is waiting for data, resolve it, and clear the resolve callback
if (this.promiseResolve) {
this.promiseResolve(data);
this.promiseResolve = null;
return;
}
// Otherwise add the data to a queue
this.received.push(data);
}
}
+130
View File
@@ -0,0 +1,130 @@
use async_trait::async_trait;
use dioxus_core::ScopeState;
use dioxus_html::prelude::{EvalError, EvalProvider, Evaluator};
use js_sys::Function;
use serde_json::Value;
use std::{cell::RefCell, rc::Rc, str::FromStr};
use wasm_bindgen::prelude::*;
/// Provides the WebEvalProvider through [`cx.provide_context`].
pub fn init_eval(cx: &ScopeState) {
let provider: Rc<dyn EvalProvider> = Rc::new(WebEvalProvider {});
cx.provide_context(provider);
}
/// Reprents the web-target's provider of evaluators.
pub struct WebEvalProvider;
impl EvalProvider for WebEvalProvider {
fn new_evaluator(&self, js: String) -> Result<Rc<dyn Evaluator>, EvalError> {
WebEvaluator::new(js).map(|eval| Rc::new(eval) as Rc<dyn Evaluator + 'static>)
}
}
/// Required to avoid blocking the Rust WASM thread.
const PROMISE_WRAPPER: &str = r#"
return new Promise(async (resolve, _reject) => {
{JS_CODE}
resolve(null);
});
"#;
/// Reprents a web-target's JavaScript evaluator.
pub struct WebEvaluator {
dioxus: Dioxus,
channel_receiver: async_channel::Receiver<serde_json::Value>,
result: RefCell<Option<serde_json::Value>>,
}
impl WebEvaluator {
/// Creates a new evaluator for web-based targets.
pub fn new(js: String) -> Result<Self, EvalError> {
let (channel_sender, channel_receiver) = async_channel::unbounded();
// This Rc cloning mess hurts but it seems to work..
let recv_value = Closure::<dyn FnMut(JsValue)>::new(move |data| {
match serde_wasm_bindgen::from_value::<serde_json::Value>(data) {
Ok(data) => _ = channel_sender.send_blocking(data),
Err(e) => {
// Can't really do much here.
log::error!("failed to serialize JsValue to serde_json::Value (eval communication) - {}", e);
}
}
});
let dioxus = Dioxus::new(recv_value.as_ref().unchecked_ref());
recv_value.forget();
// Wrap the evaluated JS in a promise so that wasm can continue running (send/receive data from js)
let code = PROMISE_WRAPPER.replace("{JS_CODE}", &js);
let result = match Function::new_with_args("dioxus", &code).call1(&JsValue::NULL, &dioxus) {
Ok(result) => {
if let Ok(stringified) = js_sys::JSON::stringify(&result) {
if !stringified.is_undefined() && stringified.is_valid_utf16() {
let string: String = stringified.into();
Value::from_str(&string).map_err(|e| {
EvalError::Communication(format!("Failed to parse result - {}", e))
})?
} else {
return Err(EvalError::Communication(
"Failed to stringify result".into(),
));
}
} else {
return Err(EvalError::Communication(
"Failed to stringify result".into(),
));
}
}
Err(err) => {
return Err(EvalError::InvalidJs(
err.as_string().unwrap_or("unknown".to_string()),
));
}
};
Ok(Self {
dioxus,
channel_receiver,
result: RefCell::new(Some(result)),
})
}
}
#[async_trait(?Send)]
impl Evaluator for WebEvaluator {
/// Runs the evaluated JavaScript.
async fn join(&self) -> Result<serde_json::Value, EvalError> {
self.result.take().ok_or(EvalError::Finished)
}
/// Sends a message to the evaluated JavaScript.
fn send(&self, data: serde_json::Value) -> Result<(), EvalError> {
let data = match serde_wasm_bindgen::to_value::<serde_json::Value>(&data) {
Ok(d) => d,
Err(e) => return Err(EvalError::Communication(e.to_string())),
};
self.dioxus.rustSend(data);
Ok(())
}
/// Gets an UnboundedReceiver to receive messages from the evaluated JavaScript.
async fn recv(&self) -> Result<serde_json::Value, EvalError> {
self.channel_receiver
.recv()
.await
.map_err(|_| EvalError::Communication("failed to receive data from js".to_string()))
}
}
#[wasm_bindgen(module = "/src/eval.js")]
extern "C" {
pub type Dioxus;
#[wasm_bindgen(constructor)]
pub fn new(recv_callback: &Function) -> Dioxus;
#[wasm_bindgen(method)]
pub fn rustSend(this: &Dioxus, data: JsValue);
}
+5 -2
View File
@@ -54,18 +54,17 @@
// - Do DOM work in the next requestAnimationFrame callback // - Do DOM work in the next requestAnimationFrame callback
pub use crate::cfg::Config; pub use crate::cfg::Config;
pub use crate::util::{use_eval, EvalResult};
use dioxus_core::{Element, Scope, VirtualDom}; use dioxus_core::{Element, Scope, VirtualDom};
use futures_util::{pin_mut, FutureExt, StreamExt}; use futures_util::{pin_mut, FutureExt, StreamExt};
mod cache; mod cache;
mod cfg; mod cfg;
mod dom; mod dom;
mod eval;
mod file_engine; mod file_engine;
mod hot_reload; mod hot_reload;
#[cfg(feature = "hydrate")] #[cfg(feature = "hydrate")]
mod rehydrate; mod rehydrate;
mod util;
// Currently disabled since it actually slows down immediate rendering // Currently disabled since it actually slows down immediate rendering
// todo: only schedule non-immediate renders through ric/raf // todo: only schedule non-immediate renders through ric/raf
@@ -168,6 +167,10 @@ pub async fn run_with_props<T: 'static>(root: fn(Scope<T>) -> Element, root_prop
let mut dom = VirtualDom::new_with_props(root, root_props); let mut dom = VirtualDom::new_with_props(root, root_props);
// Eval
let cx = dom.base_scope();
eval::init_eval(cx);
#[cfg(feature = "panic_hook")] #[cfg(feature = "panic_hook")]
if cfg.default_panic_hook { if cfg.default_panic_hook {
console_error_panic_hook::set_once(); console_error_panic_hook::set_once();
-70
View File
@@ -1,70 +0,0 @@
//! Utilities specific to websys
use std::{
future::{IntoFuture, Ready},
rc::Rc,
str::FromStr,
};
use dioxus_core::*;
use serde::de::Error;
use serde_json::Value;
/// Get a closure that executes any JavaScript in the webpage.
///
/// # Safety
///
/// Please be very careful with this function. A script with too many dynamic
/// parts is practically asking for a hacker to find an XSS vulnerability in
/// it. **This applies especially to web targets, where the JavaScript context
/// has access to most, if not all of your application data.**
///
/// # Panics
///
/// The closure will panic if the provided script is not valid JavaScript code
/// or if it returns an uncaught error.
pub fn use_eval(cx: &ScopeState) -> &Rc<dyn Fn(String) -> EvalResult> {
cx.use_hook(|| {
Rc::new(|script: String| EvalResult {
value: if let Ok(value) =
js_sys::Function::new_no_args(&script).call0(&wasm_bindgen::JsValue::NULL)
{
if let Ok(stringified) = js_sys::JSON::stringify(&value) {
if !stringified.is_undefined() && stringified.is_valid_utf16() {
let string: String = stringified.into();
Value::from_str(&string)
} else {
Err(serde_json::Error::custom("Failed to stringify result"))
}
} else {
Err(serde_json::Error::custom("Failed to stringify result"))
}
} else {
Err(serde_json::Error::custom("Failed to execute script"))
},
}) as Rc<dyn Fn(String) -> EvalResult>
})
}
/// A wrapper around the result of a JavaScript evaluation.
/// This implements IntoFuture to be compatible with the desktop renderer's EvalResult.
pub struct EvalResult {
value: Result<Value, serde_json::Error>,
}
impl EvalResult {
/// Get the result of the Javascript execution.
pub fn get(self) -> Result<Value, serde_json::Error> {
self.value
}
}
impl IntoFuture for EvalResult {
type Output = Result<Value, serde_json::Error>;
type IntoFuture = Ready<Result<Value, serde_json::Error>>;
fn into_future(self) -> Self::IntoFuture {
std::future::ready(self.value)
}
}
+1
View File
@@ -9,4 +9,5 @@ publish = false
[dependencies] [dependencies]
dioxus = { path = "../../packages/dioxus" } dioxus = { path = "../../packages/dioxus" }
dioxus-web = { path = "../../packages/web" } dioxus-web = { path = "../../packages/web" }
dioxus-html = { path = "../../packages/html" }
serde_json = "1.0.96" serde_json = "1.0.96"
+16 -9
View File
@@ -1,13 +1,13 @@
// This test is used by playwright configured in the root of the repo // This test is used by playwright configured in the root of the repo
use dioxus::prelude::*; use dioxus::prelude::*;
use dioxus_web::use_eval;
fn app(cx: Scope) -> Element { fn app(cx: Scope) -> Element {
let mut num = use_state(cx, || 0); let mut num = use_state(cx, || 0);
let eval = use_eval(cx);
let eval_result = use_state(cx, String::new); let eval_result = use_state(cx, String::new);
let eval_provider = dioxus_html::prelude::use_eval(cx);
cx.render(rsx! { cx.render(rsx! {
div { div {
"hello axum! {num}" "hello axum! {num}"
@@ -42,13 +42,20 @@ fn app(cx: Scope) -> Element {
button { button {
class: "eval-button", class: "eval-button",
onclick: move |_| { onclick: move |_| {
// Set the window title let eval = eval_provider(
let result = eval(r#"window.document.title = 'Hello from Dioxus Eval!'; r#"
return "returned eval value";"#.to_string()); window.document.title = 'Hello from Dioxus Eval!';
if let Ok(serde_json::Value::String(string)) = result.get() { dioxus.send("returned eval value");
eval_result.set(string); "#).unwrap();
} let setter = eval_result.setter();
}, async move {
// Set the window title
let result = eval.recv().await;
if let Ok(serde_json::Value::String(string)) = result {
setter(string);
}
}},
"Eval" "Eval"
} }
div { div {