FakeFileForLightClass tries to use ktFile.originalFile.virtualFile

This commit is contained in:
Nicolay Mitropolsky
2017-10-06 20:00:17 +03:00
parent 703d053157
commit d609579e55
2 changed files with 20 additions and 10 deletions
@@ -33,7 +33,9 @@ open class FakeFileForLightClass(
private val lightClass: () -> KtLightClass,
private val stub: () -> PsiClassHolderFileStub<*>,
private val packageFqName: FqName = ktFile.packageFqName
) : ClsFileImpl(ClassFileViewProvider(ktFile.manager, ktFile.viewProvider.virtualFile)) {
) : ClsFileImpl(ClassFileViewProvider(ktFile.manager, ktFile.virtualFile ?:
ktFile.originalFile.virtualFile ?:
ktFile.viewProvider.virtualFile)) {
override fun getPackageName() = packageFqName.asString()
override fun getStub() = stub()