!596 // auto-import

import javax.imageio.*;
import java.awt.image.*;
import java.awt.*;

public class main {
	public static void main(String[] args) throws Exception {
	  BufferedImage image = makeScreenshot();
	  save("output/screenshot.png", image);
	}

	static void save(String path, BufferedImage image) throws IOException {
	  saveBinaryFile(path, convertToPNG(image));
	}
	
  public static BufferedImage makeScreenshot() {
    try {
      Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
      Rectangle screenRectangle = new Rectangle(screenSize);
      Robot robot = new Robot();
      return robot.createScreenCapture(screenRectangle);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  public static byte[] convertToPNG(BufferedImage image) {
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    try {
      ImageIO.write(image, "png", baos);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
    return baos.toByteArray();
  }

  /** writes safely (to temp file, then rename) */
  public static void saveBinaryFile(String fileName, byte[] contents) throws IOException {
    File file = new File(fileName);
    File parentFile = file.getParentFile();
    if (parentFile != null)
      parentFile.mkdirs();
    String tempFileName = fileName + "_temp";
    FileOutputStream fileOutputStream = new FileOutputStream(tempFileName);
    fileOutputStream.write(contents);
    fileOutputStream.close();
    if (file.exists() && !file.delete())
      throw new IOException("Can't delete " + fileName);

    if (!new File(tempFileName).renameTo(file))
      throw new IOException("Can't rename " + tempFileName + " to " + fileName);
  }
}