implement oembed support for images

Cargo.toml

@@ -25,6 +25,7 @@ reqwest = { version = "0.12.11", features = ["json"] }
 serde = { version = "1", features = ["derive"] }
 tokio = { version = "1", features = ["net", "rt-multi-thread"] }
 uuid = { version = "1", features = ["serde", "v4"] }
+bytes = "1.9.0"
 
 [dependencies.axum]
 version = "0.7"

src/endpoints/image.rs

@@ -1,12 +1,14 @@
 use axum::{
 	body::Bytes,
-	extract::{Path, State},
+	extract::{Path, Query, State},
+	response::Html,
 	Json,
 };
 use base64::{engine::general_purpose::STANDARD_NO_PAD, Engine};
+use bytes::Buf;
 use chrono::{DateTime, Utc};
 use reqwest::StatusCode;
-use serde::Serialize;
+use serde::{Deserialize, Serialize};
 use sqlx::{query, PgPool};
 use uuid::Uuid;
 
@@ -47,7 +49,7 @@ pub async fn get_raw(
 	State(ApiState { database, .. }): State<ApiState>,
 	Path(id): Path<Id>,
 ) -> Result<Vec<u8>, ApiError> {
-	let image = query!("SELECT player, filename, file, timestamp FROM images WHERE id = $1", id as _)
+	let image = query!("SELECT file FROM images WHERE id = $1", id as _)
 		.fetch_optional(&database)
 		.await?
 		.ok_or(StatusCode::NOT_FOUND)?;
@@ -60,6 +62,10 @@ pub async fn post(
 	Path(filename): Path<String>,
 	body: Bytes,
 ) -> Result<String, ApiError> {
+	let png = PngInfo::create(&body).await;
+	if png.is_none() {
+		return Err(StatusCode::BAD_REQUEST)?;
+	}
 	let id = Id::new();
 	query!(
 		"INSERT INTO images (id, player, filename, file) VALUES ($1, $2, $3, $4)",
@@ -80,3 +86,131 @@ pub async fn evict_expired(database: &PgPool) -> Result<(), TaskError> {
 		.await?;
 	Ok(())
 }
+
+pub async fn get_view(
+	State(ApiState { database, .. }): State<ApiState>,
+	Path(id): Path<Id>,
+) -> Result<Html<String>, ApiError> {
+	let image = query!("SELECT filename, player, timestamp, file FROM images WHERE id = $1", id as _)
+		.fetch_optional(&database)
+		.await?
+		.ok_or(StatusCode::NOT_FOUND)?;
+
+	let filename = String::from_utf8(image.filename).unwrap();
+	let base_url = "https://api.axolotlclient.com/v1/";
+	let image_url = base_url.to_string() + "image/" + &id.to_string();
+
+	let username = query!("SELECT username FROM players WHERE uuid = $1", image.player)
+		.fetch_one(&database)
+		.await?
+		.username;
+
+	let time = image.timestamp.and_utc().format("%Y/%m/%d %H:%M").to_string();
+	let png = PngInfo::create(&Bytes::from(image.file.clone())).await.unwrap();
+	Ok(Html(
+		include_str!("image_view.html")
+			.replace("{filename}", &filename)
+			.replace("{image_data}", &("data:image/png;base64,".to_string() + &STANDARD_NO_PAD.encode(image.file)))
+			.replace("{image_url}", &image_url)
+			.replace("{image_width}", &png.width.to_string())
+			.replace("{image_height}", &png.height.to_string())
+			.replace("{username}", &username)
+			.replace(
+				"{time}",
+				&image
+					.timestamp
+					.and_utc()
+					.to_rfc3339_opts(chrono::SecondsFormat::Secs, true),
+			)
+			.replace("{time_formatted}", &time),
+	))
+}
+
+#[derive(Serialize)]
+pub struct OEmbed {
+	version: &'static str,
+	#[serde(rename(serialize = "type"))]
+	_type: &'static str,
+	title: String,
+	url: String,
+	width: i32,
+	height: i32,
+	provider_name: &'static str,
+	provider_url: &'static str,
+}
+
+impl OEmbed {
+	fn create(title: String, url: String, png: PngInfo) -> OEmbed {
+		OEmbed {
+			version: "1.0",
+			_type: "photo",
+			title,
+			url,
+			width: png.width,
+			height: png.height,
+			provider_name: "AxolotlClient",
+			provider_url: "https://axolotlclient.com",
+		}
+	}
+}
+
+#[derive(Deserialize)]
+pub struct OEmbedQuery {
+	format: String,
+}
+
+pub async fn get_oembed(
+	State(ApiState { database, .. }): State<ApiState>,
+	Path(id): Path<Id>,
+	Query(OEmbedQuery { format }): Query<OEmbedQuery>,
+) -> Result<Json<OEmbed>, ApiError> {
+	let image = query!("SELECT filename, file FROM images WHERE id = $1", id as _)
+		.fetch_optional(&database)
+		.await?
+		.ok_or(StatusCode::NOT_FOUND)?;
+	let png = PngInfo::create(&Bytes::from(image.file)).await;
+
+	if png.is_none() {
+		return Err(StatusCode::BAD_REQUEST)?;
+	}
+
+	let filename = String::from_utf8(image.filename).unwrap();
+
+	let embed = OEmbed::create(
+		filename,
+		"https://api.axolotlclient.com/v1/images/".to_owned() + &id.to_string() + "/raw",
+		png.unwrap(),
+	);
+	Ok(if format == "json" {
+		Json(embed)
+	} else {
+		return Err(StatusCode::NOT_IMPLEMENTED)?;
+	})
+}
+
+struct PngInfo {
+	width: i32,
+	height: i32,
+}
+
+impl PngInfo {
+	async fn create(reader: &Bytes) -> Option<PngInfo> {
+		let mut bytes = reader.clone();
+		let header = bytes.get_u64();
+		if header != 0x89504E470D0A1A0A {
+			return None;
+		}
+		let ihdr_size = bytes.get_u32();
+		if ihdr_size != 0x0D {
+			return None;
+		}
+		let ihdr_type = bytes.get_u32();
+		if ihdr_type != 0x49484452 {
+			return None;
+		}
+		Some(PngInfo {
+			width: bytes.get_i32(),
+			height: bytes.get_i32(),
+		})
+	}
+}