blob: 8fa5003d9ba7ddbd6a34e0fdb42696d389d3d067 (
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
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
|
package mill.api
import java.io.{InputStream, OutputStream}
/**
* Misc IO utilities, eventually probably should be pushed upstream into
* ammonite-ops
*/
object IO {
/**
* Pump the data from the `src` stream into the `dest` stream.
*/
def stream(src: InputStream, dest: OutputStream): Unit = {
val buffer = new Array[Byte](4096)
while ({
src.read(buffer) match {
case -1 => false
case n =>
dest.write(buffer, 0, n)
true
}
}) ()
}
/**
* Unpacks the given `src` path into the context specific destination directory.
* @param src The ZIP file
* @param dest The relative ouput folder under the context specifix destination directory.
* @param ctx The target context
* @return The [[PathRef]] to the unpacked folder.
*/
def unpackZip(src: os.Path, dest: os.RelPath = "unpacked")(implicit ctx: Ctx.Dest): PathRef = {
val byteStream = os.read.inputStream(src)
val zipStream = new java.util.zip.ZipInputStream(byteStream)
while ({
zipStream.getNextEntry match {
case null => false
case entry =>
if (!entry.isDirectory) {
val entryDest = ctx.dest / dest / os.RelPath(entry.getName)
os.makeDir.all(entryDest / os.up)
val fileOut = new java.io.FileOutputStream(entryDest.toString)
IO.stream(zipStream, fileOut)
fileOut.close()
}
zipStream.closeEntry()
true
}
}) ()
PathRef(ctx.dest / dest)
}
}
import java.io.ByteArrayInputStream
/**
* A dummy input stream containing an empty byte array.
*/
object DummyInputStream extends ByteArrayInputStream(Array())
/**
* A dummy output stream that does nothing with what it consumes (think of it as `/dev/null`).
*/
object DummyOutputStream extends java.io.OutputStream {
override def write(b: Int): Unit = ()
override def write(b: Array[Byte]): Unit = ()
override def write(b: Array[Byte], off: Int, len: Int): Unit = ()
}
|