Refactor: rename 'check' helper function with more descriptive name, remove 'var result'
This commit is contained in:
@@ -309,21 +309,19 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
|
|||||||
elementAt.textOffset
|
elementAt.textOffset
|
||||||
}
|
}
|
||||||
|
|
||||||
var result = PsiTreeUtil.findElementOfClassAtOffset(containingFile, lineStartOffset, KtExpression::class.java, false)
|
fun KtElement.takeIfAcceptedAsCodeFragmentContext() = takeIf { KotlinEditorTextProvider.isAcceptedAsCodeFragmentContext(it) }
|
||||||
if (result.check()) {
|
|
||||||
return CodeInsightUtils.getTopmostElementAtOffset(result!!, lineStartOffset, KtExpression::class.java)
|
|
||||||
}
|
|
||||||
|
|
||||||
result = KotlinEditorTextProvider.findExpressionInner(elementAt, true)
|
PsiTreeUtil.findElementOfClassAtOffset(containingFile, lineStartOffset, KtExpression::class.java, false)
|
||||||
if (result.check()) {
|
?.takeIfAcceptedAsCodeFragmentContext()
|
||||||
return result
|
?.let { return CodeInsightUtils.getTopmostElementAtOffset(it, lineStartOffset, KtExpression::class.java) }
|
||||||
}
|
|
||||||
|
KotlinEditorTextProvider.findExpressionInner(elementAt, true)
|
||||||
|
?.takeIfAcceptedAsCodeFragmentContext()
|
||||||
|
?.let { return it }
|
||||||
|
|
||||||
return containingFile
|
return containingFile
|
||||||
}
|
}
|
||||||
|
|
||||||
private fun KtElement?.check(): Boolean = this != null && KotlinEditorTextProvider.isAcceptedAsCodeFragmentContext(this)
|
|
||||||
|
|
||||||
//internal for tests
|
//internal for tests
|
||||||
fun createCodeFragmentForLabeledObjects(project: Project, markupMap: Map<*, ValueMarkup>): Pair<String, Map<String, Value>> {
|
fun createCodeFragmentForLabeledObjects(project: Project, markupMap: Map<*, ValueMarkup>): Pair<String, Map<String, Value>> {
|
||||||
@Suppress("UNCHECKED_CAST")
|
@Suppress("UNCHECKED_CAST")
|
||||||
|
|||||||
Reference in New Issue
Block a user