From 9c1e29842e65bf20895db9e19336b2ca948236ad Mon Sep 17 00:00:00 2001
From: Reinier Zwitserloot <reinier@zwitserloot.com>
Date: Thu, 23 May 2013 22:58:34 +0200
Subject: Added methods to obtain JLS support-level version information from
 AST/LombokNode. Tests updates to honour these with //version X at the top of
 any test file (now also in eclipse, which until now always said it was v6)

---
 src/core/lombok/core/AST.java           | 10 ++++++++++
 src/core/lombok/core/LombokNode.java    |  9 +++++++++
 src/core/lombok/eclipse/EclipseAST.java | 14 ++++++++++++++
 src/core/lombok/javac/JavacAST.java     | 10 ++++++++++
 4 files changed, 43 insertions(+)

(limited to 'src/core')

diff --git a/src/core/lombok/core/AST.java b/src/core/lombok/core/AST.java
index a2279efe..30297418 100644
--- a/src/core/lombok/core/AST.java
+++ b/src/core/lombok/core/AST.java
@@ -146,9 +146,19 @@ public abstract class AST<A extends AST<A, L, N>, L extends LombokNode<A, L, N>,
 		return nodeMap.get(node);
 	}
 	
+	/**
+	 * Returns the JLS spec version that the compiler uses to parse and compile this AST.
+	 * For example, if -source 1.6 is on the command line, this will return {@code 6}.
+	 */
+	public int getSourceVersion() {
+		return 6;
+	}
+	
 	/**
 	 * Returns the latest version of the java language specification supported by the host compiler.
 	 * For example, if compiling with javac v1.7, this returns {@code 7}.
+	 * 
+	 * NB: Even if -source (lower than maximum) is specified, this method still returns the maximum supported number.
 	 */
 	public int getLatestJavaSpecSupported() {
 		return 6;
diff --git a/src/core/lombok/core/LombokNode.java b/src/core/lombok/core/LombokNode.java
index aa161a8d..30bacc56 100644
--- a/src/core/lombok/core/LombokNode.java
+++ b/src/core/lombok/core/LombokNode.java
@@ -189,6 +189,15 @@ public abstract class LombokNode<A extends AST<A, L, N>, L extends LombokNode<A,
 		return ast.getLatestJavaSpecSupported();
 	}
 	
+	/**
+	 * Convenient shortcut to the owning ast object's getSourceVersion method.
+	 * 
+	 * @see AST#getSourceVersion()
+	 */
+	public int getSourceVersion() {
+		return ast.getSourceVersion();
+	}
+	
 	/**
 	 * Convenient shortcut to the owning ast object's top method.
 	 * 
diff --git a/src/core/lombok/eclipse/EclipseAST.java b/src/core/lombok/eclipse/EclipseAST.java
index d1f4ff06..612dcff7 100644
--- a/src/core/lombok/eclipse/EclipseAST.java
+++ b/src/core/lombok/eclipse/EclipseAST.java
@@ -68,6 +68,20 @@ public class EclipseAST extends AST<EclipseAST, EclipseNode, ASTNode> {
 		return pkg == null ? null : Eclipse.toQualifiedName(pkg.getImportName());
 	}
 	
+	@Override public int getSourceVersion() {
+		long sl = compilationUnitDeclaration.problemReporter.options.sourceLevel;
+		long cl = compilationUnitDeclaration.problemReporter.options.complianceLevel;
+		sl >>= 16;
+		cl >>= 16;
+		if (sl == 0) sl = cl;
+		if (cl == 0) cl = sl;
+		return Math.min((int)(sl - 44), (int)(cl - 44));
+	}
+	
+	@Override public int getLatestJavaSpecSupported() {
+		return Eclipse.getEcjCompilerVersion();
+	}
+	
 	/**
 	 * Runs through the entire AST, starting at the compilation unit, calling the provided visitor's visit methods
 	 * for each node, depth first.
diff --git a/src/core/lombok/javac/JavacAST.java b/src/core/lombok/javac/JavacAST.java
index 9e15516e..8197dae6 100644
--- a/src/core/lombok/javac/JavacAST.java
+++ b/src/core/lombok/javac/JavacAST.java
@@ -34,6 +34,7 @@ import javax.tools.JavaFileObject;
 import lombok.core.AST;
 
 import com.sun.tools.javac.code.Symtab;
+import com.sun.tools.javac.code.Source;
 import com.sun.tools.javac.model.JavacElements;
 import com.sun.tools.javac.model.JavacTypes;
 import com.sun.tools.javac.tree.JCTree;
@@ -113,6 +114,15 @@ public class JavacAST extends AST<JavacAST, JavacNode, JCTree> {
 		for (JavacNode child : node.down()) child.traverse(visitor);
 	}
 	
+	@Override public int getSourceVersion() {
+		try {
+			String nm = Source.instance(context).name();
+			int underscoreIdx = nm.indexOf('_');
+			if (underscoreIdx > -1) return Integer.parseInt(nm.substring(underscoreIdx + 1));
+		} catch (Exception ignore) {}
+		return 6;
+	}
+	
 	@Override public int getLatestJavaSpecSupported() {
 		return Javac.getJavaCompilerVersion();
 	}
-- 
cgit