Daemon run as linux service

platforms/unix/daemon/Cargo.toml

@@ -25,3 +25,4 @@ ezrtc = "0.5.0"
 webrtc = "0.11"
 async-channel = "2.3.1"
 wol-rs = "1"
+clap = { version = "4.5.23", features = ["derive"] }

platforms/unix/daemon/src/main.rs

@@ -8,6 +8,7 @@ use axum::{
     routing::get,
     Router,
 };
+use clap::Parser;
 use ezrtc::host::EzRTCHost;
 use ezrtc::socket::DataChannelHandler;
 use futures::{sink::SinkExt, stream::StreamExt};
@@ -31,6 +32,8 @@ use webrtc::data_channel::RTCDataChannel;
 use webrtc::ice_transport::ice_server::RTCIceServer;
 use wol::{send_wol, MacAddr};
 
+mod service;
+
 #[derive(Serialize, Deserialize)]
 pub struct GenericMessage<T> {
     pub r#type: String,
@@ -44,8 +47,20 @@ pub struct AppState {
     settings: Settings,
 }
 
+/// Modern hardware monitor with remote monitoring.
+#[derive(Parser, Debug)]
+#[command(version, about, long_about = None)]
+struct Args {
+    /// Setup coresd to run as a service
+    #[arg(required = false, long, short = 's')]
+    service: bool,
+}
+
 #[tokio::main]
 async fn main() {
+    // Parse arguments
+    let args = Args::parse();
+
     // Logger
     CombinedLogger::init(vec![TermLogger::new(
         LevelFilter::Info,
@@ -55,6 +70,14 @@ async fn main() {
     )])
     .unwrap();
 
+    if args.service {
+        service::setup_service();
+    } else {
+        warn!(
+            "You are running coresd as an executable, to setup it as a service, run `sudo ./coresd --service`"
+        );
+    }
+
     // Get settings
     let settings = get_settings();
     info!("Connection code: {:?}", settings.connection_code);

platforms/unix/daemon/src/service.rs

@@ -0,0 +1,57 @@
+use log::{error, info};
+
+pub fn setup_service() {
+    // copy current executable to /bin as sudo
+    let current_exe = std::env::current_exe().expect("Failed to get current executable");
+    let status = std::fs::copy(current_exe, "/bin/coresd");
+
+    if let Err(e) = status {
+        error!(
+            "Failed to copy executable to /bin, please run as root! Error: {}",
+            e
+        );
+        std::process::exit(1);
+    }
+
+    let file_contents = "[Unit]
+Description=coresd
+After=network.target
+
+[Service]
+User=root
+ExecStart=/bin/coresd
+Restart=always
+
+[Install]
+WantedBy=multi-user.target";
+
+    // create service file
+    let status = std::fs::write("/etc/systemd/system/coresd.service", file_contents);
+
+    if let Err(e) = status {
+        error!(
+            "Failed to create coresd service file, please run as root! Error: {}",
+            e
+        );
+        std::process::exit(1);
+    }
+
+    // enable service
+    std::process::Command::new("sudo")
+        .arg("systemctl")
+        .arg("enable")
+        .arg("coresd")
+        .status()
+        .expect("Failed to enable coresd service");
+
+    // start service
+    std::process::Command::new("sudo")
+        .arg("systemctl")
+        .arg("start")
+        .arg("coresd")
+        .status()
+        .expect("Failed to start coresd service");
+
+    info!("Service created successfully, please run `sudo systemctl status coresd` for more information");
+    std::process::exit(0);
+}