blob: 4a9548730ab19aa24c2d21a8bb7409eb0664bb11 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
/* NSC -- new Scala compiler
* Copyright 2005-2013 LAMP/EPFL
* @author Paul Phillips
*/
package scala.tools.nsc
package interpreter
import util.stringFromWriter
class Formatting(indent: Int) {
private val indentation = " " * indent
private def indenting(code: String): Boolean = {
/** Heuristic to avoid indenting and thereby corrupting """-strings and XML literals. */
val tokens = List("\"\"\"", "</", "/>")
val noIndent = (code contains "\n") && (tokens exists code.contains)
!noIndent
}
/** Indent some code by the width of the scala> prompt.
* This way, compiler error messages read better.
*/
def indentCode(code: String) = stringFromWriter(str =>
for (line <- code.lines) {
if (indenting(code)) str print indentation
str println line
str.flush()
}
)
}
object Formatting {
def forPrompt(prompt: String) = new Formatting(prompt.lines.toList.last.length)
}
|