/**
* Copyright 2007 Google Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.tonicsystems.jarjar.util;
import com.tonicsystems.jarjar.transform.Transformable;
import com.tonicsystems.jarjar.transform.jar.JarProcessor;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashSet;
import java.util.Set;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.taskdefs.Jar;
import org.apache.tools.ant.types.ZipFileSet;
import org.apache.tools.zip.JarMarker;
import org.apache.tools.zip.ZipExtraField;
import org.apache.tools.zip.ZipOutputStream;
public abstract class AntJarProcessor extends Jar {
private final Transformable struct = new Transformable();
private JarProcessor proc;
private final byte[] buf = new byte[0x2000];
private final Set<String> dirs = new HashSet<String>();
private boolean filesOnly;
protected boolean verbose;
private static final ZipExtraField[] JAR_MARKER = new ZipExtraField[]{
JarMarker.getInstance()
};
public void setVerbose(boolean verbose) {
this.verbose = verbose;
}
@Override
public abstract void execute() throws BuildException;
public void execute(JarProcessor proc) throws BuildException {
this.proc = proc;
super.execute();
}
@Override
public void setFilesonly(boolean f) {
super.setFilesonly(f);
filesOnly = f;
}
@Override
protected void zipDir(File dir, ZipOutputStream zOut, String vPath, int mode)
throws IOException {
}
// TODO: Rewrite this entirely.
@Override
protected void zipFile(InputStream is, ZipOutputStream zOut, String vPath,
long lastModified, File fromArchive, int mode) throws IOException {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
IoUtil.copy(is, baos, buf);
struct.data = baos.toByteArray();
struct.name = vPath;
struct.time = lastModified;
if (proc.process(struct) != JarProcessor.Result.DISCARD) {
if (mode == 0)
mode = ZipFileSet.DEFAULT_FILE_MODE;
if (!filesOnly) {
addParentDirs(struct.name, zOut);
}
super.zipFile(new ByteArrayInputStream(struct.data),
zOut, struct.name, struct.time, fromArchive, mode);
}
}
private void addParentDirs(String file, ZipOutputStream zOut) throws IOException {
int slash = file.lastIndexOf('/');
if (slash >= 0) {
String dir = file.substring(0, slash);
if (dirs.add(dir)) {
addParentDirs(dir, zOut);
super.zipDir((File) null, zOut, dir + "/", ZipFileSet.DEFAULT_DIR_MODE, JAR_MARKER);
}
}
}
@Override
public void reset() {
super.reset();
cleanHelper();
}
@Override
protected void cleanUp() {
super.cleanUp();
cleanHelper();
}
protected void cleanHelper() {
verbose = false;
filesOnly = false;
dirs.clear();
}
}