org.jetbrains.kotlin.fir.analysis.diagnostics.SourceElementPositioningStrategy.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation

Kotlin Compiler

There is a newer version: 2.0.0

/*
 * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.fir.analysis.diagnostics

import com.intellij.openapi.util.TextRange
import com.intellij.psi.PsiElement
import org.jetbrains.kotlin.diagnostics.PositioningStrategy
import org.jetbrains.kotlin.fir.FirPsiSourceElement
import org.jetbrains.kotlin.fir.FirSourceElement

open class SourceElementPositioningStrategy(
    val lightTreeStrategy: LightTreePositioningStrategy,
    val psiStrategy: PositioningStrategy
) {
    fun markDiagnostic(diagnostic: FirDiagnostic<*>): List {
        val element = diagnostic.element
        if (element is FirPsiSourceElement<*>) {
            return psiStrategy.mark(element.psi as E)
        }
        return lightTreeStrategy.mark(element.lighterASTNode, element.treeStructure)
    }

    fun isValid(element: FirSourceElement): Boolean {
        if (element is FirPsiSourceElement<*>) {
            return psiStrategy.isValid(element.psi as E)
        }
        return lightTreeStrategy.isValid(element.lighterASTNode, element.treeStructure)
    }

    companion object {
        val DEFAULT: SourceElementPositioningStrategy = SourceElementPositioningStrategies.DEFAULT
    }
}