blob: 7c1166377c24c3c5f780703936cf050e375ae84d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package org.jetbrains.kotlin.idea.structuralsearch
import com.google.common.collect.ImmutableBiMap
import com.intellij.psi.PsiComment
import com.intellij.psi.tree.IElementType
import com.intellij.structuralsearch.impl.matcher.handlers.MatchingHandler
import com.intellij.structuralsearch.impl.matcher.handlers.SubstitutionHandler
import org.jetbrains.kotlin.descriptors.CallableDescriptor
import org.jetbrains.kotlin.descriptors.ClassDescriptor
import org.jetbrains.kotlin.descriptors.annotations.Annotations
import org.jetbrains.kotlin.idea.caches.resolve.resolveToDescriptorIfAny
import org.jetbrains.kotlin.lexer.KtSingleValueToken
import org.jetbrains.kotlin.lexer.KtTokens
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.psi.KtDeclaration
import org.jetbrains.kotlin.renderer.DescriptorRenderer
import org.jetbrains.kotlin.types.KotlinType
import org.jetbrains.kotlin.types.KotlinTypeFactory
import org.jetbrains.kotlin.types.TypeAttributes
import org.jetbrains.kotlin.types.expressions.OperatorConventions
fun getCommentText(comment: PsiComment): String {
return when (comment.tokenType) {
KtTokens.EOL_COMMENT -> comment.text.drop(2).trim()
KtTokens.BLOCK_COMMENT -> comment.text.drop(2).dropLast(2).trim()
else -> ""
}
}
private val BINARY_EXPR_OP_NAMES = ImmutableBiMap.builder<KtSingleValueToken, Name>()
.putAll(OperatorConventions.ASSIGNMENT_OPERATIONS)
.putAll(OperatorConventions.BINARY_OPERATION_NAMES)
.build()
fun IElementType.binaryExprOpName(): Name? = BINARY_EXPR_OP_NAMES[this]
fun KotlinType.renderNames(): Array<String> = arrayOf(
DescriptorRenderer.FQ_NAMES_IN_TYPES.renderType(this),
DescriptorRenderer.SHORT_NAMES_IN_TYPES.renderType(this),
"$this"
)
fun String.removeTypeParameters(): String {
if (!this.contains('<') || !this.contains('>')) return this
return this.removeRange(
this.indexOfFirst { c -> c == '<' },
this.indexOfLast { c -> c == '>' } + 1
)
}
val MatchingHandler.withinHierarchyTextFilterSet: Boolean
get() = this is SubstitutionHandler && (this.isSubtype || this.isStrictSubtype)
fun KtDeclaration.resolveKotlinType(): KotlinType? =
(resolveToDescriptorIfAny() as? CallableDescriptor)?.returnType
fun ClassDescriptor.toSimpleType(nullable: Boolean = false) =
KotlinTypeFactory.simpleType(TypeAttributes.Empty, this.typeConstructor, emptyList(), nullable)
|
