blob: 22a2e8f9526fc6d9e2975287b26eb90316ebd9f0 (
plain)
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
|
package org.jetbrains.dokka.renderers
import com.intellij.util.io.isDirectory
import java.io.File
import java.io.IOException
import java.net.URI
import java.nio.file.*
class FileWriter(val root: String, override val extension: String): OutputWriter {
private val createdFiles: MutableSet<String> = mutableSetOf()
override fun write(path: String, text: String, ext: String) {
if (createdFiles.contains(path)) {
println("ERROR. An attempt to write $root/$path several times!")
return
}
createdFiles.add(path)
try {
// println("Writing $root/$path$ext")
val dir = Paths.get(root, path.dropLastWhile { it != '/' }).toFile()
dir.mkdirsOrFail()
Paths.get(root, "$path$ext").toFile().writeText(text)
} catch (e: Throwable) {
println("Failed to write $this. ${e.message}")
e.printStackTrace()
}
}
override fun writeResources(pathFrom: String, pathTo: String) {
val rebase = fun(path: String) =
"$pathTo/${path.removePrefix(pathFrom)}"
val dest = Paths.get(root, pathTo).toFile()
dest.mkdirsOrFail()
val uri = javaClass.getResource(pathFrom).toURI()
val fs = getFileSystemForURI(uri)
val path = fs.getPath(pathFrom)
for (file in Files.walk(path).iterator()) {
if (file.isDirectory()) {
val dirPath = file.toAbsolutePath().toString()
Paths.get(root, rebase(dirPath)).toFile().mkdirsOrFail()
} else {
val filePath = file.toAbsolutePath().toString()
Paths.get(root, rebase(filePath)).toFile().writeBytes(
javaClass.getResourceAsStream(filePath).readBytes()
)
}
}
}
private fun File.mkdirsOrFail() {
if (!mkdirs() && !exists()) {
throw IOException("Failed to create directory $this")
}
}
private fun getFileSystemForURI(uri: URI): FileSystem =
try {
FileSystems.newFileSystem(uri, emptyMap<String, Any>())
} catch (e: FileSystemAlreadyExistsException) {
FileSystems.getFileSystem(uri)
}
}
|
