feat(api_server): add websocket v2 api

spacemeowx2 · Feb 1, 2025 · a3eef25 · a3eef25
1 parent d04145d
commit a3eef25
Show file tree

Hide file tree

Showing 6 changed files with 358 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -43,6 +43,7 @@ async-stream = "0.3.0"
 once_cell = "1.7.2"
 
 axum = { version = "0.6.19", features = ["ws"], optional = true }
+jsonrpc-core = { version = "18.0.0", optional = true }
 serde_urlencoded = { version = "0.7.1", optional = true }
 tower-http = { version = "0.4.3", features = [
     "fs",
@@ -92,6 +93,7 @@ rusty-hook = "0.11.0"
 default = ["ss", "trojan", "rpc", "obfs", "api_server", "rhai", "raw"]
 api_server = [
     "axum",
+    "jsonrpc-core",
     "serde_urlencoded",
     "tower-http",
     "json-patch",

diff --git a/src/api_server.rs b/src/api_server.rs
@@ -7,6 +7,7 @@ use crate::config::ConfigManager;
 
 mod handlers;
 mod routes;
+pub mod v2;
 
 pub struct ApiServer {
     pub rabbit_digger: RabbitDigger,

diff --git a/src/api_server/routes.rs b/src/api_server/routes.rs
@@ -24,6 +24,7 @@ use crate::storage::{FileStorage, FolderType};
 
 use super::{
     handlers::{self, Ctx},
+    v2::handlers as v2_handlers,
     ApiServer,
 };
 
@@ -76,6 +77,7 @@ impl ApiServer {
             .route("/userdata", get(handlers::list_userdata))
             .route("/stream/connection", get(handlers::get_connection))
             .route("/stream/logs", get(handlers::ws_log))
+            .route("/v2/ws", get(v2_handlers::websocket_handler))
             .layer(Extension(ctx));
 
         if let Some(token) = &self.access_token {