diff --git a/API/src/main/java/org/sikuli/script/support/Commons.java b/API/src/main/java/org/sikuli/script/support/Commons.java index cc98d880e..126db066c 100644 --- a/API/src/main/java/org/sikuli/script/support/Commons.java +++ b/API/src/main/java/org/sikuli/script/support/Commons.java @@ -38,6 +38,7 @@ import java.nio.ByteBuffer; import java.nio.IntBuffer; import java.nio.file.*; +import java.nio.file.FileSystem; import java.security.CodeSource; import java.util.List; import java.util.*; @@ -1435,6 +1436,24 @@ public static File asFile(Object path, String option) { // // + public static boolean isValidImageFilename(String fname) { + String validEndings = ".png.jpg.jpeg"; + String ending = ""; + try { + ending = FilenameUtils.getExtension(fname); + } catch(IllegalArgumentException e) { + return false; + } + return !ending.isEmpty() && validEndings.contains(ending.toLowerCase()); + } + + public static String getValidImageFilename(String fname) { + if (isValidImageFilename(fname)) { + return fname; + } + return fname + ".png"; + } + public static String jnaPathAdd(String sFolder) { String jnaPath = System.getProperty("jna.library.path"); if (null == jnaPath) {