How to take a full page screenshot with Java and Selenium

<dependency>
  <groupId>org.seleniumhq.selenium</groupId>
  <artifactId>selenium-java</artifactId>
  <version>4.34.0</version>
</dependency>

import org.openqa.selenium.Dimension;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;

import java.io.File;
import java.nio.file.Files;
import java.nio.file.Path;

public class Screenshot {
  public static void main(String[] args) throws Exception {
    ChromeOptions options = new ChromeOptions();
    options.addArguments("--headless=new");

    ChromeDriver driver = new ChromeDriver(options);
    driver.get("https://www.htmltoimage.com/demo");

    // Chrome can't natively capture beyond the viewport, so size the window
    // to the full height of the document first.
    long width = (long) driver.executeScript("return document.documentElement.scrollWidth");
    long height = (long) driver.executeScript("return document.documentElement.scrollHeight");
    driver.manage().window().setSize(new Dimension((int) width, (int) height));

    File image = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
    Files.copy(image.toPath(), Path.of("full-page.png"));

    driver.quit();
  }
}