GRAILS-6583 added Andrew's code for GSP debugging

Author: burtbeckwith

maven/grails-web.mf

@@ -19,6 +19,9 @@ Import-Template:
   org.codehaus.groovy.ant.*;version="${groovy.version:[=.=.=, =.+1.0)}",
   org.codehaus.groovy.tools.*;version="${groovy.version:[=.=.=, =.+1.0)}",
   org.codehaus.groovy.syntax.*;version="${groovy.version:[=.=.=, =.+1.0)}",
+  org.codehaus.groovy.ast.*;version="${groovy.version:[=.=.=, =.+1.0)}",
+  org.codehaus.groovy.ast.expr.*;version="${groovy.version:[=.=.=, =.+1.0)}",
+  org.codehaus.groovy.ast.stmt.*;version="${groovy.version:[=.=.=, =.+1.0)}",
   org.codehaus.groovy.grails.*;version="${grails.version:[=.=.=, =.+1.0)}",
   org.codehaus.groovy.transform.*;version="${groovy.version:[=.=.=, =.+1.0)}",
   org.codehaus.groovy.classgen.*;version="${groovy.version:[=.=.=, =.+1.0)}",

src/java/org/codehaus/groovy/grails/web/pages/GroovyPageParser.java

@@ -77,6 +77,7 @@ public class GroovyPageParser implements Tokens {
 	private GSPWriter out;
 	private String className;
 	private String packageName;
+	private String sourceName; // last segment of the file name (eg- index.gsp)
 	private boolean finalPass = false;
 	private int tagIndex;
 	private Map tagContext;
@@ -207,6 +208,7 @@ public GroovyPageParser(String name, String uri, String filename, InputStream in
 		this.pageName = uri;
 		this.environment = Environment.getCurrent();
 		makeName(name);
+      makeSourceName(filename);
 	} // Parse()
 
     private Map<String, String> parseDirectives(String gspSource) {
@@ -642,6 +644,22 @@ else if (ix == 0 && c >= '0' && c <= '9')
 		className = buf.toString();
 	} // makeName()
 
+	/**
+     * find the simple name of this gsp
+     * @param filename the fully qualified file name
+     */
+    private void makeSourceName(String filename) {
+        if (filename != null) {
+            int lastSegmentStart = filename.lastIndexOf('/');
+            if (lastSegmentStart == -1) {
+                lastSegmentStart = filename.lastIndexOf('\\');
+            }
+            sourceName = filename.substring(lastSegmentStart + 1);
+        } else {
+            sourceName = className;
+        }
+    }
+
 	private static boolean match(CharSequence pat, CharSequence text, int start) {
 		int ix = start, ixz = text.length(), ixy = start + pat.length();
 		if (ixz > ixy)
@@ -789,6 +807,10 @@ private void page() {
 				out.println("null");
 			}
 			out.println("}");
+
+			if (shouldAddLineNumbers()) {
+				addLineNumbers();
+			}
 		} else {
 			for (int i = 0; i < DEFAULT_IMPORTS.length; i++) {
 				out.print("import ");
@@ -798,6 +820,64 @@ private void page() {
 		}
 	} // page()
 
+	/**
+     * Determines if the line numbers array should be added to the generated Groovy class.
+     * @return true if they should
+     */
+    private boolean shouldAddLineNumbers() {
+        try {
+            // for now, we support this through a system property.
+            String prop = System.getenv("GROOVY_PAGE_ADD_LINE_NUMBERS");
+            return Boolean.valueOf(prop).booleanValue();
+        } catch (Exception e) {
+            // something wild happened
+            return false;
+        }
+    }
+
+    /**
+     * Adds the line numbers array to the end of the generated Groovy ModuleNode
+     * in a way suitable for the LineNumberTransform AST transform to operate on it
+     */
+    private void addLineNumbers() {
+        out.println();
+        out.println("@org.codehaus.groovy.grails.web.transform.LineNumber(");
+        out.print("\tlines = [");
+        // get the line numbers here.  this will mean that the last 2 lines will not be captured in the
+        // line number information, but that's OK since a user cannot set a breakpoint there anyway.
+        int[] lineNumbers = filterTrailing0s(out.getLineNumbers());
+
+        for (int i = 0; i < lineNumbers.length; i++) {
+            out.print(lineNumbers[i]);
+            if (i < lineNumbers.length - 1) {
+                out.print(", ");
+            }
+        }
+        out.println("],");
+        out.println("\tsourceName = \"" + sourceName + "\"");
+        out.println(")");
+        out.println("class ___LineNumberPlaceholder { }");
+    }
+
+    /**
+     * Filters trailing 0s from the line number array
+     * @param lineNumbers the line number array
+     * @return a new array that removes all 0s from the end of it
+     */
+    private int[] filterTrailing0s(int[] lineNumbers) {
+        int startLocation = lineNumbers.length - 1;
+        for (int i = lineNumbers.length -1; i >= 0; i--) {
+            if (lineNumbers[i] > 0) {
+                startLocation = i + 1;
+                break;
+            }
+        }
+        
+        int[] newLineNumbers = new int[startLocation];
+        System.arraycopy(lineNumbers, 0, newLineNumbers, 0, startLocation);
+        return newLineNumbers;
+    }
+
 	private void endTag() {
 		if (!finalPass)
 			return;

src/java/org/codehaus/groovy/grails/web/transform/LineNumber.java

@@ -0,0 +1,35 @@
+/*
+ * Copyright 2003-2010 the original author or authors.
+ *
+ * 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.codehaus.groovy.grails.web.transform;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+import org.codehaus.groovy.transform.GroovyASTTransformationClass;
+
+/**
+ * @author Andrew Eisenberg
+ * @created Jul 22, 2010
+ */
+@Retention(RetentionPolicy.SOURCE)
+@Target({ ElementType.TYPE })
+@GroovyASTTransformationClass({"org.codehaus.groovy.grails.web.transform.LineNumberTransform"})
+public @interface LineNumber {
+    int[] lines();
+    String sourceName();
+}