Add stdlib whitelist loader.

1 files changed, 31 insertions, 0 deletions

diff --git a/compiler/test/dotty/tools/StdLibSources.scala b/compiler/test/dotty/tools/StdLibSources.scala
new file mode 100644
index 000000000..9afb81d20
--- /dev/null
+++ b/compiler/test/dotty/tools/StdLibSources.scala
@@ -0,0 +1,31 @@
+package dotty.tools
+
+import java.io.File
+
+import scala.io.Source
+
+object StdLibSources {
+
+  def whitelistFile: String = "./test/dotc/scala-collections.whitelist"
+  def blacklistFile: String = "./test/dotc/scala-collections.blacklist"
+
+  def whitelisted: List[String] = loadList(whitelistFile)
+  def blacklisted: List[String] = loadList(blacklistFile)
+
+  def all: List[String] = {
+    def collectAllFilesInDir(dir: File, acc: List[String]): List[String] = {
+      val files = dir.listFiles()
+      val acc2 = files.foldLeft(acc)((acc1, file) => if (file.isFile && file.getPath.endsWith(".scala")) file.getPath :: acc1 else acc1)
+      files.foldLeft(acc2)((acc3, file) => if (file.isDirectory) collectAllFilesInDir(file, acc3) else acc3)
+    }
+    collectAllFilesInDir(new File("../scala-scala/src/library/"), Nil)
+  }
+
+  private def loadList(path: String): List[String] = Source.fromFile(path, "UTF8").getLines()
+    .map(_.trim) // allow identation
+    .filter(!_.startsWith("#")) // allow comment lines prefixed by #
+    .map(_.takeWhile(_ != '#').trim) // allow comments in the end of line
+    .filter(_.nonEmpty)
+    .toList
+
+}