diff options
Diffstat (limited to 'src/modules')
| -rw-r--r-- | src/modules/router.rs | 69 | ||||
| -rw-r--r-- | src/modules/router/deepl.rs | 50 | ||||
| -rw-r--r-- | src/modules/router/module.rs | 98 | ||||
| -rw-r--r-- | src/modules/static.rs | 5 |
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" ) ); } |