Merge pull request #65 from zeoflow/bug/hilt

Made hilt generated class open
zeoflow · Oct 27, 2022 · a1af0ef · a1af0ef
2 parents 75ce54f + c791c75
commit a1af0ef
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) {