feat(router): add translation engine

1 files changed, 98 insertions, 0 deletions

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");
+      }
+    }
+  }));
+}