From 1019ea1ed341e3a7769c046aa0be5764789360b6 Mon Sep 17 00:00:00 2001
From: Joris
Date: Sun, 2 Jun 2024 14:38:13 +0200
Subject: Migrate to Rust and Hyper

With sanic, downloading a file locally is around ten times slower than
with Rust and hyper.

Maybe `pypy` could have helped, but I didn’t succeed to set it up
quickly with the dependencies.
---
 src/model.rs | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 src/model.rs

(limited to 'src/model.rs')

diff --git a/src/model.rs b/src/model.rs
new file mode 100644
index 0000000..ed4fbf8
--- /dev/null
+++ b/src/model.rs
@@ -0,0 +1,52 @@
+use base64::{engine::general_purpose::URL_SAFE, Engine as _};
+use chrono::{DateTime, Local, NaiveDateTime, TimeZone};
+use rand_core::{OsRng, RngCore};
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct File {
+    pub id: String,
+    pub name: String,
+    pub expires_at: DateTime<Local>,
+    pub content_length: usize,
+}
+
+pub fn local_time() -> DateTime<Local> {
+    let dt = Local::now();
+    match decode_datetime(&encode_datetime(dt)) {
+        Some(res) => res,
+        None => dt,
+    }
+}
+
+// Using 20 bytes (160 bits) to file identifiers
+// https://owasp.org/www-community/vulnerabilities/Insufficient_Session-ID_Length
+// https://www.rfc-editor.org/rfc/rfc6749.html#section-10.10
+const FILE_ID_BYTES: usize = 20;
+
+pub fn generate_file_id() -> String {
+    let mut token = [0u8; FILE_ID_BYTES];
+    OsRng.fill_bytes(&mut token);
+    URL_SAFE.encode(token)
+}
+
+const FORMAT: &str = "%Y-%m-%d %H:%M:%S";
+
+pub fn encode_datetime(dt: DateTime<Local>) -> String {
+    dt.naive_utc().format(FORMAT).to_string()
+}
+
+pub fn decode_datetime(str: &str) -> Option<DateTime<Local>> {
+    let naive_time = NaiveDateTime::parse_from_str(str, FORMAT).ok()?;
+    Some(Local.from_utc_datetime(&naive_time))
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_datetime_serialization() {
+        let dt = local_time();
+        assert_eq!(decode_datetime(&encode_datetime(dt)), Some(dt))
+    }
+}
-- 
cgit v1.2.3