From 300b0c9a3c84be1dab5efb7ea720d381f608ab22 Mon Sep 17 00:00:00 2001
From: baalajimaestro <me@baalajimaestro.me>
Date: Sun, 7 May 2023 22:33:27 +0530
Subject: [PATCH] Handle a specific case where gitlab token or project id can
 be wrong

Also make BASE_URL an env variable

Signed-off-by: baalajimaestro <me@baalajimaestro.me>
---
 src/main.rs | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 28e9766..0bd778b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -23,7 +23,7 @@ struct Paste {
 }
 
 #[derive(Serialize, Deserialize)]
-struct ResponseJSON {
+struct GitLabSuccessJSON {
     file_name: String,
     file_path: String,
     size: u64,
@@ -37,23 +37,35 @@ struct ResponseJSON {
     content: String,
 }
 
+#[derive(Serialize, Deserialize)]
+struct GitLabFailedJSON {
+    message: String,
+}
+
 #[get("/")]
 async fn paste_render(data: Data<Template>, paste: Query<Paste>) -> impl Responder {
     let my_data = data.get_ref();
     let mut context = Context::new();
     let gitlab_token = env::var("GITLAB_TOKEN").unwrap_or("".to_string());
+    let base_url = env::var("BASE_URL").unwrap_or("https://gitlab.com".to_string());
     let client = Client::new();
     let mut headers = HeaderMap::new();
     headers.insert("PRIVATE-TOKEN", gitlab_token.parse().unwrap());
     let paste_content: String;
     if paste.paste_id.is_some() {
-        let response = client.get(format!("https://git.baalajimaestro.me/api/v4/projects/12/repository/files/paste%2F{}.md?ref=master", paste.paste_id.as_ref().unwrap())).headers(headers).send().await.unwrap();
+        let response = client.get(format!("{}/repository/files/paste%2F{}.md?ref=master", base_url, paste.paste_id.as_ref().unwrap())).headers(headers).send().await.unwrap();
         let response_code = response.status().as_str().to_string();
         let response_json = response.text().await.unwrap();
         if response_code != "200" {
-            paste_content = "Paste ID Invalid".to_string();
+            let resp_struct: GitLabFailedJSON = serde_json::from_str(&response_json).unwrap();
+            if resp_struct.message == "404 Project Not Found" {
+                paste_content = "Project ID Mismatch/GitLab Token Missing".to_string();
+            }
+            else {
+                paste_content = "Paste ID is Invalid".to_string();
+            }
         } else {
-            let resp_struct: ResponseJSON = serde_json::from_str(&response_json).unwrap();
+            let resp_struct: GitLabSuccessJSON = serde_json::from_str(&response_json).unwrap();
             paste_content = String::from_utf8(
                 general_purpose::STANDARD
                     .decode(resp_struct.content)