com.prt2121.kpop.KFile.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kpop-core Show documentation

generate Kotlin extensions from Java static methods

The newest version!

package com.prt2121.kpop

import com.prt2121.kpop.internal.KMethod
import java.io.File
import java.nio.file.Files
import kotlin.properties.Delegates

/**
 * Represents a kotlin file that corresponds to a Java file/class
 */
class KFile(ignoredImports: List = emptyList()) {
  /**
   * These are imports of classes that Kotlin advises against using and are replaced in
   * {@link #resolveKotlinTypeByName}
   */
  private val IGNORED_IMPORTS = listOf(
      "java.util.List",
      "android.support.annotation.CheckResult",
      "android.support.annotation.NonNull",
      "android.support.annotation.RequiresApi"
  ) + ignoredImports

  var fileName: String by Delegates.notNull()
  var packageName: String by Delegates.notNull()
  var clazz: String by Delegates.notNull()
  var extendedClass: String by Delegates.notNull()
  val methods = mutableListOf()
  val imports = mutableListOf()

  /** Generates the code and writes it to the desired directory */
  fun generate(directory: File) {
    val finalDir = finalDir(directory, packageName)

    finalDir?.let {
      Files.createDirectories(it.toPath())
    }

    File(finalDir, fileName).bufferedWriter().use { writer ->
      writer.append("package $packageName\n\n")

      imports.filter { !IGNORED_IMPORTS.contains(it) }
          .forEach { im ->
            writer.append("import $im\n")
          }

      methods.forEach { m ->
        writer.append("\n${m.generate(clazz)}\n")
      }
    }
  }

  override fun toString(): String =
      "KFile(fileName=$fileName, packageName=$packageName, clazz=$clazz, extendedClass=$extendedClass methods=$methods, imports=$imports)"

  companion object {
    internal fun finalDir(directory: File, packageName: String): File? {
      var directoryPath = directory.absolutePath
      if (!packageName.isEmpty()) {
        packageName.split('.').forEach {
          directoryPath += File.separator + it
        }
        return File(directoryPath)
      }
      return null
    }
  }
}