Add full page screenshot example

Author: kklisura

cdt-examples/src/main/java/com/github/kklisura/cdt/examples/FullPageScreenshotExample.java

@@ -0,0 +1,98 @@
+package com.github.kklisura.cdt.examples;
+
+import com.github.kklisura.cdt.launch.ChromeLauncher;
+import com.github.kklisura.cdt.protocol.commands.Page;
+import com.github.kklisura.cdt.protocol.types.page.CaptureScreenshotFormat;
+import com.github.kklisura.cdt.protocol.types.page.LayoutMetrics;
+import com.github.kklisura.cdt.protocol.types.page.Viewport;
+import com.github.kklisura.cdt.services.ChromeDevToolsService;
+import com.github.kklisura.cdt.services.ChromeService;
+import com.github.kklisura.cdt.services.types.ChromeTab;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.Base64;
+
+/**
+ * Takes a full page screenshot. The output screenshot dimensions will be page width x page height,
+ * for example when capturing http://github.com the output screenshot image will be 1200 × 6598.
+ *
+ * @author Kenan Klisura
+ */
+public class FullPageScreenshotExample {
+
+  private static void captureFullPageScreenshot(Page page, String outputFilename) {
+    final LayoutMetrics layoutMetrics = page.getLayoutMetrics();
+
+    Viewport viewport = new Viewport();
+    viewport.setScale(1d);
+
+    // You can set offset with X, Y
+    viewport.setX(0d);
+    viewport.setY(0d);
+
+    // Set a width, height of a page to take screenshot at
+    viewport.setWidth(layoutMetrics.getContentSize().getWidth());
+    viewport.setHeight(layoutMetrics.getContentSize().getHeight());
+
+    dump(
+        outputFilename,
+        page.captureScreenshot(CaptureScreenshotFormat.PNG, 100, viewport, Boolean.TRUE));
+  }
+
+  public static void main(String[] args) {
+    // Create chrome launcher.
+    final ChromeLauncher launcher = new ChromeLauncher();
+
+    // Launch chrome either as headless (true) or regular (false).
+    final ChromeService chromeService = launcher.launch(false);
+
+    // Create empty tab ie about:blank.
+    final ChromeTab tab = chromeService.createTab();
+
+    // Get DevTools service to this tab
+    final ChromeDevToolsService devToolsService = chromeService.createDevToolsService(tab);
+
+    // Get individual commands
+    final Page page = devToolsService.getPage();
+
+    page.onLoadEventFired(
+        event -> {
+          System.out.println("Taking screenshot...");
+
+          captureFullPageScreenshot(page, "screenshot.png");
+
+          System.out.println("Done!");
+
+          devToolsService.close();
+        });
+
+    // Enable page events.
+    page.enable();
+
+    // Navigate to github.com.
+    page.navigate("http://github.com");
+
+    devToolsService.waitUntilClosed();
+  }
+
+  private static void dump(String fileName, String data) {
+    FileOutputStream fileOutputStream = null;
+    try {
+      File file = new File(fileName);
+      fileOutputStream = new FileOutputStream(file);
+      fileOutputStream.write(Base64.getDecoder().decode(data));
+    } catch (IOException e) {
+      e.printStackTrace();
+    } finally {
+      if (fileOutputStream != null) {
+        try {
+          fileOutputStream.flush();
+          fileOutputStream.close();
+        } catch (IOException e) {
+          e.printStackTrace();
+        }
+      }
+    }
+  }
+}