Move: Fix "whole file" mode detection for scripts

#KT-25857 Fixed
This commit is contained in:
Alexey Sedunov
2018-08-02 21:22:53 +03:00
parent 40a5bf0355
commit 1797a50fc9
2 changed files with 7 additions and 6 deletions
@@ -624,4 +624,6 @@ fun KtModifierKeywordToken.toVisibility(): Visibility {
KtTokens.INTERNAL_KEYWORD -> Visibilities.INTERNAL
else -> throw IllegalArgumentException("Unknown visibility modifier:$this")
}
}
}
fun KtFile.getFileOrScriptDeclarations() = if (isScript()) script!!.declarations else declarations