support abstract receriver

openrewrite · Oct 8, 2023 · 023de9c · 023de9c
1 parent fc1afa9
commit 023de9c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java b/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java
@@ -725,7 +725,7 @@ public J visitTypeParameterList(KtTypeParameterList list, ExecutionContext data)
     @Override
     public J visitTypeProjection(KtTypeProjection typeProjection, ExecutionContext data) {
         if (typeProjection.getTypeReference() != null) {
-            return typeProjection.getTypeReference().accept(this, data);
+            return typeProjection.getTypeReference().accept(this, data).withPrefix(prefix(typeProjection));
         }
 
         if (typeProjection.getProjectionKind() == KtProjectionKind.STAR) {
@@ -1737,7 +1737,7 @@ public J visitProperty(KtProperty property, ExecutionContext data) {
                 variables
         );
 
-        if (getter != null || setter != null) {
+        if (getter != null || setter != null || receiver != null) {
             return new K.Property(
                     randomId(),
                     Space.EMPTY,
@@ -1900,8 +1900,6 @@ public J visitUserType(KtUserType type, ExecutionContext data) {
             );
         }
 
-
-        // TODO: fix NPE.
         return nameTree;
     }
 
@@ -1961,6 +1959,8 @@ private J.Modifier.Type mapModifierType(PsiElement modifier) {
                 return J.Modifier.Type.Sealed;
             case "open":
                 return J.Modifier.Type.LanguageExtension;
+            case "abstract":
+                return J.Modifier.Type.Abstract;
             default:
                 throw new UnsupportedOperationException("Unsupported ModifierType : " + modifier);
         }

diff --git a/src/test/java/org/openrewrite/kotlin/tree/VariableDeclarationTest.java b/src/test/java/org/openrewrite/kotlin/tree/VariableDeclarationTest.java
@@ -443,7 +443,7 @@ void abstractReceiver() {
             """
               class SomeParameterized<T>
               abstract class Test {
-                  abstract val SomeParameterized < Int > . receivedMember : Int
+                  abstract val SomeParameterized <  Int > . receivedMember : Int
               }
               """
           )