X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2FIOUtils.java;h=bf0686babca9e7addfbbd810a7a8e4d5fd66756d;hb=59864f77f0b8d5a57f479152a3131093544af6b2;hp=973a61b0da541362a9a44069555bbe42a5adfb45;hpb=530d4062471346d6ececf76d74a0358c91323998;p=fanfix.git
diff --git a/src/be/nikiroo/utils/IOUtils.java b/src/be/nikiroo/utils/IOUtils.java
index 973a61b..bf0686b 100644
--- a/src/be/nikiroo/utils/IOUtils.java
+++ b/src/be/nikiroo/utils/IOUtils.java
@@ -1,6 +1,7 @@
package be.nikiroo.utils;
import java.io.BufferedReader;
+import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
@@ -9,6 +10,8 @@ import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
@@ -208,6 +211,41 @@ public class IOUtils {
return builder.toString();
}
+ /**
+ * Recursively delete the given {@link File}, which may of course also be a
+ * directory.
+ *
+ * Will either silently continue or throw an exception in case of error,
+ * depending upon the parameters.
+ *
+ * @param target
+ * the target to delete
+ * @param exception
+ * TRUE to throw an {@link IOException} in case of error, FALSE
+ * to silently continue
+ *
+ * @return TRUE if all files were deleted, FALSE if an error occurred
+ *
+ * @throws IOException
+ * if an error occurred and the parameters allow an exception to
+ * be thrown
+ */
+ public static boolean deltree(File target, boolean exception)
+ throws IOException {
+ List list = deltree(target, null);
+ if (exception && !list.isEmpty()) {
+ String slist = "";
+ for (File file : list) {
+ slist += "\n" + file.getPath();
+ }
+
+ throw new IOException("Cannot delete all the files from: <" //
+ + target + ">:" + slist);
+ }
+
+ return list.isEmpty();
+ }
+
/**
* Recursively delete the given {@link File}, which may of course also be a
* directory.
@@ -216,18 +254,44 @@ public class IOUtils {
*
* @param target
* the target to delete
+ *
+ * @return TRUE if all files were deleted, FALSE if an error occurred
*/
- public static void deltree(File target) {
+ public static boolean deltree(File target) {
+ return deltree(target, null).isEmpty();
+ }
+
+ /**
+ * Recursively delete the given {@link File}, which may of course also be a
+ * directory.
+ *
+ * Will collect all {@link File} that cannot be deleted in the given
+ * accumulator.
+ *
+ * @param target
+ * the target to delete
+ * @param errorAcc
+ * the accumulator to use for errors, or NULL to create a new one
+ *
+ * @return the errors accumulator
+ */
+ public static List deltree(File target, List errorAcc) {
+ if (errorAcc == null) {
+ errorAcc = new ArrayList();
+ }
+
File[] files = target.listFiles();
if (files != null) {
for (File file : files) {
- deltree(file);
+ errorAcc = deltree(file, errorAcc);
}
}
if (!target.delete()) {
- System.err.println("Cannot delete: " + target.getAbsolutePath());
+ errorAcc.add(target);
}
+
+ return errorAcc;
}
/**
@@ -246,4 +310,69 @@ public class IOUtils {
return loader.getResourceAsStream(name);
}
+
+ /**
+ * Return a resetable {@link InputStream} from this stream, and reset it.
+ *
+ * @param in
+ * the input stream
+ * @return the resetable stream, which may be the same
+ *
+ * @throws IOException
+ * in case of I/O error
+ */
+ public static InputStream forceResetableStream(InputStream in)
+ throws IOException {
+ MarkableFileInputStream tmpIn = null;
+ File tmp = null;
+
+ boolean resetable = in.markSupported();
+ if (resetable) {
+ try {
+ in.reset();
+ } catch (IOException e) {
+ resetable = false;
+ }
+ }
+
+ if (resetable) {
+ return in;
+ }
+
+ tmp = File.createTempFile(".tmp-stream", ".tmp");
+ try {
+ write(in, tmp);
+ tmpIn = new MarkableFileInputStream(new FileInputStream(tmp));
+ return tmpIn;
+ } finally {
+ try {
+ if (tmpIn != null) {
+ tmpIn.close();
+ }
+ } finally {
+ tmp.delete();
+ }
+ }
+ }
+
+ /**
+ * Convert the {@link InputStream} into a byte array.
+ *
+ * @param in
+ * the input stream
+ *
+ * @return the array
+ *
+ * @throws IOException
+ * in case of I/O error
+ */
+ public static byte[] toByteArray(InputStream in) throws IOException {
+ ByteArrayOutputStream out = new ByteArrayOutputStream();
+ write(in, out);
+
+ byte[] array = out.toByteArray();
+ out.close();
+
+ return array;
+ }
}