feat(RESTful): support kick user, list detailed online users and Bear…

…er authorization
Itsusinn · Sep 15, 2024 · 2015dcf · 2015dcf
1 parent 351bedd
commit 2015dcf
Show file tree

Hide file tree

Showing 6 changed files with 213 additions and 16 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -5,7 +5,7 @@ resolver = "2"
 [workspace.package]
 authors = ["EAimTY <[email protected]>"]
 version = "1.1.1"
-rust-version = "1.65.0"
+rust-version = "1.80.0"
 edition = "2021"
 readme = "README.md"
 license = "GPL-3.0-or-later"

diff --git a/tuic-server/Cargo.toml b/tuic-server/Cargo.toml
@@ -19,6 +19,7 @@ lexopt = { version = "0.3", default-features = false }
 socket2 = { version = "0.5", default-features = false }
 arc-swap = "1"
 uuid = { version = "1", default-features = false, features = ["serde", "std", "v4"] }
+chashmap = { package = "chashmap-async", version = "0.1" }
 
 # QUIC
 quinn = { version = "0.11", default-features = false, features = ["runtime-tokio", "rustls", "log"] }

diff --git a/tuic-server/src/config.rs b/tuic-server/src/config.rs
@@ -116,6 +116,8 @@ pub struct RestfulConfig {
     pub addr: SocketAddr,
     #[educe(Default = "YOUR_SECRET_HERE")]
     pub secret: String,
+    #[educe(Default = 0)]
+    pub maximum_clients_per_user: u64,
 }
 
 impl Config {

diff --git a/tuic-server/src/connection/mod.rs b/tuic-server/src/connection/mod.rs
@@ -148,7 +148,7 @@ impl Connection {
 
         match self.auth.get() {
             Some(uuid) => {
-                restful::client_connect(&uuid);
+                restful::client_connect(&uuid, self.inner).await;
             }
             None => {
                 warn!(
@@ -167,7 +167,7 @@ impl Connection {
 
             if self.is_closed() {
                 if let Some(uuid) = self.auth.get() {
-                    restful::client_disconnect(&uuid);
+                    restful::client_disconnect(&uuid, self.inner).await;
                 }
                 break;
             }