move caps to webdriver

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

Author: andrei-ng

plotly_static/src/lib.rs

@@ -278,71 +278,6 @@ use webdriver::WebDriver;
 
 use crate::template::{image_export_js_script, pdf_export_js_script};
 
-#[cfg(all(feature = "chromedriver", not(target_os = "windows")))]
-fn chrome_default_caps() -> Vec<&'static str> {
-    vec![
-        "--headless",
-        "--no-sandbox",
-        "--disable-gpu-sandbox",
-        "--disable-dev-shm-usage",
-        "--disable-extensions",
-        "--disable-background-networking",
-        "--disable-sync",
-        "--disable-translate",
-        "--disable-background-timer-throttling",
-        "--disable-renderer-backgrounding",
-        "--disable-features=VizDisplayCompositor",
-        "--memory-pressure-off",
-        // macOS-specific flags from choreographer
-        "--enable-unsafe-swiftshader",
-        "--use-mock-keychain",
-        "--password-store=basic",
-        "--disable-web-security",
-        "--disable-breakpad",
-        "--no-first-run",
-        "--no-default-browser-check",
-        // Additional flags for better PDF generation support
-        "--disable-backgrounding-occluded-windows",
-        "--disable-ipc-flooding-protection",
-        "--enable-logging",
-        "--v=1",
-    ]
-}
-
-#[cfg(all(feature = "chromedriver", target_os = "windows"))]
-fn chrome_default_caps() -> Vec<&'static str> {
-    vec![
-        "--headless=new",
-        "--no-sandbox",
-        "--disable-dev-shm-usage",
-        "--disable-breakpad",
-        "--no-first-run",
-        "--no-default-browser-check",
-        // Stability flags to prevent renderer crashes
-        "--disable-background-networking",
-        "--disable-sync",
-        "--disable-translate",
-        "--disable-background-timer-throttling",
-        "--disable-renderer-backgrounding",
-        "--disable-backgrounding-occluded-windows",
-        "--disable-ipc-flooding-protection",
-        "--disable-extensions",
-        // Minimal flags for Windows headless operation without disabling GPU
-        "--hide-scrollbars",
-        "--mute-audio",
-        "--use-angle=swiftshader",
-        "--disable-software-rasterizer",
-    ]
-}
-
-#[cfg(feature = "geckodriver")]
-fn firefox_default_caps() -> Vec<&'static str> {
-    vec![
-        "-headless",   // Essential for headless operation (single dash for Firefox)
-        "--no-remote", // Prevents connecting to existing Firefox instances
-    ]
-}
-
 mod template;
 mod webdriver;
 
@@ -506,14 +441,14 @@ impl Default for StaticExporterBuilder {
             webdriver_browser_caps: {
                 #[cfg(feature = "chromedriver")]
                 {
-                    chrome_default_caps()
+                    crate::webdriver::chrome_default_caps()
                         .into_iter()
                         .map(|s| s.to_string())
                         .collect()
                 }
                 #[cfg(feature = "geckodriver")]
                 {
-                    firefox_default_caps()
+                    crate::webdriver::firefox_default_caps()
                         .into_iter()
                         .map(|s| s.to_string())
                         .collect()

plotly_static/src/webdriver.rs

@@ -34,6 +34,74 @@ pub(crate) const WEBDRIVER_PORT: u32 = 4444;
 /// Default WebDriver URL
 pub(crate) const WEBDRIVER_URL: &str = "http://localhost";
 
+#[cfg(all(feature = "chromedriver", not(target_os = "windows")))]
+pub(crate) fn chrome_default_caps() -> Vec<&'static str> {
+    vec![
+        "--headless",
+        "--no-sandbox",
+        "--disable-gpu-sandbox",
+        "--disable-dev-shm-usage",
+        "--disable-extensions",
+        "--disable-background-networking",
+        "--disable-sync",
+        "--disable-translate",
+        "--disable-background-timer-throttling",
+        "--disable-renderer-backgrounding",
+        "--disable-features=VizDisplayCompositor",
+        "--memory-pressure-off",
+        // macOS-specific flags from choreographer
+        "--enable-unsafe-swiftshader",
+        "--use-mock-keychain",
+        "--password-store=basic",
+        "--disable-web-security",
+        "--allow-file-access-from-files",
+        "--disable-breakpad",
+        "--no-first-run",
+        "--no-default-browser-check",
+        // Additional flags for better PDF generation support
+        "--disable-backgrounding-occluded-windows",
+        "--disable-ipc-flooding-protection",
+        "--enable-logging",
+        "--v=1",
+    ]
+}
+
+#[cfg(all(feature = "chromedriver", target_os = "windows"))]
+pub(crate) fn chrome_default_caps() -> Vec<&'static str> {
+    vec![
+        "--headless=new",
+        "--no-sandbox",
+        "--disable-dev-shm-usage",
+        "--disable-breakpad",
+        "--no-first-run",
+        "--no-default-browser-check",
+        // Stability flags to prevent renderer crashes
+        "--disable-background-networking",
+        "--disable-sync",
+        "--disable-translate",
+        "--disable-background-timer-throttling",
+        "--disable-renderer-backgrounding",
+        "--disable-backgrounding-occluded-windows",
+        "--disable-ipc-flooding-protection",
+        "--disable-extensions",
+        "--disable-web-security",
+        "--allow-file-access-from-files",
+        // Minimal flags for Windows headless operation without disabling GPU
+        "--hide-scrollbars",
+        "--mute-audio",
+        "--use-angle=swiftshader",
+        "--disable-software-rasterizer",
+    ]
+}
+
+#[cfg(feature = "geckodriver")]
+pub(crate) fn firefox_default_caps() -> Vec<&'static str> {
+    vec![
+        "-headless",   // Essential for headless operation (single dash for Firefox)
+        "--no-remote", // Prevents connecting to existing Firefox instances
+    ]
+}
+
 /// Internal WebDriver state
 #[derive(Debug)]
 struct WdInner {