summaryrefslogtreecommitdiff
path: root/plugins/kotlin/project-wizard/core/src/org/jetbrains/kotlin/tools/projectWizard/templates/JsClientTemplate.kt
blob: 34f30bbb042e6e38a9c060cffa50db0d4d4efe89 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.

package org.jetbrains.kotlin.tools.projectWizard.templates

import org.jetbrains.annotations.NonNls
import org.jetbrains.kotlin.tools.projectWizard.WizardGradleRunConfiguration
import org.jetbrains.kotlin.tools.projectWizard.WizardRunConfiguration
import org.jetbrains.kotlin.tools.projectWizard.core.Reader
import org.jetbrains.kotlin.tools.projectWizard.core.buildList
import org.jetbrains.kotlin.tools.projectWizard.core.safeAs
import org.jetbrains.kotlin.tools.projectWizard.ir.buildsystem.*
import org.jetbrains.kotlin.tools.projectWizard.ir.buildsystem.gradle.*
import org.jetbrains.kotlin.tools.projectWizard.ir.buildsystem.gradle.multiplatform.DefaultTargetConfigurationIR
import org.jetbrains.kotlin.tools.projectWizard.moduleConfigurators.*
import org.jetbrains.kotlin.tools.projectWizard.plugins.buildSystem.BuildSystemPlugin
import org.jetbrains.kotlin.tools.projectWizard.plugins.kotlin.ModuleSubType
import org.jetbrains.kotlin.tools.projectWizard.plugins.kotlin.ModuleType
import org.jetbrains.kotlin.tools.projectWizard.plugins.kotlin.ProjectKind
import org.jetbrains.kotlin.tools.projectWizard.plugins.projectName
import org.jetbrains.kotlin.tools.projectWizard.settings.buildsystem.Module
import org.jetbrains.kotlin.tools.projectWizard.settings.buildsystem.ModuleKind
import org.jetbrains.kotlin.tools.projectWizard.transformers.interceptors.TemplateInterceptor
import org.jetbrains.kotlin.tools.projectWizard.transformers.interceptors.interceptTemplate

abstract class JsClientTemplate : Template() {
    override fun isApplicableTo(module: Module, projectKind: ProjectKind, reader: Reader): Boolean =
        module.configurator.moduleType == ModuleType.js
                && when (module.configurator) {
                    JsBrowserTargetConfigurator, MppLibJsBrowserTargetConfigurator -> true
                    BrowserJsSinglePlatformModuleConfigurator -> {
                        with(reader) {
                            inContextOfModuleConfigurator(module, module.configurator) {
                                JSConfigurator.kind.reference.notRequiredSettingValue == JsTargetKind.APPLICATION
                            }
                        }
                    }
                    else -> false
                }

    override fun Reader.createRunConfigurations(module: ModuleIR): List<WizardRunConfiguration> = buildList {
        if (module.originalModule.kind == ModuleKind.singlePlatformJsBrowser) {
            +WizardGradleRunConfiguration(
                org.jetbrains.kotlin.tools.projectWizard.KotlinNewProjectWizardBundle.message("module.template.js.simple.run.configuration.dev"),
                "browserDevelopmentRun",
                listOf("--continuous")
            )
            +WizardGradleRunConfiguration(
                org.jetbrains.kotlin.tools.projectWizard.KotlinNewProjectWizardBundle.message("module.template.js.simple.run.configuration.prod"),
                "browserProductionRun",
                listOf("--continuous")
            )
        }
    }

    override fun Reader.createInterceptors(module: ModuleIR): List<TemplateInterceptor> = buildList {
        +interceptTemplate(KtorServerTemplate) {
            applicableIf { buildFileIR ->
                if (module !is MultiplatformModuleIR) return@applicableIf false
                val tasks = buildFileIR.irsOfTypeOrNull<GradleConfigureTaskIR>() ?: return@applicableIf true
                tasks.none { it.taskAccess.safeAs<GradleNamedTaskAccessIR>()?.name?.endsWith("Jar") == true }
            }

            interceptAtPoint(template.routes) { value ->
                if (value.isNotEmpty()) return@interceptAtPoint value
                buildList {
                    +value
                    +"""
                    static("/static") {
                        resources()
                    }
                    """.trimIndent()
                }
            }

            interceptAtPoint(template.elements) { value ->
                if (value.isNotEmpty()) return@interceptAtPoint value
                buildList {
                    +value
                    +"""
                     div {
                        id = "root"
                     }
                    """.trimIndent()
                    +"""script(src = "/static/${indexFileName(module.originalModule)}.js") {}"""
                }
            }

            transformBuildFile { buildFileIR ->
                val jsSourceSetName = module.safeAs<MultiplatformModuleIR>()?.name ?: return@transformBuildFile null
                val jvmTarget = buildFileIR.targets.firstOrNull { target ->
                    target.safeAs<DefaultTargetConfigurationIR>()?.targetAccess?.type == ModuleSubType.jvm
                } as? DefaultTargetConfigurationIR ?: return@transformBuildFile null
                val jvmTargetName = jvmTarget.targetName

                val distributionTaskName = "$jsSourceSetName$BROWSER_DISTRIBUTION_TASK_SUFFIX"

                val jvmProcessResourcesTaskAccess = GradleNamedTaskAccessIR(
                    "${jvmTargetName}ProcessResources",
                    "Copy"
                )

                val jvmProcessResourcesTaskConfiguration = run {
                    val distributionTaskVariable = CreateGradleValueIR(
                        distributionTaskName,
                        GradleNamedTaskAccessIR(distributionTaskName)
                    )
                    val from = GradleCallIr(
                        "from",
                        listOf(
                            GradlePropertyAccessIR(distributionTaskName)
                        )
                    )
                    GradleConfigureTaskIR(
                        jvmProcessResourcesTaskAccess,
                        irs = listOf(
                            distributionTaskVariable,
                            from
                        )
                    )
                }

                val jvmJarTaskAccess = GradleNamedTaskAccessIR(
                    "${jvmTargetName}Jar",
                    "Jar"
                )

                val runTaskConfiguration = run {
                    val taskAccess = GradleNamedTaskAccessIR("run", "JavaExec")
                    val classpath = GradleCallIr("classpath", listOf(jvmJarTaskAccess))
                    GradleConfigureTaskIR(
                        taskAccess,
                        dependsOn = listOf(jvmJarTaskAccess),
                        irs = listOf(classpath)
                    )
                }

                buildFileIR.withIrs(jvmProcessResourcesTaskConfiguration, runTaskConfiguration)
            }
        }
    }

    override fun Reader.getAdditionalSettings(module: Module): Map<String, Any> = withSettingsOf(module) {
        jsSettings(module)
    }

    protected fun Reader.jsSettings(module: Module): Map<String, String> {
        return mapOf("indexFile" to indexFileName(module))
    }

    private fun Reader.indexFileName(
        module: Module
    ): String {
        val buildFiles = BuildSystemPlugin.buildFiles.propertyValue
        return if (buildFiles.size == 1) projectName else module.parent?.name ?: module.name
    }

    protected fun hasKtorServNeighbourTarget(module: ModuleIR) =
        module.safeAs<MultiplatformModuleIR>()
            ?.neighbourTargetModules()
            .orEmpty()
            .any { it.template is KtorServerTemplate }

    companion object {
        @NonNls
        private const val BROWSER_DISTRIBUTION_TASK_SUFFIX = "BrowserDistribution"
    }
}