2022-04-24 20:07:41 +00:00
|
|
|
use std::fs;
|
|
|
|
|
|
|
|
use axum::{extract::rejection::JsonRejection, Extension, Json};
|
|
|
|
use axum_auth::AuthBearer;
|
2022-04-29 19:33:44 +00:00
|
|
|
use byte_unit::Byte;
|
2022-04-24 20:07:41 +00:00
|
|
|
use homedisk_database::{Database, Error};
|
|
|
|
use homedisk_types::{
|
|
|
|
config::types::Config,
|
|
|
|
errors::{AuthError, FsError, ServerError},
|
2022-04-29 19:33:44 +00:00
|
|
|
fs::list::{FileInfo, Request, Response},
|
2022-04-24 20:07:41 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
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, "");
|
2022-04-29 19:33:44 +00:00
|
|
|
let filesize = Byte::from_bytes(metadata.len().into()).get_appropriate_unit(true);
|
2022-04-24 20:07:41 +00:00
|
|
|
|
|
|
|
if metadata.is_dir() {
|
|
|
|
dirs.push(name)
|
|
|
|
} else {
|
2022-04-29 19:33:44 +00:00
|
|
|
files.push(FileInfo {
|
|
|
|
name,
|
|
|
|
size: filesize.to_string(),
|
|
|
|
})
|
2022-04-24 20:07:41 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
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))
|
|
|
|
}
|