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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
/* __ *\
** ________ ___ / / ___ __ ____ Scala.js tools **
** / __/ __// _ | / / / _ | __ / // __/ (c) 2013-2014, LAMP/EPFL **
** __\ \/ /__/ __ |/ /__/ __ |/_// /_\ \ http://scala-js.org/ **
** /____/\___/_/ |_/____/_/ | |__/ /____/ **
** |/____/ **
\* */
package scala.scalajs.tools.sourcemap
import scala.annotation.tailrec
import scala.collection.mutable
import java.io._
import java.util.regex.Pattern
import java.net.{ URI, URISyntaxException }
import scala.scalajs.ir.Position
import scala.scalajs.tools.{javascript => js}
import scala.scalajs.tools.io._
import scala.scalajs.tools.optimizer.JSTreeBuilder
class JSFileBuilder(val name: String,
protected val outputWriter: Writer) extends JSTreeBuilder {
def addLine(line: String): Unit = {
outputWriter.write(line)
outputWriter.write('\n')
}
def addLines(lines: Seq[String]): Unit =
lines.foreach(addLine)
def addFile(file: VirtualJSFile): Unit =
addPartsOfFile(file)(!_.startsWith("//# sourceMappingURL="))
def addPartsOfFile(file: VirtualJSFile)(selector: String => Boolean): Unit = {
for (line <- file.readLines() if selector(line))
addLine(line)
}
/** Add a JavaScript tree representing a statement.
* The tree must be a valid JavaScript tree (typically obtained by
* desugaring a full-fledged IR tree).
*/
def addJSTree(tree: js.Trees.Tree): Unit = {
val printer = new js.Printers.JSTreePrinter(outputWriter)
printer.printTopLevelTree(tree)
// Do not close the printer: we do not have ownership of the writers
}
/** Closes the underlying writer(s).
*/
def closeWriters(): Unit = {
outputWriter.close()
}
}
class JSFileBuilderWithSourceMapWriter(n: String, ow: Writer,
protected val sourceMapWriter: SourceMapWriter)
extends JSFileBuilder(n, ow) {
override def addLine(line: String): Unit = {
super.addLine(line)
sourceMapWriter.nextLine()
}
private final val NotSelected = -1
override def addPartsOfFile(file: VirtualJSFile)(
selector: String => Boolean): Unit = {
val br = new BufferedReader(file.reader)
try {
// Select lines, and remember offsets
val offsets = new mutable.ArrayBuffer[Int] // (maybe NotSelected)
val selectedLineLengths = new mutable.ArrayBuffer[Int]
var line: String = br.readLine()
var selectedCount = 0
while (line != null) {
if (selector(line)) {
super.addLine(line) // super call not to advance line in source map
offsets += selectedCount
selectedLineLengths += line.length
selectedCount += 1
} else {
offsets += NotSelected
}
line = br.readLine()
}
/* We ignore a potential source map.
* This happens typically for corejslib.js and other helper files
* written directly in JS.
* We generate a fake line-by-line source map for these on the fly
*/
val sourceFile = file.toURI
for (lineNumber <- 0 until offsets.size) {
val offset = offsets(lineNumber)
if (offset != NotSelected) {
val originalPos = Position(sourceFile, lineNumber, 0)
sourceMapWriter.startNode(0, originalPos, None)
sourceMapWriter.endNode(selectedLineLengths(offset))
sourceMapWriter.nextLine()
}
}
} finally {
br.close()
}
}
override def addJSTree(tree: js.Trees.Tree): Unit = {
val printer = new js.Printers.JSTreePrinterWithSourceMap(
outputWriter, sourceMapWriter)
printer.printTopLevelTree(tree)
// Do not close the printer: we do not have ownership of the writers
}
override def complete(): Unit = {
super.complete()
sourceMapWriter.complete()
}
}
class JSFileBuilderWithSourceMap(n: String, ow: Writer,
sourceMapOutputWriter: Writer,
relativizeSourceMapBasePath: Option[URI] = None)
extends JSFileBuilderWithSourceMapWriter(
n, ow,
new SourceMapWriter(sourceMapOutputWriter, n,
relativizeSourceMapBasePath)) {
override def complete(): Unit = {
addLine("//# sourceMappingURL=" + name + ".map")
super.complete()
}
override def closeWriters(): Unit = {
super.closeWriters()
sourceMapOutputWriter.close()
}
}
|