com.github.shyiko.ktlint.ruleset.standard.ModifierOrderRule.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ktlint-ruleset-standard Show documentation
The newest version!
package com.github.shyiko.ktlint.ruleset.standard

import com.github.shyiko.ktlint.core.Rule
import com.github.shyiko.ktlint.core.ast.ElementType.ABSTRACT_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.ACTUAL_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.ANNOTATION_ENTRY
import com.github.shyiko.ktlint.core.ast.ElementType.ANNOTATION_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.COMPANION_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.CONST_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.DATA_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.ENUM_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.EXPECT_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.EXTERNAL_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.FINAL_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.INFIX_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.INLINE_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.INNER_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.INTERNAL_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.LATEINIT_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.OPEN_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.OPERATOR_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.OVERRIDE_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.PRIVATE_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.PROTECTED_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.PUBLIC_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.SEALED_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.SUSPEND_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.TAILREC_KEYWORD
import com.github.shyiko.ktlint.core.ast.ElementType.VARARG_KEYWORD
import org.jetbrains.kotlin.com.intellij.lang.ASTNode
import org.jetbrains.kotlin.com.intellij.psi.tree.TokenSet
import org.jetbrains.kotlin.psi.KtAnnotationEntry
import org.jetbrains.kotlin.psi.KtDeclarationModifierList
import java.util.Arrays

class ModifierOrderRule : Rule("modifier-order") {

    // subset of ElementType.MODIFIER_KEYWORDS_ARRAY (+ annotations entries)
    private val order = arrayOf(
        ANNOTATION_ENTRY,
        PUBLIC_KEYWORD, PROTECTED_KEYWORD, PRIVATE_KEYWORD, INTERNAL_KEYWORD,
        EXPECT_KEYWORD, ACTUAL_KEYWORD,
        FINAL_KEYWORD, OPEN_KEYWORD, ABSTRACT_KEYWORD, SEALED_KEYWORD, CONST_KEYWORD,
        EXTERNAL_KEYWORD,
        OVERRIDE_KEYWORD,
        LATEINIT_KEYWORD,
        TAILREC_KEYWORD,
        VARARG_KEYWORD,
        SUSPEND_KEYWORD,
        INNER_KEYWORD,
        ENUM_KEYWORD, ANNOTATION_KEYWORD,
        COMPANION_KEYWORD,
        INLINE_KEYWORD,
        INFIX_KEYWORD,
        OPERATOR_KEYWORD,
        DATA_KEYWORD
        // NOINLINE_KEYWORD, CROSSINLINE_KEYWORD, OUT_KEYWORD, IN_KEYWORD, REIFIED_KEYWORD
        // HEADER_KEYWORD, IMPL_KEYWORD
    )
    private val tokenSet = TokenSet.create(*order)

    override fun visit(
        node: ASTNode,
        autoCorrect: Boolean,
        emit: (offset: Int, errorMessage: String, canBeAutoCorrected: Boolean) -> Unit
    ) {
        if (node.psi is KtDeclarationModifierList) {
            val modifierArr = node.getChildren(tokenSet)
            val sorted = modifierArr.copyOf().apply { sortWith(compareBy { order.indexOf(it.elementType) }) }
            if (!Arrays.equals(modifierArr, sorted)) {
                // Since annotations can be fairly lengthy and/or span multiple lines we are
                // squashing them into a single placeholder text to guarantee a single line output
                emit(
                    node.startOffset,
                    "Incorrect modifier order (should be \"${
                    squashAnnotations(sorted).joinToString(" ")
                    }\")",
                    true
                )
                if (autoCorrect) {
                    modifierArr.forEachIndexed { i, n ->
                        node.replaceChild(n, sorted[i].clone() as ASTNode)
                    }
                }
            }
        }
    }

    private fun squashAnnotations(sorted: Array): List {
        val nonAnnotationModifiers = sorted.filter { it.psi !is KtAnnotationEntry }
        return if (nonAnnotationModifiers.size != sorted.size) {
            listOf("@Annotation...") + nonAnnotationModifiers.map { it.text }
        } else {
            nonAnnotationModifiers.map { it.text }
        }
    }
}