Allow customzing css classes for the ErrorsViewStackTracePrinter

Author: codeconsole

grails-bootstrap/src/main/groovy/org/grails/exceptions/reporting/DefaultStackTracePrinter.groovy

@@ -23,7 +23,7 @@ import org.codehaus.groovy.control.MultipleCompilationErrorsException
  */
 class DefaultStackTracePrinter implements StackTracePrinter {
 
-    String prettyPrint(Throwable t) {
+    String prettyPrint(Throwable t, Map attrs = [:]) {
         if (t == null) return ''
         if (!t.stackTrace) return 'No stack trace available'
         final sw = new StringWriter()

grails-core/src/main/groovy/org/grails/core/exceptions/DefaultErrorsPrinter.groovy

@@ -47,7 +47,7 @@ class DefaultErrorsPrinter extends DefaultStackTracePrinter implements CodeSnipp
         res == null ? te.className : res.getFilename()
     }
 
-    String prettyPrintCodeSnippet(Throwable exception) {
+    String prettyPrintCodeSnippet(Throwable exception, Map attrs = [:]) {
         if (exception == null) {
             return ''
         }
@@ -86,7 +86,7 @@ class DefaultErrorsPrinter extends DefaultStackTracePrinter implements CodeSnipp
                     if (lineNumbersShown[res.filename].contains(lineNumber)) continue // don't repeat the same lines twice
 
                     lineNumbersShown[res.filename] << lineNumber
-                    pw.print formatCodeSnippetStart(res, lineNumber)
+                    pw.print formatCodeSnippetStart(res, lineNumber, attrs)
                     def input = null
                     try {
                         input = res.inputStream
@@ -101,10 +101,10 @@ class DefaultErrorsPrinter extends DefaultStackTracePrinter implements CodeSnipp
                                 if (currentLineNumber in range) {
                                     boolean isErrorLine = currentLineNumber == lineNumber
                                     if (isErrorLine) {
-                                        pw.print formatCodeSnippetErrorLine(currentLineNumber, currentLine)
+                                        pw.print formatCodeSnippetErrorLine(currentLineNumber, currentLine, attrs)
                                     }
                                     else {
-                                        pw.print formatCodeSnippetLine(currentLineNumber, currentLine)
+                                        pw.print formatCodeSnippetLine(currentLineNumber, currentLine, attrs)
                                     }
                                 }
                                 else if (currentLineNumber > last) {
@@ -198,18 +198,18 @@ class DefaultErrorsPrinter extends DefaultStackTracePrinter implements CodeSnipp
         ""
     }
 
-    String formatCodeSnippetStart(Resource resource, int lineNumber) {
+    String formatCodeSnippetStart(Resource resource, int lineNumber, Map attrs = [:]) {
         """Around line $lineNumber of $resource.filename
 """
 
     }
 
-    protected String formatCodeSnippetLine(int currentLineNumber, currentLine) {
+    protected String formatCodeSnippetLine(int currentLineNumber, currentLine, Map attrs = [:]) {
         """${currentLineNumber}: ${currentLine}
 """
     }
 
-    protected String formatCodeSnippetErrorLine(int currentLineNumber, currentLine) {
+    protected String formatCodeSnippetErrorLine(int currentLineNumber, currentLine, Map attrs = [:]) {
         """${currentLineNumber}: ${currentLine}
 """
     }

grails-web-common/src/main/groovy/org/grails/web/errors/ErrorsViewStackTracePrinter.groovy

@@ -38,23 +38,23 @@ class ErrorsViewStackTracePrinter extends DefaultErrorsPrinter {
     }
 
     @Override
-    String prettyPrint(Throwable t) {
+    String prettyPrint(Throwable t, Map attrs = [:]) {
         if (t instanceof GrailsWrappedRuntimeException) {
             t = t.cause
         }
-        return super.prettyPrint(t)
+        return super.prettyPrint(t, attrs)
     }
 
     @Override
-    String prettyPrintCodeSnippet(Throwable t) {
+    String prettyPrintCodeSnippet(Throwable t, Map attrs = [:]) {
         if (t instanceof GrailsWrappedRuntimeException) {
             t = t.cause
         }
-        return super.prettyPrintCodeSnippet(t)
+        return super.prettyPrintCodeSnippet(t, attrs)
     }
 
     @Override
-    String formatCodeSnippetStart(Resource resource, int lineNumber) {
+    String formatCodeSnippetStart(Resource resource, int lineNumber, Map attrs = [:]) {
         String path = resource.filename
         // try calc better path
         try {
@@ -67,8 +67,8 @@ class ErrorsViewStackTracePrinter extends DefaultErrorsPrinter {
             path = resource.filename
         }
         """\
-<h2>Around line ${lineNumber} of <span class="filename">${path}</span></h2>
-<pre class="snippet">"""
+<h2>Around line ${lineNumber} of <span class="${attrs['filenameClass'] ?: 'filename'}">${path}</span></h2>
+<pre class="${attrs['snippetClass'] ?: 'snippet'}">"""
     }
 
     @Override
@@ -77,13 +77,13 @@ class ErrorsViewStackTracePrinter extends DefaultErrorsPrinter {
     }
 
     @Override
-    protected String formatCodeSnippetLine(int currentLineNumber, Object currentLine) {
-        """<code class="line"><span class="lineNumber">${currentLineNumber}:</span>${currentLine.encodeAsHTML()}</code>"""
+    protected String formatCodeSnippetLine(int currentLineNumber, Object currentLine, Map attrs = [:]) {
+        """<code class="${attrs['lineClass'] ?: 'line'}"><span class="${attrs['lineNumberClass'] ?: 'lineNumber'}">${currentLineNumber}:</span>${currentLine.encodeAsHTML()}</code>"""
     }
 
     @Override
-    protected String formatCodeSnippetErrorLine(int currentLineNumber, Object currentLine) {
-        """<code class="line error"><span class="lineNumber">${currentLineNumber}:</span>${currentLine.encodeAsHTML()}</code>"""
+    protected String formatCodeSnippetErrorLine(int currentLineNumber, Object currentLine, Map attrs = [:]) {
+        """<code class="${attrs['lineErrorClass'] ?: 'line error'}"><span class="${attrs['lineNumberClass'] ?: 'lineNumber'}">${currentLineNumber}:</span>${currentLine.encodeAsHTML()}</code>"""
     }
 
     @Override