feat: Add command to change current timer to specific value

Author: on3iro

src/ipc.rs

@@ -67,6 +67,9 @@ pub enum ClientToServerMsg {
 
     /// Command the server to postpone the current break, if possible
     PostPone,
+
+    /// Sets current timer to a specific time (in seconds)
+    SetTimer(u64)
 }
 
 /// Service handling communication between processes over the zentime socket.

src/main.rs

@@ -11,6 +11,7 @@ use subcommands::{
     query_server_once::query_server_once,
     reset_timer::reset_timer,
     server::{start_daemonized, status, stop},
+    set_timer::set_timer,
     skip_timer::skip_timer,
     toggle_timer::toggle_timer,
 };
@@ -165,6 +166,9 @@ enum Commands {
     /// Postpones the current break (if possible)
     Postpone,
 
+    /// Sets current timer to a specific time in seconds
+    SetTimer { time: u64 },
+
     /// Interact with the zentime server
     Server {
         #[command(subcommand)]
@@ -233,6 +237,10 @@ fn main() {
             reset_timer(config.view.silent);
         }
 
+        Some(Commands::SetTimer { time }) => {
+            set_timer(config.view.silent, time.to_owned());
+        }
+
         None => default_cmd(&cli.common_args, config),
     }
 }

src/server/start.rs

@@ -227,6 +227,13 @@ async fn handle_client_to_server_msg(
         ClientToServerMsg::Sync => {
             info!("Client synced with server");
         }
+
+        // Set timer to a specific time
+        ClientToServerMsg::SetTimer(time) => {
+            timer_input_sender
+                .send(PomodoroTimerAction::SetTimer(time))
+                .context("Could not send SetTimer to timer")?;
+        },
     }
 
     Ok(CloseConnection::No)

src/subcommands.rs

@@ -2,5 +2,6 @@ pub mod postpone;
 pub mod query_server_once;
 pub mod reset_timer;
 pub mod server;
+pub mod set_timer;
 pub mod skip_timer;
 pub mod toggle_timer;

src/subcommands/set_timer.rs

@@ -0,0 +1,39 @@
+use futures::io::BufReader;
+use zentime_rs::client::one_shot_connection::one_shot_connection;
+use zentime_rs::ipc::ClientToServerMsg;
+use zentime_rs::ipc::InterProcessCommunication;
+use zentime_rs::ipc::ServerToClientMsg;
+
+#[tokio::main]
+pub async fn set_timer(silent: bool, time: u64) {
+    let (reader, mut writer) = match one_shot_connection().await {
+        Ok(c) => c,
+        Err(error) => panic!("Could not connect to server: {}", error),
+    };
+
+    let mut reader = BufReader::new(reader);
+
+    if let Err(err) =
+        InterProcessCommunication::send_ipc_message(ClientToServerMsg::SetTimer(time), &mut writer).await
+    {
+        panic!("Could not send to the server: {}", err)
+    };
+
+    let msg_result =
+        InterProcessCommunication::recv_ipc_message::<ServerToClientMsg>(&mut reader).await;
+
+    if !silent {
+        if let Ok(ServerToClientMsg::Timer(state)) = msg_result {
+            println!(
+                "{} {} {}",
+                state.round,
+                state.time,
+                if state.is_break { "Break" } else { "Focus" }
+            );
+        }
+    }
+
+    InterProcessCommunication::send_ipc_message(ClientToServerMsg::Detach, &mut writer)
+        .await
+        .ok();
+}

timer/src/pomodoro_timer/long_break.rs

@@ -59,6 +59,7 @@ impl PomodoroActionHandler<LongBreak> for LongBreakTickHandler {
             }
             PomodoroTimerAction::PlayPause => Some(TimerAction::PlayPause),
             PomodoroTimerAction::Skip => Some(TimerAction::End),
+            PomodoroTimerAction::SetTimer(time) => Some(TimerAction::SetTimer(time)),
 
             PomodoroTimerAction::ResetTimer => {
                 PomodoroTimer::<Interval>::reset(config, callbacks).init();

timer/src/pomodoro_timer/on_tick_handler.rs

@@ -37,6 +37,8 @@ pub trait PomodoroActionHandler<S: PomodoroState> {
                 None
             }
 
+            PomodoroTimerAction::SetTimer(time) => Some(TimerAction::SetTimer(time)),
+
             _ => None,
         }
     }

timer/src/pomodoro_timer/short_break.rs

@@ -65,6 +65,8 @@ impl PomodoroActionHandler<ShortBreak> for ShortBreakTickHandler {
                 None
             }
 
+            PomodoroTimerAction::SetTimer(time) => Some(TimerAction::SetTimer(time)),
+
             _ => None,
         }
     }

timer/src/pomodoro_timer_action.rs

@@ -17,4 +17,7 @@ pub enum PomodoroTimerAction {
 
     /// Postpone a break
     PostponeBreak,
+
+    /// Set current timer to a specific time in seconds
+    SetTimer(u64),
 }

timer/src/timer.rs

@@ -160,6 +160,10 @@ impl Timer<Paused> {
                 current_time: CurrentTime(seconds_to_time(time)),
             }) {
                 match action {
+                    TimerAction::SetTimer(time) => {
+                        self.internal_state.remaining_time = Duration::from_secs(time)
+                    }
+
                     TimerAction::PlayPause => {
                         self.unpause();
                         break;
@@ -240,6 +244,9 @@ impl Timer<Running> {
                     // Returns from the blocking loop, so that the calling code
                     // can resume execution
                     TimerAction::End => return,
+                    TimerAction::SetTimer(time) => {
+                        self.internal_state.target_time = Instant::now() + Duration::from_secs(time)
+                    }
                 }
             }
         }

timer/src/timer_action.rs

@@ -3,6 +3,9 @@
 /// Various control actions to transition into new states
 #[derive(Debug, Copy, Clone)]
 pub enum TimerAction {
+    /// Set current timer to a specific time in seconds
+    SetTimer(u64),
+
     /// Either start or pause the current timer
     PlayPause,