gratatouille.processor.GratatouilleProcessor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gratatouille-processor Show documentation

Cook delicious Gradle plugins

The newest version!

package gratatouille.processor

import com.google.devtools.ksp.processing.*
import com.google.devtools.ksp.symbol.KSAnnotated
import com.google.devtools.ksp.symbol.KSFunctionDeclaration
import com.squareup.kotlinpoet.ksp.writeTo

class GratatouilleProcessor(
  private val codeGenerator: CodeGenerator,
  private val logger: KSPLogger,
  private val options: Map,
) : SymbolProcessor {
  override fun process(resolver: Resolver): List {
    val symbols = resolver.getSymbolsWithAnnotation("gratatouille.GTaskAction")

    val coordinates = options.get("implementationCoordinates")
    symbols.forEach {
      when (it) {
        is KSFunctionDeclaration -> {
          it.toGTaskAction(coordinates).apply {
            entryPoint().writeTo(codeGenerator, Dependencies.ALL_FILES)

            taskFile().let { fileSpec ->
              val file = if (implementationCoordinates != null) {
                codeGenerator.createNewFile(
                  Dependencies.ALL_FILES,
                  "",
                  "META-INF/gratatouille/${fileSpec.packageName.replace(".", "/")}/${fileSpec.name}.kt",
                  ""
                )
              } else {
                codeGenerator.createNewFile(
                  Dependencies.ALL_FILES,
                  fileSpec.packageName,
                  fileSpec.name,
                )
              }
              file.writer().use {
                fileSpec.writeTo(it)
              }
            }
          }
        }

        else -> error("@GTaskAction is only valid on functions")
      }
    }
    return emptyList()
  }
}

class GratatouilleProcessorProvider : SymbolProcessorProvider {
  override fun create(
    environment: SymbolProcessorEnvironment
  ): SymbolProcessor {
    return GratatouilleProcessor(
      environment.codeGenerator,
      environment.logger,
      environment.options
    )
  }
}

internal val classpathProperty = Property(InputFiles, classpath, false, false, false)