make the get_all api method actually work
This commit is contained in:
parent
46fbbee459
commit
3cb1fd978f
|
@ -2,6 +2,7 @@
|
|||
use std::error::Error;
|
||||
|
||||
use redis::{Client, Commands, AsyncCommands, Connection};
|
||||
use tokio::task::JoinSet;
|
||||
|
||||
use crate::env::Env;
|
||||
|
||||
|
@ -25,23 +26,18 @@ impl FileManager {
|
|||
|
||||
async fn find_all(self: &Self, predicate: String) -> Result<Vec<File>, Box<dyn Error>> {
|
||||
let mut conn = self.conn.get_async_connection().await?;
|
||||
let found: Vec<String> = conn.keys(predicate).await?;
|
||||
let serialized: Vec<File> =
|
||||
found.iter()
|
||||
.map(|x| {
|
||||
let result = serde_json::from_str(&x);
|
||||
match result {
|
||||
Ok(x) => Some(x),
|
||||
Err(err) => {
|
||||
log::error!("Error while serializing {x}: {:?}", err);
|
||||
None
|
||||
let keys: Vec<String> = conn.keys(predicate).await?;
|
||||
|
||||
let mut data: Vec<File> = vec![];
|
||||
|
||||
for key in keys.iter() {
|
||||
let raw: String = conn.get(key.clone()).await?;
|
||||
|
||||
let mut parsed: File = serde_json::from_str(raw.as_str())?;
|
||||
data.push(parsed);
|
||||
}
|
||||
}
|
||||
})
|
||||
.filter(|x| x.is_some())
|
||||
.map(|x| x.unwrap())
|
||||
.collect();
|
||||
Ok(serialized)
|
||||
|
||||
Ok(data)
|
||||
}
|
||||
|
||||
/// Filter options
|
||||
|
|
|
@ -1,10 +1,13 @@
|
|||
use warp::{reply::{Reply, json}, reject::Rejection, Filter};
|
||||
use std::net::IpAddr;
|
||||
|
||||
use crate::web::{state::SharedState, rejection::HttpReject};
|
||||
use warp::{reply::{Reply, json, with_status}, reject::Rejection, Filter, http::StatusCode};
|
||||
use warp_real_ip::real_ip;
|
||||
|
||||
use crate::web::{state::SharedState, rejection::HttpReject, api::types::{ErrorMessage, Error}};
|
||||
|
||||
use super::{check_api_enabled, function_disabled_err};
|
||||
|
||||
pub async fn get_all(state: SharedState) -> Result<Box<dyn Reply>, Rejection> {
|
||||
pub async fn get_all(state: SharedState, ip: Option<IpAddr>) -> Result<Box<dyn Reply>, Rejection> {
|
||||
if let Err(res) = check_api_enabled(&state) {
|
||||
return Ok(Box::new(res))
|
||||
}
|
||||
|
@ -13,20 +16,63 @@ pub async fn get_all(state: SharedState) -> Result<Box<dyn Reply>, Rejection> {
|
|||
return Ok(Box::new(function_disabled_err()))
|
||||
}
|
||||
|
||||
let found =
|
||||
state.file_mgr.get_all(true, true)
|
||||
.await
|
||||
.map_err(|x| x.to_string());
|
||||
|
||||
if let Err(err) = found {
|
||||
return Ok(
|
||||
Box::new(
|
||||
with_status(
|
||||
json(
|
||||
&ErrorMessage {
|
||||
error: Error::APIError,
|
||||
details: Some(
|
||||
format!("Error while getting all files: {err}")
|
||||
)
|
||||
}
|
||||
),
|
||||
StatusCode::INTERNAL_SERVER_ERROR
|
||||
)
|
||||
)
|
||||
);
|
||||
} else {
|
||||
|
||||
let mut found = found.unwrap();
|
||||
|
||||
if state.config.api.get_all_own_only {
|
||||
found = found
|
||||
.iter()
|
||||
.filter(
|
||||
|x| {
|
||||
if let Some(owner) = x.uploader_ip {
|
||||
if let Some(caller) = ip {
|
||||
println!("{owner} {caller}");
|
||||
return owner == caller
|
||||
}
|
||||
}
|
||||
false
|
||||
}
|
||||
).map(|x| x.clone()).collect();
|
||||
}
|
||||
|
||||
Ok(
|
||||
Box::new(
|
||||
json(
|
||||
&state.file_mgr.get_all(true, true)
|
||||
.await
|
||||
.map_err(|x| x.to_string())
|
||||
.map_err(|x| HttpReject::StringError(x))?
|
||||
&found
|
||||
)
|
||||
)
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get_all_f(state: SharedState) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
|
||||
|
||||
let proxy = state.env.proxy_addr;
|
||||
|
||||
warp::path!("api" / "files" / "get_all")
|
||||
.map(move || state.clone())
|
||||
.and(real_ip(vec![proxy]))
|
||||
.and_then(get_all)
|
||||
}
|
Loading…
Reference in New Issue