mirror of
https://github.com/nestriness/nestri.git
synced 2025-12-12 08:45:38 +02:00
fbaa8835a3
Replace json protocol by protobuf generate protobuf files with `bun buf generate` or just `buf generate` - [x] Implement all datatypes with proto files - [x] Map to ts types or use the generated proto types directly with: - [x] web frontend - [x] relay - [x] runner - [ ] final performance test (to be done when CI builds new images) --------- Co-authored-by: DatCaptainHorse <DatCaptainHorse@users.noreply.github.com>
121 lines
3.1 KiB
Rust
121 lines
3.1 KiB
Rust
use crate::latency::LatencyTracker;
|
|
use num_derive::{FromPrimitive, ToPrimitive};
|
|
use num_traits::{FromPrimitive, ToPrimitive};
|
|
use serde::{Deserialize, Serialize};
|
|
use std::error::Error;
|
|
use webrtc::ice_transport::ice_candidate::RTCIceCandidateInit;
|
|
use webrtc::peer_connection::sdp::session_description::RTCSessionDescription;
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageBase {
|
|
pub payload_type: String,
|
|
pub latency: Option<LatencyTracker>,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageLog {
|
|
#[serde(flatten)]
|
|
pub base: MessageBase,
|
|
pub level: String,
|
|
pub message: String,
|
|
pub time: String,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageMetrics {
|
|
#[serde(flatten)]
|
|
pub base: MessageBase,
|
|
pub usage_cpu: f64,
|
|
pub usage_memory: f64,
|
|
pub uptime: u64,
|
|
pub pipeline_latency: f64,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageICE {
|
|
#[serde(flatten)]
|
|
pub base: MessageBase,
|
|
pub candidate: RTCIceCandidateInit,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageSDP {
|
|
#[serde(flatten)]
|
|
pub base: MessageBase,
|
|
pub sdp: RTCSessionDescription,
|
|
}
|
|
|
|
#[repr(i32)]
|
|
#[derive(Debug, FromPrimitive, ToPrimitive, Copy, Clone, Serialize, Deserialize)]
|
|
#[serde(try_from = "i32", into = "i32")]
|
|
pub enum JoinerType {
|
|
JoinerNode = 0,
|
|
JoinerClient = 1,
|
|
}
|
|
impl TryFrom<i32> for JoinerType {
|
|
type Error = &'static str;
|
|
|
|
fn try_from(value: i32) -> Result<Self, Self::Error> {
|
|
JoinerType::from_i32(value).ok_or("Invalid value for JoinerType")
|
|
}
|
|
}
|
|
impl From<JoinerType> for i32 {
|
|
fn from(joiner_type: JoinerType) -> Self {
|
|
joiner_type.to_i32().unwrap()
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageJoin {
|
|
#[serde(flatten)]
|
|
pub base: MessageBase,
|
|
pub joiner_type: JoinerType,
|
|
}
|
|
|
|
#[repr(i32)]
|
|
#[derive(Debug, FromPrimitive, ToPrimitive, Copy, Clone, Serialize, Deserialize)]
|
|
#[serde(try_from = "i32", into = "i32")]
|
|
pub enum AnswerType {
|
|
AnswerOffline = 0,
|
|
AnswerInUse = 1,
|
|
AnswerOK = 2,
|
|
}
|
|
impl TryFrom<i32> for AnswerType {
|
|
type Error = &'static str;
|
|
|
|
fn try_from(value: i32) -> Result<Self, Self::Error> {
|
|
AnswerType::from_i32(value).ok_or("Invalid value for AnswerType")
|
|
}
|
|
}
|
|
impl From<AnswerType> for i32 {
|
|
fn from(answer_type: AnswerType) -> Self {
|
|
answer_type.to_i32().unwrap()
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct MessageAnswer {
|
|
#[serde(flatten)]
|
|
pub base: MessageBase,
|
|
pub answer_type: AnswerType,
|
|
}
|
|
|
|
pub fn encode_message<T: Serialize>(message: &T) -> Result<String, Box<dyn Error>> {
|
|
// Serialize the message to JSON
|
|
let json = serde_json::to_string(message)?;
|
|
Ok(json)
|
|
}
|
|
|
|
pub fn decode_message(data: String) -> Result<MessageBase, Box<dyn Error + Send + Sync>> {
|
|
println!("Data: {}", data);
|
|
let base_message: MessageBase = serde_json::from_str(&data)?;
|
|
Ok(base_message)
|
|
}
|
|
|
|
pub fn decode_message_as<T: for<'de> Deserialize<'de>>(
|
|
data: String,
|
|
) -> Result<T, Box<dyn Error + Send + Sync>> {
|
|
let message: T = serde_json::from_str(&data)?;
|
|
Ok(message)
|
|
}
|