/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.geronimo.system.plugin;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.geronimo.gbean.GBeanInfo;
import org.apache.geronimo.gbean.GBeanInfoBuilder;
import org.apache.geronimo.kernel.config.IOUtil;
import org.apache.geronimo.kernel.repository.Artifact;
import org.apache.geronimo.system.serverinfo.ServerInfo;
import org.codehaus.plexus.archiver.Archiver;
import org.codehaus.plexus.archiver.ArchiverException;
import org.codehaus.plexus.archiver.UnixStat;
import org.codehaus.plexus.archiver.tar.TarArchiver;
import org.codehaus.plexus.archiver.tar.TarLongFileMode;
import org.codehaus.plexus.archiver.zip.ZipArchiver;
//maven inconsistency -- if we can use 1.0-alpha-9 uncomment
//import org.codehaus.plexus.archiver.util.DefaultFileSet;
/**
* @version $Rev: 663681 $ $Date: 2008-06-06 01:58:00 +0800 (Fri, 06 Jun 2008) $
*/
public class ArchiverGBean implements ServerArchiver {
private final ServerInfo serverInfo;
private List<String> excludes = new ArrayList<String>();
public ArchiverGBean(ServerInfo serverInfo) {
this.serverInfo = serverInfo;
}
public void addExclude(String pattern) {
this.excludes.add(pattern);
}
public void removeExclude(String pattern) {
this.excludes.remove(pattern);
}
private void removeExcludes(File source, Map<String, File> all) {
Map<String, File> matches = new HashMap<String, File>();
for (String exclude : this.excludes) {
IOUtil.find(source, exclude, matches);
}
for (String exclude : matches.keySet()) {
all.remove(exclude);
}
}
public File archive(String sourcePath, String destPath, Artifact artifact) throws ArchiverException, IOException {
File source = serverInfo.resolve(sourcePath);
File dest = serverInfo.resolve(destPath);
String serverName = artifact.getArtifactId() + "-" + artifact.getVersion();
dest = new File(dest, serverName + "-bin." + artifact.getType());
Archiver archiver;
if ("tar.gz".equals(artifact.getType())) {
archiver = new TarArchiver();
TarArchiver.TarCompressionMethod tarCompressionMethod = new TarArchiver.TarCompressionMethod();
tarCompressionMethod.setValue("gzip");
((TarArchiver) archiver).setCompression(tarCompressionMethod);
TarLongFileMode fileMode = new TarLongFileMode();
fileMode.setValue(TarLongFileMode.GNU);
((TarArchiver) archiver).setLongfile(fileMode);
} else if ("zip".equals(artifact.getType())) {
archiver = new ZipArchiver();
} else {
throw new IllegalArgumentException("Unknown target type: " + artifact.getType());
}
archiver.setIncludeEmptyDirs(true);
archiver.setDestFile(dest);
/* see if using plexus-archiver 1.0-alpha-7 same as maven lets us share code. Following is for 1.0-alpha-9
DefaultFileSet all = new DefaultFileSet();
all.setDirectory(source);
archiver.addFileSet(all);
*/
// add in all files and mark them with default file permissions
Map<File, Boolean> emptyDirs = new HashMap<File, Boolean>();
Map<String, File> all = IOUtil.listAllFileNames(source);
removeExcludes(source, all);
for (Map.Entry<String, File> entry : all.entrySet()) {
String destFileName = serverName + "/" + entry.getKey();
File sourceFile = entry.getValue();
if (sourceFile.isFile()) {
archiver.addFile(sourceFile, destFileName, UnixStat.DEFAULT_FILE_PERM);
// mark parent directories non-empty
for (File parentDir = sourceFile.getParentFile();
parentDir != null && !parentDir.equals(source);
parentDir = parentDir.getParentFile()) {
emptyDirs.put(parentDir, Boolean.FALSE);
}
} else if (sourceFile.isDirectory()) {
Boolean isEmpty = emptyDirs.get(sourceFile);
if (isEmpty == null) {
emptyDirs.put(sourceFile, Boolean.TRUE);
// mark parent directories non-empty
for (File parentDir = sourceFile.getParentFile();
parentDir != null && !parentDir.equals(source);
parentDir = parentDir.getParentFile()) {
emptyDirs.put(parentDir, Boolean.FALSE);
}
}
}
}
if (!all.isEmpty()) {
emptyDirs.put(source, Boolean.FALSE);
}
String sourceDirPath = source.getAbsolutePath();
for (Map.Entry<File, Boolean> entry : emptyDirs.entrySet()) {
if (entry.getValue().booleanValue()) {
String emptyDirPath = entry.getKey().getAbsolutePath();
String relativeDir = emptyDirPath.substring(sourceDirPath.length());
relativeDir = relativeDir.replace('\\', '/');
archiver.addDirectory(entry.getKey(), serverName + relativeDir);
}
}
emptyDirs.clear();
all.clear();
// add execute permissions to all non-batch files in the bin/ directory
File bin = new File(source, "bin");
if (bin.exists()) {
Map<String, File> includes = IOUtil.listAllFileNames(bin);
for (Map.Entry<String, File> entry : includes.entrySet()) {
String destFileName = serverName + "/bin/" + entry.getKey();
File sourceFile = entry.getValue();
if (!destFileName.endsWith(".bat") && sourceFile.isFile()) {
archiver.addFile(sourceFile, destFileName, UnixStat.DEFAULT_DIR_PERM);
}
}
}
archiver.createArchive();
return dest;
}
public static final GBeanInfo GBEAN_INFO;
static {
GBeanInfoBuilder infoFactory = GBeanInfoBuilder.createStatic(ArchiverGBean.class);
infoFactory.addReference("ServerInfo", ServerInfo.class, "GBean");
infoFactory.setConstructor(new String[]{"ServerInfo"});
GBEAN_INFO = infoFactory.getBeanInfo();
}
public static GBeanInfo getGBeanInfo() {
return GBEAN_INFO;
}
}