Reorganize component modules

tokio-xmpp/src/component/connect.rs

@@ -1,15 +0,0 @@
-use xmpp_parsers::{jid::Jid, ns};
-
-use crate::{component::auth::auth, connect::ServerConnector, proto::XmppStream, Error};
-
-/// Log into an XMPP server as a client with a jid+pass
-pub async fn component_login<C: ServerConnector>(
-    connector: C,
-    jid: Jid,
-    password: String,
-) -> Result<XmppStream<C::Stream>, Error> {
-    let password = password;
-    let mut xmpp_stream = connector.connect(&jid, ns::COMPONENT).await?;
-    auth(&mut xmpp_stream, password).await?;
-    Ok(xmpp_stream)
-}

tokio-xmpp/src/component/login.rs

@@ -1,9 +1,24 @@
 use futures::stream::StreamExt;
 use tokio::io::{AsyncRead, AsyncWrite};
-use xmpp_parsers::{component::Handshake, ns};
+use xmpp_parsers::{component::Handshake, jid::Jid, ns};
 
-use crate::error::{AuthError, Error};
-use crate::proto::{Packet, XmppStream};
+use crate::{
+    connect::ServerConnector,
+    error::{AuthError, Error},
+    proto::{Packet, XmppStream},
+};
+
+/// Log into an XMPP server as a client with a jid+pass
+pub async fn component_login<C: ServerConnector>(
+    connector: C,
+    jid: Jid,
+    password: String,
+) -> Result<XmppStream<C::Stream>, Error> {
+    let password = password;
+    let mut xmpp_stream = connector.connect(&jid, ns::COMPONENT).await?;
+    auth(&mut xmpp_stream, password).await?;
+    Ok(xmpp_stream)
+}
 
 pub async fn auth<S: AsyncRead + AsyncWrite + Unpin>(
     stream: &mut XmppStream<S>,

tokio-xmpp/src/component/mod.rs

@@ -1,27 +1,25 @@
 //! Components in XMPP are services/gateways that are logged into an
 //! XMPP server under a JID consisting of just a domain name. They are
 //! allowed to use any user and resource identifiers in their stanzas.
-use futures::{sink::SinkExt, task::Poll, Sink, Stream};
+use futures::sink::SinkExt;
 use minidom::Element;
-use std::pin::Pin;
 use std::str::FromStr;
-use std::task::Context;
 use xmpp_parsers::{jid::Jid, ns};
 
-use self::connect::component_login;
-
-use crate::connect::ServerConnector;
-use crate::proto::{add_stanza_id, Packet, XmppStream};
-use crate::Error;
+use crate::{
+    component::login::component_login,
+    connect::ServerConnector,
+    proto::{add_stanza_id, XmppStream},
+    Error,
+};
 
 #[cfg(any(feature = "starttls", feature = "insecure-tcp"))]
 use crate::connect::DnsConfig;
 #[cfg(feature = "insecure-tcp")]
 use crate::connect::TcpServerConnector;
 
-mod auth;
-
-pub(crate) mod connect;
+mod login;
+mod stream;
 
 /// Component connection to an XMPP server
 ///
@@ -33,20 +31,34 @@ pub struct Component<C: ServerConnector> {
     stream: XmppStream<C::Stream>,
 }
 
+impl<C: ServerConnector> Component<C> {
+    /// Send stanza
+    pub async fn send_stanza(&mut self, stanza: Element) -> Result<(), Error> {
+        self.send(add_stanza_id(stanza, ns::COMPONENT_ACCEPT)).await
+    }
+
+    /// End connection
+    pub async fn send_end(&mut self) -> Result<(), Error> {
+        self.close().await
+    }
+}
+
 #[cfg(feature = "insecure-tcp")]
 impl Component<TcpServerConnector> {
     /// Start a new XMPP component over plaintext TCP to localhost:5347
+    #[cfg(feature = "insecure-tcp")]
     pub async fn new(jid: &str, password: &str) -> Result<Self, Error> {
         Self::new_plaintext(jid, password, DnsConfig::addr("127.0.0.1:5347")).await
     }
 
     /// Start a new XMPP component over plaintext TCP
+    #[cfg(feature = "insecure-tcp")]
     pub async fn new_plaintext(
         jid: &str,
         password: &str,
         dns_config: DnsConfig,
     ) -> Result<Self, Error> {
-        Self::new_with_connector(jid, password, TcpServerConnector::from(dns_config)).await
+        Component::new_with_connector(jid, password, TcpServerConnector::from(dns_config)).await
     }
 }
 
@@ -65,65 +77,4 @@ impl<C: ServerConnector> Component<C> {
         let stream = component_login(connector, jid.clone(), password).await?;
         Ok(Component { jid, stream })
     }
-
-    /// Send stanza
-    pub async fn send_stanza(&mut self, stanza: Element) -> Result<(), Error> {
-        self.send(add_stanza_id(stanza, ns::COMPONENT_ACCEPT)).await
-    }
-
-    /// End connection
-    pub async fn send_end(&mut self) -> Result<(), Error> {
-        self.close().await
-    }
-}
-
-impl<C: ServerConnector> Stream for Component<C> {
-    type Item = Element;
-
-    fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Option<Self::Item>> {
-        loop {
-            match Pin::new(&mut self.stream).poll_next(cx) {
-                Poll::Ready(Some(Ok(Packet::Stanza(stanza)))) => return Poll::Ready(Some(stanza)),
-                Poll::Ready(Some(Ok(Packet::Text(_)))) => {
-                    // retry
-                }
-                Poll::Ready(Some(Ok(_))) =>
-                // unexpected
-                {
-                    return Poll::Ready(None)
-                }
-                Poll::Ready(Some(Err(_))) => return Poll::Ready(None),
-                Poll::Ready(None) => return Poll::Ready(None),
-                Poll::Pending => return Poll::Pending,
-            }
-        }
-    }
-}
-
-impl<C: ServerConnector> Sink<Element> for Component<C> {
-    type Error = Error;
-
-    fn start_send(mut self: Pin<&mut Self>, item: Element) -> Result<(), Self::Error> {
-        Pin::new(&mut self.stream)
-            .start_send(Packet::Stanza(item))
-            .map_err(|e| e.into())
-    }
-
-    fn poll_ready(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Result<(), Self::Error>> {
-        Pin::new(&mut self.stream)
-            .poll_ready(cx)
-            .map_err(|e| e.into())
-    }
-
-    fn poll_flush(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Result<(), Self::Error>> {
-        Pin::new(&mut self.stream)
-            .poll_flush(cx)
-            .map_err(|e| e.into())
-    }
-
-    fn poll_close(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Result<(), Self::Error>> {
-        Pin::new(&mut self.stream)
-            .poll_close(cx)
-            .map_err(|e| e.into())
-    }
 }

tokio-xmpp/src/component/stream.rs

@@ -0,0 +1,60 @@
+//! Components in XMPP are services/gateways that are logged into an
+//! XMPP server under a JID consisting of just a domain name. They are
+//! allowed to use any user and resource identifiers in their stanzas.
+use futures::{task::Poll, Sink, Stream};
+use minidom::Element;
+use std::pin::Pin;
+use std::task::Context;
+
+use crate::{component::Component, connect::ServerConnector, proto::Packet, Error};
+
+impl<C: ServerConnector> Stream for Component<C> {
+    type Item = Element;
+
+    fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Option<Self::Item>> {
+        loop {
+            match Pin::new(&mut self.stream).poll_next(cx) {
+                Poll::Ready(Some(Ok(Packet::Stanza(stanza)))) => return Poll::Ready(Some(stanza)),
+                Poll::Ready(Some(Ok(Packet::Text(_)))) => {
+                    // retry
+                }
+                Poll::Ready(Some(Ok(_))) =>
+                // unexpected
+                {
+                    return Poll::Ready(None)
+                }
+                Poll::Ready(Some(Err(_))) => return Poll::Ready(None),
+                Poll::Ready(None) => return Poll::Ready(None),
+                Poll::Pending => return Poll::Pending,
+            }
+        }
+    }
+}
+
+impl<C: ServerConnector> Sink<Element> for Component<C> {
+    type Error = Error;
+
+    fn start_send(mut self: Pin<&mut Self>, item: Element) -> Result<(), Self::Error> {
+        Pin::new(&mut self.stream)
+            .start_send(Packet::Stanza(item))
+            .map_err(|e| e.into())
+    }
+
+    fn poll_ready(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Result<(), Self::Error>> {
+        Pin::new(&mut self.stream)
+            .poll_ready(cx)
+            .map_err(|e| e.into())
+    }
+
+    fn poll_flush(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Result<(), Self::Error>> {
+        Pin::new(&mut self.stream)
+            .poll_flush(cx)
+            .map_err(|e| e.into())
+    }
+
+    fn poll_close(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Result<(), Self::Error>> {
+        Pin::new(&mut self.stream)
+            .poll_close(cx)
+            .map_err(|e| e.into())
+    }
+}