David Kleingeld
95ccce3095
Adds input to the experimental rodio_audio pipeline.
Enable with:
```json
"audio": {
"experimental.rodio_audio": true
}
```
Additionally enables automatic volume
control for incoming audio:
```json
"audio": {
"experimental.control_output_volume": true
}
```
Release Notes:
- N/A
78 lines
2.7 KiB
Rust
78 lines
2.7 KiB
Rust
use anyhow::{Context, anyhow};
|
|
use async_tar::{Builder, Header};
|
|
use gpui::{BackgroundExecutor, Task};
|
|
|
|
use collections::HashMap;
|
|
use parking_lot::Mutex;
|
|
use rodio::Source;
|
|
use smol::fs::File;
|
|
use std::{io, path::PathBuf, sync::Arc, time::Duration};
|
|
|
|
use crate::{REPLAY_DURATION, rodio_ext::Replay};
|
|
|
|
#[derive(Default, Clone)]
|
|
pub(crate) struct Replays(Arc<Mutex<HashMap<String, Replay>>>);
|
|
|
|
impl Replays {
|
|
pub(crate) fn add_voip_stream(&self, stream_name: String, source: Replay) {
|
|
let mut map = self.0.lock();
|
|
map.retain(|_, replay| replay.source_is_active());
|
|
map.insert(stream_name, source);
|
|
}
|
|
|
|
pub(crate) fn replays_to_tar(
|
|
&self,
|
|
executor: BackgroundExecutor,
|
|
) -> Task<anyhow::Result<(PathBuf, Duration)>> {
|
|
let map = Arc::clone(&self.0);
|
|
executor.spawn(async move {
|
|
let recordings: Vec<_> = map
|
|
.lock()
|
|
.iter_mut()
|
|
.map(|(name, replay)| {
|
|
let queued = REPLAY_DURATION.min(replay.duration_ready());
|
|
(name.clone(), replay.take_duration(queued).record())
|
|
})
|
|
.collect();
|
|
let longest = recordings
|
|
.iter()
|
|
.map(|(_, r)| {
|
|
r.total_duration()
|
|
.expect("SamplesBuffer always returns a total duration")
|
|
})
|
|
.max()
|
|
.ok_or(anyhow!("There is no audio to capture"))?;
|
|
|
|
let path = std::env::current_dir()
|
|
.context("Could not get current dir")?
|
|
.join("replays.tar");
|
|
let tar = File::create(&path)
|
|
.await
|
|
.context("Could not create file for tar")?;
|
|
|
|
let mut tar = Builder::new(tar);
|
|
|
|
for (name, recording) in recordings {
|
|
let mut writer = io::Cursor::new(Vec::new());
|
|
rodio::wav_to_writer(recording, &mut writer).context("failed to encode wav")?;
|
|
let wav_data = writer.into_inner();
|
|
let path = name.replace(' ', "_") + ".wav";
|
|
let mut header = Header::new_gnu();
|
|
// rw permissions for everyone
|
|
header.set_mode(0o666);
|
|
header.set_size(wav_data.len() as u64);
|
|
tar.append_data(&mut header, path, wav_data.as_slice())
|
|
.await
|
|
.context("failed to apped wav to tar")?;
|
|
}
|
|
tar.into_inner()
|
|
.await
|
|
.context("Could not finish writing tar")?
|
|
.sync_all()
|
|
.await
|
|
.context("Could not flush tar file to disk")?;
|
|
Ok((path, longest))
|
|
})
|
|
}
|
|
}
|