allow user to configure html2pdf timeout

Signed-off-by: Andrei Gherghescu <[email protected]>

Author: andrei-ng

plotly_static/Cargo.toml

@@ -16,7 +16,6 @@ exclude = ["target/*"]
 webdriver_download = []
 geckodriver = []
 chromedriver = []
-html2pdf_export_timeout = []
 
 [dependencies]
 log = "0.4"

plotly_static/src/lib.rs

@@ -455,6 +455,7 @@ struct PlotData<'a> {
 ///     .webdriver_port(4444)
 ///     .spawn_webdriver(true)
 ///     .offline_mode(false)
+///     .pdf_export_timeout(500)
 ///     .build()
 ///     .expect("Failed to build StaticExporter");
 /// ```
@@ -465,6 +466,7 @@ struct PlotData<'a> {
 /// - WebDriver URL: "http://localhost"
 /// - Spawn webdriver: true (automatically manages WebDriver lifecycle)
 /// - Offline mode: false
+/// - PDF export timeout: 250ms
 /// - Browser capabilities: Default Chrome/Firefox headless options
 /// - Automatic WebDriver detection and connection reuse
 pub struct StaticExporterBuilder {
@@ -476,6 +478,8 @@ pub struct StaticExporterBuilder {
     spawn_webdriver: bool,
     /// Use bundled JS libraries instead of CDN (default: false)
     offline_mode: bool,
+    /// PDF export timeout in milliseconds (default: 250)
+    pdf_export_timeout: u32,
     /// Browser command-line flags (e.g., "--headless", "--no-sandbox")
     webdriver_browser_caps: Vec<String>,
 }
@@ -488,13 +492,15 @@ impl Default for StaticExporterBuilder {
     /// - WebDriver URL: "http://localhost"
     /// - Spawn webdriver: true
     /// - Offline mode: false
+    /// - PDF export timeout: 250ms
     /// - Default browser capabilities for headless operation
     fn default() -> Self {
         Self {
             webdriver_port: webdriver::WEBDRIVER_PORT,
             webdriver_url: webdriver::WEBDRIVER_URL.to_string(),
             spawn_webdriver: true,
             offline_mode: false,
+            pdf_export_timeout: 250,
             webdriver_browser_caps: {
                 #[cfg(feature = "chromedriver")]
                 {
@@ -596,6 +602,30 @@ impl StaticExporterBuilder {
         self
     }
 
+    /// Sets the PDF export timeout in milliseconds.
+    ///
+    /// This timeout controls how long to wait for the SVG image to load before
+    /// proceeding with PDF generation. A longer timeout may be needed for
+    /// complex plots or slower systems.
+    ///
+    /// # Examples
+    ///
+    /// ```rust
+    /// use plotly_static::StaticExporterBuilder;
+    ///
+    /// // Set a longer timeout for complex plots
+    /// let builder = StaticExporterBuilder::default()
+    ///     .pdf_export_timeout(500);
+    ///
+    /// // Use default timeout (250ms)
+    /// let builder = StaticExporterBuilder::default()
+    ///     .pdf_export_timeout(250);
+    /// ```
+    pub fn pdf_export_timeout(mut self, timeout_ms: u32) -> Self {
+        self.pdf_export_timeout = timeout_ms;
+        self
+    }
+
     /// Sets custom browser capabilities for the WebDriver.
     ///
     /// # Examples
@@ -658,6 +688,7 @@ impl StaticExporterBuilder {
             webdriver_url: self.webdriver_url.clone(),
             webdriver: wd,
             offline_mode: self.offline_mode,
+            pdf_export_timeout: self.pdf_export_timeout,
             webdriver_browser_caps: self.webdriver_browser_caps.clone(),
             runtime,
             webdriver_client: None,
@@ -735,6 +766,9 @@ pub struct StaticExporter {
     /// Use bundled JS libraries instead of CDN
     offline_mode: bool,
 
+    /// PDF export timeout in milliseconds
+    pdf_export_timeout: u32,
+
     /// Browser command-line flags (e.g., "--headless", "--no-sandbox")
     webdriver_browser_caps: Vec<String>,
 
@@ -969,7 +1003,7 @@ impl StaticExporter {
                     plot.scale.into(),
                 ];
 
-                (pdf_export_js_script(), args)
+                (pdf_export_js_script(self.pdf_export_timeout), args)
             }
             _ => {
                 let args = vec![

plotly_static/src/template.rs

@@ -38,13 +38,7 @@ pub(crate) fn image_export_js_script() -> String {
     .to_string()
 }
 
-pub(crate) fn pdf_export_js_script() -> String {
-    let timeout_ms = if cfg!(feature = "html2pdf_export_timeout") {
-        500
-    } else {
-        250
-    };
-
+pub(crate) fn pdf_export_js_script(timeout_ms: u32) -> String {
     let foreign_object_rendering = if cfg!(feature = "chromedriver") {
         "true"
     } else {