From f9974e79a7528940a95b04d45cb01e9670215db9 Mon Sep 17 00:00:00 2001
From: Savanni D'Gerinel <savanni@luminescent-dreams.com>
Date: Thu, 5 Oct 2023 00:08:27 -0400
Subject: [PATCH] Set a maximum upload to 15MB

---
 file-service/src/main.rs | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/file-service/src/main.rs b/file-service/src/main.rs
index 0489779..ffa4ad7 100644
--- a/file-service/src/main.rs
+++ b/file-service/src/main.rs
@@ -1,28 +1,24 @@
 #[macro_use]
 extern crate log;
 
-use handlers::{file, handle_auth, handle_upload, thumbnail};
-use http::status::StatusCode;
-// use mustache::{compile_path, Template};
-// use orizentic::{Permissions, ResourceName, Secret};
-use bytes::Buf;
 use cookie::Cookie;
-use futures_util::StreamExt;
+use handlers::{file, handle_auth, handle_upload, thumbnail};
 use std::{
     collections::{HashMap, HashSet},
     convert::Infallible,
-    io::Read,
     net::{IpAddr, Ipv4Addr, SocketAddr},
     path::PathBuf,
     sync::Arc,
 };
 use tokio::sync::RwLock;
-use warp::{filters::multipart::Part, Filter, Rejection};
+use warp::{Filter, Rejection};
 
 mod handlers;
 mod html;
 mod pages;
 
+const MAX_UPLOAD: u64 = 15 * 1024 * 1024;
+
 pub use file_service::{
     AuthDB, AuthError, AuthToken, FileHandle, FileId, FileInfo, ReadFileError, SessionToken, Store,
     Username, WriteFileError,
@@ -137,7 +133,7 @@ pub async fn main() {
         .and(warp::post())
         .and(with_app(app.clone()))
         .and(with_session())
-        .and(warp::multipart::form())
+        .and(warp::multipart::form().max_length(MAX_UPLOAD))
         .then(handle_upload);
 
     let thumbnail = warp::path!(String / "tn")