1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
//! Underlying layer abstraction for testability
//!
//! This module provides an abstraction over the underlying transport layer
//! (WebSocket, TCP, TLS, etc.) for testability and flexibility.
use async_trait::async_trait;
use futures::channel::{mpsc, oneshot};
/// Type alias for the connect reply sender
pub type ConnectReplySender =
std::sync::Arc<std::sync::Mutex<Option<oneshot::Sender<Result<(), crate::error::Error>>>>>;
/// Underlying layer events (sent FROM transport TO message loop)
#[derive(Debug, Clone)]
pub enum UnderlyingLayerEvent {
Connected,
Message(Vec<u8>),
Error(String),
Closed,
/// Timer expired event
/// The String is the timer kind (e.g., "PingreqSend")
TimerExpired(String),
}
/// Underlying layer commands (sent TO transport FROM message loop)
#[derive(Debug, Clone)]
pub enum UnderlyingLayerCommand {
Connect(String, ConnectReplySender),
SendData(Vec<u8>),
Close,
/// Start or reset a timer
/// When the timer expires, the underlying layer should send TimerExpired event
TimerReset {
kind: String,
duration_ms: u64,
},
/// Cancel a timer
TimerCancel {
kind: String,
},
}
/// Abstract underlying layer interface for testing (pure message-passing)
#[async_trait(?Send)]
#[cfg(target_arch = "wasm32")]
pub trait UnderlyingLayerInterface {
/// Get event receiver (events FROM transport TO message loop)
fn event_receiver(&mut self) -> mpsc::UnboundedReceiver<UnderlyingLayerEvent>;
/// Get command sender (commands TO transport FROM message loop)
fn command_sender(&self) -> mpsc::UnboundedSender<UnderlyingLayerCommand>;
/// Start the transport processor (handles commands and generates events)
async fn run(&mut self);
}
/// Abstract underlying layer interface for testing (pure message-passing)
#[async_trait(?Send)]
#[cfg(not(target_arch = "wasm32"))]
pub trait UnderlyingLayerInterface: Send {
/// Get event receiver (events FROM transport TO message loop)
fn event_receiver(&mut self) -> mpsc::UnboundedReceiver<UnderlyingLayerEvent>;
/// Get command sender (commands TO transport FROM message loop)
fn command_sender(&self) -> mpsc::UnboundedSender<UnderlyingLayerCommand>;
/// Start the transport processor (handles commands and generates events)
async fn run(&mut self);
}
/// Browser WebSocket implementation (pure message-passing)
#[cfg(target_arch = "wasm32")]
pub struct BrowserWebSocket {
event_sender: mpsc::UnboundedSender<UnderlyingLayerEvent>,
event_receiver: Option<mpsc::UnboundedReceiver<UnderlyingLayerEvent>>,
command_sender: mpsc::UnboundedSender<UnderlyingLayerCommand>,
command_receiver: mpsc::UnboundedReceiver<UnderlyingLayerCommand>,
/// Active timers: kind -> timer_id
active_timers: std::collections::HashMap<String, i32>,
}
#[cfg(target_arch = "wasm32")]
impl BrowserWebSocket {
pub fn new() -> Self {
let (event_sender, event_receiver) = mpsc::unbounded();
let (command_sender, command_receiver) = mpsc::unbounded();
Self {
event_sender,
event_receiver: Some(event_receiver),
command_sender,
command_receiver,
active_timers: std::collections::HashMap::new(),
}
}
}
#[cfg(target_arch = "wasm32")]
#[async_trait(?Send)]
impl UnderlyingLayerInterface for BrowserWebSocket {
fn event_receiver(&mut self) -> mpsc::UnboundedReceiver<UnderlyingLayerEvent> {
self.event_receiver.take().unwrap_or_else(|| {
let (_, receiver) = mpsc::unbounded();
receiver
})
}
fn command_sender(&self) -> mpsc::UnboundedSender<UnderlyingLayerCommand> {
self.command_sender.clone()
}
async fn run(&mut self) {
use futures::stream::StreamExt;
use wasm_bindgen::prelude::*;
use wasm_bindgen::JsCast;
use web_sys::{BinaryType, ErrorEvent, MessageEvent, WebSocket};
web_sys::console::log_1(&"🚀 WEBSOCKET: NEW VERSION - CONNECT TIMING FIXED 🚀".into());
let mut websocket: Option<web_sys::WebSocket> = None;
let mut _closures: Vec<wasm_bindgen::closure::Closure<dyn FnMut(wasm_bindgen::JsValue)>> =
Vec::new();
let _is_connected = false;
let _pending_data: Vec<Vec<u8>> = Vec::new();
web_sys::console::log_1(&"WebSocket processor waiting for commands".into());
while let Some(command) = self.command_receiver.next().await {
web_sys::console::log_1(
&format!("WebSocket processor received command: {:?}", command).into(),
);
match command {
UnderlyingLayerCommand::Connect(url, reply_arc) => {
web_sys::console::log_1(&format!("WebSocket connecting to: {}", url).into());
web_sys::console::log_1(&"✅ Received Connect command with reply_arc".into());
// MQTT subprotocol is required
let protocols = js_sys::Array::new();
protocols.push(&wasm_bindgen::JsValue::from_str("mqtt"));
let ws_result = WebSocket::new_with_str_sequence(&url, &protocols);
match ws_result {
Ok(ws) => {
web_sys::console::log_1(&"WebSocket created successfully".into());
ws.set_binary_type(BinaryType::Arraybuffer);
web_sys::console::log_1(&"Binary type set to ArrayBuffer".into());
let event_sender = self.event_sender.clone();
web_sys::console::log_1(&"Event sender cloned for closures".into());
// onopen
let event_sender_clone = event_sender.clone();
let reply_arc_clone = reply_arc.clone();
web_sys::console::log_1(&"Creating onopen closure".into());
let onopen = Closure::wrap(Box::new(move |_: JsValue| {
web_sys::console::log_1(
&"🔥 NEW WEBSOCKET: WebSocket onopen fired 🔥".into(),
);
// Send reply to complete the connect() await
web_sys::console::log_1(
&"Attempting to lock reply_arc in onopen".into(),
);
match reply_arc_clone.lock() {
Ok(mut reply_opt) => {
web_sys::console::log_1(
&"Successfully locked reply_arc".into(),
);
if let Some(reply) = reply_opt.take() {
match reply.send(Ok(())) {
Ok(_) => web_sys::console::log_1(&"✅ Sent connect completion reply successfully".into()),
Err(_) => web_sys::console::log_1(&"❌ Failed to send connect completion reply - receiver dropped".into()),
}
} else {
web_sys::console::log_1(
&"❌ No reply sender in Option".into(),
);
}
}
Err(_) => {
web_sys::console::log_1(
&"❌ Failed to lock reply_arc".into(),
);
}
}
match event_sender_clone
.unbounded_send(UnderlyingLayerEvent::Connected)
{
Ok(_) => web_sys::console::log_1(
&"Sent Connected event successfully".into(),
),
Err(e) => web_sys::console::log_1(
&format!("Failed to send Connected event: {:?}", e).into(),
),
}
})
as Box<dyn FnMut(JsValue)>);
web_sys::console::log_1(
&"onopen closure created, setting on WebSocket".into(),
);
ws.set_onopen(Some(onopen.as_ref().unchecked_ref()));
web_sys::console::log_1(
&"onopen set on WebSocket, pushing to closures vec".into(),
);
_closures.push(onopen);
web_sys::console::log_1(&"onopen closure pushed to vec".into());
// onmessage
let event_sender_clone = event_sender.clone();
web_sys::console::log_1(&"Creating onmessage closure".into());
let onmessage = Closure::wrap(Box::new(move |e: JsValue| {
web_sys::console::log_1(&"WebSocket onmessage fired".into());
let event: MessageEvent = e.dyn_into().unwrap();
if let Ok(array_buffer) =
event.data().dyn_into::<js_sys::ArrayBuffer>()
{
let uint8_array = js_sys::Uint8Array::new(&array_buffer);
let mut data = vec![0; uint8_array.length() as usize];
uint8_array.copy_to(&mut data);
web_sys::console::log_1(
&format!("Received {} bytes", data.len()).into(),
);
match event_sender_clone
.unbounded_send(UnderlyingLayerEvent::Message(data))
{
Ok(_) => web_sys::console::log_1(
&"Sent Message event successfully".into(),
),
Err(e) => web_sys::console::log_1(
&format!("Failed to send Message event: {:?}", e)
.into(),
),
}
}
})
as Box<dyn FnMut(JsValue)>);
web_sys::console::log_1(
&"onmessage closure created, setting on WebSocket".into(),
);
ws.set_onmessage(Some(onmessage.as_ref().unchecked_ref()));
web_sys::console::log_1(
&"onmessage set on WebSocket, pushing to closures vec".into(),
);
_closures.push(onmessage);
web_sys::console::log_1(&"onmessage closure pushed to vec".into());
// onerror - with detailed logging
let event_sender_clone = event_sender.clone();
let onerror = Closure::wrap(Box::new(move |e: JsValue| {
web_sys::console::log_1(&"WebSocket onerror fired".into());
web_sys::console::log_1(&format!("Error event: {:?}", e).into());
let error_msg = if let Ok(error_event) = e.dyn_into::<ErrorEvent>()
{
let msg = error_event.message();
web_sys::console::log_1(
&format!("ErrorEvent message: {}", msg).into(),
);
msg
} else {
web_sys::console::log_1(
&"Not an ErrorEvent - unknown error".into(),
);
"Unknown WebSocket error".to_string()
};
let _ = event_sender_clone
.unbounded_send(UnderlyingLayerEvent::Error(error_msg));
})
as Box<dyn FnMut(JsValue)>);
ws.set_onerror(Some(onerror.as_ref().unchecked_ref()));
_closures.push(onerror);
// onclose
let event_sender_clone = event_sender.clone();
let onclose = Closure::wrap(Box::new(move |e: JsValue| {
web_sys::console::log_1(&"WebSocket onclose fired".into());
web_sys::console::log_1(&format!("Close event: {:?}", e).into());
// Try to get close details
if let Ok(close_event) = e.dyn_into::<web_sys::CloseEvent>() {
let code = close_event.code();
let reason = close_event.reason();
let was_clean = close_event.was_clean();
web_sys::console::log_1(
&format!(
"Close code: {}, reason: '{}', clean: {}",
code, reason, was_clean
)
.into(),
);
}
let _ =
event_sender_clone.unbounded_send(UnderlyingLayerEvent::Closed);
})
as Box<dyn FnMut(JsValue)>);
ws.set_onclose(Some(onclose.as_ref().unchecked_ref()));
_closures.push(onclose);
websocket = Some(ws);
}
Err(e) => {
let _ = self
.event_sender
.unbounded_send(UnderlyingLayerEvent::Error(format!(
"Failed to create WebSocket: {:?}",
e
)));
}
}
}
UnderlyingLayerCommand::SendData(data) => {
web_sys::console::log_1(
&format!("WebSocket SendData command: {} bytes", data.len()).into(),
);
if let Some(ref ws) = websocket {
web_sys::console::log_1(
&"WebSocket is available, attempting to send".into(),
);
match ws.send_with_u8_array(&data) {
Ok(_) => {
web_sys::console::log_1(
&"WebSocket send_with_u8_array succeeded".into(),
);
}
Err(e) => {
web_sys::console::log_1(
&format!("WebSocket send_with_u8_array failed: {:?}", e).into(),
);
let _ =
self.event_sender
.unbounded_send(UnderlyingLayerEvent::Error(format!(
"Send failed: {:?}",
e
)));
}
}
} else {
web_sys::console::log_1(&"WebSocket not available for sending".into());
let _ = self
.event_sender
.unbounded_send(UnderlyingLayerEvent::Error(
"WebSocket not connected".to_string(),
));
}
}
UnderlyingLayerCommand::Close => {
// Clear closures first to prevent further callbacks
_closures.clear();
if let Some(ws) = websocket.take() {
// Remove event handlers before closing
ws.set_onopen(None);
ws.set_onmessage(None);
ws.set_onerror(None);
ws.set_onclose(None);
let _ = ws.close();
}
let _ = self
.event_sender
.unbounded_send(UnderlyingLayerEvent::Closed);
// Do NOT break - allow reconnection by continuing to process commands
}
UnderlyingLayerCommand::TimerReset { kind, duration_ms } => {
// Cancel existing timer if any
if let Some(old_timer_id) = self.active_timers.remove(&kind) {
web_sys::console::log_1(
&format!("Cancelling existing timer {} (ID: {})", kind, old_timer_id)
.into(),
);
crate::platform::clear_timeout(old_timer_id);
}
// Create new timer
let event_sender = self.event_sender.clone();
let timer_kind = kind.clone();
let callback = wasm_bindgen::closure::Closure::wrap(Box::new(move || {
web_sys::console::log_1(&format!("Timer expired: {}", timer_kind).into());
let _ = event_sender
.unbounded_send(UnderlyingLayerEvent::TimerExpired(timer_kind.clone()));
})
as Box<dyn Fn()>);
let timer_id = crate::platform::set_timeout(&callback, duration_ms as i32);
callback.forget();
self.active_timers.insert(kind.clone(), timer_id);
web_sys::console::log_1(
&format!(
"Timer set: {} (ID: {}) for {}ms",
kind, timer_id, duration_ms
)
.into(),
);
}
UnderlyingLayerCommand::TimerCancel { kind } => {
if let Some(timer_id) = self.active_timers.remove(&kind) {
web_sys::console::log_1(
&format!("Timer cancelled: {} (ID: {})", kind, timer_id).into(),
);
crate::platform::clear_timeout(timer_id);
} else {
web_sys::console::log_1(
&format!("Timer cancel requested but not active: {}", kind).into(),
);
}
}
}
}
}
}