filebot/source/net/sourceforge/tuned/TemporaryFolder.java

package net.sourceforge.tuned;


import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


public class TemporaryFolder {
	
	private static final String tmpdir = System.getProperty("java.io.tmpdir");
	
	private static final Map<String, TemporaryFolder> folders = new HashMap<String, TemporaryFolder>();
	
	
	public static TemporaryFolder getFolder(String name) {
		synchronized (folders) {
			TemporaryFolder folder = folders.get(name);
			
			if (folder == null) {
				folder = new TemporaryFolder(new File(tmpdir, name));
				folders.put(name, folder);
			}
			
			return folder;
		}
	}
	
	/**
	 * Delete all temporary folders on shutdown
	 */
	static {
		Runtime.getRuntime().addShutdownHook(new Thread("TemporaryFolder Cleanup") {
			
			@Override
			public void run() {
				synchronized (folders) {
					for (TemporaryFolder folder : folders.values()) {
						folder.delete();
					}
				}
			}
		});
	}
	
	private final File root;
	
	
	private TemporaryFolder(File root) {
		this.root = root;
	}
	

	/**
	 * Create an empty file in this temporary folder
	 * 
	 * @param name name of the file
	 * @return newly created file
	 * @throws IOException if an I/O error occurred
	 */
	public File createFile(String name) throws IOException {
		if (!root.exists()) {
			root.mkdir();
		}
		
		File file = new File(root, name);
		file.createNewFile();
		
		return file;
	}
	

	/**
	 * Creates an empty file in this temporary folder, using the given prefix and suffix to
	 * generate its name.
	 * 
	 * @param prefix The prefix string to be used in generating the file's name; must be at
	 *            least three characters long
	 * @param suffix The suffix string to be used in generating the file's name; may be null,
	 *            in which case the suffix ".tmp" will be used
	 * @return An abstract pathname denoting a newly-created empty file
	 * @throws IOException If a file could not be created
	 * @see File#createTempFile(String, String)
	 */
	public File createFile(String prefix, String suffix) throws IOException {
		if (!root.exists()) {
			root.mkdir();
		}
		
		return File.createTempFile(prefix, suffix, root);
	}
	

	public boolean deleteFile(String name) {
		return new File(root, name).delete();
	}
	

	/**
	 * Retrieve the {@link File} object for this {@link TemporaryFolder}
	 * 
	 * @return the {@link File} object for this {@link TemporaryFolder}
	 */
	public File getFolder() {
		return root;
	}
	

	public TemporaryFolder createFolder(String name) {
		if (!root.exists()) {
			root.mkdir();
		}
		
		TemporaryFolder folder = new TemporaryFolder(new File(root, name));
		
		folder.root.mkdir();
		
		return folder;
	}
	

	public List<File> list(boolean recursive) {
		List<File> list = new ArrayList<File>();
		
		list(root, list, recursive);
		
		return list;
	}
	

	private void list(File file, List<File> list, boolean recursive) {
		if (file.isDirectory()) {
			for (File entry : file.listFiles()) {
				if (entry.isDirectory()) {
					if (recursive) {
						list(entry, list, recursive);
					}
				} else {
					list.add(entry);
				}
			}
		}
	}
	

	public void delete() {
		delete(root);
	}
	

	/**
	 * Delete files/folders recursively
	 * 
	 * @param file file/folder that will be deleted
	 */
	private void delete(File file) {
		if (file.isDirectory()) {
			for (File entry : file.listFiles()) {
				delete(entry);
			}
		}
		
		file.delete();
	}
	
}