mirror of
https://github.com/MedzikUser/HomeDisk.git
synced 2024-08-14 21:46:53 +00:00
69 lines
2.1 KiB
Rust
69 lines
2.1 KiB
Rust
|
use std::fs;
|
||
|
|
||
|
use axum::{extract::rejection::JsonRejection, Extension, Json};
|
||
|
use axum_auth::AuthBearer;
|
||
|
use homedisk_database::{Database, Error};
|
||
|
use homedisk_types::{
|
||
|
config::types::Config,
|
||
|
errors::{AuthError, FsError, ServerError},
|
||
|
fs::list::{Request, Response},
|
||
|
};
|
||
|
|
||
|
use crate::middleware::{validate_json, validate_jwt};
|
||
|
|
||
|
pub async fn handle(
|
||
|
db: Extension<Database>,
|
||
|
config: Extension<Config>,
|
||
|
AuthBearer(token): AuthBearer,
|
||
|
request: Result<Json<Request>, JsonRejection>,
|
||
|
) -> Result<Json<Response>, ServerError> {
|
||
|
let Json(request) = validate_json::<Request>(request)?;
|
||
|
let token = validate_jwt(config.jwt.secret.as_bytes(), &token)?;
|
||
|
|
||
|
let response = match db.find_user_by_id(token.claims.sub).await {
|
||
|
Ok(res) => {
|
||
|
let user_path = format!(
|
||
|
"{path}/{username}/{request_path}",
|
||
|
path = config.storage.path,
|
||
|
username = res.username,
|
||
|
request_path = request.path,
|
||
|
);
|
||
|
|
||
|
let paths = fs::read_dir(&user_path)
|
||
|
.map_err(|err| ServerError::FsError(FsError::ReadDir(err.to_string())))?;
|
||
|
|
||
|
let mut files = vec![];
|
||
|
let mut dirs = vec![];
|
||
|
|
||
|
for path in paths {
|
||
|
let path = path
|
||
|
.map_err(|err| ServerError::FsError(FsError::UnknowError(err.to_string())))?;
|
||
|
let metadata = path
|
||
|
.metadata()
|
||
|
.map_err(|err| ServerError::FsError(FsError::UnknowError(err.to_string())))?;
|
||
|
|
||
|
let name = path.path().display().to_string().replace(&user_path, "");
|
||
|
|
||
|
if metadata.is_dir() {
|
||
|
dirs.push(name)
|
||
|
} else {
|
||
|
files.push(name)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Response { files, dirs }
|
||
|
}
|
||
|
|
||
|
Err(err) => match err {
|
||
|
Error::UserNotFound => return Err(ServerError::AuthError(AuthError::UserNotFound)),
|
||
|
_ => {
|
||
|
return Err(ServerError::AuthError(AuthError::UnknowError(
|
||
|
err.to_string(),
|
||
|
)))
|
||
|
}
|
||
|
},
|
||
|
};
|
||
|
|
||
|
Ok(Json(response))
|
||
|
}
|