aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorFuwn <[email protected]>2024-03-24 14:24:20 +0000
committerFuwn <[email protected]>2024-03-24 14:34:09 +0000
commit3970616451ab4336e37a7ddfdc67f51e01e55926 (patch)
tree6bf9fa01811004ff408ac5da40354bc34ddabc26 /src
parentfeat(crate): bump version (diff)
downloadgerm-3970616451ab4336e37a7ddfdc67f51e01e55926.tar.xz
germ-3970616451ab4336e37a7ddfdc67f51e01e55926.zip
refactor: module for blocking, default to non-blocking
Diffstat (limited to 'src')
-rw-r--r--src/request.rs52
-rw-r--r--src/request/blocking.rs63
-rw-r--r--src/request/non_blocking.rs (renamed from src/request/sync.rs)2
3 files changed, 72 insertions, 45 deletions
diff --git a/src/request.rs b/src/request.rs
index 97aa954..f1eb005 100644
--- a/src/request.rs
+++ b/src/request.rs
@@ -1,5 +1,5 @@
// This file is part of Germ <https://github.com/gemrest/germ>.
-// Copyright (C) 2022-2022 Fuwn <[email protected]>
+// Copyright (C) 2022-2023 Fuwn <[email protected]>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
@@ -22,50 +22,14 @@ mod response;
mod status;
mod verifier;
-#[cfg(feature = "sync")] pub mod sync;
+#[cfg(feature = "blocking")]
+pub mod blocking;
+
+#[cfg(feature = "request")]
+pub mod non_blocking;
-use std::io::{Read, Write};
+#[cfg(feature = "request")]
+pub use non_blocking::request;
pub(crate) use verifier::GermVerifier;
pub use {response::Response, status::Status};
-
-/// Make a request to a Gemini server. The `url` **should** be prefixed with a
-/// scheme (e.g. "gemini://").
-///
-/// # Example
-///
-/// ```rust
-/// match germ::request::request(&url::Url::parse("gemini://fuwn.me").unwrap()) {
-/// Ok(response) => println!("{:?}", response),
-/// Err(_) => {}
-/// }
-/// ```
-///
-/// # Errors
-/// - May error if the URL is invalid
-/// - May error if the TLS write fails
-/// - May error if the TLS read fails
-pub fn request(url: &url::Url) -> anyhow::Result<Response> {
- let config = rustls::ClientConfig::builder()
- .with_safe_defaults()
- .with_custom_certificate_verifier(std::sync::Arc::new(GermVerifier::new()))
- .with_no_client_auth();
- let mut connection = rustls::ClientConnection::new(
- std::sync::Arc::new(config),
- url.domain().unwrap_or("").try_into()?,
- )?;
- let mut stream = std::net::TcpStream::connect(format!(
- "{}:{}",
- url.domain().unwrap_or(""),
- url.port().unwrap_or(1965)
- ))?;
- let mut tls = rustls::Stream::new(&mut connection, &mut stream);
-
- tls.write_all(format!("{url}\r\n").as_bytes())?;
-
- let mut plain_text = Vec::new();
-
- tls.read_to_end(&mut plain_text)?;
-
- Ok(Response::new(&plain_text, tls.conn.negotiated_cipher_suite()))
-}
diff --git a/src/request/blocking.rs b/src/request/blocking.rs
new file mode 100644
index 0000000..0d78b52
--- /dev/null
+++ b/src/request/blocking.rs
@@ -0,0 +1,63 @@
+// This file is part of Germ <https://github.com/gemrest/germ>.
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 3.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+//
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+// SPDX-License-Identifier: GPL-3.0-only
+
+use {
+ crate::request::{GermVerifier, Response},
+ std::io::{Read, Write},
+};
+
+/// Make a request to a Gemini server. The `url` **should** be prefixed with a
+/// scheme (e.g. "gemini://").
+///
+/// # Example
+///
+/// ```rust
+/// match germ::request::request(&url::Url::parse("gemini://fuwn.me").unwrap()) {
+/// Ok(response) => println!("{:?}", response),
+/// Err(_) => {}
+/// }
+/// ```
+///
+/// # Errors
+/// - May error if the URL is invalid
+/// - May error if the TLS write fails
+/// - May error if the TLS read fails
+pub fn request(url: &url::Url) -> anyhow::Result<Response> {
+ let config = rustls::ClientConfig::builder()
+ .with_safe_defaults()
+ .with_custom_certificate_verifier(std::sync::Arc::new(GermVerifier::new()))
+ .with_no_client_auth();
+ let mut connection = rustls::ClientConnection::new(
+ std::sync::Arc::new(config),
+ url.domain().unwrap_or("").try_into()?,
+ )?;
+ let mut stream = std::net::TcpStream::connect(format!(
+ "{}:{}",
+ url.domain().unwrap_or(""),
+ url.port().unwrap_or(1965)
+ ))?;
+ let mut tls = rustls::Stream::new(&mut connection, &mut stream);
+
+ tls.write_all(format!("{url}\r\n").as_bytes())?;
+
+ let mut plain_text = Vec::new();
+
+ tls.read_to_end(&mut plain_text)?;
+
+ Ok(Response::new(&plain_text, tls.conn.negotiated_cipher_suite()))
+}
diff --git a/src/request/sync.rs b/src/request/non_blocking.rs
index 56950fb..c578703 100644
--- a/src/request/sync.rs
+++ b/src/request/non_blocking.rs
@@ -1,5 +1,5 @@
// This file is part of Germ <https://github.com/gemrest/germ>.
-// Copyright (C) 2022-2023 Fuwn <[email protected]>
+// Copyright (C) 2022-2024 Fuwn <[email protected]>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by