Skip to main content

oseda_cli/cmd/
export.rs

1use std::{
2    error::Error,
3    process::Command,
4    sync::{
5        atomic::{AtomicBool, Ordering},
6        Arc,
7    },
8};
9
10use clap::Args;
11
12use crate::{
13    cmd::run,
14    config::read_and_validate_config,
15    net::kill_port,
16    puppeteer::{is_puppeteer_chrome_installed, prompt_install_puppeteer_chrome},
17};
18
19/// Options struct for the export subcommand
20#[derive(Args, Debug, Clone)]
21pub struct ExportOptions {
22    /// String name of the output PDF file
23    pub output: Option<String>,
24    /// Port the project runs on
25    #[arg(long, default_value_t = 3000)]
26    pub port: u16,
27}
28
29impl ExportOptions {
30    pub fn output_or_default(&self) -> String {
31        self.output.clone().unwrap_or_else(get_default_output)
32    }
33}
34
35fn get_default_output() -> String {
36    let default = String::from("slides.pdf");
37
38    let Ok(config) = read_and_validate_config() else {
39        println!("Warning: Could not read oseda-config.json");
40        println!("Using slides.pdf as name instead");
41        return default;
42    };
43
44    format!("{}.pdf", config.title.trim())
45}
46
47/// Export the current Oseda project to a PDF file via `decktape`
48pub fn export(opts: ExportOptions) -> Result<(), Box<dyn Error>> {
49    println!("Cleaning any existing oseda processing...");
50    if kill_port(opts.port).is_err() {
51        eprintln!("Warning, could not kill value on desired port")
52    }
53
54    let output = Command::new("npm")
55        .args(["install", "decktape@3.15.0"])
56        .current_dir(".")
57        .output()?;
58
59    // prompt for puppeteer and confirm installation
60    if !is_puppeteer_chrome_installed() {
61        prompt_install_puppeteer_chrome()?;
62    }
63    println!("Puppeteer Chrome is installed, continuing...");
64
65    if !output.status.success() {
66        eprintln!(
67            "Decktape installation failure: {}",
68            String::from_utf8_lossy(&output.stderr)
69        );
70        return Err("npm init failed".into());
71    }
72
73    // decktape automatic http://localhost:3000/ Desktop/IntroToRust/slides.pdf
74
75    let shutdown_flag = Arc::new(AtomicBool::new(false));
76    let run_flag = shutdown_flag.clone();
77
78    let run_handle = std::thread::spawn(move || run::run_with_shutdown(run_flag));
79
80    // wait a moment for the localhost server to spin up
81    std::thread::sleep(std::time::Duration::from_millis(10000));
82
83    let addr = format!("http://localhost:{}", opts.port);
84
85    // run decktape, assuming the server has spun up by now
86    let export_output = Command::new("npm")
87        .args([
88            "exec",
89            "decktape",
90            "reveal",
91            &addr,
92            &opts.output_or_default(),
93        ])
94        .output()?;
95
96    // send shutdown flag, should signal to run_with_shutdown to kill the process
97    shutdown_flag.store(true, Ordering::SeqCst);
98    // wait to run to terminate (hopefully gracefully) and join the process to cur. thread
99    let _ = run_handle.join();
100
101    if !export_output.status.success() {
102        eprintln!(
103            "Decktape PDF export failure: {}",
104            String::from_utf8_lossy(&export_output.stderr)
105        );
106        return Err("npm init failed".into());
107    }
108
109    Ok(())
110}