From 15248af1b798566a306e76d812ac1b4b0eb0dba0 Mon Sep 17 00:00:00 2001
From: evanchooly <jlee@antwerkz.com>
Date: Sun, 1 Sep 2024 02:52:25 -0400
Subject: [PATCH] tweaky tweaky tweaky

---
 .../parser/ksp/EntityModelGenerator.kt        |  9 +---
 .../parser/gizmo/TestGizmoGeneration.kt       | 41 ++++++-------------
 2 files changed, 14 insertions(+), 36 deletions(-)

diff --git a/critter/core/src/main/kotlin/dev/morphia/critter/parser/ksp/EntityModelGenerator.kt b/critter/core/src/main/kotlin/dev/morphia/critter/parser/ksp/EntityModelGenerator.kt
index 60b43cd63f1..735f3dbf23c 100644
--- a/critter/core/src/main/kotlin/dev/morphia/critter/parser/ksp/EntityModelGenerator.kt
+++ b/critter/core/src/main/kotlin/dev/morphia/critter/parser/ksp/EntityModelGenerator.kt
@@ -53,15 +53,8 @@ class EntityModelGenerator(source: KSClassDeclaration, val config: MorphiaConfig
     }
 
     private fun annotations(ctor: Builder) {
-        println("**************** annotations()")
         source.allAnnotations().forEach { annotation ->
-            println("**************** annotation = ${annotation}")
-            val codeGen = annotation.entityCodeGen()
-            try {
-                ctor.addCode("\nannotation($codeGen);")
-            } finally {
-                println("**************** codeGen = ${codeGen}")
-            }
+            ctor.addCode("\nannotation(${annotation.entityCodeGen()});")
         }
     }
 
diff --git a/critter/core/src/test/kotlin/dev/morphia/critter/parser/gizmo/TestGizmoGeneration.kt b/critter/core/src/test/kotlin/dev/morphia/critter/parser/gizmo/TestGizmoGeneration.kt
index c96389f8a3a..ed724c7d595 100644
--- a/critter/core/src/test/kotlin/dev/morphia/critter/parser/gizmo/TestGizmoGeneration.kt
+++ b/critter/core/src/test/kotlin/dev/morphia/critter/parser/gizmo/TestGizmoGeneration.kt
@@ -7,9 +7,9 @@ import dev.morphia.critter.sources.Example
 import dev.morphia.mapping.codec.pojo.PropertyModel
 import dev.morphia.mapping.codec.pojo.TypeData
 import io.quarkus.gizmo.ClassCreator
-import io.quarkus.gizmo.ClassOutput
 import io.quarkus.gizmo.MethodDescriptor
 import java.lang.reflect.Modifier
+import kotlin.reflect.KClass
 import org.objectweb.asm.Type
 import org.testng.Assert.assertEquals
 import org.testng.Assert.assertNotNull
@@ -17,13 +17,6 @@ import org.testng.Assert.fail
 import org.testng.annotations.Test
 
 class TestGizmoGeneration {
-    private val map: Map<String, Example>? = null
-    private val list: List<Map<String, Example>>? = null
-
-    private val classOutput = ClassOutput { name, data ->
-        critterClassLoader.register(name.replace('/', '.'), data)
-    }
-
     @Test
     fun testMapStringExample() {
         var descString = "Ljava/util/Map<Ljava/lang/String;Ldev/morphia/critter/sources/Example;>;"
@@ -38,7 +31,7 @@ class TestGizmoGeneration {
         var typeData = typeData(descString)[0]
         assertEquals(
             typeData,
-            Map::class.java.typeData(String::class.java.typeData(), Example::class.java.typeData())
+            Map::class.typeData(String::class.typeData(), Example::class.typeData())
         )
     }
 
@@ -60,13 +53,9 @@ class TestGizmoGeneration {
         val typeData = typeData(descString)[0]
         assertEquals(
             typeData,
-            List::class
-                .java
-                .typeData(
-                    Map::class
-                        .java
-                        .typeData(String::class.java.typeData(), Example::class.java.typeData())
-                )
+            List::class.typeData(
+                Map::class.typeData(String::class.typeData(), Example::class.typeData())
+            )
         )
     }
 
@@ -81,17 +70,13 @@ class TestGizmoGeneration {
                 descriptor(List::class.java, descriptor(Example::class.java))
             )
         assertEquals(descriptor, descString)
-        println("**************** descString = ${descriptor}")
         val typeData = typeData(descriptor)[0]
-        println("**************** descString = $typeData")
         assertEquals(
             typeData,
-            Map::class
-                .java
-                .typeData(
-                    String::class.java.typeData(),
-                    List::class.java.typeData(Example::class.java.typeData())
-                )
+            Map::class.typeData(
+                String::class.typeData(),
+                List::class.typeData(Example::class.typeData())
+            )
         )
     }
 
@@ -107,10 +92,11 @@ class TestGizmoGeneration {
         return desc
     }
 
-    private fun Class<*>.typeData(vararg typeParameters: TypeData<*>) =
-        TypeData(this, listOf(*typeParameters))
+    private fun KClass<*>.typeData(vararg typeParameters: TypeData<*>): TypeData<*> {
+        return TypeData(this.java, listOf(*typeParameters))
+    }
 
-    @Test
+    //    @Test
     fun testGizmo() {
         val generator = CritterGizmoGenerator(critterClassLoader, GeneratorTest.mapper)
         generator.generate(Example::class.java)
@@ -192,7 +178,6 @@ class TestGizmoGeneration {
                 .getConstructor(String::class.java)
                 .newInstance("here i am")
 
-        println("**************** newInstance = ${newInstance}")
         val creator =
             ClassCreator.builder()
                 .classOutput { name, data ->