ZipFileType.java from Magellan-Client at Krugle

Show ZipFileType.java syntax highlighted

/*
 *  Copyright (C) 2000-2004 Roger Butenuth, Andreas Gampe,
 *                          Stefan Goetz, Sebastian Pappert,
 *                          Klaas Prause, Enno Rehling,
 *                          Sebastian Tusk, Ulrich Kuester,
 *                          Ilja Pavkovic
 *
 * This file is part of the Eressea Java Code Base, see the
 * file LICENSING for the licensing information applying to
 * this file.
 *
 */

package com.eressea.io.file;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Enumeration;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

/**
 * TODO: DOCUMENT ME!
 *
 * @author $author$
 * @version $Revision: 305 $
 */
public class ZipFileType extends FileType {
	// TODO: ZipEntry may also be a "normal" FileType
	protected ZipEntry zipentry = null;

	ZipFileType(File aFile, boolean readonly, ZipEntry aEntry) throws IOException {
		super(aFile, readonly);

		if(aEntry == null) {
			throw new FileTypeFactory.NoValidEntryException();
		}

		zipentry = new ZipEntry(aEntry);
	}

	/**
	 * Returns the most inner name of the FileType. Will be overwritten in ZipFileType
	 *
	 * @return the most inner name of a FileType.
	 */
	public String getInnerName() {
		return zipentry.getName();
	}

	/**
	 * Returns all files inside the zip ending with one of  the given endings case insensitive
	 *
	 * @param zip a ZipFile to inspect
	 * @param endings an array of valid file name endings.
	 *
	 * @return an array of valid zip file entries
	 */
	public static ZipEntry[] getZipEntries(ZipFile zip, String endings[]) {
		Collection ret = new ArrayList();

		for(Enumeration iter = zip.entries(); iter.hasMoreElements();) {
			ZipEntry entry = (ZipEntry) iter.nextElement();
			String entryName = entry.getName();

			for(int i = 0; i < endings.length; i++) {
				if(entryName.toLowerCase().endsWith(endings[i])) {
					ret.add(entry);

					break;
				}
			}
		}

		return (ZipEntry[]) ret.toArray(new ZipEntry[] {  });
	}

	protected InputStream createInputStream() throws IOException {
		InputStream is = new ZipFile(filename).getInputStream(zipentry);

		if(is == null) {
			throw new IOException("Cannot read zip entry '" + zipentry + "' in file '" + filename +
								  "',");
		}

		return is;
	}

	protected OutputStream createOutputStream() throws IOException {
		// here we need to do something special: all entries are copied expect the named zipentry, which will be overwritten
		File tmpfile = CopyFile.copy(filename);
		ZipFile zfile = new ZipFile(tmpfile);
		ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(filename));

		for(Enumeration e = zfile.entries(); e.hasMoreElements();) {
			ZipEntry oldEntry = (ZipEntry) e.nextElement();

			if(!oldEntry.getName().equals(zipentry.getName())) {
				// do not reuse oldEntry but create a new ZipEntry
				zos.putNextEntry(new ZipEntry(oldEntry.getName()));

				InputStream currIn=zfile.getInputStream(oldEntry);
				CopyFile.copyStreams(currIn, zos);
				currIn.close();
			}
		}

		// do not reuse oldEntry but create a new ZipEntry
		zos.putNextEntry(new ZipEntry(zipentry.getName()));

		return zos;
	}
}
See more files for this project here