feat(router): add translation engine

4 files changed, 156 insertions, 66 deletions

diff --git a/src/modules/router.rs b/src/modules/router.rs
index ade7d33..8f24a9b 100644
--- a/src/modules/router.rs
+++ b/src/modules/router.rs
@@ -16,70 +16,7 @@
 // Copyright (C) 2022-2022 Fuwn <[email protected]>
 // SPDX-License-Identifier: GPL-3.0-only
 
-use crate::DATABASE;
+mod deepl;
+mod module;
 
-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(),
-    );
-
-    // Just a simple demo for now.
-    if let Some(uwu) =
-      windmark::utilities::queries_from_url(context.url).get("uwu")
-    {
-      if uwu.to_lowercase() == "true" || uwu == "1" {
-        let mut lines = content
-          .lines()
-          .map(ToString::to_string)
-          .collect::<Vec<String>>();
-
-        for line in &mut lines {
-          if !line.starts_with("=>") {
-            *line = line
-              .replace('l', "w")
-              .replace('r', "w")
-              .replace('L', "W")
-              .replace('R', "W")
-              .replace('!', "! >w<");
-          }
-        }
-
-        *content = lines.join("\n");
-      }
-    }
-  }));
-}
+pub use module::module;
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");
+      }
+    }
+  }));
+}
diff --git a/src/modules/static.rs b/src/modules/static.rs
index b829b44..7197ac2 100644
--- a/src/modules/static.rs
+++ b/src/modules/static.rs
@@ -60,6 +60,11 @@ pub fn module(router: &mut windmark::Router) {
       "/stocks/telegram",
       "Stock Market Related Telegram Groups",
       "stocks_telegram"
+    ),
+    (
+      "/translate",
+      "Translate a page to your language of choice!",
+      "translate"
     )
   );
 }