feat(server): add version display for server

noctisynth · Apr 25, 2024 · 6716123 · 6716123
1 parent f74ee1b
commit 6716123
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "oblivion"
-version = "2.0.0"
+version = "2.0.1"
 authors = ["苏向夜 <fu050409@163.com>"]
 description = "Rust High Concurrency Implementation of Oblivion, an End-to-End Encryption Protocol Based on ECDHE Encryption Algorithm"
 edition = "2021"

diff --git a/src/lib.rs b/src/lib.rs
@@ -140,3 +140,5 @@ macro_rules! regex_route {
         );
     }};
 }
+
+pub static VERSION: &'static str = env!("CARGO_PKG_VERSION");
diff --git a/src/models/server.rs b/src/models/server.rs
@@ -3,6 +3,7 @@ use std::net::SocketAddr;
 use std::sync::Arc;
 
 use crate::utils::gear::Socket;
+use crate::VERSION;
 
 use anyhow::{Error, Result};
 use chrono::Local;
@@ -138,6 +139,19 @@ impl Server {
             std::process::exit(0);
         });
 
+        #[cfg(feature = "unsafe")]
+        println!(
+            "Oblivion version {}, using '{}'",
+            VERSION.bright_yellow(),
+            "p256".bright_red()
+        );
+        #[cfg(not(feature = "unsafe"))]
+        println!(
+            "Oblivion version {}, using '{}'",
+            VERSION.bright_yellow(),
+            "ring".bright_green()
+        );
+
         println!(
             "Starting server at {}",
             format!("Oblivion://{}:{}/", self.host, self.port).bright_cyan()
-Original file line number
+Diff line change
@@ Expand Up / @@ -140,3 +140,5 @@ macro_rules! regex_route { @@
             );
         }};
     }
+    pub static VERSION: &'static str = env!("CARGO_PKG_VERSION");