add CLI config options

src-tauri/Cargo.toml

@@ -19,6 +19,7 @@ prost-build = { version = "0.12" }
 [dependencies]
 anyhow = "1.0.75"
 base64 = "0.21"
+clap = { version = "4.4", features = ["derive", "env"] }
 chrono = { version = "0.4", features = ["serde"] }
 defguard_wireguard_rs = { git = "https://github.com/DefGuard/wireguard-rs.git", branch = "main" }
 dirs = "5.0"

src-tauri/src/bin/defguard-service.rs

@@ -3,7 +3,8 @@
 //! This binary is meant to run as a daemon with root privileges
 //! and communicate with the desktop client over HTTP.
 
-use defguard_client::service::run_server;
+use clap::Parser;
+use defguard_client::service::{config::Config, run_server};
 use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
 
 #[tokio::main]
@@ -18,8 +19,11 @@ async fn main() -> anyhow::Result<()> {
         .with(tracing_subscriber::fmt::layer())
         .init();
 
+    // parse config
+    let config = Config::parse();
+
     // run gRPC server
-    run_server().await?;
+    run_server(config).await?;
 
     Ok(())
 }

src-tauri/src/service/config.rs

@@ -0,0 +1,10 @@
+use clap::Parser;
+
+#[derive(Debug, Parser, Clone)]
+#[clap(about = "Defguard VPN gateway service")]
+#[command(version)]
+pub struct Config {
+    /// Defines how often (in seconds) interface statistics are sent to Defguard client
+    #[arg(long, short = 'p', env = "DEFGUARD_STATS_PERIOD", default_value = "10")]
+    pub stats_period: u64,
+}

src-tauri/src/service/mod.rs

@@ -28,9 +28,12 @@ use proto::{
     CreateInterfaceRequest, InterfaceData, ReadInterfaceDataRequest, RemoveInterfaceRequest,
 };
 
+use self::config::Config;
+
+pub mod config;
+
 pub const DAEMON_HTTP_PORT: u16 = 54127;
 pub const DAEMON_BASE_URL: &str = "http://localhost:54127";
-const STATS_PERIOD: u64 = 60;
 
 #[derive(thiserror::Error, Debug)]
 pub enum DaemonError {
@@ -43,7 +46,17 @@ pub enum DaemonError {
 }
 
 #[derive(Default)]
-pub struct DaemonService {}
+pub struct DaemonService {
+    stats_period: u64,
+}
+
+impl DaemonService {
+    pub fn new(config: Config) -> Self {
+        Self {
+            stats_period: config.stats_period,
+        }
+    }
+}
 
 type InterfaceDataStream = Pin<Box<dyn Stream<Item = Result<InterfaceData, Status>> + Send>>;
 
@@ -135,13 +148,14 @@ impl DesktopDaemonService for DaemonService {
         let ifname = request.interface_name;
         info!("Starting interface data stream for {ifname}");
 
+        let stats_period = self.stats_period;
         let (tx, rx) = mpsc::channel(64);
         tokio::spawn(async move {
             info!("Spawning stats thread for interface {ifname}");
             // setup WireGuard API
             let wgapi =
                 setup_wgapi(ifname.clone()).expect("Failed to initialize WireGuard interface API");
-            let period = Duration::from_secs(STATS_PERIOD);
+            let period = Duration::from_secs(stats_period);
             let mut interval = interval(period);
 
             loop {
@@ -172,13 +186,13 @@ impl DesktopDaemonService for DaemonService {
     }
 }
 
-pub async fn run_server() -> anyhow::Result<()> {
+pub async fn run_server(config: Config) -> anyhow::Result<()> {
     info!("Starting defguard interface management daemon");
 
     let addr = format!("127.0.0.1:{DAEMON_HTTP_PORT}")
         .parse()
         .context("Failed to parse gRPC address")?;
-    let daemon_service = DaemonService::default();
+    let daemon_service = DaemonService::new(config);
 
     info!("defguard daemon listening on {addr}");