Minor: Do not place line markers in code fragments and in non-Kotlin files

2018-05-31 00:32:14 +03:00
parent ab5614eaac
commit e334f693b2
2 changed files with 12 additions and 0 deletions
@@ -55,6 +55,12 @@ class KotlinSuspendCallLineMarkerProvider : LineMarkerProvider {
            ProgressManager.checkCanceled()

            if (element !is KtExpression) continue
+
+            val containingFile = element.containingFile
+            if (containingFile !is KtFile || containingFile is KtCodeFragment) {
+                continue
+            }
+
            val lineNumber = element.getLineNumber()
            if (lineNumber in markedLineNumbers) continue
            if (!element.hasSuspendCalls()) continue
@@ -56,6 +56,12 @@ class KotlinSuspendCallLineMarkerProvider : LineMarkerProvider {
            ProgressManager.checkCanceled()

            if (element !is KtExpression) continue
+
+            val containingFile = element.containingFile
+            if (containingFile !is KtFile || containingFile is KtCodeFragment) {
+                continue
+            }
+
            val lineNumber = element.getLineNumber()
            if (lineNumber in markedLineNumbers) continue
            if (!element.hasSuspendCalls()) continue