wip

src/routes/pdf.rs

@@ -1,12 +1,15 @@
 use headless_chrome::Browser;
 use headless_chrome::types::PrintToPdfOptions;
-use nanobyte_opentelemetry::rocket::TracingSpan;
-use rocket::{get, response::stream::ByteStream, fairing::Fairing, fs::NamedFile, futures::TryFutureExt};
+use nanobyte_opentelemetry::rocket::{TracingSpan, RequestId, OtelReqwestClient};
+use ovlach_data::cv::cv::CV;
+use reqwest::Client;
+use rocket::fs::NamedFile;
+use ::rocket::{State, http::Status};
+use ::rocket::get;
 use tempfile::NamedTempFile;
 use tera::Context;
 use tracing::{info_span, error, debug};
-use crate::{chromium::rocket::BrowserHolder, tools::{tera::NanoTera, pdf::PdfStream}};
-use urlencoding::encode;
+use crate::{chromium::rocket::BrowserHolder, tools::{tera::NanoTera, pdf::PdfStream, rocket::RequestLanguage}, services::cv::fetch_cv_data_from_backend, CVBackendConfig};
 
 // TODO: request-id
 fn generate_pdf(browser: Browser, file: &NamedTempFile) -> Vec<u8> {
@@ -38,29 +41,59 @@ fn generate_pdf(browser: Browser, file: &NamedTempFile) -> Vec<u8> {
 }
 
 #[tracing::instrument]
-fn render_template(template_name: &str, file: &NamedTempFile, tera: NanoTera) {
+fn render_template(template_name: &str, file: &NamedTempFile, tera: NanoTera, cv: CV, language: String) {
     // TODO: handle errors
-    tera.0.render_to("two_column.html.tera", &Context::new(), file);
+    let mut tera_context = Context::new();
+    tera_context.insert("cv", &cv);
+    tera_context.insert("lang", language.as_str());
+    
+    match tera.0.render_to("two_column.html.tera", &tera_context, file) {
+        Ok(_) => {
+            debug!("Rendered template to {}", file.path().to_str().unwrap());
+        },
+        Err(e) => {
+            error!("Error rendering template {}: {}", &template_name, e);
+        }
+    }
 }
 
-#[get("/cv/<username>/output.pdf")]
-pub async fn render_pdf_cv(username: &str, browser: BrowserHolder, tera: NanoTera, tracing: TracingSpan) -> PdfStream { 
+#[get("/cv/<username>/<language>/output.pdf")]
+pub async fn render_pdf_cv(username: &str, tera: NanoTera, tracing: TracingSpan, request_client: OtelReqwestClient,
+    cv_config: &State<CVBackendConfig>, language: RequestLanguage, browser: BrowserHolder) -> Result<PdfStream, Status> { 
     let entered_span = tracing.0.enter();
-    let file = tempfile::Builder::new().suffix(".html").tempfile().unwrap(); 
-    render_template("two_column", &file, tera);
-    let span = info_span!("render_pdf", username = username);
-    let pdf = span.in_scope(||{
-        generate_pdf(browser.browser, &file)
-    });
-    drop(entered_span);
-    PdfStream::new(pdf)
+    match fetch_cv_data_from_backend(cv_config.cv_backend_path.clone(), request_client.0).await {
+        Ok(cv_data) => {
+            let file = tempfile::Builder::new().suffix(".html").tempfile().unwrap(); 
+            render_template("two_column", &file, tera, cv_data, language.language);
+            let span = info_span!("render_pdf", username = username);
+            let pdf = span.in_scope(||{
+                generate_pdf(browser.browser, &file)
+            });
+            drop(entered_span);
+            Ok(PdfStream::new(pdf))
+        },
+        Err(e) => {
+            error!("Error fetching cv data: {:?}", e);
+            Err(Status::InternalServerError)
+        }
+    }
 }
 
 /// Route only for debuging
-#[get("/cv/<username>/output.html")]
-pub async fn render_html_cv(username: &str, tera: NanoTera, tracing: TracingSpan) -> NamedFile { 
-    let entered_span = tracing.0.enter();
-    let file = tempfile::Builder::new().suffix(".html").tempfile().unwrap(); 
-    render_template("two_column", &file, tera);
-    NamedFile::open(file.path()).await.unwrap()
+#[get("/cv/<username>/<language>/output.html")]
+pub async fn render_html_cv(username: &str, tera: NanoTera, tracing: TracingSpan, request_client: OtelReqwestClient,
+    cv_config: &State<CVBackendConfig>, language: RequestLanguage) -> Result<NamedFile, Status> { 
+    let _ = tracing.0.enter();
+    match fetch_cv_data_from_backend(cv_config.cv_backend_path.clone(), request_client.0).await {
+        Ok(cv_data) => {
+            let file = tempfile::Builder::new().suffix(".html").tempfile().unwrap(); 
+            render_template("two_column", &file, tera, cv_data, language.language);
+            Ok(NamedFile::open(file.path()).await.unwrap())
+        },
+        Err(e) => {
+            error!("Error fetching cv data: {:?}", e);
+            Err(Status::InternalServerError)
+        }
+    }
+    
 }