org.specs2.analysis.ClassycleDependencyFinder.scala Maven / Gradle / Ivy

Go to download
package org.specs2
package analysis

import classycle._
import reflect._
import io._

/**
 * Implementation of the dependency finder using the classycle library.
 *
 * This implementation is faster and simpler than the CompilerDependencyFinder trait
 */
trait ClassycleDependencyFinder extends DependencyFinder {

  def getPackageDependents(packageName: String, sourceDir: String, targetDir: String): Seq[Dependency] = {

    val analyser = new Analyser(fs.filePaths(targetDir, "**/*.class").toArray)
    analyser.createClassGraph()

    analyser.getClassGraph.collect {
      case classVertex if ClassName.packageName(classVertex.getAttributes.asInstanceOf[ClassAttributes].getName) == packageName =>
        (0 until classVertex.getNumberOfIncomingArcs).map { i =>
          Dependency(ClassName.className(classVertex.getAttributes.asInstanceOf[ClassAttributes].getName),
            ClassName.className(classVertex.getTailVertex(i).getAttributes.asInstanceOf[ClassAttributes].getName))
        }
    }.flatMap(identity).distinct
  }
}