org.jetbrains.kotlin.kapt3.util.PrettyWithWorkarounds.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-annotation-processing Show documentation
Show all versions of kotlin-annotation-processing Show documentation
Annotation Processor for Kotlin
/*
* Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
*/
package org.jetbrains.kotlin.kapt3.util
import com.sun.tools.javac.code.Flags
import com.sun.tools.javac.tree.JCTree
import com.sun.tools.javac.tree.JCTree.JCAnnotation
import com.sun.tools.javac.tree.JCTree.JCVariableDecl
import com.sun.tools.javac.tree.Pretty
import com.sun.tools.javac.tree.TreeMaker
import com.sun.tools.javac.util.Context
import org.jetbrains.kotlin.kapt3.base.util.isJava11OrLater
import org.jetbrains.kotlin.load.java.JvmAnnotationNames.METADATA_FQ_NAME
import java.io.StringWriter
import java.io.Writer
private class PrettyWithWorkarounds(
private val context: Context, private val out: Writer, sourceOutput: Boolean,
private val renderMetadata: ((Pretty, JCAnnotation) -> String)?,
) : Pretty(out, sourceOutput) {
companion object {
private const val ENUM = Flags.ENUM.toLong()
}
override fun print(s: Any) {
out.write(s.toString())
}
override fun visitVarDef(tree: JCVariableDecl) {
if ((tree.mods.flags and ENUM) != 0L) {
// Pretty does not print annotations for enum values for some reason
printExpr(TreeMaker.instance(context).Modifiers(0, tree.mods.annotations))
if (isJava11OrLater()) {
// Print enums fully, there is an issue when using Pretty in JDK 11.
// See https://youtrack.jetbrains.com/issue/KT-33052.
print("/*public static final*/ ${tree.name}")
tree.init?.let { print(" /* = $it */") }
return
}
}
super.visitVarDef(tree)
}
override fun visitAnnotation(tree: JCAnnotation) {
if (renderMetadata != null && tree.annotationType.toString() == METADATA_FQ_NAME.asString()) {
print(renderMetadata.invoke(this, tree))
}
print("@")
printExpr(tree.annotationType)
print("(")
printExprs(tree.args)
print(")")
}
}
fun JCTree.prettyPrint(context: Context, renderMetadata: ((Pretty, JCAnnotation) -> String)? = null): String {
return StringWriter().apply { PrettyWithWorkarounds(context, this, false, renderMetadata).printStat(this@prettyPrint) }.toString()
}