feat: todo's, add image cache control

src/server/services/images.rs

@@ -0,0 +1,60 @@
+//! Scope for the image get backend.
+//! - `/`: Gives the default image.
+//! - `/{username}`: Gives username's pfp or redirects to the default image's path (for better
+//!   cache control) if there's no image.
+//!
+//! Must be scoped at [`ws::IMAGES`]
+
+// TODO: etags
+
+use actix_web::{
+    HttpResponse, get,
+    http::header,
+    web::{self, Redirect},
+};
+
+use crate::{
+    consts::{self, web_scopes as ws},
+    server,
+};
+
+#[must_use]
+pub fn make_scope(path: &str) -> actix_web::Scope {
+    web::scope(path)
+        .service(get_default_image)
+        .service(get_image)
+}
+
+#[get("/")]
+async fn get_default_image(
+    _data: web::Data<&server::AppState>,
+    _username: web::Path<String>,
+) -> HttpResponse {
+    HttpResponse::Ok()
+        .insert_header((
+            header::CACHE_CONTROL,
+            consts::DEFAULT_USER_PFP_CACHES_HEADER,
+        ))
+        .content_type(consts::DEFAULT_USER_PFP_MIME)
+        .body(web::Bytes::from_static(consts::DEFAULT_USER_PFP))
+}
+
+#[get("/{username}")]
+async fn get_image(
+    data: web::Data<&server::AppState>,
+    username: web::Path<String>,
+) -> web::Either<Redirect, HttpResponse> {
+    let cached_pfp = data.cache.get_pfp(username.to_string()).await;
+
+    cached_pfp.as_ref().map_or_else(
+        || web::Either::Left(web::Redirect::to(ws::IMAGES).temporary()),
+        |img| {
+            web::Either::Right(
+                HttpResponse::Ok()
+                    .insert_header((header::CACHE_CONTROL, consts::USER_CACHES_HEADER))
+                    .content_type(img.mime.as_ref())
+                    .body(web::Bytes::copy_from_slice(img.bytes.as_ref())),
+            )
+        },
+    )
+}