feat(screenshot): [closes #34], Take full page screenshots of urls

pwnwriter · Nov 3, 2023 · b43753f · b43753f
1 parent abfaca6
commit b43753f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/cli/screenshot.rs b/src/cli/screenshot.rs
@@ -2,9 +2,10 @@ use super::args::{Cli, Input};
 use super::ascii::{BAR, RESET};
 use crate::log;
 use anyhow::Context;
+use chromiumoxide::page::ScreenshotParams;
 use chromiumoxide::{
     browser::{Browser, BrowserConfig},
-    cdp::browser_protocol::page::{CaptureScreenshotFormat, CaptureScreenshotParams},
+    cdp::browser_protocol::page::CaptureScreenshotFormat,
     handler::viewport::Viewport,
 };
 use colored::{Color, Colorize};
@@ -146,8 +147,10 @@ async fn take_screenshot(
     let filename = format!("{}.png", url.replace("://", "-").replace('/', "_"));
     let page = browser.new_page(parsed_url.clone()).await?;
     page.save_screenshot(
-        CaptureScreenshotParams::builder()
+        ScreenshotParams::builder()
             .format(CaptureScreenshotFormat::Png)
+            .full_page(true)
+            .omit_background(true)
             .build(),
         filename,
     )