fix nullability of lambdas in KSTypeReference.toTypeName (#1756)

square · Dec 2, 2023 · 51b218b · 51b218b
1 parent 221bbbc
commit 51b218b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/interop/ksp/src/main/kotlin/com/squareup/kotlinpoet/ksp/KsTypes.kt b/interop/ksp/src/main/kotlin/com/squareup/kotlinpoet/ksp/KsTypes.kt
@@ -188,7 +188,7 @@ public fun KSTypeReference.toTypeName(
         receiver = elem.receiverType?.toTypeName(typeParamResolver),
         parameters = elem.functionParameters.map { ParameterSpec.unnamed(it.type.toTypeName(typeParamResolver)) },
         returnType = elem.returnType.toTypeName(typeParamResolver),
-      )
+      ).copy(nullable = resolve().isMarkedNullable)
     }
     else -> resolve().toTypeName(typeParamResolver, element?.typeArguments.orEmpty())
   }

diff --git a/...processor/src/test/kotlin/com/squareup/kotlinpoet/ksp/test/processor/TestProcessorTest.kt b/...processor/src/test/kotlin/com/squareup/kotlinpoet/ksp/test/processor/TestProcessorTest.kt
@@ -133,6 +133,7 @@ class TestProcessorTest {
                param4: Function0<String>,
                param5: Function1<String, String>,
                param6: (Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int) -> Unit,
+               param7: ((String) -> String)?,
              ) {
              }
 
@@ -296,6 +297,7 @@ class TestProcessorTest {
             Int,
             Int,
           ) -> Unit,
+          param7: ((String) -> String)?,
         ) {
         }