Skylot
68 lines
2.5 KiB
Kotlin
68 lines
2.5 KiB
Kotlin
package jadx.plugins.script.ide
|
|
|
|
import jadx.plugins.script.ScriptEval
|
|
import kotlinx.coroutines.runBlocking
|
|
import org.jetbrains.kotlin.scripting.ide_services.compiler.KJvmReplCompilerWithIdeServices
|
|
import kotlin.script.experimental.api.ReplAnalyzerResult
|
|
import kotlin.script.experimental.api.ScriptCompilationConfiguration
|
|
import kotlin.script.experimental.api.ScriptDiagnostic
|
|
import kotlin.script.experimental.api.SourceCode
|
|
import kotlin.script.experimental.api.SourceCodeCompletionVariant
|
|
import kotlin.script.experimental.api.analysisDiagnostics
|
|
import kotlin.script.experimental.api.hostConfiguration
|
|
import kotlin.script.experimental.api.renderedResultType
|
|
import kotlin.script.experimental.api.valueOrNull
|
|
import kotlin.script.experimental.host.toScriptSource
|
|
import kotlin.script.experimental.jvm.defaultJvmScriptingHostConfiguration
|
|
import kotlin.script.experimental.jvm.util.isError
|
|
import kotlin.script.experimental.jvm.util.toSourceCodePosition
|
|
|
|
const val AUTO_COMPLETE_INSERT_STR = "ABCDEF" // defined at KJvmReplCompleter.INSERTED_STRING
|
|
|
|
data class ScriptCompletionResult(
|
|
val completions: List<SourceCodeCompletionVariant>,
|
|
val reports: List<ScriptDiagnostic>,
|
|
)
|
|
|
|
data class ScriptAnalyzeResult(
|
|
val success: Boolean,
|
|
val issues: List<ScriptDiagnostic>,
|
|
val renderType: String?,
|
|
)
|
|
|
|
class ScriptServices {
|
|
private val compileConf = ScriptEval.compileConf
|
|
private val replCompiler = KJvmReplCompilerWithIdeServices(
|
|
compileConf[ScriptCompilationConfiguration.hostConfiguration]
|
|
?: defaultJvmScriptingHostConfiguration,
|
|
)
|
|
|
|
fun complete(scriptName: String, code: String, cursor: Int): ScriptCompletionResult {
|
|
val snippet = code.toScriptSource(scriptName)
|
|
val result = runBlocking {
|
|
replCompiler.complete(snippet, cursor.toSourceCodePosition(snippet), compileConf)
|
|
}
|
|
return ScriptCompletionResult(
|
|
completions = result.valueOrNull()?.toList() ?: emptyList(),
|
|
reports = result.reports,
|
|
)
|
|
}
|
|
|
|
fun analyze(scriptName: String, code: String): ScriptAnalyzeResult {
|
|
val sourceCode = code.toScriptSource(scriptName)
|
|
val result = runBlocking {
|
|
val cursor = SourceCode.Position(0, 0) // not used
|
|
replCompiler.analyze(sourceCode, cursor, compileConf)
|
|
}
|
|
val analyzerResult = result.valueOrNull()
|
|
val issues = mutableListOf<ScriptDiagnostic>()
|
|
analyzerResult?.get(ReplAnalyzerResult.analysisDiagnostics)?.let(issues::addAll)
|
|
issues.addAll(result.reports)
|
|
return ScriptAnalyzeResult(
|
|
success = !result.isError(),
|
|
issues = issues,
|
|
renderType = analyzerResult?.get(ReplAnalyzerResult.renderedResultType),
|
|
)
|
|
}
|
|
}
|