/* 
 * The Apache Software License, Version 1.1
 *
 * Copyright (c) 1999 The Apache Software Foundation.  All rights
 * reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution, if
 *    any, must include the following acknowlegement:
 *       "This product includes software developed by the
 *        Apache Software Foundation (http://www.apache.org/)."
 *    Alternately, this acknowlegement may appear in the software itself,
 *    if and wherever such third-party acknowlegements normally appear.
 *
 * 4. The names "The Jakarta Project", "Tomcat", and "Apache Software
 *    Foundation" must not be used to endorse or promote products derived
 *    from this software without prior written permission. For written
 *    permission, please contact apache@apache.org.
 *
 * 5. Products derived from this software may not be called "Apache"
 *    nor may "Apache" appear in their names without prior written
 *    permission of the Apache Group.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * ====================================================================
 *
 * This software consists of voluntary contributions made by many
 * individuals on behalf of the Apache Software Foundation.  For more
 * information on the Apache Software Foundation, please see
 * <http://www.apache.org/>.
 */

package org.apache.tools.ant.taskdefs;

import org.apache.tools.ant.*;
import org.apache.tools.analyzer.*;
import org.apache.tools.analyzer.info.*;

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

/**
 * Task to compile Java source files. This task can take the following
 * arguments:
 * <ul>
 * <li>sourcedir
 * <li>destdir
 * <li>deprecation
 * <li>classpath
 * <li>bootclasspath
 * <li>extdirs
 * <li>optimize
 * <li>debug
 * <li>target
 * </ul>
 * Of these arguments, the <b>sourcedir</b> and <b>destdir</b> are required.
 * <p>
 * When this task executes, it will recursively scan the sourcedir and
 * destdir looking for Java source files to compile. This task makes its
 * compile decision based on timestamp. Any other file in the
 * sourcedir will be copied to the destdir allowing support files to be
 * located properly in the classpath.
 *
 * @author James Davidson <a href="mailto:duncan@x180.com">duncan@x180.com</a>
 */

public class Javac extends MatchingTask implements IStatusListener{

    private String[] srcDirS; 
    private File[] srcDir;
    private File destDir;
    private File sourcePath;
    private String sourcePathS;
    private String compileClasspath;
    private boolean debug = false;
    private boolean optimize = false;
    private boolean deprecation = false;
    private boolean filtering = false;
    private boolean dependingTrack = false;
    private String target;
    private String bootclasspath;
    private String extdirs;
                                       
    protected UniqueVector compileList = new UniqueVector();
    protected Hashtable filecopyList = new Hashtable();

    /**
     * Set the source dir to find the source Java files.
     */
    public void setSrcdir(String srcDirName) {
      StringTokenizer srcPath = new StringTokenizer(srcDirName,";");
      srcDir = new File[srcPath.countTokens()];
      srcDirS = new String[srcPath.countTokens()];
      int i=0;
      while(srcPath.hasMoreTokens()) {
        srcDirS[i]=srcPath.nextToken().trim();
		srcDir[i] = project.resolveFile(srcDirS[i]);
        i++;
      }
    }

    /**
     * Set the destination directory into which the Java source
     * files should be compiled.
     */
    public void setDestdir(String destDirName) {
        destDir = project.resolveFile(destDirName);
    }

    /**
     * Set the classpath to be used for this compilation.
     */
    public void setClasspath(String classpath) {
        compileClasspath = project.translatePath(classpath);
    }

    /**
     * Sets the -sourcepath option.
     */

    public void setSourcepath(String sourcepath) {
      sourcePathS = sourcepath;
      sourcePath = project.resolveFile(sourcepath);
    }

    /**
     * Sets the bootclasspath that will be used to compile the classes
     * against.
     */
    public void setBootclasspath(String bootclasspath) {
        this.bootclasspath = project.translatePath(bootclasspath);
    }

    /**
     * Sets the extension directories that will be used during the
     * compilation.
     */
    public void setExtdirs(String extdirs) {
        this.extdirs = project.translatePath(extdirs);
    }

    /**
     * Set the deprecation flag.
     */
    public void setDeprecation(String deprecationString) {
        this.deprecation = Project.toBoolean(deprecationString);
    }

    /**
     * Set the debug flag.
     */
    public void setDebug(String debugString) {
        this.debug = Project.toBoolean(debugString);
    }

    /**
     * Set the optimize flag.
     */
     public void setOptimize(String optimizeString) {
        this.optimize = Project.toBoolean(optimizeString);
     }

    /**
     * Sets the target VM that the classes will be compiled for. Valid
     * strings are "1.1", "1.2", and "1.3".
     */
    public void setTarget(String target) {
        this.target = target;
    }

    /**
     * Set the filtering flag.
     */
    public void setFiltering(String filter) {
        filtering = Project.toBoolean(filter);
    }

    public void setDependence(String dependence) {
        dependingTrack = Project.toBoolean(dependence);
    }

    /**
     * Executes the task.
     */
    public void execute() throws BuildException {
        // first off, make sure that we've got a srcdir and destdir

        if (srcDir[0] == null) {
            throw new BuildException("srcdir attribute must be set!");
        }
        if (!srcDir[0].exists()) {
            throw new BuildException("srcdir does not exist!");
        }
        if (destDir == null) {
            throw new BuildException("destdir attribute must be set!");
        }

        // scan source and dest dirs to build up both copy lists and
        // compile lists

        prepareCompileList();

        for(int counter=0; counter<srcDir.length; counter++) {
            DirectoryScanner ds = this.getDirectoryScanner(srcDir[counter]);

            String[] files = ds.getIncludedFiles();
     
            scanDir(counter, destDir, files);
        }

        // compile the source files

        String compiler = project.getProperty("build.compiler");
        if (compiler == null) {
            if (Project.getJavaVersion().startsWith("1.3")) {
                compiler = "modern";
            } else {
                compiler = "classic";
            }
        }

        if (compileList.size() > 0) {
            project.log("Compiling " + compileList.size() +
                        " source files to " + destDir);

            if (compiler.equalsIgnoreCase("classic")) {
                doClassicCompile();
            } else if (compiler.equalsIgnoreCase("modern")) {
                doModernCompile();
            } else if (compiler.equalsIgnoreCase("jikes")) {
                doJikesCompile();
            } else {
                String msg = "Don't know how to use compiler " + compiler;
                throw new BuildException(msg);
            }
        }

        // copy the support files

        if (filecopyList.size() > 0) {
            project.log("Copying " + filecopyList.size() +
                        " support files to " + destDir.getAbsolutePath());
            Enumeration enum = filecopyList.keys();
            while (enum.hasMoreElements()) {
                String fromFile = (String) enum.nextElement();
                String toFile = (String) filecopyList.get(fromFile);
                try {
                    project.copyFile(fromFile, toFile, filtering);
                } catch (IOException ioe) {
                    String msg = "Failed to copy " + fromFile + " to " + toFile
                        + " due to " + ioe.getMessage();
                    throw new BuildException(msg);
                }
            }
        }
    }

    /**
     * Scans the directory looking for source files to be compiled and
     * support files to be copied.  The results are returned in the
     * class variables compileList and filecopyList.
     */

    protected void prepareCompileList() {
        compileList.removeAllElements();
        filecopyList.clear();
    }

    protected void scanDir(int nDir, File destDir, String files[]) {
        
        long now = (new Date()).getTime();
        UniqueVector loadItems = new UniqueVector();
        HashedVector fullUnits = new HashedVector(true, 300, 100);
        AnalyzerUtilities.setUnits(fullUnits);
        ClassLoad loader = new ClassLoad(fullUnits, this);
        loader.fileList = new UniqueVector();
        Vector classFiles = new Vector();

        for (int i = 0; i < files.length; i++) {
            File classFile;
            File srcFile = new File(srcDir[nDir], files[i]);
            if (files[i].endsWith(".java")) {
                classFile = new File(destDir,
                        srcDirS[nDir].substring(sourcePathS.length())+
                        File.separator+
                        files[i].substring(0,files[i].indexOf(".java"))
                                                    + ".class");

                    if (srcFile.lastModified() > now) {
                        project.log("Warning: file modified in the future: " + 
                            files[i], project.MSG_WARN);
                    }

                    boolean forceAddClass = false;
                    if(dependingTrack) {
                        if(classFile.exists()) {
                            LoadItem item = new LoadItem(
                                   destDir+File.separator+srcDirS[nDir].substring(sourcePathS.length())+
                                   File.separator+
                                   files[i].substring(0,files[i].indexOf(".java"))
                                   + ".class", false);
                            loadItems.addElement(item);
                            forceAddClass = true;
                        }
                        if (srcFile.lastModified() > classFile.lastModified()) {
                            compileList.addElement(srcFile.getAbsolutePath());
                            if(forceAddClass) 
                                classFiles.addElement(classFile);
                        }
                    }
                    else
                      compileList.addElement(srcFile.getAbsolutePath());
                } else {
                    File destFile = new File(destDir,
                        srcDirS[nDir].substring(sourcePathS.length())+
                        File.separator+
                        files[i]);
                    if (srcFile.lastModified() > destFile.lastModified()) {
                        filecopyList.put(srcFile.getAbsolutePath(),
                                         destFile.getAbsolutePath());
                }
            }
        }
        if(dependingTrack) {
            if ( !loader.loadAllClassesFromSet(loadItems) )
                return;
            for(int k=0; k<fullUnits.size(); k++) {
                ((ClassInfo)fullUnits.get(k)).addReferencedClasses();
            }
            for(int k=0; k<classFiles.size(); k++){
                String classFilePath = ((File)classFiles.get(k)).getAbsolutePath();
                String pathToClass = classFilePath.substring(destDir.getAbsolutePath().length()+1, classFilePath.indexOf(".class"));

                ClassInfo ci = (ClassInfo) fullUnits.get(pathToClass.replace('\\','/')); 
                for(int j=0; j<ci.classesThatRefMe.size(); j++) {
                    ClassInfo cli = (ClassInfo) ci.classesThatRefMe.get(j);
                    if(ci!=cli){
                        compileList.addUnique((sourcePathS.length()!=0?sourcePathS+File.separator:"")+cli.getCompleteDisplayName()+".java");
                    }
                }
            }
        }
    }                  

    /**
     * Builds the compilation classpath.
     */

    // XXX
    // we need a way to not use the current classpath.

    private String getCompileClasspath() {
        StringBuffer classpath = new StringBuffer();

        // add dest dir to classpath so that previously compiled and
        // untouched classes are on classpath

        //classpath.append(sourceDir.getAbsolutePath());
        //classpath.append(File.pathSeparator);
        classpath.append(destDir.getAbsolutePath());

        // add our classpath to the mix

        if (compileClasspath != null) {
            addExistingToClasspath(classpath,compileClasspath);
        }

        // add the system classpath

        addExistingToClasspath(classpath,System.getProperty("java.class.path"));
        return classpath.toString();
    }


     /**
     * Takes a classpath-like string, and adds each element of
     * this string to a new classpath, if the components exist.
     * Components that don't exist, aren't added.
     * We do this, because jikes issues warnings for non-existant
     * files/dirs in his classpath, and these warnings are pretty
     * annoying.
     * @param target - target classpath
     * @param source - source classpath
     * to get file objects.
     */
    private void addExistingToClasspath(StringBuffer target,String source) {
       StringTokenizer tok = new StringTokenizer(source,
                             System.getProperty("path.separator"), false);
       while (tok.hasMoreTokens()) {
           File f = project.resolveFile(tok.nextToken());

           if (f.exists()) {
               target.append(File.pathSeparator);
               target.append(f.getAbsolutePath());
           } else {
               project.log("Dropping from classpath: "+
                   f.getAbsolutePath(),project.MSG_VERBOSE);
           }
       }

    }

    /**
     * Peforms a copmile using the classic compiler that shipped with
     * JDK 1.1 and 1.2.
     */

    private void doClassicCompile() throws BuildException {
        project.log("Using classic compiler", project.MSG_VERBOSE);
        String classpath = getCompileClasspath();
        Vector argList = new Vector();

        if (deprecation == true)
            argList.addElement("-deprecation");

        argList.addElement("-d");
        argList.addElement(destDir.getAbsolutePath());
        argList.addElement("-classpath");
        // Just add "sourcepath" to classpath ( for JDK1.1 )
        if (Project.getJavaVersion().startsWith("1.1")) {
            argList.addElement(classpath + File.pathSeparator +
                               sourcePath.getAbsolutePath());
        } else {
            argList.addElement(classpath);
            argList.addElement("-sourcepath");
            argList.addElement(sourcePath.getAbsolutePath());
            if (target != null) {
                argList.addElement("-target");
                argList.addElement(target);
            }
        }
        if (debug) {
            argList.addElement("-g");
        }
        if (optimize) {
            argList.addElement("-O");
        }
        if (bootclasspath != null) {
            argList.addElement("-bootclasspath");
            argList.addElement(bootclasspath);
        }
        if (extdirs != null) {
            argList.addElement("-extdirs");
            argList.addElement(extdirs);
        }

        project.log("Compilation args: " + argList.toString(),
                    project.MSG_VERBOSE);

        String[] args = new String[argList.size() + compileList.size()];
        int counter = 0;

        for (int i = 0; i < argList.size(); i++) {
            args[i] = (String)argList.elementAt(i);
            counter++;
        }

        // XXX
        // should be using system independent line feed!

        StringBuffer niceSourceList = new StringBuffer("Files to be compiled:"
                                                       + "\r\n");

        Enumeration enum = compileList.elements();
        while (enum.hasMoreElements()) {
            args[counter] = (String)enum.nextElement();
            niceSourceList.append("    " + args[counter] + "\r\n");
            counter++;
        }

        project.log(niceSourceList.toString(), project.MSG_VERBOSE);

        // XXX
        // provide the compiler a different message sink - namely our own

        JavacOutputStream jos = new JavacOutputStream(project);

        sun.tools.javac.Main compiler =
            new sun.tools.javac.Main(jos, "javac");
        compiler.compile(args);
        if (jos.getErrorFlag()) {
            String msg = "Compile failed, messages should have been provided.";
            throw new BuildException(msg);
        }
    }

    /**
     * Performs a compile using the newer compiler that ships with JDK 1.3
     */

    private void doModernCompile() throws BuildException {
        project.log("Performing a Modern Compile");
    }

    /**
     * Performs a compile using the Jikes compiler from IBM..
     * Mostly of this code is identical to doClassicCompile()
     * However, it does not support all options like
     * bootclasspath, extdirs, deprecation and so on, because
     * there is no option in jikes and I don't understand
     * what they should do.
     *
     * It has been successfully tested with jikes 1.10
     *
     * @author skanthak@muehlheim.de
     */

    private void doJikesCompile() throws BuildException {
        project.log("Using jikes compiler",project.MSG_VERBOSE);

        StringBuffer classpath = new StringBuffer();
        classpath.append(getCompileClasspath());

        // Jikes doesn't support an extension dir (-extdir)
        // so we'll emulate it for compatibility and convenience.
        addExtdirsToClasspath(classpath);

        // Jikes has no option for source-path so we
        // will add it to classpath.
        classpath.append(File.pathSeparator);
        classpath.append(srcDir[0].getAbsolutePath());

        Vector argList = new Vector();

        if (deprecation == true)
            argList.addElement("-deprecation");

        // We want all output on stdout to make
        // parsing easier
        argList.addElement("-Xstdout");

        argList.addElement("-d");
        argList.addElement(destDir.getAbsolutePath());
        argList.addElement("-classpath");
        argList.addElement(classpath.toString());

        if(!dependingTrack) { // Full dependence tracking wil be performed by JIKES 
	        argList.addElement("+F"); 
	        argList.addElement("+M");
	        argList.addElement("+DR=makefile");
        }



        if (debug) {
            argList.addElement("-g");
        }
        if (optimize) {
            argList.addElement("-O");
        }

       /**
        * XXX
        * Perhaps we shouldn't use properties for these
        * two options (emacs mode and warnings),
        * but include it in the javac directive?
        */

       /**
        * Jikes has the nice feature to print error
        * messages in a form readable by emacs, so
        * that emcas can directly set the cursor
        * to the place, where the error occured.
        */
       boolean emacsMode = false;
       String emacsProperty = project.getProperty("build.compiler.emacs");
       if (emacsProperty != null &&
           (emacsProperty.equalsIgnoreCase("on") ||
            emacsProperty.equalsIgnoreCase("true"))
           ) {
           emacsMode = true;
       }

       /**
        * Jikes issues more warnings that javac, for
        * example, when you have files in your classpath
        * that don't exist. As this is often the case, these
        * warning can be pretty annoying.
        */
       boolean warnings = true;
       String warningsProperty = project.getProperty("build.compiler.warnings");
       if (warningsProperty != null &&
           (warningsProperty.equalsIgnoreCase("off") ||
            warningsProperty.equalsIgnoreCase("false"))
           ) {
           warnings = false;
       }

       if (emacsMode)
           argList.addElement("+E");

       if (!warnings)
           argList.addElement("-nowarn");

        project.log("Compilation args: " + argList.toString(),
                    project.MSG_VERBOSE);

        String[] args = new String[argList.size() + compileList.size()];
        int counter = 0;

        for (int i = 0; i < argList.size(); i++) {
            args[i] = (String)argList.elementAt(i);
            counter++;
        }

        // XXX
        // should be using system independent line feed!

        StringBuffer niceSourceList = new StringBuffer("Files to be compiled:"
                                                       + "\r\n");

        Enumeration enum = compileList.elements();
        while (enum.hasMoreElements()) {
            args[counter] = (String)enum.nextElement();
            niceSourceList.append("    " + args[counter] + "\r\n");
            counter++;
        }

        project.log(niceSourceList.toString(), project.MSG_VERBOSE);

        // XXX
        // provide the compiler a different message sink - namely our own

        JikesOutputParser jop = new JikesOutputParser(project,emacsMode);

        Jikes compiler = new Jikes(jop,"jikes");
        compiler.compile(args);
        if (jop.getErrorFlag()) {
            String msg = "Compile failed, messages should have been provided.";
            throw new BuildException(msg);
        }
    }

    class JarFilenameFilter implements FilenameFilter {
        public boolean accept(File dir,String name) {
            return name.endsWith(".jar");
        }
    }

    /**
     * Emulation of extdirs feature in java >= 1.2.
     * This method adds all jar archives in the given
     * directories (but not in sub-directories!) to the classpath,
     * so that you don't have to specify them all one by one.
     * @param classpath - stringbuffer to append jar files to
     */
    private void addExtdirsToClasspath(StringBuffer classpath) {
       // FIXME
       // Should we scan files recursively? How does
       // javac handle this?

       if (extdirs != null) {
           StringTokenizer tok = new StringTokenizer(extdirs,
                                                     File.pathSeparator,
                                                     false);
           while (tok.hasMoreTokens()) {
               File dir = project.resolveFile(tok.nextToken());
               String[] files = dir.list(new JarFilenameFilter());
               for (int i=0 ; i < files.length ; i++) {
                   File f = new File(dir,files[i]);
                   if (f.exists() && f.isFile()) {
                       classpath.append(File.pathSeparator);
                       classpath.append(f.getAbsolutePath());
                   }
               }
           }
       }
    }

    /**
     * Here we can catch class file parser error and status messages.
     */
    public void status(String s) {
    }

}