Adding domain entitiesv0.1.0

1 files changed, 36 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala b/src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala
new file mode 100644
index 0000000..2be7ed7
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala
@@ -0,0 +1,36 @@
+package xyz.driver.pdsuicommon.resources
+
+import scala.io.{Codec, Source}
+
+trait ResourcesStorage {
+
+  /**
+    * @param resourcePath Don't forget / at start
+    */
+  def getFirstLine(resourcePath: String): String
+}
+
+object RealResourcesStorage extends ResourcesStorage {
+
+  def getFirstLine(resourcePath: String): String = {
+    val resourceUrl = getClass.getResource(resourcePath)
+    Option(resourceUrl) match {
+      case Some(url) =>
+        val source = Source.fromURL(resourceUrl)(Codec.UTF8)
+        try {
+          val lines = source.getLines()
+          if (lines.isEmpty) throw new RuntimeException(s"'$resourcePath' is empty")
+          else lines.next()
+        } finally {
+          source.close()
+        }
+      case None =>
+        throw new RuntimeException(s"Can not find the '$resourcePath'!")
+    }
+  }
+}
+
+object FakeResourcesStorage extends ResourcesStorage {
+
+  def getFirstLine(resourcePath: String): String = ""
+}