81 lines
2.1 KiB
Rust
81 lines
2.1 KiB
Rust
use axum::{
|
|
body::Body,
|
|
extract::{Extension, Path},
|
|
handler::get,
|
|
http::HeaderMap,
|
|
response::IntoResponse,
|
|
routing::BoxRoute,
|
|
Router,
|
|
};
|
|
use headers::{ContentType, HeaderMapExt, HeaderValue};
|
|
use reqwest::{
|
|
header::{HeaderName, CONTENT_LENGTH},
|
|
StatusCode,
|
|
};
|
|
|
|
use crate::JMService;
|
|
|
|
use self::{
|
|
error::CDNError,
|
|
templates::{DirTemplate, HtmlTemplate},
|
|
};
|
|
|
|
mod error;
|
|
mod sql;
|
|
mod templates;
|
|
|
|
pub fn routes() -> Router<BoxRoute> {
|
|
Router::new()
|
|
.route("/", get(users))
|
|
.route("/:user/", get(memes))
|
|
.route("/:user/:filename", get(image))
|
|
.boxed()
|
|
}
|
|
|
|
async fn image(
|
|
Path((user, filename)): Path<(String, String)>,
|
|
Extension(service): Extension<JMService>,
|
|
) -> Result<impl IntoResponse, CDNError> {
|
|
let filename = urlencoding::decode(&filename)?.into_owned();
|
|
let cid = sql::get_cid(user, filename.clone(), &service.db_pool).await?;
|
|
let ipfs_path = format!("/ipfs/{}", cid);
|
|
let res = service.ipfs_cat(cid).await?;
|
|
let clength = res
|
|
.headers()
|
|
.get(HeaderName::from_static("x-content-length"))
|
|
.ok_or(CDNError::Internal)?;
|
|
|
|
let mut headers = HeaderMap::new();
|
|
let ctype = ContentType::from(new_mime_guess::from_path(filename).first_or_octet_stream());
|
|
headers.typed_insert(ctype);
|
|
headers.insert(CONTENT_LENGTH, clength.clone());
|
|
headers.insert("X-Ipfs-Path", HeaderValue::from_str(ipfs_path.as_str())?);
|
|
|
|
Ok((
|
|
StatusCode::OK,
|
|
headers,
|
|
Body::wrap_stream(res.bytes_stream()),
|
|
))
|
|
}
|
|
|
|
async fn users(Extension(service): Extension<JMService>) -> Result<impl IntoResponse, CDNError> {
|
|
let users = sql::get_users(&service.db_pool).await?;
|
|
Ok(HtmlTemplate(DirTemplate {
|
|
entries: users,
|
|
prefix: service.int_cdn_url(),
|
|
suffix: "/".to_string(),
|
|
}))
|
|
}
|
|
|
|
async fn memes(
|
|
Path(user): Path<String>,
|
|
Extension(service): Extension<JMService>,
|
|
) -> Result<impl IntoResponse, CDNError> {
|
|
let memes = sql::get_memes(user, &service.db_pool).await?;
|
|
Ok(HtmlTemplate(DirTemplate {
|
|
entries: memes,
|
|
prefix: ".".to_string(),
|
|
suffix: "".to_string(),
|
|
}))
|
|
}
|