feat(request): make gemini requests

8 files changed, 334 insertions, 0 deletions

diff --git a/Cargo.toml b/Cargo.toml
index 0ec440c..b3b73f1 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -17,3 +17,9 @@ categories = ["encoding"]
 [features]
 ast = []
 convert = []
+request = ["rustls", "url", "anyhow"]
+
+[dependencies]
+rustls = { version = "0.20.5", features = ["dangerous_configuration"], optional = true } # TLS
+url = { version = "2.2.2", optional = true } # URL Validation
+anyhow = { version = "1.0.57", optional = true } # `Result`
diff --git a/examples/request.rs b/examples/request.rs
new file mode 100644
index 0000000..5791896
--- /dev/null
+++ b/examples/request.rs
@@ -0,0 +1,24 @@
+// 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
+
+fn main() {
+  match germ::request::request(url::Url::parse("gemini://fuwn.me").unwrap()) {
+    Ok(response) => println!("{:?}", response),
+    Err(_) => {}
+  }
+}
diff --git a/src/lib.rs b/src/lib.rs
index 8b47af8..0b04370 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -34,3 +34,6 @@ pub mod ast;
 
 #[cfg(feature = "convert")]
 pub mod convert;
+
+#[cfg(feature = "request")]
+pub mod request;
diff --git a/src/request.rs b/src/request.rs
new file mode 100644
index 0000000..f6297ef
--- /dev/null
+++ b/src/request.rs
@@ -0,0 +1,68 @@
+// 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
+
+//! Make Gemini requests and get sane, structured results
+
+mod response;
+mod status;
+mod verifier;
+
+use std::io::{Read, Write};
+
+pub use response::Response;
+pub use status::Status;
+use verifier::GermVerifier;
+
+/// 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(_) => {}
+/// }
+/// ```
+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!("{}\r\n", url).as_bytes())?;
+
+  let mut plain_text = Vec::new();
+
+  tls.read_to_end(&mut plain_text).unwrap();
+
+  Ok(Response::new(
+    plain_text,
+    tls.conn.negotiated_cipher_suite(),
+  ))
+}
diff --git a/src/request/response.rs b/src/request/response.rs
new file mode 100644
index 0000000..0202967
--- /dev/null
+++ b/src/request/response.rs
@@ -0,0 +1,59 @@
+// 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 rustls::SupportedCipherSuite;
+
+use crate::request::Status;
+
+#[derive(Debug)]
+pub struct Response {
+  pub status:  Status,
+  pub meta:    String,
+  pub content: Option<String>,
+  pub size:    usize,
+  pub suite:   Option<SupportedCipherSuite>,
+}
+impl Response {
+  pub(super) fn new(
+    data: Vec<u8>,
+    suite: Option<SupportedCipherSuite>,
+  ) -> Self {
+    let string_form = String::from_utf8_lossy(&data).to_string();
+    let mut content = None;
+    let header;
+
+    if !string_form.ends_with("\r\n") {
+      let mut string_split = string_form.split("\r\n");
+
+      header = string_split.next().unwrap_or("").to_string();
+      content = Some(string_split.collect());
+    } else {
+      header = string_form;
+    }
+
+    let header_split = header.split_at(2);
+
+    Self {
+      status: Status::from(header_split.0.parse::<i32>().unwrap_or(0)),
+      meta: header_split.1.trim_start().to_string(),
+      content,
+      size: data.len(),
+      suite,
+    }
+  }
+}
diff --git a/src/request/status.rs b/src/request/status.rs
new file mode 100644
index 0000000..b2065ed
--- /dev/null
+++ b/src/request/status.rs
@@ -0,0 +1,103 @@
+// 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
+
+/// Simple Gemini status reporting
+///
+/// # Examples
+///
+/// ```rust
+/// use germ::request::Status;
+///
+/// assert_eq!(Status::from(10), Status::Input);
+/// assert_eq!(i32::from(Status::Input), 10);
+/// ```
+#[derive(Debug, PartialEq)]
+pub enum Status {
+  Input,
+  SensitiveInput,
+  Success,
+  TemporaryRedirect,
+  PermanentRedirect,
+  TemporaryFailure,
+  ServerUnavailable,
+  CGIError,
+  ProxyError,
+  SlowDown,
+  PermanentFailure,
+  NotFound,
+  Gone,
+  ProxyRefused,
+  BadRequest,
+  ClientCertificateRequired,
+  CertificateNotAuthorised,
+  CertificateNotValid,
+  Unsupported,
+}
+impl Default for Status {
+  fn default() -> Self { Self::Success }
+}
+impl From<Status> for i32 {
+  fn from(n: Status) -> Self {
+    match n {
+      Status::Input => 10,
+      Status::SensitiveInput => 11,
+      Status::Success => 20,
+      Status::TemporaryRedirect => 30,
+      Status::PermanentRedirect => 31,
+      Status::TemporaryFailure => 40,
+      Status::ServerUnavailable => 41,
+      Status::CGIError => 42,
+      Status::ProxyError => 43,
+      Status::SlowDown => 44,
+      Status::PermanentFailure => 50,
+      Status::NotFound => 51,
+      Status::Gone => 52,
+      Status::ProxyRefused => 53,
+      Status::BadRequest => 59,
+      Status::ClientCertificateRequired => 60,
+      Status::CertificateNotAuthorised => 61,
+      Status::CertificateNotValid => 62,
+      _ => 0,
+    }
+  }
+}
+impl From<i32> for Status {
+  fn from(n: i32) -> Self {
+    match n {
+      10 => Self::Input,
+      11 => Self::SensitiveInput,
+      20 => Self::Success,
+      30 => Self::TemporaryRedirect,
+      31 => Self::PermanentRedirect,
+      40 => Self::TemporaryFailure,
+      41 => Self::ServerUnavailable,
+      42 => Self::CGIError,
+      43 => Self::ProxyError,
+      44 => Self::SlowDown,
+      50 => Self::PermanentFailure,
+      51 => Self::NotFound,
+      52 => Self::Gone,
+      53 => Self::ProxyRefused,
+      59 => Self::BadRequest,
+      60 => Self::ClientCertificateRequired,
+      61 => Self::CertificateNotAuthorised,
+      62 => Self::CertificateNotValid,
+      _ => Self::Unsupported,
+    }
+  }
+}
diff --git a/src/request/verifier.rs b/src/request/verifier.rs
new file mode 100644
index 0000000..f2c6a35
--- /dev/null
+++ b/src/request/verifier.rs
@@ -0,0 +1,39 @@
+// 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 std::time::SystemTime;
+
+use rustls::{client, client::ServerCertVerified, Certificate};
+
+pub(super) struct GermVerifier;
+impl GermVerifier {
+  pub fn new() -> Self { Self {} }
+}
+impl client::ServerCertVerifier for GermVerifier {
+  fn verify_server_cert(
+    &self,
+    _end_entity: &Certificate,
+    _intermediates: &[Certificate],
+    _server_name: &client::ServerName,
+    _scts: &mut dyn Iterator<Item = &[u8]>,
+    _ocsp_response: &[u8],
+    _now: SystemTime,
+  ) -> Result<ServerCertVerified, rustls::Error> {
+    Ok(ServerCertVerified::assertion())
+  }
+}
diff --git a/tests/status.rs b/tests/status.rs
new file mode 100644
index 0000000..51f3f66
--- /dev/null
+++ b/tests/status.rs
@@ -0,0 +1,32 @@
+// 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
+
+#[cfg(test)]
+mod test {
+  use germ::request::Status;
+
+  #[test]
+  fn status_from_i32() {
+    assert_eq!(Status::from(10), Status::Input);
+  }
+
+  #[test]
+  fn i32_from_status() {
+    assert_eq!(i32::from(Status::Input), 10);
+  }
+}