From 916f1b4a45639839145e8a54848ddc2ce29773b0 Mon Sep 17 00:00:00 2001
From: Timo Ley <timo.ley@tilera.dev>
Date: Fri, 15 Mar 2024 13:14:29 +0100
Subject: [PATCH] feat: seperate internal and external cdn urls

---
 src/cdn/mod.rs   |  2 +-
 src/config.rs    | 14 ++++++++++----
 src/main.rs      |  3 ++-
 src/v1/routes.rs |  8 ++++----
 4 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/src/cdn/mod.rs b/src/cdn/mod.rs
index a66128c..0a3c798 100644
--- a/src/cdn/mod.rs
+++ b/src/cdn/mod.rs
@@ -62,7 +62,7 @@ async fn users(Extension(service): Extension<JMService>) -> Result<impl IntoResp
     let users = sql::get_users(&service.db_pool).await?;
     Ok(HtmlTemplate(DirTemplate {
         entries: users,
-        prefix: service.cdn_url(),
+        prefix: service.int_cdn_url(),
         suffix: "/".to_string(),
     }))
 }
diff --git a/src/config.rs b/src/config.rs
index 9d4957f..a94dacc 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -9,7 +9,8 @@ use crate::{error::JMError, JMService, JMServiceInner};
 pub struct Config {
     pub addr: SocketAddr,
     pub database: String,
-    pub cdn: String,
+    pub int_cdn: String,
+    pub ext_cdn: String,
     pub ipfs_api: Url,
     pub matrix_url: Url,
     pub matrix_token: String,
@@ -23,7 +24,8 @@ impl Config {
             client,
             db_pool,
             ipfs_url: self.ipfs_api.clone(),
-            cdn_url: self.cdn.clone(),
+            int_cdn: self.int_cdn.clone(),
+            ext_cdn: self.ext_cdn.clone(),
             matrix_url: self.matrix_url.clone(),
             matrix_token: self.matrix_token.clone(),
             matrix_domain: self.matrix_domain.clone(),
@@ -32,7 +34,11 @@ impl Config {
 }
 
 impl JMServiceInner {
-    pub fn cdn_url(&self) -> String {
-        self.cdn_url.clone()
+    pub fn int_cdn_url(&self) -> String {
+        self.int_cdn.clone()
+    }
+
+    pub fn ext_cdn_url(&self) -> String {
+        self.ext_cdn.clone()
     }
 }
diff --git a/src/main.rs b/src/main.rs
index a30263f..8a58a9d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -37,7 +37,8 @@ pub struct JMServiceInner {
     client: Client,
     db_pool: PgPool,
     ipfs_url: Url,
-    cdn_url: String,
+    int_cdn: String,
+    ext_cdn: String,
     matrix_url: Url,
     matrix_token: String,
     matrix_domain: String,
diff --git a/src/v1/routes.rs b/src/v1/routes.rs
index 70db0bf..2145fd4 100644
--- a/src/v1/routes.rs
+++ b/src/v1/routes.rs
@@ -22,7 +22,7 @@ async fn meme(
             .get_meme(params.id)
             .await?
             .ok_or_else(|| APIError::NotFound("Meme not found".to_string()))?,
-        service.cdn_url(),
+        service.ext_cdn_url(),
     );
     Ok(Json(MemeResponse {
         status: 200,
@@ -39,7 +39,7 @@ async fn memes(
         .get_memes(params.into())
         .await?
         .into_iter()
-        .map(|meme| V1Meme::new(meme, service.cdn_url()))
+        .map(|meme| V1Meme::new(meme, service.ext_cdn_url()))
         .collect();
     Ok(Json(MemesResponse {
         status: 200,
@@ -104,7 +104,7 @@ async fn random(
 ) -> Result<impl IntoResponse, APIError> {
     let random = V1Meme::new(
         service.get_random_meme(params.into()).await?,
-        service.cdn_url(),
+        service.ext_cdn_url(),
     );
     Ok(Json(MemeResponse {
         status: 200,
@@ -181,7 +181,7 @@ async fn upload(
         service.ipfs_pin(f.hash).await?;
         links.push(format!(
             "{}/{}/{}",
-            service.cdn_url(),
+            service.ext_cdn_url(),
             user.id.clone(),
             f.name.clone()
         ));