• Home
• com.github.xuwei-k
• sbt-class-diagram_sbt2.0.0-M2_3
• 0.3.0
• source code

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

wei-k.sbt-class-diagram_sbt2.0.0-M2_3.0.3.0.source-code.Reflect.scala Maven / Gradle / Ivy

package diagram

object Reflect {

  private def getSuperClasses(clazz: Class[?]): List[Class[?]] = {

    @scala.annotation.tailrec
    def loop(o: Class[?], result: List[Class[?]]): List[Class[?]] = {
      val superClass = o.getSuperclass
      if (superClass == null) {
        result
      } else {
        loop(superClass, superClass :: result)
      }
    }

    loop(clazz, List(clazz))
  }

  def getAllClassAndTrait(classes: Class[?]*): List[Class[?]] = {

    def loop(c: Class[?], result: List[Class[?]]): List[Class[?]] = {
      val interfaces = c.getInterfaces.toList

      if (interfaces.size == 0) {
        result
      } else {
        interfaces.flatMap { i => loop(i, i :: result) }
      }
    }

    {
      {
        for {
          c <- classes
          clazz <- getSuperClasses(c)
          ret <- loop(clazz, List(clazz))
        } yield {
          ret
        }
      }.toSet -- classes
    }.toList
  }

}