feat: Add sd_notify support

Author: provokateurin

Cargo.lock

@@ -45,6 +45,7 @@ rkyv = { version = "0.8.10", features = ["little_endian"] }
 form-data = { version = "0.6.0", features = ["sync"], default-features = false }
 mime = "0.3.17"
 compact_str = "0.9.0"
+sd-notify = "0.4.5"
 
 [dev-dependencies]
 

crates/http/Cargo.toml

@@ -10,13 +10,13 @@ use common::{
     ipc::{BroadcastEvent, HousekeeperEvent},
 };
 use directory::Permission;
+use http_proto::*;
 use hyper::Method;
+use sd_notify;
 use serde_json::json;
 use std::future::Future;
 use utils::url_params::UrlParams;
 
-use http_proto::*;
-
 pub trait ManageReload: Sync + Send {
     fn handle_manage_reload(
         &self,
@@ -43,7 +43,16 @@ impl ManageReload for Server {
         // Validate the access token
         access_token.assert_has_permission(Permission::SettingsReload)?;
 
-        match (path.get(1).copied(), req.method()) {
+        let is_systemd = match sd_notify::booted() {
+            Ok(booted) => booted,
+            _ => false,
+        };
+
+        if (is_systemd) {
+            let _ = sd_notify::notify(true, &[sd_notify::NotifyState::Reloading]);
+        }
+
+        let response = match (path.get(1).copied(), req.method()) {
             (Some("lookup"), &Method::GET) => {
                 let result = self.reload_lookups().await?;
                 // Update core
@@ -117,7 +126,13 @@ impl ManageReload for Server {
                 .into_http_response())
             }
             _ => Err(trc::ResourceEvent::NotFound.into_err()),
+        };
+
+        if (is_systemd) {
+            let _ = sd_notify::notify(true, &[sd_notify::NotifyState::Ready]);
         }
+
+        response
     }
 
     async fn handle_manage_update(

crates/http/src/management/reload.rs

@@ -35,6 +35,7 @@ trc = { path = "../trc" }
 utils = { path = "../utils" }
 migration = { path = "../migration" }
 tokio = { version = "1.47", features = ["full"] }
+sd-notify = "0.4.5"
 
 [target.'cfg(not(target_env = "msvc"))'.dependencies]
 jemallocator = "0.5.0"

crates/main/Cargo.toml

@@ -14,6 +14,7 @@ use http::HttpSessionManager;
 use imap::core::ImapSessionManager;
 use managesieve::core::ManageSieveSessionManager;
 use pop3::Pop3SessionManager;
+use sd_notify;
 use services::{StartServices, broadcast::subscriber::spawn_broadcast_subscriber};
 use smtp::{StartQueueManager, core::SmtpSessionManager};
 use std::time::Duration;
@@ -97,9 +98,22 @@ async fn main() -> std::io::Result<()> {
     // Start broadcast subscriber
     spawn_broadcast_subscriber(init.inner, shutdown_rx);
 
+    let is_systemd = match sd_notify::booted() {
+        Ok(booted) => booted,
+        _ => false,
+    };
+
+    if (is_systemd) {
+        let _ = sd_notify::notify(true, &[sd_notify::NotifyState::Ready]);
+    }
+
     // Wait for shutdown signal
     wait_for_shutdown().await;
 
+    if (is_systemd) {
+        let _ = sd_notify::notify(true, &[sd_notify::NotifyState::Stopping]);
+    }
+
     // Shutdown collector
     Collector::shutdown();