Add example

openrewrite · Nov 7, 2023 · 492a94e · 492a94e
1 parent b3275c5
commit 492a94e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 8 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/Assertions.java b/src/main/java/org/openrewrite/kotlin/Assertions.java
@@ -32,6 +32,7 @@
 import org.openrewrite.marker.Markers;
 import org.openrewrite.test.SourceSpec;
 import org.openrewrite.test.SourceSpecs;
+import org.openrewrite.test.UncheckedConsumer;
 import org.opentest4j.AssertionFailedError;
 
 import java.util.Collections;
@@ -181,14 +182,20 @@ public Space visitSpace(Space space, Space.Location loc, Integer integer) {
         };
     }
 
+    public static UncheckedConsumer<SourceSpec<?>> spaceConscious() {
+        return source -> {
+            if (source.getSourceFileType() == K.CompilationUnit.class) {
+                SourceSpec<K.CompilationUnit> kotlinSourceSpec = (SourceSpec<K.CompilationUnit>) source;
+                kotlinSourceSpec.afterRecipe(spaceConscious(kotlinSourceSpec));
+            }
+        };
+    }
+
     public static ThrowingConsumer<K.CompilationUnit> spaceConscious(SourceSpec<K.CompilationUnit> spec) {
         return cu -> {
             K.CompilationUnit visited = (K.CompilationUnit) new KotlinIsoVisitor<Integer>() {
-                final int max = 20;
-                final String buf = "\n                     ";
                 int id = 0;
 
-
                 @Override
                 public Space visitSpace(Space space, KSpace.Location loc, Integer integer) {
                     return next(space);
@@ -200,9 +207,6 @@ private Space next(Space space) {
                         return space;
                     }
                     return space.withComments(Collections.singletonList(new TextComment(true, Integer.toString(id++), "", Markers.EMPTY)));
-//                    boolean nl = space.getWhitespace().indexOf('\n') >= 0;
-//                    String ws = buf.substring(nl ? 0 : 1, (id++ % max) + 2);
-//                    return space.withWhitespace(ws);
                 }
 
                 @Override
@@ -223,7 +227,6 @@ public Space visitSpace(Space space, Space.Location loc, Integer integer) {
                     } else if (getCursor().firstEnclosing(J.Package.class) != null) {
                         return space;
                     }
-//                    return space.withComments(Collections.singletonList(new TextComment(true, Integer.toString(id++), "", Markers.EMPTY)));
                     return next(space);
                 }
             }.visit(cu, 0);
@@ -233,7 +236,7 @@ public Space visitSpace(Space space, Space.Location loc, Integer integer) {
                 ctx.putMessage(ExecutionContext.REQUIRE_PRINT_EQUALS_INPUT, false);
                 SourceFile cu2 = spec.getParser().build().parse(ctx, s).findFirst().get();
                 String s1 = cu2.printAll();
-                assertEquals(s, s1, "Parser is not space print idempotent");
+                assertEquals(s, s1, "Parser is not whitespace print idempotent");
             } catch (Exception e) {
                 fail(e);
             }

diff --git a/src/test/java/org/openrewrite/kotlin/tree/AnonymousFunctionTest.java b/src/test/java/org/openrewrite/kotlin/tree/AnonymousFunctionTest.java
@@ -23,6 +23,13 @@
 
 class AnonymousFunctionTest implements RewriteTest {
 
+//    @Override
+//    public void defaults(RecipeSpec spec) {
+//        spec
+//          .recipe(Recipe.noop())
+//          .allSources(spaceConscious());
+//    }
+
     @Test
     @Issue("https://github.com/openrewrite/rewrite-kotlin/issues/287")
     void noArgs() {