minor progress
This commit is contained in:
parent
66aa62de98
commit
c5880964e7
7 changed files with 90 additions and 45 deletions
|
@ -11,9 +11,8 @@ use vcs_common::ApplicationMessage;
|
||||||
use crate::config::AppConfig;
|
use crate::config::AppConfig;
|
||||||
use crate::APP_HANDLE;
|
use crate::APP_HANDLE;
|
||||||
|
|
||||||
|
|
||||||
mod satellite_connection;
|
|
||||||
mod joystick_source;
|
mod joystick_source;
|
||||||
|
mod satellite_connection;
|
||||||
|
|
||||||
use joystick_source::joystick_loop;
|
use joystick_source::joystick_loop;
|
||||||
use satellite_connection::SatelliteConnection;
|
use satellite_connection::SatelliteConnection;
|
||||||
|
@ -21,6 +20,7 @@ use satellite_connection::SatelliteConnection;
|
||||||
pub enum ApplicationEvent {
|
pub enum ApplicationEvent {
|
||||||
WebRTCMessage((String, vcs_common::ApplicationMessage)),
|
WebRTCMessage((String, vcs_common::ApplicationMessage)),
|
||||||
JoystickMove(Point),
|
JoystickMove(Point),
|
||||||
|
RetryDisconnectedSatellites,
|
||||||
Close,
|
Close,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -82,7 +82,7 @@ impl AppState {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn check_alive_things(&mut self) {
|
pub async fn check_alive_things(&mut self) {
|
||||||
if !self
|
if !self
|
||||||
.joystick_task_is_alive
|
.joystick_task_is_alive
|
||||||
.load(std::sync::atomic::Ordering::SeqCst)
|
.load(std::sync::atomic::Ordering::SeqCst)
|
||||||
|
@ -92,6 +92,14 @@ impl AppState {
|
||||||
Arc::clone(&self.joystick_task_is_alive),
|
Arc::clone(&self.joystick_task_is_alive),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for i in self
|
||||||
|
.camera_satellites
|
||||||
|
.iter_mut()
|
||||||
|
.filter(|x| !x.is_connected() && x.try_connecting)
|
||||||
|
{
|
||||||
|
i.connect(self.runtime.clone()).await
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -104,15 +112,26 @@ pub async fn run_main_event_loop(
|
||||||
) {
|
) {
|
||||||
let mut state = AppState::new(mec, to_mec, config, rt).await;
|
let mut state = AppState::new(mec, to_mec, config, rt).await;
|
||||||
loop {
|
loop {
|
||||||
state.check_alive_things();
|
state.check_alive_things().await;
|
||||||
|
|
||||||
match state.mec.try_recv() {
|
match state.mec.try_recv() {
|
||||||
Err(TryRecvError::Empty) => tokio::time::sleep(Duration::from_millis(50)).await,
|
Err(TryRecvError::Empty) => tokio::time::sleep(Duration::from_millis(50)).await,
|
||||||
Err(TryRecvError::Closed) => break,
|
Err(TryRecvError::Closed) => break,
|
||||||
Ok(msg) => match msg {
|
Ok(msg) => {
|
||||||
|
match msg {
|
||||||
ApplicationEvent::Close => {
|
ApplicationEvent::Close => {
|
||||||
state.mec.close(); // cleanup is handled on reading from a closed mec
|
state.mec.close(); // cleanup is handled on reading from a closed mec
|
||||||
}
|
}
|
||||||
|
ApplicationEvent::RetryDisconnectedSatellites => {
|
||||||
|
state
|
||||||
|
.camera_satellites
|
||||||
|
.iter_mut()
|
||||||
|
.filter(|x| !x.is_connected() && !x.try_connecting)
|
||||||
|
.for_each(|x| {
|
||||||
|
x.try_connecting = true;
|
||||||
|
x.retry_attempts = 0;
|
||||||
|
});
|
||||||
|
}
|
||||||
ApplicationEvent::WebRTCMessage((name, msg)) => {
|
ApplicationEvent::WebRTCMessage((name, msg)) => {
|
||||||
for conn in state
|
for conn in state
|
||||||
.camera_satellites
|
.camera_satellites
|
||||||
|
@ -142,21 +161,25 @@ pub async fn run_main_event_loop(
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
for connection in state.camera_satellites.iter_mut() {
|
for connection in state.camera_satellites.iter_mut() {
|
||||||
match connection.try_next().await {
|
match connection.try_next().await {
|
||||||
Some(msg) => match msg {
|
Some(msg) => match msg {
|
||||||
ApplicationMessage::NameRequest(None) => {
|
ApplicationMessage::NameRequest(None) => {
|
||||||
if let Err(e) = connection.send(ApplicationMessage::NameRequest(Some("Controller".to_owned()))).await {
|
if let Err(e) = connection
|
||||||
|
.send(ApplicationMessage::NameRequest(Some(
|
||||||
|
"Controller".to_owned(),
|
||||||
|
)))
|
||||||
|
.await
|
||||||
|
{
|
||||||
info!("Was not able to send name to remote? {:?}", e);
|
info!("Was not able to send name to remote? {:?}", e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
ApplicationMessage::NameRequest(Some(name)) => connection.name = name,
|
ApplicationMessage::NameRequest(Some(name)) => connection.name = name,
|
||||||
ApplicationMessage::ChangeTrackingID(_) => {}
|
ApplicationMessage::ChangeTrackingID(_) => {}
|
||||||
ApplicationMessage::ManualMovementOverride(_) => {}
|
ApplicationMessage::ManualMovementOverride(_) => {}
|
||||||
ApplicationMessage::TrackingBoxes(_update) => {
|
ApplicationMessage::TrackingBoxes(_update) => {}
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
ApplicationMessage::WebRTCIceCandidateInit(pkt) => {
|
ApplicationMessage::WebRTCIceCandidateInit(pkt) => {
|
||||||
send_frontend_message(serde_json::to_string(&pkt).unwrap())
|
send_frontend_message(serde_json::to_string(&pkt).unwrap())
|
||||||
|
@ -167,7 +190,7 @@ pub async fn run_main_event_loop(
|
||||||
ApplicationMessage::WebRTCPacket(pkt) => {
|
ApplicationMessage::WebRTCPacket(pkt) => {
|
||||||
send_frontend_message(serde_json::to_string(&pkt).unwrap())
|
send_frontend_message(serde_json::to_string(&pkt).unwrap())
|
||||||
}
|
}
|
||||||
}
|
},
|
||||||
None => {}
|
None => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -191,9 +214,9 @@ pub fn send_frontend_message(message: String) {
|
||||||
return;
|
return;
|
||||||
} else {
|
} else {
|
||||||
let handle = e.take().unwrap();
|
let handle = e.take().unwrap();
|
||||||
handle.emit_all("frontend_message", message)
|
handle
|
||||||
|
.emit_all("frontend_message", message)
|
||||||
.expect("Could not send message to the tauri frontend!");
|
.expect("Could not send message to the tauri frontend!");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -15,6 +15,8 @@ pub enum SatelliteConnectionError {
|
||||||
|
|
||||||
pub struct SatelliteConnection {
|
pub struct SatelliteConnection {
|
||||||
pub name: String,
|
pub name: String,
|
||||||
|
pub retry_attempts: usize,
|
||||||
|
pub try_connecting: bool,
|
||||||
connection: ConnectionString,
|
connection: ConnectionString,
|
||||||
|
|
||||||
to_socket: Option<AppSender>,
|
to_socket: Option<AppSender>,
|
||||||
|
@ -29,6 +31,9 @@ impl SatelliteConnection {
|
||||||
name: String::new(),
|
name: String::new(),
|
||||||
connection: conn_string.clone(),
|
connection: conn_string.clone(),
|
||||||
|
|
||||||
|
retry_attempts: 0,
|
||||||
|
try_connecting: true,
|
||||||
|
|
||||||
to_socket: None,
|
to_socket: None,
|
||||||
from_socket: None,
|
from_socket: None,
|
||||||
socket_is_alive: Arc::new(AtomicBool::new(false)),
|
socket_is_alive: Arc::new(AtomicBool::new(false)),
|
||||||
|
@ -66,9 +71,14 @@ impl SatelliteConnection {
|
||||||
self.socket_is_alive = is_alive;
|
self.socket_is_alive = is_alive;
|
||||||
}
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
|
self.retry_attempts += 1;
|
||||||
|
if self.retry_attempts > 5 {
|
||||||
|
self.try_connecting = false;
|
||||||
|
}
|
||||||
error!(
|
error!(
|
||||||
"Could not connect to socket remote: '{}' \n {}",
|
"Could not connect to socket remote: '{}' \nRetry Attempt: {} \n {}",
|
||||||
self.connection.build_conn_string(),
|
self.connection.build_conn_string(),
|
||||||
|
self.retry_attempts,
|
||||||
e
|
e
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
|
@ -57,6 +57,7 @@ fn main() {
|
||||||
|
|
||||||
let rt = runtime::Runtime::new().expect("Could not start tokio runtime");
|
let rt = runtime::Runtime::new().expect("Could not start tokio runtime");
|
||||||
let handle = rt.handle().clone();
|
let handle = rt.handle().clone();
|
||||||
|
let handle2 = handle.clone();
|
||||||
|
|
||||||
let _coordinator = rt.handle().spawn(run_main_event_loop(
|
let _coordinator = rt.handle().spawn(run_main_event_loop(
|
||||||
mec,
|
mec,
|
||||||
|
@ -71,6 +72,7 @@ fn main() {
|
||||||
tauri::Builder::default()
|
tauri::Builder::default()
|
||||||
.manage(tauri_functions::TauriState {
|
.manage(tauri_functions::TauriState {
|
||||||
to_mec: to_mec.clone(),
|
to_mec: to_mec.clone(),
|
||||||
|
rt: handle2.clone(),
|
||||||
})
|
})
|
||||||
.invoke_handler(tauri::generate_handler![tauri_functions::connect_to_camera])
|
.invoke_handler(tauri::generate_handler![tauri_functions::connect_to_camera])
|
||||||
.setup(|app| {
|
.setup(|app| {
|
||||||
|
@ -102,7 +104,12 @@ fn main() {
|
||||||
};
|
};
|
||||||
|
|
||||||
if message.is_some() {
|
if message.is_some() {
|
||||||
if let Err(e) = to_mec.send_blocking(ApplicationEvent::WebRTCMessage(("first".to_owned(), message.unwrap()))) {
|
if let Err(e) =
|
||||||
|
to_mec.send_blocking(ApplicationEvent::WebRTCMessage((
|
||||||
|
"first".to_owned(),
|
||||||
|
message.unwrap(),
|
||||||
|
)))
|
||||||
|
{
|
||||||
error!("Could not send to mec! {e}");
|
error!("Could not send to mec! {e}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,16 +1,20 @@
|
||||||
use async_channel::Sender;
|
use async_channel::Sender;
|
||||||
use tauri::State;
|
use tauri::State;
|
||||||
|
use tokio::runtime::Handle;
|
||||||
use tracing::info;
|
use tracing::info;
|
||||||
|
|
||||||
use crate::coordinator::ApplicationEvent;
|
use crate::coordinator::ApplicationEvent;
|
||||||
|
|
||||||
pub struct TauriState {
|
pub struct TauriState {
|
||||||
pub to_mec: Sender<ApplicationEvent>,
|
pub to_mec: Sender<ApplicationEvent>,
|
||||||
|
pub rt: Handle,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
pub fn connect_to_camera(state: State<'_, TauriState>) {
|
pub fn connect_to_camera(state: State<'_, TauriState>) {
|
||||||
let _ = state
|
let mec = state.to_mec.clone();
|
||||||
.to_mec
|
state.rt.spawn(async move{
|
||||||
.send_blocking(ApplicationEvent::CameraConnectionPress);
|
let _ = mec
|
||||||
|
.send_blocking(ApplicationEvent::RetryDisconnectedSatellites);
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
|
@ -30,7 +30,7 @@
|
||||||
<div class="flex flex-row h-dvh w-100 text-neutral-400">
|
<div class="flex flex-row h-dvh w-100 text-neutral-400">
|
||||||
|
|
||||||
<div class="flex flex-col bg-neutral-300 h-dvh w-1/8">
|
<div class="flex flex-col bg-neutral-300 h-dvh w-1/8">
|
||||||
<button id="camera_connect_button" onclick="call_camera_connect()" class="rounded-full font-semibold mx-3 mt-2 px-4 py-2 text-white bg-cyan-600">Connect to Camera</button>
|
<button id="camera_connect_button" class="rounded-full font-semibold mx-3 mt-2 px-4 py-2 text-white bg-cyan-600">Connect to Camera</button>
|
||||||
|
|
||||||
<button class="rounded-full font-semibold mx-3 mt-2 px-4 py-2 text-white bg-cyan-600">Connect to Computer</button>
|
<button class="rounded-full font-semibold mx-3 mt-2 px-4 py-2 text-white bg-cyan-600">Connect to Computer</button>
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
const { invoke } = window.__TAURI__.tauri;
|
import { invoke } from "./@tauri-apps/api/index.js";
|
||||||
import { rtc_init } from "./rtc.js";
|
import { rtc_init } from "./rtc.js";
|
||||||
|
|
||||||
function call_camera_connect() {
|
function call_camera_connect() {
|
||||||
|
@ -15,6 +15,7 @@ function call_camera_connect() {
|
||||||
|
|
||||||
async function init() {
|
async function init() {
|
||||||
console.log("Setting up");
|
console.log("Setting up");
|
||||||
|
document.getElementById("camera_connect_button").addEventListener("click", call_camera_connect);
|
||||||
await rtc_init();
|
await rtc_init();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue