refactor: ♻️ gui connect, record

This commit is contained in:
Ahmet Kaan GÜMÜŞ 2024-05-01 19:37:52 +03:00
parent 83e0d40c0e
commit 2f5a980df8
5 changed files with 148 additions and 81 deletions

View file

@ -5,7 +5,7 @@ use iced::{
use tokio::sync::broadcast::{channel, Receiver, Sender};
use crate::{
gui_components::button_with_centered_text, gui_utils, recording, streaming, utils::get_config, Config, BUFFER_LENGTH
gui_components::button_with_centered_text, gui_utils, utils::get_config, Config, BUFFER_LENGTH,
};
#[derive(Debug, Clone)]
@ -26,7 +26,7 @@ pub enum Event {
StopAudio,
LoadConfig(Config),
IcedEvent(iced::Event),
CloseWindow(window::Id)
CloseWindow(window::Id),
}
#[derive(Debug, Clone)]
@ -113,23 +113,22 @@ impl Streamer {
Event::Connect => {
println!("Connect");
self.gui_status.are_we_connect = Condition::Loading;
let mut streaming_to_base_receiver =
self.communication_channel.streaming_to_base.subscribe();
tokio::spawn(streaming::connect(
self.data_channel.sound_stream_sender.subscribe(),
self.config.clone().unwrap(),
self.communication_channel.base_to_streaming.subscribe(),
self.communication_channel.streaming_to_base.clone(),
));
let sound_stream_receiver = self.data_channel.sound_stream_sender.subscribe();
let streamer_config = self.config.clone().unwrap();
let streaming_to_base = self.communication_channel.streaming_to_base.clone();
let base_to_streaming =
self.communication_channel.base_to_streaming.subscribe();
Command::perform(
async move {
match streaming_to_base_receiver.recv().await {
Ok(_) => State::Connected,
Err(err_val) => {
eprintln!("Error: Communication | {}", err_val);
State::Disconnected
}
}
gui_utils::connect(
sound_stream_receiver,
streamer_config,
streaming_to_base,
base_to_streaming,
)
.await
},
Message::State,
)
@ -137,31 +136,33 @@ impl Streamer {
Event::Disconnect => {
println!("Disconnect");
self.gui_status.are_we_connect = Condition::Loading;
let streaming_to_base = self.communication_channel.streaming_to_base.subscribe();
let streaming_to_base =
self.communication_channel.streaming_to_base.subscribe();
let base_to_streaming = self.communication_channel.base_to_streaming.clone();
Command::perform(async move {
gui_utils::disconnect(streaming_to_base, base_to_streaming).await
}, Message::State)
Command::perform(
async move { gui_utils::disconnect(streaming_to_base, base_to_streaming).await },
Message::State,
)
}
Event::Record => {
println!("Record");
self.gui_status.are_we_record = Condition::Loading;
let mut recording_to_base_receiver =
self.communication_channel.recording_to_base.subscribe();
tokio::spawn(recording::record(
self.data_channel.sound_stream_sender.clone(),
self.communication_channel.base_to_recording.subscribe(),
self.communication_channel.recording_to_base.clone(),
));
let sound_stream_sender = self.data_channel.sound_stream_sender.clone();
let recording_to_base = self.communication_channel.recording_to_base.clone();
let base_to_recording =
self.communication_channel.base_to_recording.subscribe();
Command::perform(
async move {
match recording_to_base_receiver.recv().await {
Ok(_) => State::Recording,
Err(err_val) => {
eprintln!("Error: Communication | Streaming | {}", err_val);
State::StopRecording
}
}
gui_utils::record(
sound_stream_sender,
recording_to_base,
base_to_recording,
)
.await
},
Message::State,
)
@ -169,11 +170,15 @@ impl Streamer {
Event::StopRecord => {
println!("Stop Record");
self.gui_status.are_we_record = Condition::Loading;
let recording_to_base = self.communication_channel.recording_to_base.subscribe();
let recording_to_base =
self.communication_channel.recording_to_base.subscribe();
let base_to_recording = self.communication_channel.base_to_recording.clone();
Command::perform(async move {
gui_utils::stop_recording(recording_to_base, base_to_recording).await
}, Message::State)
Command::perform(
async move {
gui_utils::stop_recording(recording_to_base, base_to_recording).await
},
Message::State,
)
}
Event::PlayAudio => {
println!("Play Audio");
@ -230,9 +235,7 @@ impl Streamer {
iced::Event::Touch(_) => Command::none(),
iced::Event::PlatformSpecific(_) => Command::none(),
},
Event::CloseWindow(id) => {
window::close(id)
},
Event::CloseWindow(id) => window::close(id),
},
Message::State(state) => match state {
State::None => Command::none(),
@ -316,28 +319,31 @@ impl Streamer {
fn call_closer(
streaming_to_base: Receiver<bool>,
base_to_streaming: Sender<bool>,
recording_to_base: Receiver<bool>,
recording_to_base: Receiver<bool>,
base_to_recording: Sender<bool>,
playing_to_base: Receiver<bool>,
base_to_playing: Sender<bool>,
features_in_need: Features,
window_id: window::Id,
)-> Command<Message> {
Command::perform(async move {
if features_in_need.stream {
gui_utils::disconnect(streaming_to_base, base_to_streaming).await;
}
if features_in_need.record {
gui_utils::stop_recording(recording_to_base, base_to_recording).await;
}
if features_in_need.play_audio {
gui_utils::stop_playing_audio(playing_to_base, base_to_playing).await;
}
Event::CloseWindow(window_id)
}, Message::Event)
) -> Command<Message> {
Command::perform(
async move {
if features_in_need.stream {
gui_utils::disconnect(streaming_to_base, base_to_streaming).await;
}
if features_in_need.record {
gui_utils::stop_recording(recording_to_base, base_to_recording).await;
}
if features_in_need.play_audio {
gui_utils::stop_playing_audio(playing_to_base, base_to_playing).await;
}
Event::CloseWindow(window_id)
},
Message::Event,
)
}
fn exit(&self, window_id: window::Id) -> Command<Message> {
let mut features_in_need = Features{
let mut features_in_need = Features {
stream: false,
record: false,
play_audio: false,
@ -361,6 +367,15 @@ impl Streamer {
let playing_to_base = self.communication_channel.playing_to_base.subscribe();
let base_to_playing = self.communication_channel.base_to_playing.clone();
Self::call_closer(streaming_to_base, base_to_streaming, recording_to_base, base_to_recording, playing_to_base, base_to_playing, features_in_need, window_id)
Self::call_closer(
streaming_to_base,
base_to_streaming,
recording_to_base,
base_to_recording,
playing_to_base,
base_to_playing,
features_in_need,
window_id,
)
}
}
}

View file

@ -1,8 +1,33 @@
use tokio::sync::broadcast::{Receiver, Sender};
use crate::gui::State;
use crate::{gui::State, recording, streaming, Config};
pub async fn disconnect(mut streaming_to_base:Receiver<bool>, base_to_streaming:Sender<bool>) -> State {
pub async fn connect(
sound_stream_receiver: Receiver<f32>,
streamer_config: Config,
streaming_to_base: Sender<bool>,
base_to_streaming: Receiver<bool>,
) -> State {
let mut streaming_to_base_receiver = streaming_to_base.subscribe();
tokio::spawn(streaming::connect(
sound_stream_receiver,
streamer_config,
base_to_streaming,
streaming_to_base.clone(),
));
match streaming_to_base_receiver.recv().await {
Ok(_) => State::Connected,
Err(err_val) => {
eprintln!("Error: Communication | {}", err_val);
State::Disconnected
}
}
}
pub async fn disconnect(
mut streaming_to_base: Receiver<bool>,
base_to_streaming: Sender<bool>,
) -> State {
let _ = base_to_streaming.send(false);
match streaming_to_base.recv().await {
Ok(_) => State::Disconnected,
@ -11,10 +36,33 @@ pub async fn disconnect(mut streaming_to_base:Receiver<bool>, base_to_streaming:
State::Connected
}
}
}
pub async fn stop_recording(mut recording_to_base:Receiver<bool>, base_to_recording:Sender<bool>) -> State {
pub async fn record(
sound_stream_sender: Sender<f32>,
recording_to_base: Sender<bool>,
base_to_recording: Receiver<bool>,
) -> State {
let mut recording_to_base_receiver = recording_to_base.subscribe();
tokio::spawn(recording::record(
sound_stream_sender.clone(),
recording_to_base.clone(),
base_to_recording,
));
match recording_to_base_receiver.recv().await {
Ok(_) => State::Recording,
Err(err_val) => {
eprintln!("Error: Communication | Streaming | {}", err_val);
State::StopRecording
}
}
}
pub async fn stop_recording(
mut recording_to_base: Receiver<bool>,
base_to_recording: Sender<bool>,
) -> State {
let _ = base_to_recording.send(false);
match recording_to_base.recv().await {
Ok(_) => State::StopRecording,
@ -23,10 +71,12 @@ pub async fn stop_recording(mut recording_to_base:Receiver<bool>, base_to_record
State::Recording
}
}
}
pub async fn stop_playing_audio(mut audio_to_base:Receiver<bool>, base_to_audio:Sender<bool>) -> State {
pub async fn stop_playing_audio(
mut audio_to_base: Receiver<bool>,
base_to_audio: Sender<bool>,
) -> State {
let _ = base_to_audio.send(false);
match audio_to_base.recv().await {
Ok(_) => State::StopRecording,
@ -35,5 +85,4 @@ pub async fn stop_playing_audio(mut audio_to_base:Receiver<bool>, base_to_audio:
State::Recording
}
}
}
}

View file

@ -13,4 +13,4 @@ pub struct Config {
pub quality: u8,
pub latency: u16,
pub tls: bool,
}
}

View file

@ -1,16 +1,17 @@
use streamer::gui::Streamer;
#[tokio::main]
async fn main(){
async fn main() {
println!("Hello, world!");
tokio::task::block_in_place(|| {
iced::program("Streamer GUI", Streamer::update, Streamer::view)
.centered()
.window_size((350.0, 400.0))
.load(Streamer::load_config)
.antialiasing(true)
.subscription(Streamer::subscription)
.exit_on_close_request(false)
.run().unwrap()
.centered()
.window_size((350.0, 400.0))
.load(Streamer::load_config)
.antialiasing(true)
.subscription(Streamer::subscription)
.exit_on_close_request(false)
.run()
.unwrap()
});
}

View file

@ -1,13 +1,14 @@
use cpal::traits::{DeviceTrait, HostTrait, StreamTrait};
use tokio::{sync::broadcast::{Receiver, Sender}, task};
use tokio::{
sync::broadcast::{Receiver, Sender},
task,
};
pub async fn record(
sound_stream_sender: Sender<f32>,
mut base_to_recording: Receiver<bool>,
recording_to_base: Sender<bool>,
mut base_to_recording: Receiver<bool>,
) {
let _ = recording_to_base.send(true);
let host = cpal::default_host();
let input_device = host.default_input_device().unwrap();
@ -28,12 +29,13 @@ pub async fn record(
.unwrap();
input_stream.play().unwrap();
println!("Recording Started");
tokio::spawn(let_the_base_know(recording_to_base.clone()));
task::block_in_place(|| {
let _ = base_to_recording.blocking_recv();
});
input_stream.pause().unwrap();
tokio::spawn(let_the_base_know(recording_to_base.clone()));
println!("Recording Stopped");