azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[tree generator] Extract printPropertyDeclaration util

Browse Source 
analogous to printFunctionDeclaration.

This decouples the printing logic from handling IR elements' fields,
and makes it easier to directly print arbitrary fields.
This commit is contained in:
Wojciech Litewka
2024-02-14 10:41:01 +01:00
committed by Space Team
parent 92ff782815
commit 5d7a5245ff
2 changed files with 105 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractFieldPrinter.kt
+31 -67
View File
@@ -6,10 +6,14 @@
package org.jetbrains.kotlin.generators.tree
import org.jetbrains.kotlin.descriptors.Modality
import org.jetbrains.kotlin.generators.tree.printer.VariableKind
import org.jetbrains.kotlin.generators.tree.printer.printBlock
import org.jetbrains.kotlin.generators.tree.printer.printKDoc
import org.jetbrains.kotlin.generators.tree.printer.printPropertyDeclaration
import org.jetbrains.kotlin.util.capitalizeDecapitalize.toLowerCaseAsciiOnly
import org.jetbrains.kotlin.utils.SmartPrinter
import org.jetbrains.kotlin.utils.withIndent
import java.lang.reflect.Modifier.isVolatile
abstract class AbstractFieldPrinter<Field : AbstractField<*>>(
private val printer: SmartPrinter,
@@ -34,78 +38,38 @@ abstract class AbstractFieldPrinter<Field : AbstractField<*>>(
modality: Modality? = null,
) {
printer.run {
if (!field.fromParent) {
printKDoc(field.kDoc)
}
field.deprecation?.let {
println("@Deprecated(")
withIndent {
println("message = \"", it.message, "\",")
println("replaceWith = ReplaceWith(\"", it.replaceWith.expression, "\"),")
println("level = DeprecationLevel.", it.level.name, ",")
}
println(")")
}
if (field.isVolatile) {
println("@", type<Volatile>().render())
}
val defaultValue = field.defaultValueInImplementation
printPropertyDeclaration(
name = field.name,
type = actualTypeOfField(field),
kind = if (forceMutable(field) || field.isFinal && field.isMutable) VariableKind.VAR else VariableKind.VAL,
inConstructor = inConstructor,
visibility = field.visibility,
modality = modality,
override = override,
isLateinit = field.isLateinit,
isVolatile = field.isVolatile,
optInAnnotation = field.optInAnnotation,
printOptInWrapped = defaultValue != null,
deprecation = field.deprecation,
kDoc = field.kDoc,
initializer = defaultValue.takeUnless { field.withGetter }
)
println()
field.optInAnnotation?.let {
val rendered = it.render()
when {
defaultValue != null -> println("@OptIn(", rendered, "::class)")
inConstructor -> println("@property:", rendered)
else -> println("@", rendered)
}
}
if (field.visibility != Visibility.PUBLIC) {
print(field.visibility.name.toLowerCaseAsciiOnly(), " ")
}
modality?.let {
print(it.name.toLowerCaseAsciiOnly(), " ")
}
if (override) {
print("override ")
}
if (field.isLateinit) {
print("lateinit ")
}
if (forceMutable(field) || field.isFinal && field.isMutable) {
print("var ")
} else {
print("val ")
}
print(field.name, ": ", actualTypeOfField(field).render())
if (inConstructor) {
print(",")
}
if (defaultValue == null) {
println()
return
}
if (field.withGetter) {
println()
pushIndent()
print("get()")
}
println(" = $defaultValue")
field.customSetter?.let {
println("set(value) {")
if (defaultValue != null && field.withGetter) {
withIndent {
println(it)
println("get() = $defaultValue")
}
println("}")
}
if (field.withGetter) {
popIndent()
field.customSetter?.let {
withIndent {
print("set(value)")
printBlock {
println(it)
}
}
}
}
}
generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/printer/printUtils.kt
+74
View File
@@ -190,6 +190,80 @@ inline fun SmartPrinter.printFunctionWithBlockBody(
printBlock(blockBody)
}
context(ImportCollector)
fun SmartPrinter.printPropertyDeclaration(
name: String,
type: TypeRef,
kind: VariableKind,
inConstructor: Boolean = false,
visibility: Visibility = Visibility.PUBLIC,
modality: Modality? = null,
override: Boolean = false,
isLateinit: Boolean = false,
isVolatile: Boolean = false,
kDoc: String? = null,
optInAnnotation: ClassRef<*>? = null,
printOptInWrapped: Boolean = false,
deprecation: Deprecated? = null,
initializer: String? = null,
) {
printKDoc(kDoc)
deprecation?.let {
println("@Deprecated(")
withIndent {
println("message = \"", it.message, "\",")
println("replaceWith = ReplaceWith(\"", it.replaceWith.expression, "\"),")
println("level = DeprecationLevel.", it.level.name, ",")
}
println(")")
}
if (isVolatile) {
println("@", type<Volatile>().render())
}
optInAnnotation?.let {
val rendered = it.render()
when {
printOptInWrapped -> println("@OptIn(", rendered, "::class)")
inConstructor -> println("@property:", rendered)
else -> println("@", rendered)
}
}
if (visibility != Visibility.PUBLIC) {
print(visibility.name.toLowerCaseAsciiOnly(), " ")
}
modality?.let {
print(it.name.toLowerCaseAsciiOnly(), " ")
}
if (override) {
print("override ")
}
if (isLateinit) {
print("lateinit ")
}
when (kind) {
VariableKind.PARAMETER -> {}
VariableKind.VAL -> print("val ")
VariableKind.VAR -> print("var ")
}
print(name, ": ", type.render())
if (initializer != null) {
print(" = $initializer")
}
if (inConstructor) {
print(",")
}
}
enum class VariableKind { VAL, VAR, PARAMETER }
inline fun SmartPrinter.printBlock(body: () -> Unit) {
println(" {")
withIndent(body)