Supports restarting daemon processes and viewing status

src/daemon.rs

@@ -18,9 +18,7 @@ pub(crate) const DEFAULT_STDERR_PATH: &str = "/var/run/xunlei.err";
 pub(crate) const DEFAULT_WORK_DIR: &str = "/";
 
 pub fn check_root() {
-    use nix::unistd::Uid;
-
-    if !Uid::effective().is_root() {
+    if !nix::unistd::Uid::effective().is_root() {
         println!("You must run this executable with root permissions");
         std::process::exit(-1)
     }
@@ -96,6 +94,19 @@ pub(super) fn stop() -> anyhow::Result<()> {
     Ok(())
 }
 
+pub(super) fn restart() -> anyhow::Result<()> {
+    stop()?;
+    start()
+}
+
+pub(super) fn status() -> anyhow::Result<()> {
+    match get_pid() {
+        Some(pid) => println!("Xunlei is running with pid: {}", pid),
+        None => println!("Xunlei is not running"),
+    }
+    Ok(())
+}
+
 pub(super) fn log() -> anyhow::Result<()> {
     fn read_and_print_file(file_path: &Path, placeholder: &str) -> anyhow::Result<()> {
         if !file_path.exists() {

src/main.rs

@@ -41,9 +41,13 @@ pub enum Commands {
     Run(ServeConfig),
     /// Start xunlei daemon
     Start(ServeConfig),
+    /// Restart xunlei daemon
+    Restart(ServeConfig),
     /// Stop xunlei daemon
     Stop,
-    /// Xunlei log
+    /// Status of the Http server daemon process
+    Status,
+    /// Show the Http server daemon log
     Log,
 }
 
@@ -182,9 +186,16 @@ fn main() -> anyhow::Result<()> {
             daemon::start()?;
             serve::Serve::new(config, InstallConfig::read_from_file()?).run()?;
         }
+        Commands::Restart(config) => {
+            daemon::restart()?;
+            serve::Serve::new(config, InstallConfig::read_from_file()?).run()?;
+        }
         Commands::Stop => {
             daemon::stop()?;
         }
+        Commands::Status => {
+            daemon::status()?;
+        }
         Commands::Log => {
             daemon::log()?;
         }