From 4db81e96b21bec4a0b1f0e3283605033efe8e720 Mon Sep 17 00:00:00 2001
From: Sameera <sameira@gmail.com>
Date: Mon, 17 Oct 2016 00:24:25 +0530
Subject: [PATCH] Adding Print Screen in Java

---
 printscreen/pom.xml                           | 31 ++++++++++++++
 .../org/baeldung/corejava/Screenshot.java     | 42 +++++++++++++++++++
 .../org/baeldung/corejava/ScreenshotTest.java | 39 +++++++++++++++++
 3 files changed, 112 insertions(+)
 create mode 100644 printscreen/pom.xml
 create mode 100644 printscreen/src/main/java/org/baeldung/corejava/Screenshot.java
 create mode 100644 printscreen/src/test/java/org/baeldung/corejava/ScreenshotTest.java

diff --git a/printscreen/pom.xml b/printscreen/pom.xml
new file mode 100644
index 0000000000..25316a3b48
--- /dev/null
+++ b/printscreen/pom.xml
@@ -0,0 +1,31 @@
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+
+  <groupId>com.baeldung</groupId>
+  <artifactId>corejava-printscreen</artifactId>
+  <version>1.0-SNAPSHOT</version>
+  <packaging>jar</packaging>
+
+  <name>How to Print Screen in Java</name>
+  <url>https://github.com/eugenp/tutorials</url>
+
+  <properties>
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+  </properties>
+
+  <dependencies>
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>3.8.1</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>4.12</version>
+      <scope>test</scope>
+    </dependency>
+  </dependencies>
+</project>
diff --git a/printscreen/src/main/java/org/baeldung/corejava/Screenshot.java b/printscreen/src/main/java/org/baeldung/corejava/Screenshot.java
new file mode 100644
index 0000000000..57a754bb83
--- /dev/null
+++ b/printscreen/src/main/java/org/baeldung/corejava/Screenshot.java
@@ -0,0 +1,42 @@
+package org.baeldung.corejava;;
+
+import javax.imageio.ImageIO;
+import java.awt.*;
+import java.awt.image.BufferedImage;
+import java.io.File;
+
+public class Screenshot {
+
+    private String filePath;
+    private String filenamePrefix;
+    private String fileType;
+    private int timeToWait;
+
+    public Screenshot(String filePath, String filenamePrefix,
+                      String fileType, int timeToWait) {
+        this.filePath = filePath;
+        this.filenamePrefix = filenamePrefix;
+        this.fileType = fileType;
+        this.timeToWait = timeToWait;
+    }
+
+    public void getScreenshot() {
+        try {
+            Thread.sleep(timeToWait);
+            Rectangle rectangle = new Rectangle(Toolkit.getDefaultToolkit().getScreenSize());
+            Robot robot = new Robot();
+            BufferedImage img = robot.createScreenCapture(rectangle);
+            ImageIO.write(img, fileType, setupFileNamePath());
+        } catch (Exception ex) {
+            System.out.println("Error occurred while getting the Print Screen.");
+        }
+    }
+
+    private File setupFileNamePath() {
+        return new File(filePath + filenamePrefix + "." + fileType);
+    }
+
+    private Rectangle getScreenSizedRectangle(final Dimension d) {
+        return new Rectangle(0, 0, d.width, d.height);
+    }
+}
diff --git a/printscreen/src/test/java/org/baeldung/corejava/ScreenshotTest.java b/printscreen/src/test/java/org/baeldung/corejava/ScreenshotTest.java
new file mode 100644
index 0000000000..588c2eea78
--- /dev/null
+++ b/printscreen/src/test/java/org/baeldung/corejava/ScreenshotTest.java
@@ -0,0 +1,39 @@
+package org.baeldung.corejava;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.File;
+
+import static org.junit.Assert.*;
+
+
+public class ScreenshotTest {
+
+    private Screenshot screenshot;
+    private String filePath;
+    private String fileName;
+    private String fileType;
+    private File file;
+
+    @Before
+    public void setUp() throws Exception {
+        filePath = "";
+        fileName = "Screenshot";
+        fileType = "jpg";
+        file = new File(filePath + fileName + "." + fileType);
+        screenshot = new Screenshot(filePath, fileName, fileType, 2000);
+    }
+
+    @Test
+    public void testGetScreenshot() throws Exception {
+        screenshot.getScreenshot();
+        assertTrue(file.exists());
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        file.delete();
+    }
+}
\ No newline at end of file