use actix_web::http::header::ContentType;
use actix_web::Result;
use actix_web::{
  web::{self, Data},
  Scope,
};
use bytes::Bytes;
use shared_entity::data::{AppResponse, JsonAppResponse};
use shared_entity::error_code::ErrorCode;

use crate::biz::file_storage;
use crate::{component::auth::jwt::UserUuid, state::AppState};

pub fn file_storage_scope() -> Scope {
  web::scope("/api/file_storage").service(
    web::resource("/{path}")
      .route(web::get().to(get_handler))
      .route(web::put().to(put_handler))
      .route(web::delete().to(delete_handler)),
  )
}

async fn put_handler(
  user_uuid: UserUuid,
  state: Data<AppState>,
  path: web::Path<String>,
  file_data: Bytes,
  content_type: web::Header<ContentType>,
) -> Result<JsonAppResponse<()>> {
  let file_path = path.into_inner();
  let mime = content_type.into_inner().0;
  file_storage::put_object(
    &state.pg_pool,
    &state.s3_bucket,
    &user_uuid,
    &file_path,
    &file_data,
    mime,
  )
  .await?;
  Ok(AppResponse::Ok().into())
}

async fn delete_handler(
  user_uuid: UserUuid,
  state: Data<AppState>,
  path: web::Path<String>,
) -> Result<JsonAppResponse<()>> {
  let file_path = path.into_inner();
  file_storage::delete_object(&state.pg_pool, &state.s3_bucket, &user_uuid, &file_path).await?;
  Ok(AppResponse::Ok().into())
}

async fn get_handler(
  user_uuid: UserUuid,
  state: Data<AppState>,
  path: web::Path<String>,
) -> Result<Bytes> {
  let file_path = path.into_inner();
  match file_storage::get_object(&state.pg_pool, &state.s3_bucket, &user_uuid, &file_path).await {
    Ok(data) => Ok(data),
    Err(e) => match e.code {
      ErrorCode::FileNotFound => Err(actix_web::error::ErrorNotFound(e)),
      _ => Err(actix_web::error::ErrorInternalServerError(e)),
    },
  }
}