Enable restarting the video stream.

Sometimes a client may want to restart the video stream while keeping the same config.
H-M-H · Nov 12, 2024 · d30b8b3 · d30b8b3
1 parent 00a19aa
commit d30b8b3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/protocol.rs b/src/protocol.rs
@@ -21,6 +21,7 @@ pub enum MessageInbound {
     Config(ClientConfiguration),
     PauseVideo,
     ResumeVideo,
+    RestartVideo,
     ChooseCustomInputAreas,
 }
 

diff --git a/src/websocket.rs b/src/websocket.rs
@@ -31,6 +31,7 @@ enum VideoCommands {
     Start(VideoConfig),
     Pause,
     Resume,
+    Restart,
 }
 
 fn send_message<S>(sender: &mut S, message: MessageOutbound)
@@ -120,6 +121,9 @@ impl<S, R, FnUInput> WeylusClientHandler<S, R, FnUInput> {
                         MessageInbound::ResumeVideo => {
                             self.video_sender.send(VideoCommands::Resume).unwrap()
                         }
+                        MessageInbound::RestartVideo => {
+                            self.video_sender.send(VideoCommands::Restart).unwrap()
+                        }
                         MessageInbound::ChooseCustomInputAreas => {
                             let (sender, receiver) = std::sync::mpsc::channel();
                             crate::gui::get_input_area(self.config.no_gui, sender);
@@ -372,6 +376,9 @@ fn handle_video<S: WeylusSender + Clone + 'static>(
             Ok(VideoCommands::Resume) => {
                 paused = false;
             }
+            Ok(VideoCommands::Restart) => {
+                video_encoder = None;
+            }
             Err(RecvTimeoutError::Timeout) => {
                 if recorder.is_none() {
                     warn!("Screen capture not initalized, can not send video frame!");