handlers-refactored

src/handlers/upload.rs

@@ -0,0 +1,78 @@
+use actix_multipart::Multipart;
+use actix_web::{web, HttpRequest, HttpResponse, Result};
+
+use crate::app_state::AppState;
+use crate::auth::{get_id_by_token, user_added_file};
+use crate::s3_utils::{generate_key_with_extension, upload_to_s3};
+use futures::TryStreamExt;
+use crate::handlers::MAX_WEEK_BYTES;
+
+/// Обработчик для аплоада файлов.
+pub async fn upload_handler(
+    req: HttpRequest,
+    mut payload: Multipart,
+    state: web::Data<AppState>,
+) -> Result<HttpResponse, actix_web::Error> {
+    // Получаем токен из заголовка авторизации
+    let token = req
+        .headers()
+        .get("Authorization")
+        .and_then(|header_value| header_value.to_str().ok());
+    if token.is_none() {
+        return Err(actix_web::error::ErrorUnauthorized("Unauthorized")); // Если токен отсутствует, возвращаем ошибку
+    }
+
+    let user_id = get_id_by_token(token.unwrap()).await?;
+
+    // Получаем текущую квоту пользователя
+    let this_week_amount: u64 = state.get_or_create_quota(&user_id).await.unwrap_or(0);
+
+    while let Ok(Some(field)) = payload.try_next().await {
+        let mut field = field;
+
+        let content_type = field.content_type().unwrap().to_string();
+        let file_name = field
+            .content_disposition()
+            .unwrap()
+            .get_filename()
+            .map(|f| f.to_string());
+
+        if let Some(name) = file_name {
+            let mut file_bytes = Vec::new();
+            let mut file_size: u64 = 0;
+
+            // Читаем данные файла
+            while let Ok(Some(chunk)) = field.try_next().await {
+                file_size += chunk.len() as u64;
+                file_bytes.extend_from_slice(&chunk);
+            }
+
+            // Проверяем, что добавление файла не превышает лимит квоты
+            if this_week_amount + file_size > MAX_WEEK_BYTES {
+                return Err(actix_web::error::ErrorUnauthorized("Quota exceeded"));
+                // Квота превышена
+            }
+
+            // инкрементируем квоту пользователя
+            let _ = state.increment_uploaded_bytes(&user_id, file_size).await?;
+
+            // Определяем правильное расширение и ключ для S3
+            let file_key = generate_key_with_extension(name, content_type.to_owned());
+
+            // Загружаем файл в S3
+            upload_to_s3(
+                &state.s3_client,
+                &state.s3_bucket,
+                &file_key,
+                file_bytes,
+                &content_type,
+            )
+            .await?;
+
+            // Сохраняем информацию о загруженном файле для пользователя
+            user_added_file(&mut state.redis.clone(), &user_id, &file_key).await?;
+        }
+    }
+
+    Ok(HttpResponse::Ok().json("File uploaded successfully"))
+}