GenStubsTask.java

/*
 * Copyright (c) 2009, 2012, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code 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 General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package anttasks;

import genstubs.GenStubs;

import java.io.*;
import java.util.*;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.DirectoryScanner;
import org.apache.tools.ant.taskdefs.MatchingTask;
import org.apache.tools.ant.types.Path;
import org.apache.tools.ant.types.Reference;

/**
 * Files are specified with an implicit fileset, using srcdir as a base directory.
 * The set of files to be included is specified with an includes attribute or
 * nested <includes> set. However, unlike a normal fileset, an empty includes attribute
 * means "no files" instead of "all files".  The Ant task also accepts "fork=true" and
 * classpath attribute or nested <classpath> element to run GenStubs in a separate VM
 * with the specified path. This is likely necessary if a JDK 7 parser is required to read the
 * JDK 7 input files.
 */
public class GenStubsTask extends MatchingTask {
    private File srcDir;
    private File destDir;
    private boolean fork;
    private Path classpath;
    private String includes;

    public void setSrcDir(File dir) {
        this.srcDir = dir;
    }

    public void setDestDir(File dir) {
        this.destDir = dir;
    }

    public void setFork(boolean v) {
        this.fork = v;
    }

    public void setClasspath(Path cp) {
        if (classpath == null)
            classpath = cp;
        else
            classpath.append(cp);
    }

    public Path createClasspath() {
        if (classpath == null) {
            classpath = new Path(getProject());
        }
        return classpath.createPath();
    }

    public void setClasspathRef(Reference r) {
        createClasspath().setRefid(r);
    }

    public void setIncludes(String includes) {
        super.setIncludes(includes);
        this.includes = includes;
    }

    @Override
    public void execute() {
        if (includes != null && includes.trim().isEmpty())
            return;

        DirectoryScanner s = getDirectoryScanner(srcDir);
        String[] files = s.getIncludedFiles();
//            System.err.println("Ant.execute: srcDir " + srcDir);
//            System.err.println("Ant.execute: destDir " + destDir);
//            System.err.println("Ant.execute: files " + Arrays.asList(files));

        files = filter(srcDir, destDir, files);
        if (files.length == 0)
            return;
        System.out.println("Generating " + files.length + " stub files to " + destDir);

        List<String> classNames = new ArrayList<String>();
        for (String file: files) {
            classNames.add(file.replaceAll(".java$", "").replace('/', '.'));
        }

        if (!fork) {
            GenStubs m = new GenStubs();
            boolean ok = m.run(srcDir.getPath(), destDir, classNames);
            if (!ok)
                throw new BuildException("genstubs failed");
        } else {
            List<String> cmd = new ArrayList<String>();
            String java_home = System.getProperty("java.home");
            cmd.add(new File(new File(java_home, "bin"), "java").getPath());
            if (classpath != null)
                cmd.add("-Xbootclasspath/p:" + classpath);
            cmd.add(GenStubs.class.getName());
            cmd.add("-sourcepath");
            cmd.add(srcDir.getPath());
            cmd.add("-s");
            cmd.add(destDir.getPath());
            cmd.addAll(classNames);
            //System.err.println("GenStubs exec " + cmd);
            ProcessBuilder pb = new ProcessBuilder(cmd);
            pb.redirectErrorStream(true);
            try {
                Process p = pb.start();
                BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
                try {
                    String line;
                    while ((line = in.readLine()) != null)
                        System.out.println(line);
                } finally {
                    in.close();
                }
                int rc = p.waitFor();
                if (rc != 0)
                    throw new BuildException("genstubs failed");
            } catch (IOException e) {
                throw new BuildException("genstubs failed", e);
            } catch (InterruptedException e) {
                throw new BuildException("genstubs failed", e);
            }
        }
    }

    String[] filter(File srcDir, File destDir, String[] files) {
        List<String> results = new ArrayList<String>();
        for (String f: files) {
            long srcTime = new File(srcDir, f).lastModified();
            long destTime = new File(destDir, f).lastModified();
            if (srcTime > destTime)
                results.add(f);
        }
        return results.toArray(new String[results.size()]);
    }
}