Change visibility of Sink and Stream

shadowylab · Mar 27, 2024 · eb82b2b · eb82b2b
1 parent 3c15c19
commit eb82b2b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -5,6 +5,8 @@
 
 #![forbid(unsafe_code)]
 
+#[cfg(target_arch = "wasm32")]
+pub use wasm_ws::WsMessage;
 pub use futures_util;
 pub use url;
 
@@ -14,7 +16,6 @@ pub mod native;
 pub mod wasm;
 
 #[cfg(target_arch = "wasm32")]
-pub use wasm_ws::WsMessage;
-
+pub use self::wasm::{Sink, Stream};
 #[cfg(not(target_arch = "wasm32"))]
-pub use self::native::Message as WsMessage;
+pub use self::native::{Message as WsMessage, Sink, Stream};
diff --git a/src/native/mod.rs b/src/native/mod.rs
@@ -22,8 +22,8 @@ use tokio_tungstenite::{MaybeTlsStream, WebSocketStream};
 use url::{ParseError, Url};
 
 type WebSocket = WebSocketStream<MaybeTlsStream<TcpStream>>;
-type Sink = SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>;
-type Stream = SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>;
+pub type Sink = SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>;
+pub type Stream = SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>;
 
 mod socks;
 

diff --git a/src/wasm.rs b/src/wasm.rs
@@ -12,8 +12,8 @@ use thiserror::Error;
 use url::Url;
 use wasm_ws::{WebSocket, WsErr, WsMessage, WsStream};
 
-type Sink = SplitSink<WsStream, WsMessage>;
-type Stream = SplitStream<WsStream>;
+pub type Sink = SplitSink<WsStream, WsMessage>;
+pub type Stream = SplitStream<WsStream>;
 
 #[derive(Debug, Error)]
 pub enum Error {