From b28eb4112117cf9bba703dc62e7969636642045e Mon Sep 17 00:00:00 2001
From: Kingsley Adio <adiksonline@gmail.com>
Date: Mon, 27 Jan 2025 16:26:34 +0100
Subject: [PATCH] Simplify AppComponentHandler generated code (#101)

---
 .../whetstone/compiler/handlers/AppComponentHandler.kt     | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt b/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt
index 8ac13af..812b57c 100644
--- a/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt
+++ b/whetstone-compiler/src/main/java/com/deliveryhero/whetstone/compiler/handlers/AppComponentHandler.kt
@@ -67,12 +67,11 @@ internal class AppComponentHandler : CodegenHandler {
             .addFunction(
                 FunSpec.builder("create")
                     .addParameter(appParam)
-                    .returns(generatedComponentCn)
+                    .returns(applicationComponentCn)
                     .addModifiers(KModifier.PUBLIC, KModifier.OVERRIDE)
                     .addCode(CodeBlock.of(
-                        "return DaggerGeneratedApplicationComponent.factory().create(%L) as %T",
-                        appParam.name,
-                        generatedComponentCn,
+                        "return DaggerGeneratedApplicationComponent.factory().create(%L)",
+                        appParam.name
                     ))
                     .build())
             .build()