blob: 7b3dfdc9550aca20d698575eb235be7116c0b2cd (
plain)
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
|
use crate::{Error, Result};
use futures::{
channel::{mpsc, mpsc::Receiver},
SinkExt,
};
use mullvad_management_interface::{
types::daemon_event::Event as EventType, ManagementServiceClient,
};
use mullvad_types::states::TunnelState;
// Spawns a new task that listens for tunnel state changes and forwards it through the returned
// channel. Panics if called from outside of the Tokio runtime.
pub fn state_listen(mut rpc: ManagementServiceClient) -> Receiver<Result<TunnelState>> {
let (mut sender, receiver) = mpsc::channel::<Result<TunnelState>>(1);
tokio::spawn(async move {
match rpc.events_listen(()).await {
Ok(events) => {
let mut events = events.into_inner();
loop {
let forward = match events.message().await {
Ok(Some(event)) => match event.event.unwrap() {
EventType::TunnelState(new_state) => {
Ok(TunnelState::try_from(new_state).expect("invalid tunnel state"))
}
_ => continue,
},
Ok(None) => break,
Err(status) => Err(Error::RpcFailed(status)),
};
if sender.send(forward).await.is_err() {
break;
}
}
}
Err(status) => {
let _ = sender.send(Err(Error::RpcFailed(status))).await;
}
}
});
receiver
}
|
