1 files changed, 150 insertions, 0 deletions
diff --git a/src/gemini_to_html.rs b/src/gemini_to_html.rs
new file mode 100644
index 0000000..b6dca1e
--- /dev/null
+++ b/src/gemini_to_html.rs
@@ -0,0 +1,150 @@
+// This file is part of September <https://github.com/gemrest/september>.
+// 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::env::var;
+
+use germ::ast::Node;
+use gmi::url::Url;
+
+fn link_from_host_href(url: &Url, href: &str) -> String {
+  format!(
+    "gemini://{}{}{}",
+    url.authority.host,
+    {
+      if href.starts_with('/') {
+        ""
+      } else {
+        "/"
+      }
+    },
+    href
+  )
+}
+
+pub fn gemini_to_html(
+  response: &gmi::protocol::Response,
+  url: &Url,
+  is_proxy: bool,
+) -> (String, String) {
+  let ast = germ::ast::build(&String::from_utf8_lossy(&response.data));
+  let mut html = String::new();
+  let mut title = "".to_string();
+
+  for node in ast {
+    match node {
+      Node::Text(text) => html.push_str(&format!("<p>{}</p>", text)),
+      Node::Link {
+        to,
+        text,
+      } => {
+        let mut href = to.clone();
+
+        if href.starts_with('/') || !href.contains("://") {
+          href = link_from_host_href(url, &href);
+        }
+
+        if var("PROXY_BY_DEFAULT").unwrap_or_else(|_| "true".to_string())
+          == "true"
+          && to.contains("gemini://")
+        {
+          if is_proxy
+            || href
+              .trim_start_matches("gemini://")
+              .trim_end_matches('/')
+              .split('/')
+              .collect::<Vec<_>>()
+              .get(0)
+              .unwrap()
+              != &url.authority.host.as_str()
+          {
+            href = format!("/proxy/{}", href.trim_start_matches("gemini://"));
+          } else {
+            href = href
+              .trim_start_matches("gemini://")
+              .replace(&url.authority.host, "");
+          }
+        }
+
+        if let Ok(keeps) = var("KEEP_GEMINI_EXACT") {
+          let mut keeps = keeps.split(',');
+
+          if href.starts_with('/') || !href.contains("://") {
+            let temporary_href = link_from_host_href(url, &href);
+
+            if keeps.any(|k| k == &*temporary_href) {
+              href = temporary_href;
+            }
+          }
+        }
+
+        if let Ok(keeps) = var("KEEP_GEMINI_DOMAIN") {
+          if href.starts_with('/')
+            || !href.contains("://")
+              && keeps.split(',').any(|k| k == &*url.authority.host)
+          {
+            href = link_from_host_href(url, &href);
+          }
+        }
+
+        html.push_str(&format!(
+          "<p><a href=\"{}\">{}</a></p>\n",
+          href,
+          text.unwrap_or(to)
+        ));
+      }
+      Node::Heading {
+        level,
+        text,
+      } => {
+        if title.is_empty() && level == 1 {
+          title = text.clone();
+        }
+
+        html.push_str(&format!(
+          "<{}>{}</{0}>",
+          match level {
+            1 => "h1",
+            2 => "h2",
+            3 => "h3",
+            _ => "p",
+          },
+          text
+        ));
+      }
+      Node::List(items) =>
+        html.push_str(&format!(
+          "<ul>{}</ul>",
+          items
+            .into_iter()
+            .map(|i| format!("<li>{}</li>", i))
+            .collect::<Vec<String>>()
+            .join("\n")
+        )),
+      Node::Blockquote(text) =>
+        html.push_str(&format!("<blockquote>{}</blockquote>", text)),
+      Node::PreformattedText {
+        text, ..
+      } => {
+        html.push_str(&format!("<pre>{}</pre>", text));
+      }
+      Node::Whitespace => {}
+    }
+  }
+
+  (title, html)
+}