blob: e728f02c98d3136791df57b5c7e94129c55e5ab0 (
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
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
|
package com.getbootstrap.savage
import java.nio.charset.Charset
import java.nio.file.{FileSystems, FileSystem, Path}
import java.io.{IOException, InputStream}
import java.util.Scanner
import akka.event.LoggingAdapter
import scala.util.Try
import com.typesafe.config.Config
package object util {
val utf8Name = "UTF-8"
private val utf8 = Charset.forName(utf8Name)
implicit class Utf8String(str: String) {
def utf8Bytes: Array[Byte] = str.getBytes(utf8)
}
implicit class Utf8ByteArray(bytes: Array[Byte]) {
def utf8String: Try[String] = Try { new String(bytes, utf8) }
}
implicit class PrefixedString(str: String) {
def unprefix(prefix: String): Option[String] = {
if (str.startsWith(prefix)) {
Some(str.stripPrefix(prefix))
}
else {
None
}
}
}
private object UnixFileSystemString {
private lazy val unixFileSystem: FileSystem = {
// get a Unix-y FileSystem, or fail hard
val unixFsAbstractClass = Class.forName("sun.nio.fs.UnixFileSystem")
val systemFs = FileSystems.getDefault
if (unixFsAbstractClass isInstance systemFs) {
systemFs
}
else {
throw new SecurityException("The globbing for the editable files whitelist requires a Unix-y java.nio.file.FileSystem, but we could not obtain one.")
}
}
}
implicit class UnixFileSystemString(str: String) {
def asUnixGlob = UnixFileSystemString.unixFileSystem.getPathMatcher("glob:" + str)
def asUnixPath = UnixFileSystemString.unixFileSystem.getPath(str)
}
implicit class RichInputStream(stream: InputStream) {
def readUntilEofAsSingleUtf8String: String = {
val scanner = new Scanner(stream, utf8Name).useDelimiter("\\A")
val string = if (scanner.hasNext) {
scanner.next()
}
else {
""
}
scanner.close()
string
}
}
implicit class HexByteArray(array: Array[Byte]) {
import javax.xml.bind.DatatypeConverter
def asHexBytes: String = DatatypeConverter.printHexBinary(array).toLowerCase
}
implicit class RichPath(path: Path) {
@throws[SecurityException]
def deleteRecursively()(implicit log: LoggingAdapter) {
try {
java.nio.file.Files.walkFileTree(path, new DeleterFileVisitor(log))
}
catch {
case nsfe:java.nio.file.NoSuchFileException => {
// file/directory is already nonexistent
}
case exc:IOException => log.error(exc, s"Error while deleting ${path}")
}
}
}
implicit class RichConfig(config: Config) {
import java.util.concurrent.TimeUnit
import scala.concurrent.duration.FiniteDuration
def getFiniteDuration(path: String): FiniteDuration = FiniteDuration(config.getDuration(path, TimeUnit.SECONDS), TimeUnit.SECONDS)
}
}
|
