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

[FIR] Allow to render function type with custom cone type renderer

Browse Source
This commit is contained in:
Ivan Kylchik
2021-03-11 15:55:37 +03:00
committed by TeamCityServer
parent 13d8b10be0
commit 30bd39aabf
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/cones/src/org/jetbrains/kotlin/fir/types/TypeRenderer.kt
+8 -6
View File
@@ -56,7 +56,7 @@ private fun ConeKotlinType.renderAttributes(): String {
return attributes.joinToString(" ", postfix = " ") { it.toString() }
}
private fun ConeTypeProjection.render(): String {
fun ConeTypeProjection.render(): String {
return when (this) {
ConeStarProjection -> "*"
is ConeKotlinTypeProjectionIn -> "in ${type.render()}"
@@ -65,8 +65,10 @@ private fun ConeTypeProjection.render(): String {
}
}
fun ConeKotlinType.renderFunctionType(kind: FunctionClassKind?, isExtension: Boolean): String {
if (!kind.withPrettyRender()) return render()
fun ConeKotlinType.renderFunctionType(
kind: FunctionClassKind?, isExtension: Boolean, renderType: ConeTypeProjection.() -> String = { render() }
): String {
if (!kind.withPrettyRender()) return renderType()
return buildString {
if (kind == FunctionClassKind.SuspendFunction) {
append("suspend ")
@@ -79,12 +81,12 @@ fun ConeKotlinType.renderFunctionType(kind: FunctionClassKind?, isExtension: Boo
val arguments = otherTypeArguments.subList(0, otherTypeArguments.size - 1)
val returnType = otherTypeArguments.last()
if (receiver != null) {
append(receiver.render())
append(receiver.renderType())
append(".")
}
append(arguments.joinToString(", ", "(", ")") { it.render() })
append(arguments.joinToString(", ", "(", ")") { it.renderType() })
append(" -> ")
append(returnType.render())
append(returnType.renderType())
}
}