package de.jamba.hudson.plugin.wsclean;
import hudson.Extension;
import hudson.FilePath;
import hudson.Launcher;
import hudson.model.AbstractBuild;
import hudson.model.AbstractProject;
import hudson.model.BuildListener;
import hudson.model.Hudson;
import hudson.model.Label;
import hudson.model.Node;
import hudson.model.Slave;
import hudson.remoting.RequestAbortedException;
import hudson.remoting.VirtualChannel;
import hudson.tasks.BuildWrapper;
import hudson.tasks.BuildWrapperDescriptor;
import java.io.File;
import java.io.IOException;
import java.util.Set;
import org.kohsuke.stapler.DataBoundConstructor;
public class PrePostClean extends BuildWrapper {
public boolean before;
private boolean behind;
public boolean isBefore() {
return before;
}
public void setBefore(boolean before) {
this.before = before;
this.behind = !before;
}
@DataBoundConstructor
public PrePostClean(boolean before) {
this.before = before;
this.behind = !before;
}
@SuppressWarnings("unchecked")
@Override
public Environment setUp(AbstractBuild build, Launcher launcher,
BuildListener listener) throws IOException, InterruptedException {
// TearDown
class TearDownImpl extends Environment {
@Override
public boolean tearDown(AbstractBuild build, BuildListener listener)
throws IOException, InterruptedException {
if (behind) {
executeOnSlaves(build, listener);
}
return super.tearDown(build, listener);
}
}
if (before) {
executeOnSlaves(build, listener);
}
return new TearDownImpl();
}
@SuppressWarnings("unchecked")
private void executeOnSlaves(AbstractBuild build, BuildListener listener) {
listener.getLogger().println("run PrePostClean");
// select actual running label
String runNode = build.getBuiltOnStr();
if (runNode.length() == 0) {
listener.getLogger().println("running on master");
} else {
listener.getLogger().println("running on " + runNode);
}
Label assignedLabel = build.getProject().getAssignedLabel();
if (assignedLabel == null) {
listener.getLogger().println("skipping roaming projects.");
return;
}
Set<Node> usedNodes = assignedLabel.getNodes();
if (usedNodes != null) {
for (Node node : usedNodes) {
if (!runNode.equals(node.getNodeName())) {
if (node.getNodeName().length() == 0) {
listener.getLogger().println("clean on master");
deleteOnMaster(build, listener);
} else {
listener.getLogger().println(
"clean on " + node.getNodeName());
deleteRemote(build, listener, node);
}
}
}
}
}
@SuppressWarnings("unchecked")
private void deleteOnMaster(AbstractBuild build, BuildListener listener) {
if (((Node) Hudson.getInstance()).getNumExecutors() > 0) {
FilePath fp = new FilePath(new File(Hudson.getInstance()
.getRootPath()
+ "/jobs/" + build.getProject().getName() + "/workspace"));
try {
fp.deleteContents();
} catch (IOException e) {
listener.getLogger().println(
"cat delete on Master " + e.getMessage());
listener.getLogger().print(e);
} catch (InterruptedException e) {
listener.getLogger().println(
"cat delete on Master " + e.getMessage());
listener.getLogger().print(e);
}
}
}
@SuppressWarnings({"unchecked","deprecation"})
private void deleteRemote(AbstractBuild build, BuildListener listener,
Node node) {
VirtualChannel vc = ((Slave) node).getComputer().getChannel();
if (!((Slave) node).getComputer().isConnecting()
&& !((Slave) node).getComputer().isTemporarilyOffline()) {
((Slave) node).getComputer().connect(true);
int i = 0;
while (vc == null && ++i < 120) {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
listener.getLogger().println(e.getMessage());
}
vc = ((Slave) node).getComputer().getChannel();
}
}
if (vc != null) {
FilePath fp = new FilePath(vc, ((Slave) node).getRemoteFS()
+ "/workspace/" + build.getProject().getName());
try {
fp.deleteContents();
} catch (IOException e) {
listener.getLogger().println(
"cat delete on Slave " + e.getMessage());
listener.getLogger().print(e);
} catch (InterruptedException e) {
listener.getLogger().println(
"cat delete on Slave " + e.getMessage());
listener.getLogger().print(e);
} catch (RequestAbortedException e){
listener.getLogger().println(
"cat delete on Slave " + e.getMessage());
}
} else {
listener.getLogger().println(
"no deleteChannel on " + node.getNodeName());
}
}
@Extension
public static final class DescriptorImpl extends BuildWrapperDescriptor {
public DescriptorImpl() {
super(PrePostClean.class);
}
public String getDisplayName() {
return "CleanUp all other workspaces in the same slavegroup";
}
public boolean isApplicable(AbstractProject<?, ?> item) {
return true;
}
}
}