From 0dfc4d854e2c0c459a1379eca659bfb4f8375f6b Mon Sep 17 00:00:00 2001 From: Eric-Turner Date: Fri, 20 May 2016 22:03:01 -0400 Subject: [PATCH] Made a file name duplication prevention --- src/main/java/org/parabot/core/ui/BotUI.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/main/java/org/parabot/core/ui/BotUI.java b/src/main/java/org/parabot/core/ui/BotUI.java index 3a0d669..9ec3a56 100644 --- a/src/main/java/org/parabot/core/ui/BotUI.java +++ b/src/main/java/org/parabot/core/ui/BotUI.java @@ -150,7 +150,15 @@ public class BotUI extends JFrame implements ActionListener, ComponentListener, Robot robot = new Robot(); Rectangle parabotScreen = new Rectangle((int) getLocation().getX(), (int) getLocation().getY(), getWidth(), getHeight()); BufferedImage image = robot.createScreenCapture(parabotScreen); - ImageIO.write(image, "png", new File(Directories.getScreenshotDir().getPath() + "/" + StringUtils.randomString(10) + ".png")); + String randString = StringUtils.randomString(10); + for (int x = 0; x < Directories.getScreenshotDir().listFiles().length; x++) { + if (Directories.getScreenshotDir().listFiles()[x].getAbsoluteFile().getName().contains(randString)) { + randString = randString + StringUtils.randomString(4); + } + } + File file = new File(Directories.getScreenshotDir().getPath() + "/" + randString + ".png"); + ImageIO.write(image, "png", file); + } catch (IOException | AWTException k) { k.printStackTrace(); }