Read alternative sources from project as well

Make sure to filter out those falling under output dirs.

Author: Albert Meltzer

src/main/scala/org/antipathy/mvn_scalafmt/SourcesBuilder.scala

@@ -12,6 +12,7 @@ class SourcesBuilder(project: MavenProject) {
 
   private val sources = Set.newBuilder[File]
   private val build   = project.getBuild()
+  private val outpath = getCanonicalFile(build.getDirectory()).toPath()
 
   private[mvn_scalafmt] def resultSet(): Set[File] =
     try sources.result()
@@ -22,11 +23,13 @@ class SourcesBuilder(project: MavenProject) {
   def addMain(dirs: JList[File]): Unit =
     if (!add(dirs)) {
       appendPrimary(build.getSourceDirectory())
+      appendAlternative(project.getCompileSourceRoots())
     }
 
   def addTest(dirs: JList[File]): Unit =
     if (!add(dirs)) {
       appendPrimary(build.getTestSourceDirectory())
+      appendAlternative(project.getTestCompileSourceRoots())
     }
 
   private def add(dirs: JList[File]): Boolean = {
@@ -46,4 +49,7 @@ class SourcesBuilder(project: MavenProject) {
   private def appendPrimary(dir: String): Unit =
     sources += getCanonicalFile(dir + "/../scala")
 
+  private def appendAlternative(altDirs: JList[String]): Unit =
+    sources ++= altDirs.asScala.map(getCanonicalFile).filter(!_.toPath().startsWith(outpath))
+
 }

src/test/scala/org/antipathy/mvn_scalafmt/SourcesBuilderSpec.scala

@@ -75,14 +75,41 @@ class SourcesBuilderSpec extends AnyFlatSpec with Matchers {
     builder.resultSet() shouldBe empty
   }
 
+  it should "add empty both, filtering" in {
+    val mavenBuild = new Build() {
+      setSourceDirectory("/foo/src/main/java")
+      setTestSourceDirectory("/foo/src/test/java")
+    }
+    val project = getMavenProject(mavenBuild)
+    project.addCompileSourceRoot("/foo/src/main/scala")
+    project.addCompileSourceRoot("/out/src/main/scala")
+    project.addCompileSourceRoot("/out2/src/main/scala")
+    project.addTestCompileSourceRoot("/foo/src/test/scala")
+    project.addTestCompileSourceRoot("/out/src/test/scala")
+    project.addTestCompileSourceRoot("/out2/src/test/scala")
+
+    val builder = new SourcesBuilder(project)
+    builder.addMain(null)
+    builder.addTest(null)
+    builder.resultSet() shouldBe Set[File](
+      "/foo/src/main/scala",
+      "/out2/src/main/scala",
+      "/foo/src/test/scala",
+      "/out2/src/test/scala"
+    )
+    builder.resultSet() shouldBe empty
+  }
+
 }
 
 private object SourcesBuilderSpec {
 
-  def getMavenProject(mavenBuild: Build = null) =
+  def getMavenProject(mavenBuild: Build = new Build()) = {
+    mavenBuild.setDirectory("/out")
     new MavenProject(new Model { setBuild(mavenBuild) }) {
       setFile(new File("/xyz/pom.xml")) // sets basedir
     }
+  }
 
   implicit def implicitStringToFile(file: String): File = new File(file)