Add convenience anonymousClassBuilder(CodeBlock) (square#560)

* Add convenience anonymousClassBuilder(CodeBlock) Resolves square#558 * Fix indent
monperrus · May 13, 2017 · 40dfa0c · 40dfa0c
1 parent 48378f7
commit 40dfa0c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/main/java/com/squareup/javapoet/TypeSpec.java b/src/main/java/com/squareup/javapoet/TypeSpec.java
@@ -111,11 +111,15 @@ public static Builder enumBuilder(ClassName className) {
   }
 
   public static Builder anonymousClassBuilder(String typeArgumentsFormat, Object... args) {
-    return new Builder(Kind.CLASS, null, CodeBlock.builder()
+    return anonymousClassBuilder(CodeBlock.builder()
         .add(typeArgumentsFormat, args)
         .build());
   }
 
+  public static Builder anonymousClassBuilder(CodeBlock typeArguments) {
+    return new Builder(Kind.CLASS, null, typeArguments);
+  }
+
   public static Builder annotationBuilder(String name) {
     return new Builder(Kind.ANNOTATION, checkNotNull(name, "name == null"), null);
   }

diff --git a/src/test/java/com/squareup/javapoet/TypeSpecTest.java b/src/test/java/com/squareup/javapoet/TypeSpecTest.java
@@ -127,7 +127,7 @@ private boolean isJava8() {
     ParameterSpec thungParameter = ParameterSpec.builder(thungOfSuperFoo, "thung")
         .addModifiers(Modifier.FINAL)
         .build();
-    TypeSpec aSimpleThung = TypeSpec.anonymousClassBuilder("$N", thungParameter)
+    TypeSpec aSimpleThung = TypeSpec.anonymousClassBuilder(CodeBlock.of("$N", thungParameter))
         .superclass(simpleThungOfBar)
         .addMethod(MethodSpec.methodBuilder("doSomething")
             .addAnnotation(Override.class)