pico-strike / src / net / rpc.rs

use embassy_net::tcp::TcpSocket;
use sachy_fmt::{error, info};

use crate::{
    constants::HOST_PORT,
    updates::{NetDataReceiver, UpdateConnection},
};

pub async fn rpc_loop<'device>(tcp: &mut TcpSocket<'device>, net_data: &NetDataReceiver) {
    loop {
        UpdateConnection::disconnect();

        if tcp.accept(HOST_PORT).await.is_err() {
            continue;
        }

        match super::noise::NoiseSession::initialize(tcp).await {
            Ok(session) => {
                info!("Connected!");
                UpdateConnection::connect();

                session.run(tcp).await;

                net_data.clear();

                info!("DISCONNECT");
            }
            Err(e) => error!("Session init failure: {}", e),
        }

        tcp.abort();
        tcp.flush().await.ok();
    }
}