wip user confirm

TODO: timeout for the user confirmation.
BitBoxSwiss · Jun 30, 2024 · f11c4a8 · f11c4a8
1 parent 13881ba
commit f11c4a8
Show file tree

Hide file tree

Showing 5 changed files with 425 additions and 32 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/bitbox-bridge/Cargo.toml b/bitbox-bridge/Cargo.toml
@@ -7,6 +7,7 @@ edition = "2021"
 license = "Apache-2.0"
 
 [dependencies]
+webbrowser = "1.0"
 env_logger = "0.11"
 futures =  { workspace = true }
 futures-util = { workspace = true }

diff --git a/bitbox-bridge/resources/confirmation_dialog.html b/bitbox-bridge/resources/confirmation_dialog.html
@@ -0,0 +1,20 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Confirmation</title>
+</head>
+<body>
+    <h1>Confirm Action</h1>
+    <p>{{ message }}</p>
+    <button onclick="sendResponse(true)">OK</button>
+    <button onclick="sendResponse(false)">Cancel</button>
+    <script>
+        function sendResponse(userChoice) {
+            fetch(`/confirm/response/{{ counter }}/${userChoice}`, { method: 'POST' })
+                .then(() => window.close())  // Optionally close the window/tab
+                .catch(err => console.error('Error sending response:', err));
+        }
+    </script>
+</body>
+</html>
diff --git a/bitbox-bridge/src/main.rs b/bitbox-bridge/src/main.rs
@@ -13,7 +13,9 @@
 // limitations under the License.
 
 use futures::channel::mpsc;
+use std::collections::HashSet;
 use std::net::SocketAddr;
+use std::sync::{Arc, Mutex};
 use tokio::runtime::Runtime;
 
 #[macro_use]
@@ -90,9 +92,26 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let addr = SocketAddr::new("127.0.0.1".parse()?, port);
 
     println!("listening on http://{}", addr);
-    let server = web::create(usb_devices, notify_tx, addr);
+    let state = web::ConfirmState::new();
+    let remembered_allowed_origins = Arc::new(Mutex::new(HashSet::new()));
+    let server = web::create(
+        usb_devices,
+        notify_tx,
+        state.clone(),
+        remembered_allowed_origins.clone(),
+        addr,
+    );
 
     rt.block_on(async move {
+        let s = state.clone();
+        let r = remembered_allowed_origins.clone();
+        tokio::spawn(async move {
+            let foo = web::user_confirm_origin(s, r, "ORIGIN".into(), &format!("http://{}", addr))
+                .await
+                .unwrap();
+            println!("RESULT 1 {}", foo);
+        });
+
         tokio::select! {
             _ = stop_request => info!("Requested to stop by environment"),
             _ = server => info!("Warp returned"),