Add useExplicitWhiteList for debugging.

1 files changed, 27 insertions, 1 deletions

diff --git a/compiler/test/dotty/tools/StdLibSources.scala b/compiler/test/dotty/tools/StdLibSources.scala
index 2b37235fb..78a132426 100644
--- a/compiler/test/dotty/tools/StdLibSources.scala
+++ b/compiler/test/dotty/tools/StdLibSources.scala
@@ -6,11 +6,26 @@ import scala.io.Source
 
 object StdLibSources {
 
+  /* For debug only */
+  private val useExplicitWhiteList = false
+
   private final val stdLibPath = "../scala-scala/src/library/"
 
   def blacklistFile: String = "./test/dotc/scala-collections.blacklist"
+  private def whitelistFile: String = "./test/dotc/scala-collections.whitelist"
+
+  def whitelisted: List[String] = {
+    lazy val whitelistBasedOnBlacklist = all.diff(blacklisted)
+    if (!useExplicitWhiteList) {
+      whitelistBasedOnBlacklist
+    } else if (!new File(whitelistFile).exists()) {
+      genWhitelist(whitelistBasedOnBlacklist.map(_.replace(stdLibPath, "")))
+      whitelistBasedOnBlacklist
+    } else {
+      loadList(whitelistFile)
+    }
+  }
 
-  def whitelisted: List[String] = (all.toSet -- blacklisted).toList
   def blacklisted: List[String] = loadList(blacklistFile)
 
   def all: List[String] = {
@@ -22,6 +37,17 @@ object StdLibSources {
     collectAllFilesInDir(new File(stdLibPath), Nil)
   }
 
+  private def genWhitelist(list: List[String]): Unit = {
+    println(s"Generating $whitelistFile based on $blacklistFile")
+    val whitelist = new File(whitelistFile)
+    val p = new java.io.PrintWriter(whitelist)
+    try {
+      list.foreach(p.println)
+    } finally {
+      p.close()
+    }
+  }
+
   private def loadList(path: String): List[String] = Source.fromFile(path, "UTF8").getLines()
     .map(_.trim) // allow identation
     .filter(!_.startsWith("#")) // allow comment lines prefixed by #