sculptor/src/api/v1/avatars.rs

use std::env::var;

use axum::{body::Bytes, extract::{Path, State}};
use tokio::{fs, io::{self, BufWriter}};
use tracing::warn;
use uuid::Uuid;

use crate::{api::figura::profile::send_event, auth::Token, ApiResult, AppState, AVATARS_ENV};

pub async fn upload_avatar(
    Path(uuid): Path<Uuid>,
    Token(token): Token,
    State(state): State<AppState>,
    body: Bytes,
) -> ApiResult<&'static str> {
    let request_data = body;

    state.config.read().await.clone().verify_token(&token)?;

    tracing::info!(
        "trying to upload the avatar for {}",
        uuid,
    );

    let avatar_file = format!("{}/{}.moon", var(AVATARS_ENV).unwrap(), &uuid);
    let mut file = BufWriter::new(fs::File::create(&avatar_file).await.unwrap());
    io::copy(&mut request_data.as_ref(), &mut file).await.unwrap();
    send_event(&state, &uuid).await;

    Ok("ok")
}

pub async fn delete_avatar(
    Path(uuid): Path<Uuid>,
    Token(token): Token,
    State(state): State<AppState>
) -> ApiResult<&'static str> {
    state.config.read().await.clone().verify_token(&token)?;

    tracing::info!(
        "trying to delete the avatar for {}",
        uuid,
    );

    let avatar_file = format!("{}/{}.moon", var(AVATARS_ENV).unwrap(), &uuid);
    match fs::remove_file(avatar_file).await {
        Ok(_) => {},
        Err(_) => {
            warn!("avatar doesn't exist");
            return Err(crate::ApiError::NotFound)
        }
    };
    send_event(&state, &uuid).await;

    Ok("ok")
}