Vite. Fix compilation tasks linkage

turansky · Sep 2, 2024 · a089010 · a089010
1 parent d0b9165
commit a089010
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/...c-gradle-plugin/src/main/kotlin/io/github/turansky/kfc/gradle/plugin/ApplicationPlugin.kt b/...c-gradle-plugin/src/main/kotlin/io/github/turansky/kfc/gradle/plugin/ApplicationPlugin.kt
@@ -11,11 +11,8 @@ class ApplicationPlugin : Plugin<Project> {
 
         applyKotlinMultiplatformPlugin(APPLICATION(bundler))
 
-        val productionTask = if (bundler == Webpack) Webpack.productionTask else Vite.productionTask
-        val developmentTask = if (bundler == Webpack) Webpack.developmentTask else Vite.developmentTask
-
-        linkWithModuleCompilation(productionTask, COMPILE_PRODUCTION)
-        linkWithModuleCompilation(developmentTask, COMPILE_DEVELOPMENT)
+        linkWithModuleCompilation(bundler.productionTask, COMPILE_PRODUCTION)
+        linkWithModuleCompilation(bundler.developmentTask, COMPILE_DEVELOPMENT)
 
         plugins.apply(BundlePlugin::class)
     }