sascha@480: package de.intevation.gnv.utils;
sascha@480: 
sascha@480: import java.io.BufferedInputStream;
sascha@480: import java.io.File;
sascha@480: import java.io.FileInputStream;
sascha@480: import java.io.IOException;
sascha@480: import java.io.InputStream;
sascha@480: import java.io.OutputStream;
sascha@480: 
sascha@480: import java.util.Stack;
sascha@480: 
sascha@480: import java.util.zip.ZipEntry;
sascha@480: import java.util.zip.ZipOutputStream;
sascha@480: 
sascha@480: /**
sascha@480:  * @author Sascha L. Teichmann (sascha.teichmann@intevation.de)
sascha@480:  */
sascha@480: public final class FileUtils
sascha@480: {
sascha@480:     private FileUtils() {
sascha@480:     }
sascha@480: 
sascha@480:     public final static boolean deleteRecursive(File file) {
sascha@480: 
sascha@480:         if (file == null) {
sascha@480:             return false;
sascha@480:         }
sascha@480: 
sascha@480:         if (file.isDirectory()) {
sascha@480:             File [] files = file.listFiles();
sascha@480:             if (files != null) {
sascha@480:                 for (File sub: files) {
sascha@480:                     if (!deleteRecursive(sub)) {
sascha@480:                         return false;
sascha@480:                     }
sascha@480:                 }
sascha@480:             }
sascha@480:         }
sascha@480: 
sascha@480:         return file.delete();
sascha@480:     }
sascha@480: 
sascha@480:     public static void createZipArchive(
sascha@480:         File         file,
sascha@480:         OutputStream outputStream
sascha@480:     )
sascha@480:     throws IOException
sascha@480:     {
sascha@480:         ZipOutputStream out = new ZipOutputStream(outputStream);
sascha@480: 
sascha@480:         if (file.isFile()) {
sascha@480:             copyFileToZip("", file, out);
sascha@480:         }
sascha@480:         else if (file.isDirectory()) {
sascha@480: 
sascha@480:             Stack<PrefixDir> stack = new Stack<PrefixDir>();
sascha@480:             stack.push(new PrefixDir(file.getName() + "/", file));
sascha@480: 
sascha@480:             while (!stack.isEmpty()) {
sascha@480:                 PrefixDir pd = stack.pop();
sascha@480: 
sascha@480:                 ZipEntry dirEntry = new ZipEntry(pd.prefix);
sascha@480:                 out.putNextEntry(dirEntry);
sascha@480:                 out.closeEntry();
sascha@480: 
sascha@480:                 File [] files = pd.dir.listFiles();
sascha@480:                 if (files != null) {
sascha@480:                     for (File sub: files) {
sascha@480:                         if (sub.isDirectory()) {
sascha@480:                             stack.push(new PrefixDir(
sascha@480:                                 pd.prefix + sub.getName() + "/",
sascha@480:                                 sub));
sascha@480:                         }
sascha@480:                         else if (sub.isFile()) {
sascha@480:                             copyFileToZip(pd.prefix, sub, out);
sascha@480:                         }
sascha@480:                     }
sascha@480:                 }
sascha@480:             }
sascha@480:         }
sascha@480: 
sascha@480:         out.finish();
sascha@480:     }
sascha@480: 
sascha@480:     private static final class PrefixDir {
sascha@480: 
sascha@480:         String prefix;
sascha@480:         File   dir;
sascha@480: 
sascha@480:         public PrefixDir(String prefix, File dir) {
sascha@480:             this.prefix = prefix;
sascha@480:             this.dir    = dir;
sascha@480:         }
sascha@480: 
sascha@480:     } // class PrefixDir
sascha@480: 
sascha@480:     private static void copyFileToZip(
sascha@480:         String          prefix,
sascha@480:         File            file,
sascha@480:         ZipOutputStream out
sascha@480:     )
sascha@480:     throws IOException
sascha@480:     {
sascha@480:         String   entryName = prefix + file.getName();
sascha@480:         ZipEntry entry     = new ZipEntry(entryName);
sascha@480:         out.putNextEntry(entry);
sascha@480:         InputStream in = null;
sascha@480:         try {
sascha@480:             in =
sascha@480:                 new BufferedInputStream(
sascha@480:                 new FileInputStream(file), 20*1024);
sascha@480: 
sascha@480:             byte [] buf = new byte[2048];
sascha@480: 
sascha@480:             int r;
sascha@480:             while ((r = in.read(buf)) > 0) {
sascha@480:                 out.write(buf, 0, r);
sascha@480:             }
sascha@480:         }
sascha@480:         finally {
sascha@480:             if (in != null) {
sascha@480:                 try { in.close(); }
sascha@480:                 catch (IOException ioe) {}
sascha@480:             }
sascha@480:         }
sascha@480:         out.closeEntry();
sascha@480:     }
sascha@480: }
sascha@480: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :