feat(router): add translation engine

2 files changed, 148 insertions, 0 deletions

diff --git a/src/modules/router/deepl.rs b/src/modules/router/deepl.rs
new file mode 100644
index 0000000..d4e210f
--- /dev/null
+++ b/src/modules/router/deepl.rs
@@ -0,0 +1,50 @@
+// This file is part of Locus <https://github.com/gemrest/locus>.
+// 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
+
+#[derive(serde::Deserialize)]
+pub struct DeepL {
+  translations: Vec<Translation>,
+}
+
+impl DeepL {
+  pub fn text(&self) -> String { self.translations[0].text.clone() }
+}
+
+#[derive(serde::Deserialize)]
+struct Translation {
+  pub text: String,
+}
+
+pub fn translate(text: &str, language: &str) -> DeepL {
+  let deepl_response: DeepL = serde_json::from_str(
+    &reqwest::blocking::Client::new()
+      .post("https://api-free.deepl.com/v2/translate")
+      .header(
+        "Authorization",
+        "DeepL-Auth-Key 182c8c21-730e-f99d-8f2e-61e9cc861faa:fx",
+      )
+      .form(&[("text", text), ("target_lang", language)])
+      .send()
+      .unwrap()
+      .text()
+      .unwrap(),
+  )
+  .unwrap();
+
+  deepl_response
+}
diff --git a/src/modules/router/module.rs b/src/modules/router/module.rs
new file mode 100644
index 0000000..a97c973
--- /dev/null
+++ b/src/modules/router/module.rs
@@ -0,0 +1,98 @@
+// This file is part of Locus <https://github.com/gemrest/locus>.
+// 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::{modules::router::deepl::translate, DATABASE};
+
+pub fn module(router: &mut windmark::Router) {
+  router.set_pre_route_callback(Box::new(|context| {
+    info!(
+      "accepted connection from {} to {}",
+      context.tcp.peer_addr().unwrap().ip(),
+      context.url.to_string(),
+    );
+
+    let url_path = if context.url.path().is_empty() {
+      "/"
+    } else {
+      context.url.path()
+    };
+
+    let previous_database = (*DATABASE.lock().unwrap()).get::<i32>(url_path);
+
+    match previous_database {
+      None => {
+        (*DATABASE.lock().unwrap())
+          .set::<i32>(url_path, &0)
+          .unwrap();
+      }
+      Some(_) => {}
+    }
+
+    let new_database = (*DATABASE.lock().unwrap()).get::<i32>(url_path);
+
+    (*DATABASE.lock().unwrap())
+      .set(url_path, &(new_database.unwrap() + 1))
+      .unwrap();
+  }));
+  router.set_post_route_callback(Box::new({
+    |context, content| {
+      info!(
+        "closed connection from {} to {}",
+        context.tcp.peer_addr().unwrap().ip(),
+        context.url.to_string(),
+      );
+
+      if let Some(language) =
+        windmark::utilities::queries_from_url(context.url).get("translate")
+      {
+        let mut lines = content
+          .lines()
+          .map(ToString::to_string)
+          .collect::<Vec<String>>();
+        let mut preformatted = true;
+
+        for line in &mut lines {
+          if line == "```" {
+            preformatted = !preformatted;
+          }
+
+          if line.starts_with("=>") {
+            if let Some(germ::ast::Node::Link {
+              to,
+              text,
+            }) = germ::ast::Ast::from_string(line).inner().get(0)
+            {
+              *line = format!(
+                "=> {to}?translate={language}{}",
+                text.clone().map_or_else(
+                  || "".to_string(),
+                  |text| { format!(" {}", translate(&text, language).text()) }
+                )
+              );
+            }
+          } else if !line.starts_with("=>") && preformatted && !line.is_empty()
+          {
+            *line = translate(line, language).text();
+          }
+        }
+
+        *content = lines.join("\n");
+      }
+    }
+  }));
+}