Move Gradle-related classes to idea-gradle module
This commit is contained in:
Dmitry Jemerov
-160
@@ -1,160 +0,0 @@
|
||||
/*
|
||||
* Copyright 2010-2016 JetBrains s.r.o.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.jetbrains.kotlin.idea.core.script
|
||||
|
||||
import com.intellij.execution.configurations.CommandLineTokenizer
|
||||
import com.intellij.openapi.diagnostic.Logger
|
||||
import com.intellij.openapi.externalSystem.util.ExternalSystemApiUtil
|
||||
import com.intellij.openapi.project.Project
|
||||
import org.gradle.tooling.ProjectConnection
|
||||
import org.jetbrains.kotlin.lexer.KotlinLexer
|
||||
import org.jetbrains.kotlin.lexer.KtTokens
|
||||
import org.jetbrains.kotlin.script.ScriptTemplatesProvider
|
||||
import org.jetbrains.plugins.gradle.service.execution.GradleExecutionHelper
|
||||
import org.jetbrains.plugins.gradle.settings.GradleExecutionSettings
|
||||
import java.io.File
|
||||
import java.util.*
|
||||
|
||||
abstract class AbstractGradleScriptTemplatesProvider(
|
||||
project: Project, override val id: String, private val templateClass: String, private val dependencySelector: Regex
|
||||
): ScriptTemplatesProvider {
|
||||
|
||||
private val gradleExeSettings: GradleExecutionSettings? by lazy {
|
||||
try {
|
||||
ExternalSystemApiUtil.getExecutionSettings<GradleExecutionSettings>(
|
||||
project,
|
||||
com.intellij.openapi.externalSystem.util.ExternalSystemApiUtil.toCanonicalPath(project.basePath!!),
|
||||
org.jetbrains.plugins.gradle.util.GradleConstants.SYSTEM_ID)
|
||||
}
|
||||
catch (e: Throwable) {
|
||||
// TODO: consider displaying the warning to the user
|
||||
Logger.getInstance(AbstractGradleScriptTemplatesProvider::class.java).warn("[kts] Cannot get gradle execution settings", e)
|
||||
null
|
||||
}
|
||||
}
|
||||
|
||||
private val gradleJvmOptions: List<String> by lazy {
|
||||
gradleExeSettings?.daemonVmOptions?.let { vmOptions ->
|
||||
CommandLineTokenizer(vmOptions).toList()
|
||||
.mapNotNull { it?.let { it as? String } }
|
||||
.filterNot(String::isBlank)
|
||||
.distinct()
|
||||
} ?: emptyList()
|
||||
}
|
||||
|
||||
internal val gradleLibDir: File by lazy {
|
||||
val gradleHome = gradleExeSettings?.gradleHome ?: error("Unable to get Gradle home directory")
|
||||
|
||||
File(gradleHome, "lib").let {
|
||||
it.takeIf { it.exists() && it.isDirectory } ?: error("Invalid Gradle libraries directory $it")
|
||||
}
|
||||
}
|
||||
|
||||
override val isValid: Boolean get() = true
|
||||
|
||||
override val templateClassNames get() = listOf(templateClass)
|
||||
|
||||
override val templateClasspath: List<File> get() {
|
||||
return gradleLibDir.listFiles { it ->
|
||||
/* an inference problem without explicit 'it', TODO: remove when fixed */
|
||||
dependencySelector.matches(it.name)
|
||||
}.takeIf { it.isNotEmpty() }?.asList() ?: error("Missing jars in gradle directory")
|
||||
}
|
||||
|
||||
override val environment: Map<String, Any?>? by lazy {
|
||||
mapOf(
|
||||
"gradleHome" to gradleExeSettings?.gradleHome?.let(::File),
|
||||
"projectRoot" to (project.basePath ?: project.baseDir.canonicalPath)?.let(::File),
|
||||
"gradleWithConnection" to { action: (ProjectConnection) -> Unit ->
|
||||
GradleExecutionHelper().execute(project.basePath!!, null) { action(it) } },
|
||||
"gradleJavaHome" to gradleExeSettings?.javaHome,
|
||||
"gradleJvmOptions" to gradleJvmOptions,
|
||||
"getScriptSectionTokens" to ::topLevelSectionCodeTextTokens
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
class GradleKotlinDSLTemplateProvider(project: Project) : AbstractGradleScriptTemplatesProvider(
|
||||
project,
|
||||
"Gradle Kotlin DSL",
|
||||
"org.gradle.kotlin.dsl.KotlinBuildScript",
|
||||
Regex("^gradle-(?:kotlin-dsl|core).*\\.jar\$")
|
||||
) {
|
||||
// TODO_R: check this against kotlin-dsl branch that uses daemon
|
||||
override val additionalResolverClasspath: List<File> get() =
|
||||
// additionally need compiler jar to load gradle resolver
|
||||
gradleLibDir.listFiles { file -> file.name.startsWith("kotlin-compiler-embeddable") }
|
||||
.firstOrNull()?.let(::listOf).orEmpty()
|
||||
}
|
||||
|
||||
class LegacyGradleScriptKotlinTemplateProvider(project: Project) : AbstractGradleScriptTemplatesProvider(
|
||||
project,
|
||||
"Gradle Script Kotlin",
|
||||
"org.gradle.script.lang.kotlin.KotlinBuildScript",
|
||||
Regex("^gradle-(?:script-kotlin|core).*\\.jar\$")
|
||||
)
|
||||
|
||||
class TopLevelSectionTokensEnumerator(script: CharSequence, identifier: String) : Enumeration<KotlinLexer> {
|
||||
|
||||
private val lexer = KotlinLexer().apply {
|
||||
start(script.replace(Regex.fromLiteral("\r"), ""))
|
||||
var depth = 0
|
||||
|
||||
loop@ while (tokenType != null) {
|
||||
when (tokenType) {
|
||||
KtTokens.IDENTIFIER -> if (depth == 0 && tokenText == identifier) {
|
||||
advance()
|
||||
skipWhiteSpaceAndComments()
|
||||
if (tokenType == KtTokens.LBRACE)
|
||||
break@loop
|
||||
}
|
||||
KtTokens.LBRACE -> depth += 1
|
||||
KtTokens.RBRACE -> depth -= 1
|
||||
}
|
||||
advance()
|
||||
}
|
||||
}
|
||||
|
||||
private var depth = 1
|
||||
private var finished = false
|
||||
|
||||
override fun hasMoreElements(): Boolean = !finished && lexer.tokenType != null
|
||||
|
||||
override fun nextElement(): KotlinLexer = lexer.apply {
|
||||
advance()
|
||||
when (tokenType) {
|
||||
KtTokens.LBRACE -> depth += 1
|
||||
KtTokens.RBRACE -> {
|
||||
if (depth == 1) {
|
||||
finished = true
|
||||
}
|
||||
depth -= 1
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun KotlinLexer.skipWhiteSpaceAndComments() {
|
||||
while (tokenType in KtTokens.WHITE_SPACE_OR_COMMENT_BIT_SET) {
|
||||
advance()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun topLevelSectionCodeTextTokens(script: CharSequence, sectionIdentifier: String): Sequence<CharSequence> =
|
||||
TopLevelSectionTokensEnumerator(script, sectionIdentifier).asSequence()
|
||||
.filter { it.tokenType !in KtTokens.WHITE_SPACE_OR_COMMENT_BIT_SET }
|
||||
.map { it.tokenSequence }
|
||||
Reference in New Issue
Block a user