/*
* JBoss, Home of Professional Open Source.
* Copyright 2012, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.patching.runner;
import static org.jboss.as.patching.IoUtils.NO_CONTENT;
import java.io.File;
import java.io.IOException;
import org.jboss.as.patching.HashUtils;
import org.jboss.as.patching.IoUtils;
import org.jboss.as.patching.logging.PatchLogger;
import org.jboss.as.patching.metadata.BundleItem;
import org.jboss.as.patching.metadata.ContentModification;
import org.jboss.as.patching.metadata.ModificationType;
/**
* @author Emanuel Muckenhuber
*/
class BundlePatchingTask extends AbstractPatchingTask<BundleItem> {
public BundlePatchingTask(PatchingTaskDescription description) {
super(description, BundleItem.class);
}
@Override
byte[] backup(PatchingTaskContext context) throws IOException {
// Check the bundle dir hash
final File[] repoRoots = context.getTargetBundlePath();
final String bundleName = contentItem.getName();
final String slot = contentItem.getSlot();
for(final File path : repoRoots) {
// Check the bundle path
final File bundlePath = PatchContentLoader.getModulePath(path, bundleName, slot);
if(bundlePath.exists()) {
PatchLogger.ROOT_LOGGER.debugf("found in path (%s)", bundlePath.getAbsolutePath());
// Bundles don't contain a modules.xml
final File[] children = bundlePath.listFiles();
if(children == null || children.length == 0) {
return NO_CONTENT;
}
context.invalidateRoot(bundlePath);
return HashUtils.hashFile(bundlePath);
}
}
return NO_CONTENT;
}
@Override
byte[] apply(PatchingTaskContext context, PatchContentLoader loader) throws IOException {
if (context.getCurrentMode() == PatchingTaskContext.Mode.ROLLBACK) {
return getContentItem().getContentHash();
}
// Copy the new bundle resources to the patching directory
final File targetDir = context.getTargetFile(contentItem);
final File sourceDir = loader.getFile(contentItem);
final File[] moduleResources = sourceDir.listFiles();
if(! targetDir.mkdirs() && ! targetDir.exists()) {
throw PatchLogger.ROOT_LOGGER.cannotCreateDirectory(targetDir.getAbsolutePath());
}
if(moduleResources == null || moduleResources.length == 0) {
return NO_CONTENT;
}
for(final File file : moduleResources) {
final File target = new File(targetDir, file.getName());
IoUtils.copy(file, target);
}
return contentItem.getContentHash();
}
@Override
ContentModification createRollbackEntry(ContentModification original, byte[] targetHash, byte[] itemHash) {
final BundleItem item = new BundleItem(contentItem.getName(), contentItem.getSlot(), itemHash);
switch (original.getType()) {
case ADD:
return new ContentModification(item, targetHash, ModificationType.REMOVE, original.getCondition());
case REMOVE:
return new ContentModification(item, targetHash, ModificationType.ADD, original.getCondition());
default:
return new ContentModification(item, targetHash, ModificationType.MODIFY, original.getCondition());
}
}
}