generate public component

Author: h908714124

compiler/src/main/java/io/jbock/simple/processor/writing/ComponentImpl.java

@@ -19,6 +19,7 @@
 
 import javax.annotation.processing.Generated;
 import javax.lang.model.element.ExecutableElement;
+import javax.lang.model.element.Modifier;
 import javax.lang.model.type.TypeMirror;
 import java.util.Collection;
 import java.util.Map;
@@ -48,6 +49,8 @@ private ComponentImpl(
 
     TypeSpec generate() {
         TypeSpec.Builder spec = TypeSpec.classBuilder(component.generatedClass()).addSuperinterface(component.element().asType());
+        spec.addModifiers(component.element().getModifiers().stream()
+                .filter(m -> m == PUBLIC || m == PROTECTED).toArray(Modifier[]::new));
         MethodSpec.Builder constructor = MethodSpec.constructorBuilder().addModifiers(PRIVATE);
         for (NamedBinding namedBinding : sorted.values()) {
             Binding b = namedBinding.binding();
@@ -78,6 +81,8 @@ TypeSpec generate() {
         component.factoryElement().ifPresent(factory -> {
             spec.addMethod(MethodSpec.methodBuilder("factory")
                     .addModifiers(STATIC)
+                    .addModifiers(component.element().getModifiers().stream()
+                            .filter(m -> m == PUBLIC).toArray(Modifier[]::new))
                     .returns(TypeName.get(factory.element().asType()))
                     .addStatement("return new $T()", factory.generatedClass())
                     .build());
@@ -86,6 +91,8 @@ TypeSpec generate() {
         component.builderElement().ifPresent(builder -> {
             spec.addMethod(MethodSpec.methodBuilder("builder")
                     .addModifiers(STATIC)
+                    .addModifiers(component.element().getModifiers().stream()
+                            .filter(m -> m == PUBLIC).toArray(Modifier[]::new))
                     .returns(TypeName.get(builder.element().asType()))
                     .addStatement("return new $T()", builder.generatedClass())
                     .build());

compiler/src/test/java/io/jbock/simple/processor/BuilderTest.java

@@ -142,7 +142,7 @@ void factoryParameter() {
                 "  }",
                 "",
                 "  @Component",
-                "  interface AComponent {",
+                "  public interface AComponent {",
                 "    A getA();",
                 "",
                 "    @Component.Builder",
@@ -158,7 +158,7 @@ void factoryParameter() {
                 .containsLines(
                         "package test;",
                         "",
-                        "final class TestClass_AComponent_Impl implements TestClass.AComponent {",
+                        "public final class TestClass_AComponent_Impl implements TestClass.AComponent {",
                         "  private final TestClass.A a;",
                         "",
                         "  private TestClass_AComponent_Impl(String s) {",
@@ -170,7 +170,7 @@ void factoryParameter() {
                         "    return a;",
                         "  }",
                         "",
-                        "  static TestClass.AComponent.Builder builder() {",
+                        "  public static TestClass.AComponent.Builder builder() {",
                         "    return new Builder_Impl();",
                         "  }",
                         "",