-
Notifications
You must be signed in to change notification settings - Fork 160
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
89b7fb6
commit ada323a
Showing
5 changed files
with
306 additions
and
46 deletions.
There are no files selected for viewing
216 changes: 216 additions & 0 deletions
216
plugin/src/main/kotlin/org/jlleitschuh/gradle/ktlint/worker/KtLintInvocation.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,216 @@ | ||
package org.jlleitschuh.gradle.ktlint.worker | ||
|
||
import com.pinterest.ktlint.core.LintError | ||
import com.pinterest.ktlint.core.api.FeatureInAlphaState | ||
import java.io.File | ||
import kotlin.reflect.full.declaredMemberFunctions | ||
import kotlin.reflect.full.findParameterByName | ||
import kotlin.reflect.full.instanceParameter | ||
import kotlin.reflect.full.memberFunctions | ||
import kotlin.reflect.full.memberProperties | ||
import kotlin.reflect.full.primaryConstructor | ||
|
||
internal sealed interface KtLintInvocation { | ||
fun invokeLint(file: File, cb: (LintError, Boolean) -> Unit) | ||
fun invokeFormat(file: File, cb: (LintError, Boolean) -> Unit): String | ||
} | ||
|
||
internal class LegacyParamsInvocation( | ||
var editorConfigPath: String? = null, | ||
var debug: Boolean = false, | ||
var ruleSets: Set<com.pinterest.ktlint.core.RuleSet> = emptySet(), | ||
var userData: Map<String, String> = emptyMap() | ||
) : KtLintInvocation { | ||
private fun buildParams(file: File, cb: (LintError, Boolean) -> Unit): com.pinterest.ktlint.core.KtLint.Params { | ||
val script = !file.name.endsWith(".kt", ignoreCase = true) | ||
return com.pinterest.ktlint.core.KtLint.Params( | ||
fileName = file.absolutePath, | ||
text = file.readText(), | ||
ruleSets = ruleSets, | ||
userData = userData, | ||
debug = debug, | ||
editorConfigPath = editorConfigPath, | ||
script = script, | ||
cb = cb | ||
) | ||
} | ||
|
||
override fun invokeLint(file: File, cb: (LintError, Boolean) -> Unit) { | ||
com.pinterest.ktlint.core.KtLint.lint(buildParams(file, cb)) | ||
} | ||
|
||
override fun invokeFormat(file: File, cb: (LintError, Boolean) -> Unit): String { | ||
return com.pinterest.ktlint.core.KtLint.format(buildParams(file, cb)) | ||
} | ||
|
||
} | ||
|
||
@OptIn(FeatureInAlphaState::class) | ||
internal class ExperimentalParamsInvocation( | ||
var editorConfigPath: String? = null, | ||
var debug: Boolean = false, | ||
var ruleSets: Set<com.pinterest.ktlint.core.RuleSet> = emptySet(), | ||
var userData: Map<String, String> = emptyMap() | ||
) : KtLintInvocation { | ||
|
||
private fun buildParams(file: File, cb: (LintError, Boolean) -> Unit): com.pinterest.ktlint.core.KtLint.ExperimentalParams { | ||
val script = !file.name.endsWith(".kt", ignoreCase = true) | ||
val ctor = Class.forName("com.pinterest.ktlint.core.KtLint\$ExperimentalParams").kotlin.primaryConstructor | ||
val editorConfigOverride = userDataToEditorConfigOverride(userData) | ||
return ctor!!.callBy( | ||
mapOf( | ||
ctor.findParameterByName("fileName")!! to file.absolutePath, | ||
ctor.findParameterByName("text")!! to file.readText(), | ||
ctor.findParameterByName("ruleSets")!! to ruleSets, | ||
ctor.findParameterByName("cb")!! to cb, | ||
ctor.findParameterByName("script")!! to script, | ||
ctor.findParameterByName("editorConfigPath")!! to editorConfigPath, | ||
ctor.findParameterByName("debug")!! to debug, | ||
ctor.findParameterByName("editorConfigOverride")!! to editorConfigOverride | ||
) | ||
) as com.pinterest.ktlint.core.KtLint.ExperimentalParams | ||
} | ||
|
||
override fun invokeLint(file: File, cb: (LintError, Boolean) -> Unit) { | ||
com.pinterest.ktlint.core.KtLint.lint(buildParams(file, cb)) | ||
} | ||
|
||
override fun invokeFormat(file: File, cb: (LintError, Boolean) -> Unit): String { | ||
return com.pinterest.ktlint.core.KtLint.format(buildParams(file, cb)) | ||
} | ||
} | ||
|
||
fun getCodeStyle(styleName: String): Any { | ||
return try { | ||
Class.forName("com.pinterest.ktlint.core.api.DefaultEditorConfigProperties\$CodeStyleValue").getDeclaredField(styleName).get(null) | ||
} catch (e: ClassNotFoundException) { | ||
(Class.forName("com.pinterest.ktlint.core.api.editorconfig.CodeStyleValue").enumConstants as Array<Enum<*>>).first { | ||
it.name == styleName | ||
} | ||
} | ||
} | ||
|
||
fun getEditorConfigPropertyClass(): Class<*> { | ||
return try { | ||
Class.forName("com.pinterest.ktlint.core.api.UsesEditorConfigProperties\$EditorConfigProperty") | ||
} catch (e: ClassNotFoundException) { | ||
Class.forName("com.pinterest.ktlint.core.api.editorconfig.EditorConfigProperty") | ||
} | ||
} | ||
|
||
@OptIn(FeatureInAlphaState::class) | ||
fun userDataToEditorConfigOverride(userData: Map<String, String>, useNewDisabledProp: Boolean = false): Any { | ||
val defaultEditorConfigPropertiesClass = Class.forName("com.pinterest.ktlint.core.api.DefaultEditorConfigProperties") | ||
val defaultEditorConfigProperties = defaultEditorConfigPropertiesClass.kotlin.objectInstance | ||
val codeStyle = getCodeStyle(if (userData["android"]?.toBoolean() == true) "android" else "official") | ||
val editorConfigOverrideClass = Class.forName("com.pinterest.ktlint.core.api.EditorConfigOverride") | ||
val editorConfigOverride = editorConfigOverrideClass.kotlin.primaryConstructor!!.call() | ||
val addMethod = editorConfigOverrideClass.getDeclaredMethod("add", getEditorConfigPropertyClass(), Any::class.java) | ||
addMethod.isAccessible = true | ||
val disabledRulesProperty = defaultEditorConfigPropertiesClass.kotlin.memberProperties.firstOrNull { it.name == "ktlintDisabledRulesProperty" } | ||
?: defaultEditorConfigPropertiesClass.kotlin.memberProperties.first { it.name == "disabledRulesProperty" } | ||
val codeStyleSetProperty = defaultEditorConfigPropertiesClass.kotlin.memberProperties.first { it.name == "codeStyleSetProperty" } | ||
addMethod.invoke(editorConfigOverride, disabledRulesProperty.getter.call(defaultEditorConfigProperties), userData["disabled_rules"]) | ||
addMethod.invoke(editorConfigOverride, codeStyleSetProperty.getter.call(defaultEditorConfigProperties), codeStyle) | ||
return editorConfigOverride | ||
} | ||
|
||
@OptIn(FeatureInAlphaState::class) | ||
internal class ExperimentalParamsProviderInvocation( | ||
var editorConfigPath: String? = null, | ||
var debug: Boolean = false, | ||
var ruleProviders: Set<Any> = emptySet(), | ||
var userData: Map<String, String> = emptyMap() | ||
) : KtLintInvocation { | ||
|
||
private fun buildParams(file: File, cb: (LintError, Boolean) -> Unit): com.pinterest.ktlint.core.KtLint.ExperimentalParams { | ||
val script = !file.name.endsWith(".kt", ignoreCase = true) | ||
val ctor = Class.forName("com.pinterest.ktlint.core.KtLint\$ExperimentalParams").kotlin.primaryConstructor | ||
val editorConfigOverride = userDataToEditorConfigOverride(userData, true) | ||
return ctor!!.callBy( | ||
mapOf( | ||
ctor.findParameterByName("fileName")!! to file.absolutePath, | ||
ctor.findParameterByName("text")!! to file.readText(), | ||
ctor.findParameterByName("ruleProviders")!! to ruleProviders, | ||
ctor.findParameterByName("cb")!! to cb, | ||
ctor.findParameterByName("script")!! to script, | ||
ctor.findParameterByName("editorConfigPath")!! to editorConfigPath, | ||
ctor.findParameterByName("debug")!! to debug, | ||
ctor.findParameterByName("editorConfigOverride")!! to editorConfigOverride | ||
) | ||
) as com.pinterest.ktlint.core.KtLint.ExperimentalParams | ||
} | ||
|
||
override fun invokeLint(file: File, cb: (LintError, Boolean) -> Unit) { | ||
com.pinterest.ktlint.core.KtLint.lint(buildParams(file, cb)) | ||
} | ||
|
||
override fun invokeFormat(file: File, cb: (LintError, Boolean) -> Unit): String { | ||
return com.pinterest.ktlint.core.KtLint.format(buildParams(file, cb)) | ||
} | ||
} | ||
|
||
internal class RuleEngineInvocation : KtLintInvocation { | ||
private lateinit var engine: Any | ||
fun initialize(ruleProviders: Set<Any>, userData: Map<String, String>) { | ||
val engineClass = Class.forName("com.pinterest.ktlint.core.KtLintRuleEngine") | ||
val editorConfigOverride = userDataToEditorConfigOverride(userData, true) | ||
val ctor = engineClass.kotlin.primaryConstructor | ||
engine = ctor!!.callBy( | ||
mapOf( | ||
ctor.findParameterByName("ruleProviders")!! to ruleProviders, | ||
ctor.findParameterByName("editorConfigOverride")!! to editorConfigOverride | ||
) | ||
) | ||
} | ||
|
||
override fun invokeLint(file: File, cb: (LintError, Boolean) -> Unit) { | ||
engine::class.declaredMemberFunctions.forEach { println(it.name + " " + it.parameters.map { it.name }.joinToString(",")) } | ||
val lintMethod = engine::class.memberFunctions.first { | ||
it.name == "lint" | ||
&& it.parameters.map { it.name }.containsAll(setOf("code", "filePath", "callback")) | ||
} | ||
lintMethod.callBy( | ||
mapOf( | ||
lintMethod.instanceParameter!! to engine, | ||
lintMethod.findParameterByName("code")!! to file.readText(), | ||
lintMethod.findParameterByName("filePath")!! to file.absoluteFile.toPath(), | ||
lintMethod.findParameterByName("callback")!! to { le: LintError -> cb.invoke(le, false) } | ||
) | ||
) | ||
} | ||
|
||
override fun invokeFormat(file: File, cb: (LintError, Boolean) -> Unit): String { | ||
val formatMethod = engine::class.memberFunctions.first { | ||
it.name == "format" | ||
&& it.parameters.map { it.name }.containsAll(setOf("code", "filePath", "callback")) | ||
} | ||
return formatMethod.callBy( | ||
mapOf( | ||
formatMethod.instanceParameter!! to engine, | ||
formatMethod.findParameterByName("code")!! to file.readText(), | ||
formatMethod.findParameterByName("filePath")!! to file.absoluteFile.toPath(), | ||
formatMethod.findParameterByName("callback")!! to cb | ||
) | ||
) as String | ||
} | ||
} | ||
|
||
internal fun selectInvocation(): KtLintInvocation { | ||
return try { | ||
KtLintInvocation::class.java.classLoader.loadClass("com.pinterest.ktlint.core.KtLint\$Params") | ||
LegacyParamsInvocation() | ||
} catch (e: Exception) { | ||
try { | ||
Class.forName("com.pinterest.ktlint.core.KtLintRuleEngine") | ||
RuleEngineInvocation() | ||
} catch (e: Exception) { | ||
val ctor = Class.forName("com.pinterest.ktlint.core.KtLint\$ExperimentalParams").kotlin.primaryConstructor | ||
if (ctor?.findParameterByName("ruleProviders") != null) { | ||
ExperimentalParamsProviderInvocation() | ||
} else { | ||
ExperimentalParamsInvocation() | ||
} | ||
} | ||
} | ||
} |
30 changes: 30 additions & 0 deletions
30
plugin/src/main/kotlin/org/jlleitschuh/gradle/ktlint/worker/KtLintRuleLoader.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,30 @@ | ||
package org.jlleitschuh.gradle.ktlint.worker | ||
|
||
import java.util.ServiceLoader | ||
import kotlin.reflect.full.memberProperties | ||
|
||
internal fun loadRuleSetsFromClasspathWithRuleSetProvider(): Map<String, com.pinterest.ktlint.core.RuleSet> { | ||
return ServiceLoader | ||
.load(com.pinterest.ktlint.core.RuleSetProvider::class.java) | ||
.associateBy { | ||
val key = it.get().id | ||
// Adapted from KtLint CLI module | ||
if (key == "standard") "\u0000$key" else key | ||
} | ||
.mapValues { it.value.get() } | ||
} | ||
|
||
internal fun loadRuleSetsFromClasspathWithRuleSetProviderV2(): Map<String, Set<Any>> { | ||
val ruleSetProviderV2Class = Class.forName("com.pinterest.ktlint.core.RuleSetProviderV2") | ||
val idProperty = ruleSetProviderV2Class.kotlin.memberProperties.first{it.name == "id"} | ||
val getRuleProviders = ruleSetProviderV2Class.getDeclaredMethod("getRuleProviders") | ||
return ServiceLoader | ||
.load(ruleSetProviderV2Class) | ||
.associateBy { | ||
val key = idProperty.getter.call(it) as String | ||
// Adapted from KtLint CLI module | ||
if (key == "standard") "\u0000$key" else key | ||
}.mapValues { | ||
getRuleProviders.invoke(it.value) as Set<Any> | ||
} | ||
} |
Oops, something went wrong.