From 6e8ff0294d1c34b727129fd5f0c82a9ff7016900 Mon Sep 17 00:00:00 2001
From: Yusuf Bera Ertan <y.bera003.06@protonmail.com>
Date: Mon, 17 Jul 2023 10:46:32 +0300
Subject: [PATCH] respond to discord with the image itself

---
 src/main.rs | 45 +++++++++++++++++++++++++++++++++++++--------
 1 file changed, 37 insertions(+), 8 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 39bb762..532281a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,9 @@
-use axum::{extract::State, response::Html, routing::get, Router};
+use axum::{
+    extract::State,
+    response::{Html, IntoResponse},
+    routing::get,
+    Router,
+};
 use dashmap::DashMap;
 use error::AppError;
 use http::Uri;
@@ -6,7 +11,7 @@ use std::{
     collections::HashMap,
     ops::Deref,
     str::FromStr,
-    sync::{atomic::AtomicBool, Arc, Mutex},
+    sync::{Arc, Mutex},
 };
 
 mod error;
@@ -149,18 +154,42 @@ async fn main() {
     axum::serve(listener, app).await.unwrap();
 }
 
-async fn show_art(state: State<AppState>) -> AppResult<Html<String>> {
+async fn show_art(
+    state: State<AppState>,
+    headers: http::HeaderMap,
+) -> AppResult<axum::response::Response> {
     let art = state.data.lock().unwrap().pick_random_art().clone();
-    if let Some(image_link) = state.direct_links.get(&art.url) {
-        Ok(render_page(&art, &image_link))
+    let image_link = if let Some(image_link) = state.direct_links.get(&art.url) {
+        image_link.to_string()
     } else {
         let image_link = match art.kind {
             ArtKind::Twitter => fetch_twitter_image_link(&state.http, &art.url).await?,
         };
-        let page = render_page(&art, &image_link);
-        state.direct_links.insert(art.url, image_link);
-        Ok(page)
+        state
+            .direct_links
+            .insert(art.url.clone(), image_link.clone());
+        image_link
+    };
+
+    if let Some(agent) = headers
+        .get(http::header::USER_AGENT)
+        .and_then(|h| h.to_str().ok())
+    {
+        if agent.contains("Discordbot") {
+            let request = state.http.get(&image_link).build()?;
+            let mut resp = state.http.execute(request).await?.error_for_status()?;
+            let content_type = resp.headers_mut().remove(http::header::CONTENT_TYPE);
+            let downloaded = resp.bytes().await?;
+            let mut response = axum::response::Response::new(downloaded.into());
+            if let Some(v) = content_type {
+                response.headers_mut().insert(http::header::CONTENT_TYPE, v);
+            }
+            return Ok(response);
+        }
     }
+
+    let page = render_page(&art, &image_link);
+    Ok(page.into_response())
 }
 
 fn render_page(art: &Art, image_link: &str) -> Html<String> {