/*
* Copyright 2001-2004 The Apache Software Foundation
*
* 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 org.apache.tools.ant.taskdefs.optional;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.util.Enumeration;
import java.util.Vector;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.Task;
import org.apache.tools.ant.taskdefs.Execute;
import org.apache.tools.ant.taskdefs.ExecuteStreamHandler;
import org.apache.tools.ant.taskdefs.LogOutputStream;
import org.apache.tools.ant.taskdefs.LogStreamHandler;
import org.apache.tools.ant.taskdefs.PumpStreamHandler;
import org.apache.tools.ant.taskdefs.condition.Os;
import org.apache.tools.ant.types.Commandline;
import org.apache.tools.ant.types.Path;
/**
* Invokes the rpm tool to build a Linux installation file.
*
* @author lucas@collab.net
*/
public class Rpm extends Task {
/**
* the spec file
*/
private String specFile;
/**
* the rpm top dir
*/
private File topDir;
/**
* the rpm command to use
*/
private String command = "-bb";
/**
* The executable to use for building the packages.
* @since Ant 1.6
*/
private String rpmBuildCommand = null;
/**
* clean BUILD directory
*/
private boolean cleanBuildDir = false;
/**
* remove spec file
*/
private boolean removeSpec = false;
/**
* remove sources
*/
private boolean removeSource = false;
/**
* the file to direct standard output from the command
*/
private File output;
/**
* the file to direct standard error from the command
*/
private File error;
/**
* Execute the task
*
* @throws BuildException is there is a problem in the task execution.
*/
public void execute() throws BuildException {
Commandline toExecute = new Commandline();
toExecute.setExecutable(rpmBuildCommand == null
? guessRpmBuildCommand()
: rpmBuildCommand);
if (topDir != null) {
toExecute.createArgument().setValue("--define");
toExecute.createArgument().setValue("_topdir" + topDir);
}
toExecute.createArgument().setLine(command);
if (cleanBuildDir) {
toExecute.createArgument().setValue("--clean");
}
if (removeSpec) {
toExecute.createArgument().setValue("--rmspec");
}
if (removeSource) {
toExecute.createArgument().setValue("--rmsource");
}
toExecute.createArgument().setValue("SPECS/" + specFile);
ExecuteStreamHandler streamhandler = null;
OutputStream outputstream = null;
OutputStream errorstream = null;
if (error == null && output == null) {
streamhandler = new LogStreamHandler(this, Project.MSG_INFO,
Project.MSG_WARN);
} else {
if (output != null) {
try {
BufferedOutputStream bos
= new BufferedOutputStream(new FileOutputStream(output));
outputstream = new PrintStream(bos);
} catch (IOException e) {
throw new BuildException(e, getLocation());
}
} else {
outputstream = new LogOutputStream(this, Project.MSG_INFO);
}
if (error != null) {
try {
BufferedOutputStream bos
= new BufferedOutputStream(new FileOutputStream(error));
errorstream = new PrintStream(bos);
} catch (IOException e) {
throw new BuildException(e, getLocation());
}
} else {
errorstream = new LogOutputStream(this, Project.MSG_WARN);
}
streamhandler = new PumpStreamHandler(outputstream, errorstream);
}
Execute exe = new Execute(streamhandler, null);
exe.setAntRun(getProject());
if (topDir == null) {
topDir = getProject().getBaseDir();
}
exe.setWorkingDirectory(topDir);
exe.setCommandline(toExecute.getCommandline());
try {
exe.execute();
log("Building the RPM based on the " + specFile + " file");
} catch (IOException e) {
throw new BuildException(e, getLocation());
} finally {
if (output != null) {
try {
outputstream.close();
} catch (IOException e) {
// ignore any secondary error
}
}
if (error != null) {
try {
errorstream.close();
} catch (IOException e) {
// ignore any secondary error
}
}
}
}
/**
* The directory which will have the expected
* subdirectories, SPECS, SOURCES, BUILD, SRPMS ; optional.
* If this isn't specified,
* the <tt>baseDir</tt> value is used
*
* @param td the directory containing the normal RPM directories.
*/
public void setTopDir(File td) {
this.topDir = td;
}
/**
* What command to issue to the rpm build tool; optional.
* The default is "-bb"
*/
public void setCommand(String c) {
this.command = c;
}
/**
* The name of the spec File to use; required.
*/
public void setSpecFile(String sf) {
if ((sf == null) || (sf.trim().equals(""))) {
throw new BuildException("You must specify a spec file", getLocation());
}
this.specFile = sf;
}
/**
* Flag (optional, default=false) to remove
* the generated files in the BUILD directory
*/
public void setCleanBuildDir(boolean cbd) {
cleanBuildDir = cbd;
}
/**
* Flag (optional, default=false) to remove the spec file from SPECS
*/
public void setRemoveSpec(boolean rs) {
removeSpec = rs;
}
/**
* Flag (optional, default=false)
* to remove the sources after the build.
* See the <tt>--rmsource</tt> option of rpmbuild.
*/
public void setRemoveSource(boolean rs) {
removeSource = rs;
}
/**
* Optional file to save stdout to.
*/
public void setOutput(File output) {
this.output = output;
}
/**
* Optional file to save stderr to
*/
public void setError(File error) {
this.error = error;
}
/**
* The executable to run when building; optional.
* The default is <code>rpmbuild</code>.
*
* @since Ant 1.6
* @param c the rpm build executable
*/
public void setRpmBuildCommand(String c) {
this.rpmBuildCommand = c;
}
/**
* Checks whether <code>rpmbuild</code> is on the PATH and returns
* the absolute path to it - falls back to <code>rpm</code>
* otherwise.
*
* @since 1.6
*/
protected String guessRpmBuildCommand() {
Vector env = Execute.getProcEnvironment();
String path = null;
for (Enumeration e = env.elements(); e.hasMoreElements();) {
String var = (String) e.nextElement();
if (var.startsWith("PATH=") || var.startsWith("Path=")) {
path = var.substring(6 /* "PATH=".length() + 1 */);
break;
}
}
if (path != null) {
Path p = new Path(getProject(), path);
String[] pElements = p.list();
for (int i = 0; i < pElements.length; i++) {
File f = new File(pElements[i],
"rpmbuild"
+ (Os.isFamily("dos") ? ".exe" : ""));
if (f.canRead()) {
return f.getAbsolutePath();
}
}
}
return "rpm";
}
}