made hilt generated class open

zeoflow · Oct 27, 2022 · c791c75 · c791c75
1 parent 75ce54f
commit c791c75
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/memo-processor/src/main/java/com/zeoflow/memo/processor/entity/EntityBuilder.kt b/memo-processor/src/main/java/com/zeoflow/memo/processor/entity/EntityBuilder.kt
@@ -33,8 +33,12 @@ class EntityBuilder(
         )
 
         val typeSpec = TypeSpec.classBuilder(className())
-            .addModifiers(KModifier.PUBLIC)
-            .addKdoc("Code generated by Memo's Compiler. @see [link](https://github.com/zeoflow/memo)\n\n")
+        if (entityData.isHilt) {
+            typeSpec.addModifiers(KModifier.OPEN,KModifier.PUBLIC)
+        } else {
+            typeSpec.addModifiers(KModifier.PUBLIC)
+        }
+        typeSpec.addKdoc("Code generated by Memo's Compiler. @see [link](https://github.com/zeoflow/memo)\n\n")
             .addKdoc("Code generated based on the [${entityData.entityName}]")
 
         if (entityData.isHilt) {