Lazy. Calculate data from export

seskar/seskar-gradle-plugin/src/main/kotlin/seskar/gradle/plugin/LazyItemData.kt

@@ -0,0 +1,25 @@
+package seskar.gradle.plugin
+
+import seskar.gradle.plugin.Modules.ORIGINAL_MODULE_SUFFIX
+
+data class LazyItemData(
+    val name: String,
+    val originalFilePath: String,
+    val type: LazyItemType,
+    val export: String,
+)
+
+fun LazyItemData(
+    export: String,
+): LazyItemData {
+    val fileName = export
+        .removePrefix("get_")
+        .substringBefore("__react__component")
+
+    return LazyItemData(
+        name = fileName,
+        originalFilePath = "./$fileName$ORIGINAL_MODULE_SUFFIX",
+        type = LazyItemType.LAZY_REACT_COMPONENT,
+        export = export,
+    )
+}

seskar/seskar-gradle-plugin/src/main/kotlin/seskar/gradle/plugin/LazyItemFactory.kt

@@ -2,6 +2,6 @@ package seskar.gradle.plugin
 
 interface LazyItemFactory {
     fun create(
-        export: String,
+        data: LazyItemData,
     ): LazyItem?
 }

seskar/seskar-gradle-plugin/src/main/kotlin/seskar/gradle/plugin/LazyItemType.kt

@@ -0,0 +1,8 @@
+package seskar.gradle.plugin
+
+enum class LazyItemType {
+    LAZY_FUNCTION,
+    LAZY_REACT_COMPONENT,
+
+    ;
+}

seskar/seskar-gradle-plugin/src/main/kotlin/seskar/gradle/plugin/LazyModuleReader.kt

@@ -1,5 +1,6 @@
 package seskar.gradle.plugin
 
+import seskar.gradle.plugin.LazyItemType.LAZY_REACT_COMPONENT
 import java.io.FilterReader
 import java.io.Reader
 import java.io.StringReader
@@ -33,17 +34,17 @@ private fun lazyComponentTransformer(
     return StringReader(proxyBody)
 }
 
-private val LAZY_ITEM_FACTORIES = setOf(
-    ReactLazyComponentFactory(),
+private val LAZY_ITEM_FACTORY_MAP = mapOf(
+    LAZY_REACT_COMPONENT to ReactLazyComponentFactory(),
 )
 
 private fun createLazyItem(
     export: String,
-): LazyItem =
-    LAZY_ITEM_FACTORIES.asSequence()
-        .mapNotNull { it.create(export) }
-        .firstOrNull()
-        ?: error("Unable to transform export '$export' to lazy item.")
+): LazyItem {
+    val data = LazyItemData(export)
+    val factory = LAZY_ITEM_FACTORY_MAP.getValue(data.type)
+    return factory.create(data)
+}
 
 private fun getComponentProvider(
     content: String,

seskar/seskar-gradle-plugin/src/main/kotlin/seskar/gradle/plugin/ReactLazyComponentFactory.kt

@@ -1,33 +1,30 @@
 package seskar.gradle.plugin
 
-import seskar.gradle.plugin.Modules.ORIGINAL_MODULE_SUFFIX
+// language=javascript
+private val IMPORTS = """
+import { lazy } from "react"
+""".trimIndent()
 
 class ReactLazyComponentFactory :
     LazyItemFactory {
     override fun create(
-        export: String,
+        data: LazyItemData,
     ): LazyItem {
-        val fileName = export
-            .removePrefix("get_")
-            .substringBefore("__react__component")
-
-        val originalComponentPath = "./$fileName$ORIGINAL_MODULE_SUFFIX"
-
-        val componentName = "$fileName\$\$__react__lazy__component"
+        val componentName = "${data.name}\$\$__react__lazy__component"
 
         // language=javascript
         val body = """
         const $componentName = lazy(() => 
-            import("$originalComponentPath")
-                .then(module => module.${export}())
+            import("${data.originalFilePath}")
+                .then(module => module.${data.export}())
                 .then(component => ({ default: component }))
         )
         
-        export const $export = () => $componentName
+        export const ${data.export} = () => $componentName
         """.trimIndent()
 
         return LazyItem(
-            imports = """import { lazy } from "react"""",
+            imports = IMPORTS,
             body = body,
         )
     }