/* -*- mode: java; c-basic-offset: 2; indent-tabs-mode: nil -*- */

/*

Part of the Processing project - http://processing.org

Copyright (c) 2004-12 Ben Fry and Casey Reas

Copyright (c) 2001-04 Massachusetts Institute of Technology

This program is free software; you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation; either version 2 of the License, or

(at your option) any later version.

This program 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 for more details.

You should have received a copy of the GNU General Public License

along with this program; if not, write to the Free Software Foundation,

Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

*/

package processing.mode.java;

import processing.app.*;

import processing.app.ui.Editor;

import processing.core.*;

import java.io.*;

import java.lang.reflect.Method;

import java.util.HashMap;

//import org.eclipse.jdt.core.compiler.batch.BatchCompiler;

//import org.eclipse.jdt.core.compiler.CompilationProgress;

public class Compiler {

static HashMap<String, String> importSuggestions;

static {

importSuggestions = new HashMap<String, String>();

importSuggestions.put("Arrays", "java.util.Arrays");

importSuggestions.put("Collections", "java.util.Collections");

importSuggestions.put("Date", "java.util.Date");

importSuggestions.put("Frame", "java.awt.Frame");

importSuggestions.put("Iterator", "java.util.Iterator");

}

/**

* Compile with ECJ. See http://j.mp/8paifz for documentation.

*

* @param sketch Sketch object to be compiled, used for placing exceptions

* @param buildPath Where the temporary files live and will be built from.

* @return true if successful.

* @throws RunnerException Only if there's a problem. Only then.

*/

static public boolean compile(JavaBuild build) throws SketchException {

// This will be filled in if anyone gets angry

SketchException exception = null;

boolean success = false;

String baseCommand[] = new String[] {

"-g",

"-Xemacs",

//"-noExit", // not necessary for ecj

"-source", "1.7",

"-target", "1.7",

"-classpath", build.getClassPath(),

"-nowarn", // we're not currently interested in warnings (works in ecj)

"-d", build.getBinFolder().getAbsolutePath() // output the classes in the buildPath

};

//PApplet.println(baseCommand);

String[] sourceFiles = Util.listFiles(build.getSrcFolder(), false, ".java");

String[] command = PApplet.concat(baseCommand, sourceFiles);

//PApplet.println(command);

try {

// Load errors into a local StringBuilder

final StringBuilder errorBuffer = new StringBuilder();

// Create single method dummy writer class to slurp errors from ecj

Writer internalWriter = new Writer() {

public void write(char[] buf, int off, int len) {

errorBuffer.append(buf, off, len);

}

public void flush() { }

public void close() { }

};

// Wrap as a PrintWriter since that's what compile() wants

PrintWriter writer = new PrintWriter(internalWriter);

//result = com.sun.tools.javac.Main.compile(command, writer);

PrintWriter outWriter = new PrintWriter(System.out);

// Version that's not dynamically loaded

//CompilationProgress progress = null;

//success = BatchCompiler.compile(command, outWriter, writer, progress);

// Version that *is* dynamically loaded. First gets the mode class loader

// so that it can grab the compiler JAR files from it.

ClassLoader loader = build.mode.getClassLoader();

try {

Class<?> batchClass =

Class.forName("org.eclipse.jdt.core.compiler.batch.BatchCompiler", false, loader);

Class<?> progressClass =

Class.forName("org.eclipse.jdt.core.compiler.CompilationProgress", false, loader);

Class<?>[] compileArgs =

new Class<?>[] { String[].class, PrintWriter.class, PrintWriter.class, progressClass };

Method compileMethod = batchClass.getMethod("compile", compileArgs);

success = (Boolean)

compileMethod.invoke(null, new Object[] { command, outWriter, writer, null });

} catch (Exception e) {

e.printStackTrace();

throw new SketchException("Unknown error inside the compiler.");

}

// Close out the stream for good measure

writer.flush();

writer.close();

BufferedReader reader =

new BufferedReader(new StringReader(errorBuffer.toString()));

//System.err.println(errorBuffer.toString());

String line = null;

while ((line = reader.readLine()) != null) {

//System.out.println("got line " + line); // debug

// get first line, which contains file name, line number,

// and at least the first line of the error message

String errorFormat = "([\\w\\d_]+.java):(\\d+):\\s*(.*):\\s*(.*)\\s*";

String[] pieces = PApplet.match(line, errorFormat);

//PApplet.println(pieces);

// if it's something unexpected, die and print the mess to the console

if (pieces == null) {

exception = new SketchException("Cannot parse error text: " + line);

exception.hideStackTrace();

// Send out the rest of the error message to the console.

System.err.println(line);

while ((line = reader.readLine()) != null) {

System.err.println(line);

}

break;

}

// translate the java filename and line number into a un-preprocessed

// location inside a source file or tab in the environment.

String dotJavaFilename = pieces[1];

// Line numbers are 1-indexed from javac

int dotJavaLineIndex = PApplet.parseInt(pieces[2]) - 1;

String errorMessage = pieces[4];

exception = build.placeException(errorMessage,

dotJavaFilename,

dotJavaLineIndex);

if (exception == null) {

exception = new SketchException(errorMessage);

}

if (errorMessage.startsWith("The import ") &&

errorMessage.endsWith("cannot be resolved")) {

// The import poo cannot be resolved

//import poo.shoe.blah.*;

//String what = errorMessage.substring("The import ".length());

String[] m = PApplet.match(errorMessage, "The import (.*) cannot be resolved");

//what = what.substring(0, what.indexOf(' '));

if (m != null) {

// System.out.println("'" + m[1] + "'");

if (m[1].equals("processing.xml")) {

exception.setMessage("processing.xml no longer exists, this code needs to be updated for 2.0.");

System.err.println("The processing.xml library has been replaced " +

"with a new 'XML' class that's built-in.");

handleCrustyCode();

} else {

exception.setMessage("The package " +

"\u201C" + m[1] + "\u201D" +

" does not exist. " +

"You might be missing a library.");

System.err.println("Libraries must be " +

"installed in a folder named 'libraries' " +

"inside the 'sketchbook' folder.");

}

}

} else if (errorMessage.endsWith("cannot be resolved to a type")) {

// xxx cannot be resolved to a type

//xxx c;

String what = errorMessage.substring(0, errorMessage.indexOf(' '));

if (what.equals("BFont") ||

what.equals("BGraphics") ||

what.equals("BImage")) {

exception.setMessage(what + " has been replaced with P" + what.substring(1));

handleCrustyCode();

} else {

exception.setMessage("Cannot find a class or type " +

"named \u201C" + what + "\u201D");

String suggestion = importSuggestions.get(what);

if (suggestion != null) {

System.err.println("You may need to add \"import " + suggestion + ";\" to the top of your sketch.");

System.err.println("To make sketches more portable, imports that are not part of the Processing API have been removed from Processing 2.0.");

System.err.println("See the changes page for more information: http://wiki.processing.org/w/Changes");

}

}

} else if (errorMessage.endsWith("cannot be resolved")) {

// xxx cannot be resolved

//println(xxx);

String what = errorMessage.substring(0, errorMessage.indexOf(' '));

if (what.equals("LINE_LOOP") ||

what.equals("LINE_STRIP")) {

exception.setMessage("LINE_LOOP and LINE_STRIP are not available, " +

"please update your code.");

handleCrustyCode();

} else if (what.equals("framerate")) {

exception.setMessage("framerate should be changed to frameRate.");

handleCrustyCode();

} else if (what.equals("screen")) {

exception.setMessage("Change screen.width and screen.height to " +

"displayWidth and displayHeight.");

handleCrustyCode();

} else if (what.equals("screenWidth") ||

what.equals("screenHeight")) {

exception.setMessage("Change screenWidth and screenHeight to " +

"displayWidth and displayHeight.");

handleCrustyCode();

} else {

exception.setMessage("Cannot find anything " +

"named \u201C" + what + "\u201D");

}

} else if (errorMessage.startsWith("Duplicate")) {

// "Duplicate nested type xxx"

// "Duplicate local variable xxx"

} else {

String[] parts = null;

// The method xxx(String) is undefined for the type Temporary_XXXX_XXXX

//xxx("blah");

// The method xxx(String, int) is undefined for the type Temporary_XXXX_XXXX

//xxx("blah", 34);

// The method xxx(String, int) is undefined for the type PApplet

//PApplet.sub("ding");

String undefined =

"The method (\\S+\\(.*\\)) is undefined for the type (.*)";

parts = PApplet.match(errorMessage, undefined);

if (parts != null) {

if (parts[1].equals("framerate(int)")) {

exception.setMessage("framerate() no longer exists, use frameRate() instead.");

handleCrustyCode();

} else if (parts[1].equals("push()")) {

exception.setMessage("push() no longer exists, use pushMatrix() instead.");

handleCrustyCode();

} else if (parts[1].equals("pop()")) {

exception.setMessage("pop() no longer exists, use popMatrix() instead.");

handleCrustyCode();

} else {

String mess = "The function " + parts[1] + " does not exist.";

exception.setMessage(mess);

}

break;

}

}

if (exception != null) {

// The stack trace just shows that this happened inside the compiler,

// which is a red herring. Don't ever show it for compiler stuff.

exception.hideStackTrace();

break;

}

}

} catch (IOException e) {

String bigSigh = "Error while compiling. (" + e.getMessage() + ")";

exception = new SketchException(bigSigh);

e.printStackTrace();

success = false;

}

// In case there was something else.

if (exception != null) throw exception;

return success;

}

static protected void handleCrustyCode() {

System.err.println("This code needs to be updated " +

"for this version of Processing, " +

"please read the Changes page on the Wiki.");

Editor.showChanges();

}

protected int caretColumn(String caretLine) {

return caretLine.indexOf("^");

}

}