allow specifying a file extension for getTempFile LDEV-2331 (#1148)

* Added a fix for LDEV-2331 * Update GetTempFile.java
lucee · Apr 1, 2021 · 6491aee · 6491aee
1 parent 7cf1711
commit 6491aee
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/core/src/main/java/lucee/runtime/functions/system/GetTempFile.java b/core/src/main/java/lucee/runtime/functions/system/GetTempFile.java
@@ -33,12 +33,18 @@
 
 public final class GetTempFile implements Function {
 	public static String call(PageContext pc, String strDir, String prefix) throws PageException {
+		return call(pc, strDir, prefix, ".tmp");
+	}
+
+	public static String call(PageContext pc, String strDir, String prefix, String extension) throws PageException {
 		Resource dir = ResourceUtil.toResourceExisting(pc, strDir);
 		pc.getConfig().getSecurityManager().checkFileLocation(dir);
 		if (!dir.isDirectory()) throw new ExpressionException("[" + strDir + "] is not a directory");
 		int count = 1;
 		Resource file;
-		while ((file = dir.getRealResource(prefix + pc.getId() + count + ".tmp")).exists()) {
+		if (extension.trim().isEmpty() == true) { extension = ".tmp"; }
+		if (extension.charAt(0) != '.') { extension = "." + extension; }
+		while ((file = dir.getRealResource(prefix + pc.getId() + count + extension)).exists()) {
 			count++;
 		}
 		try {

diff --git a/core/src/main/java/resource/fld/core-base.fld b/core/src/main/java/resource/fld/core-base.fld
@@ -6582,6 +6582,14 @@ if this attribute is not provided all types are returned.</description>
 			<required>Yes</required>
 		<description>Prefix of a temporary file to create in the dir directory</description>
     </argument>
+	<argument>
+    		<name>extension</name>
+			<alias>ext</alias>
+    		<type>string</type>
+			<required>No</required>
+			<Default>tmp</Default>
+		<description>Creates temporary files with user defined extension, default is .tmp</description>
+	</argument>
 		<return>
 			<type>string</type>
 		</return>