feat: add support for NoChatReports encryption

2025-03-04 20:23:41 -05:00
parent 426c19304d
commit c4454fe217
9 changed files with 341 additions and 25 deletions
--- a/src/lua/nochatreports/crypt.rs
+++ b/src/lua/nochatreports/crypt.rs
@@ -0,0 +1,41 @@
+macro_rules! crypt_with {
+    ($op:ident, $encoding:expr, $key:expr, $text:expr, $algo:ident) => {
+        match $encoding {
+            1 => $algo::<Base64Encoding>::$op($text, $key),
+            2 => $algo::<Base64rEncoding>::$op($text, $key),
+            _ => $algo::<NewBase64rEncoding>::$op($text, $key),
+        }
+        .map_err(|error| Error::external(error.to_string()))?
+    };
+}
+
+#[macro_export]
+macro_rules! crypt {
+    ($op:ident, $encoding:expr, $options:expr, $text:expr) => {
+        match $options.get("encryption").unwrap_or_default() {
+            1 => CaesarEncryption::$op(&$text, &$options.get("key")?)
+                .map_err(|error| Error::external(error.to_string()))?,
+            2 => crypt_with!(
+                $op,
+                $encoding,
+                &$options.get::<UserDataRef<AesKey>>("key")?.inner,
+                &$text,
+                EcbEncryption
+            ),
+            3 => crypt_with!(
+                $op,
+                $encoding,
+                &$options.get::<UserDataRef<AesKey>>("key")?.inner,
+                &$text,
+                GcmEncryption
+            ),
+            _ => crypt_with!(
+                $op,
+                $encoding,
+                &$options.get::<UserDataRef<AesKey>>("key")?.inner,
+                &$text,
+                Cfb8Encryption
+            ),
+        }
+    };
+}