From 19c52844677cf2e2647cc073225d0afcd000484d Mon Sep 17 00:00:00 2001 From: Vlad Koshkin Date: Tue, 30 Jul 2024 14:39:25 +0200 Subject: [PATCH] [kotlin] k2: extracted main.kts constant GitOrigin-RevId: b6ef8a4eefcaf5b195b09135a65b3de50ee17233 --- .../idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt index 3b37d0be03a2..60e6c5efed34 100644 --- a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt +++ b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/script/k2/utils.kt @@ -9,7 +9,9 @@ import com.intellij.psi.SyntaxTraverser import com.intellij.util.containers.TreeTraversal import org.jetbrains.kotlin.lexer.KtTokens.SHEBANG_COMMENT -fun isMainKtsScript(virtualFile: VirtualFile) = virtualFile.name.endsWith(".main.kts") +private const val MAIN_KTS = "main.kts" + +fun isMainKtsScript(virtualFile: VirtualFile) = virtualFile.name == MAIN_KTS || virtualFile.name.endsWith(".$MAIN_KTS") fun PsiFile.hasShebangComment(): Boolean = SyntaxTraverser.psiTraverser(/* root = */ this)