serxoz/share
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

error si non se encontra o arquivo

Browse Source
This commit is contained in:
serxoz 2022-10-21 16:18:57 +02:00
parent 1f90332fc1
commit 48b499c349
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/vistas/download.rs
View File

@ -9,8 +9,11 @@ use tokio_util::io::ReaderStream;
const BASE_PATH: &str = "uploads"; const BASE_PATH: &str = "uploads";
pub async fn get_file(axum::extract::Path(hash): axum::extract::Path<String>) -> impl IntoResponse { pub async fn get_file(axum::extract::Path(hash): axum::extract::Path<String>) -> impl IntoResponse {
// hash.into_response() // find file from hash
let (fpath, fname) = find_file(hash).await; let (fpath, fname) = match find_file(hash).await {
Ok(file) => file,
Err(err) => return Err((StatusCode::NOT_FOUND, format!("File not found: {}", err))),
};
// `File` implements `AsyncRead` // `File` implements `AsyncRead`
let file = match tokio::fs::File::open(fpath).await { let file = match tokio::fs::File::open(fpath).await {
@ -34,16 +37,16 @@ pub async fn get_file(axum::extract::Path(hash): axum::extract::Path<String>) ->
} }
// uploads/O/OfsdaDF/proba.txt // uploads/O/OfsdaDF/proba.txt
async fn find_file(hash: String) -> (String, String) { async fn find_file(hash: String) -> Result<(String, String), tokio::io::Error> {
let mut file_path = String::from(""); let mut file_path = String::from("");
let mut file_name = String::from(""); let mut file_name = String::from("");
let pai = &hash[0..1]; //primeiro char let pai = &hash[0..1]; //primeiro char
let path = format!("{}/{}/{}", BASE_PATH, pai, hash); let path = format!("{}/{}/{}", BASE_PATH, pai, hash);
let mut dir = read_dir(path).await.expect("error atopando directorio"); let mut dir = read_dir(path).await?;
while let Some(child) = dir.next_entry().await.expect("error listando") { while let Some(child) = dir.next_entry().await? {
file_path = child.path().display().to_string(); file_path = child.path().display().to_string();
file_name = child.file_name().into_string().unwrap(); file_name = child.file_name().into_string().unwrap();
} }
(file_path, file_name) Ok((file_path, file_name))
} }