✨ Adds image route

src/api/image.rs

@@ -0,0 +1,82 @@
+use crate::models::AppState;
+use crate::models::hit::Hit;
+use crate::models::tracker::Tracker;
+use crate::schema::{hits, trackers};
+use chrono::Utc;
+use diesel::{ExpressionMethods, QueryDsl, RunQueryDsl};
+use rocket::fs::NamedFile;
+use rocket::http::Status;
+use rocket::request::{FromRequest, Outcome};
+use rocket::{Request, State};
+use std::path::Path;
+use uuid::Uuid;
+
+pub struct ReqMeta {
+    pub ip: String,
+    pub agent: Option<String>,
+    pub language: Option<String>,
+}
+
+#[rocket::async_trait]
+impl<'r> FromRequest<'r> for ReqMeta {
+    type Error = ();
+
+    async fn from_request(req: &'r Request<'_>) -> Outcome<Self, Self::Error> {
+        let headers = req.headers();
+        let agent = headers.get_one("User-Agent").map(|s| s.to_string());
+        let language = headers.get_one("Accept-Language").map(|s| s.to_string());
+
+        let ip = req
+            .client_ip()
+            .map(|ip| ip.to_string())
+            .or_else(|| {
+                headers
+                    .get_one("X-Forwarded-For")
+                    .map(|s| s.split(',').next().unwrap_or("").trim().to_string())
+            })
+            .unwrap_or_else(|| "unknown".to_string());
+
+        Outcome::Success(ReqMeta {
+            ip,
+            agent,
+            language,
+        })
+    }
+}
+
+#[get("/<id>")]
+pub async fn get(id: String, meta: ReqMeta, state: &State<AppState>) -> Result<NamedFile, Status> {
+    let id = match Uuid::parse_str(id.as_str()).ok() {
+        Some(id) => id,
+        None => return Err(Status::BadRequest),
+    };
+
+    {
+        let mut db = state.db.lock().unwrap();
+
+        let result = trackers::dsl::trackers
+            .filter(trackers::id.eq(id))
+            .first::<Tracker>(&mut *db)
+            .ok();
+
+        if result.is_none() {
+            return Err(Status::NotFound);
+        }
+
+        // TODO: handle possible error
+        let _ = diesel::insert_into(hits::table)
+            .values(&Hit {
+                id: Uuid::new_v4(),
+                tracker_id: id,
+                ip: meta.ip,
+                agent: meta.agent,
+                language: meta.language,
+                created_at: Utc::now().naive_utc(),
+            })
+            .execute(&mut *db);
+    }
+
+    NamedFile::open(Path::new(state.image_path.as_str()))
+        .await
+        .map_err(|_| Status::InternalServerError)
+}

src/api/mod.rs

@@ -1,2 +1,3 @@
 pub mod tracker;
 pub mod hit;
+pub mod image;