azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[LL API] Preserve the original 'FirCodeFragment' after in-body change

Browse Source
This commit is contained in:
Yan Zhulanow
2023-08-04 02:22:22 +09:00
committed by Space Team
parent 27a15cd76d
commit 75868d5f52
1 changed files with 2 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/file/structure/FileStructureElement.kt
+2 -20
View File
@@ -147,26 +147,8 @@ internal class ReanalyzableCodeFragmentStructureElement(
override val timestamp = psi.modificationStamp override val timestamp = psi.modificationStamp
override fun reanalyze(): ReanalyzableStructureElement<KtCodeFragment, FirCodeFragmentSymbol> { override fun reanalyze(): ReanalyzableStructureElement<KtCodeFragment, FirCodeFragmentSymbol> {
val originalCodeFragment = firSymbol.fir firSymbol.lazyResolveToPhase(FirResolvePhase.BODY_RESOLVE)
val originalDesignation = originalCodeFragment.collectDesignation() return ReanalyzableCodeFragmentStructureElement(firFile, psi, firSymbol, moduleComponents)
firFile.transformDeclarations(object : FirTransformer<Nothing?>() {
override fun <E : FirElement> transformElement(element: E, data: Nothing?): E = element
override fun transformCodeFragment(codeFragment: FirCodeFragment, data: Nothing?): FirCodeFragment {
return RawFirNonLocalDeclarationBuilder.build(
session = originalCodeFragment.moduleData.session,
scopeProvider = originalCodeFragment.moduleData.session.kotlinScopeProvider,
designation = originalDesignation,
rootNonLocalDeclaration = psi,
) as FirCodeFragment
}
}, null)
val newCodeFragment = firFile.codeFragment
newCodeFragment.lazyResolveToPhase(FirResolvePhase.BODY_RESOLVE)
return ReanalyzableCodeFragmentStructureElement(firFile, psi, newCodeFragment.symbol, moduleComponents)
} }
} }