auto generated functions are now located in the same package as annot…

…ated class
Intex32 · Oct 16, 2022 · 2e254ab · 2e254ab
1 parent e87e719
commit 2e254ab
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/processor/src/main/kotlin/ron/ModelMapperProcessor.kt b/processor/src/main/kotlin/ron/ModelMapperProcessor.kt
@@ -24,15 +24,16 @@ class ModelMapperProcessor(
         if (!symbols.iterator().hasNext())
             return emptyList()
 
-        val file = codeGenerator.createNewFile(
-            dependencies = Dependencies(false, *resolver.getAllFiles().toList().toTypedArray()),
-            packageName = "ron",
-            fileName = "GeneratedMapperFunctions"
-        )
-
-        file += "package ron\n\n"
-        symbols.forEach { it.accept(Visitor(file), Unit) }
-        file.close()
+        symbols.forEach { classDeclaration ->
+            val file = codeGenerator.createNewFile(
+                dependencies = Dependencies(false, *resolver.getAllFiles().toList().toTypedArray()),
+                packageName = classDeclaration.packageName.asString(),
+                fileName = "AutoGeneratedMappers"
+            )
+            file += "package ${classDeclaration.packageName.asString()}\n"
+            classDeclaration.accept(Visitor(file), Unit)
+            file.close()
+        }
 
         return symbols
             .filterNot { it.validate() }