diff options
| author | Fuwn <[email protected]> | 2024-03-24 14:24:20 +0000 |
|---|---|---|
| committer | Fuwn <[email protected]> | 2024-03-24 14:34:09 +0000 |
| commit | 3970616451ab4336e37a7ddfdc67f51e01e55926 (patch) | |
| tree | 6bf9fa01811004ff408ac5da40354bc34ddabc26 /src | |
| parent | feat(crate): bump version (diff) | |
| download | germ-3970616451ab4336e37a7ddfdc67f51e01e55926.tar.xz germ-3970616451ab4336e37a7ddfdc67f51e01e55926.zip | |
refactor: module for blocking, default to non-blocking
Diffstat (limited to 'src')
| -rw-r--r-- | src/request.rs | 52 | ||||
| -rw-r--r-- | src/request/blocking.rs | 63 | ||||
| -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 |