Gate behind feature flag

Author: ameba23

Cargo.toml

@@ -47,14 +47,19 @@ time = "0.3.44"
 once_cell = "1.21.3"
 axum = "0.8.6"
 tower-http = { version = "0.6.7", features = ["fs"] }
-tokio-tungstenite = "0.28.0"
-futures-util = "0.3.31"
+tokio-tungstenite = { version = "0.28.0", optional = true }
+futures-util = { version = "0.3.31", optional = true }
 
 [dev-dependencies]
 rcgen = "0.14.5"
 tempfile = "3.23.0"
 tdx-quote = { version = "0.0.5", features = ["mock"] }
 
 [features]
-default = ["azure"]
+default = ["azure", "ws"]
+
+# Adds support for Microsoft Azure attestation generation and verification
 azure = ["tss-esapi", "az-tdx-vtpm"]
+
+# Adds websocket support
+ws = ["tokio-tungstenite", "futures-util"]

src/lib.rs

@@ -3,6 +3,8 @@ pub mod attested_get;
 pub mod attested_tls;
 pub mod file_server;
 pub mod health_check;
+
+#[cfg(feature = "azure")]
 pub mod websockets;
 
 pub use attestation::AttestationGenerator;