Screenshot the viewport with Java and Selenium

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

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");

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

    driver.quit();
  }
}