wip

src/routes/pdf.rs

@@ -1,17 +1,20 @@
-use core::panic;
-use std::{fs, io::Write};
-
 use headless_chrome::Browser;
 use headless_chrome::types::PrintToPdfOptions;
 use nanobyte_opentelemetry::rocket::TracingSpan;
-use rocket::{get, fs::NamedFile};
-use tracing::{info_span, info};
-use crate::chromium::rocket::BrowserHolder;
+use rocket::{get, response::stream::ByteStream, fairing::Fairing, fs::NamedFile, futures::TryFutureExt};
+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;
 
-fn generate_pdf(browser: Browser) {
+// TODO: request-id
+fn generate_pdf(browser: Browser, file: &NamedTempFile) -> Vec<u8> {
     let tab = browser.new_tab().unwrap();
+    let path = format!("file://{}", file.path().to_str().unwrap());
     info_span!("open_pdf").in_scope(|| {
-       tab.navigate_to("file:///home/6a6996c0-1609-48b6-8ca6-affbef1b4d1d/Devel/Nanobyte/ovlach/ovlach_pdf/template.html").unwrap().wait_until_navigated().unwrap();
+        debug!("Render pdf from {}", &path);
+        tab.navigate_to(&path).unwrap().wait_until_navigated().unwrap();
     });
 
     let options = PrintToPdfOptions{
@@ -31,25 +34,33 @@ fn generate_pdf(browser: Browser) {
          tab.print_to_pdf(Some(options)).unwrap()
     });
 
-    info_span!("write_temporary_file").in_scope(|| {
-        let mut file = fs::OpenOptions::new()
-            // .create(true) // To create a new file
-            .write(true)
-            // either use the ? operator or unwrap since it returns a Result
-            .open("/tmp/foo.pdf").unwrap();
-        file.write_all(&bytes).unwrap();
-    });
+    return bytes;
 }
 
-#[get("/cv/<username>/pdf")]
-pub async fn render_pdf_cv(username: &str, browser: BrowserHolder, tracing: TracingSpan) -> Option<NamedFile> { 
+#[tracing::instrument]
+fn render_template(template_name: &str, file: &NamedTempFile, tera: NanoTera) {
+    // TODO: handle errors
+    tera.0.render_to("two_column.html.tera", &Context::new(), file);
+}
+
+#[get("/cv/<username>/output.pdf")]
+pub async fn render_pdf_cv(username: &str, browser: BrowserHolder, tera: NanoTera, tracing: TracingSpan) -> PdfStream { 
     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);
-    info!("generate PDF");
-    span.in_scope(||{
-        generate_pdf(browser.browser);
+    let pdf = span.in_scope(||{
+        generate_pdf(browser.browser, &file)
     });
-    info!("done generating");
     drop(entered_span);
-    Some(NamedFile::open("/tmp/foo.pdf").await.expect("failed to open foo.pdf"))
+    PdfStream::new(pdf)
+}
+
+/// 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()
 }