feat(multi-rig): auto-discover per-rig audio ports via GetRigs

This commit is contained in:
2026-02-25 23:31:38 +01:00
parent b1c232f388
commit 47cff3f927
9 changed files with 50 additions and 12 deletions
+17 -2
View File
@@ -15,6 +15,7 @@ use tokio::net::TcpStream;
use tokio::sync::{broadcast, mpsc, watch};
use tokio::time;
use tracing::{info, warn};
use trx_frontend::RemoteRigEntry;
use trx_core::audio::{
read_audio_msg, write_audio_msg, AudioStreamInfo, AUDIO_MSG_APRS_DECODE, AUDIO_MSG_CW_DECODE,
@@ -29,6 +30,7 @@ pub async fn run_audio_client(
default_port: u16,
rig_ports: HashMap<String, u16>,
selected_rig_id: Arc<Mutex<Option<String>>>,
known_rigs: Arc<Mutex<Vec<RemoteRigEntry>>>,
rx_tx: broadcast::Sender<Bytes>,
mut tx_rx: mpsc::Receiver<Bytes>,
stream_info_tx: watch::Sender<Option<AudioStreamInfo>>,
@@ -47,6 +49,7 @@ pub async fn run_audio_client(
&server_host,
default_port,
&rig_ports,
&known_rigs,
selected_rig_id
.lock()
.ok()
@@ -63,6 +66,7 @@ pub async fn run_audio_client(
default_port,
&rig_ports,
&selected_rig_id,
&known_rigs,
&server_addr,
&rx_tx,
&mut tx_rx,
@@ -104,6 +108,7 @@ async fn handle_audio_connection(
default_port: u16,
rig_ports: &HashMap<String, u16>,
selected_rig_id: &Arc<Mutex<Option<String>>>,
known_rigs: &Arc<Mutex<Vec<RemoteRigEntry>>>,
connected_addr: &str,
rx_tx: &broadcast::Sender<Bytes>,
tx_rx: &mut mpsc::Receiver<Bytes>,
@@ -196,6 +201,7 @@ async fn handle_audio_connection(
server_host,
default_port,
rig_ports,
known_rigs,
current_rig.as_deref(),
);
if desired_addr != connected_addr {
@@ -217,11 +223,20 @@ fn resolve_audio_addr(
host: &str,
default_port: u16,
rig_ports: &HashMap<String, u16>,
known_rigs: &Arc<Mutex<Vec<RemoteRigEntry>>>,
selected_rig_id: Option<&str>,
) -> String {
let port = selected_rig_id
.and_then(|rig_id| rig_ports.get(rig_id))
.copied()
.and_then(|rig_id| {
rig_ports.get(rig_id).copied().or_else(|| {
known_rigs.lock().ok().and_then(|entries| {
entries
.iter()
.find(|entry| entry.rig_id == rig_id)
.and_then(|entry| entry.audio_port)
})
})
})
.unwrap_or(default_port);
format!("{}:{}", host, port)
}
+1
View File
@@ -295,6 +295,7 @@ async fn async_init() -> DynResult<AppState> {
cfg.frontends.audio.server_port,
audio_rig_ports,
frontend_runtime.remote_active_rig_id.clone(),
frontend_runtime.remote_rigs.clone(),
rx_audio_tx,
tx_audio_rx,
stream_info_tx,
+2
View File
@@ -278,6 +278,7 @@ fn cache_remote_rigs(config: &RemoteClientConfig, rigs: &[RigEntry]) {
.map(|entry| RemoteRigEntry {
rig_id: entry.rig_id.clone(),
state: entry.state.clone(),
audio_port: entry.audio_port,
})
.collect();
}
@@ -561,6 +562,7 @@ mod tests {
rigs: Some(vec![RigEntry {
rig_id: "default".to_string(),
state: snapshot.clone(),
audio_port: Some(4531),
}]),
error: None,
})
+1
View File
@@ -21,6 +21,7 @@ use trx_core::{DynResult, RigRequest, RigState};
pub struct RemoteRigEntry {
pub rig_id: String,
pub state: RigSnapshot,
pub audio_port: Option<u16>,
}
/// Trait implemented by concrete frontends to expose a runner entrypoint.
@@ -244,6 +244,7 @@ fn snapshot_remote_rigs(context: &FrontendRuntimeContext) -> Vec<RigEntry> {
.map(|entry| RigEntry {
rig_id: entry.rig_id.clone(),
state: entry.state.clone(),
audio_port: entry.audio_port,
})
.collect()
})
+2
View File
@@ -55,6 +55,8 @@ pub struct ClientEnvelope {
pub struct RigEntry {
pub rig_id: String,
pub state: RigSnapshot,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub audio_port: Option<u16>,
}
/// Response sent to network clients over TCP.
+12 -3
View File
@@ -244,6 +244,7 @@ async fn handle_client(
entries.push(RigEntry {
rig_id: handle.rig_id.clone(),
state: snapshot,
audio_port: Some(handle.audio_port),
});
}
}
@@ -303,7 +304,10 @@ async fn handle_client(
match time::timeout(IO_TIMEOUT, handle.rig_tx.send(req)).await {
Ok(Ok(())) => {}
Ok(Err(e)) => {
error!("Failed to send request to rig_task for '{}': {:?}", target_rig_id, e);
error!(
"Failed to send request to rig_task for '{}': {:?}",
target_rig_id, e
);
let resp = ClientResponse {
success: false,
rig_id: Some(target_rig_id.clone()),
@@ -404,8 +408,8 @@ mod tests {
use trx_core::radio::freq::Band;
use trx_core::rig::request::RigRequest;
use trx_core::rig::{RigAccessMethod, RigCapabilities, RigInfo};
use trx_core::rig::state::RigState;
use trx_core::rig::{RigAccessMethod, RigCapabilities, RigInfo};
fn loopback_addr() -> SocketAddr {
let listener = std::net::TcpListener::bind((Ipv4Addr::LOCALHOST, 0)).expect("bind");
@@ -458,6 +462,7 @@ mod tests {
rig_id: "default".to_string(),
rig_tx,
state_rx,
audio_port: 4531,
};
let mut map = HashMap::new();
map.insert("default".to_string(), handle);
@@ -568,7 +573,11 @@ mod tests {
reader.read_line(&mut line).await.expect("read");
let resp: ClientResponse = serde_json::from_str(line.trim_end()).expect("response json");
assert!(!resp.success);
assert!(resp.error.as_deref().unwrap_or("").contains("Unknown rig_id"));
assert!(resp
.error
.as_deref()
.unwrap_or("")
.contains("Unknown rig_id"));
let _ = shutdown_tx.send(true);
handle.abort();
+12 -7
View File
@@ -219,11 +219,9 @@ fn access_from_rig_instance(rig_cfg: &RigInstanceConfig) -> DynResult<RigAccess>
let args = rig_cfg.rig.access.args.clone().unwrap_or_default();
Ok(RigAccess::Sdr { args })
}
Some(other) => Err(format!(
"Unknown access type '{}' for rig '{}'",
other, rig_cfg.id
)
.into()),
Some(other) => {
Err(format!("Unknown access type '{}' for rig '{}'", other, rig_cfg.id).into())
}
}
}
@@ -455,7 +453,10 @@ fn spawn_rig_audio_stack(
if rig_cfg.audio.rx_enabled {
if let Some(mut sdr_rx) = sdr_pcm_rx {
// SDR path: the backend pipeline provides demodulated PCM.
info!("[{}] using SDR audio source — cpal capture disabled", rig_cfg.id);
info!(
"[{}] using SDR audio source — cpal capture disabled",
rig_cfg.id
);
let pcm_tx_clone = pcm_tx.clone();
handles.push(tokio::spawn(async move {
loop {
@@ -661,7 +662,10 @@ async fn main() -> DynResult<()> {
}
}
}
let callsign = cli.callsign.clone().or_else(|| cfg.general.callsign.clone());
let callsign = cli
.callsign
.clone()
.or_else(|| cfg.general.callsign.clone());
(callsign, cfg.general.latitude, cfg.general.longitude)
};
@@ -799,6 +803,7 @@ async fn main() -> DynResult<()> {
rig_id: rig_cfg.id.clone(),
rig_tx,
state_rx,
audio_port: rig_cfg.audio.port,
},
);
}
+2
View File
@@ -20,4 +20,6 @@ pub struct RigHandle {
pub rig_tx: mpsc::Sender<RigRequest>,
/// Watch the latest rig state for fast GetState/GetRigs responses.
pub state_rx: watch::Receiver<RigState>,
/// Per-rig audio listener TCP port.
pub audio_port: u16,
}