Fix jpeg extension not being correct

src/main.rs

@@ -1,7 +1,6 @@
 #[macro_use]
 extern crate rocket;
 
-use regex::Regex;
 use rocket::fs::FileServer;
 use rocket::response::Redirect;
 use reqwest::header::{HeaderMap, HeaderValue};
@@ -20,9 +19,9 @@ async fn download(url: &str) -> Redirect {
 
     println!("doc_id={}", doc_id);
 
-    let body = ureq::get(url).call().unwrap().into_string().unwrap();
-    let re = Regex::new(r##"token=(.*?)""##).unwrap();
-    let Some(caps) = re.captures(&*body) else {
+    let json = send_get_request(doc_id).await.unwrap();
+    let data = json["data"].as_object().unwrap();
+    if data["file_preview"].is_null(){
         println!("no token found");
         let json = send_get_request(doc_id).await.unwrap();
         let token = get_token().await.unwrap();
@@ -32,9 +31,16 @@ async fn download(url: &str) -> Redirect {
             .last()
             .unwrap();
         return Redirect::to(format!("https://cdn.studydrive.net/d/prod/documents/{}/original/{}.{}?token={}", doc_id, doc_id,ending, token));
-    };
-    let token = &caps[1];
-    Redirect::to(format!("https://cdn.studydrive.net/d/prod/documents/{}/original/{}.pdf?token={}", doc_id, doc_id, token))
+    }
+    let name = data["filename"].as_str().unwrap();
+    let ending=name.split(".")
+        .last()
+        .unwrap();
+    let preview = data["file_preview"].as_str().unwrap();
+    let token = preview.split("token=")
+            .last()
+            .unwrap();
+    Redirect::to(format!("https://cdn.studydrive.net/d/prod/documents/{}/original/{}.{}?token={}", doc_id, doc_id, ending, token))
 }
 
 async fn get_token() -> Result<String, Box<dyn std::error::Error>> {