Create method to obtain a `DaemonCommandSender`

1 files changed, 21 insertions, 1 deletions

diff --git a/mullvad-daemon/src/lib.rs b/mullvad-daemon/src/lib.rs
index f531ce9e36..37d6e7f6d5 100644
--- a/mullvad-daemon/src/lib.rs
+++ b/mullvad-daemon/src/lib.rs
@@ -20,7 +20,8 @@ mod relays;
 mod rpc_uniqueness_check;
 pub mod version;
 
-use crate::management_interface::{BoxFuture, ManagementCommand, ManagementInterfaceServer};
+pub use crate::management_interface::ManagementCommand;
+use crate::management_interface::{BoxFuture, ManagementInterfaceServer};
 use futures::{
     future,
     sync::{mpsc::UnboundedSender, oneshot},
@@ -59,6 +60,9 @@ pub enum Error {
     #[error(display = "Another instance of the daemon is already running")]
     DaemonIsAlreadyRunning,
 
+    #[error(display = "Failed to send command to daemon because it is not running")]
+    DaemonUnavailable,
+
     #[error(display = "Unable to initialize network event loop")]
     InitIoEventLoop(#[error(cause)] io::Error),
 
@@ -162,6 +166,17 @@ impl DaemonExecutionState {
     }
 }
 
+pub struct DaemonCommandSender(IntoSender<ManagementCommand, InternalDaemonEvent>);
+
+impl DaemonCommandSender {
+    pub(crate) fn new(internal_event_sender: mpsc::Sender<InternalDaemonEvent>) -> Self {
+        DaemonCommandSender(IntoSender::from(internal_event_sender))
+    }
+
+    pub fn send(&self, command: ManagementCommand) -> Result<()> {
+        self.0.send(command).map_err(|_| Error::DaemonUnavailable)
+    }
+}
 
 pub struct Daemon {
     tunnel_command_tx: SyncUnboundedSender<TunnelCommand>,
@@ -309,6 +324,11 @@ impl Daemon {
         });
     }
 
+    /// Retrieve a channel for sending daemon commands.
+    pub fn command_sender(&self) -> DaemonCommandSender {
+        DaemonCommandSender::new(self.tx.clone())
+    }
+
     /// Consume the `Daemon` and run the main event loop. Blocks until an error happens or a
     /// shutdown event is received.
     pub fn run(mut self) -> Result<()> {