From cc86f8d609969b40793a227b9af4b41a18657dfb Mon Sep 17 00:00:00 2001
From: Jakob Odersky <jakob@driver.xyz>
Date: Thu, 22 Mar 2018 15:47:42 -0700
Subject: Add blob storage abstractions

---
 .../core/storage/FileSystemBlobStorage.scala       | 75 ++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 src/main/scala/xyz/driver/core/storage/FileSystemBlobStorage.scala

(limited to 'src/main/scala/xyz/driver/core/storage/FileSystemBlobStorage.scala')

diff --git a/src/main/scala/xyz/driver/core/storage/FileSystemBlobStorage.scala b/src/main/scala/xyz/driver/core/storage/FileSystemBlobStorage.scala
new file mode 100644
index 0000000..80076b6
--- /dev/null
+++ b/src/main/scala/xyz/driver/core/storage/FileSystemBlobStorage.scala
@@ -0,0 +1,75 @@
+package xyz.driver.core.storage
+
+import java.nio.file.{Files, Path, StandardCopyOption}
+
+import akka.stream.scaladsl.{FileIO, Sink, Source}
+import akka.util.ByteString
+import akka.{Done, NotUsed}
+
+import scala.collection.JavaConverters._
+import scala.concurrent.{ExecutionContext, Future}
+
+/** A blob store that is backed by a local filesystem. All objects are stored relative to the given
+  * root path. Slashes ('/') in blob names are treated as usual path separators and are converted
+  * to directories. */
+class FileSystemBlobStorage(root: Path)(implicit ec: ExecutionContext) extends BlobStorage {
+
+  private def ensureParents(file: Path): Path = {
+    Files.createDirectories(file.getParent())
+    file
+  }
+
+  private def file(name: String) = root.resolve(name)
+
+  override def uploadContent(name: String, content: Array[Byte]): Future[String] = Future {
+    Files.write(ensureParents(file(name)), content)
+    name
+  }
+  override def uploadFile(name: String, content: Path): Future[String] = Future {
+    Files.copy(content, ensureParents(file(name)), StandardCopyOption.REPLACE_EXISTING)
+    name
+  }
+
+  override def exists(name: String): Future[Boolean] = Future {
+    val path = file(name)
+    Files.exists(path) && Files.isReadable(path)
+  }
+
+  override def list(prefix: String): Future[Set[String]] = Future {
+    val dir = file(prefix)
+    Files
+      .list(dir)
+      .iterator()
+      .asScala
+      .map(p => root.relativize(p))
+      .map(_.toString)
+      .toSet
+  }
+
+  override def content(name: String): Future[Option[Array[Byte]]] = exists(name) map {
+    case true =>
+      Some(Files.readAllBytes(file(name)))
+    case false => None
+  }
+
+  override def download(name: String): Future[Option[Source[ByteString, NotUsed]]] = Future {
+    if (Files.exists(file(name))) {
+      Some(FileIO.fromPath(file(name)).mapMaterializedValue(_ => NotUsed))
+    } else {
+      None
+    }
+  }
+
+  override def upload(name: String): Future[Sink[ByteString, Future[Done]]] = Future {
+    val f = ensureParents(file(name))
+    FileIO.toPath(f).mapMaterializedValue(_.map(_ => Done))
+  }
+
+  override def delete(name: String): Future[String] = exists(name).map { e =>
+    if (e) {
+      Files.delete(file(name))
+    }
+    name
+  }
+
+}
-- 
cgit v1.2.3