WispV2Extensions -> WispV2Handshake and motd helpers

client/src/stream_provider.rs

@@ -17,7 +17,7 @@ use webpki_roots::TLS_SERVER_ROOTS;
 use wisp_mux::{
 	extensions::{udp::UdpProtocolExtensionBuilder, AnyProtocolExtensionBuilder},
 	ws::{WebSocketRead, WebSocketWrite},
-	ClientMux, MuxStreamAsyncRW, MuxStreamIo, StreamType, WispV2Extensions,
+	ClientMux, MuxStreamAsyncRW, MuxStreamIo, StreamType, WispV2Handshake,
 };
 
 use crate::{
@@ -119,7 +119,7 @@ impl StreamProvider {
 				UdpProtocolExtensionBuilder,
 			)];
 		let extensions = if self.wisp_v2 {
-			Some(WispV2Extensions::new(extensions_vec))
+			Some(WispV2Handshake::new(extensions_vec))
 		} else {
 			None
 		};

server/src/config.rs

@@ -14,7 +14,7 @@ use wisp_mux::{
 		udp::UdpProtocolExtensionBuilder,
 		AnyProtocolExtensionBuilder,
 	},
-	WispV2Extensions,
+	WispV2Handshake,
 };
 
 use crate::{handle::wisp::utils::get_certificates_from_paths, CLI, CONFIG, RESOLVER};
@@ -346,7 +346,7 @@ impl Default for WispConfig {
 }
 
 impl WispConfig {
-	pub async fn to_opts(&self) -> anyhow::Result<(Option<WispV2Extensions>, Vec<u8>, u32)> {
+	pub async fn to_opts(&self) -> anyhow::Result<(Option<WispV2Handshake>, Vec<u8>, u32)> {
 		if self.wisp_v2 {
 			let mut extensions: Vec<AnyProtocolExtensionBuilder> = Vec::new();
 			let mut required_extensions: Vec<u8> = Vec::new();
@@ -391,7 +391,7 @@ impl WispConfig {
 			}
 
 			Ok((
-				Some(WispV2Extensions::new(extensions)),
+				Some(WispV2Handshake::new(extensions)),
 				required_extensions,
 				self.buffer_size,
 			))

simple-wisp-client/src/main.rs

@@ -38,7 +38,7 @@ use wisp_mux::{
 		udp::{UdpProtocolExtension, UdpProtocolExtensionBuilder},
 		AnyProtocolExtensionBuilder,
 	},
-	ClientMux, StreamType, WispError, WispV2Extensions,
+	ClientMux, StreamType, WispError, WispV2Handshake,
 };
 
 #[derive(Debug)]
@@ -198,7 +198,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
 			.await?
 			.with_no_required_extensions()
 	} else {
-		ClientMux::create(rx, tx, Some(WispV2Extensions::new(extensions)))
+		ClientMux::create(rx, tx, Some(WispV2Handshake::new(extensions)))
 			.await?
 			.with_required_extensions(extension_ids.as_slice())
 			.await?

wisp/src/extensions/motd.rs

@@ -76,6 +76,18 @@ pub enum MotdProtocolExtensionBuilder {
 	Client,
 }
 
+impl MotdProtocolExtensionBuilder {
+	/// Create a new server variant of the MOTD protocol extension builder.
+	pub fn new_server(motd: String) -> Self {
+		Self::Server(motd)
+	}
+
+	/// Create a new client variant of the MOTD protocol extension builder.
+	pub fn new_client() -> Self {
+		Self::Client
+	}
+}
+
 impl ProtocolExtensionBuilder for MotdProtocolExtensionBuilder {
 	fn get_id(&self) -> u8 {
 		MotdProtocolExtension::ID

wisp/src/mux/client.rs

@@ -20,15 +20,15 @@ use crate::{
 
 use super::{
 	get_supported_extensions, validate_continue_packet, Multiplexor, MuxResult,
-	WispHandshakeResult, WispHandshakeResultKind, WispV2Extensions,
+	WispHandshakeResult, WispHandshakeResultKind, WispV2Handshake,
 };
 
 async fn handshake<R: WebSocketRead>(
 	rx: &mut R,
 	tx: &LockedWebSocketWrite,
-	v2_info: Option<WispV2Extensions>,
+	v2_info: Option<WispV2Handshake>,
 ) -> Result<(WispHandshakeResult, u32), WispError> {
-	if let Some(WispV2Extensions {
+	if let Some(WispV2Handshake {
 		mut builders,
 		closure,
 	}) = v2_info
@@ -100,7 +100,7 @@ impl ClientMux {
 	pub async fn create<R, W>(
 		mut rx: R,
 		tx: W,
-		wisp_v2: Option<WispV2Extensions>,
+		wisp_v2: Option<WispV2Handshake>,
 	) -> Result<MuxResult<ClientMux, impl Future<Output = Result<(), WispError>> + Send>, WispError>
 	where
 		R: WebSocketRead + Send,

wisp/src/mux/mod.rs

@@ -123,14 +123,19 @@ where
 	}
 }
 
-type WispV2ClosureResult = Pin<Box<dyn Future<Output = Result<(), WispError>> + Sync + Send>>;
 /// Wisp V2 handshake and protocol extension settings wrapper struct.
-pub struct WispV2Extensions {
+pub struct WispV2Handshake {
 	builders: Vec<AnyProtocolExtensionBuilder>,
-	closure: Box<dyn Fn(&mut Vec<AnyProtocolExtensionBuilder>) -> WispV2ClosureResult + Send>,
+	#[expect(clippy::type_complexity)]
+	closure: Box<
+		dyn Fn(
+				&mut Vec<AnyProtocolExtensionBuilder>,
+			) -> Pin<Box<dyn Future<Output = Result<(), WispError>> + Sync + Send>>
+			+ Send,
+	>,
 }
 
-impl WispV2Extensions {
+impl WispV2Handshake {
 	/// Create a Wisp V2 settings struct with no middleware.
 	pub fn new(builders: Vec<AnyProtocolExtensionBuilder>) -> Self {
 		Self {
@@ -142,7 +147,11 @@ impl WispV2Extensions {
 	/// Create a Wisp V2 settings struct with some middleware.
 	pub fn new_with_middleware<C>(builders: Vec<AnyProtocolExtensionBuilder>, closure: C) -> Self
 	where
-		C: Fn(&mut Vec<AnyProtocolExtensionBuilder>) -> WispV2ClosureResult + Send + 'static,
+		C: Fn(
+				&mut Vec<AnyProtocolExtensionBuilder>,
+			) -> Pin<Box<dyn Future<Output = Result<(), WispError>> + Sync + Send>>
+			+ Send
+			+ 'static,
 	{
 		Self {
 			builders,

wisp/src/mux/server.rs

@@ -19,16 +19,16 @@ use crate::{
 
 use super::{
 	get_supported_extensions, send_info_packet, Multiplexor, MuxResult, WispHandshakeResult,
-	WispHandshakeResultKind, WispV2Extensions,
+	WispHandshakeResultKind, WispV2Handshake,
 };
 
 async fn handshake<R: WebSocketRead>(
 	rx: &mut R,
 	tx: &LockedWebSocketWrite,
 	buffer_size: u32,
-	v2_info: Option<WispV2Extensions>,
+	v2_info: Option<WispV2Handshake>,
 ) -> Result<WispHandshakeResult, WispError> {
-	if let Some(WispV2Extensions {
+	if let Some(WispV2Handshake {
 		mut builders,
 		closure,
 	}) = v2_info
@@ -95,7 +95,7 @@ impl ServerMux {
 		mut rx: R,
 		tx: W,
 		buffer_size: u32,
-		wisp_v2: Option<WispV2Extensions>,
+		wisp_v2: Option<WispV2Handshake>,
 	) -> Result<MuxResult<ServerMux, impl Future<Output = Result<(), WispError>> + Send>, WispError>
 	where
 		R: WebSocketRead + Send,