@mmmbuto/masix 0.4.1 → 0.4.3

package/packages/plugin-base/discovery/source/src/transport.rs

@@ -0,0 +1,112 @@
+use anyhow::Result;
+use reqwest::{Client, Proxy};
+use std::time::Duration;
+use tokio::{net::TcpStream, time::timeout};
+
+const TOR_DETECT_TIMEOUT_SECS: u64 = 2;
+const TOR_SOCKS_PORTS: &[u16] = &[9050, 9150];
+
+#[derive(Debug, Clone)]
+pub struct TransportLayer {
+    clearnet: Client,
+    tor: Option<Client>,
+    tor_available: bool,
+    tor_port: Option<u16>,
+}
+
+impl TransportLayer {
+    pub async fn new(
+        user_agent: &str,
+        clearnet_timeout_secs: u64,
+        tor_timeout_secs: u64,
+    ) -> Result<Self> {
+        let clearnet = Client::builder()
+            .user_agent(user_agent)
+            .timeout(Duration::from_secs(clearnet_timeout_secs))
+            .build()?;
+
+        let mut tor_port = std::env::var("MASIX_TOR_SOCKS_PORT")
+            .ok()
+            .and_then(|value| value.trim().parse::<u16>().ok());
+        let candidates = if let Some(port) = tor_port {
+            vec![port]
+        } else {
+            TOR_SOCKS_PORTS.to_vec()
+        };
+
+        for port in candidates {
+            if detect_tor_port(port).await {
+                tor_port = Some(port);
+                break;
+            }
+        }
+
+        let tor = if let Some(port) = tor_port {
+            Some(
+                Client::builder()
+                    .user_agent(user_agent)
+                    .timeout(Duration::from_secs(tor_timeout_secs))
+                    .proxy(Proxy::all(format!("socks5h://127.0.0.1:{port}"))?)
+                    .build()?,
+            )
+        } else {
+            None
+        };
+
+        Ok(Self {
+            clearnet,
+            tor_available: tor.is_some(),
+            tor_port,
+            tor,
+        })
+    }
+
+    pub fn client_for(&self, url: &str) -> (&Client, &'static str) {
+        if url.contains(".onion") {
+            if let Some(tor) = self.tor.as_ref() {
+                return (tor, "tor");
+            }
+        }
+        (&self.clearnet, "clearnet")
+    }
+
+    pub fn client_for_preference(&self, prefer_tor: bool) -> (&Client, &'static str) {
+        if prefer_tor {
+            if let Some(tor) = self.tor.as_ref() {
+                return (tor, "tor");
+            }
+        }
+        (&self.clearnet, "clearnet")
+    }
+
+    pub fn client_for_ahmia(&self, prefer_tor: bool) -> (&Client, &'static str, &'static str) {
+        if prefer_tor {
+            if let Some(tor) = self.tor.as_ref() {
+                return (
+                    tor,
+                    "tor",
+                    "http://juhanurmihxlp77nkq76byazcldy2hlmovfu2epvl5ankdibsot4csyd.onion/search/",
+                );
+            }
+        }
+        (&self.clearnet, "clearnet", "https://ahmia.fi/search/")
+    }
+
+    pub fn tor_available(&self) -> bool {
+        self.tor_available
+    }
+
+    pub fn tor_port(&self) -> Option<u16> {
+        self.tor_port
+    }
+}
+
+async fn detect_tor_port(port: u16) -> bool {
+    timeout(
+        Duration::from_secs(TOR_DETECT_TIMEOUT_SECS),
+        TcpStream::connect(("127.0.0.1", port)),
+    )
+    .await
+    .map(|result| result.is_ok())
+    .unwrap_or(false)
+}